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 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)