def goSimpleTest(): nt,ni = 481,2 # number of time samples; number of impulses freq,decay = 0.08,0.05 # peak frequency and decay for wavelet #na,ka = 11,-5 # sampling for inverse wavelet A na,ka = 81,-20 # sampling for inverse wavelet A nh,kh = 181,-90 # sampling for wavelet H dt,ft = 0.004,0.000 # used for plotting only tmin,tmax = 0,nt-1 sfac = 1.00 wha = 10.0 st = Sampling(nt,dt,ft) for mp in [True,False]: # True, for minimum-phase; False for other hk = getWavelet(freq,decay,nh,kh,mp) # known wavelet for r in [0.5,2.0]: # 0.5 for stretch; 2.0 for squeeze aw = zerofloat(na); aw[-ka] = 1.0 hw = zerofloat(nh); hw[-kh] = 1.0 p,q = makeImpulses(r,nt,ni) f = addWavelet(freq,decay,p,mp) g = addWavelet(freq,decay,q,mp) u = rampfloat(0.0,r,nt) if r<=1.0: u = add((1.0-r)*(nt-1),u) ww = WaveletWarpingHA() ww.setTimeRange(tmin,tmax) ww.setStabilityFactor(sfac) ww.setWeightHA(wha) ak = ww.getWaveletH(nh,kh,hk,na,ka) # known inverse wavelet #dump(ak) for iter in range(500): aw = ww.getInverseA(na,ka,nh,kh,hw,u,f,g) # estimated inverse hw = ww.getWaveletH(nh,kh,na,ka,aw,u,f,g) # estimated wavelet #dump(aw) #hw = ww.getWaveletH(na,ka,aw,nh,kh) # estimated wavelet sg = ww.applyS(u,g) ag = ww.applyA(na,ka,aw,g) lag = ww.applyL(u,ag) # lowpass, if squeezing slag = ww.applyS(u,lag) hslag = ww.applyH(nh,kh,hw,slag) normalizeMax(hw) normalizeMax(hk) title = "r = "+str(r) #plotSequences(st,[f,g],labels=["f","g"],title=title) #plotSequences(st,[f,sg],labels=["f","Sg"],title=title) plotSequences(st,[f,g],labels=["f","g"],title=title) plotSequences(st,[f,hslag],labels=["f","HSLAg"],title=title) plotWavelets(Sampling(nh,dt,kh*dt),[nhw,nhk],title=title)