def createPlots2D_(plot, compounddetectorname):
"""2D material budget map to know exactly what we are adding.
"""
#IBs = ["InnerServices", "Phase2PixelBarrel", "TIB", "TIDF", "TIDB"]
theDirname = "Figures"
hist_X0_detectors = OrderedDict()
if plot not in plots.keys():
print("Error: chosen plot name not known %s" % plot)
return
# We need to keep the file content alive for the lifetime of the
# full function....
subDetectorFiles = []
hist_X0_elements = OrderedDict()
prof_X0_elements = OrderedDict()
for subDetector, color in DETECTORS.iteritems():
subDetectorFilename = "matbdg_%s.root" % subDetector
if not checkFile_(subDetectorFilename):
print("Error opening file: %s" % subDetectorFilename)
continue
subDetectorFiles.append(TFile(subDetectorFilename))
subDetectorFile = subDetectorFiles[-1]
print("Opening file: %s" % subDetectorFilename)
prof_X0_XXX = subDetectorFile.Get("%d" % plots[plot].plotNumber)
#hist_X0_detectors[subDetector] = prof_X0_XXX
hist_X0_detectors[subDetector] = prof_X0_XXX.ProjectionXY("_pxy", "B")
print subDetector
# First Plot: BeamPipe + Tracker + ECAL + HCal + HGCal + MB + MGNT
# Create "null" histo
minX = 1.03 * hist_X0_detectors["BeamPipe"].GetXaxis().GetXmin()
maxX = 1.03 * hist_X0_detectors["BeamPipe"].GetXaxis().GetXmax()
minY = 1.03 * hist_X0_detectors["BeamPipe"].GetYaxis().GetXmin()
maxY = 1.03 * hist_X0_detectors["BeamPipe"].GetYaxis().GetXmax()
frame = TH2F("frame", "", 10, minX, maxX, 10, minY, maxY)
frame.SetMinimum(0.1)
frame.SetMaximum(10.)
frame.GetXaxis().SetTickLength(frame.GetXaxis().GetTickLength() * 0.50)
frame.GetYaxis().SetTickLength(frame.GetXaxis().GetTickLength() / 4.)
hist2d_X0_total = hist_X0_detectors["BeamPipe"]
# stack
hist2dTitle = (
'%s %s;%s;%s;%s' %
(plots[plot].quotaName, "All detectors", plots[plot].abscissa,
plots[plot].ordinate, plots[plot].quotaName))
hist2d_X0_total.SetTitle(hist2dTitle)
frame.SetTitle(hist2dTitle)
frame.SetTitleOffset(0.5, "Y")
#If here you put different histomin,histomaxin plot_utils you won't see anything
#for the material plots.
if plots[plot].histoMin != -1.:
hist2d_X0_total.SetMinimum(plots[plot].histoMin)
if plots[plot].histoMax != -1.:
hist2d_X0_total.SetMaximum(plots[plot].histoMax)
#
# canvas
can_SubDetectors = TCanvas("can_SubDetectors", "can_SubDetectors",
2480 + 248, 580 + 58 + 58)
can_SubDetectors.SetTopMargin(0.1)
can_SubDetectors.SetBottomMargin(0.1)
can_SubDetectors.SetLeftMargin(0.04)
can_SubDetectors.SetRightMargin(0.06)
can_SubDetectors.SetFillColor(kWhite)
gStyle.SetOptStat(0)
gStyle.SetTitleFillColor(0)
gStyle.SetTitleBorderSize(0)
gStyle.SetOptTitle(0)
hist2d_X0_total.GetYaxis().SetTickLength(
hist2d_X0_total.GetXaxis().GetTickLength() / 4.)
hist2d_X0_total.GetYaxis().SetTickLength(
hist2d_X0_total.GetXaxis().GetTickLength() / 4.)
hist2d_X0_total.SetTitleOffset(0.5, "Y")
hist2d_X0_total.GetYaxis().SetTitleOffset(0.50)
#hist2d_X0_total.GetXaxis().SetTitleOffset(1.15);
#hist2d_X0_total.GetXaxis().SetNoExponent(True)
#hist2d_X0_total.GetYaxis().SetNoExponent(True)
# colors
for det, color in DETECTORS.iteritems():
hist_X0_detectors[det].SetMarkerColor(color)
hist_X0_detectors[det].SetFillColor(color)
for det, histo in hist_X0_detectors.iteritems():
print det
histo.Draw("same")
# Legenda
theLegend_SubDetectors = TLegend(0.100, 0.7, 0.90,
0.90) #(0.180,0.8,0.98,0.90)
theLegend_SubDetectors.SetNColumns(3)
theLegend_SubDetectors.SetFillColor(0)
theLegend_SubDetectors.SetFillStyle(0)
theLegend_SubDetectors.SetBorderSize(0)
for det, histo in hist_X0_detectors.iteritems():
theLegend_SubDetectors.AddEntry(histo, det, "f")
#theLegend_SubDetectors.AddEntry(hgbound1, "HGCal Eta Boundaries [1.3, 3.0]", "l")
theLegend_SubDetectors.Draw()
# text
text_SubDetectors = TPaveText(0.100, 0.627, 0.322, 0.687,
"NDC") #(0.180,0.727,0.402,0.787,"NDC")
text_SubDetectors.SetFillColor(0)
text_SubDetectors.SetBorderSize(0)
text_SubDetectors.AddText("CMS Simulation")
text_SubDetectors.SetTextAlign(11)
text_SubDetectors.Draw()
#Add eta labels
keep_alive = []
if plots[plot].iDrawEta:
keep_alive.extend(drawEtaValues())
# Store
can_SubDetectors.Update()
if not checkFile_(theDirname):
os.mkdir(theDirname)
can_SubDetectors.SaveAs("%s/MaterialBdg_%s_%s.png" %
(theDirname, compounddetectorname, plot))
#It seems that it is too heavy to create .pdf and .root
#can_SubDetectors.SaveAs("%s/MaterialBdg_FromVertexToEndofHGCal_%s.pdf" % (theDirname, plot))
#can_SubDetectors.SaveAs("%s/MaterialBdg_FromVertexToEndofHGCal_%s.root" % (theDirname, plot))
hist2d_X0_total.GetXaxis().SetRangeUser(0., 7000.)
#Draw eta values in the zoom case
keep_alive = []
keep_alive.extend(drawHalfEtaValues())
#hist2d_X0_total.Draw("COLZ")
can_SubDetectors.Update()
can_SubDetectors.Modified()
can_SubDetectors.SaveAs("%s/MaterialBdg_%s_%s_Zpluszoom.png" %
(theDirname, compounddetectorname, plot))
def create2DPlots(detector, plot, plotnum, plotmat, dosingledetector=True):
"""Produce the requested plot for the specified detector.
Function that will plot the requested 2D-@plot for the
specified @detector. The specified detector could either be a
real detector or a compound one. The list of available plots
are the keys of plots dictionary imported from plot_utils.
"""
#gStyle.Reset()
#Better to use an underscore.
plotmat = plotmat.replace(" ", "_")
if plotmat != "":
theDirname = ('Images/%s' % plotmat).replace(" ", "")
else:
theDirname = 'Images'
if not checkFile_(theDirname):
os.mkdir(theDirname)
if not os.path.isdir(('Images/%s/ZPlusZoom' % plotmat).replace(" ", "")):
os.mkdir(('Images/%s/ZPlusZoom' % plotmat).replace(" ", ""))
if not os.path.isdir(('Images/%s/ZMinusZoom' % plotmat).replace(" ", "")):
os.mkdir(('Images/%s/ZMinusZoom' % plotmat).replace(" ", ""))
goodToGo, theDetectorFilename = paramsGood_(detector, plot)
if not goodToGo:
return
theDetectorFile = TFile(theDetectorFilename)
prof2d_X0_det_total = TProfile2D()
prof2d_X0_det_total.Reset()
# get TProfiles
#prof2d_X0_det_total = theDetectorFile.Get('%s' % plots[plot].plotNumber)
prof2d_X0_det_total = theDetectorFile.Get('%s' % plotnum)
print "=================================================================="
print plotnum
# histos
prof2d_X0_det_total.__class__ = TProfile2D
hist_X0_total = prof2d_X0_det_total.ProjectionXY()
# keep files live forever
files = []
if detector in COMPOUNDS.keys() and not dosingledetector:
#When the loop was:
#for subDetector in COMPOUNDS[detector][1:]:
#and the detector was single it never went in the loop and read the single file
#from above. I alter this to COMPOUNDS[detector] to do the multi material budget plot.
#This won't effect the single detector due to the alter in the if above
for subDetector in COMPOUNDS[detector]:
# filenames of single components
subDetectorFilename = "matbdg_%s.root" % subDetector
# open file
if not checkFile_(subDetectorFilename):
print("Error, missing file %s" % subDetectorFilename)
continue
subDetectorFile = TFile(subDetectorFilename)
files.append(subDetectorFile)
print("*** Open file... %s" % subDetectorFilename)
# subdetector profiles
prof2d_X0_det_total = subDetectorFile.Get('%s' %
plots[plot].plotNumber)
prof2d_X0_det_total.__class__ = TProfile2D
# add to summary histogram
hist_X0_total.Add(
prof2d_X0_det_total.ProjectionXY(
"B_%s" % prof2d_X0_det_total.GetName()), +1.000)
# # properties
#gStyle.SetPalette(1)
gStyle.SetStripDecimals(False)
# #
# Create "null" histo
minX = 1.03 * prof2d_X0_det_total.GetXaxis().GetXmin()
maxX = 1.03 * prof2d_X0_det_total.GetXaxis().GetXmax()
minY = 1.03 * prof2d_X0_det_total.GetYaxis().GetXmin()
maxY = 1.03 * prof2d_X0_det_total.GetYaxis().GetXmax()
frame = TH2F("frame", "", 10, minX, maxX, 10, minY, maxY)
frame.SetMinimum(0.1)
frame.SetMaximum(10.)
frame.GetXaxis().SetTickLength(frame.GetXaxis().GetTickLength() * 0.50)
frame.GetYaxis().SetTickLength(frame.GetXaxis().GetTickLength() / 4.)
# Ratio
if plots[plot].iRebin:
prof2d_X0_det_total.Rebin2D()
# stack
hist2dTitle = ('%s %s;%s;%s;%s' %
(plots[plot].quotaName, detector, plots[plot].abscissa,
plots[plot].ordinate, plots[plot].quotaName))
if dosingledetector:
hist2d_X0_total = prof2d_X0_det_total
else:
hist2d_X0_total = hist_X0_total
hist2d_X0_total.SetTitle(hist2dTitle)
frame.SetTitle(hist2dTitle)
frame.SetTitleOffset(0.5, "Y")
#If here you put different histomin,histomaxin plot_utils you won't see anything
#for the material plots.
if plots[plot].histoMin != -1.:
hist2d_X0_total.SetMinimum(plots[plot].histoMin)
if plots[plot].histoMax != -1.:
hist2d_X0_total.SetMaximum(plots[plot].histoMax)
#
can2name = "MBCan_2D_%s_%s_%s" % (detector, plot, plotmat)
can2 = TCanvas(can2name, can2name, 2480 + 248, 580 + 58 + 58)
can2.SetTopMargin(0.1)
can2.SetBottomMargin(0.1)
can2.SetLeftMargin(0.04)
can2.SetRightMargin(0.06)
can2.SetFillColor(kWhite)
gStyle.SetOptStat(0)
gStyle.SetTitleFillColor(0)
gStyle.SetTitleBorderSize(0)
#hist2d_X0_total.SetMaximum(hist2d_X0_total.GetMaximum())
# Color palette
# gStyle.SetPalette()#1
acustompalette()
ex1 = TExec("ex1", "acustompalette();")
ex1.Draw()
#for i in range(100): MyPaletteArray.append(i+1)
#gStyle.SetPalette(first_color_number);
# Log?
can2.SetLogz(plots[plot].zLog)
# Draw in colors
#frame.Draw()
#hist2d_X0_total.Draw("COLZsame") #Dummy draw to create the palette object
hist2d_X0_total.Draw("COLZ") #Dummy draw to create the palette object
# Store
can2.Update()
#Aesthetic
palette = hist2d_X0_total.GetListOfFunctions().FindObject("palette")
if palette:
palette.__class__ = TPaletteAxis
palette.SetX1NDC(0.945)
palette.SetX2NDC(0.96)
palette.SetY1NDC(0.1)
palette.SetY2NDC(0.9)
palette.GetAxis().SetTickSize(.01)
palette.GetAxis().SetTitle("")
if plots[plot].zLog:
palette.GetAxis().SetLabelOffset(-0.01)
paletteTitle = TLatex(1.12 * maxX, maxY, plots[plot].quotaName)
paletteTitle.SetTextAngle(90.)
paletteTitle.SetTextSize(0.05)
paletteTitle.SetTextAlign(31)
paletteTitle.Draw()
hist2d_X0_total.GetYaxis().SetTickLength(
hist2d_X0_total.GetXaxis().GetTickLength() / 4.)
hist2d_X0_total.GetYaxis().SetTickLength(
hist2d_X0_total.GetXaxis().GetTickLength() / 4.)
hist2d_X0_total.SetTitleOffset(0.5, "Y")
hist2d_X0_total.GetYaxis().SetTitleOffset(0.45)
#hist2d_X0_total.GetXaxis().SetTitleOffset(1.15);
#hist2d_X0_total.GetXaxis().SetNoExponent(True)
#hist2d_X0_total.GetYaxis().SetNoExponent(True)
#Add eta labels
keep_alive = []
if plots[plot].iDrawEta:
keep_alive.extend(drawEtaValues())
can2.Modified()
hist2d_X0_total.SetContour(255)
# Store
can2.Update()
can2.Modified()
can2.SaveAs("%s/%s_%s%s.pdf" % (theDirname, detector, plot, plotmat))
can2.SaveAs("%s/%s_%s%s.png" % (theDirname, detector, plot, plotmat))
#can2.SaveAs( "%s/%s_%s%s.root" % (theDirname, detector, plot, plotmat))
#Zoom in a little bit
if plot == "x_vs_z_vs_Rsum" or plot == "l_vs_z_vs_Rsum" or plot == "x_vs_z_vs_Rsumcos" or plot == "l_vs_z_vs_Rsumcos" or plot == "x_vs_z_vs_Rloc" or plot == "l_vs_z_vs_Rloc" or plot == "x_vs_z_vs_Rloccos" or plot == "l_vs_z_vs_Rloccos":
#Z+
#hist2d_X0_total.GetXaxis().SetLimits( 3100., 5200.)
if dosingledetector:
hist2d_X0_total.GetXaxis().SetRangeUser(3100., 5400.)
else:
hist2d_X0_total.GetXaxis().SetRangeUser(0., 7000.)
#Do not draw eta values in the zoom case
keep_alive = []
#hist2d_X0_total.Draw("COLZ")
can2.Update()
can2.Modified()
can2.SaveAs("%s/%s/%s_%s%s_ZplusZoom.pdf" %
(theDirname, "ZPlusZoom", detector, plot, plotmat))
can2.SaveAs("%s/%s/%s_%s%s_ZplusZoom.png" %
(theDirname, "ZPlusZoom", detector, plot, plotmat))
#Z-
#hist2d_X0_total.GetXaxis().SetLimits( 3100., 5200.)
if dosingledetector:
hist2d_X0_total.GetXaxis().SetRangeUser(-5400., -3100.)
else:
hist2d_X0_total.GetXaxis().SetRangeUser(0., -7000.)
#Do not draw eta values in the zoom case
keep_alive = []
#hist2d_X0_total.Draw("COLZ")
can2.Update()
can2.Modified()
can2.SaveAs("%s/%s/%s_%s%s_ZminusZoom.pdf" %
(theDirname, "ZMinusZoom", detector, plot, plotmat))
can2.SaveAs("%s/%s/%s_%s%s_ZminusZoom.png" %
(theDirname, "ZMinusZoom", detector, plot, plotmat))
gStyle.SetStripDecimals(True)
def create2DPlotsGeometryComparison(detector, plot,
geometryOld, geometryNew):
setTDRStyle()
print('Extracting plot: %s.'%(plot))
goodToGo, theFiles = paramsGood_(detector,plot,
geometryOld,geometryNew)
if not goodToGo:
return
gStyle.SetOptStat(False)
old2DHisto = get2DHisto_(detector,plots[plot].plotNumber,geometryOld)
new2DHisto = get2DHisto_(detector,plots[plot].plotNumber,geometryNew)
if plots[plot].iRebin:
old2DHisto.Rebin2D()
new2DHisto.Rebin2D()
def setRanges(h):
h.GetXaxis().SetRangeUser(plots[plot].xmin, plots[plot].xmax)
h.GetYaxis().SetRangeUser(plots[plot].ymin, plots[plot].ymax)
if plots[plot].histoMin != -1.:
h.SetMinimum(plots[plot].histoMin)
if plots[plot].histoMax != -1.:
h.SetMaximum(plots[plot].histoMax)
ratio2DHisto = copy.deepcopy(new2DHisto)
ratio2DHisto.Divide(old2DHisto)
# Ratio and Difference have the same call
# But different 'Palette' range so we are
# setting the range only for the Ratio
ratio2DHisto.SetMinimum(0.2)
ratio2DHisto.SetMaximum(1.8)
setRanges(ratio2DHisto)
diff2DHisto = copy.deepcopy(new2DHisto)
diff2DHisto.Add(old2DHisto,-1.0)
setRanges(diff2DHisto)
def setPadStyle():
gPad.SetLeftMargin(0.05)
gPad.SetRightMargin(0.08)
gPad.SetTopMargin(0.10)
gPad.SetBottomMargin(0.10)
gPad.SetLogz(plots[plot].zLog)
gPad.SetFillColor(kWhite)
gPad.SetBorderMode(0)
can = TCanvas('can','can',
2724,1336)
can.Divide(1,2)
can.cd(1)
setPadStyle()
gPad.SetLogz(plots[plot].zLog)
gStyle.SetOptStat(0)
gStyle.SetFillColor(kWhite)
gStyle.SetPalette(kTemperatureMap)
ratio2DHisto.SetTitle("%s, Ratio: %s/%s;%s;%s"
%(plots[plot].quotaName,
geometryOld, geometryNew,
plots[plot].abscissa,
plots[plot].ordinate))
ratio2DHisto.Draw('COLZ')
can.Update()
setUpPalette(ratio2DHisto,plot)
etasTop = []
if plots[plot].iDrawEta:
etasTop.extend(drawEtaValues())
can.cd(2)
diff2DHisto.SetTitle('%s, Difference: %s - %s %s;%s;%s'
%(plots[plot].quotaName,geometryNew,geometryOld,detector,
plots[plot].abscissa,plots[plot].ordinate))
setPadStyle()
diff2DHisto.Draw("COLZ")
can.Update()
setUpPalette(diff2DHisto,plot)
etasBottom = []
if plots[plot].iDrawEta:
etasBottom.extend(drawEtaValues())
can.Modified()
theDirname = "Images"
if not checkFile_(theDirname):
os.mkdir(theDirname)
can.SaveAs( "%s/%s_Comparison_%s_%s_vs_%s.png"
% (theDirname,detector,plot,geometryOld,geometryNew) )
gStyle.SetStripDecimals(True)
def create2DPlots(detector, plot, geometry):
"""Produce the requested plot for the specified detector.
Function that will plot the requested 2D-@plot for the
specified @detector. The specified detector could either be a
real detector or a compound one. The list of available plots
are the keys of plots dictionary (imported from plot_utils).
"""
theDirname = 'Images'
if not checkFile_(theDirname):
os.mkdir(theDirname)
goodToGo, theDetectorFilename = paramsGood_(detector, plot, geometry)
if not goodToGo:
return
theDetectorFile = TFile(theDetectorFilename)
hist_X0_total = get2DHisto_(detector,plots[plot].plotNumber,geometry)
# # properties
gStyle.SetStripDecimals(False)
# Ratio
if plots[plot].iRebin:
hist_X0_total.Rebin2D()
# stack
hist2dTitle = ('%s %s;%s;%s;%s' % (plots[plot].quotaName,
detector,
plots[plot].abscissa,
plots[plot].ordinate,
plots[plot].quotaName))
hist_X0_total.SetTitle(hist2dTitle)
hist_X0_total.SetTitleOffset(0.5,"Y")
if plots[plot].histoMin != -1.:
hist_X0_total.SetMinimum(plots[plot].histoMin)
if plots[plot].histoMax != -1.:
hist_X0_total.SetMaximum(plots[plot].histoMax)
can2name = "MBCan_2D_%s_%s" % (detector, plot)
can2 = TCanvas(can2name, can2name, 2480+248, 580+58+58)
can2.SetTopMargin(0.1)
can2.SetBottomMargin(0.1)
can2.SetLeftMargin(0.04)
can2.SetRightMargin(0.06)
can2.SetFillColor(kWhite)
gStyle.SetOptStat(0)
gStyle.SetTitleFillColor(0)
gStyle.SetTitleBorderSize(0)
# Color palette
gStyle.SetPalette(kGreyScale)
# Log?
can2.SetLogz(plots[plot].zLog)
# Draw in colors
hist_X0_total.Draw("COLZ")
# Store
can2.Update()
#Aesthetic
setUpPalette(hist_X0_total,plot)
#Add eta labels
keep_alive = []
if plots[plot].iDrawEta:
keep_alive.extend(drawEtaValues())
can2.Modified()
hist_X0_total.SetContour(255)
# Store
can2.Update()
can2.Modified()
can2.SaveAs( "%s/%s_%s_%s_bw.pdf"
% (theDirname, detector, plot, geometry))
can2.SaveAs( "%s/%s_%s_%s_bw.png"
% (theDirname, detector, plot, geometry))
gStyle.SetStripDecimals(True)
def create2DPlots(detector, plot, geometry):
"""Produce the requested plot for the specified detector.
Function that will plot the requested 2D-@plot for the
specified @detector. The specified detector could either be a
real detector or a compound one. The list of available plots
are the keys of plots dictionary (imported from plot_utils).
"""
theDirname = 'Images'
if not checkFile_(theDirname):
os.mkdir(theDirname)
goodToGo, theDetectorFilename = paramsGood_(detector, plot, geometry)
if not goodToGo:
return
theDetectorFile = TFile(theDetectorFilename)
hist_X0_total = get2DHisto_(detector,plots[plot].plotNumber,geometry)
# # properties
gStyle.SetStripDecimals(False)
# Ratio
if plots[plot].iRebin:
hist_X0_total.Rebin2D()
# stack
hist2dTitle = ('%s %s;%s;%s;%s' % (plots[plot].quotaName,
detector,
plots[plot].abscissa,
plots[plot].ordinate,
plots[plot].quotaName))
hist_X0_total.SetTitle(hist2dTitle)
hist_X0_total.SetTitleOffset(0.5,"Y")
if plots[plot].histoMin != -1.:
hist_X0_total.SetMinimum(plots[plot].histoMin)
if plots[plot].histoMax != -1.:
hist_X0_total.SetMaximum(plots[plot].histoMax)
can2name = "MBCan_2D_%s_%s" % (detector, plot)
can2 = TCanvas(can2name, can2name, 2480+248, 580+58+58)
can2.SetTopMargin(0.1)
can2.SetBottomMargin(0.1)
can2.SetLeftMargin(0.04)
can2.SetRightMargin(0.06)
can2.SetFillColor(kWhite)
gStyle.SetOptStat(0)
gStyle.SetTitleFillColor(0)
gStyle.SetTitleBorderSize(0)
# Color palette
gStyle.SetPalette(kGreyScale)
# Log?
can2.SetLogz(plots[plot].zLog)
# Draw in colors
hist_X0_total.Draw("COLZ")
# Store
can2.Update()
#Aesthetic
setUpPalette(hist_X0_total,plot)
#Add eta labels
keep_alive = []
if plots[plot].iDrawEta:
keep_alive.extend(drawEtaValues())
can2.Modified()
hist_X0_total.SetContour(255)
# Store
can2.Update()
can2.Modified()
can2.SaveAs( "%s/%s_%s_%s_bw.pdf"
% (theDirname, detector, plot, geometry))
can2.SaveAs( "%s/%s_%s_%s_bw.png"
% (theDirname, detector, plot, geometry))
gStyle.SetStripDecimals(True)
def create2DPlotsGeometryComparison(detector, plot,
geometryOld, geometryNew):
setTDRStyle()
print('Extracting plot: %s.'%(plot))
goodToGo, theFiles = paramsGood_(detector,plot,
geometryOld,geometryNew)
if not goodToGo:
return
gStyle.SetOptStat(False)
old2DHisto = get2DHisto_(detector,plots[plot].plotNumber,geometryOld)
new2DHisto = get2DHisto_(detector,plots[plot].plotNumber,geometryNew)
if plots[plot].iRebin:
old2DHisto.Rebin2D()
new2DHisto.Rebin2D()
def setRanges(h):
h.GetXaxis().SetRangeUser(plots[plot].xmin, plots[plot].xmax)
h.GetYaxis().SetRangeUser(plots[plot].ymin, plots[plot].ymax)
if plots[plot].histoMin != -1.:
h.SetMinimum(plots[plot].histoMin)
if plots[plot].histoMax != -1.:
h.SetMaximum(plots[plot].histoMax)
ratio2DHisto = copy.deepcopy(new2DHisto)
ratio2DHisto.Divide(old2DHisto)
# Ratio and Difference have the same call
# But different 'Palette' range so we are
# setting the range only for the Ratio
ratio2DHisto.SetMinimum(0.2)
ratio2DHisto.SetMaximum(1.8)
setRanges(ratio2DHisto)
diff2DHisto = copy.deepcopy(new2DHisto)
diff2DHisto.Add(old2DHisto,-1.0)
setRanges(diff2DHisto)
def setPadStyle():
gPad.SetLeftMargin(0.05)
gPad.SetRightMargin(0.08)
gPad.SetTopMargin(0.10)
gPad.SetBottomMargin(0.10)
gPad.SetLogz(plots[plot].zLog)
gPad.SetFillColor(kWhite)
gPad.SetBorderMode(0)
can = TCanvas('can','can',
2724,1336)
can.Divide(1,2)
can.cd(1)
setPadStyle()
gPad.SetLogz(plots[plot].zLog)
gStyle.SetOptStat(0)
gStyle.SetFillColor(kWhite)
gStyle.SetPalette(kTemperatureMap)
ratio2DHisto.SetTitle("%s, Ratio: %s/%s;%s;%s"
%(plots[plot].quotaName,
geometryOld, geometryNew,
plots[plot].abscissa,
plots[plot].ordinate))
ratio2DHisto.Draw('COLZ')
can.Update()
setUpPalette(ratio2DHisto,plot)
etasTop = []
if plots[plot].iDrawEta:
etasTop.extend(drawEtaValues())
can.cd(2)
diff2DHisto.SetTitle('%s, Difference: %s - %s %s;%s;%s'
%(plots[plot].quotaName,geometryNew,geometryOld,detector,
plots[plot].abscissa,plots[plot].ordinate))
setPadStyle()
diff2DHisto.Draw("COLZ")
can.Update()
setUpPalette(diff2DHisto,plot)
etasBottom = []
if plots[plot].iDrawEta:
etasBottom.extend(drawEtaValues())
can.Modified()
theDirname = "Images"
if not checkFile_(theDirname):
os.mkdir(theDirname)
can.SaveAs( "%s/%s_Comparison_%s_%s_vs_%s.png"
% (theDirname,detector,plot,geometryOld,geometryNew) )
gStyle.SetStripDecimals(True)
def create2DPlots(detector, plot, plotnum, plotmat):
"""Produce the requested plot for the specified detector.
Function that will plot the requested 2D-@plot for the
specified @detector. The specified detector could either be a
real detector or a compound one. The list of available plots
are the keys of plots dictionary (imported from plot_utils.
"""
#gStyle.Reset()
#Better to use an underscore.
plotmat = plotmat.replace(" ", "_")
if plotmat != "":
theDirname = ('Images/%s' % plotmat).replace(" ", "")
else:
theDirname = 'Images'
if not checkFile_(theDirname):
os.mkdir(theDirname)
if not os.path.isdir(('Images/%s/ZPlusZoom' % plotmat).replace(" ", "")):
os.mkdir( ('Images/%s/ZPlusZoom' % plotmat).replace(" ", "") )
if not os.path.isdir(('Images/%s/ZMinusZoom' % plotmat).replace(" ", "")):
os.mkdir( ('Images/%s/ZMinusZoom' % plotmat).replace(" ", "") )
goodToGo, theDetectorFilename = paramsGood_(detector, plot)
if not goodToGo:
return
theDetectorFile = TFile(theDetectorFilename)
prof2d_X0_det_total = TProfile2D()
prof2d_X0_det_total.Reset()
# get TProfiles
#prof2d_X0_det_total = theDetectorFile.Get('%s' % plots[plot].plotNumber)
prof2d_X0_det_total = theDetectorFile.Get('%s' % plotnum)
print "=================================================================="
print plotnum
# histos
prof2d_X0_det_total.__class__ = TProfile2D
#hist_X0_total = prof2d_X0_det_total.ProjectionXY()
# keep files live forever
files = []
if detector in COMPOUNDS.keys():
for subDetector in COMPOUNDS[detector][1:]:
# filenames of single components
subDetectorFilename = "matbdg_%s.root" % subDetector
# open file
if not checkFile_(subDetectorFilename):
print("Error, missing file %s" % subDetectorFilename)
continue
subDetectorFile = TFile(subDetectorFilename)
files.append(subDetectorFile)
print("*** Open file... %s" % subDetectorFilename)
# subdetector profiles
prof2d_X0_det_total = subDetectorFile.Get('%s' % plots[plot].plotNumber)
prof2d_X0_det_total.__class__ = TProfile2D
# add to summary histogram
hist_X0_total.Add(prof2d_X0_det_total.ProjectionXY("B_%s" % prof2d_X0_det_total.GetName()), +1.000 )
# # properties
#gStyle.SetPalette(1)
gStyle.SetStripDecimals(False)
# #
# Create "null" histo
minX = 1.03*prof2d_X0_det_total.GetXaxis().GetXmin()
maxX = 1.03*prof2d_X0_det_total.GetXaxis().GetXmax()
minY = 1.03*prof2d_X0_det_total.GetYaxis().GetXmin()
maxY = 1.03*prof2d_X0_det_total.GetYaxis().GetXmax()
frame = TH2F("frame", "", 10, minX, maxX, 10, minY, maxY);
frame.SetMinimum(0.1)
frame.SetMaximum(10.)
frame.GetXaxis().SetTickLength(frame.GetXaxis().GetTickLength()*0.50)
frame.GetYaxis().SetTickLength(frame.GetXaxis().GetTickLength()/4.)
# Ratio
if plots[plot].iRebin:
prof2d_X0_det_total.Rebin2D()
# stack
hist2dTitle = ('%s %s;%s;%s;%s' % (plots[plot].quotaName,
detector,
plots[plot].abscissa,
plots[plot].ordinate,
plots[plot].quotaName))
hist2d_X0_total = prof2d_X0_det_total
hist2d_X0_total.SetTitle(hist2dTitle)
frame.SetTitle(hist2dTitle)
frame.SetTitleOffset(0.5,"Y")
#If here you put different histomin,histomaxin plot_utils you won't see anything
#for the material plots.
if plots[plot].histoMin != -1.:
hist2d_X0_total.SetMinimum(plots[plot].histoMin)
if plots[plot].histoMax != -1.:
hist2d_X0_total.SetMaximum(plots[plot].histoMax)
#
can2name = "MBCan_2D_%s_%s_%s" % (detector, plot, plotmat)
can2 = TCanvas(can2name, can2name, 2480+248, 580+58+58)
can2.SetTopMargin(0.1)
can2.SetBottomMargin(0.1)
can2.SetLeftMargin(0.04)
can2.SetRightMargin(0.06)
can2.SetFillColor(kWhite)
gStyle.SetOptStat(0)
gStyle.SetTitleFillColor(0)
gStyle.SetTitleBorderSize(0)
#hist2d_X0_total.SetMaximum(hist2d_X0_total.GetMaximum())
# Color palette
# gStyle.SetPalette()#1
acustompalette()
ex1 = TExec("ex1","acustompalette();");
ex1.Draw();
#for i in range(100): MyPaletteArray.append(i+1)
#gStyle.SetPalette(first_color_number);
# Log?
can2.SetLogz(plots[plot].zLog)
# Draw in colors
#frame.Draw()
#hist2d_X0_total.Draw("COLZsame") #Dummy draw to create the palette object
hist2d_X0_total.Draw("COLZ") #Dummy draw to create the palette object
# Store
can2.Update()
#Aesthetic
palette = hist2d_X0_total.GetListOfFunctions().FindObject("palette")
if palette:
palette.__class__ = TPaletteAxis
palette.SetX1NDC(0.945)
palette.SetX2NDC(0.96)
palette.SetY1NDC(0.1)
palette.SetY2NDC(0.9)
palette.GetAxis().SetTickSize(.01)
palette.GetAxis().SetTitle("")
if plots[plot].zLog:
palette.GetAxis().SetLabelOffset(-0.01)
paletteTitle = TLatex(1.12*maxX, maxY, plots[plot].quotaName)
paletteTitle.SetTextAngle(90.)
paletteTitle.SetTextSize(0.05)
paletteTitle.SetTextAlign(31)
paletteTitle.Draw()
hist2d_X0_total.GetYaxis().SetTickLength(hist2d_X0_total.GetXaxis().GetTickLength()/4.)
hist2d_X0_total.GetYaxis().SetTickLength(hist2d_X0_total.GetXaxis().GetTickLength()/4.)
hist2d_X0_total.SetTitleOffset(0.5,"Y")
hist2d_X0_total.GetYaxis().SetTitleOffset(0.45);
#hist2d_X0_total.GetXaxis().SetTitleOffset(1.15);
#hist2d_X0_total.GetXaxis().SetNoExponent(True)
#hist2d_X0_total.GetYaxis().SetNoExponent(True)
#Add eta labels
keep_alive = []
if plots[plot].iDrawEta:
keep_alive.extend(drawEtaValues())
can2.Modified()
hist2d_X0_total.SetContour(255)
# Store
can2.Update()
can2.Modified()
can2.SaveAs( "%s/%s_%s%s.pdf" % (theDirname, detector, plot, plotmat))
can2.SaveAs( "%s/%s_%s%s.png" % (theDirname, detector, plot, plotmat))
#can2.SaveAs( "%s/%s_%s%s.root" % (theDirname, detector, plot, plotmat))
#Zoom in a little bit
if plot == "x_vs_z_vs_Rsum" or plot == "l_vs_z_vs_Rsum" or plot == "x_vs_z_vs_Rsumcos" or plot == "l_vs_z_vs_Rsumcos" or plot == "x_vs_z_vs_Rloc" or plot == "l_vs_z_vs_Rloc" or plot == "x_vs_z_vs_Rloccos" or plot == "l_vs_z_vs_Rloccos":
#Z+
#hist2d_X0_total.GetXaxis().SetLimits( 3100., 5200.)
hist2d_X0_total.GetXaxis().SetRangeUser( 3100., 5400.)
#Do not draw eta values in the zoom case
keep_alive = []
#hist2d_X0_total.Draw("COLZ")
can2.Update()
can2.Modified()
can2.SaveAs( "%s/%s/%s_%s%s_ZplusZoom.pdf" % (theDirname, "ZPlusZoom", detector, plot, plotmat))
can2.SaveAs( "%s/%s/%s_%s%s_ZplusZoom.png" % (theDirname, "ZPlusZoom", detector, plot, plotmat))
#Z-
#hist2d_X0_total.GetXaxis().SetLimits( 3100., 5200.)
hist2d_X0_total.GetXaxis().SetRangeUser( -5400., -3100.)
#Do not draw eta values in the zoom case
keep_alive = []
#hist2d_X0_total.Draw("COLZ")
can2.Update()
can2.Modified()
can2.SaveAs( "%s/%s/%s_%s%s_ZminusZoom.pdf" % (theDirname, "ZMinusZoom", detector, plot, plotmat))
can2.SaveAs( "%s/%s/%s_%s%s_ZminusZoom.png" % (theDirname, "ZMinusZoom", detector, plot, plotmat))
gStyle.SetStripDecimals(True)
def create2DPlots(detector, plot):
"""Produce the requested plot for the specified detector.
Function that will plot the requested 2D-@plot for the
specified @detector. The specified detector could either be a
real detector or a compound one. The list of available plots
are the keys of plots dictionary (imported from plot_utils).
"""
theDirname = 'Images'
if not checkFile_(theDirname):
os.mkdir(theDirname)
goodToGo, theDetectorFilename = paramsGood_(detector, plot)
if not goodToGo:
return
theDetectorFile = TFile(theDetectorFilename)
# get TProfiles
prof2d_X0_det_total = theDetectorFile.Get('%s' % plots[plot].plotNumber)
# histos
prof2d_X0_det_total.__class__ = TProfile2D
hist_X0_total = prof2d_X0_det_total.ProjectionXY()
# keep files live forever
files = []
if detector in COMPOUNDS.keys():
for subDetector in COMPOUNDS[detector][1:]:
# filenames of single components
subDetectorFilename = "matbdg_%s.root" % subDetector
# open file
if not checkFile_(subDetectorFilename):
print("Error, missing file %s" % subDetectorFilename)
continue
subDetectorFile = TFile(subDetectorFilename)
files.append(subDetectorFile)
print("*** Open file... %s" % subDetectorFilename)
# subdetector profiles
prof2d_X0_det_total = subDetectorFile.Get('%s' %
plots[plot].plotNumber)
prof2d_X0_det_total.__class__ = TProfile2D
# add to summary histogram
hist_X0_total.Add(
prof2d_X0_det_total.ProjectionXY(
"B_%s" % prof2d_X0_det_total.GetName()), +1.000)
# # properties
gStyle.SetPalette(1)
gStyle.SetStripDecimals(False)
# #
# Create "null" histo
minX = 1.03 * hist_X0_total.GetXaxis().GetXmin()
maxX = 1.03 * hist_X0_total.GetXaxis().GetXmax()
minY = 1.03 * hist_X0_total.GetYaxis().GetXmin()
maxY = 1.03 * hist_X0_total.GetYaxis().GetXmax()
frame = TH2F("frame", "", 10, minX, maxX, 10, minY, maxY)
frame.SetMinimum(0.1)
frame.SetMaximum(10.)
frame.GetXaxis().SetTickLength(frame.GetXaxis().GetTickLength() * 0.50)
frame.GetYaxis().SetTickLength(frame.GetXaxis().GetTickLength() / 4.)
# Ratio
if plots[plot].iRebin:
hist_X0_total.Rebin2D()
# stack
hist2dTitle = ('%s %s;%s;%s;%s' %
(plots[plot].quotaName, detector, plots[plot].abscissa,
plots[plot].ordinate, plots[plot].quotaName))
hist2d_X0_total = hist_X0_total
frame.SetTitle(hist2dTitle)
frame.SetTitleOffset(0.5, "Y")
if plots[plot].histoMin != -1.:
hist2d_X0_total.SetMinimum(plots[plot].histoMin)
if plots[plot].histoMax != -1.:
hist2d_X0_total.SetMaximum(plots[plot].histoMax)
#
can2name = "MBCan_2D_%s_%s" % (detector, plot)
can2 = TCanvas(can2name, can2name, 2480 + 248, 580 + 58 + 58)
can2.SetTopMargin(0.1)
can2.SetBottomMargin(0.1)
can2.SetLeftMargin(0.04)
can2.SetRightMargin(0.06)
can2.SetFillColor(kWhite)
gStyle.SetOptStat(0)
gStyle.SetTitleFillColor(0)
gStyle.SetTitleBorderSize(0)
# Color palette
gStyle.SetPalette(1)
# Log?
can2.SetLogz(plots[plot].zLog)
# Draw in colors
frame.Draw()
hist2d_X0_total.Draw("COLZsame") #Dummy draw to create the palette object
# Store
can2.Update()
#Aesthetic
palette = hist2d_X0_total.GetListOfFunctions().FindObject("palette")
if palette:
palette.__class__ = TPaletteAxis
palette.SetX1NDC(0.945)
palette.SetX2NDC(0.96)
palette.SetY1NDC(0.1)
palette.SetY2NDC(0.9)
palette.GetAxis().SetTickSize(.01)
palette.GetAxis().SetTitle("")
if plots[plot].zLog:
palette.GetAxis().SetLabelOffset(-0.01)
paletteTitle = TLatex(1.12 * maxX, maxY, plots[plot].quotaName)
paletteTitle.SetTextAngle(90.)
paletteTitle.SetTextSize(0.05)
paletteTitle.SetTextAlign(31)
paletteTitle.Draw()
hist2d_X0_total.GetYaxis().SetTickLength(
hist2d_X0_total.GetXaxis().GetTickLength() / 4.)
hist2d_X0_total.GetYaxis().SetTickLength(
hist2d_X0_total.GetXaxis().GetTickLength() / 4.)
hist2d_X0_total.SetTitleOffset(0.5, "Y")
hist2d_X0_total.GetXaxis().SetNoExponent(True)
hist2d_X0_total.GetYaxis().SetNoExponent(True)
#Add eta labels
keep_alive = []
if plots[plot].iDrawEta:
keep_alive.extend(drawEtaValues())
can2.Modified()
hist2d_X0_total.SetContour(255)
# Store
can2.Update()
can2.Modified()
can2.SaveAs("%s/%s_%s_bw.pdf" % (theDirname, detector, plot))
can2.SaveAs("%s/%s_%s_bw.png" % (theDirname, detector, plot))
gStyle.SetStripDecimals(True)
def create2DPlots(detector, plot):
"""Produce the requested plot for the specified detector.
Function that will plot the requested 2D-@plot for the
specified @detector. The specified detector could either be a
real detector or a compound one. The list of available plots
are the keys of plots dictionary (imported from plot_utils).
"""
theDirname = 'Images'
if not checkFile_(theDirname):
os.mkdir(theDirname)
goodToGo, theDetectorFilename = paramsGood_(detector, plot)
if not goodToGo:
return
theDetectorFile = TFile(theDetectorFilename)
# get TProfiles
prof2d_X0_det_total = theDetectorFile.Get('%s' % plots[plot].plotNumber)
# histos
prof2d_X0_det_total.__class__ = TProfile2D
hist_X0_total = prof2d_X0_det_total.ProjectionXY()
# keep files live forever
files = []
if detector in COMPOUNDS.keys():
for subDetector in COMPOUNDS[detector][1:]:
# filenames of single components
subDetectorFilename = "matbdg_%s.root" % subDetector
# open file
if not checkFile_(subDetectorFilename):
print("Error, missing file %s" % subDetectorFilename)
continue
subDetectorFile = TFile(subDetectorFilename)
files.append(subDetectorFile)
print("*** Open file... %s" % subDetectorFilename)
# subdetector profiles
prof2d_X0_det_total = subDetectorFile.Get('%s' % plots[plot].plotNumber)
prof2d_X0_det_total.__class__ = TProfile2D
# add to summary histogram
hist_X0_total.Add(prof2d_X0_det_total.ProjectionXY("B_%s" % prof2d_X0_det_total.GetName()), +1.000 )
# # properties
gStyle.SetPalette(1)
gStyle.SetStripDecimals(False)
# #
# Create "null" histo
minX = 1.03*hist_X0_total.GetXaxis().GetXmin()
maxX = 1.03*hist_X0_total.GetXaxis().GetXmax()
minY = 1.03*hist_X0_total.GetYaxis().GetXmin()
maxY = 1.03*hist_X0_total.GetYaxis().GetXmax()
# Ratio
if plots[plot].iRebin:
hist_X0_total.Rebin2D()
# stack
hist2dTitle = ('%s %s;%s;%s;%s' % (plots[plot].quotaName,
detector,
plots[plot].abscissa,
plots[plot].ordinate,
plots[plot].quotaName))
hist_X0_total.SetTitle(hist2dTitle)
hist_X0_total.SetTitleOffset(0.5,"Y")
if plots[plot].histoMin != -1.:
hist_X0_total.SetMinimum(plots[plot].histoMin)
if plots[plot].histoMax != -1.:
hist_X0_total.SetMaximum(plots[plot].histoMax)
#
can2name = "MBCan_2D_%s_%s" % (detector, plot)
can2 = TCanvas(can2name, can2name, 2480+248, 580+58+58)
can2.SetTopMargin(0.1)
can2.SetBottomMargin(0.1)
can2.SetLeftMargin(0.04)
can2.SetRightMargin(0.06)
can2.SetFillColor(kWhite)
gStyle.SetOptStat(0)
gStyle.SetTitleFillColor(0)
gStyle.SetTitleBorderSize(0)
# Color palette
gStyle.SetPalette(1)
# Log?
can2.SetLogz(plots[plot].zLog)
# Draw in colors
hist_X0_total.Draw("COLZ")
# Store
can2.Update()
#Aesthetic
palette = hist_X0_total.GetListOfFunctions().FindObject("palette")
if palette:
palette.__class__ = TPaletteAxis
palette.SetX1NDC(0.945)
palette.SetX2NDC(0.96)
palette.SetY1NDC(0.1)
palette.SetY2NDC(0.9)
palette.GetAxis().SetTickSize(.01)
palette.GetAxis().SetTitle("")
if plots[plot].zLog:
palette.GetAxis().SetLabelOffset(-0.01)
paletteTitle = TLatex(1.12*maxX, maxY, plots[plot].quotaName)
paletteTitle.SetTextAngle(90.)
paletteTitle.SetTextSize(0.05)
paletteTitle.SetTextAlign(31)
paletteTitle.Draw()
hist_X0_total.GetYaxis().SetTickLength(hist_X0_total.GetXaxis().GetTickLength()/4.)
hist_X0_total.GetYaxis().SetTickLength(hist_X0_total.GetXaxis().GetTickLength()/4.)
hist_X0_total.SetTitleOffset(0.5,"Y")
hist_X0_total.GetXaxis().SetNoExponent(True)
hist_X0_total.GetYaxis().SetNoExponent(True)
#Add eta labels
keep_alive = []
if plots[plot].iDrawEta:
keep_alive.extend(drawEtaValues())
can2.Modified()
hist_X0_total.SetContour(255)
# Store
can2.Update()
can2.Modified()
can2.SaveAs( "%s/%s_%s_bw.pdf" % (theDirname, detector, plot))
can2.SaveAs( "%s/%s_%s_bw.png" % (theDirname, detector, plot))
gStyle.SetStripDecimals(True)