diff --git a/pyfstat/core.py b/pyfstat/core.py index d2fb2a1e6e52369cb6de5d9cd08ff33c45681557..3ed1e92d31ce610221e228f1f8ca87981b43de8a 100755 --- a/pyfstat/core.py +++ b/pyfstat/core.py @@ -384,6 +384,7 @@ class ComputeFstat(BaseSearchClass): SSBprec=None, tCWFstatMapVersion="lal", cudaDeviceName=None, + computeAtoms=False, ): """ Parameters @@ -405,6 +406,7 @@ class ComputeFstat(BaseSearchClass): allow for the Fstat to be computed over a transient range. ('none' instead of None explicitly calls the transient-window function, but with the full range, for debugging) + (if not None, will also force atoms regardless of computeAtoms option) t0Band, tauBand: int if >0, search t0 in (minStartTime,minStartTime+t0Band) and tau in (tauMin,2*Tsft+tauBand). @@ -439,6 +441,8 @@ class ComputeFstat(BaseSearchClass): 'pycuda' for gpu, and some others for devel/debug. cudaDeviceName: str GPU name to be matched against drv.Device output. + computeAtoms: bool + request atoms calculations regardless of transientWindowType """ @@ -550,7 +554,7 @@ class ComputeFstat(BaseSearchClass): logging.info("Initialising FstatInput") dFreq = 0 - if self.transientWindowType: + if self.transientWindowType or self.computeAtoms: self.whatToCompute = lalpulsar.FSTATQ_ATOMS_PER_DET else: self.whatToCompute = lalpulsar.FSTATQ_2F diff --git a/pyfstat/grid_based_searches.py b/pyfstat/grid_based_searches.py index cf6a98262dad94a8a14c3d4e5a53d73f9bfd3ffb..7251013a87f894809e630a07a8a9059a6722e754 100644 --- a/pyfstat/grid_based_searches.py +++ b/pyfstat/grid_based_searches.py @@ -574,6 +574,7 @@ class TransientGridSearch(GridSearch): assumeSqrtSX=self.assumeSqrtSX, tCWFstatMapVersion=self.tCWFstatMapVersion, cudaDeviceName=self.cudaDeviceName, + computeAtoms=self.outputAtoms, ) self.search.get_det_stat = self.search.get_fullycoherent_twoF