plotBeamSpotVert.Plots Class Reference

Inheritance diagram for plotBeamSpotVert.Plots:

Collaboration diagram for plotBeamSpotVert.Plots:

Public Member Functions
def	hist (self)
def	histErr (self)
def	histErrComp (self)
def	xz (self)
def	yz (self)
def	yx (self)
def	protect (self, obj)
def	plot (self, plotName='', *args)
def	genPlot (self, what='ALL', code='plot', labels=[], *args)

Public Attributes
	name
	otherMethods
	rootObjects
	whatList
	singleCanvasSize
	allCanvasSize
	allCanvasDivs
	saveAsList
	gPadSaveAsList

Detailed Description

Definition at line 145 of file plotBeamSpotVert.py.

Member Function Documentation

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 plotBeamSpotVert.Plots.hist

(

self

)

Definition at line 147 of file plotBeamSpotVert.py.

    def hist(self):
        hname = getPlotDef(options.plot,'hname','h')
        var = getPlotDef(options.plot,'var')
        units = getPlotDef(options.plot,'units','')
        ROOT.gStyle.SetOptStat(options.optstat)
        if options.fit:
            ROOT.gStyle.SetOptFit(1111)
        c = ROOTUtils.protect(ROOTUtils.MyCanvas(var,options.canvas))
        c.SetRightMargin(0.14)
        h = ROOTUtils.protect(ROOT.TH1F(hname,'Primary vertex: %s;Primary vertex %s %s' % (var,var,units),nbins,xmin,xmax))
        nt.Draw('%s >> %s' % (var,hname),cuts)
        if options.fit:
            h.Fit(options.fit)
        ROOTUtils.drawText(0.2,0.77,0.06,runFillInfo,font=42)
        comment = options.comment if options.comment is not None else '#geq %i tracks/vertex;%s' % (options.ntracks,options.cuts)
        ROOTUtils.drawText(0.2,0.71,0.06,lhcEnergyInfo+comment,font=42)
        ROOT.gPad.SetLogy(options.logy)
        ROOT.gPad.Update()
        c.save()
 

histErr()

def plotBeamSpotVert.Plots.histErr

(

self

)

Definition at line 167 of file plotBeamSpotVert.py.

    def histErr(self):
        hname = getPlotDef(options.plot,'hname','h')
        var = getPlotDef(options.plot,'var')
        vName = var[1:] # Ugly - should take from plotDef
        ROOT.gStyle.SetOptStat(options.optstat)
        ROOT.gStyle.SetOptFit(0)
        c = ROOTUtils.protect(ROOTUtils.MyCanvas(var,options.canvas))
        c.SetRightMargin(0.14)
        h = ROOTUtils.protect(ROOT.TH1F(hname,'Primary vertex error: %s;Primary vertex error #sqrt{V_{%s}} (#mum);Number of vertices' % (hname,vName),
                                        nbins,xmin,xmax))
        nt.Draw('1e3*sqrt(%s) >> %s' % (var,hname),cuts)
        ROOTUtils.drawText(0.45,0.86,0.06,'Primary Vertex Error #sqrt{V_{%s}}' % vName)
        comment = options.comment if options.comment is not None else '#geq %i tracks/vertex;%s' % (options.ntracks,options.cuts)
        ROOTUtils.drawText(0.45,0.8,0.06,lhcEnergyInfo+comment,font=42)
        ROOT.gPad.SetLogy(options.logy)
        ROOT.gPad.Update()
        c.save()
 

histErrComp()

def plotBeamSpotVert.Plots.histErrComp

(

self

)

