Commit 0fabad94 authored by Gregory Ashton's avatar Gregory Ashton
Browse files

Adds functionality to compute and plot the cumulative twoF

parent c6b4c6b8
...@@ -261,10 +261,12 @@ class ComputeFstat(object): ...@@ -261,10 +261,12 @@ class ComputeFstat(object):
self.sftfilepath)) self.sftfilepath))
SFTCatalog = lalpulsar.SFTdataFind(self.sftfilepath, constraints) SFTCatalog = lalpulsar.SFTdataFind(self.sftfilepath, constraints)
names = list(set([ for d in])) names = list(set([ for d in]))
epochs = [d.header.epoch for d in] SFT_timestamps = [d.header.epoch for d in]
'Loaded {} data files from detectors {} spanning {} to {}'.format( 'Loaded {} data files from detectors {} spanning {} to {}'.format(
len(epochs), names, int(epochs[0]), int(epochs[-1]))) len(SFT_timestamps), names, int(SFT_timestamps[0]),
self.SFT_timestamps = SFT_timestamps'Initialising ephems')'Initialising ephems')
ephems = lalpulsar.InitBarycenter(self.earth_ephem, self.sun_ephem) ephems = lalpulsar.InitBarycenter(self.earth_ephem, self.sun_ephem)
...@@ -408,6 +410,37 @@ class ComputeFstat(object): ...@@ -408,6 +410,37 @@ class ComputeFstat(object):
else: else:
return self.BSGL_PREFACTOR * BSGL/np.log10(np.exp(1)) return self.BSGL_PREFACTOR * BSGL/np.log10(np.exp(1))
def calculate_twoF_cumulative(self, F0, F1, F2, Alpha, Delta, asini=None,
period=None, ecc=None, tp=None, argp=None,
tstart=None, tend=None, Npoints=1000,
""" Calculate the cumulative twoF along the obseration span """
duration = tend - tstart
taus = np.linspace(minfraction*duration, duration, Npoints)
twoFs = []
for tau in taus:
tstart=tstart, tend=tstart+tau, F0=F0, F1=F1, F2=F2,
Alpha=Alpha, Delta=Delta, asini=asini, period=period, ecc=ecc,
tp=tp, argp=argp))
return taus, np.array(twoFs)
def plot_twoF_cumulative(self, label, outdir, ax=None, c='k', savefig=True,
taus, twoFs = self.calculate_twoF_cumulative(**kwargs)
if ax is None:
fig, ax = plt.subplots()
ax.plot(taus/86400., twoFs, label=label, color=c)
ax.set_xlabel(r'Days from $t_{{\rm start}}={:.0f}$'.format(
ax.set_ylabel(r'$\widetilde{2\mathcal{F}}_{\rm cumulative}$')
ax.set_xlim(0, taus[-1]/86400)
if savefig:
plt.savefig('{}/{}_twoFcumulative.png'.format(outdir, label))
return ax
class SemiCoherentGlitchSearch(BaseSearchClass, ComputeFstat): class SemiCoherentGlitchSearch(BaseSearchClass, ComputeFstat):
""" A semi-coherent glitch search """ A semi-coherent glitch search
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment