grided_frequency_search.py 1.62 KB
 Gregory Ashton committed Nov 15, 2016 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 import pyfstat import numpy as np import matplotlib.pyplot as plt F0 = 30.0 F1 = 0 F2 = 0 Alpha = 1.0 Delta = 1.5 # Properties of the GW data sqrtSX = 1e-23 tstart = 1000000000 duration = 100*86400 tend = tstart+duration tref = .5*(tstart+tend) depth = 70 data_label = 'grided_frequency_depth_{:1.0f}'.format(depth) h0 = sqrtSX / depth data = pyfstat.Writer( label=data_label, outdir='data', tref=tref, tstart=tstart, F0=F0, F1=F1, F2=F2, duration=duration, Alpha=Alpha, Delta=Delta, h0=h0, sqrtSX=sqrtSX) data.make_data() m = 0.001 dF0 = np.sqrt(12*m)/(np.pi*duration) DeltaF0 = 800*dF0 F0s = [F0-DeltaF0/2., F0+DeltaF0/2., dF0] F1s = [F1] F2s = [F2] Alphas = [Alpha] Deltas = [Delta] search = pyfstat.GridSearch( 'grided_frequency_search', 'data', 'data/*'+data_label+'*sft', F0s, F1s, F2s, Alphas, Deltas, tref, tstart, tend) search.run() fig, ax = plt.subplots() xidx = search.keys.index('F0') frequencies = np.unique(search.data[:, xidx]) twoF = search.data[:, -1] #mismatch = np.sign(x-F0)*(duration * np.pi * (x - F0))**2 / 12.0 ax.plot(frequencies, twoF, 'k', lw=1) DeltaF = frequencies - F0 sinc = np.sin(np.pi*DeltaF*duration)/(np.pi*DeltaF*duration) A = np.abs((np.max(twoF)-4)*sinc**2 + 4) ax.plot(frequencies, A, '-r', lw=1) ax.set_ylabel('$\widetilde{2\mathcal{F}}$') ax.set_xlabel('Frequency') ax.set_xlim(F0s[0], F0s[1]) dF0 = np.sqrt(12*1)/(np.pi*duration) xticks = [F0-10*dF0, F0, F0+10*dF0] ax.set_xticks(xticks) xticklabels = ['$f_0 {-} 10\Delta f$', '$f_0$', '$f_0 {+} 10\Delta f$'] ax.set_xticklabels(xticklabels) plt.tight_layout() fig.savefig('{}/{}_1D.png'.format(search.outdir, search.label), dpi=300)