CoupledBunchAnalysis/myfilemanager.py at master · PyCOMPLETE/CoupledBunchAnalysis · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
import numpy as np


class obj_from_dict:
    def __init__(self, dictto):
        for kk in dictto.keys():
            exec 'self.'+kk +'= dictto[kk]'


def obj_to_dict(obj):
    dict_out={}
    members = dir(obj)
    for member in members:
        exec "dict_out['%s'] = obj.%s"%(member,member)
    return dict_out

def myloadmat(filename, squeeze = True):
    import scipy.io as sio
    dict_var=sio.loadmat(filename)
    if squeeze:
        for kk in dict_var.keys():
            try:
                dict_var[kk]=np.squeeze(dict_var[kk])
            except:
                pass
    return dict_var


def myloadmat_to_obj(filename, squeeze = True):
    return  obj_from_dict(myloadmat(filename, squeeze=squeeze))


def dict_of_arrays_and_scalar_from_h5(filename):
    import h5py
    with h5py.File(filename) as fid:
        f_dict = {}
        for kk in fid.keys():
            f_dict[kk] = np.array(fid[kk]).copy()
            if f_dict[kk].shape == ():
                f_dict[kk] = f_dict[kk].tolist()
    return  f_dict

def object_with_arrays_and_scalar_from_h5(filename):
    return  obj_from_dict(dict_of_arrays_and_scalar_from_h5(filename))

def bunchh5_to_dict(filename):
    import h5py
    with h5py.File(filename, 'r') as bunch_ev:
        bunch = bunch_ev['Bunch']
        bunch_dict = {}
        for kk in bunch.keys():
            bunch_dict[kk] = np.array(bunch[kk]).copy()

    return bunch_dict

def bunchh5_to_obj(filename):
    return  obj_from_dict(bunchh5_to_dict(filename))

def bunchh5list_to_dict(filename_list, permissive=False):
    bunch_dict = bunchh5_to_dict(filename_list[0])
    for i_file in xrange(1,len(filename_list)):
        print('Loading '+filename_list[i_file])
        try:
            bunch_dict_curr = bunchh5_to_dict(filename_list[i_file])
            for kk in bunch_dict.keys():
                bunch_dict[kk] = np.array(list(bunch_dict[kk])+list(bunch_dict_curr[kk]))
        except IOError as err:
            print('Got:')
            print(err)
            if not permissive:
                raise err

    return bunch_dict

def bunchh5list_to_obj(filename_list):
    return  obj_from_dict(bunchh5list_to_dict(filename_list))