Source code for gwemlightcurves.KNModels.io.RoFe2017

# https://arxiv.org/abs/1705.07084

import os, sys, glob
import numpy as np
import scipy.interpolate
from scipy.interpolate import interpolate as interp
from scipy.interpolate import griddata

from .model import register_model
from .. import KNTable

from gwemlightcurves import lightcurve_utils, Global, svd_utils
from gwemlightcurves.EjectaFits.DiUj2017 import calc_meje, calc_vej

[docs]def get_RoFe2017_model(table, **kwargs): if not 'n_coeff' in table.colnames: table['n_coeff'] = 100 if not Global.svd_mag_model == 0: svd_mag_model = Global.svd_mag_model else: svd_mag_model = svd_utils.calc_svd_mag(table['tini'][0], table['tmax'][0], table['dt'][0], model = "RoFe2017", n_coeff = table['n_coeff'][0]) Global.svd_mag_model = svd_mag_model if not Global.svd_lbol_model == 0: svd_lbol_model = Global.svd_lbol_model else: svd_lbol_model = svd_utils.calc_svd_lbol(table['tini'][0], table['tmax'][0], table['dt'][0], model = "RoFe2017", n_coeff = table['n_coeff'][0]) Global.svd_lbol_model = svd_lbol_model if not 'mej' in table.colnames: # calc the mass of ejecta table['mej'] = calc_meje(table['m1'], table['mb1'], table['c1'], table['m2'], table['mb2'], table['c2']) # calc the velocity of ejecta table['vej'] = calc_vej(table['m1'], table['c1'], table['m2'], table['c2']) # Throw out smaples where the mass ejecta is less than zero. mask = (table['mej'] > 0) table = table[mask] if len(table) == 0: return table # Log mass ejecta table['mej10'] = np.log10(table['mej']) # Initialize lightcurve values in table timeseries = np.arange(table['tini'][0], table['tmax'][0]+table['dt'][0], table['dt'][0]) table['t'] = [np.zeros(timeseries.size)] table['lbol'] = [np.zeros(timeseries.size)] table['mag'] = [np.zeros([9, timeseries.size])] # calc lightcurve for each sample for isample in range(len(table)): table['t'][isample], table['lbol'][isample], table['mag'][isample] = svd_utils.calc_lc(table['tini'][isample], table['tmax'][isample],table['dt'][isample], [np.log10(table['mej'][isample]),table['vej'][isample],table['Ye'][isample]],svd_mag_model = svd_mag_model, svd_lbol_model = svd_lbol_model, model = "RoFe2017") return table
register_model('RoFe2017', KNTable, get_RoFe2017_model, usage="table")