def compute_inj_optimal_snr(workflow,
inj_file,
precalc_psd_files,
out_dir,
tags=None):
"Set up a job for computing optimal SNRs of a sim_inspiral file."
if tags is None:
tags = []
try:
factor = int(
workflow.cp.get_opt_tags('workflow-optimal-snr',
'parallelization-factor', tags))
except ConfigParser.Error:
factor = 1
if factor == 1:
# parallelization factor not given - default to single optimal snr job
opt_snr_exe = PyCBCOptimalSNRExecutable(workflow.cp,
'optimal_snr',
ifos=workflow.ifos,
out_dir=out_dir,
tags=tags)
node = opt_snr_exe.create_node(workflow, inj_file, precalc_psd_files,
'0/1')
workflow += node
return node.output_files[0]
opt_snr_split_files = []
for i in range(factor):
group_str = '%s/%s' % (i, factor)
opt_snr_exe = PyCBCOptimalSNRExecutable(workflow.cp,
'optimal_snr',
ifos=workflow.ifos,
out_dir=out_dir,
tags=tags + [str(i)])
opt_snr_exe.update_current_retention_level(
Executable.INTERMEDIATE_PRODUCT)
node = opt_snr_exe.create_node(workflow, inj_file, precalc_psd_files,
group_str)
opt_snr_split_files += [node.output_files[0]]
workflow += node
llwadd_exe = LigolwAddExecutable(workflow.cp,
'optimal_snr_merge',
ifos=workflow.ifos,
out_dir=out_dir,
tags=tags)
llwadd_exe.update_current_retention_level(Executable.MERGED_TRIGGERS)
merge_node = llwadd_exe.create_node(workflow.analysis_time,
opt_snr_split_files,
use_tmp_subdirs=False)
workflow += merge_node
return merge_node.output_files[0]
