Inheritance diagram for fitman.Plots:

Collaboration diagram for fitman.Plots:

Public Member Functions
def	__init__ (self, name=options.name, whatList=[])

def	prepareMainPad (self)

def	drawRatioPad (self, frame, hmin, hmax)

def	labels (self)

def	legend (self, frame)

def	hist (self, what)

def	x (self)

def	y (self)

def	z (self)

def	protect (self, obj)

def	plot (self, plotName='', *args)

def	genPlot (self, what='ALL', code='plot', labels=[], *args)

Public Attributes
	whatList

	allCanvasSize

	allCanvasDivs

	singleCanvasSize

	name

	otherMethods

	rootObjects

	saveAsList

	gPadSaveAsList

Detailed Description

Definition at line 156 of file fitman.py.

Constructor & Destructor Documentation

◆ init()

def fitman.Plots.__init__	(	self,
		name = `options.name`,
		otherMethods = `[]`
	)

Constructor. otherMethods specifies a list of methods of the derived
   class that are not creating plots and should not be called by plot().

Reimplemented from ROOTUtils.PlotLibrary.

Definition at line 158 of file fitman.py.

     def __init__(self, name = options.name, whatList = []):
         ROOTUtils.PlotLibrary.__init__(self,name)
         self.whatList = whatList
         self.allCanvasSize = 'landscape'
         self.allCanvasDivs = (2,2)
         self.singleCanvasSize = options.canvas
  

Member Function Documentation

◆ drawRatioPad()

def fitman.Plots.drawRatioPad	(	self,
		frame,
		hmin,
		hmax
	)

Definition at line 173 of file fitman.py.

     def drawRatioPad(self,frame,hmin,hmax):
         ROOT.gROOT.ProcessLine(".L ratioPlot.C+")
         axisTitle = frame.GetXaxis().GetTitle()
         frame.GetXaxis().SetTitle('')
         frame.GetXaxis().SetLabelSize(0.)
         frame.GetYaxis().SetTitleOffset(1.0)
         canvas = ROOT.gPad.GetCanvas()
         canvas.cd()
         ratioPad = self.protect( ROOT.TPad("ratioPad","ratioPad",0.,0.,1.,0.3) )
         ratioPad.SetTopMargin(0.)
         ratioPad.SetBottomMargin(0.3)
         ratioPad.Draw()
         ratioPad.cd()
         dataHist = frame.getObject(0)
         fitCurve = frame.getObject(1)
         #global pull
         #pull = dataHist.makeResidHist(fitCurve,True,True)
         global ratio
         ratio = ROOT.ratioPlot(dataHist,fitCurve)
         ratio.Draw('AP')
         # NOTE: SetRangeUser chooses bin edges, so ratio plot would not be perfectly aligned
         ratio.GetXaxis().SetLimits(hmin,hmax)
         scale = 2.5
         size = scale*options.lsize
         ratio.GetXaxis().SetLabelSize(size)
         ratio.GetXaxis().SetTitle(axisTitle)
         ratio.GetXaxis().SetTitleSize(size)
         ratio.GetXaxis().SetTitleOffset(1.)
         ratio.GetYaxis().SetRangeUser(0.3,1.7)
         ratio.GetYaxis().SetLabelSize(size)
         ratio.GetYaxis().SetTitle('Data / fit')
         #ratio.GetYaxis().SetTitleSize(scale*ratio.GetYaxis().GetTitleSize())
         ratio.GetYaxis().SetTitleSize(size)
         ratio.GetYaxis().SetTitleOffset(1.0/scale)
         ratio.GetYaxis().SetNdivisions(504)
         ratio.SetMarkerSize(0.7)
         line = self.protect( ROOT.TLine(hmin,1,hmax,1) )
         line.Draw()
  
  

◆ genPlot()

def ROOTUtils.PlotLibrary.genPlot	(		self,
			what = `'ALL'`,
			code = `'plot'`,
			labels = `[]`,
		*	args
	)

inherited

Make plots using a general code. genPlot makes either a single plot defined
   by argument what, or all plots (if what=''). The plots made in the latter
   case are specified by whatList. A canvas that is subdivided if necessary
   is created before the plotting code is called.

