From 981927582e8ed3ec8998cdb71f37e37396c7bd6d Mon Sep 17 00:00:00 2001
From: Cody Messick <cody.messick@ligo.org>
Date: Tue, 20 Feb 2018 15:59:48 -0500
Subject: [PATCH] First commit of gstlal_injsplitter (Note: Is actually
Surabhis code, committed with permission)
---
gstlal-ugly/bin/Makefile.am | 3 +-
gstlal-ugly/bin/gstlal_injsplitter | 93 ++++++++++++++++++++++++++++++
2 files changed, 95 insertions(+), 1 deletion(-)
create mode 100755 gstlal-ugly/bin/gstlal_injsplitter
diff --git a/gstlal-ugly/bin/Makefile.am b/gstlal-ugly/bin/Makefile.am
index b9d70c6e02..159fc7835a 100644
--- a/gstlal-ugly/bin/Makefile.am
+++ b/gstlal-ugly/bin/Makefile.am
@@ -34,4 +34,5 @@ dist_bin_SCRIPTS = \
gstlal_ll_inspiral_state \
gstlal_condor_top \
gstlal_etg \
- gstlal_etg_pipe
+ gstlal_etg_pipe \
+ gstlal_injsplitter
diff --git a/gstlal-ugly/bin/gstlal_injsplitter b/gstlal-ugly/bin/gstlal_injsplitter
new file mode 100755
index 0000000000..cbeed6d59d
--- /dev/null
+++ b/gstlal-ugly/bin/gstlal_injsplitter
@@ -0,0 +1,93 @@
+#! /usr/bin/python
+# Copyright (C) 2015 Surabhi Sachdev, Tjonnie Li
+#
+# This program is free software; you can redistribute it and/or modify it
+# under the terms of the GNU General Public License as published by the
+# Free Software Foundation; either version 2 of the License, or (at your
+# option) any later version.
+#
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
+# Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+
+###############################################################################
+#
+# IMPORT MODULES
+#
+###############################################################################
+
+from time import strftime
+from optparse import OptionParser
+from glue.ligolw import lsctables
+from glue.ligolw import utils
+from glue.ligolw import ligolw
+from glue.ligolw.utils import process as ligolw_process
+import numpy
+
+###############################################################################
+#
+# IMPORT MODULES
+#
+###############################################################################
+
+class ContentHandler(ligolw.LIGOLWContentHandler):
+ pass
+lsctables.use_in(ContentHandler)
+
+###############################################################################
+#
+# COMMAND LINE PARSING
+#
+###############################################################################
+
+def parse_command_line():
+ parser = OptionParser()
+ parser.add_option("-o", "--output-path", metavar = "path", default = ".", help = "Set the path to the directory where output files will be written. Default is \".\".")
+ parser.add_option("-u", "--usertag", metavar = "usertag", help = "Set the user tag",default="INJSPLITTER")
+ parser.add_option("-n", "--nsplit", metavar = "count", type = "int", help = "Number of files you want the original file to be split into",default=1)
+ parser.add_option("-v", "--verbose", action = "store_true", help = "Be verbose.")
+ options, filenames = parser.parse_args()
+
+ if len(filenames)!=1:
+ raise ValueError, "Provide one injection file"
+
+ return options, filenames
+
+options, filenames = parse_command_line()
+
+###############################################################################
+#
+# MAIN CODE
+#
+###############################################################################
+
+# GETTING COMMAND LINE OPTIONS FOR PRINTING INTO THE TABLE
+opts_dict = dict((k, v) for k, v in options.__dict__.iteritems() if v is not False and v is not None)
+
+# LOAD INJECTION TABLE
+xmldoc=utils.load_filename(filenames[0], gz=filenames[0].endswith(".gz"), verbose = options.verbose, contenthandler=ContentHandler)
+sim_inspiral_table=lsctables.table.get_table(xmldoc, lsctables.SimInspiralTable.tableName)
+# Sort the sim inspiral table based on geocent end time
+sim_inspiral_table.sort(key = lambda row: (row.geocent_end_time + 1e-9 * row.geocent_end_time_ns))
+process_params_table = lsctables.table.get_table(xmldoc, lsctables.ProcessParamsTable.tableName)
+
+# PREPARE PROCESS TABLE WITH INFORMATION ABOUT THE CURRENT PROGRAM
+process = ligolw_process.register_to_xmldoc(xmldoc,
+"gstlal_injsplitter", opts_dict,
+version="no version", cvs_repository="gstlal",
+cvs_entry_time=strftime('%Y/%m/%d %H:%M:%S'))
+
+sim_inspiral_table_split = lsctables.New(lsctables.SimInspiralTable)
+sim_inspiral_table.parentNode.replaceChild(sim_inspiral_table_split, sim_inspiral_table)
+
+evensplit = numpy.array_split(sim_inspiral_table,options.nsplit);
+for i in xrange(options.nsplit):
+ sim_inspiral_table_split[:] = evensplit[i]
+ ligolw_process.set_process_end_time(process)
+ utils.write_filename(xmldoc, "%s/%s_INJ_SPLIT_%04d.xml"%(options.output_path,options.usertag,i), gz = False, verbose = options.verbose)
+
--
GitLab