From 845c55c269a59620bbf8a6c0d8adab575e1185dc Mon Sep 17 00:00:00 2001 From: shivesh Date: Thu, 8 Nov 2018 15:02:27 -0600 Subject: Thu 8 Nov 15:02:27 CST 2018 --- plot_llh/angles_to_fr.py | 30 ++++++++++++++++++++++++------ 1 file changed, 24 insertions(+), 6 deletions(-) (limited to 'plot_llh/angles_to_fr.py') diff --git a/plot_llh/angles_to_fr.py b/plot_llh/angles_to_fr.py index 7a842af..ce1e4ed 100644 --- a/plot_llh/angles_to_fr.py +++ b/plot_llh/angles_to_fr.py @@ -7,8 +7,16 @@ sys.path.extend(['.', '../']) import numpy as np from utils import fr as fr_utils +from utils.enums import MixingScenario + +SOURCE = [0, 1, 0] + +bsm = True +SCALE = 1E-45 +DIMENSION = 6 +FIX_MIXING = MixingScenario.T13 +ENERGY = 1E6 -SOURCE = [1, 2, 0] if len(sys.argv)< 2: print sys.argv @@ -21,11 +29,21 @@ outfile = infile[:-4] + '_proc.npy' d = np.load(infile) def m_fr(theta): - s_12_2, c_13_4, s_23_2, dcp, m21_2, m3x_2 = theta - sm_u = np.array( - fr_utils.angles_to_u((s_12_2, c_13_4, s_23_2, dcp)), dtype=np.complex256 - ) - return fr_utils.u_to_fr(SOURCE, sm_u) + if not bsm: + s_12_2, c_13_4, s_23_2, dcp, m21_2, m3x_2 = theta + sm_u = fr_utils.angles_to_u((s_12_2, c_13_4, s_23_2, dcp)) + sm_u = np.array(sm_u, dtype=np.complex256) + return fr_utils.u_to_fr(SOURCE, sm_u) + elif bsm: + s_12_2, c_13_4, s_23_2, dcp, m21_2, m3x_2 = theta[:6] + sm_u = fr_utils.angles_to_u((s_12_2, c_13_4, s_23_2, dcp)) + bsm_u = np.array( + fr_utils.params_to_BSMu( + theta[6:], fix_scale=True, scale=SCALE, dim=DIMENSION, + energy=ENERGY, sm_u=sm_u + ), dtype=np.complex256 + ) + return fr_utils.u_to_fr(SOURCE, bsm_u) pd = np.array(map(m_fr, d)) np.save(outfile, pd) -- cgit v1.2.3