Skip to content
Snippets Groups Projects
Commit 3260a519 authored by Chad Hanna's avatar Chad Hanna
Browse files

gstlal_s5_pbh_trigger_pipe: expose coinciddence threshold option at the dag level

parent 23d4537a
No related branches found
No related tags found
No related merge requests found
......@@ -235,7 +235,7 @@ class gstlal_inspiral_node(InspiralNode):
A gstlal_inspiral node
"""
#FIXME add veto segments name
def __init__(self, job, dag, frame_cache, frame_segments_file, frame_segments_name, gps_start_time, gps_end_time, channel_dict, reference_psd, svd_bank, tmp_space=inspiral_pipe.log_path(), ht_gate_thresh=20.0, injections=None, control_peak_time = 8, vetoes=None, time_slide_file=None, fir_stride = 8, instruments = "H1H2L1", number = 1, p_node=[]):
def __init__(self, job, dag, frame_cache, frame_segments_file, frame_segments_name, gps_start_time, gps_end_time, channel_dict, reference_psd, svd_bank, tmp_space=inspiral_pipe.log_path(), ht_gate_thresh=20.0, injections=None, control_peak_time = 8, coincidence_threshold = 0.020, vetoes=None, time_slide_file=None, fir_stride = 8, instruments = "H1H2L1", number = 1, p_node=[]):
InspiralNode.__init__(self, job, dag, p_node)
if time_slide_file is not None:
......@@ -251,6 +251,7 @@ class gstlal_inspiral_node(InspiralNode):
self.add_var_opt("tmp-space", tmp_space)
self.add_var_opt("track-psd", "")
self.add_var_opt("control-peak-time", control_peak_time)
self.add_var_opt("coincidence-threshold", coincidence_threshold)
self.add_var_opt("fir-stride", fir_stride)
self.injections = injections
if self.injections is not None:
......@@ -381,6 +382,7 @@ def parse_command_line():
parser.add_option("--web-dir", metavar = "directory", help = "Set the web directory like /home/USER/public_html")
parser.add_option("--fir-stride", type="int", metavar = "secs", default = 8, help = "Set the duration of the fft output blocks, default 8")
parser.add_option("--control-peak-time", type="int", default = 8, metavar = "secs", help = "Set the peak finding time for the control signal, default 8")
parser.add_option("--coincidence-threshold", metavar = "value", type = "float", default = 0.020, help = "Set the coincidence window in seconds (default = 0.020). The light-travel time between instruments will be added automatically in the coincidence test.")
parser.add_option("--do-iir-pipeline", action="store_true", help = "run the iir pipeline instead of lloid")
parser.add_option("--num-banks", type="int", metavar = "banks", default = 1, help = "Set the number of template banks per job, default 1")
parser.add_option("--verbose", action = "store_true", help = "Be verbose")
......@@ -465,7 +467,7 @@ for i, (s, trials_factor) in enumerate(inspiral_pipe.build_bank_string(bank_cach
# inspiral jobs by segment
for seg in segs:
noninjnode = gstlal_inspiral_node(gstlalInspiralJob, dag, options.frame_cache, options.frame_segments_file, options.frame_segments_name, seg[0].seconds, seg[1].seconds, channel_dict, reference_psd=options.reference_psd, svd_bank=s, injections=None, vetoes=options.vetoes, time_slide_file=options.time_slide_file, control_peak_time = options.control_peak_time, fir_stride = options.fir_stride, number = i)
noninjnode = gstlal_inspiral_node(gstlalInspiralJob, dag, options.frame_cache, options.frame_segments_file, options.frame_segments_name, seg[0].seconds, seg[1].seconds, channel_dict, reference_psd=options.reference_psd, svd_bank=s, injections=None, vetoes=options.vetoes, time_slide_file=options.time_slide_file, control_peak_time = options.control_peak_time, fir_stride = options.fir_stride, coincidence_threshold=options.coincidence_threshold, number = i)
inspiral_nodes.append((None, noninjnode))
......@@ -476,7 +478,7 @@ for i, (s, trials_factor) in enumerate(inspiral_pipe.build_bank_string(bank_cach
for injections in options.injections:
# inspiral jobs by segment
for seg in segs:
injnode = gstlal_inspiral_node(gstlalInspiralInjJob, dag, options.frame_cache, options.frame_segments_file, options.frame_segments_name, seg[0].seconds, seg[1].seconds, channel_dict, reference_psd=options.reference_psd, svd_bank=s, injections=injections, vetoes=options.vetoes, control_peak_time = options.control_peak_time, fir_stride = options.fir_stride, number = i)
injnode = gstlal_inspiral_node(gstlalInspiralInjJob, dag, options.frame_cache, options.frame_segments_file, options.frame_segments_name, seg[0].seconds, seg[1].seconds, channel_dict, reference_psd=options.reference_psd, svd_bank=s, injections=injections, vetoes=options.vetoes, control_peak_time = options.control_peak_time, fir_stride = options.fir_stride, coincidence_threshold=options.coincidence_threshold, number = i)
inspiral_nodes.append((injections, injnode))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment