From 222ed2d3c6c4307c643a083e701b929fc16c81a0 Mon Sep 17 00:00:00 2001
From: Duncan Meacher <duncan.meacher@ligo.org>
Date: Mon, 10 Dec 2018 12:03:18 -0800
Subject: [PATCH] gstlal_inspiral_add_template_ids: Can loop over multiple bank
files
---
.../bin/gstlal_inspiral_add_template_ids | 15 +++++++++------
1 file changed, 9 insertions(+), 6 deletions(-)
diff --git a/gstlal-inspiral/bin/gstlal_inspiral_add_template_ids b/gstlal-inspiral/bin/gstlal_inspiral_add_template_ids
index f82ee14aa7..b812b0ad09 100755
--- a/gstlal-inspiral/bin/gstlal_inspiral_add_template_ids
+++ b/gstlal-inspiral/bin/gstlal_inspiral_add_template_ids
@@ -31,12 +31,15 @@ from glue.ligolw import utils as ligolw_utils
class LIGOLWContentHandler(ligolw.LIGOLWContentHandler):
pass
-fname = sys.argv[1]
+cnt = 0
-xmldoc = ligolw_utils.load_filename(fname, verbose = True, contenthandler = LIGOLWContentHandler)
-sngl_inspiral_table = lsctables.SnglInspiralTable.get_table(xmldoc)
+for fname in sys.argv[1:]:
-for n, row in enumerate(sngl_inspiral_table):
- row.template_id = n
+ xmldoc = ligolw_utils.load_filename(fname, verbose = True, contenthandler = LIGOLWContentHandler)
+ sngl_inspiral_table = lsctables.SnglInspiralTable.get_table(xmldoc)
-ligolw_utils.write_filename(xmldoc, fname, gz = fname.endswith('gz'), verbose = True)
+ for row in sngl_inspiral_table:
+ row.template_id = cnt
+ cnt += 1
+
+ ligolw_utils.write_filename(xmldoc, fname, gz = fname.endswith('gz'), verbose = True)
--
GitLab