def rapidpe_to_hdf(basegrp, bankfile, samplefiles): # import (or look up template bank) #bank_xmldoc = utils.load_filename(args.tmplt_bank_file, contenthandler=ligolw.LIGOLWContentHandler) bank_xmldoc = utils.load_filename(bankfile, contenthandler=ligolw.LIGOLWContentHandler) try: tmplt_bank = lsctables.SimInspiralTable.get_table(bank_xmldoc) except: tmplt_bank = lsctables.SnglInspiralTable.get_table(bank_xmldoc) for sample_file in samplesfiles: print "Processing %s" % sample_file # Get tables from xmldoc xmldoc = utils.load_filename(sample_file, contenthandler=ligolw.LIGOLWContentHandler) process_params = lsctables.ProcessParamsTable.get_table(xmldoc) samples = lsctables.SimInspiralTable.get_table(xmldoc) run_result = lsctables.SnglInspiralTable.get_table(xmldoc)[0] # Get intrinsic parameter used intr_prms = get_intr_prms_from_pp_table(process_params) # Identify intrinisic ID tmplt_id = obtain_tmplt_id(tmplt_bank, **intr_prms) # Append sample data and metadata subgrp = base_grp.create_group(str(tmplt_id)) xmlutils.append_samples_to_hdf5_group(subgrp, samples) def pp_table_to_dict(pptable): return dict([(pp.param, pp.value) for pp in pptable]) run_info = pp_table_to_dict(process_params) for key in ("snr", "tau0", "ttotal"): run_info[KEY_MAPPING[key]] = getattr(run_result, key) xmlutils.append_metadata_to_hdf5_group(subgrp, run_info) xmldoc.unlink()
def txt_to_hdf(basegrp, samplefile): samplefile = samplefile[0] print "Processing %s" % samplefile samples = numpy.genfromtxt(samplefile, names=True) # Append sample data and metadata #subgrp = base_grp.create_group(str(tmplt_id)) subgrp = base_grp.create_group("run_identifier") append_lalinf_samples_to_hdf5_group(subgrp, samples) return def pp_table_to_dict(pptable): return dict([(pp.param, pp.value) for pp in pptable]) run_info = pp_table_to_dict(process_params) for key in ("snr", "tau0", "ttotal"): run_info[KEY_MAPPING[key]] = getattr(run_result, key) xmlutils.append_metadata_to_hdf5_group(subgrp, run_info) xmldoc.unlink()