def write_to_xml(cells, intr_prms, fvals=None, fname=None, verbose=False):
"""
Write a set of cells, with dimensions corresponding to intr_prms to an XML file as sim_inspiral rows.
"""
xmldoc = ligolw.Document()
xmldoc.appendChild(ligolw.LIGO_LW())
procrow = process.append_process(xmldoc, program=sys.argv[0])
procid = procrow.process_id
process.append_process_params(xmldoc, procrow, process.process_params_from_dict(opts.__dict__))
rows = ["simulation_id", "process_id", "numrel_data"] + list(intr_prms)
if fvals is not None:
rows.append("alpha1")
sim_insp_tbl = lsctables.New(lsctables.SimInspiralTable, rows)
for itr, intr_prm in enumerate(cells):
sim_insp = sim_insp_tbl.RowType()
# FIXME: Need better IDs
sim_insp.numrel_data = "INTR_SET_%d" % itr
sim_insp.simulation_id = ilwd.ilwdchar("sim_inspiral:sim_inspiral_id:%d" % itr)
sim_insp.process_id = procid
if fvals:
sim_insp.alpha1 = fvals[itr]
for p, v in zip(intr_prms, intr_prm._center):
setattr(sim_insp, p, v)
sim_insp_tbl.append(sim_insp)
xmldoc.childNodes[0].appendChild(sim_insp_tbl)
if fname is None:
channel_name = ["H=H", "L=L"]
ifos = "".join([o.split("=")[0][0] for o in channel_name])
#start = int(event_time)
start = 0
fname = "%s-MASS_POINTS-%d-1.xml.gz" % (ifos, start)
utils.write_filename(xmldoc, fname, gz=True, verbose=verbose)
def write_to_xml(cells,
intr_prms,
pin_prms={},
fvals=None,
fname=None,
verbose=False):
"""
Write a set of cells, with dimensions corresponding to intr_prms to an XML file as sim_inspiral rows.
"""
xmldoc = ligolw.Document()
xmldoc.appendChild(ligolw.LIGO_LW())
procrow = process.append_process(xmldoc, program=sys.argv[0])
procid = procrow.process_id
process.append_process_params(
xmldoc, procrow, process.process_params_from_dict(opts.__dict__))
rows = ["simulation_id", "process_id", "numrel_data"]
# Override eff_lambda to with psi0, its shoehorn column
if "eff_lambda" in intr_prms:
intr_prms[intr_prms.index("eff_lambda")] = "psi0"
if "deff_lambda" in intr_prms:
intr_prms[intr_prms.index("deff_lambda")] = "psi3"
rows += list(intr_prms)
rows += list(pin_prms)
if fvals is not None:
rows.append("alpha1")
sim_insp_tbl = lsctables.New(lsctables.SimInspiralTable, rows)
for itr, intr_prm in enumerate(cells):
sim_insp = sim_insp_tbl.RowType()
# FIXME: Need better IDs
sim_insp.numrel_data = "INTR_SET_%d" % itr
sim_insp.simulation_id = ilwd.ilwdchar(
"sim_inspiral:sim_inspiral_id:%d" % itr)
sim_insp.process_id = procid
if fvals:
sim_insp.alpha1 = fvals[itr]
for p, v in zip(intr_prms, intr_prm._center):
setattr(sim_insp, p, v)
for p, v in pin_prms.iteritems():
setattr(sim_insp, p, v)
sim_insp_tbl.append(sim_insp)
xmldoc.childNodes[0].appendChild(sim_insp_tbl)
if fname is None:
channel_name = ["H=H", "L=L"]
ifos = "".join([o.split("=")[0][0] for o in channel_name])
#start = int(event_time)
start = 0
fname = "%s-MASS_POINTS-%d-1.xml.gz" % (ifos, start)
utils.write_filename(xmldoc, fname, gz=True, verbose=verbose)
# If clusters get upgraded, add this header to output:
# header='grid center; 2x2 effective Fisher matrix, 4rd row: ellipsoid axes, 5th row: total ellipse area, estimated physical area'
if opts.save_ellipsoid_data:
np.savetxt('ellipsoid.dat', test)
# Convert to m1, m2
m1m2_grid = np.array([lsu.m1m2(cart_grid[i][0], cart_grid[i][1])
for i in xrange(len(cart_grid))])
m1m2_grid /= lal.MSUN_SI
if opts.mass_points_xml:
xmldoc = ligolw.Document()
xmldoc.appendChild(ligolw.LIGO_LW())
procrow = process.append_process(xmldoc, program=sys.argv[0])
procid = procrow.process_id
process.append_process_params(xmldoc, procrow, process.process_params_from_dict(opts.__dict__))
sim_insp_tbl = lsctables.New(lsctables.SimInspiralTable, ["simulation_id", "process_id", "numrel_data", "mass1", "mass2", "psi0", "psi3"])
for itr, (m1, m2) in enumerate(m1m2_grid):
for l1 in np.linspace(common_cl.param_limits["lam_tilde"][0], common_cl.param_limits["lam_tilde"][1], Nlam):
sim_insp = sim_insp_tbl.RowType()
sim_insp.numrel_data = "MASS_SET_%d" % itr
sim_insp.simulation_id = ilwd.ilwdchar("sim_inspiral:sim_inspiral_id:%d" % itr)
sim_insp.process_id = procid
sim_insp.mass1, sim_insp.mass2 = m1, m2
sim_insp.psi0, sim_insp.psi3 = opts.eff_lambda or l1, opts.delta_eff_lambda or 0
sim_insp_tbl.append(sim_insp)
xmldoc.childNodes[0].appendChild(sim_insp_tbl)
if opts.channel_name:
ifos = "".join([o.split("=")[0][0] for o in opts.channel_name])
else:
# Format and add the process information:
from glue.ligolw.utils import process
#from pylal.xlal.datatypes.ligotimegps import LIGOTimeGPS
#formated_cvs_gps = LIGOTimeGPS(proc_cvs_entry_time)
#from pylal.date import XLALGPSToUTC
#cvs_time_tuple=XLALGPSToUTC(formated_cvs_gps)
#proc_cvs_entry_time="%d-%d-%d %d:%d:%d +0000" %(cvs_time_tuple[0],cvs_time_tuple[1],cvs_time_tuple[2],cvs_time_tuple[3],cvs_time_tuple[4],cvs_time_tuple[5])
#append_process_gpssane
#proc_out=process.append_process(doc,program=proc_program,version=proc_version,cvs_repository=proc_cvs_repository,cvs_entry_time=proc_cvs_entry_time,comment=proc_comment,is_online=proc_is_online,jobid=proc_jobid,domain=proc_domain,ifos=proc_ifos)
proc_out=append_process_gpssane(doc,program=proc_program,version=proc_version,cvs_repository=proc_cvs_repository,cvs_entry_time=proc_cvs_entry_time,comment=proc_comment,is_online=proc_is_online,jobid=proc_jobid,domain=proc_domain,ifos=proc_ifos)
params=process.process_params_from_dict({"start_time": proc_start_time, "end_time": proc_end_time, "creator_db":proc_creator_db, "node":proc_node,"username":proc_username,"unix_procid":proc_unix_procid,"param_set":proc_param_set, "insertion_time":proc_insertion_time})
process_doc_out=process.append_process_params(doc,proc_out,params)
# Now add segment definer to doc
# Fix!!! Why didn't I just call segmentdb_utils.add_to_segment_definer? This is that code duplicated:
from glue.ligolw import lsctables
seg_def_table = lsctables.New(lsctables.SegmentDefTable,columns = ['segment_def_id', 'process_id','ifos','name', 'version','comment'])
# adds this table to the doc:
doc.childNodes[0].appendChild(seg_def_table)
# creates (local) table data container to be appended to table we just made
segment_definer = lsctables.SegmentDef()
# Read the datas and such
#
ifos = list(set([c[:2] for c in channel_cond.keys()]))
cache = Cache.fromfile(open(opts.frame_cache))
seg = cache.to_segmentlistdict()[ifos[0][0]][0]
if opts.verbose:
print "Loaded %s, total coverage time: %f" % (opts.frame_cache, abs(seg))
#
# Set up the XML document
#
xmldoc = ligolw.Document()
xmldoc.appendChild(ligolw.LIGO_LW())
# Append the process information
procrow = utils.process.append_process(xmldoc, program=sys.argv[0])
utils.process.append_process_params(xmldoc, procrow, process.process_params_from_dict(opts.__dict__))
#
# Segment storage
#
lwsegs = lw_segs.LigolwSegments(xmldoc)
all_segs = {}
#
# Loop over each channel, applying the set of conditions appropriate to it
#
for channel, opthresholds in channel_cond.iteritems():
#
# Can't have both an equality as well as an inequality via threshold
#
if len(opthresholds) > 1 and any([op == "==" for (op, t) in opthresholds]):
if opts.save_ellipsoid_data:
np.savetxt('ellipsoid.dat', test)
# Convert to m1, m2
m1m2_grid = np.array([
lsu.m1m2(cart_grid[i][0], cart_grid[i][1]) for i in xrange(len(cart_grid))
])
m1m2_grid /= lal.MSUN_SI
if opts.mass_points_xml:
xmldoc = ligolw.Document()
xmldoc.appendChild(ligolw.LIGO_LW())
procrow = process.append_process(xmldoc, program=sys.argv[0])
procid = procrow.process_id
process.append_process_params(
xmldoc, procrow, process.process_params_from_dict(opts.__dict__))
sim_insp_tbl = lsctables.New(lsctables.SimInspiralTable, [
"simulation_id", "process_id", "numrel_data", "mass1", "mass2", "psi0",
"psi3"
])
for itr, (m1, m2) in enumerate(m1m2_grid):
for l1 in np.linspace(common_cl.param_limits["lam_tilde"][0],
common_cl.param_limits["lam_tilde"][1], Nlam):
sim_insp = sim_insp_tbl.RowType()
sim_insp.numrel_data = "MASS_SET_%d" % itr
sim_insp.simulation_id = ilwd.ilwdchar(
"sim_inspiral:sim_inspiral_id:%d" % itr)
sim_insp.process_id = procid
sim_insp.mass1, sim_insp.mass2 = m1, m2
sim_insp.psi0, sim_insp.psi3 = opts.eff_lambda or l1, opts.delta_eff_lambda or 0
