def transform_m1m2_mceta(m1, m2): return lalsimutils.Mceta(m1, m2)
required_params = ("mass1", "mass2", "spin1x", "spin1y", "spin1z", "spin2x", "spin2y", "spin2z") # lalapps_tmpltbank assigns 0 ID to all events, so we remap # FIXME: Check for tmplt_bank: All others do assign IDs for tmplt in tmplt_bank: if not args.no_overwrite_id: try: tmplt.event_id = tmplt_bank.get_next_id() except AttributeError: tmplt.simulation_id = tmplt_bank.get_next_id() for p in required_params: if not hasattr(tmplt, p): setattr(tmplt, p, 0.0) if not hasattr(tmplt, "mchirp"): tmplt.mchirp, tmplt.eta = lalsimutils.Mceta(tmplt.mass1, tmplt.mass2) # FIXME: Unhardcode wtype = "%s_%s" % (args.approximant1, args.approximant2) toc = {"types": {wtype: []}} bdir = "%s/" % wtype if not os.path.exists(bdir): os.mkdir(bdir) # FIXME: This code should probably be libized # FIXME: Unhardcode intr_prms = ("mass1", "mass2", "spin1z", "spin2z") pts = numpy.array( [tuple(getattr(t, a) for a in intr_prms) for t in tmplt_bank]) # Dump full m1/m2 bank to JSON