Definition at line 116 of file roofit/ROOTUtils.py.

     def genPlot(self,what='ALL',code='plot',labels=[],*args):
         """Make plots using a general code. genPlot makes either a single plot defined
            by argument what, or all plots (if what=''). The plots made in the latter
            case are specified by whatList. A canvas that is subdivided if necessary
            is created before the plotting code is called."""
         if what=='' or what.upper()=='ALL':
             c = self.protect( MyCanvas('%s-%s-%s' % (self.name,what,code),
                                        self.allCanvasSize,self.allCanvasDivs[0],self.allCanvasDivs[1]) )
             iCanvas = 0
             for w in self.whatList:
                 iCanvas += 1
                 c.cd(iCanvas)
                 try:
                     self.__class__.__dict__[code](self,w,*args)
                 except:
                     self.__class__.__bases__[0].__dict__[code](self,w,*args)
                 ROOT.gPad.Update()
                 # For some strange reason, this works only for .eps, but not e.g. for gif files...???
                 for o in self.gPadSaveAsList:
                     if o[0]=='.':
                         ROOT.gPad.SaveAs('%s-%s-%s%s' % (self.name,code,w,o))
                     else:
                         ROOT.gPad.SaveAs(o)
  
             # If we want to put some labels on an empty pad, add them now.
             if labels!=[] and self.allCanvasDivs[0]*self.allCanvasDivs[1] > len(self.whatList):
                 iCanvas+=1
                 c.cd(iCanvas)
                 xtext=0.0
                 ytext=0.8
                 for label in labels:
                     drawText(xtext,ytext,0.06,label)
                     ytext=ytext-0.1
             elif labels!=[]:
                 print "ERROR: can't add labels unless we have an empty pad to use.  Ignoring labels."
  
             for o in self.saveAsList:
                 if o[0]=='.':
                     c.SaveAs('%s-%s-%s%s' % (self.name,code,what,o))
                 else:
                     c.SaveAs(o)
  
         else:
             c = self.protect( MyCanvas(what,self.singleCanvasSize) )
             try:
                 self.__class__.__dict__[code](self,what,*args)
             except:
                 self.__class__.__bases__[0].__dict__[code](self,what,*args)
             ROOT.gPad.Update()
             for o in self.saveAsList:
                 if o[0]=='.':
                     c.SaveAs('%s-%s-%s%s' % (self.name,code,what,o))
                 else:
                     c.SaveAs(o)
  

◆ hist()

def fitman.Plots.hist	(	self,
		what
	)

Definition at line 233 of file fitman.py.

     def hist(self,what):
         if not what:
             return
         if options.hmin is not None or options.hmax is not None or options.hnbins is not None:
             options.hmin = options.hmin if options.hmin is not None else 0.
             options.hmax = options.hmax if options.hmax is not None else 1.
             options.hnbins = options.hnbins if options.hnbins is not None else 40
             h = ROOTUtils.protect( ROOT.TH1F('tmp','%s;%s' % (what,what),
                                              options.hnbins,options.hmin,options.hmax))
             vtxData.Draw('%s >> tmp' % what)
         else:
             vtxData.Draw(what)
         if options.logy:
             ROOT.gPad.SetLogy(options.logy)
  

◆ labels()

def fitman.Plots.labels ( self )

Definition at line 213 of file fitman.py.

     def labels(self):
         # ATLAS labels and other text
         if options.prelim:
             ROOTUtils.atlasLabel(options.atlasx,options.atlasy,True,offset=options.atlasdx,energy=8,size=options.lsize)
         if options.approval:
             ROOTUtils.atlasLabel(options.atlasx,options.atlasy,False,offset=options.atlasdx,isForApproval=True,energy=8,size=options.lsize)
         if options.published:
             ROOTUtils.atlasLabel(options.atlasx,options.atlasy,False,offset=options.atlasdx,energy=8,size=options.lsize)
         if options.comment:
             self.protect( ROOTUtils.drawText(options.commentx,options.commenty,0.06,options.comment,font=42) )
  

◆ legend()

def fitman.Plots.legend	(	self,
		frame
	)

Definition at line 224 of file fitman.py.

     def legend(self,frame):
         legendList = []
         legendList.append([frame.getObject(0),'Data','P'])
         legendList.append([frame.getObject(1),'Fit projection','L'])
         if options.showbs:
             legendList.append([frame.getObject(2),'Beam spot','L'])
         legendMinY = max(options.legendy-options.lsize*1.4*len(legendList),0.2)
         self.protect( ROOTUtils.drawLegend(options.legendx,legendMinY,0.92,options.legendy,legendList,textSize=options.lsize) )
  

◆ plot()

def ROOTUtils.PlotLibrary.plot	(		self,
			plotName = `''`,
		*	args
	)

inherited

Make one or all (if plotName=='') plots.

