Commit ca31578f authored by David Keitel's avatar David Keitel

ComputeFstat: extra flag computeAtoms independent of transientWindowType

parent 24b8f256
Pipeline #9566 passed with stages
in 3 minutes and 43 seconds
......@@ -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
......
......@@ -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
......
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