def poisson_safety(segs, injTable, livetime):
"""
Return a tuple containing the number of vetoed injections, the number
expected, and the Poisson safety probability based on the number of
injections vetoed relative to random chance according to Poisson statistics.
Arguments:
segs : glue.segments.segmentlist
list of segments to be tested
injTable : glue.ligolw.table.Table
table of injections
livetime : [ float ]
livetime of search
"""
deadtime = segs.__abs__()
get_time = def_get_time(injTable.tableName)
injvetoed = len([ inj for inj in injTable if get_time(inj) in segs ])
injexp = len(injTable)*float(deadtime) / float(livetime)
prob = 1 - poisson.cdf(injvetoed-1, injexp)
return injvetoed, injexp, prob
def poisson_safety(segs, injTable, livetime):
"""
Return a tuple containing the number of vetoed injections, the number
expected, and the Poisson safety probability based on the number of
injections vetoed relative to random chance according to Poisson statistics.
Arguments:
segs : glue.segments.segmentlist
list of segments to be tested
injTable : glue.ligolw.table.Table
table of injections
livetime : [ float ]
livetime of search
"""
deadtime = segs.__abs__()
get_time = def_get_time(injTable.tableName)
injvetoed = len([inj for inj in injTable if get_time(inj) in segs])
injexp = len(injTable) * float(deadtime) / float(livetime)
prob = 1 - poisson.cdf(injvetoed - 1, injexp)
return injvetoed, injexp, prob
def frominjectionfile(file, type, ifo=None, start=None, end=None):
"""
Read generic injection file object file containing injections of the given
type string. Returns an 'Sim' lsctable of the corresponding type.
Arguments:
file : file object
type : [ "inspiral" | "burst" | "ringdown" ]
Keyword arguments:
ifo : [ "G1" | "H1" | "H2" | "L1" | "V1" ]
"""
# read type
type = type.lower()
# read injection xml
xml = re.compile('(xml$|xml.gz$)')
if re.search(xml,file.name):
xmldoc,digest = utils.load_fileobj(file)
injtable = table.get_table(xmldoc,'sim_%s:table' % (type))
# read injection txt
else:
cchar = re.compile('[#%<!()_\[\]{}:;\'\"]+')
#== construct new Sim{Burst,Inspiral,Ringdown}Table
injtable = lsctables.New(lsctables.__dict__['Sim%sTable' % (type.title())])
if type=='inspiral':
columns = ['geocent_end_time.geocent_end_time_ns',\
'h_end_time.h_end_time_ns',\
'l_end_time.l_end_time_ns',\
'v_end_time.v_end_time_ns',\
'distance']
for line in file.readlines():
if re.match(cchar,line):
continue
# set up siminspiral object
inj = lsctables.SimInspiral()
# split data
sep = re.compile('[\s,=]+')
data = sep.split(line)
# set attributes
inj.geocent_end_time = int(data[0].split('.')[0])
inj.geocent_end_time_ns = int(data[0].split('.')[1])
inj.h_end_time = int(data[1].split('.')[0])
inj.h_end_time_ns = int(data[1].split('.')[1])
inj.l_end_time = int(data[2].split('.')[0])
inj.l_end_time_ns = int(data[2].split('.')[1])
inj.v_end_time = int(data[3].split('.')[0])
inj.v_end_time_ns = int(data[3].split('.')[1])
inj.distance = float(data[4])
injtable.append(inj)
if type=='burst':
if file.readlines()[0].startswith('filestart'):
# if given parsed burst file
file.seek(0)
snrcol = { 'G1':23, 'H1':19, 'L1':21, 'V1':25 }
for line in file.readlines():
inj = lsctables.SimBurst()
# split data
sep = re.compile('[\s,=]+')
data = sep.split(line)
# set attributes
# gps time
if 'burstgps' in data:
idx = data.index('burstgps')+1
geocent = LIGOTimeGPS(data[idx])
inj.time_geocent_gps = geocent.seconds
inj.time_geocent_gps_ns = geocent.nanoseconds
else:
continue
#inj.waveform = data[4]
#inj.waveform_number = int(data[5])
# frequency
if 'freq' in data:
idx = data.index('freq')+1
inj.frequency = float(data[idx])
else:
continue
# SNR a.k.a. amplitude
if ifo and 'snr%s' % ifo in data:
idx = data.index('snr%s' % ifo)+1
inj.amplitude = float(data[idx])
elif 'rmsSNR' in data:
idx = data.index('rmsSNR')+1
inj.amplitude = float(data[idx])
else:
continue
if 'phi' in data:
idx = data.index('phi' )+1
inj.ra = float(data[idx])*24/(2*math.pi)
if 'theta' in data:
idx = data.index('theta' )+1
inj.ra = 90-(float(data[idx])*180/math.pi)
if ifo and 'hrss%s' % ifo in data:
idx = data.index('hrss%s' % ifo)+1
inj.hrss = float(data[idx])
elif 'hrss' in data:
idx = data.index('hrss')+1
inj.hrss = float(data[idx])
# extra columns to be added when I know how
#inj.q = 0
#inj.q = float(data[11])
#h_delay = LIGOTimeGPS(data[41])
#inj.h_peak_time = inj.time_geocent_gps+h_delay.seconds
#inj.h_peak_time_ns = inj.time_geocent_gps_ns+h_delay.nanoseconds
#l_delay = LIGOTimeGPS(data[43])
#inj.l_peak_time = inj.time_geocent_gps+l_delay.seconds
#inj.l_peak_time_ns = inj.time_geocent_gps_ns+l_delay.nanoseconds
#v_delay = LIGOTimeGPS(data[43])
#inj.v_peak_time = inj.time_geocent_gps+v_delay.seconds
#inj.v_peak_time_ns = inj.time_geocent_gps_ns+v_delay.nanoseconds
injtable.append(inj)
else:
# if given parsed burst file
file.seek(0)
for line in file.readlines():
inj = lsctables.SimBurst()
# split data
sep = re.compile('[\s,]+')
data = sep.split(line)
# set attributes
geocent = LIGOTimeGPS(data[0])
inj.time_geocent_gps = geocent.seconds
inj.time_geocent_gps_ns = geocent.nanoseconds
injtable.append(inj)
injections = table.new_from_template(injtable)
if not start: start = 0
if not end: end = 9999999999
span = segments.segmentlist([ segments.segment(start, end) ])
get_time = dqTriggerUtils.def_get_time(injections.tableName)
injections.extend(inj for inj in injtable if get_time(inj) in span)
return injections
