diff options
Diffstat (limited to 'fig2.py')
| -rwxr-xr-x | fig2.py | 252 |
1 files changed, 0 insertions, 252 deletions
diff --git a/fig2.py b/fig2.py deleted file mode 100755 index 8bf108b..0000000 --- a/fig2.py +++ /dev/null @@ -1,252 +0,0 @@ -#! /usr/bin/env python -# author : S. Mandalia -# s.p.mandalia@qmul.ac.uk -# -# date : February 24, 2019 - -""" -HESE BSM Flavour Figure 2 -""" - -from __future__ import absolute_import, division - -import argparse -from functools import partial - -import numpy as np - -from utils import fr as fr_utils -from utils import misc as misc_utils -from utils import plot as plot_utils -from utils.plot import PLANCK_SCALE - -from matplotlib import pyplot as plt -from matplotlib.patches import Circle -from matplotlib.legend_handler import HandlerPatch -import matplotlib.gridspec as gridspec - - -DIM = 6 -NBINS = 25 - - -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 original object - self.update_prop(p, orig_handle, legend) - - # move xy to legend - p.set_transform(trans) - return [p] - - -def cmap_discretize(cmap, N): - colors_i = np.concatenate((np.linspace(0, 1., N), (0.,0.,0.,0.))) - colors_rgba = cmap(colors_i) - indices = np.linspace(0, 1., N+1) - cdict = {} - for ki, key in enumerate(('red','green','blue')): - cdict[key] = [ - (indices[i], colors_rgba[i-1,ki], colors_rgba[i,ki]) for i in xrange(N+1) - ] - # Return colormap object. - return mpl.colors.LinearSegmentedColormap(cmap.name + "_%d"%N, cdict, 1024) - - -def alp(x): - y = list(x) - y[-1] = 0.7 - return y - - -def process_args(args): - """Process the input args.""" - pass - - -def parse_args(args=None): - """Parse command line arguments""" - parser = argparse.ArgumentParser( - description="HESE BSM Flavour Figure 2", - formatter_class=misc_utils.SortingHelpFormatter, - ) - parser.add_argument( - '--datadir', type=str, - help='Path to directory containing contour chains' - ) - if args is None: return parser.parse_args() - else: return parser.parse_args(args.split()) - - -def main(): - args = parse_args() - process_args(args) - misc_utils.print_args(args) - - prefix = '' - - # Load HESE contour. - contour_infile = args.datadir + '/contour' + prefix + '/contour_REAL.npy' - contour_angles = np.load(contour_infile)[:,-2:] - contour_frs = np.array(map(fr_utils.angles_to_fr, contour_angles)) - - # Load mc_unitary. - mcu_basefile = args.datadir + '/mc_unitary' + prefix + '/mc_unitary_SRC_' - mcu_frs_120 = np.load(mcu_basefile + '1_2_0.npy') - mcu_frs_100 = np.load(mcu_basefile + '1_0_0.npy') - mcu_frs_010 = np.load(mcu_basefile + '0_1_0.npy') - - # Load mc_texture. - mct_basefile = args.datadir + '/mc_texture' + prefix + '/mc_texture_SRC_' - mct_chains_010_OET = np.load(mct_basefile + '0_1_0_OET.npy') - mct_chains_100_OUT = np.load(mct_basefile + '1_0_0_OUT.npy') - - # Caculate min/max scales. - min_scale = 1E100 - max_scale = 1E-100 - - for i in (mct_chains_010_OET, mct_chains_100_OUT): - min_scale = min_scale if min_scale < np.min(i[:,-1]) else np.min(i[:,-1]) - max_scale = max_scale if max_scale > np.max(i[:,-1]) else np.max(i[:,-1]) - min_scale -= np.log10(PLANCK_SCALE[DIM]) - max_scale -= np.log10(PLANCK_SCALE[DIM]) - - cmap_green = cmap_discretize(mpl.colors.LinearSegmentedColormap.from_list( - "", ["lime", "gold", "darkorange"] - ), 10) - cmap_blue = cmap_discretize(mpl.colors.LinearSegmentedColormap.from_list( - "", ["blue", "fuchsia", "darkmagenta"] - ), 10) - - norm = mpl.colors.Normalize(vmin=min_scale, vmax=max_scale) - color_010 = map(alp, map(cmap_green, map( - norm, mct_chains_010_OET[:,-1]-np.log10(PLANCK_SCALE[DIM]) - ))) - color_100 = map(alp, map(cmap_blue, map( - norm, mct_chains_100_OUT[:,-1]-np.log10(PLANCK_SCALE[DIM]) - ))) - - fontsize = 23 - ax_labels = [r'$f_{e}^{\oplus}$', r'$f_{\mu}^{\oplus}$', r'$f_{\tau}^{\oplus}$'] - - # Figure - fig = plt.figure(figsize=(10, 10)) - gs = gridspec.GridSpec(2, 1, height_ratios=[40, 1]) - gs.update(hspace=0.3, wspace=0.15) - - # Axis - ax = fig.add_subplot(gs[0]) - tax = plot_utils.get_tax(ax, scale=NBINS, ax_labels=ax_labels) - - # Plot HESE contour - coverages = {68: 'grey', 90: 'darkgrey'} - for cov in coverages.iterkeys(): - plot_utils.flavour_contour( - frs = contour_frs, - ax = ax, - nbins = NBINS, - coverage = cov, - linewidth = 2.3, - color = coverages[cov], - alpha=0.6, - zorder=0 - ) - ax.text(0.34*NBINS, 0.143*NBINS, r'$68\%$', fontsize=fontsize, rotation=3) - ax.text(0.34*NBINS, 0.038*NBINS, r'$90\%$', fontsize=fontsize, rotation=0) - - # Plot unitary contours - mcu_kwargs = {mcu_frs_120:{'color':'red', 'alpha':0.2, 'zorder':3}, - mcu_frs_100:{'color':'black', 'alpha':0.1, 'zorder':2}, - mcu_frs_010:{'color':'black', 'alpha':0.1, 'zorder':2}} - for frs, kwargs in mcu_kwargs.itervals(): - plot_utils.flavour_contour( - frs = frs, - ax = ax, - fill = True, - nbins = NBINS, - coverage = 100, - linewidth = 1, - **kwargs - ) - - # Plot BSM points - tax.scatter( - mct_chains_010_OET[:,:-1]*NBINS, marker='o', s=0.03, color=color_010, - zorder=5 - ) - tax.scatter( - mct_chains_100_OUT[:,:-1]*NBINS, marker='o', s=0.03, color=color_100, - zorder=5 - ) - - # Legend - legend_elements = [] - legend_elements.append( - Circle((0., 0.), 0.1, facecolor='lime', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (0:1:0\right )\:w/\:{\rm New\:Physics}$') - ) - legend_elements.append( - Circle((0., 0.), 0.1, facecolor='blue', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (1:0:0\right )\:w/\:{\rm New\:Physics}$') - ) - 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='grey', alpha=0.7, edgecolor='k', - linewidth=2., label=r'$\left (0:1:0\right ) + \left (1:0:0\right )$') - ) - legend = plt.legend(handles=legend_elements, loc=(0.65, 0.8), - title='Source composition', - fontsize=fontsize, - handler_map={Circle: HandlerCircle()}) - plt.setp(legend.get_title(), fontsize=fontsize) - legend.get_frame().set_linestyle('-') - - # Colorbar - gs00 = gridspec.GridSpecFromSubplotSpec(1, 2, subplot_spec=gs[1]) - ax0 = fig.add_subplot(gs00[0]) - cb = mpl.colorbar.ColorbarBase( - ax0, cmap=cmap_010, norm=norm, orientation='horizontal' - ) - cb.ax.tick_params(labelsize=fontsize-5) - ax0.text( - 0.5, 2, r'$\mathcal{O}_{e\tau}\:texture$', fontsize=fontsize, - rotation=0, verticalalignment='center', horizontalalignment='center' - ) - - - ax1 = fig.add_subplot(gs00[1]) - cb = mpl.colorbar.ColorbarBase(ax1, cmap=cmap_100, norm=norm, orientation='horizontal') - cb.ax.tick_params(labelsize=fontsize-5) - ax1.text(0.5, 2, r'$\mathcal{O}_{\mu\tau}\:texture$', fontsize=fontsize, - rotation=0, verticalalignment='center', horizontalalignment='center') - - fig.text(0.5, 0.038, r'${\rm New\:Physics\:Scale}\:[\:{\rm log}_{10} (\Lambda_6\:/\:{\rm M}^{2}_{\rm Planck})\: ]$', - fontsize=fontsize+5, horizontalalignment='center') - - outformat = ['png'] - outfile = args.datadir[:5]+args.datadir[5:].replace('data', 'plots') - outfile += '/fig2' + prefix - make_dir(outfile) - for of in outformat: - print 'Saving plot as {0}'.format(outfile + '.' + of) - fig.savefig(outfile + '.' + of, bbox_inches='tight', dpi=150) - - print "DONE!" - - -main.__doc__ = __doc__ - - -if __name__ == '__main__': - main() |