From 5bab69fe11c096c4099034d838d1be1188a78c09 Mon Sep 17 00:00:00 2001
From: Ryan Michael Magee <ryan.magee@comp-hd-001.gwave.ics.psu.edu>
Date: Wed, 7 Aug 2019 16:53:47 -0400
Subject: [PATCH] construct_skymap_test_dag : added command line options
---
gstlal-ugly/share/construct_skymap_test_dag | 49 ++++++++++++++++-----
1 file changed, 38 insertions(+), 11 deletions(-)
diff --git a/gstlal-ugly/share/construct_skymap_test_dag b/gstlal-ugly/share/construct_skymap_test_dag
index 9ce7d8997d..df55895bf6 100755
--- a/gstlal-ugly/share/construct_skymap_test_dag
+++ b/gstlal-ugly/share/construct_skymap_test_dag
@@ -1,10 +1,10 @@
#!/usr/bin/env python
'''
-./construct_skymap_test_dag path/to/injection/database path/to/tmp/space max_number_inspiral_jobs
+construct_skymap_test_dag --injection-database=path/to/injection/database --tmp-space=path/to/tmp/space --far-threshold=1e-7 --max-number-inspiral-jobs=max_number_inspiral_jobs
'''
+from optparse import OptionParser
import sqlite3
-import sys
import os
import glob
@@ -15,6 +15,30 @@ from ligo.lw.utils import process as ligolw_process
from ligo.lw import dbtables
from gstlal import dagparts
+def parse_command_line():
+ parser = OptionParser()
+ parser.add_option("-i", "--injection-database", metavar = "filename", help = "Path to the injection database.")
+ parser.add_option("-t", "--tmp-space", metavar = "path", help = "Path to temp space. Defaults to _CONDOR_SCRATCH_DIR .")
+ parser.add_option("-f", "--far-threshold", type = "float", default = 3.86e-7, help = "Threshold for rerunning an injection job to produce the time series.")
+ parser.add_option("-m", "--max-number-inspiral-jobs", type = "int", default = 10000, help = "Maximum number of inspiral jobs to queue. If this option is not provided, the program defaults to the top 10000 that pass the imposed FAR threshold.")
+
+ options, filenames = parser.parse_args()
+
+ return options, filenames
+
+options, filenames = parse_command_line()
+
+if not options.injection_database:
+ raise ValueError("You must provide an injection database.")
+
+if not options.tmp_space:
+ if '_CONDOR_SCRATCH_DIR' in os.environ:
+ options.tmp_space = os.environ['_CONDOR_SCRATCH_DIR']
+ else:
+ options.tmp_space = os.environ['TMPDIR']
+
+
+
# copied from gstlal_inspiral_plotsummary
def create_sim_coinc_view(connection):
"""
@@ -68,10 +92,6 @@ AS
sim_coinc_map_helper
WHERE
sid = simulation_id
- AND
- nevents < 3
- AND
- instruments = "H1,L1,V1"
ORDER BY
lr
DESC
@@ -138,9 +158,9 @@ CREATE TEMPORARY TABLE
connection.cursor().execute("DROP INDEX sim_id_combined_far_index")
connection.cursor().execute("DROP INDEX sim_id_sngl_id_index")
-inj_db = sys.argv[1]
-tmp_space = sys.argv[2]
-num_inspiral_jobs = int(sys.argv[3])
+inj_db = options.injection_database
+tmp_space = options.tmp_space
+num_inspiral_jobs = options.max_number_inspiral_jobs
analysis_dir = os.path.dirname(inj_db)
working_filename = dbtables.get_connection_filename(inj_db, tmp_path = tmp_space, verbose = True)
@@ -156,11 +176,11 @@ SELECT
FROM
sim_sngl_far
WHERE
- far <= 3.86e-7
+ far <= %e
ORDER BY
far ASC
LIMIT ?
-""", (int(num_inspiral_jobs),)):
+""" % (options.far_threshold,), (int(num_inspiral_jobs),)):
process_id = record[0]
bank_id = record[1]
far = record[2]
@@ -260,6 +280,13 @@ try:
except OSError:
pass
+mass_model = analysis_dir + '/gstlal_inspiral_mass_model'
+try:
+ os.mkdir("gstlal_inspiral_mass_model")
+ os.system('cp -r %s .' % (mass_model,))
+except OSError:
+ pass
+
bayesdir = "bayestar_input"
try:
os.mkdir(bayesdir)
--
GitLab