def SetStyle():
gROOT.SetStyle('Plain')
gROOT.ForceStyle()
gStyle.SetTextFont(42)
gStyle.SetOptTitle(0)
gStyle.SetOptFit(1112)
gStyle.SetOptStat(1110)
gStyle.SetPadRightMargin(0.08)
gStyle.SetPadTopMargin(0.11)
gStyle.SetPadBottomMargin(0.12)
gStyle.SetTitleFont(42, 'x')
gStyle.SetTitleFont(42, 'y')
gStyle.SetTitleFont(42, 'z')
gStyle.SetTitleOffset(1.05, 'x')
gStyle.SetTitleOffset(1.00, 'y')
gStyle.SetTitleSize(0.05, 'x')
gStyle.SetTitleSize(0.05, 'y')
gStyle.SetTitleSize(0.05, 'z')
gStyle.SetLabelFont(42, 'x')
gStyle.SetLabelFont(42, 'y')
gStyle.SetLabelFont(42, 'z')
gStyle.SetLabelSize(0.05, 'x')
gStyle.SetLabelSize(0.05, 'y')
gStyle.SetLabelSize(0.05, 'z')
TGaxis.SetMaxDigits(3)
gStyle.SetStatY(0.9)
def Styling():
# Center title
gStyle.SetTitleAlign(22)
gStyle.SetTitleX(.5)
gStyle.SetTitleY(.95)
gStyle.SetTitleBorderSize(0)
# Remove stats box
gStyle.SetOptStat(0)
# Set background color to white
gStyle.SetFillColor(10)
gStyle.SetFrameFillColor(10)
gStyle.SetCanvasColor(10)
gStyle.SetPadColor(10)
gStyle.SetTitleFillColor(0)
gStyle.SetStatColor(10)
# No colored frames around plots
gStyle.SetFrameBorderMode(0)
gStyle.SetCanvasBorderMode(0)
gStyle.SetPadBorderMode(0)
# Set the default line color for a fit function to be red
gStyle.SetFuncColor(2)
# Marker settings
gStyle.SetMarkerStyle(20)
# No border on legends
gStyle.SetLegendBorderSize(0)
# Disabled for violating NOvA style guidelines
# Scientific notation on axes
TGaxis.SetMaxDigits(3)
# Axis titles
gStyle.SetTitleSize(.055, "xyz")
gStyle.SetTitleOffset(2.2, "xyz")
gStyle.SetTitleOffset(2.2, "y")
gStyle.SetTitleSize(.05, "")
gStyle.SetTitleOffset(2.2, "")
# Axis labels (numbering)
gStyle.SetLabelSize(.05, "xyz")
gStyle.SetLabelOffset(.005, "xyz")
# Thicker lines
gStyle.SetHistLineWidth(2)
gStyle.SetFrameLineWidth(2)
gStyle.SetFuncWidth(2)
# Set the number of tick marks
gStyle.SetNdivisions(506, "xyz")
# Set tick marks on all sides
gStyle.SetPadTickX(1)
gStyle.SetPadTickY(1)
# Fonts
kNovaFont = 42
gStyle.SetStatFont(kNovaFont)
gStyle.SetLabelFont(kNovaFont, "xyz")
gStyle.SetTitleFont(kNovaFont, "xyz")
gStyle.SetTitleFont(kNovaFont, "")
# Apply same settings to titles
gStyle.SetTextFont(kNovaFont)
gStyle.SetLegendFont(kNovaFont)
def Init():
from ROOT import gStyle
from ROOT import gROOT
gROOT.ProcessLine("gErrorIgnoreLevel = 2000;")
gStyle.SetCanvasColor(0);
gStyle.SetCanvasBorderMode(0);
gStyle.SetCanvasBorderSize(3);
gStyle.SetPadLeftMargin(0.125);
gStyle.SetPadBottomMargin(0.12);
gStyle.SetPadColor(0);
gStyle.SetPadBorderMode(0);
gStyle.SetFrameBorderMode(0);
gStyle.SetFrameBorderSize(0);
gStyle.SetFrameFillColor(0);
gStyle.SetOptStat(0);
gStyle.SetLabelOffset(0.005,"X");
gStyle.SetLabelSize(0.03,"X");
gStyle.SetLabelFont(42,"X");
gStyle.SetTitleOffset(.85,"X");
gStyle.SetTitleSize(0.04,"X");
gStyle.SetTitleFont(42,"X");
gStyle.SetLabelOffset(0.005,"Y");
gStyle.SetLabelSize(0.03,"Y");
gStyle.SetLabelFont(42,"Y");
gStyle.SetTitleOffset(.98,"Y");
gStyle.SetTitleSize(0.04,"Y");
gStyle.SetTitleFont(42,"Y");
gStyle.SetStatColor(0);
gStyle.SetStatBorderSize(0);
gStyle.SetTextFont(2);
gStyle.SetTextSize(.05);
gStyle.SetLegendBorderSize(1);
gStyle.SetHistLineWidth(2);
gStyle.SetTitleFillColor(0);
gStyle.SetTitleFontSize(0.06);
gStyle.SetTitleBorderSize(0);
gStyle.SetTitleAlign(13);
gStyle.SetTextAlign(22);
#RJ
gStyle.SetPadTickX(1);
gStyle.SetPadTickY(1);
gStyle.SetPadTopMargin(0.05);
gStyle.SetPadBottomMargin(0.13);
gStyle.SetPadLeftMargin(0.16);
gStyle.SetPadRightMargin(0.03);
def setPlotStyle(): gStyle.SetPadGridX(1) gStyle.SetPadGridY(1) gStyle.SetOptStat(1110) gStyle.SetLineWidth(2) gStyle.SetHistLineWidth(3) gStyle.SetPadTickX(1) gStyle.SetPadTickY(1) gStyle.SetTextFont(62) gStyle.SetLabelFont(62) gStyle.SetTitleFont(62) gStyle.SetLegendBorderSize(1) gStyle.SetLegendFont(62)
def setStyle():
gROOT.Reset()
icol = 0
# WHITE
font = 42
# Helvetica
tsize = 0.05
gStyle.SetFrameBorderMode(icol)
gStyle.SetFrameFillColor(icol)
gStyle.SetCanvasBorderMode(icol)
gStyle.SetCanvasColor(icol)
gStyle.SetPadBorderMode(icol)
gStyle.SetPadColor(icol)
gStyle.SetStatColor(icol)
gStyle.SetPaperSize(20, 26)
gStyle.SetPadTopMargin(0.05)
gStyle.SetPadRightMargin(0.08)
gStyle.SetPadBottomMargin(0.15)
gStyle.SetPadLeftMargin(0.12)
gStyle.SetTitleXOffset(1.05)
gStyle.SetTitleYOffset(0.95)
gStyle.SetTextFont(font)
gStyle.SetTextSize(tsize)
gStyle.SetLabelFont(font, "x")
gStyle.SetTitleFont(font, "x")
gStyle.SetLabelFont(font, "y")
gStyle.SetTitleFont(font, "y")
gStyle.SetLabelFont(font, "z")
gStyle.SetTitleFont(font, "z")
gStyle.SetLabelSize(tsize * 0.85, "x")
gStyle.SetTitleSize(tsize * 1.10, "x")
gStyle.SetLabelSize(tsize * 0.85, "y")
gStyle.SetTitleSize(tsize * 1.10, "y")
gStyle.SetLabelSize(tsize * 0.85, "z")
gStyle.SetTitleSize(tsize * 1.10, "z")
gStyle.SetMarkerStyle(20)
gStyle.SetMarkerSize(1.)
gStyle.SetHistLineWidth(2)
gStyle.SetLineStyleString(2, "[12 12]")
# postscript dashes
gStyle.SetEndErrorSize(0.)
# gStyle.SetOptTitle(0);
gStyle.SetOptStat(0)
gStyle.SetOptFit(0)
gStyle.SetPadTickX(1)
gStyle.SetPadTickY(1)
gStyle.SetPaintTextFormat("4.1f")
def setStyle():
TColor.InitializeColors()
stops = array('d', [
0.0000, 0.1250, 0.2500, 0.3750, 0.5000, 0.6250, 0.7500, 0.8750, 1.0000
])
red = array('d', [
0.9764, 0.9956, 0.8186, 0.5301, 0.1802, 0.0232, 0.0780, 0.0592, 0.2082
])
green = array('d', [
0.9832, 0.7862, 0.7328, 0.7492, 0.7178, 0.6419, 0.5041, 0.3599, 0.1664
])
blue = array('d', [
0.0539, 0.1968, 0.3499, 0.4662, 0.6425, 0.7914, 0.8385, 0.8684, 0.5293
])
TColor.CreateGradientColorTable(9, stops, red, green, blue, 255, 1.0)
gStyle.SetHatchesLineWidth(1)
gStyle.SetNumberContours(255)
gStyle.SetNdivisions(505, "xyz")
gStyle.SetTitleBorderSize(0)
gStyle.SetTitleColor(1)
gStyle.SetTitleStyle(3013)
gStyle.SetTitleFillColor(0)
gStyle.SetTitleX(0.05)
gStyle.SetTitleW(0.4)
gStyle.SetTitleY(0.965)
gStyle.SetTitleH(0.065)
gStyle.SetCanvasBorderMode(0)
gStyle.SetCanvasBorderSize(3)
gStyle.SetCanvasColor(0)
gStyle.SetPadColor(0)
gStyle.SetPadBorderMode(0)
gStyle.SetFuncWidth(3)
gStyle.SetPadGridY(False)
gStyle.SetPadGridX(False)
gStyle.SetFrameLineWidth(1)
gStyle.SetMarkerSize(5)
gStyle.SetPadTickX(True)
gStyle.SetPadTickY(True)
#gStyle.SetPalette(1)
gStyle.SetHistLineWidth(3)
gStyle.SetHistLineColor(1)
gStyle.SetOptStat(0)
gStyle.SetOptFit(0)
gStyle.SetStatW(0.25)
gStyle.SetStatH(0.25)
gStyle.SetStatX(0.9)
gStyle.SetStatY(0.9)
gStyle.SetStatColor(0)
gStyle.SetStatFormat("6.4g")
gStyle.SetPadTopMargin(0.05)
gStyle.SetPadRightMargin(0.05)
gStyle.SetPadBottomMargin(0.16)
gStyle.SetPadLeftMargin(0.12)
font = 42
gStyle.SetTextSize(.055)
gStyle.SetTextFont(font)
gStyle.SetLabelFont(font, "x")
gStyle.SetTitleFont(font, "x")
gStyle.SetLabelFont(font, "y")
gStyle.SetTitleFont(font, "y")
gStyle.SetLabelFont(font, "z")
gStyle.SetTitleFont(font, "z")
gStyle.SetTitleSize(.055, "x")
gStyle.SetTitleSize(.055, "y")
gStyle.SetTitleSize(.05, "z")
gStyle.SetLabelSize(.05, "x")
gStyle.SetLabelSize(.05, "y")
gStyle.SetLabelSize(.05, "z")
gStyle.SetLabelOffset(0.014, "x")
gStyle.SetLabelOffset(0.006, "y")
gStyle.SetLabelOffset(0.008, "z")
gStyle.SetTitleOffset(1, "y")
gStyle.SetTitleXOffset(1.2)
# use bold lines and markers
gStyle.SetMarkerStyle(20)
gStyle.SetMarkerColor(1)
gStyle.SetMarkerSize(1.2)
gStyle.SetLineStyleString(2, "[12 12]")
# postscript dashes
gStyle.SetPadTickX(1)
gStyle.SetPadTickY(1)
def applyRootStyle():
from ROOT import gROOT, gStyle, kWhite, kBlack, TColor
# Start from a plain default
gROOT.SetStyle("Plain")
lhcbMarkerType = 8
lhcbMarkerSize = 0.8
lhcbFont = 62
lhcbStatFontSize = 0.02
lhcbStatBoxWidth = 0.12
lhcbStatBoxHeight = 0.12
lhcbWidth = 1
lhcbTextSize = 0.05
lhcbLabelSize = 0.035
lhcbAxisLabelSize = 0.035
lhcbForeColour = kBlack
gStyle.SetFrameBorderMode(0)
gStyle.SetPadBorderMode(0)
# canvas options
gStyle.SetCanvasBorderSize(0)
gStyle.SetCanvasBorderMode(0)
# fonts
gStyle.SetTextFont(lhcbFont)
gStyle.SetTextSize(lhcbTextSize)
gStyle.SetLabelFont(lhcbFont, "x")
gStyle.SetLabelFont(lhcbFont, "y")
gStyle.SetLabelFont(lhcbFont, "z")
gStyle.SetLabelSize(lhcbLabelSize, "x")
gStyle.SetLabelSize(lhcbLabelSize, "y")
gStyle.SetLabelSize(lhcbLabelSize, "z")
gStyle.SetTitleFont(lhcbFont)
gStyle.SetTitleSize(lhcbAxisLabelSize, "x")
gStyle.SetTitleSize(lhcbAxisLabelSize, "y")
gStyle.SetTitleSize(lhcbAxisLabelSize, "z")
gStyle.SetTitleColor(kWhite)
gStyle.SetTitleFillColor(kWhite)
gStyle.SetTitleColor(kBlack)
gStyle.SetTitleBorderSize(0)
gStyle.SetTitleTextColor(kBlack)
# set title position
gStyle.SetTitleX(0.15)
gStyle.SetTitleY(0.97)
# turn off Title box
gStyle.SetTitleBorderSize(0)
gStyle.SetTitleTextColor(lhcbForeColour)
gStyle.SetTitleColor(lhcbForeColour)
# use bold lines and markers
gStyle.SetLineWidth(lhcbWidth)
gStyle.SetFrameLineWidth(lhcbWidth)
gStyle.SetHistLineWidth(lhcbWidth)
gStyle.SetFuncWidth(lhcbWidth)
gStyle.SetGridWidth(lhcbWidth)
gStyle.SetLineStyleString(2, "[12 12]")
gStyle.SetMarkerStyle(lhcbMarkerType)
gStyle.SetMarkerSize(lhcbMarkerSize)
# Grid
# gStyle.SetGridStyle(3)
# label offsets
gStyle.SetLabelOffset(0.015)
# by default, do not display histogram decorations:
gStyle.SetOptStat(1111)
# show probability, parameters and errors
gStyle.SetOptFit(1011)
# look of the statistics box:
gStyle.SetStatBorderSize(1)
gStyle.SetStatFont(lhcbFont)
gStyle.SetStatFontSize(lhcbStatFontSize)
gStyle.SetStatX(0.9)
gStyle.SetStatY(0.9)
gStyle.SetStatW(lhcbStatBoxWidth)
gStyle.SetStatH(lhcbStatBoxHeight)
# put tick marks on top and RHS of plots
gStyle.SetPadTickX(1)
gStyle.SetPadTickY(1)
# histogram divisions
gStyle.SetNdivisions(505, "x")
gStyle.SetNdivisions(510, "y")
# Style for 2D zcol plots
NRGBs = 5
NCont = 255
from array import array
stops = array('d', [0.00, 0.34, 0.61, 0.84, 1.00])
red = array('d', [0.00, 0.00, 0.87, 1.00, 0.51])
green = array('d', [0.00, 0.81, 1.00, 0.20, 0.00])
blue = array('d', [0.51, 1.00, 0.12, 0.00, 0.00])
TColor().CreateGradientColorTable(NRGBs, stops, red, green, blue, NCont)
gStyle.SetNumberContours(NCont)
# Force the style
gROOT.ForceStyle()
def rootStyle(batchMode=True):
from ROOT import gROOT, gStyle, kWhite, kBlack
# No info messages
gROOT.ProcessLine("gErrorIgnoreLevel = kWarning;")
# Batch mode (no TCanvas)
gROOT.SetBatch(batchMode)
# Start from a plain default
gROOT.SetStyle("Plain")
lhcbMarkerType = 8
lhcbMarkerSize = 0.8
lhcbFont = 62
lhcbStatFontSize = 0.02
lhcbStatBoxWidth = 0.12
lhcbStatBoxHeight = 0.12
lhcbWidth = 1
lhcbTextSize = 0.05
lhcbLabelSize = 0.035
lhcbAxisLabelSize = 0.035
lhcbForeColour = kBlack
gStyle.SetFrameBorderMode(0)
gStyle.SetPadBorderMode(0)
# canvas options
gStyle.SetCanvasBorderSize(0)
gStyle.SetCanvasBorderMode(0)
# fonts
gStyle.SetTextFont(lhcbFont)
gStyle.SetTextSize(lhcbTextSize)
gStyle.SetLabelFont(lhcbFont, "x")
gStyle.SetLabelFont(lhcbFont, "y")
gStyle.SetLabelFont(lhcbFont, "z")
gStyle.SetLabelSize(lhcbLabelSize, "x")
gStyle.SetLabelSize(lhcbLabelSize, "y")
gStyle.SetLabelSize(lhcbLabelSize, "z")
gStyle.SetTitleFont(lhcbFont)
gStyle.SetTitleSize(lhcbAxisLabelSize, "x")
gStyle.SetTitleSize(lhcbAxisLabelSize, "y")
gStyle.SetTitleSize(lhcbAxisLabelSize, "z")
gStyle.SetTitleColor(kWhite)
gStyle.SetTitleFillColor(kWhite)
gStyle.SetTitleColor(kBlack)
gStyle.SetTitleBorderSize(0)
gStyle.SetTitleTextColor(kBlack)
# set title position
gStyle.SetTitleX(0.15)
gStyle.SetTitleY(0.97)
# turn off Title box
gStyle.SetTitleBorderSize(0)
gStyle.SetTitleTextColor(lhcbForeColour)
gStyle.SetTitleColor(lhcbForeColour)
# use bold lines and markers
gStyle.SetLineWidth(lhcbWidth)
gStyle.SetFrameLineWidth(lhcbWidth)
gStyle.SetHistLineWidth(lhcbWidth)
gStyle.SetFuncWidth(lhcbWidth)
gStyle.SetGridWidth(lhcbWidth)
gStyle.SetLineStyleString(2, "[12 12]")
gStyle.SetMarkerStyle(lhcbMarkerType)
gStyle.SetMarkerSize(lhcbMarkerSize)
# label offsets
gStyle.SetLabelOffset(0.015)
# by default, do not display histogram decorations:
gStyle.SetOptStat(1111)
# show probability, parameters and errors
gStyle.SetOptFit(1011)
# look of the statistics box:
gStyle.SetStatBorderSize(1)
gStyle.SetStatFont(lhcbFont)
gStyle.SetStatFontSize(lhcbStatFontSize)
gStyle.SetStatX(0.9)
gStyle.SetStatY(0.9)
gStyle.SetStatW(lhcbStatBoxWidth)
gStyle.SetStatH(lhcbStatBoxHeight)
# put tick marks on top and RHS of plots
gStyle.SetPadTickX(1)
gStyle.SetPadTickY(1)
# histogram divisions
gStyle.SetNdivisions(505, "x")
gStyle.SetNdivisions(510, "y")
# Force the style
gROOT.ForceStyle()
def set_root_env():
#//TStyle* genieStyle = new TStyle("genieStyle", "GENIE Style")
#//set the background color to white
gStyle.SetFillColor(10)
gStyle.SetFrameFillColor(10)
gStyle.SetCanvasColor(10)
gStyle.SetPadColor(10)
gStyle.SetTitleFillColor(0)
gStyle.SetStatColor(10)
#dont put a colored frame around the plots
gStyle.SetFrameBorderMode(0)
gStyle.SetCanvasBorderMode(0)
gStyle.SetPadBorderMode(0)
gStyle.SetLegendBorderSize(3)
#use the primary color palette
#gStyle.SetPalette(1,0)
#set the default line color for a histogram to be black
gStyle.SetHistLineColor(ROOT.kBlack)
#set the default line color for a fit function to be red
gStyle.SetFuncColor(ROOT.kRed)
#make the axis labels black
gStyle.SetLabelColor(ROOT.kBlack, "xyz")
#set the default title color to be black
gStyle.SetTitleColor(ROOT.kBlack)
#set the margins
gStyle.SetPadBottomMargin(0.18)
gStyle.SetPadTopMargin(0.08)
gStyle.SetPadRightMargin(0.08)
gStyle.SetPadLeftMargin(0.17)
#set axis label and title text sizes
gStyle.SetLabelFont(42, "xyz")
gStyle.SetLabelSize(0.04, "xyz")
gStyle.SetLabelOffset(0.015, "xyz")
gStyle.SetTitleFont(42, "xyz")
gStyle.SetTitleSize(0.04, "xyz")
gStyle.SetTitleOffset(1.4, "y")
gStyle.SetTitleOffset(1.3, "x")
gStyle.SetStatFont(42)
gStyle.SetStatFontSize(0.07)
gStyle.SetTitleBorderSize(1)
gStyle.SetStatBorderSize(0)
gStyle.SetTextFont(42)
gStyle.SetTitleW(0.5)
gStyle.SetTitleH(0.1)
#set line widths
gStyle.SetFrameLineWidth(2)
gStyle.SetFuncWidth(2)
gStyle.SetHistLineWidth(2)
#set the number of divisions to show
gStyle.SetNdivisions(506, "xy")
#gStyle.SetPadTickX(-50202)
#turn off xy grids
gStyle.SetPadGridX(0)
gStyle.SetPadGridY(0)
#set the tick mark style
gStyle.SetPadTickX(1)
gStyle.SetPadTickY(1)
#turn off stats
gStyle.SetOptStat(0)
gStyle.SetOptFit(0)
#marker/line settings
#gStyle.SetMarkerStyle(20)
gStyle.SetMarkerSize(.95) #0.7
gStyle.SetLineWidth(2)
gStyle.SetErrorX(0)
gStyle.SetHistLineStyle(0) #It was 3 for a dotted line
#done
gStyle.cd()
gROOT.ForceStyle()
def draw_model_perf():
gROOT.SetBatch()
gStyle.SetOptStat(0)
gStyle.SetOptTitle(0)
gStyle.SetTextFont(42)
gStyle.SetTitleFont(42)
gStyle.SetLabelFont(42)
gStyle.SetPalette(kDarkBodyRadiator)
trees_dir = "/mnt/temp/mkabus/val-20201209/trees"
suffix = "filter4_poo0_drop0.00_depth4_batch0_scaler0_useSCMean1_useSCFluc1_pred_doR1" \
"_dophi0_doz0/"
pdf_dir_90 = "%s/phi90_r17_z17_%s" % (trees_dir, suffix)
pdf_dir_180 = "%s/phi180_r33_z33_%s" % (trees_dir, suffix)
filename = "model_perf_90-180"
file_formats = ["png"] # "pdf" - lasts long
nevs_90 = [10000, 18000] # 5000
nevs_180 = [10000, 18000]
nevs = nevs_90 + nevs_180
pdf_files_90 = [
"%s/pdfmaps_nEv%d.root" % (pdf_dir_90, nev) for nev in nevs_90
]
pdf_files_180 = [
"%s/pdfmaps_nEv%d.root" % (pdf_dir_180, nev) for nev in nevs_180
]
pdf_file_names = pdf_files_90 + pdf_files_180
grans = [90, 90, 180, 180]
colors = [kBlue, kOrange, kGreen, kRed]
markers_list = [(kFullSquare, kOpenSquare), (kFullCircle, kOpenCircle),
(kFullTriangleUp, kOpenTriangleUp),
(kFullDiamond, kOpenDiamond)]
var_name = "flucDistRDiff"
y_vars = ["rmsd", "means"]
y_vars_names = ["RMSE", "#mu"]
# y_labels = ["#it{RMSE} (cm)", "Mean (cm)"]
x_vars = ["rBinCenter", "fsector"]
x_vars_short = ["r"] #, "fsector"]
x_labels = ["#it{r} (cm)", "fsector"] # TODO: what units?
# "r_rmsd": 33, 195.0, 245.5, 20, # 83.5, 254.5, 200,
# "r_rmsd": "33, 83.5, 110, 200, 0.000, 0.06",
hist_strs = {
"r_rmsd": "33, 83.5, 245.5, 200, 0.000, 0.06",
"fsector_rmsd": "90, -1.0, 19, 200, 0.00, 0.1",
"r_means": "33, 83.5, 245.5, 200, -0.06, 0.06",
"fsector_means": "90, -1.0, 19, 200, -0.07, 0.01",
"r_rmsd_means": "33, 83.5, 245.5, 200, -0.06, 0.06"
}
date = datetime.date.today().strftime("%Y%m%d")
# flucDistR_entries>50
# deltaSCBinCenter>0.0121 && deltaSCBinCenter<0.0122
# deltaSCBinCenter>0.020 && deltaSCBinCenter<0.023
# rBinCenter > 200.0 && deltaSCBinCenter>0.04 && deltaSCBinCenter<0.057
# deltaSCBinCenter > 0.06 && deltaSCBinCenter < 0.07
delta_sc_str = "#int_{#it{r}, #it{#varphi}, #it{z}} " +\
"#frac{#it{#rho}_{SC} - #it{<#rho>}_{SC}}{#it{<#rho>}_{SC}}"
cuts = {
"deltaSC": {
"r_lim": (0.05, 0.07),
"fsector_lim": (0.00, 0.05),
"desc": lambda x: "%.2f < %s < %.2f" % (x[0], delta_sc_str, x[1])
},
"z": {
"r_lim": (0.0, 5.0),
"fsector_lim": (0.0, 5.0),
"desc": lambda x: "%.2f < #it{z} < %.2f cm" % x
},
"r": {
"r_lim": (0.0, 110.0),
"fsector_lim": (86.0, 86.1),
"desc": lambda x: "%.2f cm < #it{r} < %.2f cm" % x
},
"fsector": {
"r_lim": (9.00, 9.05),
"fsector_lim": (0.0, 20.0),
"desc": lambda x: "sector %d" % int(round(x[0]))
},
"phi": {
"r_lim": (3.1, 2.9),
"fsector_lim": (3.1, 2.9),
"desc": lambda x: "%.2f < #it{#varphi} < %.2f" % x
}
}
cut_fsector = "zBinCenter > %.2f && zBinCenter < %.2f" % cuts["z"]["fsector_lim"] +\
" && rBinCenter > %.2f && rBinCenter < %.2f" % cuts["r"]["fsector_lim"] +\
" && deltaSCBinCenter > %.2f && deltaSCBinCenter < %.2f" %\
cuts["deltaSC"]["fsector_lim"] +\
" && %s_rmsd > 0.0" % var_name
cut_r = "zBinCenter > %.2f && zBinCenter < %.2f" % cuts["z"]["r_lim"] +\
" && abs(phiBinCenter - %.1f) < %.1f" % cuts["phi"]["r_lim"] +\
" && deltaSCBinCenter > %.2f && deltaSCBinCenter < %.2f" %\
cuts["deltaSC"]["r_lim"] +\
" && %s_rmsd > 0.0" % var_name
cuts_list = [cut_r, cut_fsector]
# for y_var, y_label in zip(y_vars, y_labels):
y_label = "#it{RMSE} and #it{#mu} (cm)"
canvas, leg = setup_canvas("perf_%s" % y_label)
pdf_files = [
TFile.Open(pdf_file_name, "read") for pdf_file_name in pdf_file_names
]
trees = [pdf_file.Get("pdfmaps") for pdf_file in pdf_files]
variables = zip(x_vars, x_vars_short, x_labels, cuts_list)
for x_var, x_var_short, x_label, cut in variables:
hist_str = hist_strs["%s_%s_%s" % (x_var_short, y_vars[0], y_vars[1])]
hists = []
styles = enumerate(zip(nevs, colors, markers_list, trees, grans))
for ind, (nev, color, markers, tree, gran) in styles:
tree.SetMarkerColor(color)
tree.SetMarkerSize(2)
gran_str = "180#times 33 #times 33" if gran == 180 else "90#times 17 #times 17"
for y_ind, (y_var, y_var_name,
marker) in enumerate(zip(y_vars, y_vars_names,
markers)):
tree.SetMarkerStyle(marker)
hist_def = "th_%d_%d(%s)" % (ind, y_ind, hist_str)
tree.Draw("%s_%s:%s>>%s" % (var_name, y_var, x_var, hist_def),
cut, "prof")
hists.append(tree.GetHistogram())
leg.AddEntry(hists[-1], "%s" % y_var_name, "P")
leg.AddEntry("", "%d, %s" % (nev, gran_str), "")
for hist in hists:
setup_hist(x_label, y_label, hist)
hist.SetMaximum(0.06)
hist.SetMinimum(-0.06)
hist.Draw("same")
leg.Draw()
#tex = add_alice_text()
#tex.Draw()
txt = add_cut_desc(cuts, x_var_short)
txt.Draw()
for ff in file_formats:
canvas.SaveAs(
"%s_%s_%s_%s_%s.%s" %
(date, filename, x_var_short, y_vars[0], y_vars[1], ff))
for pdf_file in pdf_files:
pdf_file.Close()
def initialization(batch=True, font=default_font):
'''-- ROOT initialization --'''
print "Initializing ROOT ..."
# general
gROOT.Reset()
gROOT.SetBatch(batch)
gROOT.SetStyle("Plain")
# gStyle
gStyle.SetFillColor(0)
gStyle.SetCanvasColor(10)
gStyle.SetLineWidth(1)
gStyle.SetPalette(8)
gStyle.SetTextFont(font)
#gStyle.SetTextSize(30)
# Frame
gStyle.SetFrameBorderMode(0)
gStyle.SetFrameFillColor(0)
# Pad
gStyle.SetPadBorderMode(0)
gStyle.SetPadColor(0)
gStyle.SetPadBottomMargin(0.1)
gStyle.SetPadTopMargin(0.01)
gStyle.SetPadLeftMargin(0.1)
gStyle.SetPadRightMargin(0.01)
gStyle.SetPadTickX(1) # make ticks be on all 4 sides.
gStyle.SetPadTickY(1)
gStyle.SetPadGridX(0)
gStyle.SetPadGridY(0)
# histogram
gStyle.SetHistFillStyle(0)
gStyle.SetOptTitle(0)
gStyle.SetTitleSize(0.22)
gStyle.SetTitleFontSize(10)
gStyle.SetTitleFont(font)
gStyle.SetTitleFont(font, "xyz")
gStyle.SetTitleYOffset(1.0)
gStyle.SetTitleXOffset(1.0)
gStyle.SetTitleXSize(0.04)
gStyle.SetTitleYSize(0.04)
gStyle.SetTitleX(.15)
gStyle.SetTitleY(.98)
gStyle.SetTitleW(.70)
gStyle.SetTitleH(.05)
# statistics box
gStyle.SetOptStat(0)
gStyle.SetStatFont(font)
gStyle.SetStatFontSize(10)
gStyle.SetStatX(.91)
gStyle.SetStatY(.90)
gStyle.SetStatW(.15)
gStyle.SetStatH(.15)
# axis labels
gStyle.SetLabelFont(font)
gStyle.SetLabelFont(font, "xyz")
gStyle.SetLabelSize(10, "xyz")
# gStyle.SetGridColor(1)
gStyle.SetLegendBorderSize(1)
#----------------------------------------------------------------------------- # pad margins #----------------------------------------------------------------------------- gStyle.SetPadTopMargin(0.05) gStyle.SetPadRightMargin(0.05) gStyle.SetPadBottomMargin(0.16) gStyle.SetPadLeftMargin(0.16) gStyle.SetTitleOffset(1.2, 'x') gStyle.SetTitleOffset(1.2, 'y') #----------------------------------------------------------------------------- # fonts #----------------------------------------------------------------------------- font = 42 # 22 = Times, 42 = Helvetica gStyle.SetTextFont(font) gStyle.SetLabelFont(font, 'xyz') gStyle.SetTitleFont(font, 'xyz') gStyle.SetTitleFont(font, 't') gStyle.SetStatFont(font) #----------------------------------------------------------------------------- # text sizes #----------------------------------------------------------------------------- gStyle.SetTextSize(0.06) gStyle.SetLabelSize(0.06, 'xyz') gStyle.SetTitleSize(0.06, 'xyz') gStyle.SetTitleSize(0.06, 't') gStyle.SetStatFontSize(0.04) #gStyle.SetTextSizePixels(30)
def tdrstyle():
gROOT.SetStyle("Plain")
gStyle.SetAxisColor(1, "XYZ")
gStyle.SetCanvasColor(0)
#gStyle.SetCanvasBorderSize(10)
gStyle.SetCanvasBorderMode(0)
gStyle.SetCanvasDefH(700)
gStyle.SetCanvasDefW(700)
gStyle.SetCanvasDefX(0)
gStyle.SetCanvasDefY(0)
gStyle.SetFitFormat("5.4g")
gStyle.SetFuncColor(2)
gStyle.SetFuncStyle(1)
gStyle.SetFuncWidth(1)
gStyle.SetFrameBorderMode(0)
gStyle.SetFrameBorderSize(1)
gStyle.SetFrameFillStyle(0)
gStyle.SetFrameFillColor(0)
gStyle.SetFrameLineColor(1)
gStyle.SetFrameLineStyle(1) # 0?
gStyle.SetFrameLineWidth(1) # 1?
gStyle.SetGridColor(0)
gStyle.SetGridStyle(3)
gStyle.SetGridWidth(1)
#gStyle.SetHistFillColor(1)
#gStyle.SetHistFillStyle(0)
gStyle.SetHistLineColor(1)
gStyle.SetHistLineStyle(0)
gStyle.SetHistLineWidth(1)
gStyle.SetLabelColor(1, "XYZ")
gStyle.SetLabelFont(42,"XYZ")
gStyle.SetLabelOffset(0.007,"XYZ") # 0.010?
gStyle.SetLabelSize(0.05,"XYZ") # 0.04?
gStyle.SetLegendBorderSize(0)
gStyle.SetLegendFillColor(0)
gStyle.SetLegendFont(42)
gStyle.SetMarkerSize(1.0)
gStyle.SetMarkerStyle(20)
gStyle.SetLineColor(1)
gStyle.SetLineWidth(2)
#gStyle.SetLineScalePS(2)
gStyle.SetOptDate(0)
gStyle.SetOptFile(0)
gStyle.SetOptFit(1)
gStyle.SetOptStat(0)
gStyle.SetOptTitle(0)
#gStyle.SetOptLogx(0)
#gStyle.SetOptLogy(0)
#gStyle.SetOptLogz(0)
gStyle.SetPadColor(0)
gStyle.SetPadBorderMode(0)
gStyle.SetPadBorderSize(10)
gStyle.SetPadTopMargin(0.05) # 0.08?
gStyle.SetPadBottomMargin(0.13)
gStyle.SetPadLeftMargin(0.16)
gStyle.SetPadRightMargin(0.03) # 0.05?
gStyle.SetPadGridX(0)
gStyle.SetPadGridY(0)
gStyle.SetPadTickX(1)
gStyle.SetPadTickY(1)
gStyle.SetStatColor(0)
gStyle.SetStatFont(42)
gStyle.SetStatFontSize(0.025)
gStyle.SetStatTextColor(1)
gStyle.SetStatFormat("6.4g")
gStyle.SetStatBorderSize(1)
gStyle.SetStatH(0.1)
gStyle.SetStatW(0.15)
#gStyle.SetStatX(0)
#gStyle.SetStatY(0)
#gStyle.SetTextSize(0.055)
gStyle.SetTextFont(42)
gStyle.SetTitleBorderSize(0)
gStyle.SetTitleColor(1)
gStyle.SetTitleFont(42)
gStyle.SetTitleColor(1,"XYZ")
gStyle.SetTitleFont(42,"XYZ")
gStyle.SetTitleSize(0.06,"XYZ") # 0.05?
#gStyle.SetTitleOffset(1.4,"XYZ")
gStyle.SetTitleOffset(0.9,"X")
gStyle.SetTitleOffset(1.20,"Y")
gStyle.SetTitleFillColor(10)
gStyle.SetTitleFontSize(0.05)
gStyle.SetTitleTextColor(1)
#gStyle.SetTitleH(0)
#gStyle.SetTitleW(0)
#gStyle.SetTitleX(0)
#gStyle.SetTitleY(0.985)
#gStyle.SetTitleStyle(1001)
gStyle.SetPalette(1)
#gStyle.SetNdivisions(510, "XYZ") # 505?
gStyle.SetNdivisions(505, "XYZ")
gStyle.SetEndErrorSize(2) # 2?
#gStyle.SetErrorMarker(20)
#gStyle.SetErrorX(0.)
#gStyle.SetPaperSize(20.,20.)
gStyle.SetStripDecimals(1)
gStyle.SetTickLength(0.03, "XYZ")
return 1
def setStyle():
gStyle.SetCanvasColor(0)
gStyle.SetCanvasBorderSize(10)
gStyle.SetCanvasBorderMode(0)
gStyle.SetCanvasDefH(700)
gStyle.SetCanvasDefW(700)
gStyle.SetPadColor(0)
gStyle.SetPadBorderSize(10)
gStyle.SetPadBorderMode(0)
gStyle.SetPadBottomMargin(0.13)
gStyle.SetPadTopMargin(0.08)
gStyle.SetPadLeftMargin(0.15)
gStyle.SetPadRightMargin(0.05)
gStyle.SetPadGridX(0)
gStyle.SetPadGridY(0)
gStyle.SetPadTickX(1)
gStyle.SetPadTickY(1)
gStyle.SetFrameFillStyle(0)
gStyle.SetFrameFillColor(0)
gStyle.SetFrameLineColor(1)
gStyle.SetFrameLineStyle(0)
gStyle.SetFrameLineWidth(1)
gStyle.SetFrameBorderSize(10)
gStyle.SetFrameBorderMode(0)
gStyle.SetNdivisions(505)
gStyle.SetLineWidth(2)
gStyle.SetHistLineWidth(2)
gStyle.SetFrameLineWidth(2)
gStyle.SetLegendFillColor(root.kWhite)
gStyle.SetLegendFont(42)
gStyle.SetMarkerSize(1.2)
gStyle.SetMarkerStyle(20)
gStyle.SetLabelSize(0.040, "X")
gStyle.SetLabelSize(0.040, "Y")
gStyle.SetLabelOffset(0.010, "X")
gStyle.SetLabelOffset(0.010, "Y")
gStyle.SetLabelFont(42, "X")
gStyle.SetLabelFont(42, "Y")
gStyle.SetTitleBorderSize(0)
gStyle.SetTitleFont(42)
gStyle.SetTitleFont(42, "X")
gStyle.SetTitleFont(42, "Y")
gStyle.SetTitleSize(0.045, "X")
gStyle.SetTitleSize(0.045, "Y")
gStyle.SetTitleOffset(1.4, "X")
gStyle.SetTitleOffset(1.4, "Y")
gStyle.SetTextSize(0.055)
gStyle.SetTextFont(42)
gStyle.SetOptStat(0)
def CLICdpStyle():
gROOT.SetStyle("Plain")
gStyle.SetCanvasColor(root.kWhite)
gStyle.SetFrameFillColor(root.kWhite)
gStyle.SetStatColor(root.kWhite)
gStyle.SetPadColor(root.kWhite)
gStyle.SetFillColor(10)
gStyle.SetTitleFillColor(root.kWhite)
gStyle.SetPaperSize(20, 26)
gStyle.SetDrawBorder(0)
gStyle.SetCanvasBorderMode(0)
gStyle.SetPadBorderMode(0)
gStyle.SetFrameBorderMode(0)
gStyle.SetLegendBorderSize(0)
gStyle.SetTextSize(0.05)
gStyle.SetTitleSize(0.06, "xyz")
gStyle.SetLabelSize(0.06, "xyz")
gStyle.SetLabelOffset(0.015, "xyz")
gStyle.SetTitleOffset(1.2, "yz")
gStyle.SetTitleOffset(1.17, "x")
font = 42
gStyle.SetTitleFont(font)
gStyle.SetTitleFontSize(0.06)
gStyle.SetStatFont(font)
gStyle.SetStatFontSize(0.07)
gStyle.SetTextFont(font)
gStyle.SetLabelFont(font, "xyz")
gStyle.SetTitleFont(font, "xyz")
gStyle.SetTitleBorderSize(0)
gStyle.SetStatBorderSize(1)
gStyle.SetMarkerStyle(1)
gStyle.SetLineWidth(2)
gStyle.SetMarkerSize(1.2)
gStyle.SetPalette(1)
gStyle.SetOptTitle(0)
gStyle.SetOptStat(0)
gStyle.SetOptFit(0)
gStyle.SetEndErrorSize(5)
gStyle.SetHistLineWidth(2)
gStyle.SetFrameLineWidth(2)
gStyle.SetFuncWidth(2)
gStyle.SetHistLineColor(root.kBlack)
gStyle.SetFuncColor(root.kBlack)
gStyle.SetLabelColor(root.kBlack, "xyz")
gStyle.SetPadBottomMargin(0.18)
gStyle.SetPadTopMargin(0.11)
gStyle.SetPadRightMargin(0.08)
gStyle.SetPadLeftMargin(0.17)
gStyle.SetNdivisions(506, "xy")
gStyle.SetPadGridX(0)
gStyle.SetPadGridY(0)
gStyle.SetPadTickX(1)
gStyle.SetPadTickY(1)
gStyle.SetCanvasDefW(800)
gStyle.SetCanvasDefH(700)
gROOT.ForceStyle()
import ROOT
from ROOT import gROOT, gStyle
import argparse, os, json
import numpy as np
gROOT.SetBatch(True)
gStyle.SetOptStat(0)
gStyle.SetOptFit(0)
gStyle.SetOptTitle(0)
gStyle.SetTextFont(43)
gStyle.SetTitleOffset(0.86, "X")
gStyle.SetTitleOffset(1.8, "Y")
# gStyle.SetPadLeftMargin(0.18)
gStyle.SetPadLeftMargin(0.19)
# gStyle.SetPadBottomMargin(0.15)
gStyle.SetPadBottomMargin(0.12)
gStyle.SetPadTopMargin(0.08)
# gStyle.SetPadRightMargin(0.08)
gStyle.SetPadRightMargin(0.1)
gStyle.SetMarkerSize(0.5)
gStyle.SetHistLineWidth(2)
gStyle.SetTitleSize(0.05, "XYZ")
gStyle.SetLabelSize(0.04, "XYZ")
gStyle.SetNdivisions(506, "XYZ")
gStyle.SetLegendBorderSize(0)
leftMargin = 0.14
colors = {
from ROOT import TH1F, TH2F, TColor, gStyle, SetOwnership, TCanvas, TGraph
from numpy import array, pi, log10, zeros, arange, arccos, cos
import pyIrfLoader
d2r = pi / 180.
r2d = 180. / pi
gStyle.SetTextFont(132)
gStyle.SetTitleFont(132, 'xyz')
gStyle.SetLabelFont(132, 'xyz')
gStyle.SetCanvasColor(0)
gStyle.SetFrameBorderMode(0)
gStyle.SetFrameFillColor(0)
gStyle.SetCanvasBorderMode(0)
gStyle.SetPadBorderSize(1)
gStyle.SetLegendBorderSize(1)
gStyle.SetOptStat(0)
gStyle.SetTitleBorderSize(1)
gStyle.SetMarkerSize(0.9)
gStyle.SetFuncWidth(1)
#function to create npts points from xmin to xmax (inclusive) with equal spacing in log10 space
def log10_array(npts, xmin, xmax):
xstep = log10(xmax / xmin) / (npts - 1)
return xmin * 10**(arange(npts, dtype=float) * xstep)
####################################################################################
####################################################################################
#############start functions for making effective area plots########################