Definition at line 185 of file plotBeamSpotVert.py.

    def histErrComp(self):
        hname = getPlotDef(options.plot,'hname','h')
        var = getPlotDef(options.plot,'var')
        vName = var[1:] # Ugly - should take from plotDef
        ROOT.gStyle.SetOptStat(0)
        ROOT.gStyle.SetOptFit(0)
        c = ROOTUtils.protect(ROOTUtils.MyCanvas(var,options.canvas))
        c.SetRightMargin(0.14)
        h = ROOTUtils.protect(ROOT.TH1F(hname,'Vertex Error: %s;Vertex error #sqrt{V_{%s}} [#mum];Fraction of entries' % (hname,vName),
                                        nbins,xmin,xmax))
        nt.Draw('1e3*sqrt(%s) >> %s' % (var,hname),cuts)
        norm = h.Integral()
        h.Scale(1./norm)
        pname = hname+'_pileup'
        p = ROOTUtils.protect(ROOT.TH1F(pname,'Vertex Error: %s;Vertex error #sqrt{V_{%s}} [#mum];Fraction of entries' % (hname,vName),
                                        nbins,xmin,xmax))
        pcuts = cuts.replace('vType==1','vType==3')
        print (pcuts)
        nt.Draw('1e3*sqrt(%s) >> %s' % (var,pname),pcuts,"SAME")
        pnorm = p.Integral()
        p.Scale(1./pnorm)
        p.SetLineColor(4)
        p.SetLineStyle(2)
        ROOTUtils.drawText(0.51,0.86,0.06,'Vertex error #sqrt{V_{%s}}' % vName)
        comment = options.comment if options.comment is not None else '#geq %i tracks/vertex;%s' % (options.ntracks,options.cuts)
        ROOTUtils.drawText(0.51,0.8,0.06,lhcEnergyInfo+comment,font=42)
        legend = ROOT.TLegend(.50,.6,.85,.72)
        legend.AddEntry(h,'Primary vertices','L')
        legend.AddEntry(p,'Pileup vertices','L')
        legend.SetFillColor(0)
        legend.SetBorderSize(0)
        legend.Draw()
        ROOT.gPad.SetLogy(options.logy)
        ROOT.gPad.Update()
        c.save()
 

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)
 

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
 

xz()

def plotBeamSpotVert.Plots.xz

(

self

)

Definition at line 221 of file plotBeamSpotVert.py.

    def xz(self):
        #ROOT.gStyle.SetOptStat(1010)       
        ROOT.gStyle.SetOptStat(0)
        c = ROOTUtils.protect(ROOTUtils.MyCanvas('xz',options.canvas))
        c.SetRightMargin(0.14)
        h = ROOTUtils.protect(ROOT.TH2F('pvXZ','Primary vertex: x vs z;Primary vertex z [mm];Primary vertex x [mm]',nbins,xmin,xmax,nbins,ymin,ymax))
        nt.Draw('x:z >> pvXZ',cuts)
        h.Draw('COLZ')
        h.GetYaxis().SetTitleOffset(1.0)
        drawLabels()
        #ROOTUtils.moveStats(h,-0.16,-0.08)
        drawInfo(c,h,options.more)
        ROOT.gPad.Update()
        c.save()
 

yx()

def plotBeamSpotVert.Plots.yx

(

self

)

Definition at line 249 of file plotBeamSpotVert.py.

    def yx(self):
        ROOT.gStyle.SetOptStat(0)
        c = ROOTUtils.protect(ROOTUtils.MyCanvas('yx',options.canvas))
        c.SetRightMargin(0.14)
        h = ROOTUtils.protect(ROOT.TH2F('pvYX','Primary vertex: y vs x;Primary vertex x [mm];Primary vertex y [mm]',nbins,xmin,xmax,nbins,ymin,ymax))
        nt.Draw('y:x >> pvYX',cuts)
        h.Draw('COLZ')
        h.GetYaxis().SetTitleOffset(1.0)
        drawLabels()
        drawInfo(c,h,options.more)
        ROOT.gPad.Update()
        c.save()
 
 
# Start processing

yz()

def plotBeamSpotVert.Plots.yz

(

self

)

Definition at line 236 of file plotBeamSpotVert.py.

    def yz(self):
        ROOT.gStyle.SetOptStat(0)
        c = ROOTUtils.protect(ROOTUtils.MyCanvas('yz',options.canvas))
        c.SetRightMargin(0.14)
        h = ROOTUtils.protect(ROOT.TH2F('pvYZ','Primary vertex: y vs z;Primary vertex z [mm];Primary vertex y [mm]',nbins,xmin,xmax,nbins,ymin,ymax))
        nt.Draw('y:z >> pvYZ',cuts)
        h.Draw('COLZ')
        h.GetYaxis().SetTitleOffset(1.0)
        drawLabels()
        drawInfo(c,h,options.more)
        ROOT.gPad.Update()
        c.save()
 

Member Data Documentation

allCanvasDivs

ROOTUtils.PlotLibrary.allCanvasDivs

inherited

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

allCanvasSize

ROOTUtils.PlotLibrary.allCanvasSize

inherited

Definition at line 96 of file roofit/ROOTUtils.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

ROOTUtils.PlotLibrary.singleCanvasSize

inherited

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

whatList

ROOTUtils.PlotLibrary.whatList

inherited

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

---

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

• plotBeamSpotVert.py