redback 1.0.1__py3-none-any.whl → 1.0.3__py3-none-any.whl

redback/model_library.py

@@ -1,13 +1,13 @@
 from redback.transient_models import afterglow_models, \
     extinction_models, kilonova_models, fireball_models, \
     gaussianprocess_models, magnetar_models, magnetar_driven_ejecta_models, phase_models, phenomenological_models, \
-    prompt_models, shock_powered_models, supernova_models, tde_models, integrated_flux_afterglow_models, combined_models
+    prompt_models, shock_powered_models, supernova_models, tde_models, integrated_flux_afterglow_models, combined_models, general_synchrotron_models
 from redback.utils import get_functions_dict
 
 modules = [afterglow_models, extinction_models, fireball_models,
            gaussianprocess_models,  integrated_flux_afterglow_models, kilonova_models,
            magnetar_models, magnetar_driven_ejecta_models,
-           phase_models, phenomenological_models, prompt_models, shock_powered_models, supernova_models, tde_models, combined_models]
+           phase_models, phenomenological_models, prompt_models, shock_powered_models, supernova_models, tde_models, combined_models, general_synchrotron_models]
 
 all_models_dict = dict()
 modules_dict = dict()

redback/plotting.py

@@ -346,7 +346,7 @@ class IntegratedFluxPlotter(Plotter):
                 axes.plot(times, ys, color=self.random_sample_color, alpha=self.random_sample_alpha, lw=self.linewidth,
                           zorder=self.zorder)
         elif self.uncertainty_mode == "credible_intervals":
-            lower_bound, upper_bound, _ = redback.utils.calc_credible_intervals(samples=random_ys_list)
+            lower_bound, upper_bound, _ = redback.utils.calc_credible_intervals(samples=random_ys_list, interval=self.credible_interval_level)
             axes.fill_between(
                 times, lower_bound, upper_bound, alpha=self.uncertainty_band_alpha, color=self.max_likelihood_color)
 
@@ -391,11 +391,11 @@ class LuminosityPlotter(IntegratedFluxPlotter):
 
 class MagnitudePlotter(Plotter):
 
-    xlim_low_phase_model_multiplier = 0.9
-    xlim_high_phase_model_multiplier = 1.1
-    xlim_high_multiplier = 1.2
-    ylim_low_magnitude_multiplier = 0.8
-    ylim_high_magnitude_multiplier = 1.2
+    xlim_low_phase_model_multiplier = KwargsAccessorWithDefault("xlim_low_multiplier", 0.9)
+    xlim_high_phase_model_multiplier = KwargsAccessorWithDefault("xlim_high_multiplier", 1.1)
+    xlim_high_multiplier = KwargsAccessorWithDefault("xlim_high_multiplier", 1.2)
+    ylim_low_magnitude_multiplier = KwargsAccessorWithDefault("ylim_low_multiplier", 0.8)
+    ylim_high_magnitude_multiplier = KwargsAccessorWithDefault("ylim_high_multiplier", 1.2)
     ncols = KwargsAccessorWithDefault("ncols", 2)
 
     @property
@@ -635,11 +635,11 @@ class MagnitudePlotter(Plotter):
             elif self.uncertainty_mode == "credible_intervals":
                 if band in self.band_scaling:
                     if self.band_scaling.get("type") == 'x':
-                        lower_bound, upper_bound, _ = redback.utils.calc_credible_intervals(samples=np.array(random_ys_list) * self.band_scaling.get(band))
+                        lower_bound, upper_bound, _ = redback.utils.calc_credible_intervals(samples=np.array(random_ys_list) * self.band_scaling.get(band), interval=self.credible_interval_level)
                     elif self.band_scaling.get("type") == '+':
-                        lower_bound, upper_bound, _ = redback.utils.calc_credible_intervals(samples=np.array(random_ys_list) + self.band_scaling.get(band))
+                        lower_bound, upper_bound, _ = redback.utils.calc_credible_intervals(samples=np.array(random_ys_list) + self.band_scaling.get(band), interval=self.credible_interval_level)
                 else:
-                    lower_bound, upper_bound, _ = redback.utils.calc_credible_intervals(samples=np.array(random_ys_list))
+                    lower_bound, upper_bound, _ = redback.utils.calc_credible_intervals(samples=np.array(random_ys_list), interval=self.credible_interval_level)
                 axes.fill_between(
                     times - self._reference_mjd_date, lower_bound, upper_bound,
                     alpha=self.uncertainty_band_alpha, color=color_sample)
@@ -789,7 +789,7 @@ class MagnitudePlotter(Plotter):
                     axes[ii].plot(times - self._reference_mjd_date, random_ys, color=color_sample,
                                   alpha=self.random_sample_alpha, lw=self.linewidth, zorder=self.zorder)
             elif self.uncertainty_mode == "credible_intervals":
-                lower_bound, upper_bound, _ = redback.utils.calc_credible_intervals(samples=random_ys_list)
+                lower_bound, upper_bound, _ = redback.utils.calc_credible_intervals(samples=random_ys_list, interval=self.credible_interval_level)
                 axes[ii].fill_between(
                     times - self._reference_mjd_date, lower_bound, upper_bound,
                     alpha=self.uncertainty_band_alpha, color=color_sample)

redback/priors/bazin_sne.prior

