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 configure_draw():
gROOT.SetStyle('Plain')
gStyle.SetEndErrorSize(0.)
gStyle.SetOptStat(0)
gStyle.SetPalette(1)
gStyle.SetTitleBorderSize(0)
gStyle.SetTitleFont(22, '')
gStyle.SetTitleSize(0.06, '')
gStyle.SetTitleX(0.1)
gStyle.SetTitleW(0.8)
def looks():
gROOT.ProcessLine('TGaxis::SetMaxDigits(3)')
gStyle.SetFrameBorderMode(0)
gStyle.SetFrameFillColor(0)
gStyle.SetCanvasBorderMode(0)
gStyle.SetCanvasColor(0)
gStyle.SetPadBorderMode(0)
gStyle.SetPadColor(0)
gStyle.SetStatColor(0)
gStyle.SetTitleFont(42, '')
gStyle.SetLabelFont(42, 'x')
gStyle.SetTitleFont(42, 'x')
gStyle.SetLabelFont(42, 'y')
gStyle.SetTitleFont(42, 'y')
gStyle.SetLabelFont(42, 'z')
gStyle.SetTitleFont(42, 'z')
gStyle.SetLabelSize(0.048, 'x')
gStyle.SetTitleSize(0.048, 'x')
gStyle.SetLabelSize(0.048, 'y')
gStyle.SetTitleSize(0.048, 'y')
gStyle.SetLabelSize(0.048, 'z')
gStyle.SetTitleSize(0.048, 'z')
gStyle.SetTitleSize(0.048, '')
gStyle.SetTitleAlign(23)
gStyle.SetTitleX(0.5)
gStyle.SetTitleBorderSize(0)
gStyle.SetTitleFillColor(0)
gStyle.SetTitleStyle(0)
gStyle.SetOptStat('emr')
gStyle.SetOptStat(0)
gStyle.SetStatBorderSize(0)
gStyle.SetStatFont(42)
gStyle.SetStatFontSize(0.048)
gStyle.SetStatY(0.9)
gStyle.SetStatX(0.86)
gStyle.SetHistLineColor(ROOT.kBlue + 2)
gStyle.SetLegendBorderSize(0)
gStyle.SetLegendFillColor(0)
gStyle.SetFuncWidth(2)
gStyle.SetFuncColor(2)
gStyle.SetPadTopMargin(0.08)
gStyle.SetPadBottomMargin(0.12)
gStyle.SetPadLeftMargin(0.12)
gStyle.SetPadRightMargin(0.06)
gStyle.SetCanvasDefX(400)
gStyle.SetCanvasDefY(20)
gStyle.SetCanvasDefH(420)
gStyle.SetCanvasDefW(610)
gStyle.SetFrameBorderMode(0)
gStyle.SetFrameLineWidth(2)
gStyle.SetHistLineWidth(2)
gStyle.SetTitleOffset(1.16, 'y')
gStyle.SetTitleOffset(1.20, 'x')
from ROOT import TH1F, TH2F, TFile, TTree
from ROOT import gROOT, gStyle
from ROOT import TCanvas
gROOT.SetBatch(1)
gROOT.Reset()
gStyle.SetCanvasColor(0)
gStyle.SetFrameBorderMode(0)
gStyle.SetOptStat(0)
gStyle.SetTitleX(0.5) # title X location
gStyle.SetTitleY(0.96) # title Y location
gStyle.SetPaintTextFormat(".2f")
def draw_underflow_overflow(h1):
h1.GetXaxis().SetRange(0, h1.GetNbinsX() + 1)
h1.Draw()
return h1
def fill_underflow_overflow(h1):
nbin = h1.GetNbinsX()
h1.Fill(h1.GetBinCenter(1),h1.GetBinContent(0))
h1.Fill(h1.GetBinCenter(nbin),h1.GetBinContent(nbin+1))
h1.Draw()
return h1
Canv = TCanvas("c1","c1",0,0,800,600)
f_out = TFile(filename,"recreate")
for sample in sampleName:
# dest='ML',
# help='Which ML method? DNN, BDT, MLP, or kNN')
(options, args) = parser.parse_args()
argv = []
folder = "projections_" + str(options.n) + "_" + str(options.layers)
#ML = str(options.ML)
eff = options.eff
gStyle.SetPadLeftMargin(0.1)
gStyle.SetPadBottomMargin(0.1)
gStyle.SetPadRightMargin(0.18)
gStyle.SetPadTopMargin(0.1)
gStyle.SetTitleX(0.3)
gStyle.SetTitleAlign(23)
planes = [
"Hit_plane0", "Hit_plane1", "Hit_plane2", "Hit_plane3", "Hit_plane4",
"Hit_plane5", "Hit_plane6", "Hit_plane7"
]
names = [("Hit_plane0", "X0"), ("Hit_plane1", "X1"), ("Hit_plane2", "U0"),
("Hit_plane3", "V0"), ("Hit_plane4", "U1"), ("Hit_plane5", "V1"),
("Hit_plane6", "X2"), ("Hit_plane7", "X3")]
samples = [("sig", "Signal"), ("bkg", "Background")]
MLlist = ["DNN", "BDT", "MLP"]
#MLlist = ["DNN"]
for ML in MLlist:
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()
from ROOT import TCanvas, TPad, TFormula, TF1, TPaveLabel, TH1F, TFile
from ROOT import TGraphErrors
from ROOT import gROOT, gBenchmark, gStyle, gPad
from array import array
import math
#plot the final kinematic distributions for the particles in exclusive phi production
gStyle.SetOptStat(0000)
gStyle.SetPalette(55)
gStyle.SetStatStyle(1001)
gStyle.SetStatFont(62)
gStyle.SetStatBorderSize(2)
gStyle.SetTitleX(.20)
gStyle.SetTitleY(.99)
gStyle.SetPadBottomMargin(0.1)
gStyle.SetPadLeftMargin(0.12)
gROOT.ForceStyle()
fin = TFile("../../analysis_out_skim4_rgaf18_final.root", "read")
hw = fin.Get("W")
hmissemm2 = fin.Get("missemm2")
hphim = fin.Get("phim")
hmm2phim = fin.Get("missephim")
hmismm2phim = fin.Get("mismm2phim")
hmissmm2 = fin.Get("missmm2")
cmm = TCanvas("cmm", "cmm", 900, 900)
cmm.Divide(2, 2)
cmm.cd(1)
hmissemm2.Draw("colz")
from ROOT import gROOT, gStyle
# old mva > 0.795, 0.884, 0.921
# mva2 > 0.85 , 0.90 , 0.94
# 3hit < 2 , 1 , 0.8
### style parameters
gROOT.SetStyle("Plain")
gROOT.SetBatch(True)
gStyle.SetLegendFillColor(0)
gStyle.SetLegendBorderSize(0)
gStyle.SetStatBorderSize(0)
gStyle.SetTitleBorderSize(0)
gStyle.SetPadTickX(1)
gStyle.SetPadTickY(1)
gStyle.SetTitleX(0.125)
gStyle.SetTitleY(0.96)
gStyle.SetTitleW(0.8)
gStyle.SetTextFont(42)
gStyle.SetLegendFont(42)
#gStyle.SetLabelFont(42, 'xy')
#gStyle.SetTitleFont(42, 'xy')
tauScale = '0.03'
shiftedMet = 'sqrt(pow(mex+' + tauScale + '*l1Px+' + tauScale + '*l2Px,2)+pow(mey+' + tauScale + '*l1Py+' + tauScale + '*l2Py,2))'
baseline = ' l1Pt>35 && l2Pt>35 && abs(l1Eta)<2.1 && abs(l2Eta)<2.1 && diTauCharge==0 '
#### should be redundant once skimmed
baseline += ' && l2againstElectronLooseMVA3>0.5 '
baseline += ' && l1LooseEle>0.5 && l2LooseEle>0.5'
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 drawMultiGraph(self,TMG,figdir='',SetLogy=False, SetLogx=False, abscissaIsTime = True, drawLines=True, xAxisLabel=None,yAxisLabel=None):
'''
draw TMultiGraph with legend and output as pdf
Default is that abscissa is calendar time.
'''
debugMG = False
if not TMG.GetListOfGraphs(): return # empty
title = TMG.GetTitle()
name = TMG.GetName()
if SetLogx: name += '_logx'
if SetLogy: name += '_logy'
if debugMG: print 'graphUtils.drawMultiGraph',title,name,'TMG.GetListOfGraphs()',TMG.GetListOfGraphs(),'TMG.GetListOfGraphs().GetSize()',TMG.GetListOfGraphs().GetSize()
nGraphs = TMG.GetListOfGraphs().GetSize()
pdf = os.path.join( figdir , name + '.pdf' )
ps = os.path.join( figdir , name + '.ps')
xsize,ysize = 1100,850 # landscape style
noPopUp = True
if noPopUp : gROOT.ProcessLine("gROOT->SetBatch()")
canvas = TCanvas(pdf,title,xsize,ysize)
canvas.SetLogy(SetLogy)
canvas.SetLogx(SetLogx)
# move title to left in order to put legend above plot
gStyle.SetTitleX(0.3)
x1 = 0.5
x2 = x1 + .5
y1 = 0.9
y2 = y1 + .1
lg = TLegend(x1,y1,x2,y2)
NGraph = 0
for g in TMG.GetListOfGraphs():
NGraph += 1
t = g.GetTitle()
lg.AddEntry(g, t, "LP" )
if abscissaIsTime : self.fixTimeDisplay(g)
if NGraph>6: lg.SetNColumns(2)
dOption = "AP"
if drawLines: dOption += "L"
# complicated monkey business because of idiotic way that logY is set
if SetLogy:
ymi,yma = 1.e20,1.e-20
for g in TMG.GetListOfGraphs():
x,y = self.getPoints(g)
ymi = min(ymi,min(y))
yma = max(yma,max(y))
if ymi<=0: ymi = 0.1
ymi = ymi/2.
yma = 2.*yma
TMG.SetMinimum(ymi)
TMG.SetMaximum(yma)
for g in TMG.GetListOfGraphs():
g.SetMinimum(ymi)
g.SetMaximum(yma)
if "A" in dOption:
g.SetTitle( TMG.GetTitle() )
if xAxisLabel is not None: g.GetXaxis().SetTitle(xAxisLabel)
if yAxisLabel is not None: g.GetYaxis().SetTitle(yAxisLabel)
g.Draw(dOption)
dOption = dOption.replace("A","")
else:
TMG.Draw(dOption)
if xAxisLabel is not None: TMG.GetXaxis().SetTitle(xAxisLabel)
if yAxisLabel is not None: TMG.GetYaxis().SetTitle(yAxisLabel)
if abscissaIsTime : self.fixTimeDisplay(TMG)
self.labelTMultiGraph(TMG,debug=debugMG)
lg.Draw()
canvas.Draw()
canvas.SetGrid(1)
canvas.SetTicks(1)
canvas.cd()
canvas.Modified()
canvas.Update()
if 0:
canvas.Print(pdf,'pdf')
else:
canvas.Print(ps,'Landscape')
os.system('ps2pdf ' + ps + ' ' + pdf)
if os.path.exists(pdf): os.system('rm ' + ps)
if debugMG: print 'graphUtils.drawMultiGraph',title,'complete'
return
from ROOT import gStyle gStyle.SetStatColor(0) gStyle.SetTitleColor(0) gStyle.SetCanvasColor(0) gStyle.SetPadColor(0) gStyle.SetPadBorderMode(0) gStyle.SetCanvasBorderMode(0) gStyle.SetFrameBorderMode(0) gStyle.SetOptStat(1110) gStyle.SetStatH(0.3) gStyle.SetStatW(0.3) gStyle.SetTitleColor(1) gStyle.SetTitleFillColor(0) gStyle.SetTitleY(1.) gStyle.SetTitleX(.1) gStyle.SetTitleBorderSize(0) gStyle.SetHistLineWidth(2) gStyle.SetLineWidth(2) #gStyle.SetFillColor(0) gStyle.SetFrameFillColor(0)
from ROOT import gROOT, gStyle
# old mva > 0.795, 0.884, 0.921
# mva2 > 0.85 , 0.90 , 0.94
# 3hit < 2 , 1 , 0.8
### style parameters
gROOT.SetStyle("Plain")
gROOT.SetBatch(True)
gStyle.SetLegendFillColor(0)
gStyle.SetLegendBorderSize(0)
gStyle.SetStatBorderSize(0)
gStyle.SetTitleBorderSize(0)
gStyle.SetPadTickX(1)
gStyle.SetPadTickY(1)
gStyle.SetTitleX(0.14) ## to be changed to 0.15
gStyle.SetTitleY(0.96)
gStyle.SetTitleW(0.8)
gStyle.SetTextFont(42)
gStyle.SetLegendFont(42)
gStyle.SetLabelFont(42, 'xy')
gStyle.SetTitleFont(42, 'xy')
tauScale = '0.03'
shiftedMet = 'sqrt(pow(mex+'+tauScale+'*l1Px+'+tauScale+'*l2Px,2)+pow(mey+'+tauScale+'*l1Py+'+tauScale+'*l2Py,2))'
baseline = ' l1Pt>45 && l2Pt>45 && abs(l1Eta)<2.1 && abs(l2Eta)<2.1 && diTauCharge!=0 '
#### should be redundant once skimmed
baseline += ' && l1LooseEle>0.5 && l2LooseEle>0.5 '
baseline += ' && l1LooseMu >0.5 && l2LooseMu >0.5 '
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)
def drawHist(hist, confL, name, colz=True, contour=False, dec=False):
can = TCanvas()
can.cd()
if (dec):
gStyle.SetPaintTextFormat("3.2f")
else:
gStyle.SetPaintTextFormat("3.0f")
if "metOmeff" in hist.GetName():
gStyle.SetPaintTextFormat("3.2f")
gStyle.SetStatX(50)
#colors = array("i",[3,30, 40, 20, 46, 5, 6, 7, 8, 9,11,4]) #colors >= #levels - 1
#gStyle.SetPalette(len(colors), colors)
#levels = array("d",[.01, .20, 0.50, 1, 2, 3, 4, 7,9,10])
#hist.SetContour(len(levels), levels)
hist.SetMarkerSize(1.5)
if colz:
hist.Draw("colz,same")
hist.Draw("text30, same")
if SigSample == "GG_ttn1":
xlabel = 850
ylabel = 1400
elif SigSample == "GG_2stepWZ":
xlabel = 600
ylabel = 1400
elif SigSample == "GG_SLN1":
xlabel = 400
ylabel = 1000
elif SigSample == "BB_onestepC1":
xlabel = 350
ylabel = 300
xlabel = 650
ylabel = 1500
if contour:
g = TGraph2D()
TH2toTGraph2D(hist, g)
Ghisto = g.GetHistogram()
level = array('d', [1.64, 3])
Ghisto.SetLineWidth(4)
Ghisto.SetContour(2, level)
Ghisto.Draw("cont1,list ,same")
# confL.SetLineColor(1)
# confL.Draw("colz,same")
tex1 = TLatex()
if SigSample == "BB_onestepC1":
tex1.DrawLatex(xlabel, ylabel + 50, "#color[3]{1.64#sigma}")
else:
tex1.DrawLatex(xlabel, ylabel + 100, "#color[3]{1.64#sigma}")
tex1.DrawLatex(xlabel, ylabel, "#color[2]{3#sigma}")
#sprintf(&title[0],"MET>%d GeV",met);
#myText(.2,.84,kBlue,title);
#sprintf(&title[0],"MEFF>%d GeV",meff);
#myText(.2,.79,kBlue,title);
#sprintf(&title[0],"Bkg: %3.1f evts",_bkg);
#myText(.2,.74,kBlue,title);
# myLineText(.26, .8 , 0.05, kGreen,"1.64#sigma")
# myLineText(.26, .75, 0.05, kRed, "3#sigma")
hist.GetYaxis().SetTitleOffset(1.3)
hist.GetXaxis().SetTitle("Gluino Mass (GeV)")
hist.GetYaxis().SetTitle("Neutralino Mass (GeV)")
gStyle.SetTitleX(100.0) # move title off plot
gStyle.SetTitleW(0.0) # title width =0
tex2 = TLatex()
if SigSample == "BB_onestepC1":
tex2.DrawLatex(
xlabel, ylabel + 100,
name + " " + str(luminosity) + " fb^{-1}: " + hist.GetName())
else:
tex2.DrawLatex(
xlabel, ylabel + 200,
name + " " + str(luminosity) + " fb^{-1}: " + hist.GetName())
# Draw m_gl - m_N1 = 2 (m_W+m_b)
line = TLine(600, 429.6, 1800, 1629.6)
line.SetLineColor(16)
line.SetLineStyle(7)
line.Draw("same")
th1 = TLatex(
1450, 1340,
"m_{#tilde{g}} - m_{#tilde{#chi}_{1}^{0}} = 2 #left(m_{W}+m_{b}#right)"
)
th1.SetTextAngle(30)
th1.SetTextColor(16)
th1.SetTextSize(0.03)
th1.Draw("same")
# Draw m_gl - m_N1 = 2 m_t
line2 = TLine(600, 255, 2050, 1680)
line2.SetLineColor(16)
line2.SetLineStyle(7)
line2.Draw("same")
th2 = TLatex(1550, 1240,
"m_{#tilde{g}} - m_{#tilde{#chi}_{1}^{0}} = 2 m_{t}")
th2.SetTextAngle(30)
th2.SetTextColor(16)
th2.SetTextSize(0.03)
# th2.Draw("same")
if "Dm" in hist.GetName():
line3 = TLine(600, 25, 2100, 25)
line3.SetLineColor(16)
line3.SetLineStyle(7)
line3.Draw("same")
th3 = TLatex(
1600, 40,
"#Delta m = m_{#tilde{#chi}_{1}^{0}} - #left(m_{#tilde{g}} -2 m_{t} #right)"
)
# th3.SetTextAngle(32)
th3.SetTextColor(16)
th3.SetTextSize(0.03)
th3.Draw("same")
hist.GetYaxis().SetTitle(
"#Delta m_{#tilde{#chi}_{1}^{0}} (above diagonal) (GeV)")
can.Update()
directory = "/afs/cern.ch/user/o/othrif/workarea/results/v48/Plots/" + Dirbase #+"/Lumi"+str(luminosity)
if not os.path.exists(directory):
os.makedirs(directory)
can.SaveAs(directory + "/" + name + "_" + hist.GetName() + ".pdf")
def initStyle():
gROOT.SetStyle("Plain")
gStyle.SetOptFit()
# For the canvas:
gStyle.SetCanvasBorderMode(0)
gStyle.SetCanvasColor(TStyle.kWhite)
gStyle.SetCanvasDefH(600) #Height of canvas
gStyle.SetCanvasDefW(600) #Width of canvas
gStyle.SetCanvasDefX(0) #POsition on screen
gStyle.SetCanvasDefY(0)
# For the Pad:
gStyle.SetPadBorderMode(0)
# gStyle.SetPadBorderSize(Width_t size = 1)
gStyle.SetPadColor(TStyle.kWhite)
gStyle.SetPadGridX(False)
gStyle.SetPadGridY(False)
gStyle.SetGridColor(0)
gStyle.SetGridStyle(3)
gStyle.SetGridWidth(1)
# For the frame:
gStyle.SetFrameBorderMode(0)
gStyle.SetFrameBorderSize(1)
gStyle.SetFrameFillColor(0)
gStyle.SetFrameFillStyle(0)
gStyle.SetFrameLineColor(1)
gStyle.SetFrameLineStyle(1)
gStyle.SetFrameLineWidth(1)
# For the histo:
# gStyle.SetHistFillColor(1)
# gStyle.SetHistFillStyle(0)
gStyle.SetHistLineColor(1)
gStyle.SetHistLineStyle(0)
gStyle.SetHistLineWidth(1)
# gStyle.SetLegoInnerR(Float_t rad = 0.5)
# gStyle.SetNumberContours(Int_t number = 20)
gStyle.SetEndErrorSize(2)
#gStyle.SetErrorMarker(20)
gStyle.SetErrorX(0.)
gStyle.SetMarkerStyle(20)
#gStyle.SetMarkerStyle(20)
#For the fit/function:
gStyle.SetOptFit(1)
gStyle.SetFitFormat("5.4g")
gStyle.SetFuncColor(2)
gStyle.SetFuncStyle(1)
gStyle.SetFuncWidth(1)
#For the date:
gStyle.SetOptDate(0)
# gStyle.SetDateX(Float_t x = 0.01)
# gStyle.SetDateY(Float_t y = 0.01)
# For the statistics box:
gStyle.SetOptStat(1)
gStyle.SetStatBorderSize(0)
gStyle.SetOptStat('emr')
gStyle.SetOptFile(0)
gStyle.SetStatColor(TStyle.kWhite)
gStyle.SetStatFont(42)
gStyle.SetStatFontSize(0.025)
gStyle.SetStatTextColor(1)
gStyle.SetStatFormat("6.4g")
gStyle.SetStatH(0.1)
gStyle.SetStatW(0.15)
# gStyle.SetStatStyle(Style_t style = 1001)
# gStyle.SetStatX(Float_t x = 0)
# gStyle.SetStatY(Float_t y = 0)
# Margins:
#gStyle.SetPadTopMargin(0.15)
gStyle.SetPadTopMargin(0.05)
#gStyle.SetPadBottomMargin(0.13)
gStyle.SetPadLeftMargin(0.16)
gStyle.SetPadRightMargin(0.04) # top group adaption, original is 0.02
gStyle.SetPadBottomMargin(0.13)
# For the Global title:
gStyle.SetOptTitle(1)
gStyle.SetTitleFont(42)
#gStyle.SetTitleColor(1)
gStyle.SetTitleTextColor(1)
#gStyle.SetTitleFillColor(0)
gStyle.SetTitleFontSize(0.07)
# gStyle.SetTitleH(0) # Set the height of the title box
# gStyle.SetTitleW(0) # Set the width of the title box
gStyle.SetTitleX(0.14) # Set the position of the title box
gStyle.SetTitleY(1.025) # Set the position of the title box
# gStyle.SetTitleStyle(Style_t style = 1001)
gStyle.SetTitleBorderSize(0)
# For the axis titles:
gStyle.SetTitleColor(1, "XYZ")
gStyle.SetTitleFont(42, "XYZ")
#gStyle.SetTitleSize(0.06, "XYZ")
gStyle.SetTitleSize(0.05, "XYZ")
# gStyle.SetTitleXSize(Float_t size = 0.02) # Another way to set the size?
# gStyle.SetTitleYSize(Float_t size = 0.02)
gStyle.SetTitleXOffset(0.9)
gStyle.SetTitleYOffset(1.25)
#gStyle.SetTitleOffset(1.1, "Y") # Another way to set the Offset
# For the axis labels:
gStyle.SetLabelColor(1, "XYZ")
gStyle.SetLabelFont(42, "XYZ")
gStyle.SetLabelOffset(0.007, "XYZ")
gStyle.SetLabelSize(0.04, "XYZ")
#gStyle.SetLabelSize(0.04, "XYZ")
# For the axis:
gStyle.SetAxisColor(1, "XYZ")
gStyle.SetStripDecimals(True)
gStyle.SetTickLength(0.03, "XYZ")
gStyle.SetNdivisions(510, "XYZ")
gStyle.SetPadTickX(
1) # To get tick marks on the opposite side of the frame
gStyle.SetPadTickY(1)
# Change for log plots:
gStyle.SetOptLogx(0)
gStyle.SetOptLogy(0)
gStyle.SetOptLogz(0)
gStyle.SetPalette(1) #(1,0)
# another top group addition
gStyle.SetHatchesSpacing(1.0)
# Postscript options:
gStyle.SetPaperSize(20., 20.)
#gStyle.SetPaperSize(TStyle.kA4)
#gStyle.SetPaperSize(27., 29.7)
#TGaxis.SetMaxDigits(3)
# gStyle.SetLineScalePS(Float_t scale = 3)
# gStyle.SetLineStyleString(Int_t i, const char* text)
# gStyle.SetHeaderPS(const char* header)
# gStyle.SetTitlePS(const char* pstitle)
#gStyle.SetColorModelPS(1)
# gStyle.SetBarOffset(Float_t baroff = 0.5)
# gStyle.SetBarWidth(Float_t barwidth = 0.5)
# gStyle.SetPaintTextFormat(const char* format = "g")
# gStyle.SetPalette(Int_t ncolors = 0, Int_t* colors = 0)
# gStyle.SetTimeOffset(Double_t toffset)
# gStyle.SetHistMinimumZero(kTRUE)
#gStyle.cd()
print "TDR Style initialized"
