Source code for particlespy.particle_load

# -*- coding: utf-8 -*-
"""
Created on Wed Nov 28 17:11:22 2018

@author: qzo13262
"""

import h5py
from particlespy.ptcl_class import particle, particle_list
import hyperspy as hs
import numpy as np

def load_plist(filename):
    f = h5py.File(filename,'r')
    p_list = particle_list()
    
    for p_name in list(f.keys()):
        if p_name[:8] == 'Particle':
            p_group = f[p_name]
            p = particle()
            
            for attr in p_group.attrs:
                if "units" not in attr:
                    if attr+' units' in p_group.attrs:
                        units = p_group.attrs[attr+' units']
                    else:
                        units = None
                    p.set_property(attr, p_group.attrs[attr], units)
            
            p.set_mask(np.array(p_group['Mask'][:]))
            
            if "Image" in p_group:
                p.store_im(hs.signals.Signal2D(np.array(p_group['Image'][:])))
                
            p_list.append(p)
    
    f.close()
    return(p_list)