def set_meeting_style():
from ROOT import gStyle
gStyle.SetOptDate(0)
gStyle.SetOptTitle(0)
gStyle.SetOptStat(1111)
gStyle.SetStatBorderSize(1)
gStyle.SetStatColor(10)
gStyle.SetStatFont(42)
gStyle.SetStatFontSize(0.03)
gStyle.SetOptFit(1111)
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 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')
def initialize2(self):
# For the canvas:
gStyle.SetCanvasBorderMode(0)
gStyle.SetCanvasColor(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(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.SetFrameFillStyle(1000)
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)
#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.SetOptFile(0)
gStyle.SetOptStat(
0) # To display the mean and RMS: SetOptStat("mr")
gStyle.SetStatColor(kWhite)
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.SetStatStyle(Style_t style = 1001)
# gStyle.SetStatX(Float_t x = 0)
# gStyle.SetStatY(Float_t y = 0)
# Margins:
gStyle.SetPadTopMargin(0.05)
gStyle.SetPadBottomMargin(0.13)
gStyle.SetPadLeftMargin(0.16)
gStyle.SetPadRightMargin(0.02)
# For the Global title:
gStyle.SetOptTitle(0)
gStyle.SetTitleFont(42)
gStyle.SetTitleColor(1)
gStyle.SetTitleTextColor(1)
gStyle.SetTitleFillColor(10)
gStyle.SetTitleFontSize(0.05)
# gStyle.SetTitleH(0) # Set the height of the title box
# gStyle.SetTitleW(0) # Set the width of the title box
# gStyle.SetTitleX(0) # Set the position of the title box
# gStyle.SetTitleY(0.985) # Set the position of the title box
# gStyle.SetTitleStyle(Style_t style = 1001)
# gStyle.SetTitleBorderSize(2)
# For the axis titles:
gStyle.SetTitleColor(1, "XYZ")
gStyle.SetTitleFont(42, "XYZ")
gStyle.SetTitleSize(0.06, "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.05, "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)
# Postscript options:
gStyle.SetPaperSize(20., 20.)
def setTDRStyle(force):
gStyle.SetCanvasBorderMode(0)
gStyle.SetCanvasColor(kWhite)
gStyle.SetCanvasDefH(600)
gStyle.SetCanvasDefW(600)
gStyle.SetCanvasDefX(0)
gStyle.SetCanvasDefY(0)
gStyle.SetPadBorderMode(0)
gStyle.SetPadColor(kWhite)
gStyle.SetPadGridX(False)
gStyle.SetPadGridY(False)
gStyle.SetGridColor(0)
gStyle.SetGridStyle(3)
gStyle.SetGridWidth(1)
gStyle.SetFrameBorderMode(0)
gStyle.SetFrameBorderSize(1)
gStyle.SetFrameFillColor(0)
gStyle.SetFrameFillStyle(0)
gStyle.SetFrameLineColor(1)
gStyle.SetFrameLineStyle(1)
gStyle.SetFrameLineWidth(1)
if force:
gStyle.SetHistLineColor(1)
gStyle.SetHistLineStyle(0)
gStyle.SetHistLineWidth(1)
gStyle.SetEndErrorSize(2)
gStyle.SetErrorX(0.)
gStyle.SetMarkerStyle(20)
gStyle.SetOptFit(1)
gStyle.SetFitFormat("5.4g")
gStyle.SetFuncColor(2)
gStyle.SetFuncStyle(1)
gStyle.SetFuncWidth(1)
gStyle.SetOptDate(0)
gStyle.SetOptFile(0)
gStyle.SetOptStat(0)
gStyle.SetStatColor(kWhite)
gStyle.SetStatFont(42)
gStyle.SetStatFontSize(0.04)
gStyle.SetStatTextColor(1)
gStyle.SetStatFormat("6.4g")
gStyle.SetStatBorderSize(1)
gStyle.SetStatH(0.1)
gStyle.SetStatW(0.2)
gStyle.SetPadTopMargin(0.05)
gStyle.SetPadBottomMargin(0.13)
gStyle.SetPadLeftMargin(0.16)
gStyle.SetPadRightMargin(0.04)
gStyle.SetOptTitle(0)
gStyle.SetTitleFont(42)
gStyle.SetTitleColor(1)
gStyle.SetTitleTextColor(1)
gStyle.SetTitleFillColor(10)
gStyle.SetTitleFontSize(0.05)
gStyle.SetTitleColor(1, "XYZ")
gStyle.SetTitleFont(42, "XYZ")
gStyle.SetTitleSize(0.06, "XYZ")
gStyle.SetTitleXOffset(0.9)
gStyle.SetTitleYOffset(1.25)
gStyle.SetLabelColor(1, "XYZ")
gStyle.SetLabelFont(42, "XYZ")
gStyle.SetLabelOffset(0.007, "XYZ")
gStyle.SetLabelSize(0.05, "XYZ")
gStyle.SetAxisColor(1, "XYZ")
gStyle.SetStripDecimals(True)
gStyle.SetTickLength(0.03, "XYZ")
gStyle.SetNdivisions(510, "XYZ")
gStyle.SetPadTickX(1)
gStyle.SetPadTickY(1)
gStyle.SetOptLogx(0)
gStyle.SetOptLogy(0)
gStyle.SetOptLogz(0)
gStyle.SetPaperSize(20.,20.)
gROOT.ForceStyle()
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")
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 studyVqqResolution(rootFile):
#get all from file
histos = {}
inF = TFile.Open(rootFile)
keys = inF.GetListOfKeys()
for k in keys:
obj = inF.Get(k.GetName())
obj.SetDirectory(0)
histos[k.GetName()] = obj
inF.Close()
#plot
gROOT.SetBatch()
gROOT.SetStyle('Plain')
gStyle.SetOptStat(0)
gStyle.SetOptFit(1111)
gStyle.SetOptTitle(0)
gStyle.SetStatFont(42)
kin = ['', '30to40', '40to50', '50to75', '75to100', '100toInf']
for k in kin:
c = TCanvas('c', 'c', 600, 600)
c.cd()
c.SetCanvasSize(1000, 500)
c.SetWindowSize(1000, 500)
c.Divide(2, 1)
c.cd(1)
histos['deta' + k + 'barrel'].SetLineWidth(2)
histos['deta' + k + 'barrel'].SetTitle('barrel')
histos['deta' + k + 'barrel'].Draw('hist')
histos['deta' + k + 'endcap'].SetLineWidth(2)
histos['deta' + k + 'endcap'].SetLineStyle(7)
histos['deta' + k + 'endcap'].SetTitle('endcap')
histos['deta' + k + 'endcap'].Draw('histsame')
leg = TLegend(0.6, 0.92, 0.9, 0.98)
leg.SetFillStyle(0)
leg.SetBorderSize(0)
leg.SetTextFont(42)
leg.AddEntry(histos['deta' + k + 'barrel'], 'barrel', 'f')
leg.AddEntry(histos['deta' + k + 'endcap'], 'endcap', 'f')
leg.SetNColumns(2)
leg.Draw()
drawHeader()
c.cd(2)
histos['dphi' + k + 'barrel'].SetLineWidth(2)
histos['dphi' + k + 'barrel'].SetTitle('barrel')
histos['dphi' + k + 'barrel'].Draw('hist')
histos['dphi' + k + 'endcap'].SetLineWidth(2)
histos['dphi' + k + 'endcap'].SetLineStyle(7)
histos['dphi' + k + 'endcap'].SetTitle('endcap')
histos['dphi' + k + 'endcap'].Draw('histsame')
c.Modified()
c.Update()
c.SaveAs('dr_%s.png' % k)
labels = []
responseVars = ['dpt', 'den', 'dphi', 'deta', 'dr']
for r in responseVars:
barrelResponse = TGraphErrors()
barrelResponse.SetName(r + 'barrelresponse')
barrelResponse.SetLineWidth(2)
barrelResponse.SetFillStyle(0)
barrelResponse.SetMarkerStyle(20)
barrelCoreResponse = barrelResponse.Clone(r + 'barrelcoreresponse')
endcapResponse = TGraphErrors()
endcapResponse.SetName(r + 'endcapresponse')
endcapResponse.SetLineWidth(2)
endcapResponse.SetFillStyle(0)
endcapResponse.SetMarkerStyle(24)
endcapCoreResponse = endcapResponse.Clone(r + 'endcapresponse')
for k in kin:
c.cd()
c.Clear()
c.SetWindowSize(1000, 500)
c.Divide(2, 1)
for i in [1, 2]:
c.cd(i)
reg = 'barrel'
if i == 2: reg = 'endcap'
h = histos[r + k + reg]
x = RooRealVar("x",
h.GetXaxis().GetTitle(),
h.GetXaxis().GetXmin(),
h.GetXaxis().GetXmax())
data = RooDataHist("data", "dataset with x", RooArgList(x), h)
frame = x.frame()
RooAbsData.plotOn(data, frame,
RooFit.DataError(RooAbsData.SumW2))
mean1 = RooRealVar("mean1", "mean1", 0, -0.5, 0.5)
sigma1 = RooRealVar("sigma1", "sigma1", 0.1, 0.01, 1.0)
gauss1 = RooGaussian("g1", "g", x, mean1, sigma1)
if r == 'dpt' or r == 'den':
mean2 = RooRealVar("mean2", "mean2", 0, -0.5, 0.5)
sigma2 = RooRealVar("sigma2", "sigma2", 0.1, 0.01, 1.0)
alphacb = RooRealVar("alphacb", "alphacb", 1, 0.1, 3)
ncb = RooRealVar("ncb", "ncb", 4, 1, 100)
gauss2 = RooCBShape("cb2", "cb", x, mean2, sigma2, alphacb,
ncb)
else:
mean1.setRange(0, 0.5)
mean2 = RooRealVar("mean2", "mean", 0, 0, 1)
sigma2 = RooRealVar("sigma2", "sigma", 0.1, 0.01, 1.0)
gauss2 = RooGaussian("g2", "g", x, mean2, sigma2)
frac = RooRealVar("frac", "fraction", 0.9, 0.0, 1.0)
if data.sumEntries() < 100:
frac.setVal(1.0)
frac.setConstant(True)
model = RooAddPdf("sum", "g1+g2", RooArgList(gauss1, gauss2),
RooArgList(frac))
status = model.fitTo(data, RooFit.Save()).status()
if status != 0: continue
model_cdf = model.createCdf(RooArgSet(x))
cl = 0.90
ul = 0.5 * (1.0 + cl)
closestToCL = 1.0
closestToUL = -1
closestToMedianCL = 1.0
closestToMedian = -1
for ibin in xrange(1, h.GetXaxis().GetNbins() * 10):
xval = h.GetXaxis().GetXmin() + (
ibin - 1) * h.GetXaxis().GetBinWidth(ibin) / 10.
x.setVal(xval)
cdfValToCL = math.fabs(model_cdf.getVal() - ul)
if cdfValToCL < closestToCL:
closestToCL = cdfValToCL
closestToUL = xval
cdfValToCL = math.fabs(model_cdf.getVal() - 0.5)
if cdfValToCL < closestToMedianCL:
closestToMedianCL = cdfValToCL
closestToMedian = xval
RooAbsPdf.plotOn(model, frame)
frame.Draw()
if i == 1: drawHeader()
labels.append(TPaveText(0.6, 0.92, 0.9, 0.98, 'brNDC'))
ilab = len(labels) - 1
labels[ilab].SetName(r + k + 'txt')
labels[ilab].SetBorderSize(0)
labels[ilab].SetFillStyle(0)
labels[ilab].SetTextFont(42)
labels[ilab].SetTextAlign(12)
kinReg = k.replace('to', '-')
kinReg = kinReg.replace('Inf', '#infty')
labels[ilab].AddText('[' + reg + '] ' + kinReg)
labels[ilab].Draw()
resolutionVal = math.fabs(closestToUL - closestToMedian)
responseGr = barrelResponse
responseCoreGr = barrelCoreResponse
coreResolutionVal = sigma1.getVal()
coreResolutionErr = sigma1.getError()
if frac.getVal() < 0.7 and (sigma2.getVal() < sigma1.getVal()):
coreResolutionVal = sigma2.getVal()
coreResolutionErr = sigma2.getError()
if i == 2:
responseGr = endcapResponse
responseCoreGr = endcapCoreResponse
if k != '':
nrespPts = responseGr.GetN()
kinAvg = 150
kinWidth = 50
if k == '30to40':
kinAvg = 35
kinWidth = 5
if k == '40to50':
kinAvg = 45
kinWidth = 5
if k == '50to75':
kinAvg = 62.5
kinWidth = 12.5
elif k == '75to100':
kinAvg = 87.5
kinWidth = 12.5
responseGr.SetPoint(nrespPts, kinAvg, resolutionVal)
responseCoreGr.SetPoint(nrespPts, kinAvg,
coreResolutionVal)
responseCoreGr.SetPointError(nrespPts, kinWidth,
coreResolutionErr)
labels.append(TPaveText(0.15, 0.7, 0.4, 0.9, 'brNDC'))
ilab = len(labels) - 1
labels[ilab].SetName(r + k + 'fitrestxt')
labels[ilab].SetBorderSize(0)
labels[ilab].SetFillStyle(0)
labels[ilab].SetTextFont(42)
labels[ilab].SetTextAlign(12)
labels[ilab].AddText('Gaussian #1 (f=%3.3f)' % frac.getVal())
labels[ilab].AddText('#mu=%3.3f#pm%3.3f' %
(mean1.getVal(), mean1.getError()))
labels[ilab].AddText('#sigma=%3.3f#pm%3.3f' %
(sigma1.getVal(), sigma1.getError()))
labels[ilab].AddText('Gaussian #2 (f=%3.3f)' %
(1 - frac.getVal()))
labels[ilab].AddText('#mu=%3.3f#pm%3.3f' %
(mean2.getVal(), mean2.getError()))
labels[ilab].AddText('#sigma=%3.3f#pm%3.3f' %
(sigma2.getVal(), sigma2.getError()))
labels[ilab].Draw()
c.Modified()
c.Update()
c.SaveAs(r + 'res_' + k + '.png')
frame = TGraphErrors()
frame.SetPoint(0, 0, 0)
frame.SetPoint(1, 200, 0.3)
frame.SetMarkerStyle(1)
frame.SetFillStyle(0)
frame.SetName('frame')
cresp = TCanvas('cresp', 'cresp', 500, 500)
cresp.cd()
frame.Draw('ap')
barrelResponse.Draw('pl')
endcapResponse.Draw('pl')
frame.GetXaxis().SetTitle("Quark transverse momentum [GeV]")
frame.GetYaxis().SetTitle("Resolution %3.2f C.L." % cl)
frame.GetYaxis().SetTitleOffset(1.4)
frame.GetYaxis().SetNdivisions(10)
drawHeader()
leg = TLegend(0.6, 0.92, 0.9, 0.98)
leg.SetFillStyle(0)
leg.SetBorderSize(0)
leg.SetTextFont(42)
leg.AddEntry(barrelResponse, 'barrel', 'fp')
leg.AddEntry(endcapResponse, 'endcap', 'fp')
leg.SetNColumns(2)
leg.Draw()
cresp.Modified()
cresp.Update()
cresp.SaveAs(r + 'res_evol.png')
frameCore = frame.Clone('framecore')
cresp.Clear()
frameCore.Draw('ap')
barrelCoreResponse.Draw('pl')
endcapCoreResponse.Draw('pl')
frameCore.GetXaxis().SetTitle("Quark transverse momentum [GeV]")
frameCore.GetYaxis().SetTitle("Core resolution")
frameCore.GetYaxis().SetTitleOffset(1.4)
frameCore.GetYaxis().SetNdivisions(10)
frameCore.GetYaxis().SetRangeUser(0, 0.2)
drawHeader()
leg = TLegend(0.6, 0.92, 0.9, 0.98)
leg.SetFillStyle(0)
leg.SetBorderSize(0)
leg.SetTextFont(42)
leg.AddEntry(barrelCoreResponse, 'barrel', 'fp')
leg.AddEntry(endcapCoreResponse, 'endcap', 'fp')
leg.SetNColumns(2)
leg.Draw()
cresp.Modified()
cresp.Update()
cresp.SaveAs(r + 'rescore_evol.png')
bosons = ['h', 'z', 'w']
kin = ['', '50', '100']
region = ['', 'bb', 'eb', 'ee']
for k in kin:
for r in region:
c = TCanvas('c', 'c', 600, 600)
c.cd()
histos['mjj' + k + r].Rebin()
histos['mjj' + k + r].Draw()
ic = 1
leg = TLegend(0.6, 0.92, 0.9, 0.98)
leg.SetFillStyle(0)
leg.SetBorderSize(0)
leg.SetTextFont(42)
leg.AddEntry(histos['mjj' + k + r], 'inclusive', 'f')
for b in bosons:
if histos[b + 'mjj' + k + r].Integral() <= 0: continue
ic = ic + 1
histos[b + 'mjj' + k + r].Rebin()
histos[b + 'mjj' + k + r].SetLineColor(ic)
histos[b + 'mjj' + k + r].SetLineWidth(2)
histos[b + 'mjj' + k + r].SetMarkerColor(ic)
histos[b + 'mjj' + k + r].SetMarkerStyle(1)
histos[b + 'mjj' + k + r].SetFillStyle(3000 + ic)
histos[b + 'mjj' + k + r].SetFillColor(ic)
histos[b + 'mjj' + k + r].Draw('histsame')
leg.AddEntry(histos[b + 'mjj' + k + r], b, "f")
leg.SetNColumns(ic)
leg.Draw()
drawHeader()
labels.append(TPaveText(0.65, 0.8, 0.9, 0.9, 'brNDC'))
ilab = len(labels) - 1
labels[ilab].SetName(k + r + 'mjj')
labels[ilab].SetBorderSize(0)
labels[ilab].SetFillStyle(0)
labels[ilab].SetTextFont(42)
labels[ilab].SetTextAlign(12)
regionTitle = "inclusive"
if r == 'bb': regionTitle = 'barrel-barrel'
if r == 'eb': regionTitle = 'endcap-barrel'
if r == 'ee': regionTitle = 'endcap-endcap'
labels[ilab].AddText(regionTitle)
ptthreshold = 30
if k != '': ptthreshold = float(k)
labels[ilab].AddText('p_{T}>%3.0f GeV' % ptthreshold)
labels[ilab].Draw()
c.Modified()
c.Update()
c.SaveAs('mjj' + k + r + '.png')
massResolutionGrs = []
for r in region:
massResolution = TGraphErrors()
massResolution.SetName(r + 'dm')
massResolution.SetLineWidth(2)
massResolution.SetFillStyle(0)
massResolution.SetMarkerStyle(20 + len(massResolutionGrs))
massResolution.SetMarkerColor(1 + len(massResolutionGrs))
massResolution.SetLineColor(1 + len(massResolutionGrs))
massResolution.SetFillColor(1 + len(massResolutionGrs))
massResolutionGrs.append(massResolution)
for k in kin:
c = TCanvas('c', 'c', 600, 600)
c.cd()
h = histos['dmjj' + k + r]
x = RooRealVar("x",
h.GetXaxis().GetTitle(),
h.GetXaxis().GetXmin(),
h.GetXaxis().GetXmax())
data = RooDataHist("data", "dataset with x", RooArgList(x), h)
frame = x.frame()
RooAbsData.plotOn(data, frame, RooFit.DataError(RooAbsData.SumW2))
mean1 = RooRealVar("mean1", "mean1", 0, -0.5, 0.5)
sigma1 = RooRealVar("sigma1", "sigma1", 0.1, 0.01, 1.0)
gauss1 = RooGaussian("g1", "g", x, mean1, sigma1)
mean2 = RooRealVar("mean2", "mean2", 0, -0.5, 0.5)
sigma2 = RooRealVar("sigma2", "sigma2", 0.1, 0.01, 1.0)
alphacb = RooRealVar("alphacb", "alphacb", 1, 0.1, 3)
ncb = RooRealVar("ncb", "ncb", 4, 1, 100)
gauss2 = RooCBShape("cb2", "cb", x, mean2, sigma2, alphacb, ncb)
frac = RooRealVar("frac", "fraction", 0.9, 0.0, 1.0)
model = RooAddPdf("sum", "g1+g2", RooArgList(gauss1, gauss2),
RooArgList(frac))
status = model.fitTo(data, RooFit.Save()).status()
if status != 0: continue
RooAbsPdf.plotOn(model, frame)
frame.Draw()
labels.append(TPaveText(0.6, 0.65, 0.85, 0.9, 'brNDC'))
ilab = len(labels) - 1
labels[ilab].SetName(r + k + 'dmfitrestxt')
labels[ilab].SetBorderSize(0)
labels[ilab].SetFillStyle(0)
labels[ilab].SetTextFont(42)
labels[ilab].SetTextAlign(12)
labels[ilab].AddText('Gaussian #1 (f=%3.3f)' % frac.getVal())
labels[ilab].AddText('#mu=%3.3f#pm%3.3f' %
(mean1.getVal(), mean1.getError()))
labels[ilab].AddText('#sigma=%3.3f#pm%3.3f' %
(sigma1.getVal(), sigma1.getError()))
labels[ilab].AddText('Gaussian #2 (f=%3.3f)' % (1 - frac.getVal()))
labels[ilab].AddText('#mu=%3.3f#pm%3.3f' %
(mean2.getVal(), mean2.getError()))
labels[ilab].AddText('#sigma=%3.3f#pm%3.3f' %
(sigma2.getVal(), sigma2.getError()))
labels[ilab].Draw()
drawHeader()
labels.append(TPaveText(0.15, 0.8, 0.4, 0.9, 'brNDC'))
ilab = len(labels) - 1
labels[ilab].SetName(k + r + 'dmjj')
labels[ilab].SetBorderSize(0)
labels[ilab].SetFillStyle(0)
labels[ilab].SetTextFont(42)
labels[ilab].SetTextAlign(12)
regionTitle = "inclusive"
if r == 'bb': regionTitle = 'barrel-barrel'
if r == 'eb': regionTitle = 'endcap-barrel'
if r == 'ee': regionTitle = 'endcap-endcap'
labels[ilab].AddText(regionTitle)
ptthreshold = 30
if k != '': ptthreshold = float(k)
labels[ilab].AddText('p_{T}>%3.0f GeV' % ptthreshold)
labels[ilab].Draw()
c.Modified()
c.Update()
c.SaveAs('dmjj' + k + r + '.png')
massResolution.SetTitle(regionTitle)
ip = massResolution.GetN()
x = 40
xerr = 10
if k == '50':
x = 75
xerr = 25
elif k == '100':
x = 150
xerr = 50
y = sigma1.getVal()
yerr = sigma1.getError()
if frac.getVal() < 0.8:
if sigma2.getVal() < sigma1.getVal():
y = sigma2.getVal()
ey = sigma2.getError()
massResolution.SetPoint(ip, x, y)
massResolution.SetPointError(ip, xerr, yerr)
frame = TGraphErrors()
frame.SetPoint(0, 0, 0)
frame.SetPoint(1, 200, 0.2)
frame.SetMarkerStyle(1)
frame.SetFillStyle(0)
frame.SetName('dmframe')
cdmevol = TCanvas('cdmevol', 'cdmevol', 500, 500)
cdmevol.cd()
frame.Draw('ap')
leg = TLegend(0.6, 0.92, 0.9, 0.98)
leg.SetFillStyle(0)
leg.SetBorderSize(0)
leg.SetTextFont(42)
for dmGr in massResolutionGrs:
dmGr.Draw('pl')
leg.AddEntry(dmGr, dmGr.GetTitle(), 'fp')
frame.GetXaxis().SetTitle("Leading quark transverse momentum [GeV]")
frame.GetYaxis().SetTitle("Core resolution")
frame.GetYaxis().SetTitleOffset(1.4)
frame.GetYaxis().SetNdivisions(10)
drawHeader()
leg.SetNColumns(2)
leg.Draw()
cdmevol.Modified()
cdmevol.Update()
cdmevol.SaveAs('dm_evol.png')
c = TCanvas('c', 'c', 600, 600)
c.cd()
histos['sel'].Draw('histtext')
drawHeader()
c.Modified()
c.Update()
c.SaveAs('selection.png')
return
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()
def initStyle():
gROOT.SetStyle("Plain")
# For the canvas:
gStyle.SetCanvasBorderMode(0)
gStyle.SetCanvasColor(kWhite)
gStyle.SetCanvasDefH(700) #Height of canvas
gStyle.SetCanvasDefW(700) #Width of canvas
gStyle.SetCanvasDefX(0) #Position on screen
gStyle.SetCanvasDefY(0)
# For the line:
gStyle.SetLineWidth(2)
# For the Pad:
gStyle.SetPadBorderMode(0)
# gStyle.SetPadBorderSize(Width_t size = 1)
gStyle.SetPadColor(kWhite)
gStyle.SetPadGridX(True)
gStyle.SetPadGridY(True)
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(2)
# gStyle.SetLegoInnerR(Float_t rad = 0.5)
# gStyle.SetNumberContours(Int_t number = 20)
gStyle.SetEndErrorSize(2)
#gStyle.SetErrorMarker(20)
gStyle.SetErrorX(0.)
gStyle.SetMarkerStyle(8)
gStyle.SetMarkerSize(1)
#For the fit/function:
gStyle.SetOptFit(0)
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.SetOptFile(0)
gStyle.SetOptStat(0) # To display the mean and RMS: SetOptStat("mr")
gStyle.SetStatColor(kWhite)
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.SetStatStyle(Style_t style = 1001)
# gStyle.SetStatX(Float_t x = 0)
# gStyle.SetStatY(Float_t y = 0)
# Margins:
gStyle.SetPadTopMargin(0.11)
gStyle.SetPadBottomMargin(0.13)
gStyle.SetPadLeftMargin(0.17)
gStyle.SetPadRightMargin(0.07)
# For the Global title:
gStyle.SetOptTitle(0)
gStyle.SetTitleFont(42)
gStyle.SetTitleColor(1)
gStyle.SetTitleTextColor(1)
gStyle.SetTitleFillColor(10)
gStyle.SetTitleFontSize(0.04)
# gStyle.SetTitleH(0) # Set the height of the title box
# gStyle.SetTitleW(0) # Set the width of the title box
#gStyle.SetTitleX(0.35) # Set the position of the title box
#gStyle.SetTitleY(0.986) # 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.05, "XYZ")
# gStyle.SetTitleXSize(Float_t size = 0.02) # Another way to set the size?
# gStyle.SetTitleYSize(Float_t size = 0.02)
gStyle.SetTitleXOffset(1.)
gStyle.SetTitleYOffset(1.3)
#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.035, "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"
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 initialize(fitresults=True, grid=False):
gROOT.SetStyle("Plain")
gStyle.SetOptFit()
gStyle.SetOptStat(0)
# For the canvas:
gStyle.SetCanvasBorderMode(0)
gStyle.SetCanvasColor(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.SetPadColor(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(kWhite)
gStyle.SetFrameFillStyle(1000)
gStyle.SetFrameLineColor(1)
gStyle.SetFrameLineStyle(1)
gStyle.SetFrameLineWidth(1)
# For the histo:
gStyle.SetHistLineColor(1)
gStyle.SetHistLineStyle(0)
gStyle.SetHistLineWidth(2)
gStyle.SetEndErrorSize(2)
gStyle.SetErrorX(0.)
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)
# For the statistics box:
gStyle.SetOptFile(0)
gStyle.SetOptStat(0) # To display the mean and RMS: SetOptStat("mr")
gStyle.SetStatColor(kWhite)
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)
# Margins:
gStyle.SetPadTopMargin(0.05)
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(0)
gStyle.SetTitleFont(42)
gStyle.SetTitleColor(1)
gStyle.SetTitleTextColor(1)
gStyle.SetTitleFillColor(10)
gStyle.SetTitleFontSize(0.05)
# For the axis titles:
gStyle.SetTitleColor(1, "XYZ")
gStyle.SetTitleFont(42, "XYZ")
gStyle.SetTitleSize(0.06, "XYZ")
gStyle.SetTitleXOffset(0.9)
gStyle.SetTitleYOffset(1.25)
# For the axis labels:
gStyle.SetLabelColor(1, "XYZ")
gStyle.SetLabelFont(42, "XYZ")
gStyle.SetLabelOffset(0.007, "XYZ")
gStyle.SetLabelSize(0.05, "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.)
# For graphs
gStyle.SetErrorX(0) # suppress error along x
if grid:
gStyle.SetPadGridX(gridOn)
gStyle.SetPadGridY(gridOn)
gStyle.SetGridColor(kGray)
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) #gStyle.SetLabelSize(30, 'xyz') #gStyle.SetTitleSize(30, 'xyz') #gStyle.SetTitleSize(30, 't') #gStyle.SetStatFontSize(15)
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
