Skip to content
Snippets Groups Projects
Commit f052a6f2 authored by Wanting Niu's avatar Wanting Niu :speech_balloon:
Browse files

write process parameter to svd_bank file

parent dcad2abb
No related branches found
No related tags found
1 merge request!302write process parameter to svd_bank file
Pipeline #456576 passed
......@@ -192,4 +192,5 @@ for (template_bank, bank_id, clipleft, clipright, bank_type) in zip(template_ban
)
banks.append(bank)
svd_bank.write_bank(options.write_svd_bank, banks, psd)
process_param_dict = options.__dict__.copy()
svd_bank.write_bank(options.write_svd_bank, banks, psd, process_param_dict)
......@@ -54,6 +54,7 @@ from ligo.lw import lsctables
from ligo.lw import array as ligolw_array
from ligo.lw import param as ligolw_param
from ligo.lw import utils as ligolw_utils
from ligo.lw.utils import process as ligolw_process
Attributes = ligolw.sax.xmlreader.AttributesImpl
......@@ -401,13 +402,15 @@ def build_bank(template_bank_url, psd, flow, max_duration, ortho_gate_fap, snr_t
return bank
def write_bank(filename, banks, psd_input, verbose = False):
def write_bank(filename, banks, psd_input, process_param_dict = None, verbose = False):
"""Write SVD banks to a LIGO_LW xml file."""
# Create new document
xmldoc = ligolw.Document()
lw = xmldoc.appendChild(ligolw.LIGO_LW())
# add Process Parameter Table into svd_bank file
if process_param_dict:
process = ligolw_process.register_to_xmldoc(xmldoc, program="gstlal_inspiral_svd_bank", paramdict=process_param_dict, comment="Process parameter tables for further calculation")
for bank in banks:
# set up root for this sub bank
root = lw.appendChild(ligolw.LIGO_LW(Attributes({u"Name": u"gstlal_svd_bank_Bank"})))
......
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