From bf39f6d29c290d246de943579a85fdd1d9eb62e4 Mon Sep 17 00:00:00 2001
From: Chad Hanna <chad.hanna@ligo.org>
Date: Wed, 11 Jul 2018 19:15:47 -0700
Subject: [PATCH] gstlal_play: add vetoes
---
gstlal/bin/gstlal_play | 18 +++++++++++++++---
1 file changed, 15 insertions(+), 3 deletions(-)
diff --git a/gstlal/bin/gstlal_play b/gstlal/bin/gstlal_play
index 42a1a983ec..ca109fd17b 100755
--- a/gstlal/bin/gstlal_play
+++ b/gstlal/bin/gstlal_play
@@ -34,6 +34,7 @@ from gstlal import multirate_datasource
from gstlal import pipeparts
from gstlal import simplehandler
from glue.ligolw.utils import segments as ligolw_segments
+from glue.ligolw import lsctables, ligolw
### This program will play data in a variety of ways
###
@@ -129,6 +130,11 @@ from glue.ligolw.utils import segments as ligolw_segments
###
+@lsctables.use_in
+class LIGOLWContentHandler(ligolw.LIGOLWContentHandler):
+ pass
+
+
def parse_command_line():
parser = OptionParser(description = __doc__)
@@ -143,6 +149,8 @@ def parse_command_line():
parser.add_option("--sample-format", metavar = "name", help = "Force a specific sample format for the output. If not specified, the format is chosen by auto-negotiation with the encoder. Allowed values are any GStreamer-recognized format that is compatible with the requested encoder. Examples include \"F32LE\", \"F64LE\".")
parser.add_option("--rate", metavar = "Hz", type = "int", default = 4096, help = "Downsample input to this sample rate. Default = 4096 Hz. Must be <= input sample rate or else you will get a caps negotiation error.")
parser.add_option("--whiten", action = "store_true", help = "Whiten the time series (default = do not whiten).")
+ parser.add_option("--veto-segments-file", metavar = "filename", help = "Set the name of the LIGO light-weight XML file from which to load vetoes (optional). Must coincide with --whiten")
+ parser.add_option("--veto-segments-name", metavar = "name", help = "Set the name of the segments to extract from the segment tables and use as the veto list. Must coincide with --whiten", default = "vetoes")
parser.add_option("--reference-psd", metavar = "file", help = "When whitening, normalize the time series to the spectrum described in this XML file. If this option is not given, the spectrum is measured from the data.")
parser.add_option("--low-pass-filter", metavar = "Hz", type = "float", help = "Low pass filter frequency (default = no low-pass filter). Low-pass filter is applied after whitening.")
parser.add_option("--high-pass-filter", metavar = "Hz", type = "float", help = "High pass filter frequency (default = no high-pass filter). High-pass filter is applied after whitening.")
@@ -159,8 +167,8 @@ def parse_command_line():
if options.low_pass_filter <= options.high_pass_filter:
raise ValueError("--low-pass-filter must be > --high-pass-filter")
- if options.reference_psd and not options.whiten:
- raise ValueError("--reference-psd requires --whiten")
+ if options.reference_psd or options.veto_segments_file and not options.whiten:
+ raise ValueError("--reference-psd and --veto-segments-file requires --whiten")
if len(options.channel_name) > 1:
raise ValueError("only one --channel-name allowed")
@@ -179,6 +187,10 @@ if options.reference_psd is not None:
else:
psd = None
+if options.veto_segments_file is not None:
+ veto_segments = ligolw_segments.segmenttable_get_by_name(ligolw_utils.load_filename(options.veto_segments_file, verbose = options.verbose, contenthandler = LIGOLWContentHandler), options.veto_segments_name).coalesce()[instrument]
+else:
+ veto_segments = None
# building the event loop and pipeline
mainloop = GObject.MainLoop()
@@ -193,7 +205,7 @@ head, _, _ = datasource.mkbasicsrc(pipeline, gw_data_source_info, instrument, ve
# if whitening, leverage mkwhitened_multirate_src() otherwise just resample
if options.whiten:
- head = multirate_datasource.mkwhitened_multirate_src(pipeline, head, [options.rate], instrument, psd, track_psd = True)[options.rate]
+ head = multirate_datasource.mkwhitened_multirate_src(pipeline, head, [options.rate], instrument, psd, track_psd = True, veto_segments = veto_segments)[options.rate]
else:
head = pipeparts.mkresample(pipeline, head, quality = 9)
head = pipeparts.mkcapsfilter(pipeline, head, "audio/x-raw, rate=%d" % options.rate)
--
GitLab