From 670e625124d846e982cba6cf1e7bb362d62eed24 Mon Sep 17 00:00:00 2001
From: ChiWai Chan <chiwai.chan@ligo.org>
Date: Wed, 4 Sep 2019 19:05:54 -0700
Subject: [PATCH] gstlal_inspiral_plot_snr: drop format checking & fix output
naming when --center and --span is given
---
gstlal-inspiral/bin/gstlal_inspiral_plot_snr | 13 +++++++++----
1 file changed, 9 insertions(+), 4 deletions(-)
diff --git a/gstlal-inspiral/bin/gstlal_inspiral_plot_snr b/gstlal-inspiral/bin/gstlal_inspiral_plot_snr
index 36d10e1f27..73799c6ad0 100755
--- a/gstlal-inspiral/bin/gstlal_inspiral_plot_snr
+++ b/gstlal-inspiral/bin/gstlal_inspiral_plot_snr
@@ -29,15 +29,16 @@ def parse_command_line():
raise ValueError("Missing --input.")
if options.outdir is None:
raise ValueError("Missing --outdir.")
- if options.format is not None:
- if options.format != ".png" and options.format != ".svg":
- raise ValueError("--format can only be '.png' or '.svg'.")
+ if options.center is not None and options.span is None:
+ raise ValueError("--span should also be provided if --center is provided.")
+ if options.span is not None and options.center is None:
+ raise ValueError("--center should also be provided if --span is provided.")
return options
options = parse_command_line()
-suffix = ".svg" if (options.format is None) else options.format
+suffix = ".png" if (options.format is None) else options.format
urls = []
description = None
segment = None
@@ -51,6 +52,10 @@ with open(options.input) as cache:
else:
assert description + segment == entry.description + "-" + str(entry.segment[0]) + "-" + str(entry.segment[1] - entry.segment[0]), "Cannot plot snrs with different segments or for different template."
+# done checking file consistency, overwrite output name with customized segment
+if options.center and options.span:
+ segment = "-" + str(int(options.center - options.span)) + "-" + str(int(options.span * 2))
+
SNRs_dict = {}
for url in urls:
SNRs_dict.update(svd_bank_snr.read_xmldoc(svd_bank_snr.read_url(url, svd_bank_snr.SNRContentHandler, verbose = options.verbose)))
--
GitLab