Skip to content
Snippets Groups Projects
Commit 6188c03a authored by ChiWai Chan's avatar ChiWai Chan
Browse files

gstlal_inspiral_calc_snr & svd_bank_snr.py: change spaces to tabs

parent c69f733a
No related branches found
No related tags found
No related merge requests found
Pipeline #84806 passed with warnings
Stage: level0
Stage: level1
Stage: level2
Stage: test-gstlal
Stage: test-inspiral
Stage: test-burst
Stage: test-offline
Stage: docker
Stage: nightly-pages
Stage: deploy
Hide whitespace changes
Inline Side-by-side
gstlal-inspiral/bin/gstlal_inspiral_calc_snr
+ 13
− 13
View file @ 6188c03a
......@@ -145,7 +145,7 @@ class ContentHandler(ligolw.LIGOLWContentHandler):
@lsctables.use_in
class LIGOLWContentHandler(ligolw.LIGOLWContentHandler):
pass
pass
# disable mkchecktimestamps()
# FIXME: python plugin broken until we switch to Python 3
......@@ -265,13 +265,13 @@ def parse_command_line():
# Check SNRs series output
if options.start is None or options.end is None:
raise ValueError("Must have --start and --end.")
elif options.start >= options.end:
raise ValueError("--start must less than --end.")
# Extra handle for SNRs output because SNRs are not stable initially and have padding at the end
# FIXME: the 50s is hardcoded and only use to avoid snr being unstable due to edge effect when doing convoluion
if options.start - gw_data_source_info.seg[0] <= 50 or gw_data_source_info.seg[1] - options.end <= 50:
raise ValueError("Check your inputted --start / --end or your frame file. You should have a long enough data such that, the --start/--end is larger/less than the start/end of your data at least 50s. ")
raise ValueError("Must have --start and --end.")
elif options.start >= options.end:
raise ValueError("--start must less than --end.")
# Extra handle for SNRs output because SNRs are not stable initially and have padding at the end
# FIXME: the 50s is hardcoded and only use to avoid snr being unstable due to edge effect when doing convoluion
if options.start - gw_data_source_info.seg[0] <= 50 or gw_data_source_info.seg[1] - options.end <= 50:
raise ValueError("Check your inputted --start / --end or your frame file. You should have a long enough data such that, the --start/--end is larger/less than the start/end of your data at least 50s. ")
# Setting up PSD
if options.reference_psd:
......@@ -298,9 +298,9 @@ def parse_command_line():
# Setting up SVD bank
bank_urls = inspiral.parse_svdbank_string(options.svd_bank)
# Check if --svd-bank contains svd banks from --instrument
for instrument in options.instrument:
if instrument not in set(bank_urls):
raise ValueError("No SVD Banks for --instrument=%s." % instrument)
for instrument in options.instrument:
if instrument not in set(bank_urls):
raise ValueError("No SVD Banks for --instrument=%s." % instrument)
banks_dict = dict([(ifo, svd_bank.read_banks(svdbanks, svd_bank.DefaultContentHandler)) for ifo, svdbanks in bank_urls.items()])
......@@ -343,9 +343,9 @@ def parse_command_line():
options, gw_data_source_info, template, psds_dict = parse_command_line()
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()
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()
else:
veto_segments = None
veto_segments = None
#====================================================================================================
#
......
gstlal-inspiral/python/svd_bank_snr.py
+ 60
− 60
View file @ 6188c03a
......@@ -54,17 +54,17 @@ class SNR_Pipeline(object):
self.handler = simplehandler.Handler(self.mainloop, self.pipeline)
self.verbose = verbose
self.lock = threading.Lock()
self.row_number = row_number
self.start = start
self.end = end
self.row_number = row_number
self.start = start
self.end = end
self.snr_info = {
"epoch": None,
"instrument": None,
"deltaT": None,
"data": [],
}
if self.start >= self.end:
raise ValueError("Start time must be less than end time.")
if self.start >= self.end:
raise ValueError("Start time must be less than end time.")
def run(self, segments):
if self.verbose:
......@@ -89,12 +89,12 @@ class SNR_Pipeline(object):
raise RuntimeError("pipeline could not be set to NULL.")
def make_series(self, data):
para = {"name" : self.snr_info["instrument"],
"epoch" : self.snr_info["epoch"],
"deltaT" : self.snr_info["deltaT"],
"f0": 0,
"sampleUnits" : lal.DimensionlessUnit,
"length" : len(data)}
para = {"name" : self.snr_info["instrument"],
"epoch" : self.snr_info["epoch"],
"deltaT" : self.snr_info["deltaT"],
"f0": 0,
"sampleUnits" : lal.DimensionlessUnit,
"length" : len(data)}
if data.dtype == numpy.float32:
tseries = lal.CreateREAL4TimeSeries(**para)
elif data.dtype == numpy.float64:
......@@ -110,18 +110,18 @@ class SNR_Pipeline(object):
return tseries
def get_snr_series(self, COMPLEX = False):
assert self.snr_info["epoch"] is not None, "No SNRs are obtained, check your start time."
assert self.snr_info["epoch"] is not None, "No SNRs are obtained, check your start time."
gps_start = self.snr_info["epoch"].gpsSeconds + self.snr_info["epoch"].gpsNanoSeconds * 10.**-9
gps = gps_start + numpy.arange(len(self.snr_info["data"])) * self.snr_info["deltaT"]
if self.start - gps[0] < 0 or self.end - gps[-1] > 0:
raise ValueError("Invalid choice of start time or end time. The data spans from %f to %f." % (gps[0], gps[-1]))
else:
s = abs(gps - self.start).argmin()
e = abs(gps - self.end).argmin()
if self.start - gps[0] < 0 or self.end - gps[-1] > 0:
raise ValueError("Invalid choice of start time or end time. The data spans from %f to %f." % (gps[0], gps[-1]))
else:
s = abs(gps - self.start).argmin()
e = abs(gps - self.end).argmin()
self.snr_info["epoch"] = gps[s]
self.snr_info["data"] = self.snr_info["data"][s:e].T
self.snr_info["epoch"] = gps[s]
self.snr_info["data"] = self.snr_info["data"][s:e].T
if self.row_number is None:
temp = []
......@@ -141,17 +141,17 @@ class SNR_Pipeline(object):
return [self.make_series(numpy.abs(self.snr_info["data"]))]
def new_preroll_handler(self, elem):
with self.lock:
# ignore preroll buffers
elem.emit("pull-preroll")
return Gst.FlowReturn.OK
def new_preroll_handler(self, elem):
with self.lock:
# ignore preroll buffers
elem.emit("pull-preroll")
return Gst.FlowReturn.OK
def pull_snr_buffer(self, elem):
with self.lock:
sample = elem.emit("pull-sample")
if sample is None:
return Gst.FlowReturn.OK
def pull_snr_buffer(self, elem):
with self.lock:
sample = elem.emit("pull-sample")
if sample is None:
return Gst.FlowReturn.OK
success, rate = sample.get_caps().get_structure(0).get_int("rate")
......@@ -162,21 +162,21 @@ class SNR_Pipeline(object):
else:
assert self.snr_info["deltaT"] == 1. / rate, "data have different sampling rate."
buf = sample.get_buffer()
if buf.mini_object.flags & Gst.BufferFlags.GAP or buf.n_memory() == 0:
return Gst.FlowReturn.OK
# drop snrs that are irrelevant
cur_time_stamp = LIGOTimeGPS(0, sample.get_buffer().pts)
if self.start >= cur_time_stamp and self.end > cur_time_stamp:
# record the first timestamp closet to start time
self.snr_info["epoch"] = cur_time_stamp
# FIXME: check timestamps
self.snr_info["data"] = [pipeio.array_from_audio_sample(sample)]
elif self.start <= cur_time_stamp < self.end:
buf = sample.get_buffer()
if buf.mini_object.flags & Gst.BufferFlags.GAP or buf.n_memory() == 0:
return Gst.FlowReturn.OK
# drop snrs that are irrelevant
cur_time_stamp = LIGOTimeGPS(0, sample.get_buffer().pts)
if self.start >= cur_time_stamp and self.end > cur_time_stamp:
# record the first timestamp closet to start time
self.snr_info["epoch"] = cur_time_stamp
# FIXME: check timestamps
self.snr_info["data"] = [pipeio.array_from_audio_sample(sample)]
elif self.start <= cur_time_stamp < self.end:
self.snr_info["data"].append(pipeio.array_from_audio_sample(sample))
else:
Gst.FlowReturn.OK
else:
Gst.FlowReturn.OK
return Gst.FlowReturn.OK
......@@ -226,7 +226,7 @@ class LLOID_SNR(SNR_Pipeline):
assert handler_id > 0
self.run(gw_data_source_info.seg)
self.snr_info["data"] = numpy.concatenate(numpy.array(self.snr_info["data"]), axis = 0)
self.snr_info["data"] = numpy.concatenate(numpy.array(self.snr_info["data"]), axis = 0)
def __call__(self, COMPLEX = False):
return self.get_snr_series(COMPLEX)
......@@ -268,7 +268,7 @@ class FIR_SNR(SNR_Pipeline):
assert handler_id > 0
self.run(gw_data_source_info.seg)
self.snr_info["data"] = numpy.concatenate(numpy.array(self.snr_info["data"]), axis = 0)
self.snr_info["data"] = numpy.concatenate(numpy.array(self.snr_info["data"]), axis = 0)
self.snr_info["data"] = numpy.vectorize(complex)(self.snr_info["data"][:,0], self.snr_info["data"][:,1])
self.snr_info["data"].shape = len(self.snr_info["data"]), 1
......@@ -320,7 +320,7 @@ class FIR_SNR(SNR_Pipeline):
#=============================================================================================
#
# Output Utilities
# Output Utilities
#
#=============================================================================================
......@@ -385,7 +385,7 @@ def read_url(filename, contenthandler = SNRContentHandler, verbose = False):
#=============================================================================================
#
# Gracedb Events Utilities
# Gracedb Events Utilities
#
#=============================================================================================
......@@ -396,14 +396,14 @@ def scan_svd_banks_for_row(coinc_xmldoc, banks_dict):
sub_bank_id = None
row_number = None
for i, bank in enumerate(banks_dict.values()[0]):
for j, row in enumerate(bank.sngl_inspiral_table):
if row.template_id == eventid_trigger_dict.values()[0].template_id:
sub_bank_id = i
row_number = j
break
if sub_bank_id is not None:
break
for i, bank in enumerate(banks_dict.values()[0]):
for j, row in enumerate(bank.sngl_inspiral_table):
if row.template_id == eventid_trigger_dict.values()[0].template_id:
sub_bank_id = i
row_number = j
break
if sub_bank_id is not None:
break
assert sub_bank_id is not None, "Cannot find the template listed in the coinc.xml."
return sub_bank_id, row_number
......@@ -453,19 +453,19 @@ def framecache_from_event(gid, observatories, frame_types, time_span = 1000, out
# FIXME: This is not reliable, have a better way to map frame_type to observatory?
obs_type_dict = dict([(obs, frame_type) for obs in observatories for frame_type in frame_types if obs == frame_type[0]])
gracedb_client = gracedb.GraceDb()
coinc_xmldoc = lvalert_helper.get_coinc_xmldoc(gracedb_client, gid)
eventid_trigger_dict = dict((row.ifo, row) for row in lsctables.SnglInspiralTable.get_table(coinc_xmldoc))
gracedb_client = gracedb.GraceDb()
coinc_xmldoc = lvalert_helper.get_coinc_xmldoc(gracedb_client, gid)
eventid_trigger_dict = dict((row.ifo, row) for row in lsctables.SnglInspiralTable.get_table(coinc_xmldoc))
channel_names_dict = dict([(row.value.split("=")[0], row.value) for row in lsctables.ProcessParamsTable.get_table(coinc_xmldoc) if row.param == "--channel-name"])
gwdata_metavar_headers = ["instruments", "trigger_times", "gps_start_time", "gps_end_time", "channels_name"]
gwdata_metavar_values = []
urls = []
for observatory, frame_type in obs_type_dict.items():
for observatory, frame_type in obs_type_dict.items():
trigger_time = eventid_trigger_dict[obs2ifo[observatory]].end
gps_start_time = int(trigger_time - time_span)
gps_end_time = int(trigger_time + time_span)
gwdata_metavar_values.append((obs2ifo[observatory], trigger_time, gps_start_time, gps_end_time, channel_names_dict[obs2ifo[observatory]]))
gwdata_metavar_values.append((obs2ifo[observatory], trigger_time, gps_start_time, gps_end_time, channel_names_dict[obs2ifo[observatory]]))
urls += gwdatafind.find_urls(observatory, frame_type, gps_start_time, gps_end_time)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment