def user_DisplayParameter(self, parameter):
if parameter == "weight":
logging.getLogger('MA5').info(
" User-imposed weight of the set = " + str(self.weight))
elif parameter == "xsection":
logging.getLogger('MA5').info(" User-imposed cross section = " +
str(self.xsection))
elif parameter == "type":
logging.getLogger('MA5').info(" Type = " + self.GetStringTag())
elif parameter == "weighted_events":
if self.weighted_events:
logging.getLogger('MA5').info(
" Taking account of event weight: true")
else:
logging.getLogger('MA5').info(
" Taking account of event weight: false")
elif parameter == "title":
logging.getLogger('MA5').info(" Title = '" + self.title + "'")
elif parameter == "linecolor":
msg = ColorType.convert2string(self.linecolor)
if self.lineshade != 0 and self.linecolor != ColorType.AUTO:
if self.lineshade > 0:
msg += "+" + str(self.lineshade)
else:
msg += str(self.lineshade)
logging.getLogger('MA5').info(" Line color in histograms = " +
msg)
elif parameter == "linestyle":
logging.getLogger(
'MA5').info(" Line style in histograms = " +
LineStyleType.convert2string(self.linestyle))
elif parameter == "linewidth":
logging.getLogger('MA5').info(" Line width in histograms = " +
str(self.linewidth))
elif parameter == "backcolor":
msg = ColorType.convert2string(self.backcolor)
if self.backshade != 0 and self.backcolor != ColorType.AUTO:
if self.backshade > 0:
msg += "+" + str(self.backshade)
else:
msg += str(self.backshade)
logging.getLogger('MA5').info(
" Background color in histograms = " + msg)
elif parameter == "backstyle":
logging.getLogger(
'MA5').info(" Background style in histograms = " +
BackStyleType.convert2string(self.backstyle))
else:
logging.getLogger('MA5').error(
" the class dataset has no attribute denoted by '" +
parameter + "'")
def user_DisplayParameter(self,parameter):
if parameter=="weight":
logging.info(" User-imposed weight of the set = "+str(self.weight))
elif parameter=="xsection":
logging.info(" User-imposed cross section = "+str(self.xsection))
elif parameter=="type":
logging.info(" Type = "+self.GetStringTag())
elif parameter=="weighted_events":
if self.weighted_events:
logging.info(" Taking account of event weight: true")
else:
logging.info(" Taking account of event weight: false")
elif parameter=="title":
logging.info(" Title = '"+self.title+"'")
elif parameter=="linecolor":
msg=ColorType.convert2string(self.linecolor)
if self.lineshade!=0 and self.linecolor!=ColorType.AUTO:
if self.lineshade>0:
msg+="+"+str(self.lineshade)
else:
msg+=str(self.lineshade)
logging.info(" Line color in histograms = "+msg)
elif parameter=="linestyle":
logging.info(" Line style in histograms = "+LineStyleType.convert2string(self.linestyle))
elif parameter=="linewidth":
logging.info(" Line width in histograms = "+str(self.linewidth))
elif parameter=="backcolor":
msg=ColorType.convert2string(self.backcolor)
if self.backshade!=0 and self.backcolor!=ColorType.AUTO:
if self.backshade>0:
msg+="+"+str(self.backshade)
else:
msg+=str(self.backshade)
logging.info(" Background color in histograms = "+msg)
elif parameter=="backstyle":
logging.info(" Background style in histograms = "+BackStyleType.convert2string(self.backstyle))
else:
logging.error(" the class dataset has no attribute denoted by '"+parameter+"'")
def DrawMATPLOTLIB(self, histos, scales, ref, irelhisto, filenamePy,
outputnames):
# Is there any legend?
legendmode = False
if len(self.main.datasets) > 1:
legendmode = True
# Type of histogram
frequencyhisto = True
for histo in histos:
if histo.__class__.__name__ != 'HistogramFrequency':
frequencyhisto = False
break
logxhisto = True
for histo in histos:
if histo.__class__.__name__ != 'HistogramLogX':
logxhisto = False
break
# Stacking or superimposing histos ?
stackmode = False
if ref.stack==StackingMethodType.STACK or \
( ref.stack==StackingMethodType.AUTO and \
self.main.stack==StackingMethodType.STACK ):
stackmode = True
# Open the file in write-mode
try:
outputPy = file(filenamePy, 'w')
except:
logging.getLogger('MA5').error('Impossible to write the file: ' +
filenamePy)
return False
# File header
function_name = filenamePy[:-3]
function_name = function_name.split('/')[-1]
outputPy.write('def ' + function_name + '():\n')
outputPy.write('\n')
# Import Libraries
outputPy.write(' # Library import\n')
outputPy.write(' import numpy\n')
outputPy.write(' import matplotlib\n')
# outputPy.write(" matplotlib.use('Agg')\n")
outputPy.write(' import matplotlib.pyplot as plt\n')
outputPy.write(' import matplotlib.gridspec as gridspec\n')
outputPy.write('\n')
# Matplotlib & numpy version
outputPy.write(' # Library version\n')
outputPy.write(' matplotlib_version = matplotlib.__version__\n')
outputPy.write(' numpy_version = numpy.__version__\n')
outputPy.write('\n')
# Binning
# Loop over datasets and histos
xnbin = histos[0].nbins
xmin = histos[0].xmin
xmax = histos[0].xmax
outputPy.write(' # Histo binning\n')
if logxhisto:
outputPy.write(' xBinning = [')
for bin in range(1, xnbin + 2):
if bin != 1:
outputPy.write(',')
outputPy.write(str(histos[0].GetBinLowEdge(bin)))
outputPy.write(']\n')
outputPy.write('\n')
else:
outputPy.write(' xBinning = numpy.linspace('+\
str(xmin)+','+str(xmax)+','+str(xnbin+1)+\
',endpoint=True)\n')
outputPy.write('\n')
# Data
outputPy.write(' # Creating data sequence: middle of each bin\n')
outputPy.write(' xData = numpy.array([')
for bin in range(0, xnbin):
if bin != 0:
outputPy.write(',')
outputPy.write(str(histos[0].GetBinMean(bin)))
outputPy.write('])\n\n')
# Loop over datasets and histos
ntot = 0
for ind in range(0, len(histos)):
# Ntot
# Creating a new histo
histoname = 'y' + histos[ind].name + '_' + str(ind)
outputPy.write(' # Creating weights for histo: ' + histoname +
'\n')
outputPy.write(' ' + histoname + '_weights = numpy.array([')
for bin in range(1, xnbin + 1):
ntot += histos[ind].summary.array[bin - 1] * scales[ind]
if bin != 1:
outputPy.write(',')
outputPy.write(
str(histos[ind].summary.array[bin - 1] * scales[ind]))
outputPy.write('])\n\n')
# Canvas
outputPy.write(' # Creating a new Canvas\n')
dpi = 80
height = 500
widthx = 700
if legendmode:
widthx = 1000
outputPy.write(' fig = plt.figure(figsize=('+\
str(widthx/dpi)+','+str(height/dpi)+\
'),dpi='+str(dpi)+')\n')
if not legendmode:
outputPy.write(' frame = gridspec.GridSpec(1,1)\n')
else:
outputPy.write(' frame = gridspec.GridSpec(1,1,right=0.7)\n')
# subplot argument: nrows, ncols, plot_number
# outputPy.write(' pad = fig.add_subplot(111)\n')
outputPy.write(' pad = fig.add_subplot(frame[0])\n')
outputPy.write('\n')
# Stack
outputPy.write(' # Creating a new Stack\n')
for ind in range(len(histos) - 1, -1, -1):
myweight = 'y' + histos[ind].name + '_' + str(ind) + '_weights'
mytitle = '"' + PlotFlow.NiceTitleMatplotlib(
self.main.datasets[ind].title) + '"'
mytitle = mytitle.replace('_', '\_')
if not stackmode:
myweights = 'y' + histos[ind].name + '_' + str(
ind) + '_weights'
else:
myweights = ''
for ind2 in range(0, ind + 1):
if ind2 >= 1:
myweights += '+'
myweights += 'y' + histos[ind2].name + '_' + str(
ind2) + '_weights'
# reset
linecolor = 0
linestyle = 0
backcolor = 0
backstyle = 0
linewidth = 1
# Setting AUTO settings
if len(histos) == 1:
linecolor1 = [9]
linecolor = linecolor1[ind]
if stackmode:
backstyle1 = [3004]
backstyle = backstyle1[ind]
backcolor = linecolor1[ind]
elif len(histos) == 2:
linecolor2 = [9, 46]
linecolor = linecolor2[ind]
if stackmode:
backstyle2 = [3004, 3005]
backstyle = backstyle2[ind]
backcolor = linecolor2[ind]
elif len(histos) == 3:
linecolor3 = [9, 46, 8]
linecolor = linecolor3[ind]
if stackmode:
backstyle3 = [3004, 3005, 3006]
backstyle = backstyle3[ind]
backcolor = linecolor3[ind]
elif len(histos) == 4:
linecolor4 = [9, 46, 8, 4]
linecolor = linecolor4[ind]
if stackmode:
backstyle4 = [3004, 3005, 3006, 3007]
backstyle = backstyle4[ind]
backcolor = linecolor4[ind]
elif len(histos) == 5:
linecolor5 = [9, 46, 8, 4, 6]
linecolor = linecolor5[ind]
if stackmode:
backstyle5 = [3004, 3005, 3006, 3007, 3013]
backstyle = backstyle5[ind]
backcolor = linecolor5[ind]
elif len(histos) == 6:
linecolor6 = [9, 46, 8, 4, 6, 2]
linecolor = linecolor6[ind]
if stackmode:
backstyle6 = [3004, 3005, 3006, 3007, 3013, 3017]
backstyle = backstyle6[ind]
backcolor = linecolor6[ind]
elif len(histos) == 7:
linecolor7 = [9, 46, 8, 4, 6, 2, 7]
linecolor = linecolor7[ind]
if stackmode:
backstyle7 = [3004, 3005, 3006, 3007, 3013, 3017, 3022]
backstyle = backstyle7[ind]
backcolor = linecolor7[ind]
elif len(histos) == 8:
linecolor8 = [9, 46, 8, 4, 6, 2, 7, 3]
linecolor = linecolor8[ind]
if stackmode:
backstyle8 = [
3004, 3005, 3006, 3007, 3013, 3017, 3022, 3315
]
backstyle = backstyle8[ind]
backcolor = linecolor8[ind]
elif len(histos) == 9:
linecolor9 = [9, 46, 8, 4, 6, 2, 7, 3, 42]
linecolor = linecolor9[ind]
if stackmode:
backstyle9 = [
3004, 3005, 3006, 3007, 3013, 3017, 3022, 3315, 3351
]
backstyle = backstyle9[ind]
backcolor = linecolor9[ind]
elif len(histos) == 10:
linecolor10 = [9, 46, 8, 4, 6, 2, 7, 3, 42, 48]
linecolor = linecolor10[ind]
if stackmode:
backstyle10 = [
3004, 3005, 3006, 3007, 3013, 3017, 3022, 3315, 3351,
3481
]
backstyle = backstyle10[ind]
backcolor = linecolor10[ind]
else:
linecolor = self.color
self.color += 1
# lineStyle
linestyle = LineStyleType.convert2code(
self.main.datasets[ind].linestyle)
# linewidth
linewidth = self.main.datasets[ind].linewidth
# background color
if self.main.datasets[ind].backcolor != ColorType.AUTO:
backcolor=ColorType.convert2root( \
self.main.datasets[ind].backcolor,\
self.main.datasets[ind].backshade)
# background style
if self.main.datasets[ind].backstyle != BackStyleType.AUTO:
backstyle=BackStyleType.convert2matplotlib( \
self.main.datasets[ind].backstyle)
mylinecolor = '"' + madanalysis.enumeration.color_hex.color_hex[
linecolor] + '"'
mybackcolor = '"' + madanalysis.enumeration.color_hex.color_hex[
backcolor] + '"'
filledmode = '"stepfilled"'
rWidth = 1.
if backcolor == 0: #invisible
filledmode = '"step"'
mybackcolor = 'None'
# if frequencyhisto:
# filledmode='"bar"'
# rWidth=0.8
mylinewidth = self.main.datasets[ind].linewidth
mylinestyle = LineStyleType.convert2matplotlib(
self.main.datasets[ind].linestyle)
outputPy.write(' pad.hist('+\
'x=xData, '+\
'bins=xBinning, '+\
'weights='+myweights+',\\\n'+\
' label='+mytitle+', ')
if ntot != 0:
outputPy.write('histtype=' + filledmode + ', ')
outputPy.write( 'rwidth='+str(rWidth)+',\\\n'+\
' color='+mybackcolor+', '+\
'edgecolor='+mylinecolor+', '+\
'linewidth='+str(mylinewidth)+', '+\
'linestyle='+mylinestyle+',\\\n'+\
' bottom=None, '+\
'cumulative=False, normed=False, align="mid", orientation="vertical")\n\n')
outputPy.write('\n')
# Label
outputPy.write(' # Axis\n')
outputPy.write(" plt.rc('text',usetex=False)\n")
# X-axis
if ref.titleX == "":
axis_titleX = ref.GetXaxis_Matplotlib()
else:
axis_titleX = ref.titleX
axis_titleX = axis_titleX.replace('#DeltaR', '#Delta R')
axis_titleX = axis_titleX.replace('#', '\\')
outputPy.write(' plt.xlabel(r"' + axis_titleX + '",\\\n')
outputPy.write(' fontsize=16,color="black")\n')
# Y-axis
axis_titleY = ref.GetYaxis_Matplotlib()
# Scale to one ?
scale2one = False
if ref.stack==StackingMethodType.NORMALIZE2ONE or \
(self.main.stack==StackingMethodType.NORMALIZE2ONE and \
ref.stack==StackingMethodType.AUTO):
scale2one = True
if scale2one:
axis_titleY += " $(#mathrm{scaled}\ #mathrm{to}# #mathrm{one})$"
elif self.main.normalize == NormalizeType.LUMI or \
self.main.normalize == NormalizeType.LUMI_WEIGHT:
axis_titleY += " $(#mathcal{L}_{#mathrm{int}} = " + str(
self.main.lumi) + "# #mathrm{fb}^{-1})$ "
elif self.main.normalize == NormalizeType.NONE:
axis_titleY += " $(#mathrm{not}# #mathrm{normalized})$"
if ref.titleY != "":
axis_titleY = PlotFlow.NiceTitle(ref.titleY)
axis_titleY = axis_titleY.replace('#', '\\')
outputPy.write(' plt.ylabel(r"' + axis_titleY + '",\\\n')
outputPy.write(' fontsize=16,color="black")\n')
outputPy.write('\n')
# Tag Log/Linear
is_logx = False
if ref.logX and ntot != 0:
is_logx = True
is_logy = False
if ref.logY and ntot != 0:
is_logy = True
# Bound y
outputPy.write(' # Boundary of y-axis\n')
myweights = ''
if stackmode:
for ind in range(0, len(histos)):
if ind >= 1:
myweights += '+'
myweights += 'y' + histos[ind].name + '_' + str(
ind) + '_weights'
else:
myweights = 'numpy.array(['
for ind in range(0, len(histos)):
if ind >= 1:
myweights += ','
myweights += 'y' + histos[ind].name + '_' + str(
ind) + '_weights.max()'
myweights += '])'
outputPy.write(' ymax=(' + myweights + ').max()*1.1\n')
outputPy.write(' ')
if is_logy:
outputPy.write('#')
outputPy.write('ymin=0 # linear scale\n')
myweights = ''
if stackmode:
for ind in range(0, len(histos)):
if ind >= 1:
myweights += '+'
myweights += 'y' + histos[ind].name + '_' + str(
ind) + '_weights'
else:
myweights = 'numpy.array(['
for ind in range(0, len(histos)):
if ind >= 1:
myweights += ','
myweights += 'y' + histos[ind].name + '_' + str(
ind) + '_weights.min()'
myweights += ',1.])'
outputPy.write(' ')
if not is_logy:
outputPy.write('#')
outputPy.write('ymin=min([x for x in (' + myweights +
') if x])/100. # log scale\n')
outputPy.write(' plt.gca().set_ylim(ymin,ymax)\n')
outputPy.write('\n')
# X axis
outputPy.write(' # Log/Linear scale for X-axis\n')
# - Linear
outputPy.write(' ')
if is_logx:
outputPy.write('#')
outputPy.write('plt.gca().set_xscale("linear")\n')
# - Log
outputPy.write(' ')
if not is_logx:
outputPy.write('#')
outputPy.write('plt.gca().set_xscale("log",nonposx="clip")\n')
outputPy.write('\n')
# Y axis
outputPy.write(' # Log/Linear scale for Y-axis\n')
# - Linear
outputPy.write(' ')
if is_logy:
outputPy.write('#')
outputPy.write('plt.gca().set_yscale("linear")\n')
# - Log
outputPy.write(' ')
if not is_logy:
outputPy.write('#')
outputPy.write('plt.gca().set_yscale("log",nonposy="clip")\n')
outputPy.write('\n')
# Labels
if frequencyhisto:
outputPy.write(' # Labels for x-Axis\n')
outputPy.write(' xLabels = numpy.array([')
for bin in range(0, xnbin):
if bin >= 1:
outputPy.write(',')
outputPy.write(
'"' + str(histos[0].stringlabels[bin]).replace('_', '\_') +
'"')
outputPy.write('])\n')
outputPy.write(
' plt.xticks(xData, xLabels, rotation="vertical")\n')
outputPy.write('\n')
### BENJ: not necessary for getting the png and pdf files
# Draw
# outputPy.write(' # Draw\n')
# outputPy.write(' plt.show()\n')
# outputPy.write('\n')
# Legend
if legendmode:
# Reminder for 'loc'
# -'best' : 0, (only implemented for axes legends)
# -'upper right' : 1,
# -'upper left' : 2,
# -'lower left' : 3,
# -'lower right' : 4,
# -'right' : 5,
# -'center left' : 6,
# -'center right' : 7,
# -'lower center' : 8,
# -'upper center' : 9,
# -'center' : 10,
outputPy.write(' # Legend\n')
outputPy.write(' plt.legend(bbox_to_anchor=(1.05,1), loc=2,'+\
' borderaxespad=0.)\n')
outputPy.write('\n')
# Producing the image
outputPy.write(' # Saving the image\n')
for outputname in outputnames:
outputPy.write(" plt.savefig('" + outputname + "')\n")
outputPy.write('\n')
# Call the function
outputPy.write('# Running!\n')
outputPy.write("if __name__ == '__main__':\n")
outputPy.write(' ' + function_name + '()\n')
# Close the file
try:
outputPy.close()
except:
logging.getLogger('MA5').error('Impossible to close the file: ' +
outputPy)
return False
# Ok
return True
def DrawROOT(self, histos, scales, ref, irelhisto, filenameC, outputnames):
# Is there any legend?
legendmode = False
if len(self.main.datasets) > 1:
legendmode = True
# Type of histogram
frequencyhisto = True
for histo in histos:
if histo.__class__.__name__ != 'HistogramFrequency':
frequencyhisto = False
break
logxhisto = True
for histo in histos:
if histo.__class__.__name__ != 'HistogramLogX':
logxhisto = False
break
# Stacking or superimposing histos ?
stackmode = False
if ref.stack==StackingMethodType.STACK or \
( ref.stack==StackingMethodType.AUTO and \
self.main.stack==StackingMethodType.STACK ):
stackmode = True
# Open the file in write-mode
try:
outputC = file(filenameC, 'w')
except:
logging.getLogger('MA5').error('Impossible to write the file: ' +
filenameC)
return False
# File header
function_name = filenameC[:-2]
function_name = function_name.split('/')[-1]
outputC.write('void ' + function_name + '()\n')
outputC.write('{\n\n')
# ROOT version
outputC.write(' // ROOT version\n')
outputC.write(' Int_t root_version = gROOT->GetVersionInt();\n')
outputC.write('\n')
# Creating the TCanvas
PlotFlow.counter = PlotFlow.counter + 1
canvas_name = 'canvas_plotflow_tempo' + str(PlotFlow.counter)
outputC.write(' // Creating a new TCanvas\n')
widthx = 700
if legendmode:
widthx = 1000
outputC.write(' TCanvas* canvas = new TCanvas("' + canvas_name +
'","' + canvas_name + '",0,0,' + str(widthx) +
',500);\n')
outputC.write(' gStyle->SetOptStat(0);\n')
outputC.write(' gStyle->SetOptTitle(0);\n')
outputC.write(' canvas->SetHighLightColor(2);\n')
# outputC.write(' canvas->Range(-2.419355,-0.005372711,16.93548,0.03939988);\n')
outputC.write(' canvas->SetFillColor(0);\n')
outputC.write(' canvas->SetBorderMode(0);\n')
outputC.write(' canvas->SetBorderSize(3);\n')
outputC.write(' canvas->SetFrameBorderMode(0);\n')
outputC.write(' canvas->SetFrameBorderSize(0);\n')
outputC.write(' canvas->SetTickx(1);\n')
outputC.write(' canvas->SetTicky(1);\n')
outputC.write(' canvas->SetLeftMargin(0.14);\n')
margin = 0.05
if legendmode:
margin = 0.3
outputC.write(' canvas->SetRightMargin(' + str(margin) + ');\n')
outputC.write(' canvas->SetBottomMargin(0.15);\n')
outputC.write(' canvas->SetTopMargin(0.05);\n')
outputC.write('\n')
# Binning
xnbin = histos[0].nbins
if logxhisto:
outputC.write(' // Histo binning\n')
outputC.write(' Double_t xBinning[' + str(xnbin + 1) + '] = {')
for bin in range(1, xnbin + 2):
if bin != 1:
outputC.write(',')
outputC.write(str(histos[0].GetBinLowEdge(bin)))
outputC.write('};\n')
outputC.write('\n')
# Loop over datasets and histos
ntot = 0
for ind in range(0, len(histos)):
# Creating TH1F
outputC.write(' // Creating a new TH1F\n')
histoname = "S" + histos[ind].name + '_' + str(ind)
xmin = histos[ind].xmin
xmax = histos[ind].xmax
if logxhisto:
outputC.write(' TH1F* '+histoname+' = new TH1F("'+histoname+'","'+\
histoname+'",'+str(xnbin)+',xBinning);\n')
else:
outputC.write(' TH1F* '+histoname+' = new TH1F("'+histoname+'","'+\
histoname+'",'+str(xnbin)+','+\
str(xmin)+','+str(xmax)+');\n')
# TH1F content
outputC.write(' // Content\n')
outputC.write(' '+histoname+'->SetBinContent(0'+\
','+str(histos[ind].summary.underflow*scales[ind])+'); // underflow\n')
for bin in range(1, xnbin + 1):
ntot += histos[ind].summary.array[bin - 1] * scales[ind]
outputC.write(' '+histoname+'->SetBinContent('+str(bin)+\
','+str(histos[ind].summary.array[bin-1]*scales[ind])+');\n')
nentries = histos[ind].summary.nentries
outputC.write(' '+histoname+'->SetBinContent('+str(xnbin+1)+\
','+str(histos[ind].summary.overflow*scales[ind])+'); // overflow\n')
outputC.write(' ' + histoname + '->SetEntries(' + str(nentries) +
');\n')
# reset
linecolor = 0
linestyle = 0
backcolor = 0
backstyle = 0
linewidth = 1
# Setting AUTO settings
if len(histos) == 1:
linecolor1 = [9]
linecolor = linecolor1[ind]
if stackmode:
backstyle1 = [3004]
backstyle = backstyle1[ind]
backcolor = linecolor1[ind]
elif len(histos) == 2:
linecolor2 = [9, 46]
linecolor = linecolor2[ind]
if stackmode:
backstyle2 = [3004, 3005]
backstyle = backstyle2[ind]
backcolor = linecolor2[ind]
elif len(histos) == 3:
linecolor3 = [9, 46, 8]
linecolor = linecolor3[ind]
if stackmode:
backstyle3 = [3004, 3005, 3006]
backstyle = backstyle3[ind]
backcolor = linecolor3[ind]
elif len(histos) == 4:
linecolor4 = [9, 46, 8, 4]
linecolor = linecolor4[ind]
if stackmode:
backstyle4 = [3004, 3005, 3006, 3007]
backstyle = backstyle4[ind]
backcolor = linecolor4[ind]
elif len(histos) == 5:
linecolor5 = [9, 46, 8, 4, 6]
linecolor = linecolor5[ind]
if stackmode:
backstyle5 = [3004, 3005, 3006, 3007, 3013]
backstyle = backstyle5[ind]
backcolor = linecolor5[ind]
elif len(histos) == 6:
linecolor6 = [9, 46, 8, 4, 6, 2]
linecolor = linecolor6[ind]
if stackmode:
backstyle6 = [3004, 3005, 3006, 3007, 3013, 3017]
backstyle = backstyle6[ind]
backcolor = linecolor6[ind]
elif len(histos) == 7:
linecolor7 = [9, 46, 8, 4, 6, 2, 7]
linecolor = linecolor7[ind]
if stackmode:
backstyle7 = [3004, 3005, 3006, 3007, 3013, 3017, 3022]
backstyle = backstyle7[ind]
backcolor = linecolor7[ind]
elif len(histos) == 8:
linecolor8 = [9, 46, 8, 4, 6, 2, 7, 3]
linecolor = linecolor8[ind]
if stackmode:
backstyle8 = [
3004, 3005, 3006, 3007, 3013, 3017, 3022, 3315
]
backstyle = backstyle8[ind]
backcolor = linecolor8[ind]
elif len(histos) == 9:
linecolor9 = [9, 46, 8, 4, 6, 2, 7, 3, 42]
linecolor = linecolor9[ind]
if stackmode:
backstyle9 = [
3004, 3005, 3006, 3007, 3013, 3017, 3022, 3315, 3351
]
backstyle = backstyle9[ind]
backcolor = linecolor9[ind]
elif len(histos) == 10:
linecolor10 = [9, 46, 8, 4, 6, 2, 7, 3, 42, 48]
linecolor = linecolor10[ind]
if stackmode:
backstyle10 = [
3004, 3005, 3006, 3007, 3013, 3017, 3022, 3315, 3351,
3481
]
backstyle = backstyle10[ind]
backcolor = linecolor10[ind]
else:
linecolor = self.color
self.color += 1
# linecolor
if self.main.datasets[ind].linecolor != ColorType.AUTO:
linecolor=ColorType.convert2root( \
self.main.datasets[ind].linecolor,\
self.main.datasets[ind].lineshade)
# lineStyle
linestyle = LineStyleType.convert2code(
self.main.datasets[ind].linestyle)
# linewidth
linewidth = self.main.datasets[ind].linewidth
# background color
if self.main.datasets[ind].backcolor != ColorType.AUTO:
backcolor=ColorType.convert2root( \
self.main.datasets[ind].backcolor,\
self.main.datasets[ind].backshade)
# background color
if self.main.datasets[ind].backstyle != BackStyleType.AUTO:
backstyle=BackStyleType.convert2code( \
self.main.datasets[ind].backstyle)
# style
outputC.write(' // Style\n')
outputC.write(' ' + histoname + '->SetLineColor(' +
str(linecolor) + ');\n')
outputC.write(' ' + histoname + '->SetLineStyle(' +
str(linestyle) + ');\n')
outputC.write(' ' + histoname + '->SetLineWidth(' +
str(linewidth) + ');\n')
outputC.write(' ' + histoname + '->SetFillColor(' +
str(backcolor) + ');\n')
outputC.write(' ' + histoname + '->SetFillStyle(' +
str(backstyle) + ');\n')
if frequencyhisto:
outputC.write(' ' + histoname + '->SetBarWidth(0.8);\n')
outputC.write(' ' + histoname + '->SetBarOffset(0.1);\n')
outputC.write('\n')
# Creating the THStack
outputC.write(' // Creating a new THStack\n')
PlotFlow.counter += 1
outputC.write(' THStack* stack = new THStack("mystack_' +
str(PlotFlow.counter) + '","mystack");\n')
# Loop over datasets and histos
for ind in range(0, len(histos)):
histoname = 'S' + histos[ind].name + '_' + str(ind)
outputC.write(' stack->Add(' + histoname + ');\n')
drawoptions = []
if not stackmode:
drawoptions.append('nostack')
if frequencyhisto:
drawoptions.append('bar1')
outputC.write(' stack->Draw("' + ''.join(drawoptions) + '");\n')
outputC.write('\n')
# Setting Y axis label
outputC.write(' // Y axis\n')
axis_titleY = ref.GetYaxis()
# Scale to one ?
scale2one = False
if ref.stack==StackingMethodType.NORMALIZE2ONE or \
(self.main.stack==StackingMethodType.NORMALIZE2ONE and \
ref.stack==StackingMethodType.AUTO):
scale2one = True
if scale2one:
axis_titleY += " ( scaled to one )"
elif self.main.normalize == NormalizeType.LUMI or \
self.main.normalize == NormalizeType.LUMI_WEIGHT:
axis_titleY += " ( L_{int} = " + str(self.main.lumi) + " fb^{-1} )"
elif self.main.normalize == NormalizeType.NONE:
axis_titleY += " (not normalized)"
if ref.titleY != "":
axis_titleY = PlotFlow.NiceTitle(ref.titleY)
if (len(axis_titleY) > 35):
titlesize = 0.04
else:
titlesize = 0.06
outputC.write(' stack->GetYaxis()->SetLabelSize(0.04);\n')
outputC.write(' stack->GetYaxis()->SetLabelOffset(0.005);\n')
outputC.write(' stack->GetYaxis()->SetTitleSize(' + str(titlesize) +
');\n')
outputC.write(' stack->GetYaxis()->SetTitleFont(22);\n')
outputC.write(' stack->GetYaxis()->SetTitleOffset(1);\n')
outputC.write(' stack->GetYaxis()->SetTitle("' + axis_titleY +
'");\n')
outputC.write('\n')
outputC.write(' // X axis\n')
# Setting X axis label
if ref.titleX == "":
axis_titleX = ref.GetXaxis_Root()
else:
axis_titleX = PlotFlow.NiceTitle(ref.titleX)
# Setting X axis label
outputC.write(' stack->GetXaxis()->SetLabelSize(0.04);\n')
outputC.write(' stack->GetXaxis()->SetLabelOffset(0.005);\n')
outputC.write(' stack->GetXaxis()->SetTitleSize(0.06);\n')
outputC.write(' stack->GetXaxis()->SetTitleFont(22);\n')
outputC.write(' stack->GetXaxis()->SetTitleOffset(1);\n')
outputC.write(' stack->GetXaxis()->SetTitle("' + axis_titleX +
'");\n')
if frequencyhisto:
for bin in range(1, xnbin + 1):
outputC.write(' stack->GetXaxis()->SetBinLabel('+str(bin)+','\
'"'+str(histos[ind].stringlabels[bin-1])+'");\n')
outputC.write('\n')
# Setting Log scale
outputC.write(' // Finalizing the TCanvas\n')
logx = 0
if ref.logX and ntot != 0:
logx = 1
logy = 0
if ref.logY and ntot != 0:
logy = 1
outputC.write(' canvas->SetLogx(' + str(logx) + ');\n')
outputC.write(' canvas->SetLogy(' + str(logy) + ');\n')
outputC.write('\n')
# Displaying a legend
if legendmode:
outputC.write(' // Creating a TLegend\n')
outputC.write(' TLegend* legend = new TLegend(.73,.5,.97,.95);\n')
for ind in range(0, len(histos)):
histoname = 'S' + histos[ind].name + '_' + str(ind)
nicetitle = PlotFlow.NiceTitle(self.main.datasets[ind].title)
outputC.write(' legend->AddEntry(' + histoname + ',"' +
nicetitle + '");\n')
outputC.write(' legend->SetFillColor(0);\n')
outputC.write(' legend->SetTextSize(0.05);\n')
outputC.write(' legend->SetTextFont(22);\n')
outputC.write(
' legend->SetY1(TMath::Max(0.15,0.97-0.10*legend->GetListOfPrimitives()->GetSize()));\n'
)
outputC.write(' legend->Draw();\n')
outputC.write('\n')
# Producing the image
outputC.write(' // Saving the image\n')
for outputname in outputnames:
outputC.write(' canvas->SaveAs("' + outputname + '");\n')
outputC.write('\n')
# File foot
outputC.write('}\n')
# Close the file
try:
outputC.close()
except:
logging.getLogger('MA5').error('Impossible to close the file: ' +
outputC)
return False
# Ok
return True
def Draw(self,histos,scales,ref,irelhisto,preview=False):
from ROOT import TH1
from ROOT import TH1F
from ROOT import THStack
from ROOT import TLegend
from ROOT import TCanvas
# Creating a canvas
canvas = TCanvas("tempo","")
# Special case : NPID, NAPID
if ref.observable.name in ['NPID','NAPID']:
# New collection with labels
labels=[]
# Loop over histos
for ind in range(len(histos)):
nBins = histos[ind].GetNbinsX()
for b in range(1,nBins+1):
# Get the bin label
theLabel = histos[ind].GetXaxis().GetBinLabel(b)
# Add in the collection
if theLabel not in labels:
labels.append(theLabel)
# Sorting labels (alphabetical order)
labels = sorted(labels)
# New histos
newhistos = []
for ind in range(len(histos)):
nBins = histos[ind].GetNbinsX()
# New histo
newhisto = TH1F(histos[ind].GetName()+'_'+str(ind),\
histos[ind].GetTitle(),\
len(labels),
0,
len(labels))
for ilabel in range(len(labels)):
# Set new labels
newhisto.GetXaxis().SetBinLabel(ilabel+1,labels[ilabel])
fill=False
for bin in range(1,nBins+1):
# Get the bin label
if histos[ind].GetXaxis().GetBinLabel(bin)==labels[ilabel]:
tmp = histos[ind].GetBinContent(bin)
newhisto.SetBinContent(ilabel+1,tmp)
fill=True
break
if not fill:
newhisto.SetBinContent(ilabel+1,0)
newhistos.append(newhisto)
# overwrite histos
histos=newhistos
# Loop over datasets and histos
for ind in range(0,len(histos)):
if ref.observable.name in ['NPID','NAPID']:
nBins = histos[ind].GetNbinsX()
for b in range(1,nBins+1):
pid = int(histos[ind].GetXaxis().GetBinLabel(b))
if ref.observable.name=='NPID':
spid = self.main.multiparticles.GetName(pid)
else:
spid = self.main.multiparticles.GetAName(-pid,pid)
if spid!="":
histos[ind].GetXaxis().SetBinLabel(b,spid)
# Scaling
histos[ind].Scale(scales[ind])
# Stacking or superimposing histos ?
stackmode = False
if ref.stack==StackingMethodType.STACK or \
( ref.stack==StackingMethodType.AUTO and \
self.main.stack==StackingMethodType.STACK ):
stackmode=True
# Setting AUTO settings
if len(histos)==1:
histos[0].SetLineColor(9)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
elif len(histos)==2:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
elif len(histos)==3:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
elif len(histos)==4:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
elif len(histos)==5:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
elif len(histos)==6:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
elif len(histos)==7:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
histos[6].SetLineColor(7)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
histos[6].SetFillColor(7)
histos[6].SetFillStyle(3022)
elif len(histos)==8:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
histos[6].SetLineColor(7)
histos[7].SetLineColor(3)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
histos[6].SetFillColor(7)
histos[6].SetFillStyle(3022)
histos[7].SetFillColor(3)
histos[7].SetFillStyle(3315)
elif len(histos)==9:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
histos[6].SetLineColor(7)
histos[7].SetLineColor(3)
histos[8].SetLineColor(42)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
histos[6].SetFillColor(7)
histos[6].SetFillStyle(3022)
histos[7].SetFillColor(3)
histos[7].SetFillStyle(3315)
histos[8].SetFillColor(42)
histos[8].SetFillStyle(3351)
elif len(histos)==10:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
histos[6].SetLineColor(7)
histos[7].SetLineColor(3)
histos[8].SetLineColor(42)
histos[9].SetLineColor(48)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
histos[6].SetFillColor(7)
histos[6].SetFillStyle(3022)
histos[7].SetFillColor(3)
histos[7].SetFillStyle(3315)
histos[8].SetFillColor(42)
histos[8].SetFillStyle(3351)
histos[9].SetFillColor(48)
histos[9].SetFillStyle(3481)
else:
histos[ind].SetLineColor(self.color)
self.color += 1
# Setting USER color
for ind in range(0,len(histos)):
# linecolor
if self.main.datasets[ind].linecolor!=ColorType.AUTO:
histos[ind].SetLineColor(ColorType.convert2root( \
self.main.datasets[ind].linecolor,\
self.main.datasets[ind].lineshade))
# lineStyle
histos[ind].SetLineStyle(LineStyleType.convert2code( \
self.main.datasets[ind].linestyle))
# linewidth
histos[ind].SetLineWidth(self.main.datasets[ind].linewidth)
# background color
if self.main.datasets[ind].backcolor!=ColorType.AUTO:
histos[ind].SetFillColor(ColorType.convert2root( \
self.main.datasets[ind].backcolor,\
self.main.datasets[ind].backshade))
# background color
if self.main.datasets[ind].backstyle!=BackStyleType.AUTO:
histos[ind].SetFillStyle(BackStyleType.convert2code( \
self.main.datasets[ind].backstyle))
# Creating and filling the stack; computing the total number of events
stack = THStack("mystack","")
ntot = 0
for item in histos:
ntot+=item.GetEntries()
stack.Add(item)
# Drawing
if stackmode:
stack.Draw()
else:
stack.Draw("nostack")
# Setting Y axis label
axis_titleY = ref.GetYaxis()
# Scale to one ?
scale2one = False
if ref.stack==StackingMethodType.NORMALIZE2ONE or \
(self.main.stack==StackingMethodType.NORMALIZE2ONE and \
ref.stack==StackingMethodType.AUTO):
scale2one = True
if scale2one:
axis_titleY += " ( scaled to one )"
elif self.main.normalize == NormalizeType.LUMI or \
self.main.normalize == NormalizeType.LUMI_WEIGHT:
axis_titleY += " ( L_{int} = " + str(self.main.lumi)+ " fb^{-1} )"
elif self.main.normalize == NormalizeType.NONE:
axis_titleY += " (not normalized)"
if ref.titleY!="":
axis_titleY = self.NiceTitle(ref.titleY)
stack.GetYaxis().SetTitle(axis_titleY)
if(len(axis_titleY) > 35):
stack.GetYaxis().SetTitleSize(0.04)
else:
stack.GetYaxis().SetTitleSize(0.06)
stack.GetYaxis().SetTitleFont(22)
stack.GetYaxis().SetLabelSize(0.04)
# Setting X axis label
if ref.titleX=="":
axis_titleX = ref.GetXaxis()
else:
axis_titleX = self.NiceTitle(ref.titleX)
# Setting X axis label
stack.GetXaxis().SetTitle(axis_titleX)
stack.GetXaxis().SetTitleSize(0.06)
stack.GetXaxis().SetTitleFont(22)
stack.GetXaxis().SetLabelSize(0.04)
# Setting Log scale
if ref.logX and ntot != 0:
canvas.SetLogx()
if ref.logY and ntot != 0:
canvas.SetLogy()
# Displaying a legend
if len(self.main.datasets)>1:
ymin_legend = .9-.055*len(histos)
if ymin_legend<0.1:
ymin_legend = 0.1
legend = TLegend(.65,ymin_legend,.9,.9)
legend.SetTextSize(0.05);
legend.SetTextFont(22);
for ind in range(0,len(histos)):
legend.AddEntry(histos[ind],self.NiceTitle(self.main.datasets[ind].title))
legend.SetFillColor(0)
legend.Draw()
if not preview:
# Save the canvas in the report format
canvas.SaveAs(self.output_path+"/selection_"+str(irelhisto)+"."+\
ReportFormatType.convert2filetype(self.mode))
# Save the canvas in the C format
canvas.SaveAs(self.output_path+"/selection_"+str(irelhisto)+".C")
else:
# break
answer=raw_input("Press enter to continue : ")
def Draw(self,
histos,
myfile,
scales,
ref,
irelhisto,
mode,
output_path,
preview=False):
from ROOT import TH1
from ROOT import TH1F
from ROOT import THStack
from ROOT import TLegend
from ROOT import TPaveText #RJ
from ROOT import TCanvas
from ROOT import TASImage
from ROOT import TAttImage
from ROOT import TPad
from ROOT import TFile #RJ
# Creating a canvas
PlotFlow.counter = PlotFlow.counter + 1
canvas = TCanvas("tempo" + str(PlotFlow.counter), "")
# Loop over datasets and histos
for ind in range(0, len(histos)):
# Scaling
histos[ind].Scale(scales[ind])
# Stacking or superimposing histos ?
stackmode = False
if ref.stack==StackingMethodType.STACK or \
( ref.stack==StackingMethodType.AUTO and \
self.main.stack==StackingMethodType.STACK ):
stackmode = True
# Setting AUTO settings
if len(histos) == 1:
histos[0].SetLineColor(9)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
elif len(histos) == 2:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
elif len(histos) == 3:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
elif len(histos) == 4:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
elif len(histos) == 5:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
elif len(histos) == 6:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
elif len(histos) == 7:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
histos[6].SetLineColor(7)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
histos[6].SetFillColor(7)
histos[6].SetFillStyle(3022)
elif len(histos) == 8:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
histos[6].SetLineColor(7)
histos[7].SetLineColor(3)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
histos[6].SetFillColor(7)
histos[6].SetFillStyle(3022)
histos[7].SetFillColor(3)
histos[7].SetFillStyle(3315)
elif len(histos) == 9:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
histos[6].SetLineColor(7)
histos[7].SetLineColor(3)
histos[8].SetLineColor(42)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
histos[6].SetFillColor(7)
histos[6].SetFillStyle(3022)
histos[7].SetFillColor(3)
histos[7].SetFillStyle(3315)
histos[8].SetFillColor(42)
histos[8].SetFillStyle(3351)
elif len(histos) == 10:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
histos[6].SetLineColor(7)
histos[7].SetLineColor(3)
histos[8].SetLineColor(42)
histos[9].SetLineColor(48)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
histos[6].SetFillColor(7)
histos[6].SetFillStyle(3022)
histos[7].SetFillColor(3)
histos[7].SetFillStyle(3315)
histos[8].SetFillColor(42)
histos[8].SetFillStyle(3351)
histos[9].SetFillColor(48)
histos[9].SetFillStyle(3481)
else:
histos[ind].SetLineColor(self.color)
self.color += 1
# Setting USER color
for ind in range(0, len(histos)):
# linecolor
if self.main.datasets[ind].linecolor != ColorType.AUTO:
histos[ind].SetLineColor(ColorType.convert2root( \
self.main.datasets[ind].linecolor,\
self.main.datasets[ind].lineshade))
# lineStyle
histos[ind].SetLineStyle(LineStyleType.convert2code( \
self.main.datasets[ind].linestyle))
# linewidth
histos[ind].SetLineWidth(self.main.datasets[ind].linewidth)
# background color
if self.main.datasets[ind].backcolor != ColorType.AUTO:
histos[ind].SetFillColor(ColorType.convert2root( \
self.main.datasets[ind].backcolor,\
self.main.datasets[ind].backshade))
# background color
if self.main.datasets[ind].backstyle != BackStyleType.AUTO:
histos[ind].SetFillStyle(BackStyleType.convert2code( \
self.main.datasets[ind].backstyle))
# Creating and filling the stack; computing the total number of events
stack = THStack("mystack", "")
ntot = 0
for item in histos:
ntot += item.Integral()
stack.Add(item)
# Drawing
if stackmode:
stack.Draw()
else:
stack.Draw("nostack")
# Setting Y axis label
axis_titleY = ref.GetYaxis()
# Scale to one ?
scale2one = False
if ref.stack==StackingMethodType.NORMALIZE2ONE or \
(self.main.stack==StackingMethodType.NORMALIZE2ONE and \
ref.stack==StackingMethodType.AUTO):
scale2one = True
if scale2one:
axis_titleY += " ( scaled to one )"
elif self.main.normalize == NormalizeType.LUMI or \
self.main.normalize == NormalizeType.LUMI_WEIGHT:
axis_titleY += " ( L_{int} = " + str(self.main.lumi) + " fb^{-1} )"
elif self.main.normalize == NormalizeType.NONE:
axis_titleY += " (not normalized)"
if ref.titleY != "":
axis_titleY = PlotFlow.NiceTitle(ref.titleY)
stack.GetYaxis().SetTitle(axis_titleY)
if (len(axis_titleY) > 35):
stack.GetYaxis().SetTitleSize(0.04)
else:
stack.GetYaxis().SetTitleSize(0.06)
stack.GetYaxis().SetTitleFont(42)
stack.GetYaxis().SetLabelSize(0.04)
# Setting X axis label
if ref.titleX == "":
axis_titleX = ref.GetXaxis()
else:
axis_titleX = PlotFlow.NiceTitle(ref.titleX)
# Setting X axis label
stack.GetXaxis().SetTitle(axis_titleX)
stack.GetXaxis().SetTitleSize(0.06)
stack.GetXaxis().SetTitleFont(42)
stack.GetXaxis().SetLabelSize(0.04)
# Setting Log scale
if ref.logX and ntot != 0:
canvas.SetLogx()
if ref.logY and ntot != 0:
canvas.SetLogy()
#RJ
T = TPaveText(0.1, 0.995, 0.5, 0.95, 'NDC')
T.AddText('#bf{CMS} Simulation')
T.SetTextFont(42)
T.Draw("same")
T.SetBorderSize(0)
T.SetFillColor(0)
T.SetFillStyle(0)
T.SetLineColor(0)
T.SetTextAlign(22)
# Save all hists -- R.J. Wang
# Loop over datasets and histos
for ind in range(0, len(histos)):
if not myfile.GetListOfKeys().Contains(
PlotFlow.NiceTitle(self.main.datasets[ind].title)):
myfile.mkdir(PlotFlow.NiceTitle(self.main.datasets[ind].title))
myfile.cd(PlotFlow.NiceTitle(self.main.datasets[ind].title))
newhistosNAME = axis_titleX
newhistosNAME = newhistosNAME.replace('(GeV/c^{2})', '')
newhistosNAME = newhistosNAME.replace('(GeV/c)', '')
newhistosNAME = newhistosNAME.replace(' ', '')
newhistosNAME = newhistosNAME.replace('+', '')
newhistosNAME = newhistosNAME.replace('-', '')
newhistosNAME = newhistosNAME.replace('[', '')
newhistosNAME = newhistosNAME.replace(']', '')
newhistosNAME = newhistosNAME.replace('#it', '')
newhistosNAME = newhistosNAME.replace('#slashET', 'MET')
newhistosNAME = newhistosNAME.replace('(GeV)', '')
newhistosNAME = newhistosNAME.replace('{', '')
newhistosNAME = newhistosNAME.replace('}', '')
newhistosNAME = newhistosNAME.replace('~', '')
newhistosNAME = newhistosNAME.replace('_', '')
#print newhistosNAME
histos[ind].SetName(newhistosNAME)
histos[ind].SetTitle(newhistosNAME)
histos[ind].GetXaxis().SetTitle(axis_titleX)
histos[ind].Write()
# Displaying a legend
if len(self.main.datasets) > 1:
ymin_legend = .95 - .035 * len(histos)
if ymin_legend < 0.1:
ymin_legend = 0.1
ymin_legend = 0.8
legend = TLegend(.65, ymin_legend, .95, .93)
legend.SetNColumns(3)
#legend.SetTextSize(0.05);
legend.SetTextFont(42)
for ind in range(0, len(histos)):
legend.AddEntry(
histos[ind],
PlotFlow.NiceTitle(self.main.datasets[ind].title), "L")
legend.SetFillColor(0)
legend.SetFillStyle(0)
legend.SetLineColor(0)
legend.Draw("same")
if not preview:
# Put the MA5 logo
# logo = TASImage.Open(self.main.ma5dir+\
# "/madanalysis/input/logo.eps")
# if not logo.IsValid():
# logging.warning("file called '"+self.main.ma5dir+\
# "/madanalysis/input/logo.eps' " +\
# "is not found")
# else:
# logo.SetConstRatio(0)
# logo.SetImageQuality(TAttImage.kImgBest)
# logo.Vectorize(256)
# w = logo.GetWidth()
# h = logo.GetHeight()
# logo.Scale(int(w*0.2),int(h*0.2))
# mypad = TPad("i1", "i1", 0.75, 0.9, 0.85, 1)
# mypad.Draw()
# mypad.cd()
# logo.Draw()
# # Save the canvas in the report format
# canvas.Update()
#
# thepicture = TASImage.Create()
# thepicture.FromPad(canvas)
# thepicture.SetConstRatio(0)
# thepicture.SetImageQuality(TAttImage.kImgBest)
# thepicture.WriteImage(output_path+"/selection_"+str(irelhisto)+\
# "."+ReportFormatType.convert2filetype(mode))
canvas.SaveAs(output_path+"/selection_"+str(irelhisto)+\
"."+ReportFormatType.convert2filetype(mode))
# Save the canvas in the C format
canvas.SaveAs(output_path + "/selection_" + str(irelhisto) + ".C")
# Save the canvas in the pdf format
canvas.SaveAs(output_path + "/selection_" + str(irelhisto) +
".pdf")
else:
# break
answer = raw_input("Press enter to continue : ")
def Draw(self, histos, scales, ref, irelhisto, preview=False):
from ROOT import TH1
from ROOT import TH1F
from ROOT import THStack
from ROOT import TLegend
from ROOT import TCanvas
# Creating a canvas
canvas = TCanvas("tempo", "")
# Special case : NPID, NAPID
if ref.observable.name in ['NPID', 'NAPID']:
# New collection with labels
labels = []
# Loop over histos
for ind in range(len(histos)):
nBins = histos[ind].GetNbinsX()
for b in range(1, nBins + 1):
# Get the bin label
theLabel = histos[ind].GetXaxis().GetBinLabel(b)
# Add in the collection
if theLabel not in labels:
labels.append(theLabel)
# Sorting labels (alphabetical order)
labels = sorted(labels)
# New histos
newhistos = []
for ind in range(len(histos)):
nBins = histos[ind].GetNbinsX()
# New histo
newhisto = TH1F(histos[ind].GetName()+'_'+str(ind),\
histos[ind].GetTitle(),\
len(labels),
0,
len(labels))
for ilabel in range(len(labels)):
# Set new labels
newhisto.GetXaxis().SetBinLabel(ilabel + 1, labels[ilabel])
fill = False
for bin in range(1, nBins + 1):
# Get the bin label
if histos[ind].GetXaxis().GetBinLabel(
bin) == labels[ilabel]:
tmp = histos[ind].GetBinContent(bin)
newhisto.SetBinContent(ilabel + 1, tmp)
fill = True
break
if not fill:
newhisto.SetBinContent(ilabel + 1, 0)
newhistos.append(newhisto)
# overwrite histos
histos = newhistos
# Loop over datasets and histos
for ind in range(0, len(histos)):
if ref.observable.name in ['NPID', 'NAPID']:
nBins = histos[ind].GetNbinsX()
for b in range(1, nBins + 1):
pid = int(histos[ind].GetXaxis().GetBinLabel(b))
if ref.observable.name == 'NPID':
spid = self.main.multiparticles.GetName(pid)
else:
spid = self.main.multiparticles.GetAName(-pid, pid)
if spid != "":
histos[ind].GetXaxis().SetBinLabel(b, spid)
# Scaling
histos[ind].Scale(scales[ind])
# Stacking or superimposing histos ?
stackmode = False
if ref.stack==StackingMethodType.STACK or \
( ref.stack==StackingMethodType.AUTO and \
self.main.stack==StackingMethodType.STACK ):
stackmode = True
# Setting AUTO settings
if len(histos) == 1:
histos[0].SetLineColor(9)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
elif len(histos) == 2:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
elif len(histos) == 3:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
elif len(histos) == 4:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
elif len(histos) == 5:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
elif len(histos) == 6:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
elif len(histos) == 7:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
histos[6].SetLineColor(7)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
histos[6].SetFillColor(7)
histos[6].SetFillStyle(3022)
elif len(histos) == 8:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
histos[6].SetLineColor(7)
histos[7].SetLineColor(3)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
histos[6].SetFillColor(7)
histos[6].SetFillStyle(3022)
histos[7].SetFillColor(3)
histos[7].SetFillStyle(3315)
elif len(histos) == 9:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
histos[6].SetLineColor(7)
histos[7].SetLineColor(3)
histos[8].SetLineColor(42)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
histos[6].SetFillColor(7)
histos[6].SetFillStyle(3022)
histos[7].SetFillColor(3)
histos[7].SetFillStyle(3315)
histos[8].SetFillColor(42)
histos[8].SetFillStyle(3351)
elif len(histos) == 10:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
histos[6].SetLineColor(7)
histos[7].SetLineColor(3)
histos[8].SetLineColor(42)
histos[9].SetLineColor(48)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
histos[6].SetFillColor(7)
histos[6].SetFillStyle(3022)
histos[7].SetFillColor(3)
histos[7].SetFillStyle(3315)
histos[8].SetFillColor(42)
histos[8].SetFillStyle(3351)
histos[9].SetFillColor(48)
histos[9].SetFillStyle(3481)
else:
histos[ind].SetLineColor(self.color)
self.color += 1
# Setting USER color
for ind in range(0, len(histos)):
# linecolor
if self.main.datasets[ind].linecolor != ColorType.AUTO:
histos[ind].SetLineColor(ColorType.convert2root( \
self.main.datasets[ind].linecolor,\
self.main.datasets[ind].lineshade))
# lineStyle
histos[ind].SetLineStyle(LineStyleType.convert2code( \
self.main.datasets[ind].linestyle))
# linewidth
histos[ind].SetLineWidth(self.main.datasets[ind].linewidth)
# background color
if self.main.datasets[ind].backcolor != ColorType.AUTO:
histos[ind].SetFillColor(ColorType.convert2root( \
self.main.datasets[ind].backcolor,\
self.main.datasets[ind].backshade))
# background color
if self.main.datasets[ind].backstyle != BackStyleType.AUTO:
histos[ind].SetFillStyle(BackStyleType.convert2code( \
self.main.datasets[ind].backstyle))
# Creating and filling the stack; computing the total number of events
stack = THStack("mystack", "")
ntot = 0
for item in histos:
ntot += item.GetEntries()
stack.Add(item)
# Drawing
if stackmode:
stack.Draw()
else:
stack.Draw("nostack")
# Setting Y axis label
axis_titleY = ref.GetYaxis()
# Scale to one ?
scale2one = False
if ref.stack==StackingMethodType.NORMALIZE2ONE or \
(self.main.stack==StackingMethodType.NORMALIZE2ONE and \
ref.stack==StackingMethodType.AUTO):
scale2one = True
if scale2one:
axis_titleY += " ( scaled to one )"
elif self.main.normalize == NormalizeType.LUMI or \
self.main.normalize == NormalizeType.LUMI_WEIGHT:
axis_titleY += " ( L_{int} = " + str(self.main.lumi) + " fb^{-1} )"
elif self.main.normalize == NormalizeType.NONE:
axis_titleY += " (not normalized)"
if ref.titleY != "":
axis_titleY = self.NiceTitle(ref.titleY)
stack.GetYaxis().SetTitle(axis_titleY)
if (len(axis_titleY) > 35):
stack.GetYaxis().SetTitleSize(0.04)
else:
stack.GetYaxis().SetTitleSize(0.06)
stack.GetYaxis().SetTitleFont(22)
stack.GetYaxis().SetLabelSize(0.04)
# Setting X axis label
if ref.titleX == "":
axis_titleX = ref.GetXaxis()
else:
axis_titleX = self.NiceTitle(ref.titleX)
# Setting X axis label
stack.GetXaxis().SetTitle(axis_titleX)
stack.GetXaxis().SetTitleSize(0.06)
stack.GetXaxis().SetTitleFont(22)
stack.GetXaxis().SetLabelSize(0.04)
# Setting Log scale
if ref.logX and ntot != 0:
canvas.SetLogx()
if ref.logY and ntot != 0:
canvas.SetLogy()
# Displaying a legend
if len(self.main.datasets) > 1:
ymin_legend = .9 - .055 * len(histos)
if ymin_legend < 0.1:
ymin_legend = 0.1
legend = TLegend(.65, ymin_legend, .9, .9)
legend.SetTextSize(0.05)
legend.SetTextFont(22)
for ind in range(0, len(histos)):
legend.AddEntry(histos[ind],
self.NiceTitle(self.main.datasets[ind].title))
legend.SetFillColor(0)
legend.Draw()
if not preview:
# Save the canvas in the report format
canvas.SaveAs(self.output_path+"/selection_"+str(irelhisto)+"."+\
ReportFormatType.convert2filetype(self.mode))
# Save the canvas in the C format
canvas.SaveAs(self.output_path + "/selection_" + str(irelhisto) +
".C")
else:
# break
answer = raw_input("Press enter to continue : ")
def Draw(self,histos,myfile,scales,ref,irelhisto,mode,output_path,preview=False):
from ROOT import TH1
from ROOT import TH1F
from ROOT import THStack
from ROOT import TLegend
from ROOT import TPaveText #RJ
from ROOT import TCanvas
from ROOT import TASImage
from ROOT import TAttImage
from ROOT import TPad
from ROOT import TFile #RJ
# Creating a canvas
PlotFlow.counter=PlotFlow.counter+1
canvas = TCanvas("tempo"+str(PlotFlow.counter),"")
# Loop over datasets and histos
for ind in range(0,len(histos)):
# Scaling
histos[ind].Scale(scales[ind])
# Stacking or superimposing histos ?
stackmode = False
if ref.stack==StackingMethodType.STACK or \
( ref.stack==StackingMethodType.AUTO and \
self.main.stack==StackingMethodType.STACK ):
stackmode=True
# Setting AUTO settings
if len(histos)==1:
histos[0].SetLineColor(9)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
elif len(histos)==2:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
elif len(histos)==3:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
elif len(histos)==4:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
elif len(histos)==5:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
elif len(histos)==6:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
elif len(histos)==7:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
histos[6].SetLineColor(7)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
histos[6].SetFillColor(7)
histos[6].SetFillStyle(3022)
elif len(histos)==8:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
histos[6].SetLineColor(7)
histos[7].SetLineColor(3)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
histos[6].SetFillColor(7)
histos[6].SetFillStyle(3022)
histos[7].SetFillColor(3)
histos[7].SetFillStyle(3315)
elif len(histos)==9:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
histos[6].SetLineColor(7)
histos[7].SetLineColor(3)
histos[8].SetLineColor(42)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
histos[6].SetFillColor(7)
histos[6].SetFillStyle(3022)
histos[7].SetFillColor(3)
histos[7].SetFillStyle(3315)
histos[8].SetFillColor(42)
histos[8].SetFillStyle(3351)
elif len(histos)==10:
histos[0].SetLineColor(9)
histos[1].SetLineColor(46)
histos[2].SetLineColor(8)
histos[3].SetLineColor(4)
histos[4].SetLineColor(6)
histos[5].SetLineColor(2)
histos[6].SetLineColor(7)
histos[7].SetLineColor(3)
histos[8].SetLineColor(42)
histos[9].SetLineColor(48)
if stackmode:
histos[0].SetFillColor(9)
histos[0].SetFillStyle(3004)
histos[1].SetFillColor(46)
histos[1].SetFillStyle(3005)
histos[2].SetFillColor(8)
histos[2].SetFillStyle(3006)
histos[3].SetFillColor(4)
histos[3].SetFillStyle(3007)
histos[4].SetFillColor(6)
histos[4].SetFillStyle(3013)
histos[5].SetFillColor(2)
histos[5].SetFillStyle(3017)
histos[6].SetFillColor(7)
histos[6].SetFillStyle(3022)
histos[7].SetFillColor(3)
histos[7].SetFillStyle(3315)
histos[8].SetFillColor(42)
histos[8].SetFillStyle(3351)
histos[9].SetFillColor(48)
histos[9].SetFillStyle(3481)
else:
histos[ind].SetLineColor(self.color)
self.color += 1
# Setting USER color
for ind in range(0,len(histos)):
# linecolor
if self.main.datasets[ind].linecolor!=ColorType.AUTO:
histos[ind].SetLineColor(ColorType.convert2root( \
self.main.datasets[ind].linecolor,\
self.main.datasets[ind].lineshade))
# lineStyle
histos[ind].SetLineStyle(LineStyleType.convert2code( \
self.main.datasets[ind].linestyle))
# linewidth
histos[ind].SetLineWidth(self.main.datasets[ind].linewidth)
# background color
if self.main.datasets[ind].backcolor!=ColorType.AUTO:
histos[ind].SetFillColor(ColorType.convert2root( \
self.main.datasets[ind].backcolor,\
self.main.datasets[ind].backshade))
# background color
if self.main.datasets[ind].backstyle!=BackStyleType.AUTO:
histos[ind].SetFillStyle(BackStyleType.convert2code( \
self.main.datasets[ind].backstyle))
# Creating and filling the stack; computing the total number of events
stack = THStack("mystack","")
ntot = 0
for item in histos:
ntot+=item.Integral()
stack.Add(item)
# Drawing
if stackmode:
stack.Draw()
else:
stack.Draw("nostack")
# Setting Y axis label
axis_titleY = ref.GetYaxis()
# Scale to one ?
scale2one = False
if ref.stack==StackingMethodType.NORMALIZE2ONE or \
(self.main.stack==StackingMethodType.NORMALIZE2ONE and \
ref.stack==StackingMethodType.AUTO):
scale2one = True
if scale2one:
axis_titleY += " ( scaled to one )"
elif self.main.normalize == NormalizeType.LUMI or \
self.main.normalize == NormalizeType.LUMI_WEIGHT:
axis_titleY += " ( L_{int} = " + str(self.main.lumi)+ " fb^{-1} )"
elif self.main.normalize == NormalizeType.NONE:
axis_titleY += " (not normalized)"
if ref.titleY!="":
axis_titleY = PlotFlow.NiceTitle(ref.titleY)
stack.GetYaxis().SetTitle(axis_titleY)
if(len(axis_titleY) > 35):
stack.GetYaxis().SetTitleSize(0.04)
else:
stack.GetYaxis().SetTitleSize(0.06)
stack.GetYaxis().SetTitleFont(42)
stack.GetYaxis().SetLabelSize(0.04)
# Setting X axis label
if ref.titleX=="":
axis_titleX = ref.GetXaxis()
else:
axis_titleX = PlotFlow.NiceTitle(ref.titleX)
# Setting X axis label
stack.GetXaxis().SetTitle(axis_titleX)
stack.GetXaxis().SetTitleSize(0.06)
stack.GetXaxis().SetTitleFont(42)
stack.GetXaxis().SetLabelSize(0.04)
# Setting Log scale
if ref.logX and ntot != 0:
canvas.SetLogx()
if ref.logY and ntot != 0:
canvas.SetLogy()
#RJ
T = TPaveText(0.1,0.995,0.5,0.95,'NDC')
T.AddText('#bf{CMS} Simulation')
T.SetTextFont(42)
T.Draw("same")
T.SetBorderSize(0)
T.SetFillColor(0)
T.SetFillStyle(0)
T.SetLineColor(0)
T.SetTextAlign(22)
# Save all hists -- R.J. Wang
# Loop over datasets and histos
for ind in range(0,len(histos)):
if not myfile.GetListOfKeys().Contains(PlotFlow.NiceTitle(self.main.datasets[ind].title)):
myfile.mkdir(PlotFlow.NiceTitle(self.main.datasets[ind].title))
myfile.cd(PlotFlow.NiceTitle(self.main.datasets[ind].title))
newhistosNAME = axis_titleX
newhistosNAME = newhistosNAME.replace('(GeV/c^{2})','')
newhistosNAME = newhistosNAME.replace('(GeV/c)','')
newhistosNAME = newhistosNAME.replace(' ','')
newhistosNAME = newhistosNAME.replace('+','')
newhistosNAME = newhistosNAME.replace('-','')
newhistosNAME = newhistosNAME.replace('[','')
newhistosNAME = newhistosNAME.replace(']','')
newhistosNAME = newhistosNAME.replace('#it','')
newhistosNAME = newhistosNAME.replace('#slashET','MET')
newhistosNAME = newhistosNAME.replace('(GeV)','')
newhistosNAME = newhistosNAME.replace('{','')
newhistosNAME = newhistosNAME.replace('}','')
newhistosNAME = newhistosNAME.replace('~','')
newhistosNAME = newhistosNAME.replace('_','')
#print newhistosNAME
histos[ind].SetName(newhistosNAME)
histos[ind].SetTitle(newhistosNAME)
histos[ind].GetXaxis().SetTitle(axis_titleX)
histos[ind].Write()
# Displaying a legend
if len(self.main.datasets)>1:
ymin_legend = .95-.035*len(histos)
if ymin_legend<0.1:
ymin_legend = 0.1
ymin_legend = 0.8
legend = TLegend(.65,ymin_legend,.95,.93)
legend.SetNColumns(3);
#legend.SetTextSize(0.05);
legend.SetTextFont(42);
for ind in range(0,len(histos)):
legend.AddEntry(histos[ind],PlotFlow.NiceTitle(self.main.datasets[ind].title),"L")
legend.SetFillColor(0)
legend.SetFillStyle(0)
legend.SetLineColor(0)
legend.Draw("same")
if not preview:
# Put the MA5 logo
# logo = TASImage.Open(self.main.ma5dir+\
# "/madanalysis/input/logo.eps")
# if not logo.IsValid():
# logging.warning("file called '"+self.main.ma5dir+\
# "/madanalysis/input/logo.eps' " +\
# "is not found")
# else:
# logo.SetConstRatio(0)
# logo.SetImageQuality(TAttImage.kImgBest)
# logo.Vectorize(256)
# w = logo.GetWidth()
# h = logo.GetHeight()
# logo.Scale(int(w*0.2),int(h*0.2))
# mypad = TPad("i1", "i1", 0.75, 0.9, 0.85, 1)
# mypad.Draw()
# mypad.cd()
# logo.Draw()
# # Save the canvas in the report format
# canvas.Update()
#
# thepicture = TASImage.Create()
# thepicture.FromPad(canvas)
# thepicture.SetConstRatio(0)
# thepicture.SetImageQuality(TAttImage.kImgBest)
# thepicture.WriteImage(output_path+"/selection_"+str(irelhisto)+\
# "."+ReportFormatType.convert2filetype(mode))
canvas.SaveAs(output_path+"/selection_"+str(irelhisto)+\
"."+ReportFormatType.convert2filetype(mode))
# Save the canvas in the C format
canvas.SaveAs(output_path+"/selection_"+str(irelhisto)+".C")
# Save the canvas in the pdf format
canvas.SaveAs(output_path+"/selection_"+str(irelhisto)+".pdf")
else:
# break
answer=raw_input("Press enter to continue : ")
