From c4badcf8416da277785583032ef69ce420654f80 Mon Sep 17 00:00:00 2001
From: David Keitel <david.keitel@ligo.org>
Date: Tue, 9 Jan 2018 18:29:05 +0000
Subject: [PATCH] TransientGridSearch: new F_mn writer function matching the
format of lalpulsar.write_transientFstatMap_to_fp
---
pyfstat/grid_based_searches.py | 11 ++++++++++-
1 file changed, 10 insertions(+), 1 deletion(-)
diff --git a/pyfstat/grid_based_searches.py b/pyfstat/grid_based_searches.py
index feeb97a..29d590f 100644
--- a/pyfstat/grid_based_searches.py
+++ b/pyfstat/grid_based_searches.py
@@ -449,7 +449,7 @@ class TransientGridSearch(GridSearch):
lalpulsar.write_transientFstatMap_to_fp ( fo, FstatMap, windowRange, None )
del fo # instead of lal.FileClose() which is not SWIG-exported
else:
- np.savetxt(tCWfile, 2.0*F_mn, delimiter=' ')
+ self.write_F_mn ( tCWfile, F_mn, windowRange)
maxidx = np.unravel_index(F_mn.argmax(), F_mn.shape)
thisCand += [windowRange.t0+maxidx[0]*windowRange.dt0,
windowRange.tau+maxidx[1]*windowRange.dtau]
@@ -464,6 +464,15 @@ class TransientGridSearch(GridSearch):
self.save_array_to_disk(data)
self.data = data
+ def write_F_mn (self, tCWfile, F_mn, windowRange ):
+ with open(tCWfile, 'w') as tfp:
+ tfp.write('# t0 [s] tau [s] 2F\n')
+ for m, F_m in enumerate(F_mn):
+ this_t0 = windowRange.t0 + m * windowRange.dt0
+ for n, this_F in enumerate(F_m):
+ this_tau = windowRange.tau + n * windowRange.dtau;
+ tfp.write(' %10d %10d %- 11.8g\n' % (this_t0, this_tau, 2.0*this_F))
+
class SliceGridSearch(GridSearch):
""" Slice gridded search using ComputeFstat """
--
GitLab