Reppy/loadmat2trace.py at master · LuigiaCristiano/Reppy · 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
#!/usr/bin/env python
################################
### 130816 (CW): cleaned a bit and minor bugfixes (endtime is read-only)
#
#########################in the previous version are present more entries in the file header than allowed by some obspy routines handling stream object and traces########
from scipy.io import loadmat
from obspy.core import AttribDict, UTCDateTime, Trace,Stream

def loadmat2trace(infile, debug=False):
        '''
        wrapper function for scipy.io.loadmat
        reformats seismogram read from .mat file to ObsPy Trace object
        including all available tr.stats entries
        '''
        ## conversion libraries
        tofloat = ['azimuth', 'calib', 'delta', 'sampling_rate']
        toUTCDateTime = ['starttime']
        toremove=['__header__','__globals__', '__version__', 'endtime'] # endtime is read-only
        st = loadmat(infile, squeeze_me=True)

        if debug:
                print "1) after loadmat"
                print st

        # remove unwanted keys
        for i in st.keys():
                if i in toremove: del st[i]

        # remove Unicode from strings
        for i in st.keys():
                if i[:2] != "__" and st[i].dtype.kind == "U":  st[i] = st[i].item().encode()
        if debug:
                print ""
                print "2) after unicode removal"
                print st

        # convert strings to AttribDicts, floats, UTCDateTime
        for i in st.keys():
                if st[i][:10] == "AttribDict" or i in ['processing']: # AttribDict or list
                        st[i] = eval(st[i])
                elif i in tofloat:
                        st[i] = float(st[i])
                elif i in toUTCDateTime:
                        st[i] = UTCDateTime(st[i])
                elif i == 'location' and len(st[i]) == 0: # location == ''
                        st[i] = ''

        if debug:
                print ""
                print "3) after data type assignment"
                print st

        ## create Trace
        tr = Trace(data=st['data']) # adds default header, starttime 1970-1-1-0:0:0, sampling rate 1
        for i in st.keys():
                if i == 'data': continue
                tr.stats[i] = st[i]

	if debug:
                print ""
                print "4) new trace"
                print tr.stats
                print tr.data

        return tr #, st :: is that needed?

if __name__ == '__main__':

        #tr = loadmat2trace(infile, debug=True)
        tr = loadmat2trace('/media/Elements/data/2009/200909301016/GR.GRA1.BHZ_VEL_2009-09-30.10-16-08_9600.mat')
        print tr
        print tr.stats