1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
#!/usr/bin/python
import sys
sys.path.append('/Users/Teps/Theory')
#import header as h
#sys.path.append('/Users/Teps/Theory/HESE')
#import anarchy_header as ah
sys.path.append('/Users/Teps/Theory/HESE/Carlos')
import matplotlib as mpl
mpl.use('Agg')
import matplotlib.pyplot as plt
from matplotlib import rc, rcParams
import MinimalTools as MT
import PhysConst as PC
import LVCPT as LVCPT
import numpy as np
import sys,os
rc('text', usetex=True)
rc('font', **{'family':'serif', 'serif':['Computer Modern'], 'size':18})
cols = ['#29A2C6','#FF6D31','#FFCB18','#73B66B','#EF597B', '#333333']
font = {'family' : 'serif',
'weight' : 'bold',
'size' : 18}
if len(sys.argv)< 2:
print sys.argv
print "Usage: make_plots.py input_filepath."
exit(1)
#colors_schemes = ["Greens","Reds","Blues","PuRd","summer"]
colors_schemes = ["Greens","Reds","Blues","spring","summer"]
#colors_schemes = ["Greens","Reds","Blues","cool","summer"]
#colors_schemes = ["Greens","Reds","Blues","PuRd","summer"]
#colors_schemes = ["Blues","Greens","Reds","PuRd","summer"]
#colors_schemes = ["Greys","Greens","Reds","PuRd","summer"]
#colors_schemes = ["Greys","Greys","Greys","Greys","summer"]
#colors_schemes = ["PuRd","summer"]
output_format = "pdf"
# if True then will plot all files in the same figure
use_same_canvas = True
figure = None
for i in range(len(sys.argv)-1):
infile = sys.argv[i+1]
print "Load data: " + str(infile)
if infile[-3:] == 'txt':
flavor_list = np.genfromtxt(infile)
else:
flavor_list = np.load(infile)
if len(flavor_list[~np.isfinite(flavor_list)]) != 0:
fl = []
for x in flavor_list:
if np.sum(~np.isfinite(x)) == 0:
fl.append(x.tolist())
flavor_list = np.array(fl)
print flavor_list
print "Done loading data"
if not use_same_canvas :
filename = "triangle_plot_"+os.path.splitext(sys.argv[i+1])[0]
else :
filename = "triangle_plot"
# plots scale and diviions
scale = 8
divisions = 40
print "Begin making plot ..."
triangle_collection = LVCPT.GenerateTriangles(scale,divisions*2)
map(lambda f : LVCPT.AddPointToTriangleCollection(f,triangle_collection, scale = scale, divisions = divisions*2),flavor_list)
if use_same_canvas:
figure = LVCPT.MakeFlavorTriangle(flavor_list, divisions = 5, save_file=True,
filename = filename + "_hist", icolor = "g", icolormap = colors_schemes[i],
triangle_collection=triangle_collection, figure = figure, alpha = 0.8,
initial_flavor_ratio = [0,1,0], subdivisions = True, color_scale = "log",
output_format = output_format, inner_line_color ="silver",add_default_text = False,
plot_color_bar =True)
else:
figure = LVCPT.MakeFlavorTriangle(flavor_list, divisions = 5, save_file=True,
filename = filename + "_hist", icolor = "g", icolormap = colors_schemes[i],
triangle_collection=triangle_collection, alpha = 0.8,
initial_flavor_ratio = [0,1,0], subdivisions = True, color_scale = "log",
output_format = output_format, inner_line_color = "silver",add_default_text = False,
plot_color_bar =True)
print "Done making plot: " + filename + "_hist."+output_format
ax = figure.get_axes()[0]
#ax.plot(6.5-0.35,5.5+0.3+0.35,"o", color = "grey", ms = 10, markeredgewidth = 0.1, alpha = 0.9)
#ax.text(6.7-0.35,5.44+0.3+0.35,r"$(1-x:x:0)$", fontsize = 16)
#ax.axvline(x = 7.9)
fsz = 32
ax.plot(6.5-0.1,5.5+0.3+0.35+0.2+0.2,"o", color = "gold", ms = 25, markeredgewidth = 0.1, alpha = 0.9)
ax.text(6.7-0.1,5.44+0.3+0.35+0.2+0.2,r"$(1:2:0)$", fontsize = fsz)
ax.plot(6.5-0.1,5.5+0.35+0.2,"o", color = "#2B653E", ms = 25, markeredgewidth = 0.1, alpha = 0.9)
ax.text(6.7-0.1,5.44+0.35+0.2,r"$(1:0:0)$", fontsize = fsz)
ax.plot(6.5-0.1,5.5-0.3+0.35-0.2+0.2,"o", color = "#CA323D", ms = 25, markeredgewidth = 0.1, alpha = 0.9)
ax.text(6.7-0.1,5.44-0.3+0.35-0.2+0.2,r"$(0:1:0)$", fontsize = fsz)
ax.plot(6.5-0.1,5.5-0.3+0.35-0.3-0.4+0.2,"o", color = "#2D4676", ms = 25, markeredgewidth = 0.1, alpha = 0.9)
ax.text(6.7-0.1,5.44-0.3+0.35-0.3-0.4+0.2,r"$(0:0:1)$", fontsize = fsz)
plt.savefig("./plots/"+filename+"."+output_format, dpi = 600, bbox_inches='tight')
exit(1)
##os.system("cd plots")
##os.system("gs triangle_plot_hist.eps")
##os.system("cd ..")
|