@@ -0,0 +1,5 @@
+aa = LogUniform(1e-15,1e15,name='aa', latex_label = r'$A$')
+bb = LogUniform(1e-15,1e15,name='bb', latex_label = r'$B$')
+t0 = Uniform(0,50,name='t0', latex_label = r'$t_0$')
+tau_rise = Uniform(0.01, 50, name='tau_rise', latex_label = r'$\\tau_{\\mathrm{rise}}$')
+tau_fall = Uniform(1, 300, name='tau_fall', latex_label = r'$\\tau_{\\mathrm{fall}}$')

redback/priors/csm_interaction.prior

@@ -1,10 +1,9 @@
 redshift = Uniform(1e-3,3,name='redshift', latex_label = r'$z$')
-mej = LogUniform(1e-4, 100, 'mej', latex_label = r'$M_{\mathrm{ej} }~(M_\odot)$')
-csm_mass = LogUniform(1e-4, 100, 'csm_mass', latex_label = r'$M_{\mathrm{csm}}~(M_\odot)$')
+mej = LogUniform(1, 100, 'mej', latex_label = r'$M_{\mathrm{ej} }~(M_\odot)$')
+csm_mass = LogUniform(1, 100, 'csm_mass', latex_label = r'$M_{\mathrm{csm}}~(M_\odot)$')
 vej = LogUniform(1e3, 1e5, 'vej', latex_label = r'$v_{\mathrm{ej}}~(\mathrm{km}/\mathrm{s})$')
-eta = Uniform(0, 1, '\\eta', latex_label = r'$\\eta$')
-rho = Uniform(8, 12, 'rho', latex_label = r'$\\rho$')
-r0 = Uniform(4, 8, 'r0', latex_label=r'$r_0~({\mathrm{AU}})$')
+eta = Uniform(0, 2, '\\eta', latex_label = r'$\\eta$')
+rho = LogUniform(1e-14, 1e-12, 'rho', latex_label = r'$\\rho$')
+r0 = Uniform(50, 700, 'r0', latex_label=r'$r_0~({\mathrm{AU}})$')
 kappa = Uniform(0.05, 2, 'kappa', latex_label = r'$\\kappa~(\mathrm{cm}^{2}/\mathrm{g})$')
-kappa_gamma = LogUniform(1e-4, 1e4, 'kappa_gamma', latex_label = r'$\\kappa_{\\gamma}~(\mathrm{cm}^{2}/\mathrm{g})$')
-temperature_floor = LogUniform(1e3,1e5,name = 'temperature_floor', latex_label = r'$T_{\mathrm{floor}}~(\mathrm{K})$')
+temperature_floor = LogUniform(100,1e4,name = 'temperature_floor', latex_label = r'$T_{\mathrm{floor}}~(\mathrm{K})$')

redback/priors/csm_nickel.prior

