Add generic get_diffuse_flux function that gives f(E)

icecube · Aug 4, 2020 · 1a50594 · 1a50594
1 parent dd13508
commit 1a50594
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/flarestack/cosmo/__init__.py b/flarestack/cosmo/__init__.py
@@ -1,5 +1,6 @@
 from flarestack.cosmo.icecube_diffuse_flux import get_diffuse_flux_at_100TeV, \
-    get_diffuse_flux_at_1GeV, get_diffuse_flux_contour, plot_diffuse_flux, contours
+    get_diffuse_flux_at_1GeV, get_diffuse_flux_contour, plot_diffuse_flux, contours, \
+    get_diffuse_flux
 from flarestack.cosmo.rates import get_rate
 from flarestack.cosmo.neutrino_cosmology import calculate_transient_cosmology, define_cosmology_functions, cumulative_z
 from flarestack.cosmo.simulate_catalogue import simulate_transient_catalogue
diff --git a/flarestack/cosmo/icecube_diffuse_flux/__init__.py b/flarestack/cosmo/icecube_diffuse_flux/__init__.py
@@ -54,6 +54,16 @@ def get_diffuse_flux_at_1GeV(fit="joint_15"):
     diffuse_flux, diffuse_gamma = get_diffuse_flux_at_100TeV(fit)
     return diffuse_flux * (10 ** 5) ** diffuse_gamma, diffuse_gamma
 
+def get_diffuse_flux(e_gev=1., fit="joint_15"):
+    """Returns the IceCube diffuse flux at a given energy.
+
+    :param e_gev: Energy to evaluate flux at
+    :param fit: Fit of diffuse flux to be used
+    :return: Best fit diffuse flux at 1 GeV, and best fit spectral index
+    """
+    diffuse_flux, diffuse_gamma = get_diffuse_flux_at_1GeV(fit)
+    return diffuse_flux * e_gev ** -diffuse_gamma, diffuse_gamma
+
 def flux_f(energy, norm, index):
     """Flux function