Skip to content

Commit

Permalink
Add files via upload
Browse files Browse the repository at this point in the history
  • Loading branch information
@CharlotteRosenstroem
CharlotteRosenstroem authored Sep 10, 2019
1 parent 7eca32d commit 77767d3
Show file tree
Hide file tree
Showing 2 changed files with 70 additions and 97 deletions.
63 changes: 30 additions & 33 deletions Likelihood/Prob_dist_astro.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -26,33 +26,34 @@
Spectral index (E^-gamma).
"""

mix_params_data_set ='nufit_4_0_with_sk'
mix_params_mass_ordering ='no'

# lst_mix_params is passed to NuSQuIDS and contains the mixing parameters
# [s12sq_bf, s23sq_bf, s13sq_bf, deltaCP_bf, Delta_m2_21_bf, Delta_m2_31_bf, Delta_m2_32_bf]
lst_mix_params = Mixing_Parameters(mix_params_data_set, mix_params_mass_ordering)[0]


def Prob_dist_astro_den(nu_energy_min, nu_energy_max, nu_energy_num_nodes, costhz_npts, Nu_Fluxes_Initial, log10_energy_dep_int_min,
log10_energy_dep_int_max, log10_energy_dep_min, log10_energy_dep_max, log10_energy_dep_npts, time_det_yr, volume_total, energy_nu_max, epsabs, epsrel, verbose):

mix_params_data_set ='nufit_4_0_with_sk'
mix_params_mass_ordering ='no'

# lst_mix_params is passed to NuSQuIDS and contains the mixing parameters
# [s12sq_bf, s23sq_bf, s13sq_bf, deltaCP_bf, Delta_m2_21_bf, Delta_m2_31_bf, Delta_m2_32_bf]
lst_mix_params = Mixing_Parameters(mix_params_data_set, mix_params_mass_ordering)[0]

# Array of cos(theta_z) to compute
lst_costhz = np.linspace(-1.0, 1.0, costhz_npts)

lst_flux_det, lst_flux_earth = \
Nu_Fluxes_At_Detector(nu_energy_min, nu_energy_max, nu_energy_num_nodes, lst_costhz, Nu_Fluxes_Initial,
Nu_Fluxes_At_Detector(nu_energy_min, nu_energy_max, nu_energy_num_nodes, lst_costhz, Nu_Fluxes_Initial,
error_rel=1.e-7, error_abs=1.e-7, h_max=500., verbose_level=verbose, lst_mix_params=lst_mix_params, epsabs=1.e-8, epsrel=1.e-8, flag_return_nusquids_format=True)

log10_energy_dep_int_step = log10_energy_dep_int_max - log10_energy_dep_int_min

Event_spec = \
Generate_Event_Spectrum_All_Sky(lst_flux_det, lst_costhz, filename_data_out_suffix='',
flag_save_data=False, flag_plot_histogram=False, flag_initialize_cross_sections=False,
flag_compute_shower_rate=True, flag_compute_track_rate=True,
flag_compute_shower_rate=True, flag_compute_track_rate=True,
flag_sh_nux_nc=True, flag_sh_nue_cc=True, flag_sh_nutau_cc=True, flag_sh_nul_electron_to_electron=True,
flag_sh_nuebar_electron_to_tau=False, lag_sh_nuebar_electron_to_hadrons=True, flag_sh_nutau_electron_to_tau=False,
flag_tr_numu_cc=True, flag_tr_nutau_cc=True, flag_tr_nuebar_electron_to_tau=False,
flag_tr_numu_cc=True, flag_tr_nutau_cc=True, flag_tr_nuebar_electron_to_tau=False,
flag_tr_nutau_electron_to_tau=False, flag_tr_nuebar_electron_to_muon=False, flag_tr_numu_electron_to_muon=False,
log10_energy_dep_min=log10_energy_dep_min, log10_energy_dep_max=log10_energy_dep_max, log10_energy_dep_npts=log10_energy_dep_npts,
log10_energy_dep_int_min=log10_energy_dep_int_min, log10_energy_dep_int_max=log10_energy_dep_int_max, log10_energy_dep_int_step=log10_energy_dep_int_step,
Expand All @@ -65,21 +66,17 @@ def Prob_dist_astro_den(nu_energy_min, nu_energy_max, nu_energy_num_nodes, costh

print("Denominator=", Denominator)

return Denominator
return Denominator


def Prob_dist_astro_num(nu_energy_min, nu_energy_max, nu_energy_num_nodes, costhz_val, Nu_Fluxes_Initial, energy_dep,
time_det_yr, volume_total, energy_nu_max, epsabs, epsrel, verbose, flag_compute_shower_rate, flag_compute_track_rate):

mix_params_data_set ='nufit_4_0_with_sk'
mix_params_mass_ordering ='no'
time_det_yr, volume_total, energy_nu_max, epsabs, epsrel, verbose, flag_compute_shower_rate, flag_compute_track_rate):

lst_mix_params = Mixing_Parameters(mix_params_data_set, mix_params_mass_ordering)[0]

lst_costhz = np.array([costhz_val])

lst_flux_det, lst_flux_earth = \
Nu_Fluxes_At_Detector(nu_energy_min, nu_energy_max, nu_energy_num_nodes, lst_costhz, Nu_Fluxes_Initial,
Nu_Fluxes_At_Detector(nu_energy_min, nu_energy_max, nu_energy_num_nodes, lst_costhz, Nu_Fluxes_Initial,
error_rel=1.e-7, error_abs=1.e-7, h_max=500., verbose_level=verbose, lst_mix_params=lst_mix_params, epsabs=1.e-8, epsrel=1.e-8, flag_return_nusquids_format=True)


Expand All @@ -90,7 +87,7 @@ def Prob_dist_astro_num(nu_energy_min, nu_energy_max, nu_energy_num_nodes, costh
lst_energy_nu = lst_flux_det[0][0] # [GeV]

lst_flux_det_nue = lst_flux_det[0][1] # nu_e [GeV^{-1} cm^{-2} s^{-1} sr^{-1}]

lst_flux_det_nuebar = lst_flux_det[0][2] # nu_e-bar

lst_flux_det_numu = lst_flux_det[0][3] # nu_mu
Expand Down Expand Up @@ -154,7 +151,7 @@ def Prob_dist_astro_num(nu_energy_min, nu_energy_max, nu_energy_num_nodes, costh
miniter=1, maxiter=500, epsabs=epsabs, epsrel=epsrel,
flag_nux_nc=True, flag_nue_cc=True, flag_nutau_cc=True,
flag_nul_electron_to_electron= True, flag_nuebar_electron_to_tau= False,
flag_nuebar_electron_to_hadrons= True, flag_nutau_electron_to_tau= False)
flag_nuebar_electron_to_hadrons= True, flag_nutau_electron_to_tau= False)

# === Track rate ===

Expand All @@ -170,20 +167,20 @@ def Prob_dist_astro_num(nu_energy_min, nu_energy_max, nu_energy_num_nodes, costh
energy_nu_max=energy_nu_max, integration_method='quad',
miniter=1, maxiter=500, epsabs=epsabs, epsrel=epsrel,
flag_numu_cc=True, flag_nutau_cc=True, flag_nuebar_electron_to_tau=False,
flag_nutau_electron_to_tau=False, flag_nuebar_electron_to_muon= False, flag_numu_electron_to_muon= False)
flag_nutau_electron_to_tau=False, flag_nuebar_electron_to_muon= False, flag_numu_electron_to_muon= False)

Numerator = np.array([lst_diff_rate[0] + lst_diff_rate[1] + lst_diff_rate[2]])
Numerator = np.array([lst_diff_rate[0] + lst_diff_rate[1] + lst_diff_rate[2]])

return Numerator


def Prob_dist_astro(g, M, z_min, z_max, E_min, E_max, E_npts, gamma, nu_energy_min, nu_energy_max, nu_energy_num_nodes, costhz_val, costhz_npts, energy_dep,
log10_energy_dep_int_min, log10_energy_dep_int_max, log10_energy_dep_min, log10_energy_dep_max, log10_energy_dep_npts,
time_det_yr, volume_total, energy_nu_max, epsabs, epsrel, verbose, flag_compute_shower_rate, flag_compute_track_rate):

def Prob_dist_astro(g, M, z_min, z_max, E_min, E_max, E_npts, gamma, nu_energy_min, nu_energy_max, nu_energy_num_nodes, costhz_val, costhz_npts, energy_dep,
log10_energy_dep_int_min, log10_energy_dep_int_max, log10_energy_dep_min, log10_energy_dep_max, log10_energy_dep_npts,
time_det_yr, volume_total, energy_nu_max, epsabs, epsrel, verbose, flag_compute_shower_rate, flag_compute_track_rate):

flux_array = Neutrino_Flux(g, M, z_min, z_max, E_min, E_max, E_npts, gamma, m=1.e-10)

lst_energy_nu = flux_array[:,0]
lst_energy_nu = flux_array[:,0]

lst_nu_flux = flux_array[:,1]

Expand All @@ -192,18 +189,18 @@ def Prob_dist_astro(g, M, z_min, z_max, E_min, E_max, E_npts, gamma, nu_energy_m
Nu_Fluxes_Initial = lambda lst_energy_nu, **kwargs: Nu_Fluxes_Initial_Format_NuSQuIDS(lst_energy_nu, interp_nu_flux, **kwargs)

num = \
Prob_dist_astro_num(nu_energy_min, nu_energy_max, nu_energy_num_nodes, costhz_val, Nu_Fluxes_Initial, energy_dep,
Prob_dist_astro_num(nu_energy_min, nu_energy_max, nu_energy_num_nodes, costhz_val, Nu_Fluxes_Initial, energy_dep,
time_det_yr, volume_total, energy_nu_max, epsabs, epsrel, verbose, flag_compute_shower_rate = flag_compute_shower_rate, flag_compute_track_rate = flag_compute_track_rate)

den = \
Prob_dist_astro_den(nu_energy_min, nu_energy_max, nu_energy_num_nodes, costhz_npts, Nu_Fluxes_Initial, log10_energy_dep_int_min,
log10_energy_dep_int_max, log10_energy_dep_min, log10_energy_dep_max, log10_energy_dep_npts, time_det_yr, volume_total, energy_nu_max, epsabs, epsrel, verbose)

prob = num/den
prob = num/den

print("Prob_astro=", prob)

return prob
return prob



Expand All @@ -225,15 +222,15 @@ def Prob_dist_astro(g, M, z_min, z_max, E_min, E_max, E_npts, gamma, nu_energy_m
log10_nu_energy_min = 2.8
log10_nu_energy_max = 9.2
test = Prob_dist_astro(nu_energy_min = 10**log10_nu_energy_min, nu_energy_max = 10**log10_nu_energy_max, nu_energy_num_nodes = 150, costhz = 0.5, costhz_npts = 2, Nu_Fluxes_Initial = test_flux_initial,
energy_dep = 1e5, log10_energy_dep_int_min = 4, log10_energy_dep_int_max = 7, log10_energy_dep_min = 3.8, log10_energy_dep_max = 7.2, log10_energy_dep_npts = 50,
test = Prob_dist_astro(nu_energy_min = 10**log10_nu_energy_min, nu_energy_max = 10**log10_nu_energy_max, nu_energy_num_nodes = 150, costhz = 0.5, costhz_npts = 2, Nu_Fluxes_Initial = test_flux_initial,
energy_dep = 1e5, log10_energy_dep_int_min = 4, log10_energy_dep_int_max = 7, log10_energy_dep_min = 3.8, log10_energy_dep_max = 7.2, log10_energy_dep_npts = 50,
time_det_yr = 8, volume_total = 6.44e14, energy_nu_max = 1e8, epsabs =1e-3, epsrel = 1e-3, verbose=1, flag_compute_shower_rate = True, flag_compute_track_rate = False)
np.savetxt('test_prob.txt', test)
def Full_likelihood(g, M, external_flux_filename, z_min, z_max, E_min, E_max, E_npts, gamma):
def Full_likelihood(g, M, external_flux_filename, z_min, z_max, E_min, E_max, E_npts, gamma):
flux_array = Neutrino_Flux(g, M, external_flux_filename, z_min, z_max, E_min, E_max, E_npts, gamma, m=1.e-10)
lst_energy_nu, lst_nu_flux = Read_Data_File(os.getcwd()+'/'+external_flux_filename)
Expand All @@ -251,6 +248,6 @@ def Full_likelihood(g, M, external_flux_filename, z_min, z_max, E_min, E_max, E_
verbose=verbose)
# Detection time
time_det_yr = 8
time_det_yr = 8
"""
Loading

0 comments on commit 77767d3

Please sign in to comment.