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.1)
tdrStyle.SetPadBottomMargin(0.1)
tdrStyle.SetPadLeftMargin(0.1)
tdrStyle.SetPadRightMargin(0.1)
# 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 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)
# For the Pad:
tdrStyle.SetPadBorderMode(0)
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(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(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.05, "XYZ")
# tdrStyle.SetTitleXSize(0.05) # Another way to set the size?
# tdrStyle.SetTitleYSize(0.05)
tdrStyle.SetTitleXOffset(0.9)
tdrStyle.SetTitleYOffset(1.0)
# 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.04, "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(kTRUE)
tdrStyle.SetHatchesLineWidth(5)
tdrStyle.SetHatchesSpacing(0.05)
tdrStyle.cd()
pass
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 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
#
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.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(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 lhcbStyle
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 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()
import os
import numpy as np
from collections import OrderedDict
from ROOT import TCanvas, TText, TStyle, kBird, kWhite
from matplotlib import pyplot as plt
import matplotlib.gridspec as gridspec
nthreads = int(os.cpu_count() / 2)
# style
my_style = TStyle("Default", "My Root Style")
my_style.SetOptStat(0)
my_style.SetFrameLineWidth(2)
my_style.SetTitleSize(0.1, 'p')
my_style.SetTitleW(0.35)
my_style.SetTitleX(0.6)
my_style.SetTitleAlign(13)
my_style.SetTitleColor(2, 'xyz')
my_style.SetTitleSize(0.08, 'xyz')
my_style.SetTitleOffset(0.5, 'xyz')
my_style.SetTitleStyle(0)
my_style.SetTitleBorderSize(0)
my_style.SetLabelSize(0.06, 'xyz')
my_style.SetPalette(kBird)
my_style.SetFrameFillColor(kWhite)
my_style.SetPadGridX(0)
my_style.SetPadGridY(0)
my_style.SetPadBorderMode(0)
my_style.SetCanvasColor(0)
my_style.SetPadColor(0)
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
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)
tdrStyle.SetOptFit(1)
tdrStyle.SetFitFormat("5.4g")
tdrStyle.SetFuncColor(2)
tdrStyle.SetFuncStyle(1)
tdrStyle.SetFuncWidth(1)
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)
#~ 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])
#~ ROOT.TColor.CreateGradientColorTable(NRGBs,stops,red,green,blue,NCont)
#~ 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.15)
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.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(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():
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
