From de279d92b4766b0b8658576dc21bdcb061cc1697 Mon Sep 17 00:00:00 2001 From: Shivesh Mandalia Date: Fri, 28 Feb 2020 22:41:53 +0000 Subject: move ipynbs to separate repo --- ipynbs/testing.py | 583 ------------------------------------------------------ 1 file changed, 583 deletions(-) delete mode 100644 ipynbs/testing.py (limited to 'ipynbs/testing.py') diff --git a/ipynbs/testing.py b/ipynbs/testing.py deleted file mode 100644 index 5577f4b..0000000 --- a/ipynbs/testing.py +++ /dev/null @@ -1,583 +0,0 @@ - -# coding: utf-8 - -# In[1]: - - -get_ipython().magic(u'matplotlib inline') -get_ipython().magic(u'load_ext autoreload') -get_ipython().magic(u'autoreload 2') - - -# In[2]: - - -import numpy as np - - -# In[3]: - - -sm_010 = np.load('./mcmc_DIM3_sfr_0_1_0_mfr_3.29685E+15_7.88411E+15_6.83344E+15_fix_scale_0.1_sigma_100000_proc.npy') -sm_100 = np.load('./mcmc_DIM3_sfr_1_0_0_mfr_1.98171E+16_6.59371E+15_9.61795E+15_fix_scale_0.1_sigma_100000_proc.npy') - -bsm_010 = np.load('./mcmc_chains_DIM3_sfr_0_1_0_mfr_0_1_0_MixingScenario.T13_sigma_010.npy') -bsm_100 = np.load('./mcmc_chains_DIM3_sfr_1_0_0_mfr_1_0_0_MixingScenario.T23_sigma_010.npy') - - -# In[81]: - - -import matplotlib as mpl -import matplotlib.pyplot as plt -from matplotlib.patches import Circle -from matplotlib.legend_handler import HandlerPatch - -from scipy import interpolate - -import ternary - - -# In[105]: - - -bc = np.genfromtxt('./bayes_contours.csv', delimiter=',', skip_header=2) -print bc - -def swap(x): - y = x.T - return np.vstack([y[1], y[-1], y[0]]).T - -def interp(i): - x, y, z = i.T - p = np.linspace(0, 1, len(x)) - q = np.linspace(0, 1, 100) - s = 0.2 - x_ = interpolate.splev(q, interpolate.splrep(p, x, s=s)) - y_ = interpolate.splev(q, interpolate.splrep(p, y, s=s)) - z_ = interpolate.splev(q, interpolate.splrep(p, z, s=s)) - return np.vstack([x_, y_, z_]).T - -contour_68_upper = interp(swap(bc[:,:3])) -contour_68_lower = interp(swap(bc[:,3:6])) -contour_90_upper = interp(swap(bc[:,6:9])) -contour_90_lower = interp(swap(bc[:,9:])) - - -# In[67]: - - -plt.style.use('./paper.mplstyle') - - -# In[68]: - - -class HandlerCircle(HandlerPatch): - def create_artists(self, legend, orig_handle, - xdescent, ydescent, width, height, - fontsize, trans): - r = 10 - x = r + width//2 + 10 - y = height//2 - 3 - - # create - p = Circle(xy=(x, y), radius=r) - - # update with data from oryginal object - self.update_prop(p, orig_handle, legend) - - # move xy to legend - p.set_transform(trans) - - return [p] - - -# In[116]: - - -# Figure -fig = plt.figure(figsize=(12, 12)) -ax = fig.add_subplot(111) -ax.set_aspect('equal') - -# Boundary and Gridlines -scale = 1 -fig, tax = ternary.figure(ax=ax, scale=scale) - -# Draw Boundary and Gridlines -tax.boundary(linewidth=2.0) -tax.gridlines(color='grey', multiple=scale/5., linewidth=1.0, alpha=0.4, ls='--') -tax.gridlines(color='grey', multiple=scale/10., linewidth=0.5, alpha=0.4, ls=':') - -# Set Axis labels and Title -fontsize = 23 -tax.left_axis_label(r"$f_{\tau}^{\oplus}$", fontsize=fontsize+8, offset=0.2, rotation=0) -tax.right_axis_label(r"$f_{\mu}^{\oplus}$", fontsize=fontsize+8, offset=0.2, rotation=0) -tax.bottom_axis_label(r"$f_{e}^{\oplus}$", fontsize=fontsize+8, position=(0.55, -0.20/2, 0.5), rotation=0) - -# Remove default Matplotlib axis -tax.get_axes().axis('off') -tax.clear_matplotlib_ticks() - -# Plot -tax.scatter(sm_010, marker='.', s=0.2, alpha=0.2, color='green') -tax.scatter(sm_100, marker='.', s=0.2, alpha=0.2, color='blue') -tax.scatter(bsm_010, marker='.', s=0.2, alpha=0.2, color='green') -tax.scatter(bsm_100, marker='.', s=0.2, alpha=0.2, color='blue') - -# Contour -tax.plot(contour_68_upper, linewidth=2.3, color='grey', zorder=0, alpha=0.6) -tax.plot(contour_68_lower, linewidth=2.3, color='grey', zorder=0, alpha=0.6) -tax.plot(contour_90_upper, linewidth=2.3, color='darkgrey', zorder=0, alpha=0.6) -tax.plot(contour_90_lower, linewidth=2.3, color='darkgrey', zorder=0, alpha=0.6) - -# Lines -marker_style = dict( - linestyle=' ', color='darkorange', linewidth=1.2, - markersize=14, zorder=0 -) - -p1 = (0.18301213, 0.43765598, 0.37933189) -p2 = (0, 1, 0) -divisions = 46 -x_d = np.linspace(p1[0], p2[0], divisions+1) -y_d = np.linspace(p1[1], p2[1], divisions+1) -z_d = np.linspace(p1[2], p2[2], divisions+1) -for n in xrange(divisions-2): - p = (x_d[n], y_d[n], z_d[n]) - q = (x_d[n+1], y_d[n+1], z_d[n+1]) - tax.line(p, q, marker=(3, 2, 180+46.5), **marker_style) - -p1 = (0.55003613, 0.18301213, 0.26695174) -p2 = (1, 0, 0) -divisions = 35 -x_d = np.linspace(p1[0], p2[0], divisions+1) -y_d = np.linspace(p1[1], p2[1], divisions+1) -z_d = np.linspace(p1[2], p2[2], divisions+1) -for n in xrange(divisions-2): - p = (x_d[n], y_d[n], z_d[n]) - q = (x_d[n+1], y_d[n+1], z_d[n+1]) - tax.line(p, q, marker=(3, 2, 180+63), **marker_style) - -# Text -ax.text(0.36, 0.53, r'$\mathcal{O}_{e\tau}$', fontsize=fontsize, - rotation=0, verticalalignment='center') -ax.text(0.445, 0.54, r'$\Lambda_d\Longrightarrow$', fontsize=fontsize+5, - rotation=80, verticalalignment='center') -ax.text(0.68, 0.09, r'$\mathcal{O}_{\mu\tau}$', fontsize=fontsize, - rotation=0, verticalalignment='center') -ax.text(0.7, 0.14, r'$\Lambda_d\Longrightarrow$', fontsize=fontsize+5, - rotation=-23, verticalalignment='center') -ax.text(0.34, 0.2, r'$68\%$', fontsize=fontsize, rotation=5) -ax.text(0.34, 0.1, r'$90\%$', fontsize=fontsize, rotation=5) - -# Legend -l_size = fontsize -legend_elements = [] -legend_elements.append( - Circle((0., 0.), 0.1, facecolor='green', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (0:1:0\right )$') -) -legend_elements.append( - Circle((0., 0.), 0.1, facecolor='blue', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (1:0:0\right )$') -) -legend = plt.legend(handles=legend_elements, loc=(0.65, 0.8), - title='Source composition', - fontsize=l_size, - handler_map={Circle: HandlerCircle()}) -plt.setp(legend.get_title(), fontsize=l_size) -legend.get_frame().set_linestyle('-') - -# Set ticks -tax.ticks(axis='blr', multiple=scale/5., linewidth=1, offset=0.03, - fontsize=fontsize, tick_formats='%.1f') - -tax._redraw_labels() - - -# In[10]: - - -import sys -sys.path.extend(['.', '..']) - - -# In[35]: - - -from utils import fr as fr_utils -from utils import misc as misc_utils -from utils.enums import MixingScenario - - -# In[19]: - - -s = [1, 0, 0] -print s, '->', fr_utils.u_to_fr(s, np.array(fr_utils.NUFIT_U, dtype=np.complex128)) -s = [0, 1, 0] -print s, '->', fr_utils.u_to_fr(s, np.array(fr_utils.NUFIT_U, dtype=np.complex128)) - - -# In[47]: - - -sc = np.linspace(-60, -20, 4000) - -s = [1, 0, 0] -frs_100 = [] -for x in sc: - u = fr_utils.params_to_BSMu(x, fix_mixing=MixingScenario.T23, dim=6, energy=1e6) - frs_100.append(fr_utils.u_to_fr(s, np.array(u, dtype=np.complex128))) -frs_100 = np.vstack(frs_100) - -s = [0, 1, 0] -frs_010 = [] -for x in sc: - u = fr_utils.params_to_BSMu(x, fix_mixing=MixingScenario.T13, dim=6, energy=1e6) - frs_010.append(fr_utils.u_to_fr(s, np.array(u, dtype=np.complex128))) -frs_010 = np.vstack(frs_010) - -s = [1, 2, 0] -frs_120 = [] -for x in sc: - u = fr_utils.params_to_BSMu(x, fix_mixing=MixingScenario.T12, dim=6, energy=1e6) - frs_120.append(fr_utils.u_to_fr(s, np.array(u, dtype=np.complex128))) -frs_120 = np.vstack(frs_120) - - -# In[50]: - - -# Figure -fig = plt.figure(figsize=(12, 12)) -ax = fig.add_subplot(111) -ax.set_aspect('equal') - -# Boundary and Gridlines -scale = 1 -fig, tax = ternary.figure(ax=ax, scale=scale) - -# Draw Boundary and Gridlines -tax.boundary(linewidth=2.0) -tax.gridlines(color='grey', multiple=scale/5., linewidth=1.0, alpha=0.4, ls='--') -tax.gridlines(color='grey', multiple=scale/10., linewidth=0.5, alpha=0.4, ls=':') - -# Set Axis labels and Title -fontsize = 23 -tax.left_axis_label(r"$f_{\tau}^{\oplus}$", fontsize=fontsize+8, offset=0.2, rotation=0) -tax.right_axis_label(r"$f_{\mu}^{\oplus}$", fontsize=fontsize+8, offset=0.2, rotation=0) -tax.bottom_axis_label(r"$f_{e}^{\oplus}$", fontsize=fontsize+8, position=(0.55, -0.20/2, 0.5), rotation=0) - -# Remove default Matplotlib axis -tax.get_axes().axis('off') -tax.clear_matplotlib_ticks() - -# Plot -tax.scatter(frs_010, marker='o', s=2, alpha=1, color='green') -tax.scatter(frs_100, marker='o', s=2, alpha=1, color='blue') -tax.scatter(frs_120, marker='o', s=2, alpha=1, color='red') - -# Legend -l_size = fontsize -legend_elements = [] -legend_elements.append( - Circle((0., 0.), 0.1, facecolor='red', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (1:2:0\right )$') -) -legend_elements.append( - Circle((0., 0.), 0.1, facecolor='green', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (0:1:0\right )$') -) -legend_elements.append( - Circle((0., 0.), 0.1, facecolor='blue', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (1:0:0\right )$') -) -legend = plt.legend(handles=legend_elements, loc=(0.65, 0.8), - title='Source composition', - fontsize=l_size, - handler_map={Circle: HandlerCircle()}) -plt.setp(legend.get_title(), fontsize=l_size) -legend.get_frame().set_linestyle('-') - -# Set ticks -tax.ticks(axis='blr', multiple=scale/5., linewidth=1, offset=0.03, - fontsize=fontsize, tick_formats='%.1f') - -tax._redraw_labels() - - -# In[69]: - - -sc = np.linspace(-60, -20, 4000) - -scen = MixingScenario.T12 - -s = [1, 0, 0] -frs_100 = [] -for x in sc: - u = fr_utils.params_to_BSMu(x, fix_mixing=scen, dim=6, energy=1e6) - frs_100.append(fr_utils.u_to_fr(s, np.array(u, dtype=np.complex128))) -frs_100 = np.vstack(frs_100) - -s = [0, 1, 0] -frs_010 = [] -for x in sc: - u = fr_utils.params_to_BSMu(x, fix_mixing=scen, dim=6, energy=1e6) - frs_010.append(fr_utils.u_to_fr(s, np.array(u, dtype=np.complex128))) -frs_010 = np.vstack(frs_010) - -s = [1, 2, 0] -frs_120 = [] -for x in sc: - u = fr_utils.params_to_BSMu(x, fix_mixing=scen, dim=6, energy=1e6) - frs_120.append(fr_utils.u_to_fr(s, np.array(u, dtype=np.complex128))) -frs_120 = np.vstack(frs_120) - - -# In[70]: - - -# Figure -fig = plt.figure(figsize=(12, 12)) -ax = fig.add_subplot(111) -ax.set_aspect('equal') - -# Boundary and Gridlines -scale = 1 -fig, tax = ternary.figure(ax=ax, scale=scale) - -# Draw Boundary and Gridlines -tax.boundary(linewidth=2.0) -tax.gridlines(color='grey', multiple=scale/5., linewidth=1.0, alpha=0.4, ls='--') -tax.gridlines(color='grey', multiple=scale/10., linewidth=0.5, alpha=0.4, ls=':') - -# Set Axis labels and Title -fontsize = 23 -tax.set_title(r'$\mathcal{O}_{e\mu}\:operator$', fontsize=fontsize+10, pad=35) -tax.left_axis_label(r"$f_{\tau}^{\oplus}$", fontsize=fontsize+8, offset=0.2, rotation=0) -tax.right_axis_label(r"$f_{\mu}^{\oplus}$", fontsize=fontsize+8, offset=0.2, rotation=0) -tax.bottom_axis_label(r"$f_{e}^{\oplus}$", fontsize=fontsize+8, position=(0.55, -0.20/2, 0.5), rotation=0) - -# Remove default Matplotlib axis -tax.get_axes().axis('off') -tax.clear_matplotlib_ticks() - -# Plot -tax.scatter(frs_010, marker='o', s=2, alpha=1, color='green') -tax.scatter(frs_100, marker='o', s=2, alpha=1, color='blue') -tax.scatter(frs_120, marker='o', s=2, alpha=1, color='red') - -# Legend -l_size = fontsize -legend_elements = [] -legend_elements.append( - Circle((0., 0.), 0.1, facecolor='red', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (1:2:0\right )$') -) -legend_elements.append( - Circle((0., 0.), 0.1, facecolor='green', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (0:1:0\right )$') -) -legend_elements.append( - Circle((0., 0.), 0.1, facecolor='blue', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (1:0:0\right )$') -) -legend = plt.legend(handles=legend_elements, loc=(0.65, 0.8), - title='Source composition', - fontsize=l_size, - handler_map={Circle: HandlerCircle()}) -plt.setp(legend.get_title(), fontsize=l_size) -legend.get_frame().set_linestyle('-') - -# Set ticks -tax.ticks(axis='blr', multiple=scale/5., linewidth=1, offset=0.03, - fontsize=fontsize, tick_formats='%.1f') - -tax._redraw_labels() - - -# In[64]: - - -sc = np.linspace(-60, -20, 4000) - -scen = MixingScenario.T23 - -s = [1, 0, 0] -frs_100 = [] -for x in sc: - u = fr_utils.params_to_BSMu(x, fix_mixing=scen, dim=6, energy=1e6) - frs_100.append(fr_utils.u_to_fr(s, np.array(u, dtype=np.complex128))) -frs_100 = np.vstack(frs_100) - -s = [0, 1, 0] -frs_010 = [] -for x in sc: - u = fr_utils.params_to_BSMu(x, fix_mixing=scen, dim=6, energy=1e6) - frs_010.append(fr_utils.u_to_fr(s, np.array(u, dtype=np.complex128))) -frs_010 = np.vstack(frs_010) - -s = [1, 2, 0] -frs_120 = [] -for x in sc: - u = fr_utils.params_to_BSMu(x, fix_mixing=scen, dim=6, energy=1e6) - frs_120.append(fr_utils.u_to_fr(s, np.array(u, dtype=np.complex128))) -frs_120 = np.vstack(frs_120) - - -# In[65]: - - -# Figure -fig = plt.figure(figsize=(12, 12)) -ax = fig.add_subplot(111) -ax.set_aspect('equal') - -# Boundary and Gridlines -scale = 1 -fig, tax = ternary.figure(ax=ax, scale=scale) - -# Draw Boundary and Gridlines -tax.boundary(linewidth=2.0) -tax.gridlines(color='grey', multiple=scale/5., linewidth=1.0, alpha=0.4, ls='--') -tax.gridlines(color='grey', multiple=scale/10., linewidth=0.5, alpha=0.4, ls=':') - -# Set Axis labels and Title -fontsize = 23 -tax.set_title(r'$\mathcal{O}_{\mu\tau}\:operator$', fontsize=fontsize+10, pad=35) -tax.left_axis_label(r"$f_{\tau}^{\oplus}$", fontsize=fontsize+8, offset=0.2, rotation=0) -tax.right_axis_label(r"$f_{\mu}^{\oplus}$", fontsize=fontsize+8, offset=0.2, rotation=0) -tax.bottom_axis_label(r"$f_{e}^{\oplus}$", fontsize=fontsize+8, position=(0.55, -0.20/2, 0.5), rotation=0) - -# Remove default Matplotlib axis -tax.get_axes().axis('off') -tax.clear_matplotlib_ticks() - -# Plot -tax.scatter(frs_010, marker='o', s=2, alpha=1, color='green') -tax.scatter(frs_100, marker='o', s=2, alpha=1, color='blue') -tax.scatter(frs_120, marker='o', s=2, alpha=1, color='red') - -# Legend -l_size = fontsize -legend_elements = [] -legend_elements.append( - Circle((0., 0.), 0.1, facecolor='red', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (1:2:0\right )$') -) -legend_elements.append( - Circle((0., 0.), 0.1, facecolor='green', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (0:1:0\right )$') -) -legend_elements.append( - Circle((0., 0.), 0.1, facecolor='blue', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (1:0:0\right )$') -) -legend = plt.legend(handles=legend_elements, loc=(0.65, 0.8), - title='Source composition', - fontsize=l_size, - handler_map={Circle: HandlerCircle()}) -plt.setp(legend.get_title(), fontsize=l_size) -legend.get_frame().set_linestyle('-') - -# Set ticks -tax.ticks(axis='blr', multiple=scale/5., linewidth=1, offset=0.03, - fontsize=fontsize, tick_formats='%.1f') - -tax._redraw_labels() - - -# In[66]: - - -sc = np.linspace(-60, -20, 4000) - -scen = MixingScenario.T13 - -s = [1, 0, 0] -frs_100 = [] -for x in sc: - u = fr_utils.params_to_BSMu(x, fix_mixing=scen, dim=6, energy=1e6) - frs_100.append(fr_utils.u_to_fr(s, np.array(u, dtype=np.complex128))) -frs_100 = np.vstack(frs_100) - -s = [0, 1, 0] -frs_010 = [] -for x in sc: - u = fr_utils.params_to_BSMu(x, fix_mixing=scen, dim=6, energy=1e6) - frs_010.append(fr_utils.u_to_fr(s, np.array(u, dtype=np.complex128))) -frs_010 = np.vstack(frs_010) - -s = [1, 2, 0] -frs_120 = [] -for x in sc: - u = fr_utils.params_to_BSMu(x, fix_mixing=scen, dim=6, energy=1e6) - frs_120.append(fr_utils.u_to_fr(s, np.array(u, dtype=np.complex128))) -frs_120 = np.vstack(frs_120) - - -# In[68]: - - -# Figure -fig = plt.figure(figsize=(12, 12)) -ax = fig.add_subplot(111) -ax.set_aspect('equal') - -# Boundary and Gridlines -scale = 1 -fig, tax = ternary.figure(ax=ax, scale=scale) - -# Draw Boundary and Gridlines -tax.boundary(linewidth=2.0) -tax.gridlines(color='grey', multiple=scale/5., linewidth=1.0, alpha=0.4, ls='--') -tax.gridlines(color='grey', multiple=scale/10., linewidth=0.5, alpha=0.4, ls=':') - -# Set Axis labels and Title -fontsize = 23 -tax.set_title(r'$\mathcal{O}_{e\tau}\:operator$', fontsize=fontsize+10, pad=35) -tax.left_axis_label(r"$f_{\tau}^{\oplus}$", fontsize=fontsize+8, offset=0.2, rotation=0) -tax.right_axis_label(r"$f_{\mu}^{\oplus}$", fontsize=fontsize+8, offset=0.2, rotation=0) -tax.bottom_axis_label(r"$f_{e}^{\oplus}$", fontsize=fontsize+8, position=(0.55, -0.20/2, 0.5), rotation=0) - -# Remove default Matplotlib axis -tax.get_axes().axis('off') -tax.clear_matplotlib_ticks() - -# Plot -tax.scatter(frs_010, marker='o', s=2, alpha=1, color='green') -tax.scatter(frs_100, marker='o', s=2, alpha=1, color='blue') -tax.scatter(frs_120, marker='o', s=2, alpha=1, color='red') - -# Legend -l_size = fontsize -legend_elements = [] -legend_elements.append( - Circle((0., 0.), 0.1, facecolor='red', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (1:2:0\right )$') -) -legend_elements.append( - Circle((0., 0.), 0.1, facecolor='green', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (0:1:0\right )$') -) -legend_elements.append( - Circle((0., 0.), 0.1, facecolor='blue', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (1:0:0\right )$') -) -legend = plt.legend(handles=legend_elements, loc=(0.65, 0.8), - title='Source composition', - fontsize=l_size, - handler_map={Circle: HandlerCircle()}) -plt.setp(legend.get_title(), fontsize=l_size) -legend.get_frame().set_linestyle('-') - -# Set ticks -tax.ticks(axis='blr', multiple=scale/5., linewidth=1, offset=0.03, - fontsize=fontsize, tick_formats='%.1f') - -tax._redraw_labels() - -- cgit v1.2.3