ShipStyle.py

Go to the documentation of this file.

from __future__ import print_function
# Global scope variables
from ROOT import *
from ROOT import Double
lhcbstyle = TStyle()     # general lhcb style
lhcbName  = TPaveText()  # standard lhcb text for plot
lhcbLabel = TText  # style for Ttext
lhcbLatex = TLatex #style for TLatex
 
# define names for colours
black=1
red=2
green=3
blue=4
yellow=5 
magenta=6
cyan=7
purple=9
 
def lhcbstyleSetup():
 
 
 
 lhcbstyle=TStyle("lhcbstyle","Standard LHCb plots style")
 
# use helvetica-bold-r-normal, precision 2 (rotatable)
 lhcbFont = 62
# line thickness
 lhcbWidth = int(3.00)
 
# 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)
 
# set the paper & margin sizes
 lhcbstyle.SetPaperSize(20,26)
 lhcbstyle.SetPadTopMargin(0.05)
 lhcbstyle.SetPadRightMargin(0.05) # increase for colz plots
 lhcbstyle.SetPadBottomMargin(0.16)
 lhcbstyle.SetPadLeftMargin(0.14)
 
# use large fonts
 lhcbstyle.SetTextFont(lhcbFont)
 lhcbstyle.SetTextSize(0.08)
 lhcbstyle.SetLabelFont(lhcbFont,"x")
 lhcbstyle.SetLabelFont(lhcbFont,"y")
 lhcbstyle.SetLabelFont(lhcbFont,"z")
 lhcbstyle.SetLabelSize(0.04,"x")
 lhcbstyle.SetLabelSize(0.04,"y")
 lhcbstyle.SetLabelSize(0.04,"z")
 lhcbstyle.SetTitleFont(lhcbFont)
 lhcbstyle.SetTitleSize(0.05,"x")
 lhcbstyle.SetTitleSize(0.05,"y")
 lhcbstyle.SetTitleSize(0.05,"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]") # postscript dashes
 lhcbstyle.SetMarkerStyle(20)
 lhcbstyle.SetMarkerSize(1.5)
 
# label offsets
 lhcbstyle.SetLabelOffset(0.015)
 
# by default, do not display histogram decorations:
 lhcbstyle.SetOptStat(0)  
 lhcbstyle.SetOptStat("emr")  # show only nent -e , mean - m , rms -r
# full opts at http:#root.cern.ch/root/html/TStyle.html#TStyle:SetOptStat
 lhcbstyle.SetStatFormat("6.3g") # specified as c printf options
 lhcbstyle.SetOptTitle(0)
 lhcbstyle.SetOptFit(0)
#lhcbstyle.SetOptFit(1011) # order is probability, Chi2, errors, parameters
 
# look of the statistics box:
 lhcbstyle.SetStatBorderSize(0)
 lhcbstyle.SetStatFont(lhcbFont)
 lhcbstyle.SetStatFontSize(0.05)
 lhcbstyle.SetStatX(0.9)
 lhcbstyle.SetStatY(0.9)
 lhcbstyle.SetStatW(0.25)
 lhcbstyle.SetStatH(0.15)
# put tick marks on top and RHS of plots
 lhcbstyle.SetPadTickX(1)
 lhcbstyle.SetPadTickY(1)
 
# histogram divisions: only 5 in x to avoid label overlaps
 lhcbstyle.SetNdivisions(505,"x")
 lhcbstyle.SetNdivisions(510,"y")
 
 
#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()
 
 
def printLHCb(optLR="L", optPrelim="Final", optText=""):
 
  if optLR=="R" :   
    lhcbName = TPaveText(0.70 - lhcbstyle.GetPadRightMargin(),
                         0.75 - lhcbstyle.SetPadTopMargin(0.05),
                         0.95 - lhcbstyle.GetPadRightMargin(),
                         0.85 - lhcbstyle.SetPadTopMargin(0.05),
                         "BRNDC")
  elif optLR=="L":
    lhcbName = TPaveText(lhcbstyle.GetPadLeftMargin() + 0.05,
                         0.87 - lhcbstyle.GetPadTopMargin(),
                         lhcbstyle.GetPadLeftMargin() + 0.30,
                         0.95 - lhcbstyle.GetPadTopMargin(),
                                        "BRNDC")
  else :
   print("printLHCb: option unknown" , optLR)  
  if optPrelim=="Final":
    lhcbName.AddText("LHCb")
  elif optPrelim=="Prelim":
    lhcbName.AddText("#splitline{LHCb}{#scale[1.0]{Preliminary}}")  
  elif optPrelim=="Other":
    lhcbName.AddText(optText)
  else :
    print("printLHCb: option unknown " , optPrelim)
  lhcbName.SetFillColor(0)
  lhcbName.SetTextAlign(12)
  lhcbName.SetBorderSize(0)
  lhcbName.Draw()