diff --git a/examples/transient_examples/short_transient_search_gridded.py b/examples/transient_examples/short_transient_search_gridded.py index ceea5e26a6251efb935b51d06fef1f51365f9e4f..fd9b83b2f85d9e076231042efbe1c0a8b9b495a5 100644 --- a/examples/transient_examples/short_transient_search_gridded.py +++ b/examples/transient_examples/short_transient_search_gridded.py @@ -1,9 +1,12 @@ #!/usr/bin/env python import pyfstat +import os import numpy as np import matplotlib.pyplot as plt +datadir = 'data_s' + F0 = 30.0 F1 = -1e-10 F2 = 0 @@ -26,17 +29,29 @@ F2s = [F2] Alphas = [Alpha] Deltas = [Delta] -search = pyfstat.GridSearch( - label='grid_search', outdir='data_s', - sftfilepattern='data_s/*simulated_transient_signal*sft', +print('Standard CW search:') +search1 = pyfstat.GridSearch( + label='CW', outdir=datadir, + sftfilepattern=os.path.join(datadir,'*simulated_transient_signal*sft'), F0s=F0s, F1s=F1s, F2s=F2s, Alphas=Alphas, Deltas=Deltas, tref=tref, minStartTime=minStartTime, maxStartTime=maxStartTime, -# transientWindowType='rect', t0Band=Tspan-2*Tsft, tauBand=Tspan, BSGL=False) -search.run() -search.print_max_twoF() +search1.run() +search1.print_max_twoF() -search.plot_1D(xkey='F0', +search1.plot_1D(xkey='F0', xlabel='freq [Hz]', ylabel='$2\mathcal{F}$') -search.plot_2D(xkey='F0', ykey='F1') +print('with t0,tau bands:') +search2 = pyfstat.GridSearch( + label='tCW', outdir=datadir, + sftfilepattern=os.path.join(datadir,'*simulated_transient_signal*sft'), + F0s=F0s, F1s=F1s, F2s=F2s, Alphas=Alphas, Deltas=Deltas, tref=tref, + minStartTime=minStartTime, maxStartTime=maxStartTime, + transientWindowType='rect', t0Band=Tspan-2*Tsft, tauBand=Tspan, + BSGL=False) +search2.run() +search2.print_max_twoF() + +search2.plot_1D(xkey='F0', + xlabel='freq [Hz]', ylabel='$2\mathcal{F}$')