tmag, lbol, mag, Tobs = arnett_model_ejecta(
mej_best, vej_best, beta_best, kappa_r_best)
tmag = tmag + t0_best
else:
print "Not implemented..."
exit(0)
model_data[model] = {}
model_data[model]["tmag"] = tmag
model_data[model]["lbol"] = lbol
model_data[model]["mag"] = mag
model_data[model]["Tobs"] = Tobs
if opts.doEvent:
filename = "%s/%s.dat" % (lightcurvesDir, opts.event)
data_out = lightcurve_utils.loadEvent(filename)
for ii, key in enumerate(data_out.iterkeys()):
if key == "t":
continue
else:
data_out[key][:, 0] = data_out[key][:, 0] - opts.T0
data_out[key][:, 1] = data_out[key][:, 1] - 5 * (
np.log10(opts.distance * 1e6) - 1)
#colors = ["g","r","c","y","m"]
colors = cm.rainbow(np.linspace(0, 1, len(names)))
plotName = "%s/models.pdf" % (plotDir)
plt.figure(figsize=(10, 8))
for ii, name in enumerate(names):
mag_d = mags[name]
indexes = np.where(~np.isnan(mag_d["g"]))[0]
lightcurvesDir = opts.lightcurvesDir
filename = "%s/GRB.dat" % lightcurvesDir
lines = [line.rstrip('\n') for line in open(filename)]
lightcurveFiles = glob.glob(os.path.join(lightcurvesDir, 'GRB*.dat')) + [
os.path.join(lightcurvesDir, 'GW170817.dat')
]
data = {}
data_peak = {}
data_peak_app = {}
for lightcurveFile in lightcurveFiles:
grbname = lightcurveFile.split('/')[-1].replace('.dat', '')
if grbname == "GRB": continue
data_out = lightcurve_utils.loadEvent(lightcurveFile)
data[grbname] = data_out
if grbname == "GW170817":
grb, filts, mjd, dist = "GW170817", "r", 57982.5285236896, 40.0
else:
for line in lines:
lineSplit = line.split(" ")
if lineSplit[0] == grbname:
grb = lineSplit[0]
filts = lineSplit[1]
mjd = float(lineSplit[2])
dist = float(lineSplit[3])
break
print(grbname)
