from src.dictionary       import generateDictionary
from data.signalGenerator import generateTestSignal , simpleValues , advancedValues , masterValues
from src.processing       import calculateMP
from src.drawing          import calculateTFMap

import matplotlib.pyplot as plt
import numpy             as np

from matplotlib import gridspec, ticker
# from scipy.io   import savemat, loadmat


if __name__ == '__main__':
# create a synthetic signal
	
	(gaborParams , sinusParams , asymetricParams , rectParams , noiseRatio , samplingFrequency , numberOfSamples) = simpleValues()
	(signal,time) = generateTestSignal(gaborParams,sinusParams,asymetricParams,rectParams,numberOfSamples,samplingFrequency,noiseRatio)

	# signal.astype('float32')
	# signal.tofile('dupa.raw')

	# (gaborParams , sinusParams , asymetricParams , rectParams , noiseRatio , samplingFrequency , numberOfSamples) = masterValues()
	# (signal,time) = generateTestSignal(gaborParams,sinusParams,asymetricParams,rectParams,numberOfSamples,samplingFrequency,noiseRatio)

# quick saving to matlab
	# tmp = {}
	# tmp['signal'] = signal
	# tmp['czas']   = time
	# savemat('../syg_asym.mat' , tmp)

# config for a dictionary and MP
예제 #2
0
e-mail: [email protected]
University of Warsaw, July 06, 2015
'''

import numpy as np
import unittest

from src.dictionary import generateDictionary
from data.signalGenerator import generateTestSignal, simpleValues, advancedValues, masterValues
from src.processing import calculateMP

freqThreshold = 0.05
amplThreshold = 0.10

(gaborParams, sinusParams, asymetricParams, rectParams, noiseRatio,
 samplingFrequency, numberOfSamples) = simpleValues()
(signal, time) = generateTestSignal(gaborParams, sinusParams, asymetricParams,
                                    rectParams, numberOfSamples,
                                    samplingFrequency, noiseRatio)

(gaborParamsAdvanced, sinusParamsAdvanced, asymetricParamsAdvanced,
 rectParamsAdvanced, noiseRatio, samplingFrequency,
 numberOfSamples) = advancedValues()
(advancedSignal,
 time) = generateTestSignal(gaborParamsAdvanced, sinusParamsAdvanced,
                            asymetricParamsAdvanced, rectParamsAdvanced,
                            numberOfSamples, samplingFrequency, noiseRatio)

(gaborParamsMaster, sinusParamsMaster, asymetricParamsMaster, rectParamsMaster,
 noiseRatio, samplingFrequency, numberOfSamples) = masterValues()
(masterSignal, time) = generateTestSignal(gaborParamsMaster, sinusParamsMaster,