Definition at line 106 of file roofit/ROOTUtils.py.

     def plot(self, plotName='',*args):
         """Make one or all (if plotName=='') plots."""
         if plotName:
             self.__class__.__dict__[plotName](self,*args)
         else:
             for i in self.__class__.__dict__:
                 if i[0]=='_': continue  # skip private methods
                 if i in self.otherMethods: continue
                 self.__class__.__dict__[i](self,*args)
  

◆ prepareMainPad()

def fitman.Plots.prepareMainPad ( self )

Definition at line 165 of file fitman.py.

     def prepareMainPad(self):
         mainPad = self.protect( ROOT.TPad("mainPad","mainPad",0.,0.3,1.,1.) )
         mainPad.SetBottomMargin(0.03) # with space between the two pads
         #mainPad.SetBottomMargin(0.00) # w/o space between the two pads
         mainPad.Draw()
         mainPad.cd()
         return mainPad
  

◆ protect()

def ROOTUtils.PlotLibrary.protect	(	self,
		obj
	)

inherited

Protect ROOT object from garbage collection.

Definition at line 101 of file roofit/ROOTUtils.py.

     def protect(self,obj):
         """Protect ROOT object from garbage collection."""
         self.rootObjects.append(obj)
         return obj
  

◆ x()

def fitman.Plots.x ( self )

Definition at line 248 of file fitman.py.

     def x(self):
         if options.showratio:
             self.prepareMainPad()
         dx = options.hrange/2 if options.hrange else max(options.minrange/2.,8.*sx.getVal())
         hmin = options.hmin if options.hmin is not None else mx.getVal()-dx
         hmax = options.hmax if options.hmax is not None else mx.getVal()+dx
         global xframe
         xframe = self.protect( x.frame(hmin,hmax) )
         if options.hnbins is not None:
             binning = ROOT.RooBinning(hmin, hmax)
             binning.addUniform(options.hnbins, hmin, hmax)
             data.plotOn(xframe,RooFit.Binning(binning))
             xbinWidth = (hmax-hmin)/float(options.hnbins)
         else:
             data.plotOn(xframe)
             xbinWidth = xframe.GetXaxis().GetBinWidth(1)
         #xframe.GetYaxis().SetTitle("Number of primary vertices per %3.1f #mum" % round(1000.*xbinWidth,1))
         xframe.GetYaxis().SetTitle("Events / %1.0f #mum" % round(1000.*xbinWidth,1))
         xframe.GetYaxis().SetLabelSize(options.lsize)
         xframe.GetYaxis().SetTitleSize(options.lsize)
         if options.stats:
             data.statOn(xframe)
         cov = RooArgSet(vxx,vyy,vxy)
         fitmodel.plotOn(xframe,RooFit.ProjWData(cov,data),
                         RooFit.NumCPU(options.nCpu))
         if options.showbs:
             k.setVal(0.0)
             fitmodel.plotOn(xframe,RooFit.ProjWData(cov,data),
                             RooFit.NumCPU(options.nCpu),
                             RooFit.LineStyle(RooFit.kDashed))
         xframe.Draw()
         if options.logy:
             ROOT.gPad.SetLogy(options.logy)
         self.labels()
         self.legend(xframe)
         if options.showratio:
             self.drawRatioPad(xframe,hmin,hmax)
  

◆ y()

def fitman.Plots.y ( self )

Definition at line 286 of file fitman.py.

     def y(self):
         if options.showratio:
             self.prepareMainPad()
         dy = options.hrange/2 if options.hrange else max(options.minrange/2.,8.*sy.getVal())
         hmin = options.hmin if options.hmin is not None else my.getVal()-dy
         hmax = options.hmax if options.hmax is not None else my.getVal()+dy
         global yframe
         yframe = self.protect( y.frame(hmin,hmax) )
         if options.hnbins is not None:
             binning = ROOT.RooBinning(hmin, hmax)
             binning.addUniform(options.hnbins, hmin, hmax)
             data.plotOn(yframe,RooFit.Binning(binning))
             ybinWidth = (hmax-hmin)/float(options.hnbins)
         else:
             data.plotOn(yframe)
             ybinWidth = yframe.GetXaxis().GetBinWidth(1)
         #yframe.GetYaxis().SetTitle("Number of primary vertices per %3.1f #mum" % round(1000.*ybinWidth,1))
         yframe.GetYaxis().SetTitle("Events / %1.0f #mum" % round(1000.*ybinWidth,1))
         yframe.GetYaxis().SetLabelSize(options.lsize)
         yframe.GetYaxis().SetTitleSize(options.lsize)
         if options.stats:
             data.statOn(yframe)
         cov = RooArgSet(vxx,vyy,vxy)
         fitmodel.plotOn(yframe,RooFit.ProjWData(cov,data),
                         RooFit.NumCPU(options.nCpu))
         if options.showbs:
             k.setVal(0.0)
             fitmodel.plotOn(yframe,RooFit.ProjWData(cov,data),
                             RooFit.NumCPU(options.nCpu),
                             RooFit.LineStyle(RooFit.kDashed))
         yframe.Draw()
         if options.logy:
             ROOT.gPad.SetLogy(options.logy)
         self.labels()
         self.legend(yframe)
         if options.showratio:
             self.drawRatioPad(yframe,hmin,hmax)
  

