pasf/demo_PES.py at master · khabbazian/pasf · 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
import time
import numpy as np
import matplotlib.pyplot as plt
import matlab.engine

eng = matlab.engine.start_matlab()

data = eng.propagating_energy_sources();

dirname   = 'PES-rn'
bfilename = dirname + '/pes'
eng.mkdir( dirname )

myOpts = eng.struct( 'cmethod', 'phase',
        'spans', 21, 'detrend', 2,
        'errorRate', 0.1,
        'boolParfor', 0,
        'saveData', 0, 'boolUseSavedData', 0,
        'bfilename', bfilename);


eng.tic;
Components, Clusters, ClusterInfo, SDFInfo = eng.pasf(data, 4, myOpts, nargout=4)
eng.toc;

Components = np.array(Components._data).reshape(Components.size, order='F')
print( Components.shape )
d1, d2, d3, d4 = Components.shape

for i in range(d3):
    fig = Components[:,:,i,0]
    for c in range(1, d4):
        fig = np.hstack((fig, Components[:,:,i,c]))

    plt.imshow(fig)
    plt.show(block=False)
    time.sleep(0.5)
    plt.close()