def plot_cov_output(args):
# Covariance, fractional covariance and correlation
covfile = TFile(args.covfile)
covcanvas = TCanvas()
gStyle = TStyle()
gStyle.SetPadLeftMargin(0.25)
gStyle.SetPadRightMargin(0.15)
gStyle.SetPalette(56)
for matname in ('cov', 'fcov', 'corr'):
mat = covfile.Get(matname)
mat.GetYaxis().LabelsOption('v') # doesn't work...
mat.GetYaxis().SetLabelSize(0.07)
mat.GetXaxis().SetLabelSize(0.07)
mat.SetTitleSize(0.3, 't') # doesn't work...
if matname == 'corr':
mat.GetZaxis().SetRangeUser(-0.4, 1)
mat.SetTitle("Flux Correlation Matrix")
mat.Draw("colz")
mat.SetStats(False)
covcanvas.SetLeftMargin(0.25)
covcanvas.Update()
covcanvas.SaveAs(args.outdir + matname + "_plot.pdf")
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 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 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()
from ROOT import TStyle, TColor, kWhite, kFullCircle, kTRUE
tdrStyle = TStyle("tdrStyle", "Style for P-TDR")
def setTDRStyle():
global tdrStyle
## Canvas
##----------------------------------------------------------------------------
tdrStyle.SetCanvasBorderMode(0)
tdrStyle.SetCanvasColor(kWhite)
tdrStyle.SetCanvasDefH(500)
tdrStyle.SetCanvasDefW(700)
tdrStyle.SetCanvasDefX(0)
tdrStyle.SetCanvasDefY(0)
## Pad
##----------------------------------------------------------------------------
tdrStyle.SetPadBorderMode(0)
tdrStyle.SetPadColor(kWhite)
tdrStyle.SetPadGridX(False)
tdrStyle.SetPadGridY(False)
tdrStyle.SetGridColor(0)
tdrStyle.SetGridStyle(3)
tdrStyle.SetGridWidth(1)
## Frame
##----------------------------------------------------------------------------
tdrStyle.SetFrameBorderMode(0)
#coding=utf-8
from __future__ import division
from ROOT import TH1F, TStyle, TSpectrum, TCanvas
style = TStyle("style", "style")
style.SetCanvasColor(0)
style.SetMarkerStyle(5)
style.SetTitleFillColor(0)
style.cd()
class TrovaPicchi:
def __init__(self, nomefile, sigma):
n, nstep, step = [int(i) for i in nomefile.split(".")[0].split("_")]
self.nfenditure = n
self.canvas = TCanvas("canvas{0}".format(n),
"canvas {0} fenditure".format(n))
self.isto = TH1F("isto{0}".format(n),
"{0} fenditure;bin ;-log(intensity)".format(n), nstep,
-nstep / 2 * step / 100, (nstep / 2 + 1) * step / 100)
self.isto.SetStats(0)
self.file2isto(nomefile)
self.spectrum = TSpectrum()
self.spectrum.Search(self.isto, sigma, "nobackground", 0.01)
self.peaks = [
self.spectrum.GetPositionX()[i]
for i in range(self.spectrum.GetNPeaks())
]
def file2isto(self, nomefile):
file = open(nomefile)
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 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()
#!/usr/bin/python
# -*- coding: utf-8 -*-
from ROOT import TStyle
atlasStyle = TStyle("myStyle", "myStyle")
atlasStyle.SetOptStat(0)
atlasStyle.SetPalette(1)
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.SetTitleFillColor(icol)
#atlasStyle.SetPaperSize(20,26);
#atlasStyle.SetPadTopMargin(0.02);
#atlasStyle.SetPadRightMargin(0.10);
#atlasStyle.SetPadBottomMargin(0.10);
#atlasStyle.SetPadLeftMargin(0.14);
atlasStyle.SetPaperSize(20, 26)
atlasStyle.SetPadTopMargin(0.05)
atlasStyle.SetPadRightMargin(0.05)
atlasStyle.SetPadBottomMargin(0.16)
def compare_w_proposal(args):
chi2file = TFile(args.chifile)
gStyle = TStyle()
gStyle.SetPadLeftMargin(0.15)
gStyle.SetPadRightMargin(0.15)
colours = [30, 38, 46]
contours = [
chi2file.Get('90pct'),
chi2file.Get('3sigma'),
chi2file.Get('5sigma')
]
propcontours = []
contournames = ['90pct', '3s', '5s']
contourtitles = [
'90% Confidence Level', '3#sigma Confidence Level',
'5#sigma Confidence Level'
]
gr_range = TGraph()
gr_range.SetPoint(0, 0.001, 0.01)
gr_range.SetPoint(1, 1, 100)
gr_range.SetMarkerColor(0)
bestfit = TGraph()
bestfit.SetPoint(0, 0.062, 1.7)
bestfit.SetMarkerStyle(29)
bestfit.SetMarkerSize(1.6)
bestfit.SetMarkerColor(40)
print("contours has length " + str(len(contours)))
for i in range(len(contours)):
x = []
y = []
with open(args.compdir + 'numu' + contournames[i] + '.txt') as f:
for l, line in enumerate(f):
if l == 0: continue
x.append(float(line.split(', ')[0]))
y.append(float(line.split(', ')[1].replace("\n", "")))
propcontours.append(TGraph())
for j in range(len(x)):
propcontours[i].SetPoint(j, x[j], y[j])
tempcanvas = TCanvas('temp_canvas', '', 1020, 990)
templegend = TLegend()
templegend.AddEntry(contours[i], 'Our contour', 'l')
templegend.AddEntry(propcontours[i], 'From proposal', 'l')
templegend.AddEntry(bestfit, 'Best Fit Point', 'p')
tempcanvas.SetLogy()
tempcanvas.SetLogx()
gr_range.SetTitle(
contourtitles[i] +
' Comparison; sin^{2}(2#theta); #Delta m^{2} (eV^{2})')
gr_range.Draw('AP')
gr_range.GetXaxis().SetRangeUser(0.001, 1)
gr_range.GetYaxis().SetRangeUser(0.01, 100)
for lst in (contours, propcontours):
lst[i].SetMarkerStyle(20)
lst[i].SetMarkerSize(0.25)
lst[i].SetMarkerColor(colours[i] if lst == contours else 1)
lst[i].SetLineColor(colours[i] if lst == contours else 1)
contours[i].Draw('P same')
propcontours[i].Draw('P same')
templegend.Draw()
bestfit.Draw('P same')
tempcanvas.SaveAs(args.outdir + contournames[i] + '_comparison.pdf')
def plot_chi2_output(args):
# Chi squareds
chi2file = TFile(args.chifile)
gStyle = TStyle()
chi2 = chi2file.Get('chisq')
chi2canvas = TCanvas()
chi2.SetTitle(
'#chi^{2}; log_{10}(sin^{2}(2#theta)); log_{10}(#Delta m^{2}); #chi^{2}'
)
gStyle.SetPalette(1)
chi2.Draw('surf1')
chi2canvas.SaveAs(args.outdir + "chisq.pdf")
# Contours
contcanvas = TCanvas('cont_canvas', '', 1020, 990)
gStyle.SetPadLeftMargin(0.15)
gStyle.SetPadRightMargin(0.15)
colours = [30, 38, 46]
contours = [
chi2file.Get('90pct'),
chi2file.Get('3sigma'),
chi2file.Get('5sigma')
]
for g in range(len(contours)):
contours[g].SetMarkerStyle(20)
contours[g].SetMarkerSize(0.25)
contours[g].SetMarkerColor(colours[g])
contours[g].SetLineColor(colours[g])
gr_range = TGraph()
gr_range.SetPoint(0, 0.001, 0.01)
gr_range.SetPoint(1, 1, 100)
gr_range.SetMarkerColor(0)
bestfit = TGraph()
bestfit.SetPoint(0, 0.062, 1.7)
bestfit.SetMarkerStyle(29)
bestfit.SetMarkerSize(1.6)
bestfit.SetMarkerColor(40)
gr_range.SetTitle(
'SBN Sensitivity; sin^{2}(2#theta); #Delta m^{2} (eV^{2})')
legend = TLegend()
legend.AddEntry(contours[0], '90% CL', 'l')
legend.AddEntry(contours[1], '3#sigma CL', 'l')
legend.AddEntry(contours[2], '5#sigma CL', 'l')
legend.AddEntry(bestfit, 'Best Fit Point', 'p')
contcanvas.SetLogy()
contcanvas.SetLogx()
gr_range.Draw('AP')
gr_range.GetXaxis().SetRangeUser(0.001, 1)
gr_range.GetYaxis().SetRangeUser(0.01, 100)
contours[0].Draw('P same')
contours[1].Draw('P same')
contours[2].Draw('P same')
legend.Draw()
bestfit.Draw('P same')
contcanvas.SetLeftMargin(0.15)
contcanvas.Update()
contcanvas.SaveAs(args.outdir + 'Sensitivity.pdf')
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)
import sys, os, math, argparse
from ROOT import gROOT, gStyle, TStyle
tdrStyle = TStyle("tdrStyle","Style for P-TDR")
kWhite = 0
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)
#!/usr/bin/python
# -*- coding: utf-8 -*-
from ROOT import TStyle
atlasStyle = TStyle("myStyle","myStyle");
atlasStyle.SetOptStat(0)
atlasStyle.SetPalette(1)
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.SetTitleFillColor(icol);
#atlasStyle.SetPaperSize(20,26);
#atlasStyle.SetPadTopMargin(0.02);
#atlasStyle.SetPadRightMargin(0.10);
#atlasStyle.SetPadBottomMargin(0.10);
#atlasStyle.SetPadLeftMargin(0.14);
atlasStyle.SetPaperSize(20,26);
atlasStyle.SetPadTopMargin(0.05);
atlasStyle.SetPadRightMargin(0.05);
atlasStyle.SetPadBottomMargin(0.16);
atlasStyle.SetPadLeftMargin(0.16);
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():
"""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 setTdrStyle():
myStyle = TStyle('myStyle','a better style for my plot')
# For the canvas:
myStyle.SetCanvasBorderMode(0);
myStyle.SetCanvasColor(0);
myStyle.SetCanvasDefH(600); #Height of canvas
myStyle.SetCanvasDefW(600); #Width of canvas
myStyle.SetCanvasDefX(0); #POsition on screen
myStyle.SetCanvasDefY(0);
# For the Pad:
myStyle.SetPadBorderMode(0);
myStyle.SetPadColor(0);
myStyle.SetPadGridX(False);
myStyle.SetPadGridY(False);
myStyle.SetGridColor(0);
myStyle.SetGridStyle(3);
myStyle.SetGridWidth(1);
# For the frame:
myStyle.SetFrameBorderMode(0);
myStyle.SetFrameBorderSize(1);
myStyle.SetFrameFillColor(0);
myStyle.SetFrameFillStyle(0);
myStyle.SetFrameLineColor(1);
myStyle.SetFrameLineStyle(1);
myStyle.SetFrameLineWidth(1);
# For the histo:
myStyle.SetHistLineColor(1);
myStyle.SetHistLineStyle(0);
myStyle.SetHistLineWidth(1);
myStyle.SetEndErrorSize(2);
#For the fit/function:
myStyle.SetOptFit(1);
myStyle.SetFitFormat("5.4g");
myStyle.SetFuncColor(2);
myStyle.SetFuncStyle(1);
myStyle.SetFuncWidth(1);
#For the date:
myStyle.SetOptDate(0);
# For the statistics box:
myStyle.SetOptFile(0);
myStyle.SetOptStat(0); # To display the mean and RMS: SetOptStat("mr");
myStyle.SetStatColor(0);
myStyle.SetStatFont(40);
myStyle.SetStatFontSize(0.025);
myStyle.SetStatTextColor(1);
myStyle.SetStatFormat("6.4g");
myStyle.SetStatBorderSize(1);
myStyle.SetStatH(0.1);
myStyle.SetStatW(0.15);
# Margins:
myStyle.SetPadTopMargin(0.07);
myStyle.SetPadBottomMargin(0.15);
myStyle.SetPadLeftMargin(0.2);
myStyle.SetPadRightMargin(0.05);
# For the Global title:
myStyle.SetOptTitle(0);
myStyle.SetTitleColor(1);
myStyle.SetTitleTextColor(1);
myStyle.SetTitleFillColor(10);
myStyle.SetTitleFontSize(0.05);
# For the axis titles:
myStyle.SetTitleColor(1, "XYZ");
#myStyle.SetTitleFont(40, "XYZ");
myStyle.SetTitleSize(0.055, "XYZ");
# For the axis labels:
myStyle.SetLabelColor(1, "XYZ");
myStyle.SetLabelFont(40, "XYZ");
myStyle.SetLabelOffset(0.007, "XYZ");
myStyle.SetLabelSize(0.05, "XYZ");
# For the axis:
myStyle.SetAxisColor(1, "XYZ");
myStyle.SetStripDecimals(True);
myStyle.SetTickLength(0.03, "XYZ");
myStyle.SetNdivisions(510, "XYZ");
myStyle.SetPadTickX(1); # To get tick marks on the opposite side of the frame
myStyle.SetPadTickY(1);
# Change for log plots:
myStyle.SetOptLogx(0);
myStyle.SetOptLogy(0);
myStyle.SetOptLogz(0);
# Postscript options:
myStyle.SetPaperSize(20.,20.);
myStyle.cd()
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(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.SetHistLineColor(1)
tdrStyle.SetHistLineStyle(0)
tdrStyle.SetHistLineWidth(1)
tdrStyle.SetEndErrorSize(2)
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)
# 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.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)
# For the axis titles:
tdrStyle.SetTitleColor(1, "XYZ")
tdrStyle.SetTitleFont(42, "XYZ")
tdrStyle.SetTitleSize(0.06, "XYZ")
tdrStyle.SetTitleXOffset(0.9)
tdrStyle.SetTitleYOffset(1.25)
# 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(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.cd()
# SpartyJet is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# 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)
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()
#!/usr/bin/env python
#coding=utf-8
from ROOT import TStyle
# style:
ele_style = TStyle("eleStyle", "Style for electron validation")
ele_style.SetCanvasBorderMode(0)
ele_style.SetCanvasColor(0)
#ele_style.SetFillColor(0)
ele_style.SetCanvasDefH(768)
ele_style.SetCanvasDefW(1024)
ele_style.SetCanvasDefX(0)
ele_style.SetCanvasDefY(0)
ele_style.SetPadBorderMode(0)
ele_style.SetPadColor(0)
ele_style.SetPadGridX(False)
ele_style.SetPadGridY(False)
ele_style.SetGridColor(0)
ele_style.SetGridStyle(3)
ele_style.SetGridWidth(1)
ele_style.SetOptStat(1)
ele_style.SetPadTickX(0)
ele_style.SetPadTickY(0)
ele_style.SetHistLineColor(1)
ele_style.SetHistLineStyle(0)
ele_style.SetHistLineWidth(2)
ele_style.SetEndErrorSize(2)
ele_style.SetErrorX(0)
ele_style.SetOptStat(1)
ele_style.SetStatColor(0)
#!/usr/bin/env python
from ROOT import gROOT , TCanvas, TPad, gStyle,TFile, gPad, TLegend, TMath, TH1F, TStyle,TColor
import ROOT as ro
from array import array
#from env import *
from math import sqrt, fabs
tdrStyle = TStyle("tdrStyle","Style for P-TDR");
tdrStyle2 = TStyle("tdrStyle2","Style for P-TDR");
def xfrange(start, stop, step):
while start < stop:
yield start
start += step
def setTDRStyle(logy):
global tdrStyle
#ro.gROOT.Macro( "setTDRStyle.C")
# For the canvas:
tdrStyle.SetCanvasBorderMode(0);
tdrStyle.SetCanvasColor(ro.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);
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 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 looks_atlas():
atlasStyle = TStyle("ATLAS","Atlas style");
# use plain black on white colors
icol=int(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);
atlasStyle.SetTitleYOffset(1);
# use large fonts
#Int_t font=72; # Helvetica italics
font=int(42); # Helvetica
tsize=int(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.cd();
gROOT.ForceStyle();
gStyle.ls();
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 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();
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()
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 setTDRStyle():
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(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.0, 20.0)
# 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 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 SetAtlasStyle():
from ROOT import TROOT, gROOT, TStyle, gStyle
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.04)
atlasStyle.SetPadRightMargin(0.04)
atlasStyle.SetPadBottomMargin(0.13)
atlasStyle.SetPadLeftMargin(0.13)
atlasStyle.SetTitleXOffset(1.0)
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(1)
atlasStyle.SetMarkerSize(1)
atlasStyle.SetHistLineWidth(2)
atlasStyle.SetLineStyleString(2,"[12 12]")
atlasStyle.SetEndErrorSize(2)
atlasStyle.SetOptTitle(0)
atlasStyle.SetOptStat(0)
atlasStyle.SetOptFit(0)
atlasStyle.SetPadTickX(0)
atlasStyle.SetPadTickY(0)
gROOT.SetStyle("ATLAS")
gROOT.ForceStyle()
gStyle.SetPalette(1)
#!/usr/bin/env python
#coding=utf-8
from ROOT import TStyle
# style:
style = TStyle("MyStyle", "style for my ROOT graphs and histograms")
style.SetCanvasBorderMode(0)
style.SetCanvasColor(0)
#style.SetFillColor(0)
style.SetCanvasDefH(768)
style.SetCanvasDefW(1024)
style.SetCanvasDefX(0)
style.SetCanvasDefY(0)
style.SetPadBorderMode(0)
style.SetPadColor(0)
style.SetPadGridX(False)
style.SetPadGridY(False)
style.SetGridColor(0)
style.SetGridStyle(3)
style.SetGridWidth(1)
style.SetOptStat(1)
style.SetPadTickX(0)
style.SetPadTickY(0)
style.SetHistLineColor(1)
style.SetHistLineStyle(0)
style.SetHistLineWidth(2)
style.SetEndErrorSize(2)
style.SetErrorX(0)
style.SetOptStat(111111)
style.SetStatColor(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)
#//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 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 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 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()
import ROOT
import os
from array import array
from ROOT import TStyle, TLatex, TH1F, TCanvas, TFile, TLegend
myStyle = TStyle()
myStyle.SetLabelSize(0.05, "xyz")
myStyle.SetTitleSize(0.06, "xyz")
myStyle.SetTitleOffset(1.0, "x")
myStyle.SetTitleOffset(1.2, "y")
myStyle.SetPalette(1)
myStyle.SetPadRightMargin(0.14)
myStyle.SetOptStat(0)
t1 = TLatex()
t1.SetNDC()
t1.SetTextAlign(26)
t1.SetTextSize(0.06)
##f1=TFile.Open("histoTPG_251643_eg12.root")
##f2=TFile.Open("histoTPG_000_eg12.root");
f1 = TFile.Open(
"~/eos/cms/store/caf/user/taroni/TPG/histoTPG_254833_eg12.root")
#f2=TFile.Open("~/eos/cms/store/caf/user/taroni/TPG/newhistoTPG_251244_notranscorr_eg12.root")
f2 = TFile.Open(
"~/eos/cms/store/caf/user/taroni/TPG/newhistoTPG_251244_withtranscorr_eg12.root"
)
f3 = TFile.Open("etaphi_ee.root")
scaling = 1.
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()
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
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 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
