def looks_atlas2():
atlasStyle = TStyle("ATLAS", "Atlas style")
icol = 0
atlasStyle.SetFrameBorderMode(icol)
atlasStyle.SetFrameFillColor(icol)
atlasStyle.SetCanvasBorderMode(icol)
atlasStyle.SetCanvasColor(icol)
atlasStyle.SetPadBorderMode(icol)
atlasStyle.SetPadColor(icol)
atlasStyle.SetStatColor(icol)
atlasStyle.SetPaperSize(20, 26)
atlasStyle.SetPadTopMargin(0.05)
atlasStyle.SetPadRightMargin(0.05)
atlasStyle.SetPadBottomMargin(0.16)
atlasStyle.SetPadLeftMargin(0.16)
atlasStyle.SetTitleXOffset(1.4)
atlasStyle.SetTitleYOffset(1.4)
font = 42
tsize = 0.05
atlasStyle.SetTextFont(font)
atlasStyle.SetTextSize(tsize)
atlasStyle.SetLabelFont(font, 'x')
atlasStyle.SetTitleFont(font, 'x')
atlasStyle.SetLabelFont(font, 'y')
atlasStyle.SetTitleFont(font, 'y')
atlasStyle.SetLabelFont(font, 'z')
atlasStyle.SetTitleFont(font, 'z')
atlasStyle.SetLabelSize(tsize, 'x')
atlasStyle.SetTitleSize(tsize, 'x')
atlasStyle.SetLabelSize(tsize, 'y')
atlasStyle.SetTitleSize(tsize, 'y')
atlasStyle.SetLabelSize(tsize, 'z')
atlasStyle.SetTitleSize(tsize, 'z')
atlasStyle.SetMarkerStyle(20)
atlasStyle.SetMarkerSize(1.2)
atlasStyle.SetHistLineWidth(2)
atlasStyle.SetLineStyleString(2, '[12 12]')
atlasStyle.SetEndErrorSize(0.)
atlasStyle.SetOptTitle(0)
atlasStyle.SetOptStat(0)
atlasStyle.SetOptFit(0)
atlasStyle.SetPadTickX(1)
atlasStyle.SetPadTickY(1)
atlasStyle.cd()
gROOT.ForceStyle()
atlasStyle.ls()
def chips_style():
print("\nLet's make it look cheap... \n")
chips_style = TStyle("chips_style", "CHIPS Style")
# Center title
chips_style.SetTitleAlign(22)
chips_style.SetTitleX(.5)
chips_style.SetTitleY(.95)
chips_style.SetTitleBorderSize(0)
# Remove stats box
chips_style.SetOptStat(0)
# Set background color to white
chips_style.SetFillColor(10)
chips_style.SetFrameFillColor(10)
chips_style.SetCanvasColor(10)
chips_style.SetPadColor(10)
chips_style.SetTitleFillColor(0)
chips_style.SetStatColor(10)
# No colored frames around plots
chips_style.SetFrameBorderMode(0)
chips_style.SetCanvasBorderMode(0)
chips_style.SetPadBorderMode(0)
# Set the default line color for a fit function to be red
chips_style.SetFuncColor(2)
# Marker settings
chips_style.SetMarkerStyle(20)
# No border on legends
chips_style.SetLegendBorderSize(0)
# Disabled for violating NOvA style guidelines
# Scientific notation on axes
TGaxis.SetMaxDigits(3)
# Axis titles
chips_style.SetTitleSize(.055, "xyz")
chips_style.SetTitleOffset(.8, "xyz")
chips_style.SetTitleOffset(.9, "y")
chips_style.SetTitleSize(.055, "")
chips_style.SetTitleOffset(.8, "")
# Axis labels (numbering)
chips_style.SetLabelSize(.04, "xyz")
chips_style.SetLabelOffset(.005, "xyz")
# Set histogram minimum to exactly zero
chips_style.SetHistMinimumZero()
def setAtlasStyle():
from ROOT import TStyle
atlasStyle = TStyle("ATLAS","Atlas style")
atlasStyle.SetFrameBorderMode(0)
atlasStyle.SetFrameFillColor(0)
atlasStyle.SetCanvasBorderMode(0)
atlasStyle.SetCanvasColor(0)
atlasStyle.SetPadBorderMode(0)
atlasStyle.SetPadColor(0)
atlasStyle.SetStatColor(0)
atlasStyle.SetPaperSize(20,26)
atlasStyle.SetPadTopMargin(0.07)
atlasStyle.SetPadRightMargin(0.05)
atlasStyle.SetPadBottomMargin(0.16)
atlasStyle.SetPadLeftMargin(0.16)
atlasStyle.SetTitleXOffset(1.4)
atlasStyle.SetTitleYOffset(1.2)
font=43
tsize=0.05
atlasStyle.SetTextFont(font)
atlasStyle.SetTextSize(tsize)
atlasStyle.SetLabelFont(font,"x")
atlasStyle.SetTitleFont(font,"x")
atlasStyle.SetLabelFont(font,"y")
atlasStyle.SetTitleFont(font,"y")
atlasStyle.SetLabelFont(font,"z")
atlasStyle.SetTitleFont(font,"z")
atlasStyle.SetLabelSize(tsize,"x")
atlasStyle.SetTitleSize(tsize,"x")
atlasStyle.SetLabelSize(tsize,"y")
atlasStyle.SetTitleSize(tsize,"y")
atlasStyle.SetLabelSize(tsize,"z")
atlasStyle.SetTitleSize(tsize,"z")
atlasStyle.SetMarkerStyle(20)
atlasStyle.SetMarkerSize(1.2)
atlasStyle.SetHistLineWidth(2)
atlasStyle.SetLineStyleString(2,"[12 12]")
atlasStyle.SetEndErrorSize(0.)
atlasStyle.SetOptTitle(0)
atlasStyle.SetOptStat(0)
atlasStyle.SetOptFit(0)
atlasStyle.SetPadTickX(1)
atlasStyle.SetPadTickY(1)
gROOT.SetStyle("ATLAS")
gROOT.ForceStyle()
def myStyle():
myStyle = TStyle("MyStyle", "My Root Styles")
#Canvas
myStyle.SetCanvasBorderMode(0) # Transparent
myStyle.SetCanvasColor(0) # Transparent
#Paper, Pad, Palette, Frame
myStyle.SetPadBorderMode(0) # Transparent
myStyle.SetPadColor(0) # Transparent
myStyle.SetPalette(1) # Default
myStyle.SetFrameBorderMode(1) # Border
# Axis
myStyle.SetLabelSize(0.04, "xyz") # size of axis values
myStyle.SetTitleSize(0.04, "xyz")
myStyle.SetPadTickX(1)
myStyle.SetPadTickY(1)
# Title
myStyle.SetTitleColor(1) # Black
myStyle.SetTitleStyle(0) # Transparent
myStyle.SetTitleBorderSize(0) # Transparent
myStyle.SetTitleY(0.97) # Set y-position (fraction of pad size)
myStyle.SetTitleX(0.4) # Set x-position (fraction of pad size)
# #Stat box dimensions, position and style
myStyle.SetStatY(0.89) # Set y-position (fraction of pad size)
myStyle.SetStatX(0.89) # Set x-position (fraction of pad size)
myStyle.SetStatW(0.36) # Set width of stat-box (fraction of pad size)
myStyle.SetStatH(0.12) # Set height of stat-box (fraction of pad size)
myStyle.SetStatStyle(0) # Transparent
myStyle.SetStatColor(0) # Transparent
myStyle.SetStatBorderSize(1) # Transparent
# Histo Filling (visual)
myStyle.SetHistFillColor(3)
myStyle.SetHistLineColor(3)
myStyle.SetHistFillStyle(1001)
# Stats display options
#myStyle.SetOptStat("ourRmMe") #over/under -flows, Rms and Means with errors, number of entries
myStyle.SetOptStat("neouRM") #over/under -flows, Rms and Means with errors, number of entries
myStyle.SetOptFit(1111) #probability, Chi2, errors, name/values of parameters
myStyle.SetStatFormat("11.4f") # 4 sig.fig, f=float
return myStyle
def setTDRStyle():
import ROOT
from ROOT import TStyle
from ROOT import kWhite
from ROOT import kTRUE
tdrStyle = TStyle("tdrStyle","Style for P-TDR")
# For the canvas:
tdrStyle.SetCanvasBorderMode(0)
tdrStyle.SetCanvasColor(kWhite)
# For the canvas:
tdrStyle.SetCanvasBorderMode(0)
tdrStyle.SetCanvasColor(kWhite)
tdrStyle.SetCanvasDefH(600) #Height of canvas
tdrStyle.SetCanvasDefW(600)#Width of canvas
tdrStyle.SetCanvasDefX(0) #POsition on screen
tdrStyle.SetCanvasDefY(0)
# For the Pad:
tdrStyle.SetPadBorderMode(0)
# tdrStyle->SetPadBorderSize(Width_t size = 1);
tdrStyle.SetPadColor(kWhite)
tdrStyle.SetPadGridX(0)
tdrStyle.SetPadGridY(0)
tdrStyle.SetGridColor(0)
tdrStyle.SetGridStyle(3)
tdrStyle.SetGridWidth(1)
# For the frame:
tdrStyle.SetFrameBorderMode(0)
tdrStyle.SetFrameBorderSize(1)
tdrStyle.SetFrameFillColor(0)
tdrStyle.SetFrameFillStyle(0)
tdrStyle.SetFrameLineColor(1)
tdrStyle.SetFrameLineStyle(1)
tdrStyle.SetFrameLineWidth(1)
# For the histo:
# tdrStyle->SetHistFillColor(1);
# tdrStyle->SetHistFillStyle(0);
tdrStyle.SetHistLineColor(1)
tdrStyle.SetHistLineStyle(0)
tdrStyle.SetHistLineWidth(1)
# tdrStyle->SetLegoInnerR(Float_t rad = 0.5);
# define the palette for z axis
palette = []
NRGBs = 5
NCont = 255
stops = array("d",[0.00,0.34,0.61,0.84,1.00])
red = array("d",[0.50,0.50,1.00,1.00,1.00])
green = array("d",[0.50,1.00,1.00,0.60,0.50])
blue = array("d",[1.00,1.00,0.50,0.40,0.50])
FI = ROOT.TColor.CreateGradientColorTable(NRGBs,stops,red,green,blue,NCont)
for i in range(0,NCont):
palette.append(FI+i)
tdrStyle.SetPalette(NCont,array("i",palette));
tdrStyle.SetNumberContours(NCont);
tdrStyle.SetEndErrorSize(2)
# tdrStyle->SetErrorMarker(20);
tdrStyle.SetErrorX(0.)
tdrStyle.SetMarkerStyle(20)
#For the fit/function:
tdrStyle.SetOptFit(0)
tdrStyle.SetFitFormat("5.4g")
tdrStyle.SetFuncColor(2)
tdrStyle.SetFuncStyle(1)
tdrStyle.SetFuncWidth(1)
#For the date:
tdrStyle.SetOptDate(0)
# tdrStyle->SetDateX(Float_t x = 0.01);
# tdrStyle->SetDateY(Float_t y = 0.01);
# For the statistics box:
tdrStyle.SetOptFile(0)
tdrStyle.SetOptStat(0) # To display the mean and RMS: SetOptStat("mr");
tdrStyle.SetStatColor(kWhite)
tdrStyle.SetStatFont(42)
tdrStyle.SetStatFontSize(0.025)
tdrStyle.SetStatTextColor(1)
tdrStyle.SetStatFormat("6.4g")
tdrStyle.SetStatBorderSize(1)
tdrStyle.SetStatH(0.1)
tdrStyle.SetStatW(0.15)
# tdrStyle->SetStatStyle(Style_t style = 100.1);
# tdrStyle->SetStatX(Float_t x = 0);
# tdrStyle->SetStatY(Float_t y = 0);
# Margins:
tdrStyle.SetPadTopMargin(0.05)
tdrStyle.SetPadBottomMargin(0.14)
tdrStyle.SetPadLeftMargin(0.16)
tdrStyle.SetPadRightMargin(0.05)
# For the Global title:
tdrStyle.SetOptTitle(0)
tdrStyle.SetTitleFont(42)
tdrStyle.SetTitleColor(1)
tdrStyle.SetTitleTextColor(1)
tdrStyle.SetTitleFillColor(10)
tdrStyle.SetTitleFontSize(0.05)
# tdrStyle->SetTitleH(0); # Set the height of the title box
# tdrStyle->SetTitleW(0); # Set the width of the title box
# tdrStyle->SetTitleX(0); # Set the position of the title box
# tdrStyle->SetTitleY(0.985); # Set the position of the title box
# tdrStyle->SetTitleStyle(Style_t style = 100.1);
# tdrStyle->SetTitleBorderSize(2);
# For the axis titles:
tdrStyle.SetTitleColor(1, "XYZ")
tdrStyle.SetTitleFont(42, "XYZ")
tdrStyle.SetTitleSize(0.06, "XYZ")
# tdrStyle->SetTitleXSize(Float_t size = 0.02); # Another way to set the size?
# tdrStyle->SetTitleYSize(Float_t size = 0.02);
tdrStyle.SetTitleXOffset(0.95)
tdrStyle.SetTitleYOffset(1.3)
# tdrStyle->SetTitleOffset(1.1, "Y"); # Another way to set the Offset
# For the axis labels:
tdrStyle.SetLabelColor(1, "XYZ")
tdrStyle.SetLabelFont(42, "XYZ")
tdrStyle.SetLabelOffset(0.007, "XYZ")
tdrStyle.SetLabelSize(0.05, "XYZ")
# For the axis:
tdrStyle.SetAxisColor(1, "XYZ")
tdrStyle.SetStripDecimals(kTRUE)
tdrStyle.SetTickLength(0.03, "XYZ")
tdrStyle.SetNdivisions(408, "XYZ")
#~ tdrStyle->SetNdivisions(510, "XYZ");
tdrStyle.SetPadTickX(1) # To get tick marks on the opposite side of the frame
tdrStyle.SetPadTickY(1)
# Change for log plots:
tdrStyle.SetOptLogx(0)
tdrStyle.SetOptLogy(0)
tdrStyle.SetOptLogz(0)
# Postscript options:
tdrStyle.SetPaperSize(20.,20.);
# tdrStyle->SetLineScalePS(Float_t scale = 3);
# tdrStyle->SetLineStyleString(Int_t i, const char* text);
# tdrStyle->SetHeaderPS(const char* header);
# tdrStyle->SetTitlePS(const char* pstitle);
#tdrStyle->SetBarOffset(Float_t baroff = 0.5);
#tdrStyle->SetBarWidth(Float_t barwidth = 0.5);
#tdrStyle->SetPaintTextFormat(const char* format = "g");
#~ tdrStyle.SetPalette(1)
#tdrStyle->SetTimeOffset(Double_t toffset);
#tdrStyle->SetHistMinimumZero(kTRUE);
ROOT.gROOT.ForceStyle()
tdrStyle.cd()
return tdrStyle
def setLHCbStyle():
# configuring the plot-style
gROOT.SetStyle("Plain")
lhcbStyle = TStyle("lhcbstyle", "LHCb plots style")
GlobalLHCbTSize = 0.06
lhcbTSize = 0.06
kLHCbFont = 132
lhcbWidth = 2
lhcbStyle.SetFillColor(1)
lhcbStyle.SetFillStyle(1001)
lhcbStyle.SetFrameFillColor(0)
lhcbStyle.SetFrameBorderMode(0)
lhcbStyle.SetPadBorderMode(0)
lhcbStyle.SetPadColor(0)
lhcbStyle.SetCanvasBorderMode(0)
lhcbStyle.SetCanvasColor(0)
lhcbStyle.SetStatColor(0)
lhcbStyle.SetLegendBorderSize(0)
lhcbStyle.SetPalette(1)
# colors = [0,5,7,3,6,2,4,1]
# lhcbStyle.SetPalette(8,colors)
lhcbStyle.SetPaperSize(20, 26)
lhcbStyle.SetPadTopMargin(0.05)
lhcbStyle.SetPadRightMargin(0.05)
lhcbStyle.SetPadBottomMargin(0.16)
lhcbStyle.SetPadLeftMargin(0.14)
lhcbStyle.SetTextFont(kLHCbFont)
lhcbStyle.SetTextSize(lhcbTSize)
lhcbStyle.SetLabelFont(kLHCbFont, "x")
lhcbStyle.SetLabelFont(kLHCbFont, "y")
lhcbStyle.SetLabelFont(kLHCbFont, "z")
lhcbStyle.SetLabelSize(lhcbTSize, "x")
lhcbStyle.SetLabelSize(lhcbTSize, "y")
lhcbStyle.SetLabelSize(lhcbTSize, "z")
lhcbStyle.SetTitleFont(kLHCbFont)
lhcbStyle.SetTitleFont(kLHCbFont, "x")
lhcbStyle.SetTitleFont(kLHCbFont, "y")
lhcbStyle.SetTitleFont(kLHCbFont, "z")
lhcbStyle.SetTitleSize(1.2 * lhcbTSize, "x")
lhcbStyle.SetTitleSize(1.2 * lhcbTSize, "y")
lhcbStyle.SetTitleSize(1.2 * lhcbTSize, "z")
lhcbStyle.SetLineWidth(lhcbWidth)
lhcbStyle.SetHistLineWidth(lhcbWidth)
lhcbStyle.SetFuncWidth(lhcbWidth)
lhcbStyle.SetGridWidth(lhcbWidth)
lhcbStyle.SetLineStyleString(2, "[12 12]")
lhcbStyle.SetMarkerStyle(20)
lhcbStyle.SetMarkerSize(1.0)
lhcbStyle.SetLabelOffset(0.010, "X")
lhcbStyle.SetLabelOffset(0.010, "Y")
lhcbStyle.SetOptStat(0)
lhcbStyle.SetStatFormat("6.3g")
lhcbStyle.SetOptTitle(0)
lhcbStyle.SetOptFit(0)
lhcbStyle.SetTitleOffset(0.95, "X")
lhcbStyle.SetTitleOffset(0.95, "Y")
lhcbStyle.SetTitleOffset(1.2, "Z")
lhcbStyle.SetTitleFillColor(0)
lhcbStyle.SetTitleStyle(0)
lhcbStyle.SetTitleBorderSize(0)
lhcbStyle.SetTitleFont(kLHCbFont, "title")
lhcbStyle.SetTitleX(0.0)
lhcbStyle.SetTitleY(1.0)
lhcbStyle.SetTitleW(1.0)
lhcbStyle.SetTitleH(0.05)
lhcbStyle.SetStatBorderSize(0)
lhcbStyle.SetStatFont(kLHCbFont)
lhcbStyle.SetStatFontSize(0.05)
lhcbStyle.SetStatX(0.9)
lhcbStyle.SetStatY(0.9)
lhcbStyle.SetStatW(0.25)
lhcbStyle.SetStatH(0.15)
lhcbStyle.SetPadTickX(1)
lhcbStyle.SetPadTickY(1)
lhcbStyle.SetNdivisions(505, "x")
lhcbStyle.SetNdivisions(510, "y")
gROOT.SetStyle("lhcbstyle")
gROOT.ForceStyle()
def setTDRStyle(drawGrid):
tdrStyle = TStyle("tdrStyle", "Style for P-TDR")
# For the canvas:
tdrStyle.SetCanvasBorderMode(0)
tdrStyle.SetCanvasColor(0)
tdrStyle.SetCanvasDefH(600) #Height of canvas
tdrStyle.SetCanvasDefW(600) #Width of canvas
tdrStyle.SetCanvasDefX(0) #POsition on screen
tdrStyle.SetCanvasDefY(0)
# For the Pad:
tdrStyle.SetPadBorderMode(0)
tdrStyle.SetPadColor(0)
tdrStyle.SetPadGridX(drawGrid)
tdrStyle.SetPadGridY(drawGrid)
tdrStyle.SetGridColor(0)
tdrStyle.SetGridStyle(3) #3
tdrStyle.SetGridWidth(1)
# For the frame:
tdrStyle.SetFrameBorderMode(0)
tdrStyle.SetFrameBorderSize(1)
tdrStyle.SetFrameFillColor(0)
tdrStyle.SetFrameFillStyle(0)
tdrStyle.SetFrameLineColor(1)
tdrStyle.SetFrameLineStyle(1)
tdrStyle.SetFrameLineWidth(1)
# For the histo:
# tdrStyle.SetHistFillColor(1)
# tdrStyle.SetHistFillStyle(0)
tdrStyle.SetHistLineColor(1)
tdrStyle.SetHistLineStyle(0)
tdrStyle.SetHistLineWidth(1)
tdrStyle.SetEndErrorSize(2)
#tdrStyle.SetErrorMarker(20)
tdrStyle.SetErrorX(0.)
tdrStyle.SetMarkerStyle(20)
#For the fit/function:
tdrStyle.SetOptFit(0)
tdrStyle.SetFitFormat("5.4g")
tdrStyle.SetFuncColor(2)
tdrStyle.SetFuncStyle(1)
tdrStyle.SetFuncWidth(1)
#For the date:
tdrStyle.SetOptDate(0)
# For the statistics box:
tdrStyle.SetOptFile(0)
tdrStyle.SetOptStat(0) # To display the mean and RMS: SetOptStat("mr")
tdrStyle.SetStatColor(0)
tdrStyle.SetStatFont(42)
tdrStyle.SetStatFontSize(0.025)
tdrStyle.SetStatTextColor(1)
tdrStyle.SetStatFormat("6.4g")
tdrStyle.SetStatBorderSize(1)
tdrStyle.SetStatH(0.1)
tdrStyle.SetStatW(0.15)
tdrStyle.SetPadTopMargin(0.05)
tdrStyle.SetPadBottomMargin(0.13)
tdrStyle.SetPadLeftMargin(0.13)
tdrStyle.SetPadRightMargin(0.05)
tdrStyle.SetTitleFont(42)
tdrStyle.SetTitleColor(1)
tdrStyle.SetTitleTextColor(1)
tdrStyle.SetTitleFillColor(10)
tdrStyle.SetTitleFontSize(0.05)
tdrStyle.SetTitleColor(1, "XYZ")
tdrStyle.SetTitleFont(42, "XYZ")
tdrStyle.SetTitleSize(0.06, "XYZ")
tdrStyle.SetTitleXOffset(0.9)
tdrStyle.SetTitleYOffset(1.05)
tdrStyle.SetLabelColor(1, "XYZ")
tdrStyle.SetLabelFont(42, "XYZ")
tdrStyle.SetLabelOffset(0.007, "XYZ")
tdrStyle.SetLabelSize(0.05, "XYZ")
tdrStyle.SetAxisColor(1, "XYZ")
tdrStyle.SetStripDecimals(True)
tdrStyle.SetTickLength(0.03, "XYZ")
tdrStyle.SetNdivisions(510, "XYZ") #510
tdrStyle.SetPadTickX(
1) # To get tick marks on the opposite side of the frame
tdrStyle.SetPadTickY(1)
# Change for log plots:
tdrStyle.SetOptLogx(0)
tdrStyle.SetOptLogy(0)
tdrStyle.SetOptLogz(0)
# Postscript options:
tdrStyle.SetPaperSize(15., 15.)
# tdrStyle.SetLineScalePS(Float_t scale = 3)
# tdrStyle.SetLineStyleString(Int_t i, const char* text)
# tdrStyle.SetHeaderPS(const char* header)
# tdrStyle.SetTitlePS(const char* pstitle)
# tdrStyle.SetBarOffset(Float_t baroff = 0.5)
# tdrStyle.SetBarWidth(Float_t barwidth = 0.5)
# tdrStyle.SetPaintTextFormat(const char* format = "g")
# tdrStyle.SetPalette(Int_t ncolors = 0, Int_t* colors = 0)
# tdrStyle.SetTimeOffset(Double_t toffset)
# tdrStyle.SetHistMinimumZero(kTRUE)
tdrStyle.cd()
def AtlasStyle():
atlasStyle = TStyle("ATLAS", "Atlas style")
# use plain black on white colors
icol = 0 # WHITE
atlasStyle.SetFrameBorderMode(icol)
atlasStyle.SetFrameFillColor(icol)
atlasStyle.SetCanvasBorderMode(icol)
atlasStyle.SetCanvasColor(icol)
atlasStyle.SetPadBorderMode(icol)
atlasStyle.SetPadColor(icol)
atlasStyle.SetStatColor(icol)
#atlasStyle.SetFillColor(icol) # don't use: white fill color for *all* objects
# set the paper & margin sizes
atlasStyle.SetPaperSize(20, 26)
# set margin sizes
atlasStyle.SetPadTopMargin(0.05)
atlasStyle.SetPadRightMargin(0.05)
atlasStyle.SetPadBottomMargin(0.16)
atlasStyle.SetPadLeftMargin(0.16)
# set title offsets (for axis label)
atlasStyle.SetTitleXOffset(1.4)
atlasStyle.SetTitleYOffset(1.4)
# use large fonts
#Int_t font=72 # Helvetica italics
font = 42 # Helvetica
tsize = 0.05
atlasStyle.SetTextFont(font)
atlasStyle.SetTextSize(tsize)
atlasStyle.SetLabelFont(font, "x")
atlasStyle.SetTitleFont(font, "x")
atlasStyle.SetLabelFont(font, "y")
atlasStyle.SetTitleFont(font, "y")
atlasStyle.SetLabelFont(font, "z")
atlasStyle.SetTitleFont(font, "z")
atlasStyle.SetLabelSize(tsize, "x")
atlasStyle.SetTitleSize(tsize, "x")
atlasStyle.SetLabelSize(tsize, "y")
atlasStyle.SetTitleSize(tsize, "y")
atlasStyle.SetLabelSize(tsize, "z")
atlasStyle.SetTitleSize(tsize, "z")
# use bold lines and markers
atlasStyle.SetMarkerStyle(20)
atlasStyle.SetMarkerSize(1.2)
atlasStyle.SetHistLineWidth(2)
atlasStyle.SetLineStyleString(2, "[12 12]") # postscript dashes
# get rid of X error bars
#atlasStyle.SetErrorX(0.001)
# get rid of error bar caps
atlasStyle.SetEndErrorSize(0.)
# do not display any of the standard histogram decorations
atlasStyle.SetOptTitle(0)
#atlasStyle.SetOptStat(1111)
atlasStyle.SetOptStat(0)
#atlasStyle.SetOptFit(1111)
atlasStyle.SetOptFit(0)
# put tick marks on top and RHS of plots
atlasStyle.SetPadTickX(1)
atlasStyle.SetPadTickY(1)
atlasStyle.SetPalette(ROOT.kBird)
return atlasStyle
def setTDRStyle():
"""Function to setup a TDR-like style"""
tdrStyle = TStyle("tdrStyle", "Style for P-TDR")
# For the canvas:
tdrStyle.SetCanvasBorderMode(0)
tdrStyle.SetCanvasColor(kWhite)
tdrStyle.SetCanvasDefH(600) #Height of canvas
tdrStyle.SetCanvasDefW(600) #Width of canvas
tdrStyle.SetCanvasDefX(0) #Position on screen
tdrStyle.SetCanvasDefY(0)
# For the Pad:
tdrStyle.SetPadBorderMode(0)
tdrStyle.SetPadColor(kWhite)
tdrStyle.SetPadGridX(False)
tdrStyle.SetPadGridY(False)
tdrStyle.SetGridColor(kWhite)
tdrStyle.SetGridStyle(3)
tdrStyle.SetGridWidth(1)
tdrStyle.SetPadTickX(True)
tdrStyle.SetPadTickY(True)
# For the frame:
tdrStyle.SetFrameBorderMode(0)
tdrStyle.SetFrameBorderSize(1)
tdrStyle.SetFrameFillColor(0)
tdrStyle.SetFrameFillStyle(0)
tdrStyle.SetFrameLineColor(1)
tdrStyle.SetFrameLineStyle(1)
tdrStyle.SetFrameLineWidth(0)
# For the histo:
tdrStyle.SetHistLineColor(1)
tdrStyle.SetHistLineStyle(0)
tdrStyle.SetHistLineWidth(1)
tdrStyle.SetEndErrorSize(1)
#tdrStyle.SetErrorX(0.)
tdrStyle.SetMarkerStyle(20)
#For the fit/function:
tdrStyle.SetOptFit(0)
tdrStyle.SetFitFormat("5.4g")
tdrStyle.SetFuncColor(2)
tdrStyle.SetFuncStyle(1)
tdrStyle.SetFuncWidth(1)
#For the date:
tdrStyle.SetOptDate(0)
# For the statistics box:
tdrStyle.SetOptFile(0)
tdrStyle.SetOptStat(0)
# To display the mean and RMS: SetOptStat("mr")
tdrStyle.SetStatColor(kWhite)
tdrStyle.SetStatFont(42)
tdrStyle.SetStatFontSize(0.025)
tdrStyle.SetStatTextColor(1)
tdrStyle.SetStatFormat("6.4g")
tdrStyle.SetStatBorderSize(1)
tdrStyle.SetStatH(0.1)
tdrStyle.SetStatW(0.15)
# Margins:
tdrStyle.SetPadTopMargin(0.05)
tdrStyle.SetPadBottomMargin(0.1)
tdrStyle.SetPadLeftMargin(0.12)
tdrStyle.SetPadRightMargin(0.05)
# For the Global title:
tdrStyle.SetOptTitle(1)
tdrStyle.SetTitleFont(42)
tdrStyle.SetTitleColor(1)
tdrStyle.SetTitleTextColor(1)
tdrStyle.SetTitleFillColor(10)
tdrStyle.SetTitleFontSize(0.0525)
tdrStyle.SetTitleH(0)
# Set the height of the title box
tdrStyle.SetTitleW(0)
# Set the width of the title box
tdrStyle.SetTitleX(0.5)
# Set the position of the title box
tdrStyle.SetTitleY(1.0)
# Set the position of the title box
tdrStyle.SetTitleStyle(1001)
tdrStyle.SetTitleBorderSize(0)
tdrStyle.SetTitleAlign(23)
# For the axis titles:
tdrStyle.SetTitleColor(1, "XYZ")
tdrStyle.SetTitleFont(42, "XYZ")
tdrStyle.SetTitleSize(0.05, "XY")
tdrStyle.SetTitleSize(0.035, "Z")
tdrStyle.SetTitleXOffset(1.0)
tdrStyle.SetTitleYOffset(1.0)
# For the axis labels:
tdrStyle.SetLabelColor(1, "XYZ")
tdrStyle.SetLabelFont(42, "XYZ")
tdrStyle.SetLabelOffset(5e-3, "XYZ")
tdrStyle.SetLabelSize(0.03, "XYZ")
# For the axis:
tdrStyle.SetAxisColor(1, "XYZ")
tdrStyle.SetStripDecimals(kTRUE)
tdrStyle.SetTickLength(0.03, "XYZ")
tdrStyle.SetNdivisions(510, "XYZ")
tdrStyle.SetPadTickX(
1) # To get tick marks on the opposite side of the frame
tdrStyle.SetPadTickY(1)
# Change for log plots:
tdrStyle.SetOptLogx(0)
tdrStyle.SetOptLogy(0)
tdrStyle.SetOptLogz(0)
# Miscellaneous
tdrStyle.SetLegendBorderSize(0)
# Postscript options:
tdrStyle.SetPaperSize(20., 20.)
tdrStyle.cd()
def looks_minos():
minosStyle = TStyle("minosStyle", "MINOS Style")
# Colors
#set the background color to white
minosStyle.SetFillColor(10)
minosStyle.SetFrameFillColor(10)
minosStyle.SetCanvasColor(10)
minosStyle.SetPadColor(10)
minosStyle.SetTitleFillColor(0)
minosStyle.SetStatColor(10)
#dont put a colored frame around the plots
minosStyle.SetFrameBorderMode(0)
minosStyle.SetCanvasBorderMode(0)
minosStyle.SetPadBorderMode(0)
#use the primary color palette
#minosStyle.SetPalette(1,0);
#set the default line color for a histogram to be black
minosStyle.SetHistLineColor(ROOT.kBlack)
#set the default line color for a fit function to be red
minosStyle.SetFuncColor(ROOT.kRed)
#make the axis labels black
minosStyle.SetLabelColor(ROOT.kBlack, "xyz")
#set the default title color to be black
minosStyle.SetTitleColor(ROOT.kBlack)
# Sizes
#set the margins
minosStyle.SetPadBottomMargin(0.2)
minosStyle.SetPadTopMargin(0.075)
minosStyle.SetPadLeftMargin(0.15)
#set axis label and title text sizes
minosStyle.SetLabelSize(0.05, "xyz")
minosStyle.SetTitleSize(0.06, "xyz")
minosStyle.SetTitleOffset(0.9, "x")
minosStyle.SetTitleOffset(0.8, "yz")
minosStyle.SetStatFontSize(0.05)
minosStyle.SetTextSize(0.06)
minosStyle.SetTitleBorderSize(0)
minosStyle.SetStatBorderSize(0)
#set line widths
minosStyle.SetHistLineWidth(2)
minosStyle.SetFrameLineWidth(2)
minosStyle.SetFuncWidth(2)
# Misc
#align the titles to be centered
minosStyle.SetTitleAlign(22)
#set the number of divisions to show
minosStyle.SetNdivisions(506, "xy")
#turn off xy grids
minosStyle.SetPadGridX(0)
minosStyle.SetPadGridY(0)
#set the tick mark style
#minosStyle.SetPadTickX(1);
#minosStyle.SetPadTickY(1);
#show the fit parameters in a box
minosStyle.SetOptFit(1111)
#turn off all other stats
minosStyle.SetOptStat(0000000)
#marker settings
minosStyle.SetMarkerStyle(8)
minosStyle.SetMarkerSize(0.9)
# Fonts
kMinosFont = int(42)
minosStyle.SetStatFont(kMinosFont)
minosStyle.SetLabelFont(kMinosFont, "xyz")
minosStyle.SetTitleFont(kMinosFont, "xyz")
minosStyle.SetTextFont(kMinosFont)
#done
minosStyle.cd()
gROOT.ForceStyle()
gStyle.ls()
def setStyle():
style = TStyle( 'style', 'style' )
style.SetCanvasBorderMode(0)
style.SetCanvasColor(ROOT.kWhite)
style.SetCanvasDefH(600)
style.SetCanvasDefW(600)
style.SetCanvasDefX(0)
style.SetCanvasDefY(0)
style.SetPadBorderMode(0)
style.SetPadColor(ROOT.kWhite)
style.SetPadGridX(False)
style.SetPadGridY(False)
style.SetGridColor(0)
style.SetGridStyle(3)
style.SetGridWidth(1)
style.SetFrameBorderMode(0)
style.SetFrameBorderSize(1)
style.SetFrameFillColor(0)
style.SetFrameFillStyle(0)
style.SetFrameLineColor(1)
style.SetFrameLineStyle(1)
style.SetFrameLineWidth(1)
style.SetHistLineColor(1)
style.SetHistLineStyle(0)
style.SetHistLineWidth(1)
style.SetEndErrorSize(2)
style.SetMarkerStyle(20)
style.SetOptFile(0)
style.SetOptStat(0)
style.SetStatColor(ROOT.kWhite)
style.SetStatFont(42)
style.SetLegendFont(63) #43 to have the non-bold font
style.SetStatFontSize(0.08)
style.SetStatTextColor(1)
style.SetStatFormat("6.4g")
style.SetStatBorderSize(1)
style.SetStatH(0.7)
style.SetStatW(0.15)
#Margins
style.SetPadTopMargin(0.05)
style.SetPadBottomMargin(0.13)
style.SetPadLeftMargin(0.16)
style.SetPadRightMargin(0.04)
# For the Global title
style.SetOptTitle(0)
#For the axis titles:
style.SetTitleColor(1, "XYZ")
style.SetTitleFont(42, "XYZ")
style.SetTitleSize(0.06, "XYZ")
style.SetTitleXOffset(0.9)
style.SetTitleYOffset(1.25)
#For the axis labels:
style.SetLabelColor(1, "XYZ")
style.SetLabelFont(42, "XYZ")
style.SetLabelOffset(0.007, "XYZ")
style.SetLabelSize(0.05, "XYZ")
#For the axis:
style.SetAxisColor(1, "XYZ")
style.SetStripDecimals(ROOT.kTRUE)
style.SetTickLength(0.03, "XYZ")
style.SetNdivisions(505, "XYZ")
style.SetPadTickX(1)
style.SetPadTickY(1)
style.SetLegendBorderSize(0)
gROOT.SetStyle('style')
#important to force the style for histograms made before this style was set
gROOT.ForceStyle()
return style
def setTDRStyle():
tdrStyle = TStyle("tdrStyle", "Style for P-TDR")
# For the canvas:
tdrStyle.SetCanvasBorderMode(0)
tdrStyle.SetCanvasColor(TStyle.kWhite)
tdrStyle.SetCanvasDefH(600) #Height of canvas
tdrStyle.SetCanvasDefW(600) #Width of canvas
tdrStyle.SetCanvasDefX(0) #POsition on screen
tdrStyle.SetCanvasDefY(0)
# For the Pad:
tdrStyle.SetPadBorderMode(0)
# tdrStyle.SetPadBorderSize(Width_t size = 1)
tdrStyle.SetPadColor(TStyle.kWhite)
tdrStyle.SetPadGridX(False)
tdrStyle.SetPadGridY(False)
tdrStyle.SetGridColor(0)
tdrStyle.SetGridStyle(3)
tdrStyle.SetGridWidth(1)
# For the frame:
tdrStyle.SetFrameBorderMode(0)
tdrStyle.SetFrameBorderSize(1)
tdrStyle.SetFrameFillColor(0)
tdrStyle.SetFrameFillStyle(0)
tdrStyle.SetFrameLineColor(1)
tdrStyle.SetFrameLineStyle(1)
tdrStyle.SetFrameLineWidth(1)
# For the histo:
# tdrStyle.SetHistFillColor(1)
# tdrStyle.SetHistFillStyle(0)
tdrStyle.SetHistLineColor(1)
tdrStyle.SetHistLineStyle(0)
tdrStyle.SetHistLineWidth(1)
# tdrStyle.SetLegoInnerR(Float_t rad = 0.5)
# tdrStyle.SetNumberContours(Int_t number = 20)
tdrStyle.SetEndErrorSize(2)
# tdrStyle.SetErrorMarker(20)
#tdrStyle.SetErrorX(0.)
tdrStyle.SetMarkerStyle(20)
#For the fit/function:
tdrStyle.SetOptFit(1)
tdrStyle.SetFitFormat("5.4g")
tdrStyle.SetFuncColor(2)
tdrStyle.SetFuncStyle(1)
tdrStyle.SetFuncWidth(1)
#For the date:
tdrStyle.SetOptDate(0)
# tdrStyle.SetDateX(Float_t x = 0.01)
# tdrStyle.SetDateY(Float_t y = 0.01)
# For the statistics box:
tdrStyle.SetOptFile(0)
tdrStyle.SetOptStat(0) # To display the mean and RMS: SetOptStat("mr")
tdrStyle.SetStatColor(TStyle.kWhite)
tdrStyle.SetStatFont(42)
tdrStyle.SetStatFontSize(0.025)
tdrStyle.SetStatTextColor(1)
tdrStyle.SetStatFormat("6.4g")
tdrStyle.SetStatBorderSize(1)
tdrStyle.SetStatH(0.1)
tdrStyle.SetStatW(0.15)
# tdrStyle.SetStatStyle(Style_t style = 1001)
# tdrStyle.SetStatX(Float_t x = 0)
# tdrStyle.SetStatY(Float_t y = 0)
# Margins:
tdrStyle.SetPadTopMargin(0.05)
tdrStyle.SetPadBottomMargin(0.13)
tdrStyle.SetPadLeftMargin(0.16)
tdrStyle.SetPadRightMargin(0.02)
# For the Global title:
tdrStyle.SetOptTitle(0)
tdrStyle.SetTitleFont(42)
tdrStyle.SetTitleColor(1)
tdrStyle.SetTitleTextColor(1)
tdrStyle.SetTitleFillColor(10)
tdrStyle.SetTitleFontSize(0.05)
# tdrStyle.SetTitleH(0) # Set the height of the title box
# tdrStyle.SetTitleW(0) # Set the width of the title box
# tdrStyle.SetTitleX(0) # Set the position of the title box
# tdrStyle.SetTitleY(0.985) # Set the position of the title box
# tdrStyle.SetTitleStyle(Style_t style = 1001)
# tdrStyle.SetTitleBorderSize(2)
# For the axis titles:
tdrStyle.SetTitleColor(1, "XYZ")
tdrStyle.SetTitleFont(42, "XYZ")
tdrStyle.SetTitleSize(0.06, "XYZ")
# tdrStyle.SetTitleXSize(Float_t size = 0.02) # Another way to set the size?
# tdrStyle.SetTitleYSize(Float_t size = 0.02)
tdrStyle.SetTitleXOffset(0.9)
tdrStyle.SetTitleYOffset(1.)
# tdrStyle.SetTitleOffset(1.1, "Y") # Another way to set the Offset
# For the axis labels:
tdrStyle.SetLabelColor(1, "XYZ")
tdrStyle.SetLabelFont(42, "XYZ")
tdrStyle.SetLabelOffset(0.007, "XYZ")
tdrStyle.SetLabelSize(0.05, "XYZ")
# For the axis:
tdrStyle.SetAxisColor(1, "XYZ")
tdrStyle.SetStripDecimals(True)
tdrStyle.SetTickLength(0.03, "XYZ")
tdrStyle.SetNdivisions(510, "XYZ")
tdrStyle.SetPadTickX(
1) # To get tick marks on the opposite side of the frame
tdrStyle.SetPadTickY(1)
# Change for log plots:
tdrStyle.SetOptLogx(0)
tdrStyle.SetOptLogy(0)
tdrStyle.SetOptLogz(0)
# Postscript options:
tdrStyle.SetPaperSize(20., 20.)
tdrStyle.SetHatchesLineWidth(5)
tdrStyle.SetHatchesSpacing(0.05)
tdrStyle.cd()
def plot_multiple_components(work_dir,
root_file_path,
plot_config,
info_file_path,
info_file_index=0):
import os
import pandas as pd
# - Plotting style
myStyle = TStyle("myStyle", "My own Root Style")
myStyle.SetTitleSize(0.3)
myStyle.SetTitleX(0.3)
myStyle.SetTitleXOffset(1.5)
myStyle.SetTitleYOffset(1.5)
myStyle.SetTitleSize(0.04, "xy")
myStyle.SetOptStat(0)
myStyle.SetCanvasColor(0)
myStyle.SetFrameBorderMode(0)
gROOT.SetStyle("myStyle")
line_width = 2
text_size = 0.025
df = pd.read_table(info_file_path, delim_whitespace=True)
print("plot_config['components']", plot_config['components'])
nComponents = len(plot_config['components'])
th = nested_dict(3, dict)
tf = TFile(root_file_path)
print(plot_config)
for level, level_opts in plot_config['levels'].iteritems():
for type, type_opts in plot_config['types'].iteritems():
for component, comp_opts in plot_config['components'].iteritems():
print("xsec_{}".format(comp_opts['name']))
xsec = df.loc[info_file_index, "xsec_{}".format(
comp_opts['name'])] * comp_opts['K_factor'] * 1000
th[component][type][level] = tf.Get(
plot_config['histo_name_template_format'].format(
comp_opts['name'], type, level))
print('histo name: {}'.format(
plot_config['histo_name_template_format'].format(
comp_opts['name'], type, level)))
print('level: {}'.format(level))
w = get_weight_to_xsec(
xsec, comp_opts['nEvents']
) # xsec * 1000 to convert from pb to fb
th[component][type][level].Scale(w, 'width')
th[component][type][level].SetLineColor(comp_opts['color'])
th[component][type][level].SetMarkerColor(comp_opts['color'])
th[component][type][level].SetLineWidth(
comp_opts['line_width'])
th[component][type][level].SetLineStyle(
comp_opts['line_style'])
print('\nw[key]:', w)
# - Plotting
c = TCanvas('canvas', 'canvas', 800, 600)
th[plot_config['first_plot']][type][level].GetXaxis().SetTitle(
type_opts['xlabel'])
th[plot_config['first_plot']][type][level].GetYaxis().SetTitle(
type_opts['ylabel'])
# int = {}
# int['gg'] = th['gg'][type][level].Integral(bin1, bin2, 'width' )
# int['qq'] = th['qq'][type][level].Integral(bin1, bin2, 'width' )
#
# print("integral1: {:.2f}".format(int['gg']) )
# print("integral2: {:.2f}".format(int['qq']) )
#th[ plot_config['first_plot'] ][type][level].GetYaxis().SetRangeUser(-1.5,8)
if 'ranges' in plot_config:
th[plot_config['first_plot']][type][level].GetYaxis(
).SetRangeUser(plot_config['ranges']['y_min'],
plot_config['ranges']['y_max'])
#th['all'].GetXaxis().SetRangeUser(200,1000)
#th['all'].GetYaxis().SetRangeUser(0.0006,20)
#h_all.GetYaxis().SetRangeUser(200,1000)
gPad.SetLeftMargin(0.14)
gPad.SetBottomMargin(0.13)
gPad.SetRightMargin(0.03)
gPad.SetTopMargin(0.03)
label_posx, label_posy = plot_config['label'][
'pos_x'], plot_config['label']['pos_y']
label_int_posx, label_int_posy = plot_config['label_int'][
'pos_x'], plot_config['label_int']['pos_y']
text_vshift = plot_config['label']['shift_y']
labels = []
labels_int = []
labels_param = []
labels.append(
TLatex(plot_config['label']['pos_x'],
plot_config['label']['pos_y'],
plot_config['label']['text']))
labels[0].SetNDC(True)
labels[0].SetTextSize(plot_config['label']['text_size'])
for i, (comp, comp_opts) in enumerate(
plot_config['components'].iteritems(), 1):
xsec = df.loc[info_file_index, "xsec_{}".format(
comp_opts['name'])] * comp_opts['K_factor'] * 1000
text = "#sigma_{{tot}}({}) = {:.2f} fb".format(
comp_opts['label'], xsec)
labels.append(
TLatex(label_posx, label_posy - i * text_vshift, text))
labels[i].SetNDC(True)
labels[i].SetTextSize(plot_config['label']['text_size'])
labels_int.append(
TLatex(plot_config['label_int']['pos_x'],
plot_config['label_int']['pos_y'], level_opts['name']))
labels_int[0].SetNDC(True)
labels_int[0].SetTextSize(plot_config['label_int']['text_size'])
for i, (comp, comp_opts) in enumerate(
plot_config['components'].iteritems(), 1):
bin1 = th[comp_opts['name']][type][level].FindBin(000.0)
bin2 = th[comp_opts['name']][type][level].FindBin(1500.0)
xsec = th[comp_opts['name']][type][level].Integral(
bin1, bin2, 'width')
text = "#sigma_{{int}}({}) = {:.2f} fb".format(
comp_opts['label'], xsec)
labels_int.append(
TLatex(label_int_posx, label_int_posy - i * text_vshift,
text))
labels_int[i].SetNDC(True)
labels_int[i].SetTextSize(
plot_config['label_int']['text_size'])
# for i, (row, row_opts) in enumerate(plot_config['label_param']['rows'].iteritems(), 0):
# text = row_opts['text'].format( df.loc[info_file_index, row_opts['input'][0]],
# df.loc[info_file_index, row_opts['input'][1]])
# labels_param.append( TLatex(plot_config['label_param']['pos_x'], plot_config['label_param']['pos_y']-i*plot_config['label_param']['shift_y'], text ) )
# labels_param[i].SetNDC( True )
# labels_param[i].SetTextSize( plot_config['label_param']['text_size'] )
# text2 = "#sigma_{{tot}}(gg #rightarrow Zh)={:.2f} fb".format( xsec['gg'] )
# text2 = "#sigma_{{tot}}(gg #rightarrow Zh)={:.2f} fb".format( xsec['gg'] )
# text3 = "#sigma_{{tot}}(qq #rightarrow Zh)={:.2f} fb".format( xsec['qq'] )
# text4 = "#sigma_{{integral}}(gg #rightarrow Zh)={:.2f} fb".format( int['gg'] )
# text5 = "#sigma_{{integral}}(qq #rightarrow Zh)={:.2f} fb".format( int['qq'] )
# label1 = TLatex(label_posx, label_posy, text1 )
# label1.SetNDC( True )
# label1.SetTextSize( text_size )
# label2 = TLatex(label_posx, label_posy-text_vshift, text2 )
# label2.SetNDC( True )
# label2.SetTextSize( text_size )
# label3 = TLatex(label_posx, label_posy-2.0*text_vshift, text3 )
# label3.SetNDC( True )
# label3.SetTextSize( text_size )
# label4 = TLatex(label_posx, label_posy-3.0*text_vshift, text4 )
# label4.SetNDC( True )
# label4.SetTextSize( text_size )
# label5 = TLatex(label_posx, label_posy-4.0*text_vshift, text5 )
# label5.SetNDC( True )
# label5.SetTextSize( text_size )
legend = TLegend(plot_config['legend']['bl_corner_x'],
plot_config['legend']['bl_corner_y'],
plot_config['legend']['tr_corner_x'],
plot_config['legend']['tr_corner_y'])
legend.SetTextSize(plot_config['legend']['textsize'])
for comp, comp_opts in plot_config['components'].iteritems():
legend.AddEntry(th[comp][type][level], comp_opts['label'])
# - Label
#label = TLatex( plot_config['label']['pos_x'], plot_config['label']['pos_y'],
# plot_config['label']['text'] )
#label.SetNDC( True )
#label.SetTextSize( plot_config['label']['text_size'] )
############################
## --- Linear plot
c.SetLogy(0)
#th['all'].GetYaxis().SetRangeUser(0.0006,3.5)
th[plot_config['first_plot']][type][level].Draw("hist")
for comp in plot_config['remaining_plots']:
th[comp][type][level].Draw("SAME hist")
for label in labels:
label.Draw("SAME")
for label in labels_int:
label.Draw("SAME")
# for label in labels_param:
# label.Draw("SAME")
# label1.Draw("SAME")
# label2.Draw("SAME")
# label3.Draw("SAME")
# label4.Draw("SAME")
# label5.Draw("SAME")
legend.Draw("SAME")
# label.Draw("SAME")
c.SaveAs(
os.path.join(work_dir,
'{}_{}_lin.pdf'.format(type_opts['name'], level)))
############################
## --- Log plot
c.Clear()
c.SetLogy()
th[plot_config['first_plot']][type][level].Draw("hist")
for comp in plot_config['remaining_plots']:
th[comp][type][level].Draw("SAME hist")
# label1.Draw("SAME")
# label2.Draw("SAME")
# label3.Draw("SAME")
# label4.Draw("SAME")
# label5.Draw("SAME")
legend.Draw("SAME")
c.SaveAs(
os.path.join(work_dir,
'{}_{}_log.pdf'.format(type_opts['name'], level)))
tdrStyle.SetCanvasBorderMode(0) tdrStyle.SetCanvasColor(kWhite) tdrStyle.SetCanvasDefH(600) tdrStyle.SetCanvasDefW(600) tdrStyle.SetCanvasDefX(0) tdrStyle.SetCanvasDefY(0) tdrStyle.SetPadBorderMode(0) tdrStyle.SetPadColor(kWhite) tdrStyle.SetPadGridX(False) tdrStyle.SetPadGridY(False) tdrStyle.SetGridColor(0) tdrStyle.SetGridStyle(3) tdrStyle.SetGridWidth(1) tdrStyle.SetFrameBorderMode(0) tdrStyle.SetFrameBorderSize(1) tdrStyle.SetFrameFillColor(0) tdrStyle.SetFrameFillStyle(0) tdrStyle.SetFrameLineColor(1) tdrStyle.SetFrameLineStyle(1) tdrStyle.SetFrameLineWidth(1) tdrStyle.SetHistFillColor(63) tdrStyle.SetHistLineColor(1) tdrStyle.SetHistLineStyle(0) tdrStyle.SetHistLineWidth(1) tdrStyle.SetErrorX(0.) tdrStyle.SetMarkerStyle(20)
def producePlots(filelist_handle,identifier):
"""
produce plots for filelist
"""
files = {}
plots = []
failures = []
for line in filelist_handle.readlines():
array = line.split()
if len(array) > 3 :
name = str(array[0])
size = float(float(array[1])/1000000000000)
events = int(array[2])
run = int(array[3])
lumi = int(array[4])
if array[-1] != 'GMT' :
failures.append(line)
else :
thetime = int(time.mktime(time.strptime(' '.join(array[5:]),'%a, %d %b %Y %H:%M:%S %Z')))
if name in files.keys():
files[name]['lumi'].append(lumi)
else :
entry = {}
entry['size'] = size
entry['events'] = events
entry['run'] = [run]
entry['lumi'] = [lumi]
entry['time'] = thetime
files[name] = entry
if debug == 1 :
for name in files.keys():
print name,'size:',files[name]['size'],'events:',files[name]['events'],'runs:',','.join(map(str,files[name]['run'])),'lumis:',','.join(map(str,files[name]['lumi'])),'date:',files[name]['time']
# print failures
if len(failures) > 0 :
print ""
print "failures:"
for line in failures :
print line
# define time axis
times = [ files[name]['time'] for name in files ]
times.sort()
first_day = times[0]-times[0]%86400
last_day = times[len(times)-1]+(86400-times[len(times)-1]%86400)
days = int((last_day - first_day)/86400.)
# determine x axis division
if days%10 == 0:
bigdiv = 10
smalldiv = days/10
elif days%9 == 0:
bigdiv = 9
smalldiv = days/9
elif days%8 == 0:
bigdiv = 8
smalldiv = days/8
elif days%7 == 0:
bigdiv = 7
smalldiv = days/7
elif days%6 == 0:
bigdiv = 6
smalldiv = days/6
elif days%5 == 0:
bigdiv = 5
smalldiv = days/5
elif days%4 == 0:
bigdiv = 4
smalldiv = days/4
elif days%3 == 0:
bigdiv = 3
smalldiv = days/3
elif days%2 == 0:
bigdiv = 2
smalldiv = days/2
else :
bigdiv = 1
smalldiv = days
gROOT.SetBatch(1)
myStyle = TStyle("Default","Default");
myStyle.cd();
myStyle.SetFrameBorderMode(0);
myStyle.SetCanvasBorderMode(0);
myStyle.SetPadBorderMode(0);
myStyle.SetPadColor(0);
myStyle.SetCanvasColor(0);
myStyle.SetTitleFillColor(0);
myStyle.SetTitleBorderSize(1);
myStyle.SetStatColor(0);
myStyle.SetStatBorderSize(1);
myStyle.SetOptStat(0);
myStyle.SetTimeOffset(first_day+86400)
c = TCanvas()
c.SetLeftMargin(0.171)
c.SetRightMargin(0.04)
c.SetBottomMargin(0.093)
c.SetTopMargin(0.08)
c.SetGridx(1)
c.SetGridy(1)
# events histogram
print 'Drawing histograms for average number of events per day'
h_events = TH1D("NumberOfEvents" + identifier,"",days,0.,days*86400.)
h_events.GetYaxis().SetTitle("Number of events [1/24h]")
h_events.GetYaxis().SetTitleSize(0.08)
h_events.GetYaxis().SetTitleColor(4)
h_events.GetYaxis().SetTitleOffset(1.14)
h_events.GetYaxis().SetLabelSize(0.07)
h_events.GetYaxis().SetLabelOffset(0.01)
h_events.GetXaxis().SetTimeDisplay(1)
h_events.GetXaxis().SetTimeFormat("%m/%d")
h_events.GetXaxis().SetLabelSize(0.05)
h_events.GetXaxis().SetLabelOffset(0.035)
h_events.GetXaxis().SetNdivisions(smalldiv * 100 + bigdiv,0)
h_events.SetLineColor(4)
h_events.SetLineWidth(3)
total_events = 0
for name in files.keys():
total_events += files[name]['events']
h_events.Fill(files[name]['time']-first_day,files[name]['events'])
h_events.SetMaximum(h_events.GetMaximum()*1.3)
h_events.Draw()
t_events = TPaveText(0.2,0.75,0.9,0.9,'brNDC')
t_events.AddText('CMS 2010: %s' % identifier)
t_events.AddText('Total number of events: %.2E' % (total_events))
t_events.AddText('Last updated on: %s' % (timestring))
t_events.SetFillColor(0)
t_events.SetFillStyle(0)
t_events.SetBorderSize(0)
t_events.SetTextAlign(12)
t_events.Draw()
name = timefilestring + '_events' + identifier.replace('/','_') + '.png'
plots.append(name)
c.SaveAs(name)
c.Clear()
# rate histogram
print 'Drawing histograms for average rate per day'
h_rate = TH1D("Rate" + identifier,"",days,0.,days*86400.)
h_rate.GetYaxis().SetTitle("Average Rate [Hz/24h]")
h_rate.GetYaxis().SetTitleSize(0.08)
h_rate.GetYaxis().SetTitleColor(4)
h_rate.GetYaxis().SetTitleOffset(1.14)
h_rate.GetYaxis().SetLabelSize(0.07)
h_rate.GetYaxis().SetLabelOffset(0.01)
h_rate.GetXaxis().SetTimeDisplay(1)
h_rate.GetXaxis().SetTimeFormat("%m/%d")
h_rate.GetXaxis().SetLabelSize(0.05)
h_rate.GetXaxis().SetLabelOffset(0.035)
h_rate.GetXaxis().SetNdivisions(smalldiv * 100 + bigdiv,0)
h_rate.SetLineColor(4)
h_rate.SetLineWidth(3)
h_rate_max = 0.
for bin in range(days):
rate = h_events.GetBinContent(bin+1)/24./3600.
if rate > h_rate_max :
h_rate_max = rate
h_rate.SetBinContent(bin+1,rate)
h_rate.SetMaximum(h_rate_max*1.3)
h_rate.Draw()
t_rate = TPaveText(0.2,0.75,0.9,0.9,'brNDC')
t_rate.AddText('CMS 2010: %s' % identifier)
t_rate.AddText('Average rate: %.2f Hz' % (float(total_events)/float(days)/24./3600.))
t_rate.AddText('Last updated on: %s' % (timestring))
t_rate.SetFillColor(0)
t_rate.SetFillStyle(0)
t_rate.SetBorderSize(0)
t_rate.SetTextAlign(12)
t_rate.Draw()
name = timefilestring + '_rate' + identifier.replace('/','_') + '.png'
plots.append(name)
c.SaveAs(name)
c.Clear()
# size histogram
print 'Drawing histograms for average size per day'
h_size = TH1D("Size"+identifier,"",days,0.,days*86400.)
h_size.GetYaxis().SetTitle("Size [TB/24h]")
h_size.GetYaxis().SetTitleSize(0.08)
h_size.GetYaxis().SetTitleColor(4)
h_size.GetYaxis().SetTitleOffset(1.12)
h_size.GetYaxis().SetLabelSize(0.07)
h_size.GetYaxis().SetLabelOffset(0.01)
h_size.GetXaxis().SetTimeDisplay(1)
h_size.GetXaxis().SetTimeFormat("%m/%d")
h_size.GetXaxis().SetLabelSize(0.05)
h_size.GetXaxis().SetLabelOffset(0.035)
h_size.GetXaxis().SetNdivisions(smalldiv * 100 + bigdiv,0)
h_size.SetLineColor(4)
h_size.SetLineWidth(3)
total_size = 0.
for name in files.keys():
total_size += files[name]['size']
h_size.Fill(files[name]['time']-first_day,files[name]['size'])
h_size.SetMaximum(h_size.GetMaximum()*1.3)
h_size.Draw()
t_size = TPaveText(0.2,0.75,0.9,0.9,'brNDC')
t_size.AddText('CMS 2010: %s' % identifier)
t_size.AddText('Total size: %.2f TB' % (total_size))
t_size.AddText('Last updated on: %s' % (timestring))
t_size.SetFillColor(0)
t_size.SetFillStyle(0)
t_size.SetBorderSize(0)
t_size.SetTextAlign(12)
t_size.Draw()
#raw_input('Press Enter...')
name = timefilestring + '_size' + identifier.replace('/','_') + '.png'
plots.append(name)
c.SaveAs(name)
c.Clear()
return plots
def setTDRStyle():
print("Using TDR Style")
tdrStyle = TStyle("tdrStyle","Style for P-TDR")
tdrStyle.SetPalette(1)
# For the canvas:
tdrStyle.SetCanvasBorderMode(0)
tdrStyle.SetCanvasColor(0)
tdrStyle.SetCanvasDefH(900) #Height of canvas
tdrStyle.SetCanvasDefW(600) #Width of canvas
tdrStyle.SetCanvasDefX(0) #POsition on screen
tdrStyle.SetCanvasDefY(0)
# For the Pad:
tdrStyle.SetPadBorderMode(0)
tdrStyle.SetPadColor(0)
#tdrStyle.SetPadGridX(false)
#tdrStyle.SetPadGridY(false)
tdrStyle.SetGridColor(0)
tdrStyle.SetGridStyle(3)
tdrStyle.SetGridWidth(1)
# For the frame:
tdrStyle.SetFrameBorderMode(0)
tdrStyle.SetFrameBorderSize(1)
tdrStyle.SetFrameFillColor(0)
tdrStyle.SetFrameFillStyle(0)
tdrStyle.SetFrameLineColor(1)
tdrStyle.SetFrameLineStyle(1)
tdrStyle.SetFrameLineWidth(1)
# For the histo:
#tdrStyle.SetHistFillColor(0)
#tdrStyle.SetHistFillStyle(0)
#tdrStyle.SetHistLineColor(1)
#tdrStyle.SetHistLineStyle(0)
#tdrStyle.SetHistLineWidth(1)
tdrStyle.SetEndErrorSize(2)
#tdrStyle.SetErrorMarker(20)
#tdrStyle.SetErrorX(0.)
tdrStyle.SetMarkerStyle(20)
#For the fit/function:
#tdrStyle.SetOptFit(0010) # display fit parameters values only
tdrStyle.SetOptFit(0000)
tdrStyle.SetFitFormat("5.4g")
tdrStyle.SetFuncColor(2)
tdrStyle.SetFuncStyle(1)
tdrStyle.SetFuncWidth(2)
# for the legends
tdrStyle.SetLegendBorderSize(0)
tdrStyle.SetLegendFillColor(0)
tdrStyle.SetLegendFont(42)
#For the date:
tdrStyle.SetOptDate(0)
# tdrStyle.SetDateX(Float_t x = 0.01)
# tdrStyle.SetDateY(Float_t y = 0.01)
# For the statistics box:
tdrStyle.SetOptFile(0)
tdrStyle.SetOptStat(0) # To display the mean and RMS: SetOptStat("mr")
tdrStyle.SetStatColor(0)
tdrStyle.SetStatFont(42)
tdrStyle.SetStatFontSize(0.025)
tdrStyle.SetStatTextColor(1)
tdrStyle.SetStatFormat("6.4g")
tdrStyle.SetStatBorderSize(1)
tdrStyle.SetStatH(0.1)
tdrStyle.SetStatW(0.15)
# tdrStyle.SetStatStyle(Style_t style = 1001)
# tdrStyle.SetStatX(Float_t x = 0)
# tdrStyle.SetStatY(Float_t y = 0)
# Margins:
tdrStyle.SetPadTopMargin(0.06)
tdrStyle.SetPadBottomMargin(0.13)
tdrStyle.SetPadLeftMargin(0.16)
tdrStyle.SetPadRightMargin(0.05)
# For the Global title:
tdrStyle.SetOptTitle(0)
tdrStyle.SetTitleFont(42)
tdrStyle.SetTitleColor(1)
tdrStyle.SetTitleTextColor(1)
tdrStyle.SetTitleFillColor(10)
tdrStyle.SetTitleFontSize(0.05)
# tdrStyle.SetTitleH(0) # Set the height of the title box
# tdrStyle.SetTitleW(0) # Set the width of the title box
# tdrStyle.SetTitleX(0) # Set the position of the title box
# tdrStyle.SetTitleY(0.985) # Set the position of the title box
# tdrStyle.SetTitleStyle(Style_t style = 1001)
# tdrStyle.SetTitleBorderSize(2)
# For the axis titles:
tdrStyle.SetTitleColor(1, "XYZ")
tdrStyle.SetTitleFont(42, "XYZ")
tdrStyle.SetTitleSize(0.06, "XYZ")
# tdrStyle.SetTitleXSize(Float_t size = 0.02) # Another way to set the size?
# tdrStyle.SetTitleYSize(Float_t size = 0.02)
tdrStyle.SetTitleXOffset(0.9)
tdrStyle.SetTitleYOffset(1.25)
# tdrStyle.SetTitleOffset(1.1, "Y") # Another way to set the Offset
# For the axis labels:
tdrStyle.SetLabelColor(1, "XYZ")
tdrStyle.SetLabelFont(42, "XYZ")
tdrStyle.SetLabelOffset(0.007, "XYZ")
tdrStyle.SetLabelSize(0.05, "XYZ")
# For the axis:
tdrStyle.SetAxisColor(1, "XYZ")
tdrStyle.SetStripDecimals(True)
tdrStyle.SetTickLength(0.03, "XYZ")
tdrStyle.SetNdivisions(508, "XYZ")
tdrStyle.SetPadTickX(1) # To get tick marks on the opposite side of the frame
tdrStyle.SetPadTickY(1)
# Change for log plots:
tdrStyle.SetOptLogx(0)
tdrStyle.SetOptLogy(0)
tdrStyle.SetOptLogz(0)
# Postscript options:
tdrStyle.SetPaperSize(20.,20.)
# tdrStyle.SetLineScalePS(Float_t scale = 3)
# tdrStyle.SetLineStyleString(Int_t i, const char* text)
# tdrStyle.SetHeaderPS(const char* header)
# tdrStyle.SetTitlePS(const char* pstitle)
# tdrStyle.SetBarOffset(Float_t baroff = 0.5)
# tdrStyle.SetBarWidth(Float_t barwidth = 0.5)
# tdrStyle.SetPaintTextFormat(const char* format = "g")
# tdrStyle.SetPalette(Int_t ncolors = 0, Int_t* colors = 0)
# tdrStyle.SetTimeOffset(Double_t toffset)
# tdrStyle.SetHistMinimumZero(kTRUE)
tdrStyle.cd()
def looks_CmsTDR():
tdrStyle = TStyle("tdrStyle", "Style for P-TDR")
# For the canvas:
tdrStyle.SetCanvasBorderMode(0)
tdrStyle.SetCanvasColor(ROOT.kWhite)
tdrStyle.SetCanvasDefH(600)
#Height of canvas
tdrStyle.SetCanvasDefW(600)
#Width of canvas
tdrStyle.SetCanvasDefX(0)
#POsition on screen
tdrStyle.SetCanvasDefY(0)
# For the Pad:
tdrStyle.SetPadBorderMode(0)
# tdrStyle.SetPadBorderSize(Width_t size = 1);
tdrStyle.SetPadColor(ROOT.kWhite)
#tdrStyle.SetPadGridX(false);
#tdrStyle.SetPadGridY(false);
tdrStyle.SetGridColor(0)
tdrStyle.SetGridStyle(3)
tdrStyle.SetGridWidth(1)
# For the frame:
tdrStyle.SetFrameBorderMode(0)
tdrStyle.SetFrameBorderSize(1)
tdrStyle.SetFrameFillColor(0)
tdrStyle.SetFrameFillStyle(0)
tdrStyle.SetFrameLineColor(1)
tdrStyle.SetFrameLineStyle(1)
tdrStyle.SetFrameLineWidth(1)
# For the histo:
# tdrStyle.SetHistFillColor(1);
# tdrStyle.SetHistFillStyle(0);
tdrStyle.SetHistLineColor(1)
tdrStyle.SetHistLineStyle(0)
tdrStyle.SetHistLineWidth(1)
# tdrStyle.SetLegoInnerR(Float_t rad = 0.5);
# tdrStyle.SetNumberContours(Int_t number = 20);
tdrStyle.SetEndErrorSize(2)
#tdrStyle.SetErrorMarker(20); # Seems to give an error
tdrStyle.SetErrorX(0.)
tdrStyle.SetMarkerStyle(20)
#For the fit/function:
tdrStyle.SetOptFit(1)
tdrStyle.SetFitFormat("5.4g")
tdrStyle.SetFuncColor(2)
tdrStyle.SetFuncStyle(1)
tdrStyle.SetFuncWidth(1)
#For the date:
tdrStyle.SetOptDate(0)
# tdrStyle.SetDateX(Float_t x = 0.01);
# tdrStyle.SetDateY(Float_t y = 0.01);
# For the statistics box:
tdrStyle.SetOptFile(0)
tdrStyle.SetOptStat(0)
# To display the mean and RMS: SetOptStat("mr");
tdrStyle.SetStatColor(ROOT.kWhite)
tdrStyle.SetStatFont(42)
tdrStyle.SetStatFontSize(0.025)
tdrStyle.SetStatTextColor(1)
tdrStyle.SetStatFormat("6.4g")
tdrStyle.SetStatBorderSize(1)
tdrStyle.SetStatH(0.1)
tdrStyle.SetStatW(0.15)
# tdrStyle.SetStatStyle(Style_t style = 1001);
# tdrStyle.SetStatX(Float_t x = 0);
# tdrStyle.SetStatY(Float_t y = 0);
# Margins:
tdrStyle.SetPadTopMargin(0.05)
tdrStyle.SetPadBottomMargin(0.13)
tdrStyle.SetPadLeftMargin(0.16)
tdrStyle.SetPadRightMargin(0.02)
# For the Global title:
tdrStyle.SetOptTitle(1)
# 0=No Title
tdrStyle.SetTitleFont(42)
tdrStyle.SetTitleColor(1)
tdrStyle.SetTitleTextColor(1)
tdrStyle.SetTitleFillColor(10)
tdrStyle.SetTitleFontSize(0.05)
# tdrStyle.SetTitleH(0); # Set the height of the title box
# tdrStyle.SetTitleW(0); # Set the width of the title box
# tdrStyle.SetTitleX(0); # Set the position of the title box
# tdrStyle.SetTitleY(0.985); # Set the position of the title box
# tdrStyle.SetTitleStyle(Style_t style = 1001);
# tdrStyle.SetTitleBorderSize(2);
# For the axis titles:
tdrStyle.SetTitleColor(1, "XYZ")
tdrStyle.SetTitleFont(42, "XYZ")
tdrStyle.SetTitleSize(0.06, "XYZ")
# tdrStyle.SetTitleXSize(Float_t size = 0.02); # Another way to set the size?
# tdrStyle.SetTitleYSize(Float_t size = 0.02);
tdrStyle.SetTitleXOffset(0.9)
tdrStyle.SetTitleYOffset(1.25)
# tdrStyle.SetTitleOffset(1.1, "Y"); # Another way to set the Offset
# For the axis labels:
tdrStyle.SetLabelColor(1, "XYZ")
tdrStyle.SetLabelFont(42, "XYZ")
tdrStyle.SetLabelOffset(0.007, "XYZ")
tdrStyle.SetLabelSize(0.05, "XYZ")
# For the axis:
tdrStyle.SetAxisColor(1, "XYZ")
tdrStyle.SetStripDecimals(ROOT.kTRUE)
tdrStyle.SetTickLength(0.03, "XYZ")
tdrStyle.SetNdivisions(510, "XYZ")
tdrStyle.SetPadTickX(0)
# 0=Text labels (and tics) only on bottom, 1=Text labels on top and bottom
tdrStyle.SetPadTickY(1)
# Change for log plots:
tdrStyle.SetOptLogx(0)
tdrStyle.SetOptLogy(0)
tdrStyle.SetOptLogz(0)
# Postscript options:
tdrStyle.SetPaperSize(20., 20.)
# tdrStyle.SetLineScalePS(Float_t scale = 3);
# tdrStyle.SetLineStyleString(Int_t i, const char* text);
# tdrStyle.SetHeaderPS(const char* header);
# tdrStyle.SetTitlePS(const char* pstitle);
# tdrStyle.SetBarOffset(Float_t baroff = 0.5);
# tdrStyle.SetBarWidth(Float_t barwidth = 0.5);
# tdrStyle.SetPaintTextFormat(const char* format = "g");
# tdrStyle.SetPalette(Int_t ncolors = 0, Int_t* colors = 0);
# tdrStyle.SetTimeOffset(Double_t toffset);
# tdrStyle.SetHistMinimumZero(kTRUE);
tdrStyle.cd()
gROOT.ForceStyle()
# Try this if stuff doesn't work right
gStyle.ls()
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with SpartyJet; if not, write to the Free Software
# Foundation, Inc.:
# 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
#----------------------------------------------------------------------
from ROOT import gROOT, TStyle
print "\nApplying SpartyJet style settings for ROOT...\n"
SpartyStyle = TStyle("SpartyJet", "SpartyJet style")
# whitening
SpartyStyle.SetFrameBorderMode(0)
SpartyStyle.SetFrameFillColor(0)
SpartyStyle.SetCanvasBorderMode(0)
SpartyStyle.SetCanvasColor(0)
SpartyStyle.SetPadBorderMode(0)
SpartyStyle.SetPadColor(0)
SpartyStyle.SetStatColor(0)
SpartyStyle.SetGridColor(0)
# pad margins
SpartyStyle.SetPadTopMargin(0.05)
SpartyStyle.SetPadRightMargin(0.05)
SpartyStyle.SetPadBottomMargin(0.16)
SpartyStyle.SetPadLeftMargin(0.16)
SpartyStyle.SetTitleOffset(1.4, 'x')
SpartyStyle.SetTitleOffset(1.4, 'y')
# stat box
def setLHCbStyle():
global lhcbStyle
global lhcbText
global lhcbLatex
lhcbStyle = TStyle("lhcbStyle", "Standard LHCb plots style")
# use times new roman
lhcbFont = 132
# line thickness
lhcbWidth = 2
lhcbTSize = 0.06
#// use plain black on white colors
lhcbStyle.SetFrameBorderMode(0)
lhcbStyle.SetCanvasBorderMode(0)
lhcbStyle.SetPadBorderMode(0)
lhcbStyle.SetPadColor(0)
lhcbStyle.SetCanvasColor(0)
lhcbStyle.SetStatColor(0)
lhcbStyle.SetPalette(1)
lhcbStyle.SetLegendBorderSize(0)
lhcbStyle.SetLegendFont(132)
lhcbStyle.SetFillColor(1)
lhcbStyle.SetFillStyle(1001)
# set the paper & margin sizes
lhcbStyle.SetPaperSize(20, 26)
lhcbStyle.SetPadTopMargin(0.1)
lhcbStyle.SetPadRightMargin(0.05)
lhcbStyle.SetPadBottomMargin(0.16)
lhcbStyle.SetPadLeftMargin(0.14)
# use large fonts
lhcbStyle.SetTextFont(lhcbFont)
lhcbStyle.SetTextSize(lhcbTSize)
# lhcbStyle.SetTextSize(0.08)
lhcbStyle.SetLabelFont(lhcbFont, "x")
lhcbStyle.SetLabelFont(lhcbFont, "y")
lhcbStyle.SetLabelFont(lhcbFont, "z")
lhcbStyle.SetLabelSize(lhcbTSize, "x")
lhcbStyle.SetLabelSize(lhcbTSize, "y")
lhcbStyle.SetLabelSize(lhcbTSize, "z")
lhcbStyle.SetTitleFont(lhcbFont)
lhcbStyle.SetTitleFont(lhcbFont, "x")
lhcbStyle.SetTitleFont(lhcbFont, "y")
lhcbStyle.SetTitleFont(lhcbFont, "z")
lhcbStyle.SetTitleSize(1.2 * lhcbTSize, "x")
lhcbStyle.SetTitleSize(1.2 * lhcbTSize, "y")
lhcbStyle.SetTitleSize(1.2 * lhcbTSize, "z")
# use bold lines and markers
lhcbStyle.SetLineWidth(lhcbWidth)
lhcbStyle.SetFrameLineWidth(lhcbWidth)
lhcbStyle.SetHistLineWidth(lhcbWidth)
lhcbStyle.SetFuncWidth(lhcbWidth)
lhcbStyle.SetGridWidth(lhcbWidth)
lhcbStyle.SetLineStyleString(2, "[12 12]")
lhcbStyle.SetMarkerStyle(20)
lhcbStyle.SetMarkerSize(1.0)
# label offsets
lhcbStyle.SetLabelOffset(0.010)
#titles
lhcbStyle.SetTitleOffset(0.95, "X")
lhcbStyle.SetTitleOffset(0.95, "Y")
lhcbStyle.SetTitleOffset(1.2, "Z")
lhcbStyle.SetTitleFillColor(0)
lhcbStyle.SetTitleStyle(0)
lhcbStyle.SetTitleBorderSize(0)
lhcbStyle.SetTitleFont(lhcbFont, "title")
lhcbStyle.SetTitleX(0.0)
lhcbStyle.SetTitleY(1.0)
lhcbStyle.SetTitleW(1.0)
lhcbStyle.SetTitleH(0.05)
# by default, do not display histogram decorations:
lhcbStyle.SetOptStat(0)
#lhcbStyle.SetOptStat("emr") # show only nent -e , mean - m , rms -r
#lhcbStyle.SetStatFormat("6.3g") # specified as c printf options
lhcbStyle.SetOptTitle(0)
lhcbStyle.SetOptFit(0)
#lhcbStyle.SetOptFit(1011) # order is probability, Chi2, errors, parameters
# look of the statistics box:
lhcbStyle.SetStatBorderSize(0)
lhcbStyle.SetStatFont(lhcbFont)
lhcbStyle.SetStatFontSize(0.05)
lhcbStyle.SetStatX(0.93)
lhcbStyle.SetStatY(0.88)
lhcbStyle.SetStatW(0.20)
lhcbStyle.SetStatH(0.15)
# put tick marks on top and RHS of plots
lhcbStyle.SetPadTickX(1)
lhcbStyle.SetPadTickY(1)
# histogram divisions: only 5 in x to avoid label overlaps
lhcbStyle.SetNdivisions(505, "x")
lhcbStyle.SetNdivisions(505, "y")
lhcbStyle.SetNdivisions(505, "z")
# define style for text
lhcbLabel = TText()
lhcbLabel.SetTextFont(lhcbFont)
lhcbLabel.SetTextColor(1)
lhcbLabel.SetTextSize(0.04)
lhcbLabel.SetTextAlign(12)
# define style of latex text
lhcbLatex = TLatex()
lhcbLatex.SetTextFont(lhcbFont)
lhcbLatex.SetTextColor(1)
lhcbLatex.SetTextSize(0.04)
lhcbLatex.SetTextAlign(12)
# set this style
gROOT.SetStyle("lhcbStyle")
gROOT.ForceStyle()
return
def setTDRStyle():
tdrStyle = TStyle("tdrStyle", "Style for P-TDR")
tdrStyle.SetCanvasBorderMode(0)
# tdrStyle.SetCanvasColor(kWhite)
tdrStyle.SetCanvasColor(0)
tdrStyle.SetCanvasDefH(600)
tdrStyle.SetCanvasDefW(600)
tdrStyle.SetCanvasDefX(0)
tdrStyle.SetCanvasDefY(0)
tdrStyle.SetPadBorderMode(0)
# tdrStyle.SetPadColor(kWhite)
tdrStyle.SetPadColor(0)
# tdrStyle.SetPadGridX(kFALSE)
tdrStyle.SetPadGridX(0)
# tdrStyle.SetPadGridY(false)
tdrStyle.SetPadGridY(0)
tdrStyle.SetGridColor(0)
tdrStyle.SetGridStyle(3)
tdrStyle.SetGridWidth(1)
tdrStyle.SetFrameBorderMode(0)
tdrStyle.SetFrameBorderSize(1)
tdrStyle.SetFrameFillColor(0)
tdrStyle.SetFrameFillStyle(0)
tdrStyle.SetFrameLineColor(1)
tdrStyle.SetFrameLineStyle(1)
tdrStyle.SetFrameLineWidth(1)
tdrStyle.SetHistLineColor(1)
tdrStyle.SetHistLineStyle(0)
tdrStyle.SetHistLineWidth(1)
tdrStyle.SetEndErrorSize(2)
tdrStyle.SetErrorX(0.)
tdrStyle.SetMarkerStyle(20)
tdrStyle.SetOptFit(0)
tdrStyle.SetFitFormat("5.4g")
tdrStyle.SetFuncColor(2)
tdrStyle.SetFuncStyle(1)
tdrStyle.SetFuncWidth(1)
tdrStyle.SetOptDate(0)
tdrStyle.SetOptFile(0)
tdrStyle.SetOptStat(0000000)
tdrStyle.SetStatColor(0)
tdrStyle.SetStatFont(42)
tdrStyle.SetStatFontSize(0.025)
tdrStyle.SetStatTextColor(1)
tdrStyle.SetStatFormat("6.4g")
tdrStyle.SetStatBorderSize(1)
tdrStyle.SetStatH(0.1)
tdrStyle.SetStatW(0.15)
tdrStyle.SetPadTopMargin(0.07)
# tdrStyle.SetPadBottomMargin(0.13)
tdrStyle.SetPadBottomMargin(0.16)
tdrStyle.SetPadLeftMargin(0.16)
# tdrStyle.SetPadRightMargin(0.02)
tdrStyle.SetPadRightMargin(0.06)
tdrStyle.SetOptTitle(0)
tdrStyle.SetTitleFont(42)
tdrStyle.SetTitleColor(1)
tdrStyle.SetTitleTextColor(1)
tdrStyle.SetTitleFillColor(10)
tdrStyle.SetTitleFontSize(0.05)
tdrStyle.SetTitleColor(1, "XYZ")
tdrStyle.SetTitleFont(42, "XYZ")
tdrStyle.SetTitleSize(0.06, "XYZ")
# tdrStyle.SetTitleXOffset(0.9)
tdrStyle.SetTitleXOffset(1.1)
# tdrStyle.SetTitleYOffset(1.25)
# tdrStyle.SetTitleYOffset(1.2)
tdrStyle.SetTitleYOffset(1.1)
tdrStyle.SetTitleColor(1, "XYZ")
tdrStyle.SetTitleFont(42, "XYZ")
tdrStyle.SetTitleSize(0.06, "XYZ")
tdrStyle.SetTitleXOffset(1.1)
# tdrStyle.SetTitleYOffset(1.25)
tdrStyle.SetTitleYOffset(1.23)
tdrStyle.SetLabelColor(1, "XYZ")
tdrStyle.SetLabelFont(42, "XYZ")
tdrStyle.SetLabelOffset(0.007, "XYZ")
# tdrStyle.SetLabelSize(0.05, "XYZ")
tdrStyle.SetLabelSize(0.045, "XYZ")
tdrStyle.SetAxisColor(1, "XYZ")
# tdrStyle.SetStripDecimals(kTRUE)
tdrStyle.SetStripDecimals(1)
tdrStyle.SetTickLength(0.03, "XYZ")
# tdrStyle.SetNdivisions(510, "XYZ")
tdrStyle.SetNdivisions(509, "XYZ")
tdrStyle.SetPadTickX(1)
tdrStyle.SetPadTickY(1)
tdrStyle.SetOptLogx(0)
tdrStyle.SetOptLogy(0)
tdrStyle.SetOptLogz(0)
tdrStyle.SetPaperSize(20., 20.)
tdrStyle.cd()
def setStyle():
style = TStyle('style', 'style')
style.SetCanvasBorderMode(0)
style.SetCanvasColor(ROOT.kWhite)
style.SetCanvasDefH(600)
style.SetCanvasDefW(600)
style.SetCanvasDefX(0)
style.SetCanvasDefY(0)
style.SetPadBorderMode(0)
style.SetPadColor(ROOT.kWhite)
style.SetPadGridX(False)
style.SetPadGridY(False)
style.SetGridColor(0)
style.SetGridStyle(3)
style.SetGridWidth(1)
style.SetFrameBorderMode(0)
style.SetFrameBorderSize(1)
style.SetFrameFillColor(0)
style.SetFrameFillStyle(0)
style.SetFrameLineColor(1)
style.SetFrameLineStyle(1)
style.SetFrameLineWidth(1)
style.SetHistLineColor(1)
style.SetHistLineStyle(0)
style.SetHistLineWidth(1)
style.SetEndErrorSize(2)
style.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:
style.SetOptFile(0)
style.SetOptStat(0)
style.SetStatColor(ROOT.kWhite)
style.SetStatFont(42)
style.SetLegendFont(42)
style.SetStatFontSize(0.08)
style.SetStatTextColor(1)
style.SetStatFormat("6.4g")
style.SetStatBorderSize(1)
style.SetStatH(0.7)
style.SetStatW(0.15)
#//gStyle->SetStatTextSize(2.5);
#gStyle->SetStatX(0.96);
#//gStyle->SetStatY(0.35);
#// gStyle->SetStatStyle(Style_t style = 1001);
#// gStyle->SetStatX(Float_t x = 0);
#// gStyle->SetStatY(Float_t y = 0);
#Margins:
style.SetPadTopMargin(0.05)
style.SetPadBottomMargin(0.13)
style.SetPadLeftMargin(0.16)
style.SetPadRightMargin(0.04)
# For the Global title:
style.SetOptTitle(0)
style.SetTitleFont(42)
style.SetTitleColor(1)
style.SetTitleTextColor(1)
style.SetTitleFillColor(10)
style.SetTitleFontSize(0.05)
#For the axis titles:
style.SetTitleColor(1, "XYZ")
style.SetTitleFont(42, "XYZ")
style.SetTitleSize(0.06, "XYZ")
style.SetTitleXOffset(0.9)
style.SetTitleYOffset(1.25)
#For the axis labels:
style.SetLabelColor(1, "XYZ")
style.SetLabelFont(42, "XYZ")
style.SetLabelOffset(0.007, "XYZ")
style.SetLabelSize(0.05, "XYZ")
#For the axis:
style.SetAxisColor(1, "XYZ")
style.SetStripDecimals(ROOT.kTRUE)
style.SetTickLength(0.03, "XYZ")
style.SetNdivisions(505, "XYZ")
style.SetPadTickX(1)
style.SetPadTickY(1)
style.SetEndErrorSize(0.)
##Change for log plots:
#gStyle->SetOptLogx(0);
#gStyle->SetOptLogy(0);
#gStyle->SetOptLogz(0);
#// Postscript options:
#gStyle->SetPaperSize(20.,20.);
#// 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->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);
#//For error in SF maps
#gStyle->SetMarkerSize(0.9); //TEMPORARY SIZE FOR DILEPTON PLOTS. SET BACK TO DEFAULT FOR TRILEPTON
#gStyle->SetPaintTextFormat("4.2f"); //4.2
#//gStyle->SetHatchesLineWidth(5);
#//gStyle->SetHatchesSpacing(0.05);
style.SetLegendBorderSize(0)
gROOT.SetStyle('style')
gROOT.ForceStyle()
return style
def officialStyle(style):
style = TStyle("tdrStyle","Style for P-TDR")
style.SetCanvasColor (0)
style.SetCanvasBorderSize(10)
style.SetCanvasBorderMode(0)
style.SetCanvasDefH (700)
style.SetCanvasDefW (700)
style.SetCanvasDefX (100)
style.SetCanvasDefY (100)
# color palette for 2D temperature plots
# style.SetPalette(1,0)
# Pads
style.SetPadColor (0)
style.SetPadBorderSize (10)
style.SetPadBorderMode (0)
style.SetPadBottomMargin(0.15)
style.SetPadTopMargin (0.05)
style.SetPadLeftMargin (0.17)
# style.SetPadRightMargin (0.03566265)
style.SetPadRightMargin (0.065)
style.SetPadGridX (0)
style.SetPadGridY (0)
style.SetPadTickX (1)
style.SetPadTickY (1)
# Frames
style.SetLineWidth(3)
style.SetFrameFillStyle ( 0)
style.SetFrameFillColor ( 0)
style.SetFrameLineColor ( 1)
style.SetFrameLineStyle ( 0)
style.SetFrameLineWidth ( 2)
style.SetFrameBorderSize(10)
style.SetFrameBorderMode( 0)
# Histograms
style.SetHistFillColor(2)
style.SetHistFillStyle(0)
style.SetHistLineColor(1)
style.SetHistLineStyle(0)
style.SetHistLineWidth(3)
style.SetNdivisions(505)
# Functions
style.SetFuncColor(1)
style.SetFuncStyle(0)
style.SetFuncWidth(2)
# Various
style.SetMarkerStyle(20)
style.SetMarkerColor(kBlack)
style.SetMarkerSize (1.4)
style.SetTitleBorderSize(0)
style.SetTitleFillColor (0)
style.SetTitleX (0.2)
style.SetTitleSize (0.055,"X")
style.SetTitleOffset(1.100,"X")
style.SetLabelOffset(0.005,"X")
style.SetLabelSize (0.050,"X")
style.SetLabelFont (42 ,"X")
style.SetStripDecimals(False)
style.SetLineStyleString(11,"20 10")
style.SetTitleSize (0.055,"Y")
style.SetTitleOffset(1.5,"Y")
style.SetLabelOffset(0.010,"Y")
style.SetLabelSize (0.050,"Y")
style.SetLabelFont (42 ,"Y")
style.SetTextSize (0.055)
style.SetTextFont (42)
style.SetStatFont (42)
style.SetTitleFont (42)
style.SetTitleFont (42,"X")
style.SetTitleFont (42,"Y")
style.SetOptStat (0)
style.SetLegendBorderSize(0)
style.cd()
def setStyle():
print "setting TDR style"
tdrStyle = TStyle("tdrStyle","Style for P-TDR")
kWhite = 0
tdrStyle.SetPalette(1)
tdrStyle.SetCanvasBorderMode(0)
tdrStyle.SetCanvasColor(kWhite)
tdrStyle.SetCanvasDefH(800)
tdrStyle.SetCanvasDefW(1000)
tdrStyle.SetCanvasDefX(0)
tdrStyle.SetCanvasDefY(0)
tdrStyle.SetPadBorderMode(0)
tdrStyle.SetPadColor(kWhite)
tdrStyle.SetPadGridX(False)
tdrStyle.SetPadGridY(False)
tdrStyle.SetGridColor(0)
tdrStyle.SetGridStyle(3)
tdrStyle.SetGridWidth(1)
tdrStyle.SetFrameBorderMode(0)
tdrStyle.SetFrameBorderSize(1)
tdrStyle.SetFrameFillColor(0)
tdrStyle.SetFrameFillStyle(0)
tdrStyle.SetFrameLineColor(1)
tdrStyle.SetFrameLineStyle(1)
tdrStyle.SetFrameLineWidth(1)
tdrStyle.SetHistFillColor(0)
tdrStyle.SetHistLineColor(1)
tdrStyle.SetHistLineStyle(0)
tdrStyle.SetHistLineWidth(1)
tdrStyle.SetErrorX(0.)
tdrStyle.SetMarkerStyle(20)
tdrStyle.SetOptFit(1)
tdrStyle.SetFitFormat("5.4g")
tdrStyle.SetFuncColor(2)
tdrStyle.SetFuncStyle(1)
tdrStyle.SetFuncWidth(1)
tdrStyle.SetOptDate(0)
tdrStyle.SetOptFile(0)
#tdrStyle.SetOptStat(0111)
tdrStyle.SetOptStat(0)
tdrStyle.SetStatColor(kWhite)
tdrStyle.SetStatFont(42)
tdrStyle.SetStatFontSize(0.025)
tdrStyle.SetStatTextColor(1)
tdrStyle.SetStatFormat("6.4g")
tdrStyle.SetStatBorderSize(0)
tdrStyle.SetStatH(0.1)
tdrStyle.SetStatW(0.15)
tdrStyle.SetPadTopMargin(0.05)
tdrStyle.SetPadBottomMargin(0.1)
tdrStyle.SetPadLeftMargin(0.1)
tdrStyle.SetPadRightMargin(0.05)
tdrStyle.SetTitleFont(42)
tdrStyle.SetTitleColor(1)
tdrStyle.SetTitleTextColor(1)
tdrStyle.SetTitleFillColor(10)
tdrStyle.SetTitleFontSize(0.05)
tdrStyle.SetTitleColor(1, "XYZ")
tdrStyle.SetTitleFont(42, "XYZ")
tdrStyle.SetTitleSize(0.06, "XYZ")
tdrStyle.SetTitleXOffset(0.9)
tdrStyle.SetTitleYOffset(1.05)
tdrStyle.SetLabelColor(1, "XYZ")
tdrStyle.SetLabelFont(42, "XYZ")
tdrStyle.SetLabelOffset(0.007, "XYZ")
tdrStyle.SetLabelSize(0.05, "XYZ")
tdrStyle.SetAxisColor(1, "XYZ")
tdrStyle.SetStripDecimals(1)
tdrStyle.SetTickLength(0.03, "XYZ")
tdrStyle.SetNdivisions(505, "XYZ")
tdrStyle.SetPadTickX(1)
tdrStyle.SetPadTickY(1)
tdrStyle.SetOptLogx(0)
tdrStyle.SetOptLogy(0)
tdrStyle.SetOptLogz(0)
tdrStyle.cd()
def setE5Style():
style = TStyle("E5Style","Standard E5 plots style")
font_id = 132
basic_text_size = 0.06
basic_fg_colour = 1
basic_bg_colour= 0
basic_line_width = 2.00
style.SetTextFont(font_id)
style.SetTextSize(basic_text_size)
style.SetFrameBorderMode(0)
style.SetFrameFillColor(basic_bg_colour)
style.SetFrameLineWidth( int(basic_line_width) )
style.SetPaperSize(20,26)
style.SetCanvasBorderMode(0)
style.SetCanvasColor(basic_bg_colour)
style.SetOptStat(0)
style.SetOptTitle(0)
style.SetOptFit(0)
style.SetOptDate(0)
style.SetPadBorderMode(0)
style.SetPadColor(basic_bg_colour)
style.SetPadTopMargin(0.07)
style.SetPadRightMargin(0.10101010101010101010)
style.SetPadBottomMargin(0.16)
style.SetPadLeftMargin(0.18)
style.SetPadTickX(1)
style.SetPadTickY(1)
style.SetTickLength(0.03,"x")
style.SetTickLength(0.03,"y")
style.SetTickLength(0.03,"z")
style.SetPadGridX(False)
style.SetPadGridY(False)
style.SetGridWidth(int(basic_line_width) )
style.SetGridColor(basic_fg_colour)
style.SetTitleStyle(0)
style.SetTitleBorderSize(0)
style.SetTitleTextColor(basic_fg_colour)
style.SetTitleFillColor(basic_bg_colour)
style.SetTitleFont(font_id,"title")
style.SetTitleFont(font_id,"x")
style.SetTitleFont(font_id,"y")
style.SetTitleFont(font_id,"z")
style.SetTitleSize(1.2*basic_text_size,"x")
style.SetTitleSize(1.2*basic_text_size,"y")
style.SetTitleSize(1.2*basic_text_size,"z")
style.SetTitleOffset(0.95,"x")
style.SetTitleOffset(1.20,"y")
style.SetTitleOffset(1.20,"z")
style.SetTitleX(0.00)
style.SetTitleY(1.00)
style.SetTitleW(1.00)
style.SetTitleH(0.05)
style.SetLabelFont(font_id,"x")
style.SetLabelFont(font_id,"y")
style.SetLabelFont(font_id,"z")
style.SetLabelSize(basic_text_size,"x")
style.SetLabelSize(basic_text_size,"y")
style.SetLabelSize(basic_text_size,"z")
style.SetLabelOffset(0.010,"X")
style.SetLabelOffset(0.005,"Y")
style.SetStatColor(0)
style.SetStatBorderSize(0)
style.SetStatFont(font_id)
style.SetStatFontSize(0.05)
style.SetStatFormat("6.3g")
style.SetStatX(0.9)
style.SetStatY(0.9)
style.SetStatW(0.25)
style.SetStatH(0.15)
style.SetLegendBorderSize(0)
style.SetFillColor(1)
style.SetFillStyle(1001)
style.SetPalette(1)
style.SetHistLineColor(basic_fg_colour)
style.SetHistLineWidth(int(basic_line_width) )
style.SetHistLineStyle(0)
#style.SetHistFillColor(basic_bg_colour)
#style.SetHistFillStyle(0)
style.SetHistMinimumZero(False)
style.SetHistTopMargin(0.05)
style.SetNdivisions(505,"x")
style.SetNdivisions(510,"y")
style.SetMarkerStyle(20)
style.SetMarkerSize(1.0)
style.SetMarkerColor(basic_fg_colour)
style.SetErrorX(0.)
style.SetEndErrorSize(2)
style.SetFuncColor(basic_fg_colour)
style.SetFuncWidth(int(basic_line_width) )
gROOT.SetStyle("E5Style")
gROOT.ForceStyle()
return style
def setTDRStyle():
tdrStyle = TStyle("tdrStyle", "Style for P-TDR")
# For the canvas:
tdrStyle.SetCanvasBorderMode(0)
tdrStyle.SetCanvasColor(kWhite)
tdrStyle.SetCanvasDefH(600) # height of canvas
tdrStyle.SetCanvasDefW(600) # width of canvas
tdrStyle.SetCanvasDefX(0) # position on screen
tdrStyle.SetCanvasDefY(0)
tdrStyle.SetPadBorderMode(0)
#tdrStyle.SetPadBorderSize(Width_t size = 1)
tdrStyle.SetPadColor(kWhite)
tdrStyle.SetPadGridX(False)
tdrStyle.SetPadGridY(False)
tdrStyle.SetGridColor(0)
tdrStyle.SetGridStyle(3)
tdrStyle.SetGridWidth(1)
# For the frame:
tdrStyle.SetFrameBorderMode(0)
tdrStyle.SetFrameBorderSize(1)
tdrStyle.SetFrameFillColor(0)
tdrStyle.SetFrameFillStyle(0)
tdrStyle.SetFrameLineColor(1)
tdrStyle.SetFrameLineStyle(1)
tdrStyle.SetFrameLineWidth(1)
# For the histo:
#tdrStyle.SetHistFillColor(1)
#tdrStyle.SetHistFillStyle(0)
tdrStyle.SetHistLineColor(1)
tdrStyle.SetHistLineStyle(1)
tdrStyle.SetHistLineWidth(1)
#tdrStyle.SetLegoInnerR(Float_t rad = 0.5)
#tdrStyle.SetNumberContours(Int_t number = 20)
tdrStyle.SetEndErrorSize(2)
#tdrStyle.SetErrorMarker(20)
tdrStyle.SetErrorX(0)
tdrStyle.SetMarkerStyle(20)
# For the fit/function:
tdrStyle.SetOptFit(1)
tdrStyle.SetFitFormat("5.4g")
tdrStyle.SetFuncColor(2)
tdrStyle.SetFuncStyle(1)
tdrStyle.SetFuncWidth(1)
# For the date:
tdrStyle.SetOptDate(0)
#tdrStyle.SetDateX(Float_t x = 0.01)
#tdrStyle.SetDateY(Float_t y = 0.01)
# For the statistics box:
tdrStyle.SetOptFile(0)
tdrStyle.SetOptStat(0) # to display the mean and RMS: SetOptStat("mr")
tdrStyle.SetStatColor(kWhite)
tdrStyle.SetStatFont(42)
tdrStyle.SetStatFontSize(0.025)
tdrStyle.SetStatTextColor(1)
tdrStyle.SetStatFormat("6.4g")
tdrStyle.SetStatBorderSize(1)
tdrStyle.SetStatH(0.1)
tdrStyle.SetStatW(0.15)
#tdrStyle.SetStatStyle(Style_t style = 1001)
#tdrStyle.SetStatX(Float_t x = 0)
#tdrStyle.SetStatY(Float_t y = 0)
# For pad margins:
tdrStyle.SetPadTopMargin(0.05)
tdrStyle.SetPadBottomMargin(0.13)
tdrStyle.SetPadLeftMargin(0.16)
tdrStyle.SetPadRightMargin(0.02)
# For the Global title:
tdrStyle.SetOptTitle(0)
tdrStyle.SetTitleFont(42)
tdrStyle.SetTitleColor(1)
tdrStyle.SetTitleTextColor(1)
tdrStyle.SetTitleFillColor(10)
tdrStyle.SetTitleFontSize(0.05)
#tdrStyle.SetTitleH(0) # set the height of the title box
#tdrStyle.SetTitleW(0) # set the width of the title box
#tdrStyle.SetTitleX(0) # set the position of the title box
#tdrStyle.SetTitleY(0.985) # set the position of the title box
#tdrStyle.SetTitleStyle(Style_t style = 1001)
#tdrStyle.SetTitleBorderSize(2)
# For the axis titles:
tdrStyle.SetTitleColor(1, 'XYZ')
tdrStyle.SetTitleFont(42, 'XYZ')
tdrStyle.SetTitleSize(0.06, 'XYZ')
# tdrStyle.SetTitleXSize(Float_t size = 0.02) # another way to set the size?
# tdrStyle.SetTitleYSize(Float_t size = 0.02)
tdrStyle.SetTitleXOffset(0.9)
tdrStyle.SetTitleYOffset(1.25)
# tdrStyle.SetTitleOffset(1.1, 'Y') # another way to set the Offset
# For the axis labels:
tdrStyle.SetLabelColor(1, 'XYZ')
tdrStyle.SetLabelFont(42, 'XYZ')
tdrStyle.SetLabelOffset(0.007, 'XYZ')
tdrStyle.SetLabelSize(0.05, 'XYZ')
# For the axis:
tdrStyle.SetAxisColor(1, 'XYZ')
tdrStyle.SetStripDecimals(True)
tdrStyle.SetTickLength(0.03, 'XYZ')
tdrStyle.SetNdivisions(510, 'XYZ')
tdrStyle.SetPadTickX(
1) # to get tick marks on the opposite side of the frame
tdrStyle.SetPadTickY(1)
# Change for log plots:
tdrStyle.SetOptLogx(0)
tdrStyle.SetOptLogy(0)
tdrStyle.SetOptLogz(0)
# Postscript options:
tdrStyle.SetPaperSize(20., 20.)
#tdrStyle.SetLineScalePS(Float_t scale = 3)
#tdrStyle.SetLineStyleString(Int_t i, const char* text)
#tdrStyle.SetHeaderPS(const char* header)
#tdrStyle.SetTitlePS(const char* pstitle)
#tdrStyle.SetBarOffset(Float_t baroff = 0.5)
#tdrStyle.SetBarWidth(Float_t barwidth = 0.5)
#tdrStyle.SetPaintTextFormat(const char* format = "g")
#tdrStyle.SetPalette(Int_t ncolors = 0, Int_t* colors = 0)
#tdrStyle.SetTimeOffset(Double_t toffset)
#tdrStyle.SetHistMinimumZero(True)
tdrStyle.SetHatchesLineWidth(5)
tdrStyle.SetHatchesSpacing(0.05)
tdrStyle.cd()
from ROOT import TStyle
atlasStyle = TStyle("atlasStyle", "atlasStyle")
atlasStyle.SetOptStat(0)
iCol = 0
atlasStyle.SetFrameBorderMode(iCol)
atlasStyle.SetFrameFillColor(iCol)
atlasStyle.SetCanvasBorderMode(iCol)
atlasStyle.SetCanvasColor(iCol)
atlasStyle.SetPadBorderMode(iCol)
atlasStyle.SetPadColor(iCol)
atlasStyle.SetStatColor(iCol)
atlasStyle.SetPaperSize(20, 26)
atlasStyle.SetPadTopMargin(0.05)
atlasStyle.SetPadRightMargin(0.05)
atlasStyle.SetPadBottomMargin(0.16)
atlasStyle.SetPadLeftMargin(0.16)
atlasStyle.SetTitleXOffset(1.4)
atlasStyle.SetTitleYOffset(1.4)
font = 42
tSize = 0.05
atlasStyle.SetTextFont(font)
atlasStyle.SetTextSize(tSize)
atlasStyle.SetLabelFont(font, "x")
atlasStyle.SetTitleFont(font, "x")
atlasStyle.SetLabelFont(font, "y")
atlasStyle.SetTitleFont(font, "y")
atlasStyle.SetLabelFont(font, "z")
atlasStyle.SetTitleFont(font, "z")
atlasStyle.SetLabelSize(tSize, "x")
from ROOT import gROOT,gPad,gStyle,TCanvas,TFile,TLine,TLatex,TAxis,TLegend,TPostScript
from ROOT import TH2D,TArrow,TCut,TPad,TPaveText,TGraph,TGraphAsymmErrors
from ROOT import TGraph2D,TStyle,TBranch,gSystem,gDirectory
from ROOT import TPave,TPaveStats
atlasStyle= TStyle("ATLAS","Atlas style")
# use plain black on white colors
icol=0
atlasStyle.SetFrameBorderMode(icol)
atlasStyle.SetCanvasBorderMode(icol)
atlasStyle.SetPadBorderMode(icol)
atlasStyle.SetPadColor(icol)
atlasStyle.SetCanvasColor(icol)
atlasStyle.SetStatColor(icol)
#atlasStyle.SetFillColor(icol)
# set the paper & margin sizes
atlasStyle.SetPaperSize(20,26)
atlasStyle.SetPadTopMargin(0.1)
atlasStyle.SetPadRightMargin(0.05)
atlasStyle.SetPadBottomMargin(0.16)
atlasStyle.SetPadLeftMargin(0.12)
# use large fonts
font=42
tsize=0.05
atlasStyle.SetTextFont(font)
def setLHCbStyle():
global lhcbStyle
lhcbFont = 132
lhcbTSize = 0.06
lhcbWidth = 2
lhcbStyle = TStyle("lhcbStyle", "LHCb plots style")
lhcbStyle.SetFillColor(1)
lhcbStyle.SetFillStyle(1001) # solid
lhcbStyle.SetFrameFillColor(0)
lhcbStyle.SetFrameBorderMode(0)
lhcbStyle.SetPadBorderMode(0)
lhcbStyle.SetPadColor(0)
lhcbStyle.SetCanvasBorderMode(0)
lhcbStyle.SetCanvasColor(0)
lhcbStyle.SetStatColor(0)
lhcbStyle.SetLegendBorderSize(0)
lhcbStyle.SetLegendFont(132)
# use large fonts
lhcbStyle.SetTextFont(lhcbFont)
lhcbStyle.SetTitleFont(lhcbFont)
lhcbStyle.SetTextSize(lhcbTSize)
lhcbStyle.SetLabelFont(lhcbFont, "x")
lhcbStyle.SetLabelFont(lhcbFont, "y")
lhcbStyle.SetLabelFont(lhcbFont, "z")
lhcbStyle.SetLabelSize(lhcbTSize, "x")
lhcbStyle.SetLabelSize(lhcbTSize, "y")
lhcbStyle.SetLabelSize(lhcbTSize, "z")
lhcbStyle.SetTitleFont(lhcbFont)
lhcbStyle.SetTitleFont(lhcbFont, "x")
lhcbStyle.SetTitleFont(lhcbFont, "y")
lhcbStyle.SetTitleFont(lhcbFont, "z")
lhcbStyle.SetTitleSize(1.2 * lhcbTSize, "x")
lhcbStyle.SetTitleSize(1.2 * lhcbTSize, "y")
lhcbStyle.SetTitleSize(1.2 * lhcbTSize, "z")
# set the paper & margin sizes
lhcbStyle.SetPaperSize(20, 26)
lhcbStyle.SetPadTopMargin(0.05)
lhcbStyle.SetPadRightMargin(0.05) # increase for colz plots
lhcbStyle.SetPadBottomMargin(0.16)
lhcbStyle.SetPadLeftMargin(0.14)
# use medium bold lines and thick markers
lhcbStyle.SetLineWidth(lhcbWidth)
lhcbStyle.SetFrameLineWidth(lhcbWidth)
lhcbStyle.SetHistLineWidth(lhcbWidth)
lhcbStyle.SetFuncWidth(lhcbWidth)
lhcbStyle.SetGridWidth(lhcbWidth)
lhcbStyle.SetLineStyleString(2, "[12 12]")
# postscript dashes
lhcbStyle.SetMarkerStyle(20)
lhcbStyle.SetMarkerSize(1.0)
# label offsets
lhcbStyle.SetLabelOffset(0.010, "X")
lhcbStyle.SetLabelOffset(0.010, "Y")
# by default, do not display histogram decorations:
lhcbStyle.SetOptStat(0)
#lhcbStyle.SetOptStat("emr") # show only nent -e , mean - m , rms -r
# full opts at http:#root.cern.ch/root/html/TStyle.html#TStyle:SetOptStat
lhcbStyle.SetStatFormat("6.3g") # specified as c printf options
lhcbStyle.SetOptTitle(0)
lhcbStyle.SetOptFit(0)
#lhcbStyle.SetOptFit(1011) # order is probability, Chi2, errors, parameters
#titles
lhcbStyle.SetTitleOffset(0.85, "X")
lhcbStyle.SetTitleOffset(0.85, "Y")
lhcbStyle.SetTitleOffset(1.2, "Z")
lhcbStyle.SetTitleFillColor(0)
lhcbStyle.SetTitleStyle(0)
lhcbStyle.SetTitleBorderSize(0)
lhcbStyle.SetTitleFont(lhcbFont, "title")
lhcbStyle.SetTitleX(0.0)
lhcbStyle.SetTitleY(1.0)
lhcbStyle.SetTitleW(1.0)
lhcbStyle.SetTitleH(0.05)
# look of the statistics box:
lhcbStyle.SetStatBorderSize(0)
lhcbStyle.SetStatFont(lhcbFont)
lhcbStyle.SetStatFontSize(0.05)
lhcbStyle.SetStatX(0.9)
lhcbStyle.SetStatY(0.9)
lhcbStyle.SetStatW(0.25)
lhcbStyle.SetStatH(0.15)
# put tick marks on top and RHS of plots
lhcbStyle.SetPadTickX(1)
lhcbStyle.SetPadTickY(1)
# histogram divisions: only 5 in x to avoid label overlaps
lhcbStyle.SetNdivisions(505, "x")
lhcbStyle.SetNdivisions(510, "y")
gROOT.SetStyle("lhcbStyle")
return