@@ -3,9 +3,9 @@ mej = LogUniform(1e-4, 100, 'mej', latex_label = r'$M_{\mathrm{ej} }~(M_\odot)$'
 f_nickel = LogUniform(1e-3,1,name='f_nickel', latex_label = r'$f_{\mathrm{Ni}}$')
 csm_mass = LogUniform(1e-4, 100, 'csm_mass', latex_label = r'$M_{\mathrm{csm}}~(M_\odot)$')
 ek = LogUniform(1e48, 1e52, 'ek', latex_label = r'$E_{\mathrm{kin}}~(\mathrm{ erg})$')
-eta = Uniform(0, 1, '\\eta', latex_label = r'$\\eta$')
-rho = Uniform(8, 12, 'rho', latex_label = r'$\\rho$')
+eta = Uniform(0, 2, '\\eta', latex_label = r'$\\eta$')
+rho = LogUniform(1e-15, 1e-12, 'rho', latex_label = r'$\\rho$')
 r0 = Uniform(4, 8, 'r0', latex_label=r'$r_0~({\mathrm{AU}})$')
 kappa = Uniform(0.05, 2, 'kappa', latex_label = r'$\\kappa~(\mathrm{cm}^{2}/\mathrm{g})$')
 kappa_gamma = LogUniform(1e-4, 1e4, 'kappa_gamma', latex_label = r'$\\kappa_{\\gamma}~(\mathrm{cm}^{2}/\mathrm{g})$')
-temperature_floor = LogUniform(1e3,1e5,name = 'temperature_floor', latex_label = r'$T_{\mathrm{floor}}~(\mathrm{K})$')
+temperature_floor = LogUniform(100,1e4,name = 'temperature_floor', latex_label = r'$T_{\mathrm{floor}}~(\mathrm{K})$')

redback/priors/csm_shock_and_arnett.prior

@@ -0,0 +1,11 @@
+redshift = Uniform(0.01, 1, 'redshift', latex_label=r'$z$')
+mej = LogUniform(1e-4, 30, 'mej', latex_label = r'$M_{\mathrm{ej}}~(M_\odot)$')
+f_nickel = LogUniform(1e-3,1,name='f_nickel', latex_label = r'$f_{\mathrm{Ni}}$')
+csm_mass = LogUniform(1e-3, 5, 'csm_mass', latex_label=r'$M_{\mathrm{CSM}}~(M_{\odot})$')
+v_min = LogUniform(1e3, 3e4, 'vej', latex_label = r'$v_{\mathrm{ej}}~(\mathrm{km}/\mathrm{s})$')
+beta = Uniform(0.4, 0.5, 'beta', latex_label=r'$\\beta$')
+kappa = Uniform(0.01, 1, 'kappa', latex_label = r'$\\kappa~(\mathrm{cm}^{2}/\mathrm{g})$')
+shell_radius = Uniform(1e-2, 10, 'shell_radius', latex_label=r'$R_{\mathrm{shell}}~(10^{14}~\mathrm{cm})$')
+shell_width_ratio = Uniform(0.1, 0.5, 'shell_width_ratio', latex_label=r'$\\Delta~R_{\mathrm{shell}}/R_{\mathrm{shell}}$')
+kappa_gamma = LogUniform(1e-4, 1e4, 'kappa_gamma', latex_label = r'$\\kappa_{\\gamma}~(\mathrm{cm}^{2}/\mathrm{g})$')
+temperature_floor = LogUniform(1e3,1e5,name = 'temperature_floor', latex_label = r'$T_{\mathrm{floor}}~(\mathrm{K})$')

redback/priors/csm_shock_and_arnett_bolometric.prior

@@ -0,0 +1,10 @@
+mej = LogUniform(1e-4, 30, 'mej', latex_label = r'$M_{\mathrm{ej}}~(M_\odot)$')
+f_nickel = LogUniform(1e-3,1,name='f_nickel', latex_label = r'$f_{\mathrm{Ni}}$')
+csm_mass = LogUniform(1e-3, 5, 'csm_mass', latex_label=r'$M_{\mathrm{CSM}}~(M_{\odot})$')
+v_min = LogUniform(1e3, 3e4, 'vej', latex_label = r'$v_{\mathrm{ej}}~(\mathrm{km}/\mathrm{s})$')
+beta = Uniform(0.4, 0.5, 'beta', latex_label=r'$\\beta$')
+kappa = Uniform(0.01, 1, 'kappa', latex_label = r'$\\kappa~(\mathrm{cm}^{2}/\mathrm{g})$')
+shell_radius = Uniform(1e-2, 10, 'shell_radius', latex_label=r'$R_{\mathrm{shell}}~(10^{14}~\mathrm{cm})$')
+shell_width_ratio = Uniform(0.1, 0.5, 'shell_width_ratio', latex_label=r'$\\Delta~R_{\mathrm{shell}}/R_{\mathrm{shell}}$')
+kappa_gamma = LogUniform(1e-4, 1e4, 'kappa_gamma', latex_label = r'$\\kappa_{\\gamma}~(\mathrm{cm}^{2}/\mathrm{g})$')
+temperature_floor = LogUniform(1e3,1e5,name = 'temperature_floor', latex_label = r'$T_{\mathrm{floor}}~(\mathrm{K})$')

redback/priors/csm_shock_breakout.prior

@@ -0,0 +1,7 @@
+redshift = Uniform(0.01, 1, 'redshift', latex_label=r'$z$')
+csm_mass = LogUniform(1e-3, 2, 'csm_mass', latex_label=r'$M_{\mathrm{CSM}}~(M_{\odot})$')
+v_min = LogUniform(1e3, 3e4, 'vej', latex_label = r'$v_{\mathrm{ej}}~(\mathrm{km}/\mathrm{s})$')
+beta = Uniform(0.4, 0.5, 'beta', latex_label=r'$\\beta$')
+kappa = Uniform(0.01, 1, 'kappa', latex_label = r'$\\kappa~(\mathrm{cm}^{2}/\mathrm{g})$')
+shell_radius = Uniform(1e-2, 10, 'shell_radius', latex_label=r'$R_{\mathrm{shell}}~(10^{14}~\mathrm{cm})$')
+shell_width_ratio = Uniform(0.1, 0.5, 'shell_width_ratio', latex_label=r'$\\Delta~R_{\mathrm{shell}}/R_{\mathrm{shell}}$')

redback/priors/nicholl_bns.prior

@@ -8,7 +8,8 @@ mtov = Uniform(2.05, 2.4, 'mtov', latex_label = r'$M_{\mathrm{TOV}}~(M_\odot)$')
 epsilon = Uniform(0.05, 0.6, 'zeta', latex_label = r'$\\epsilon$')
 alpha = Uniform(0.1, 1, 'alpha', latex_label = r'$\\alpha$')
 cos_theta = Uniform(0.0, 1.0, 'cos_theta', latex_label = r'$\cos~\\theta$')
-cos_theta_cocoon = Uniform(0.707, 1.0, 'cos_theta_cocoon', latex_label = r'$\cos~\\theta_{\mathrm{cocoon}}$')
+cos_theta_open = Uniform(0.707, 0.866, 'cos_theta_open', latex_label = r'$\cos~\\theta_{\mathrm{open}}$')
+cos_theta_cocoon = Uniform(0.866, 1.0, 'cos_theta_cocoon', latex_label = r'$\cos~\\theta_{\mathrm{cocoon}}$')
 temperature_floor_1 = LogUniform(100, 6000, 'temperature_floor_1', latex_label = r'$T_{\mathrm{floor}~1}$ (K)')
 temperature_floor_2 = LogUniform(100, 6000, 'temperature_floor_2', latex_label = r'$T_{\mathrm{floor}~2}$ (K)')
 temperature_floor_3 = LogUniform(100, 6000, 'temperature_floor_3', latex_label = r'$T_{\mathrm{floor}~3}$ (K)')

redback/priors/one_comp_kne_rosswog_heatingrate.prior

@@ -0,0 +1,5 @@
+redshift = Uniform(1e-6, 0.1, 'redshift', latex_label = r'$z$')
+mej = Uniform(1e-2, 0.05, 'mej', latex_label = r'$M_{\mathrm{ej} }~(M_\odot)$')
+vej = Uniform(0.05, 0.3, 'vej', latex_label = r'$v_{\mathrm{ej}}~(c)$')
+ye = Uniform(0.05, 0.4, 'ye', latex_label = r'$Y_{e}$')
+temperature_floor = LogUniform(100, 6000, 'temperature_floor', latex_label = r'$T_{\mathrm{floor}}~(\mathrm{K})$')

redback/priors/pwn.prior

@@ -0,0 +1,7 @@
+redshift = Uniform(1e-6,3,name='redshift', latex_label = r'$z$')
+l0 = LogUniform(1e40, 1e50, 'l0', latex_label = r'$L_{\mathrm{0}} [erg s^{-1}]$')
+tau_sd = LogUniform(1e2, 1e8, 'tau_sd', latex_label = r'$t_{\mathrm{SD}} [s]$')
+nn = LogUniform(1.5, 10, 'nn', latex_label = r'$n$')
+mej = LogUniform(1e-1, 100, 'mej', latex_label = r'$M_{\mathrm{ej}} [M_{\odot}]$')
+eps_b = LogUniform(1e-7, 1, 'eps_b', latex_label = r'$\epsilon_{\mathrm{B}}$')
+gamma_b = LogUniform(1e2, 1e7, 'gamma_b', latex_label = r'$\gamma_{\mathrm{b}}$')

redback/priors/shocked_cocoon.prior

@@ -1,8 +1,8 @@
 redshift = Uniform(1e-3,3,name='redshift', latex_label = r'$z$')
-mej = Uniform(1e-4, 0.1, 'mej', latex_label = r'$M_{\mathrm{ej} }~(M_\odot)$')
-vej = Uniform(0.1, 0.7, 'vej', latex_label = r'$v_{\mathrm{ej}}~(c)$')
+mej = Uniform(1e-4, 0.2, 'mej', latex_label = r'$M_{\mathrm{ej} }~(M_\odot)$')
+vej = Uniform(0.01, 0.7, 'vej', latex_label = r'$v_{\mathrm{ej}}~(c)$')
 eta = Uniform(1, 5, '\\eta', latex_label = r'$\\eta$')
-tshock = Uniform(1, 2, 'tshock', latex_label = r'$t_{\mathrm{shock}}~(\mathrm{s})$')
-shocked_fraction = Uniform(0.1, 1, 'shocked_fraction', latex_label = r'$f_{\mathrm{shocked}}$')
-cos_theta_cocoon = Uniform(0.2, 0.9, 'cos_theta_cocoon', latex_label = r'$\cos~\\theta_{\mathrm{cocoon}}$')
-kappa = Uniform(1, 30, 'kappa', latex_label = r'$\\kappa~(\mathrm{cm}^{2}/\mathrm{g})$')
+tshock = Uniform(0.1, 10, 'tshock', latex_label = r'$t_{\mathrm{shock}}~(\mathrm{s})$')
+shocked_fraction = Uniform(0.01, 1, 'shocked_fraction', latex_label = r'$f_{\mathrm{shocked}}$')
+cos_theta_cocoon = Uniform(0.866, 1.0, 'cos_theta_cocoon', latex_label = r'$\cos~\\theta_{\mathrm{cocoon}}$')
+kappa = Uniform(0.1, 30, 'kappa', latex_label = r'$\\kappa~(\mathrm{cm}^{2}/\mathrm{g})$')

redback/priors/sn_fallback.prior

@@ -0,0 +1,8 @@
+redshift = Uniform(1e-3,3,name='redshift', latex_label = r'$z$')
+logl1 = Uniform(51, 57, name='logl1', latex_label=r'$\log_{10}L_{1} [erg~s^{-1}]$')
+tr = LogUniform(1e-4, 100, name='tr', latex_label=r'T_{r} [day]')
+mej = LogUniform(1e-4, 100, 'mej', latex_label = r'$M_{\mathrm{ej}}~(M_\odot)$')
+vej = LogUniform(1e3, 1e5, 'vej', latex_label = r'$v_{\mathrm{ej}}~(\mathrm{km}/\mathrm{s})$')
+kappa = Uniform(0.05, 2, 'kappa', latex_label = r'$\\kappa~(\mathrm{cm}^{2}/\mathrm{g})$')
+kappa_gamma = LogUniform(1e-4, 1e4, 'kappa_gamma', latex_label = r'$\\kappa_{\\gamma}~(\mathrm{cm}^{2}/\mathrm{g})$')
+temperature_floor = LogUniform(1e3,1e5,name = 'temperature_floor', latex_label = r'$T_{\mathrm{floor}}~(\mathrm{K})$')

redback/priors/stream_stream_tde.prior

@@ -0,0 +1,10 @@
+redshift = Uniform(1e-6, 3, 'redshift', latex_label = r'$z$')
+peak_time = LogUniform(0.1,60, name='peak_time', latex_label = r'$t_{\mathrm{peak}}~(\mathrm{day})$')
+sigma_t = LogUniform(10,60, name='sigma_t', latex_label = r'$\\sigma~(\mathrm{day})$')
+mbh_6 = LogUniform(0.01, 20, name='mbh_6', latex_label = r'$M_{\mathrm{BH}}~(10^{6}~M_\odot)$')
+mstar = LogUniform(0.1, 10, name='stellar_mass', latex_label = r'$M_{\mathrm{star}}~(M_\odot)$')
+c1 = 1.0
+del_omega = 2.0
+f = 0.2
+h_r = 0.2
+inc_tcool = 1

redback/priors/stream_stream_tde_bolometric.prior

@@ -0,0 +1,9 @@
+peak_time = LogUniform(0.1,60, name='peak_time', latex_label = r'$t_{\mathrm{peak}}~(\mathrm{day})$')
+sigma_t = LogUniform(10,60, name='sigma_t', latex_label = r'$\\sigma~(\mathrm{day})$')
+mbh_6 = LogUniform(0.01, 20, name='mbh_6', latex_label = r'$M_{\mathrm{BH}}~(10^{6}~M_\odot)$')
+mstar = LogUniform(0.1, 10, name='stellar_mass', latex_label = r'$M_{\mathrm{star}}~(M_\odot)$')
+c1 = 1.0
+del_omega = 2.0
+f = 0.2
+h_r = 0.2
+inc_tcool = 1

redback/priors/tde_analytical.prior

@@ -1,8 +1,8 @@
 redshift = Uniform(0.01, 3, 'redshift', latex_label=r'$z$')
-mej = LogUniform(1e-4, 100, 'mej', latex_label = r'$M_{\mathrm{ej} }~(M_\odot)$')
+mej = LogUniform(0.1, 100, 'mej', latex_label = r'$M_{\mathrm{ej} }~(M_\odot)$')
 vej = LogUniform(1e3, 1e5, 'vej', latex_label = r'$v_{\mathrm{ej}}~(\mathrm{km}/\mathrm{s})$')
-kappa = Uniform(0.05, 2, 'kappa', latex_label = r'$\\kappa~(\mathrm{cm}^{2}/\mathrm{g})$')
+kappa = Uniform(0.05, 1e2, 'kappa', latex_label = r'$\\kappa~(\mathrm{cm}^{2}/\mathrm{g})$')
 kappa_gamma = LogUniform(1e-4, 1e4, 'kappa_gamma', latex_label = r'$\\kappa_{\\gamma}~(\mathrm{cm}^{2}/\mathrm{g})$')
-temperature_floor = LogUniform(1e3,1e5,name = 'temperature_floor', latex_label = r'$T_{\mathrm{floor}}~(\mathrm{K})$')
-l0 = LogUniform(1e51, 1e58, "l0", latex_label="$l_0$")
-t_0_turn = LogUniform(1e-4, 1e2, "t_0_turn", latex_label="$t_{0 t}$")
+temperature_floor = LogUniform(1e3,1e4,name = 'temperature_floor', latex_label = r'$T_{\mathrm{floor}}~(\mathrm{K})$')
+l0 = LogUniform(1e51, 1e60, "l0", latex_label="$l_0$")
+t_0_turn = LogUniform(1e-4, 5e2, "t_0_turn", latex_label="$t_{0 t}$")

redback/priors/tde_analytical_bolometric.prior

@@ -1,6 +1,8 @@
-mej = LogUniform(1e-4, 100, 'mej', latex_label = r'$M_{\mathrm{ej} }~(M_\odot)$')
+redshift = Uniform(0.01, 3, 'redshift', latex_label=r'$z$')
+mej = LogUniform(0.1, 100, 'mej', latex_label = r'$M_{\mathrm{ej} }~(M_\odot)$')
 vej = LogUniform(1e3, 1e5, 'vej', latex_label = r'$v_{\mathrm{ej}}~(\mathrm{km}/\mathrm{s})$')
-kappa = Uniform(0.05, 2, 'kappa', latex_label = r'$\\kappa~(\mathrm{cm}^{2}/\mathrm{g})$')
+kappa = Uniform(0.05, 1e2, 'kappa', latex_label = r'$\\kappa~(\mathrm{cm}^{2}/\mathrm{g})$')
 kappa_gamma = LogUniform(1e-4, 1e4, 'kappa_gamma', latex_label = r'$\\kappa_{\\gamma}~(\mathrm{cm}^{2}/\mathrm{g})$')
-l0 = LogUniform(1e51, 1e58, "l0", latex_label="$l_0$")
-t_0_turn = LogUniform(1e-4, 1e2, "t_0_turn", latex_label="$t_{0 t}$")
+temperature_floor = LogUniform(1e3,1e4,name = 'temperature_floor', latex_label = r'$T_{\mathrm{floor}}~(\mathrm{K})$')
+l0 = LogUniform(1e51, 1e60, "l0", latex_label="$l_0$")
+t_0_turn = LogUniform(1e-4, 5e2, "t_0_turn", latex_label="$t_{0 t}$")

redback/priors/tde_fallback.prior

@@ -0,0 +1,9 @@
+redshift = Uniform(1e-6, 3, 'redshift', latex_label = r'$z$')
+mbh6 = Uniform(0.01, 20, name='mbh_6', latex_label = r'$M_{\mathrm{BH}}~[10^{6}~M_{\odot}]$')
+mstar = Uniform(0.1, 30, name='mstar', latex_label = r'$M_{\mathrm{star}} [M_{\odot}]$')
+tvisc = LogUniform(1e-3, 1e2, name='tvisc', latex_label = r'$t_{\mathrm{visc}}$')
+bb = Uniform(0, 2, name='bb', latex_label = r'$\\beta^{\star}$')
+eta = LogUniform(1e-4, 0.4, name='eta', latex_label=r'$\eta$')
+leddlimit = 1
+rph0 = LogUniform(1e-4, 1e4, name='rph_0', latex_label = r'$R_{\mathrm{ph-0}}$')
+lphoto = Uniform(0, 2, name='lphoto', latex_label = r'$L_{\mathrm{photo}}^{\star}$')

redback/priors/tde_fallback_bolometric.prior

@@ -0,0 +1,6 @@
+mbh6 = Uniform(0.01, 20, name='mbh_6', latex_label = r'$M_{\mathrm{BH}}~[10^{6}~M_{\odot}]$')
+mstar = Uniform(0.1, 30, name='mstar', latex_label = r'$M_{\mathrm{star}} [M_{\odot}]$')
+tvisc = LogUniform(1e-3, 1e2, name='tvisc', latex_label = r'$t_{\mathrm{visc}}$')
+bb = Uniform(0, 2, name='bb', latex_label = r'$\\beta^{\star}$')
+eta = LogUniform(1e-4, 0.4, name='eta', latex_label=r'$\eta$')
+leddlimit = 1

redback/priors/tde_synchrotron.prior

@@ -0,0 +1,6 @@
+redshift = Uniform(0.01, 3, 'redshift', latex_label=r'$z$')
+Mej = Uniform(0.1, 10, 'Mej', latex_label = r'$M_{\mathrm{ej}}~(M_\odot)$')
+vej = Uniform(1e3, 3e4, 'vej', latex_label = r'$v_{\mathrm{ej}}~(\mathrm{km}/\mathrm{s})$')
+logepse = Uniform(-5, 0, 'logepse', latex_label=r'$\log_{10}~\\epsilon_{e}$')
+logepsb = Uniform(-5, 0, 'logepsb', latex_label=r'$\log_{10}~\\epsilon_{B}$')
+p = Uniform(2, 3, 'p', latex_label=r'$p$')

redback/priors/tophat_from_emulator.prior

@@ -0,0 +1,9 @@
+redshift = Uniform(0.01, 3, 'redshift', latex_label=r'$z$')
+thv = Sine(name='thv', maximum=np.pi/2, latex_label=r'$\\theta_{\mathrm{observer}}~(\mathrm{rad})$')
+loge0 = Uniform(44, 54, 'loge0', latex_label=r'$\log_{10}~E_{0} / {\mathrm{erg}}$')
+thc = Uniform(0.01, 0.1, 'thc', latex_label=r'$\\theta_{\mathrm{core}}~({\mathrm{rad}})$')
+logn0 = Uniform(-5, 2, 'logn0', latex_label=r'$\log_{10}~n_{\mathrm{ism}} / {\mathrm{cm}}^{-3}$')
+p = Uniform(2, 3, 'p', latex_label=r'$p$')
+logepse = Uniform(-5, 0, 'logepse', latex_label=r'$\log_{10}~\\epsilon_{e}$')
+logepsb = Uniform(-5, 0, 'logepsb', latex_label=r'$\log_{10}~\\epsilon_{B}$')
+g0 = Uniform(100,2000, 'g0', latex_label=r'$\\Gamma_{0}$')

redback/priors/two_comp_kne_rosswog_heatingrate.prior

@@ -0,0 +1,9 @@
+redshift = Uniform(1e-6, 0.1, 'redshift', latex_label = r'$z$')
+mej_1 = Uniform(1e-2, 0.05, 'mej', latex_label = r'$M_{\mathrm{ej}~1}~(M_\odot)$')
+vej_1 = Uniform(0.05, 0.3, 'vej', latex_label = r'$v_{\mathrm{ej}~1}~(c)$')
+ye_1 = Uniform(0.05, 0.5, 'ye', latex_label = r'$Y_{e~1}$')
+temperature_floor_1 = LogUniform(100, 6000, 'temperature_floor', latex_label = r'$T_{\mathrm{floor}~1}~(\mathrm{K})$')
+mej_2 = Uniform(1e-2, 0.05, 'mej', latex_label = r'$M_{\mathrm{ej}~2}~(M_\odot)$')
+vej_2 = Uniform(0.05, 0.3, 'vej', latex_label = r'$v_{\mathrm{ej}~2}~(c)$')
+ye_2 = Uniform(0.05, 0.5, 'ye', latex_label = r'$Y_{e~2}$')
+temperature_floor_2 = LogUniform(100, 6000, 'temperature_floor', latex_label = r'$T_{\mathrm{floor}~2}~(\mathrm{K})$')

redback/priors/two_layer_stratified_kilonova.prior

@@ -3,4 +3,4 @@ mej = Uniform(1e-2, 0.05, 'mej', latex_label = r'$M_{\mathrm{ej}}~(M_\odot)$')
 vej_1 = Uniform(0.05, 0.2, 'vej_1', latex_label = r'$v_{\mathrm{ej}~1}~(c)$')
 vej_2 = Uniform(0.3, 0.5, 'vej_2', latex_label = r'$v_{\mathrm{ej}~2}~(c)$')
 kappa = Uniform(1, 30, 'kappa', latex_label = r'$\\kappa~(\mathrm{cm}^{2}/\mathrm{g})$')
-beta = Uniform(1.5, 8, 'beta', latex_label = r'$\\beta$')
+beta = Uniform(3.1, 8, 'beta', latex_label = r'$\\beta$')

redback/priors/villar_sne.prior

@@ -0,0 +1,7 @@
+aa = LogUniform(1e-15,1e15,name='aa', latex_label = r'$A$')
+cc = LogUniform(1e-15,1e15,name='bb', latex_label = r'$C$')
+t0 = Uniform(0,50,name='t0', latex_label = r'$t_0$')
+tau_rise = Uniform(0.01, 50, name='tau_rise', latex_label = r'$\\tau_{\\mathrm{rise}}$')
+tau_fall = Uniform(1, 300, name='tau_fall', latex_label = r'$\\tau_{\\mathrm{fall}}$')
+gamma = Uniform(0.1, 100, name='gamma', latex_label = r'$\\gamma$')
+nu = Uniform(0, 1, name='nu', latex_label = r'$\\nu$')

redback/priors.py

@@ -9,7 +9,18 @@ from redback.utils import logger
 
 
 def get_priors(model, times=None, y=None, yerr=None, dt=None, **kwargs):
-    prompt_prior_functions = dict(gaussian=get_gaussian_priors, skew_gaussian=get_skew_gaussian_priors,
+    """
+    Get the prior for the given model. If the model is a prompt model, the times, y, and yerr must be provided.
+
+    :param model: String referring to a name of a model implemented in Redback.
+    :param times: Time array
+    :param y: Y values, arbitrary units
+    :param yerr: Error on y values, arbitrary units
+    :param dt: time interval
+    :param kwargs: Extra arguments to be passed to the prior function
+    :return: priors: PriorDict object
+    """
+    prompt_prior_functions = dict(gaussian_prompt=get_gaussian_priors, skew_gaussian=get_skew_gaussian_priors,
                                   skew_exponential=get_skew_exponential_priors, fred=get_fred_priors,
                                   fred_extended=get_fred_extended_priors)
 

redback/sed.py

@@ -6,6 +6,193 @@ from sncosmo import TimeSeriesSource
 from redback.constants import *
 from redback.utils import nu_to_lambda, bandpass_magnitude_to_flux
 
+def _bandflux_single_redback(model, band, time_or_phase):
+    """
+
+    Synthetic photometry of a model through a single bandpass
+
+    :param model: Source object
+    :param band: Bandpass
+    :param time_or_phase: Time or phase numpy array
+    :return: bandflux through the bandpass
+    """
+    from sncosmo.utils import integration_grid
+    HC_ERG_AA = 1.9864458571489284e-08 # planck * speed_of_light in AA/s
+    MODEL_BANDFLUX_SPACING = 5.0  # Angstroms
+
+    if (band.minwave() < model.minwave() or band.maxwave() > model.maxwave()):
+        raise ValueError('bandpass {0!r:s} [{1:.6g}, .., {2:.6g}] '
+                         'outside spectral range [{3:.6g}, .., {4:.6g}]'
+                         .format(band.name, band.minwave(), band.maxwave(),
+                                 model.minwave(), model.maxwave()))
+
+        # Set up wavelength grid. Spacing (dwave) evenly divides the bandpass,
+        # closest to 5 angstroms without going over.
+    wave, dwave = integration_grid(band.minwave(), band.maxwave(),
+                                   MODEL_BANDFLUX_SPACING)
+    trans = band(wave)
+    f = model._flux(time_or_phase, wave)
+    f = np.abs(f)
+    return np.sum(wave * trans * f, axis=1) * dwave / HC_ERG_AA
+
+
+def _bandflux_redback(model, band, time_or_phase, zp, zpsys):
+    """
+    Support function for bandflux in Source and Model. Follows SNCOSMO
+
+    This is necessary to have outside because ``phase`` is used in Source
+    and ``time`` is used in Model, and we want the method signatures to
+    have the right variable name.
+    """
+    from sncosmo.magsystems import get_magsystem
+    from sncosmo.bandpasses import get_bandpass
+
+    if zp is not None and zpsys is None:
+        raise ValueError('zpsys must be given if zp is not None')
+
+    # broadcast arrays
+    if zp is None:
+        time_or_phase, band = np.broadcast_arrays(time_or_phase, band)
+    else:
+        time_or_phase, band, zp, zpsys = \
+            np.broadcast_arrays(time_or_phase, band, zp, zpsys)
+
+    # Convert all to 1-d arrays.
+    ndim = time_or_phase.ndim  # Save input ndim for return val.
+    time_or_phase = np.atleast_1d(time_or_phase)
+    band = np.atleast_1d(band)
+    if zp is not None:
+        zp = np.atleast_1d(zp)
+        zpsys = np.atleast_1d(zpsys)
+
+    # initialize output arrays
+    bandflux = np.zeros(time_or_phase.shape, dtype=float)
+
+    # Loop over unique bands.
+    for b in set(band):
+        mask = band == b
+        b = get_bandpass(b)
+
+        fsum = _bandflux_single_redback(model, b, time_or_phase[mask])
+
+        if zp is not None:
+            zpnorm = 10. ** (0.4 * zp[mask])
+            bandzpsys = zpsys[mask]
+            for ms in set(bandzpsys):
+                mask2 = bandzpsys == ms
+                ms = get_magsystem(ms)
+                zpnorm[mask2] = zpnorm[mask2] / ms.zpbandflux(b)
+            fsum *= zpnorm
+
+        bandflux[mask] = fsum
+
+    if ndim == 0:
+        return bandflux[0]
+    return bandflux
+
+def _bandmag_redback(model, band, magsys, time_or_phase):
+    """
+    Support function for bandflux in Source and Model.
+    This is necessary to have outside the models because ``phase`` is used in
+    Source and ``time`` is used in Model.
+    """
+    from sncosmo.magsystems import get_magsystem
+
+    bandflux = _bandflux_redback(model, band, time_or_phase, None, None)
+    band, magsys, bandflux = np.broadcast_arrays(band, magsys, bandflux)
+    return_scalar = (band.ndim == 0)
+    band = band.ravel()
+    magsys = magsys.ravel()
+    bandflux = bandflux.ravel()
+
+    result = np.empty(bandflux.shape, dtype=float)
+    for i, (b, ms, f) in enumerate(zip(band, magsys, bandflux)):
+        ms = get_magsystem(ms)
+        zpf = ms.zpbandflux(b)
+        result[i] = -2.5 * np.log10(f / zpf)
+
+    if return_scalar:
+        return result[0]
+    return result
+
+class RedbackTimeSeriesSource(TimeSeriesSource):
+        def __init__(self, phase, wave, flux, **kwargs):
+            """
+            RedbackTimeSeriesSource is a subclass of sncosmo.TimeSeriesSource that adds the ability to return the
+            flux density at a given time and wavelength, and changes
+            the behaviour of the _flux method to better handle models with very low flux values.
+
+            :param phase: phase/time array
+            :param wave: wavelength array in Angstrom
+            :param spectra: spectra in erg/cm^2/s/A evaluated at all times and frequencies; shape (len(times), len(frequency_array))
+            :param kwargs: additional arguments
+            """
+            super(RedbackTimeSeriesSource, self).__init__(phase=phase, wave=wave, flux=flux, **kwargs)
+
+        def get_flux_density(self, time, wavelength):
+            """
+            Get the flux density at a given time and wavelength.
+
+            :param time: time in days
+            :param wavelength: wavelength in Angstrom
+            :return: flux density in erg/cm^2/s/A
+            """
+            return self._flux(time, wavelength)
+
+        def bandflux(self, band, phase, zp=None, zpsys=None):
+            """
+            Flux through the given bandpass(es) at the given phase(s).
+
+            Default return value is flux in photons / s / cm^2. If zp and zpsys
+            are given, flux(es) are scaled to the requested zeropoints.
+
+            Parameters
+            ----------
+            band : str or list_like
+                Name(s) of bandpass(es) in registry.
+            phase : float or list_like, optional
+                Phase(s) in days. Default is `None`, which corresponds to the full
+                native phase sampling of the model.
+            zp : float or list_like, optional
+                If given, zeropoint to scale flux to (must also supply ``zpsys``).
+                If not given, flux is not scaled.
+            zpsys : str or list_like, optional
+                Name of a magnitude system in the registry, specifying the system
+                that ``zp`` is in.
+
+            Returns
+            -------
+            bandflux : float or `~numpy.ndarray`
+                Flux in photons / s /cm^2, unless `zp` and `zpsys` are
+                given, in which case flux is scaled so that it corresponds
+                to the requested zeropoint. Return value is `float` if all
+                input parameters are scalars, `~numpy.ndarray` otherwise.
+            """
+            return _bandflux_redback(self, band, phase, zp, zpsys)
+
+        def bandmag(self, band, magsys, phase):
+            """Magnitude at the given phase(s) through the given
+            bandpass(es), and for the given magnitude system(s).
+
+            Parameters
+            ----------
+            band : str or list_like
+                Name(s) of bandpass in registry.
+            magsys : str or list_like
+                Name(s) of `~sncosmo.MagSystem` in registry.
+            phase : float or list_like
+                Phase(s) in days.
+
+            Returns
+            -------
+            mag : float or `~numpy.ndarray`
+                Magnitude for each item in band, magsys, phase.
+                The return value is a float if all parameters are not iterables.
+                The return value is an `~numpy.ndarray` if any are iterable.
+            """
+            return _bandmag_redback(self, band, magsys, phase)
+
+
 
 def blackbody_to_flux_density(temperature, r_photosphere, dl, frequency):
     """
@@ -298,13 +485,18 @@ def get_correct_output_format_from_spectra(time, time_eval, spectra, lambda_arra
     :param time: times in observer frame in days to evaluate the model on
     :param time_eval: times in observer frame where spectra are evaluated. A densely sampled array for accuracy
     :param bands: band array - must be same length as time array or a single band
-    :param spectra: spectra in mJy evaluated at all times and frequencies; shape (len(times), len(frequency_array))
+    :param spectra: spectra in erg/cm^2/s/A evaluated at all times and frequencies; shape (len(times), len(frequency_array))
     :param lambda_array: wavelenth array in Angstrom in observer frame
     :param kwargs: Additional parameters
     :param output_format: 'flux', 'magnitude', 'sncosmo_source', 'flux_density'
     :return: flux, magnitude or SNcosmo TimeSeries Source depending on output format kwarg
     """
-    source = TimeSeriesSource(phase=time_eval, wave=lambda_array, flux=spectra)
+    # clean up spectrum to remove nonsensical values before creating sncosmo source
+    spectra = np.nan_to_num(spectra)
+    spectra[spectra.value == np.nan_to_num(np.inf)] = 1e-30 * np.mean(spectra[5])
+    spectra[spectra.value == 0.] = 1e-30 * np.mean(spectra[5])
+
+    source = RedbackTimeSeriesSource(phase=time_eval, wave=lambda_array, flux=spectra)
     if kwargs['output_format'] == 'flux':
         bands = kwargs['bands']
         magnitude = source.bandmag(phase=time, band=bands, magsys='ab')