◆ z()

def fitman.Plots.z ( self )

Definition at line 324 of file fitman.py.

     def z(self):
         if options.showratio:
             self.prepareMainPad()
         dz = options.hrange/2 if options.hrange else max(options.minrange/2.,8.*sz.getVal())
         hmin = options.hmin if options.hmin is not None else mz.getVal()-dz
         hmax = options.hmax if options.hmax is not None else mz.getVal()+dz
         global zframe
         zframe = self.protect( z.frame(hmin,hmax) )
         if options.hnbins is not None:
             binning = ROOT.RooBinning(hmin, hmax)
             binning.addUniform(options.hnbins, hmin, hmax)
             data.plotOn(zframe,RooFit.Binning(binning))
             zbinWidth = (hmax-hmin)/float(options.hnbins)
         else:
             data.plotOn(zframe)
             zbinWidth = zframe.GetXaxis().GetBinWidth(1)
         #zframe.GetYaxis().SetTitle("Number of primary vertices per %3.1f mm" % round(zbinWidth,1))
         zframe.GetYaxis().SetLabelSize(options.lsize)
         zframe.GetYaxis().SetTitleSize(options.lsize)
         zframe.GetYaxis().SetTitle("Events / %1.0f mm" % round(zbinWidth,1))
         if options.stats:
             data.statOn(zframe)
         cov = RooArgSet(vxx,vyy,vxy)
         fitmodel.plotOn(zframe,RooFit.ProjWData(cov,data),
                         RooFit.NumCPU(options.nCpu))   # could also use e.g. RooFit.Range(-100.,100.)
         if options.showbs:
             k.setVal(0.0)
             fitmodel.plotOn(zframe,RooFit.ProjWData(cov,data),
                             RooFit.NumCPU(options.nCpu),
                             RooFit.LineStyle(RooFit.kDashed))
         zframe.Draw()
         if options.logy:
             ROOT.gPad.SetLogy(options.logy)
         self.labels()
         self.legend(zframe)
         if options.showratio:
             self.drawRatioPad(zframe,hmin,hmax)
  
  
 #
 # If extract command is given, must process it here, before setting up
 # RooFit
 #

Member Data Documentation

◆ allCanvasDivs

fitman.Plots.allCanvasDivs

Definition at line 162 of file fitman.py.

◆ allCanvasSize

fitman.Plots.allCanvasSize

Definition at line 161 of file fitman.py.

◆ gPadSaveAsList

ROOTUtils.PlotLibrary.gPadSaveAsList

inherited

Definition at line 99 of file roofit/ROOTUtils.py.

◆ name

ROOTUtils.PlotLibrary.name

inherited

Definition at line 88 of file roofit/ROOTUtils.py.

◆ otherMethods

ROOTUtils.PlotLibrary.otherMethods

inherited

Definition at line 89 of file roofit/ROOTUtils.py.

◆ rootObjects

ROOTUtils.PlotLibrary.rootObjects

inherited

Definition at line 91 of file roofit/ROOTUtils.py.

◆ saveAsList

ROOTUtils.PlotLibrary.saveAsList

inherited

Definition at line 98 of file roofit/ROOTUtils.py.

◆ singleCanvasSize

fitman.Plots.singleCanvasSize

Definition at line 163 of file fitman.py.

◆ whatList

fitman.Plots.whatList

Definition at line 160 of file fitman.py.

The documentation for this class was generated from the following file:

fitman.py

Public Member Functions

Public Attributes

Detailed Description

Constructor & Destructor Documentation

◆ __init__()

Member Function Documentation

◆ drawRatioPad()

◆ genPlot()

◆ hist()

◆ labels()

◆ legend()

◆ plot()

◆ prepareMainPad()

◆ protect()

◆ x()

◆ y()

◆ z()

Member Data Documentation

◆ allCanvasDivs

◆ allCanvasSize

◆ gPadSaveAsList

◆ name

◆ otherMethods

◆ rootObjects

◆ saveAsList

◆ singleCanvasSize

◆ whatList

◆ init()