Commit ca31578f authored by David Keitel's avatar David Keitel
Browse files

ComputeFstat: extra flag computeAtoms independent of transientWindowType

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