tcanvasTDR.py

from ROOT import *
import CMS_lumi, tdrstyle


def TDRify(canvas, inset, name, writeOutput=False, outputdir="~/physics/_tests"):
    
    for i in range(0,1):   
      canvas.Draw()
      #canvas=originalcanvas.Clone()
      #print "\n --->TDRifying!"
      tdrstyle.setTDRStyle()
      gStyle.SetOptStat(0)
      CMS_lumi.lumi_13TeV = "35.9 fb^{-1}"
      CMS_lumi.writeExtraText = 1
      CMS_lumi.extraText = "Preliminary"
      CMS_lumi.lumi_sqrtS = "13 TeV" # used with iPeriod = 0, e.g. for simulation-only plots (default is an empty string)
  
      iPos = 11
      if( iPos==0 ): CMS_lumi.relPosX = 0.12
  
      H_ref = 600;
      W_ref = 800;
      W = W_ref
      H  = H_ref
  
      iPeriod = 4
      T = 0.08*H_ref
      B = 0.12*H_ref
      L = 0.12*W_ref
      R = 0.04*W_ref
  
      #tgraph = tfile.Get("")
      #tgraph.Draw()
      #canvas.Draw()
  
  
  
      #canvas.SetFillColor(0)
      #canvas.SetBorderMode(0)
      #canvas.SetFrameFillStyle(0)
      #canvas.SetFrameBorderMode(0)
      canvas.SetLeftMargin( L/W )
      canvas.SetRightMargin( R/W )
      canvas.SetTopMargin( T/H )
      canvas.SetBottomMargin( B/H )
      canvas.SetTickx(0)
      canvas.SetTicky(0)
  
      if not inset:
          CMS_lumi.CMS_lumi(canvas, iPeriod, iPos)
      gStyle.SetOptStat(0)
      for primitive in canvas.GetListOfPrimitives():
          #print primitive
          if primitive.GetName() in ["data"]:
              primitive.SetStats(kFALSE)
      canvas.cd()
      #print "cd'd to canvas"
      canvas.Update()
      #print "updated canvas"
      canvas.RedrawAxis()
      #print "redrew axis"
  
      if writeOutput:
        canvas.Print("%s/%s_plot.pdf"%(outputdir, name))
        outputFilename = "%s/%s_plot.root"%(outputdir, name)
        outfile = TFile(outputFilename, "RECREATE")
        outfile.cd()
        canvas.SetName("ddPlot_%s" % name)
        canvas.Write()
        outfile.Close()
      #print " --->done TDRifying! \n\n"
      return canvas