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