Inheritance diagram for python.BeamSpotData.BeamSpotFinderNt:

Collaboration diagram for python.BeamSpotData.BeamSpotFinderNt:

Public Member Functions
	__init__ (self, fileName, treeName='BeamSpotNt', fullCorrelations=True)
	__del__ (self)
	allData (self)
	summary (self)
	initStatistics (self)
	__iter__ (self)
	next (self)
	__next__ (self)
	selectedData (self)
	getDataCache (self)
	cutSummary (self)

Public Attributes
	fileName = fileName
	treeName = treeName
	rootFile = ROOT.TFile(fileName)
	nt = self.rootFile.Get(treeName)
	fullCorrelations = fullCorrelations
int	runMin = 0
int	runMax = 9999999
list	runList = []
list	runListExclude = []
int	fillMin = 0
int	fillMax = 9999999
int	bcidMin = 0
int	bcidMax = 9999999
int	lbMin = 0
int	lbMax = 9999999999
int	timeMin = 0
int	timeMax = 2000000000
list	statusList = []
	acqFlag = None
str	grl = ''
	grlIOVs = None
int	previousGRLIndex = 0
	iter = None
int	nTot = 0
int	nSel = 0
int	selRunMin = 9999999
int	selRunMax = 0
int	selFillMin = 9999999
int	selFillMax = 0
int	selBcidMin = 9999999
int	selBcidMax = 0
int	selTimeMin = 2000000000
int	selTimeMax = -1

Static Public Attributes
dict	fitResultToStatusMap = {0: 0, 1: 3, 2: 0, 3: 0}
dict	fitIdToStatusMap = {1: 0x38, 2: 0x40, 3: 0x10}

Private Attributes
	__del__

Detailed Description

BeamSpotContainer for ntuples created by InDetBeamSpotFinder.

Definition at line 983 of file BeamSpotData.py.

Constructor & Destructor Documentation

◆ init()

python.BeamSpotData.BeamSpotFinderNt.__init__	(	self,
		fileName,
		treeName = 'BeamSpotNt',
		fullCorrelations = True )

Definition at line 994 of file BeamSpotData.py.

    def __init__(self,fileName,treeName = 'BeamSpotNt',fullCorrelations=True):
        BeamSpotContainer.__init__(self)
        self.fileName = fileName
        self.treeName = treeName
        self.rootFile = ROOT.TFile(fileName)
        self.nt = self.rootFile.Get(treeName)
        self.fullCorrelations = fullCorrelations
        if not self.nt:
            raise ValueError ('Tree %s not found in ntuple file %s' % (treeName,fileName))
 
        # Register instance's __del__() with exit handler
        # This is needed for ROOT >= 5.28 to prevent ROOTs exit handler cleaning up the file
        # before we have closed/written to it. Only needed if use TFile in __del__
        atexit.register(self.__del__)
 

◆ del()

python.BeamSpotData.BeamSpotFinderNt.__del__ ( self )

Definition at line 1009 of file BeamSpotData.py.

    def __del__(self):
        # Prevent caling more than one (atexit/explicit del)
        if self.rootFile is not None:
            self.rootFile.Close()
            self.rootFile = None
 

Member Function Documentation

◆ iter()

python.BeamSpotData.BeamSpotContainer.__iter__ ( self )

inherited

Iterator to iterate over selected elements in the container.

Definition at line 792 of file BeamSpotData.py.

    def __iter__(self):
        """Iterator to iterate over selected elements in the container."""
        self.iter = self.selectedData()
        return self
 

◆ next()

python.BeamSpotData.BeamSpotContainer.__next__ ( self )

inherited

Return next selected element in the container.

Definition at line 801 of file BeamSpotData.py.

    def __next__(self):
        """Return next selected element in the container."""
        return self.next()
 

◆ allData()

python.BeamSpotData.BeamSpotFinderNt.allData ( self )

Default generator to iterate over all data. Must be overridden by derived classes.

Reimplemented from python.BeamSpotData.BeamSpotContainer.

Definition at line 1015 of file BeamSpotData.py.

    def allData(self):
        for j in range(self.nt.GetEntries()):
            self.nt.GetEntry(j)
            bs = BeamSpotValue(self.fullCorrelations)
            #try:
            #    bs.status = BeamSpotFinderNt.fitIdToStatusMap[self.nt.fitID]+BeamSpotFinderNt.fitResultToStatusMap[self.nt.fitStatus]
            #except Exception:
            #    bs.status = 0
            #    print ("ERROR: can't translate (fitID,fitStatus) = (%i,%i) into status word" % (self.nt.fitID,self.nt.fitStatus))
            bs.run = self.nt.run
            try:
                bs.bcid = self.nt.bcid
            except Exception:
                pass
            bs.lbStart = self.nt.lumiStart
            bs.lbEnd = self.nt.lumiStart+self.nt.lumiRange
            bs.nEvents = self.nt.nEvents
            try:
                bs.nValid = self.nt.nValid
            except Exception:
                pass
            try:
                bs.nVtxAll = self.nt.nVtxAll
            except Exception:
                pass
            try:
                bs.nVtxPrim = self.nt.nVtxPrim
            except Exception:
                pass                        
            bs.posX = self.nt.xc
            bs.posY = self.nt.yc
            bs.posZ = self.nt.z
            bs.sigmaX = self.nt.sx
            bs.sigmaY = self.nt.sy
            bs.sigmaZ = self.nt.sz
            bs.tiltX = self.nt.ax
            bs.tiltY = self.nt.ay
            bs.rhoXY = self.nt.rhoxy
            bs.k = self.nt.k
            bs.posXErr = sqrt(self.nt.xcxc)
            bs.posYErr = sqrt(self.nt.ycyc)
            bs.posZErr = sqrt(self.nt.zz)
            bs.sigmaXErr = sqrt(self.nt.sxsx)
            bs.sigmaYErr = sqrt(self.nt.sysy)
            bs.sigmaZErr = sqrt(self.nt.szsz)
            bs.tiltXErr = sqrt(self.nt.axax)
            bs.tiltYErr = sqrt(self.nt.ayay)
            bs.rhoXYErr = sqrt(self.nt.rhoxyrhoxy)
            bs.kErr = sqrt(self.nt.kk)
            bs.covSxSy = self.nt.sxsy
            bs.covSxRhoXY = self.nt.sxrhoxy
            bs.covSyRhoXY = self.nt.syrhoxy
 
            bs.covXY = self.nt.x0y0
            bs.covXZ = self.nt.x0z
            bs.covXSx = self.nt.x0sx
            bs.covXSy = self.nt.x0sy
            bs.covXSz = self.nt.x0sz
            bs.covXTiltX = self.nt.x0ax
            bs.covXTiltY = self.nt.x0ay
            bs.covXRhoXY = self.nt.x0rhoxy
            bs.covXk = self.nt.x0k
 
            bs.covYZ = self.nt.y0z
            bs.covYSx = self.nt.y0sx
            bs.covYSy = self.nt.y0sy
            bs.covYSz = self.nt.y0sz
            bs.covYTiltX = self.nt.y0ax
            bs.covYTiltY = self.nt.y0ay
            bs.covYRhoXY = self.nt.y0rhoxy
            bs.covYk = self.nt.y0k
 
            bs.covZSx = self.nt.zsx
            bs.covZSy = self.nt.zsy
            bs.covZSz = self.nt.zsz
            bs.covZTiltX = self.nt.zax
            bs.covZTiltY = self.nt.zay
            bs.covZRhoXY = self.nt.zrhoxy
            bs.covZk = self.nt.zk
 
            bs.covTiltXTiltY = self.nt.axay
            bs.covTiltXSx = self.nt.axsx
            bs.covTiltXSy = self.nt.axsy
            bs.covTiltXSz = self.nt.axsz
            bs.covTiltXRhoXY = self.nt.axrhoxy
            bs.covTiltXk = self.nt.axk
            
            bs.covTiltYSx = self.nt.aysx
            bs.covTiltYSy = self.nt.aysy
            bs.covTiltYSz = self.nt.aysz
            bs.covTiltYRhoXY = self.nt.ayrhoxy
            bs.covTiltYk = self.nt.ayk
            
            bs.covSxSz = self.nt.sxsz
            bs.covSxk = self.nt.sxk
            
            bs.covSySz = self.nt.sysz
            bs.covSyk = self.nt.syk
            
            bs.covSzRhoXY = self.nt.szrhoxy
            bs.covSzk = self.nt.szk
            
            bs.covRhoXYk = self.nt.rhoxyk
            
            yield bs
 

◆ cutSummary()

python.BeamSpotData.BeamSpotContainer.cutSummary ( self )

inherited

Get summary of cuts made when looping over selected data.

Definition at line 894 of file BeamSpotData.py.

    def cutSummary(self):
        """Get summary of cuts made when looping over selected data."""
        s = 'Cuts:\n'
        s += '    run                 %7i ... %7i\n' % (self.runMin,self.runMax)
        s += '    runList             %7s\n' % str(self.runList)
        s += '    runListExcluded     %7s\n' % str(self.runListExclude)        
        s += '    fill                %7i ... %7i\n' % (self.fillMin,self.fillMax)
        s += '    bcid                %7i ... %7i\n' % (self.bcidMin,self.bcidMax)
        s += '    LB                  %7i ... %7i\n' % (self.lbMin,self.lbMax)
        s += '    fit status          %7s\n' % str(self.statusList)
        s += '    %s - %s' % (time.strftime('%a %b %d %X %Z %Y',time.localtime(self.timeMin)),
                               time.strftime('%a %b %d %X %Z %Y',time.localtime(self.timeMax)))
        s += '   acquisition flag     %7s' % self.acqFlag 
        if self.grl:
            s += '    GRL                  %s\n' % self.grl
        return s
 
 
 

◆ getDataCache()

python.BeamSpotData.BeamSpotContainer.getDataCache ( self )

inherited

Get a cache of all data in the form of a dict of runs, where each element
   is a dict with a BeamSpotValue for each individual lumi blocks.

Definition at line 874 of file BeamSpotData.py.

    def getDataCache(self):
        """Get a cache of all data in the form of a dict of runs, where each element
           is a dict with a BeamSpotValue for each individual lumi blocks."""
        cache = {}
        for b in self:
            r = b.run
            if r not in cache:
                cache[r] = {}
            if b.lbEnd-b.lbStart > 500:
                print ('WARNING: Cannot cache LB range %i ... %i for run %i' % (b.lbStart,b.lbEnd,r))
            else:
                for i in range(b.lbStart,b.lbEnd+1):
                    if b.status in self.statusList or not self.statusList:
                        cache[r][i] = b
        return cache
 

◆ initStatistics()

python.BeamSpotData.BeamSpotContainer.initStatistics ( self )

inherited

Definition at line 776 of file BeamSpotData.py.

    def initStatistics(self):
        self.nTot = 0
        self.nSel = 0
        self.selRunMin = 9999999
        self.selRunMax = 0
        self.selFillMin = 9999999
        self.selFillMax = 0
        self.selBcidMin = 9999999
        self.selBcidMax = 0
        self.selTimeMin = 2000000000
        self.selTimeMax = -1
 

◆ next()

python.BeamSpotData.BeamSpotContainer.next ( self )

inherited

Return next selected element in the container.

Definition at line 797 of file BeamSpotData.py.

    def next(self):
        """Return next selected element in the container."""
        return next(self.iter)
 

◆ selectedData()

python.BeamSpotData.BeamSpotContainer.selectedData ( self )

inherited

Generator to iterate over selected elements in the container.

Definition at line 809 of file BeamSpotData.py.

    def selectedData(self):
        """Generator to iterate over selected elements in the container."""
        self.initStatistics()
 
        self.previousGRLIndex = 0
 
        for b in self.allData():
            self.nTot +=1
            if b.run<self.runMin: continue
            if b.run>self.runMax: continue
            if self.runList and b.run not in self.runList: continue
            if self.runListExclude and b.run in self.runListExclude: continue
            if b.fill<self.fillMin: continue
            if b.fill>self.fillMax: continue
            if b.bcid<self.bcidMin: continue
            if b.bcid>self.bcidMax: continue
            if b.lbStart<self.lbMin: continue
            if b.lbEnd-1>self.lbMax: continue   # lbEnd is exclusive
            if b.timeStart<self.timeMin: continue
            if b.timeEnd>self.timeMax: continue
            if self.statusList and b.status not in self.statusList: continue            
            #if self.acqFlag is not None and b.acquisitionFlag != self.acqFlag: continue
            
            if self.grl:
                # Check if in GRL
                from DQUtils.sugar import RANGEIOV_VAL, RunLumi
                from DQUtils import IOVSet
 
                # Get IoVs from GRL
                if self.grlIOVs is None:
                    self.grlIOVs = IOVSet.from_grl(self.grl)
 
                idx = self.previousGRLIndex
 
                # Get current IoV
                test_iov = RANGEIOV_VAL(RunLumi(b.run, b.lbStart), RunLumi(b.run, b.lbEnd)) # COOL convention, ie lbEnd is exclusive
 
                # Search for IoV in GRL IoVs
                for i, iov in enumerate(self.grlIOVs[idx:]):
                    if (test_iov.since >= iov.since and test_iov.until <= iov.until):
                        self.previousGRLIndex = idx + i
                        break
                else:
                    # Only if don't break ->  IOV not found so skip
                    continue
 
 
            self.nSel += 1
            self.selRunMin = min(self.selRunMin,b.run)
            self.selRunMax = max(self.selRunMax,b.run)
            self.selFillMin = min(self.selFillMin,b.fill)
            self.selFillMax = max(self.selFillMax,b.fill)
            self.selBcidMin = min(self.selBcidMin,b.bcid)
            self.selBcidMax = max(self.selBcidMax,b.bcid)
            self.selTimeMin = min(self.selTimeMin,b.timeStart)
            self.selTimeMax = max(self.selTimeMax,b.timeEnd)
            yield b
        print ('\n%i entries selected out of total of %i entries in ntuple:' % (self.nSel,self.nTot))
        print ('... runs   %6i - %6i' % (self.selRunMin,self.selRunMax))
        print ('... fills  %6i - %6i' % (self.selFillMin,self.selFillMax))
        print ('... bcids  %6i - %6i' % (self.selBcidMin,self.selBcidMax))
        print ('... %s - %s' % (time.strftime('%a %b %d %X %Z %Y',time.localtime(self.selTimeMin)),
                               time.strftime('%a %b %d %X %Z %Y',time.localtime(self.selTimeMax))))
        print()
 

◆ summary()

python.BeamSpotData.BeamSpotFinderNt.summary ( self )

Get one-line info of Ntuple. Should be overridden by derived classes.

Reimplemented from python.BeamSpotData.BeamSpotContainer.

Definition at line 1121 of file BeamSpotData.py.

    def summary(self):
        s = '%s:\n' % (self.fileName)
        s += '    %s ntuple in tree %s\n' % (self.__class__.__name__,self.treeName)
        s += '    %i entries\n' % (self.nt.GetEntries())
        return s
 
 

Member Data Documentation

◆ del

python.BeamSpotData.BeamSpotFinderNt.__del__

private

Definition at line 1007 of file BeamSpotData.py.

◆ acqFlag

python.BeamSpotData.BeamSpotContainer.acqFlag = None

inherited

Definition at line 765 of file BeamSpotData.py.

◆ bcidMax

int python.BeamSpotData.BeamSpotContainer.bcidMax = 9999999

inherited

Definition at line 759 of file BeamSpotData.py.

◆ bcidMin

int python.BeamSpotData.BeamSpotContainer.bcidMin = 0

inherited

Definition at line 758 of file BeamSpotData.py.

◆ fileName

python.BeamSpotData.BeamSpotFinderNt.fileName = fileName

Definition at line 996 of file BeamSpotData.py.

◆ fillMax

int python.BeamSpotData.BeamSpotContainer.fillMax = 9999999

inherited

Definition at line 757 of file BeamSpotData.py.

◆ fillMin

int python.BeamSpotData.BeamSpotContainer.fillMin = 0

inherited

Definition at line 756 of file BeamSpotData.py.

◆ fitIdToStatusMap

dict python.BeamSpotData.BeamSpotFinderNt.fitIdToStatusMap = {1: 0x38, 2: 0x40, 3: 0x10}

static

Definition at line 992 of file BeamSpotData.py.

◆ fitResultToStatusMap

dict python.BeamSpotData.BeamSpotFinderNt.fitResultToStatusMap = {0: 0, 1: 3, 2: 0, 3: 0}

static

Definition at line 991 of file BeamSpotData.py.

◆ fullCorrelations

python.BeamSpotData.BeamSpotFinderNt.fullCorrelations = fullCorrelations

Definition at line 1000 of file BeamSpotData.py.

◆ grl

str python.BeamSpotData.BeamSpotContainer.grl = ''

inherited

Definition at line 766 of file BeamSpotData.py.

◆ grlIOVs

python.BeamSpotData.BeamSpotContainer.grlIOVs = None

inherited

Definition at line 767 of file BeamSpotData.py.

◆ iter

python.BeamSpotData.BeamSpotContainer.iter = None

inherited

Definition at line 771 of file BeamSpotData.py.

◆ lbMax

int python.BeamSpotData.BeamSpotContainer.lbMax = 9999999999

inherited

Definition at line 761 of file BeamSpotData.py.

◆ lbMin

int python.BeamSpotData.BeamSpotContainer.lbMin = 0

inherited

Definition at line 760 of file BeamSpotData.py.

◆ nSel

python.BeamSpotData.BeamSpotContainer.nSel = 0

inherited

Definition at line 778 of file BeamSpotData.py.

◆ nt

python.BeamSpotData.BeamSpotFinderNt.nt = self.rootFile.Get(treeName)

Definition at line 999 of file BeamSpotData.py.

◆ nTot

python.BeamSpotData.BeamSpotContainer.nTot = 0

inherited

Definition at line 777 of file BeamSpotData.py.

◆ previousGRLIndex

int python.BeamSpotData.BeamSpotContainer.previousGRLIndex = 0

inherited

Definition at line 768 of file BeamSpotData.py.

◆ rootFile

python.BeamSpotData.BeamSpotFinderNt.rootFile = ROOT.TFile(fileName)

Definition at line 998 of file BeamSpotData.py.

◆ runList

list python.BeamSpotData.BeamSpotContainer.runList = []

inherited

Definition at line 754 of file BeamSpotData.py.

◆ runListExclude

list python.BeamSpotData.BeamSpotContainer.runListExclude = []

inherited

Definition at line 755 of file BeamSpotData.py.

◆ runMax

int python.BeamSpotData.BeamSpotContainer.runMax = 9999999

inherited

Definition at line 753 of file BeamSpotData.py.

◆ runMin

int python.BeamSpotData.BeamSpotContainer.runMin = 0

inherited

Definition at line 752 of file BeamSpotData.py.

◆ selBcidMax

python.BeamSpotData.BeamSpotContainer.selBcidMax = 0

inherited

Definition at line 784 of file BeamSpotData.py.

◆ selBcidMin

python.BeamSpotData.BeamSpotContainer.selBcidMin = 9999999

inherited

Definition at line 783 of file BeamSpotData.py.

◆ selFillMax

python.BeamSpotData.BeamSpotContainer.selFillMax = 0

inherited

Definition at line 782 of file BeamSpotData.py.

◆ selFillMin

python.BeamSpotData.BeamSpotContainer.selFillMin = 9999999

inherited

Definition at line 781 of file BeamSpotData.py.

◆ selRunMax

python.BeamSpotData.BeamSpotContainer.selRunMax = 0

inherited

Definition at line 780 of file BeamSpotData.py.

◆ selRunMin

python.BeamSpotData.BeamSpotContainer.selRunMin = 9999999

inherited

Definition at line 779 of file BeamSpotData.py.

◆ selTimeMax

python.BeamSpotData.BeamSpotContainer.selTimeMax = -1

inherited

Definition at line 786 of file BeamSpotData.py.

◆ selTimeMin

python.BeamSpotData.BeamSpotContainer.selTimeMin = 2000000000

inherited

Definition at line 785 of file BeamSpotData.py.

◆ statusList

list python.BeamSpotData.BeamSpotContainer.statusList = []

inherited

Definition at line 764 of file BeamSpotData.py.

◆ timeMax

int python.BeamSpotData.BeamSpotContainer.timeMax = 2000000000

inherited

Definition at line 763 of file BeamSpotData.py.

◆ timeMin

int python.BeamSpotData.BeamSpotContainer.timeMin = 0

inherited

Definition at line 762 of file BeamSpotData.py.

◆ treeName

python.BeamSpotData.BeamSpotFinderNt.treeName = treeName

Definition at line 997 of file BeamSpotData.py.

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

BeamSpotData.py

Public Member Functions

Public Attributes

Static Public Attributes

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ __init__()

◆ __del__()

Member Function Documentation

◆ __iter__()

◆ __next__()

◆ allData()

◆ cutSummary()

◆ getDataCache()

◆ initStatistics()

◆ next()

◆ selectedData()

◆ summary()

Member Data Documentation

◆ __del__

◆ acqFlag

◆ bcidMax

◆ bcidMin

◆ fileName

◆ fillMax

◆ fillMin

◆ fitIdToStatusMap

◆ fitResultToStatusMap

◆ fullCorrelations

◆ grl

◆ grlIOVs

◆ iter

◆ lbMax

◆ lbMin

◆ nSel

◆ nt

◆ nTot

◆ previousGRLIndex

◆ rootFile

◆ runList

◆ runListExclude

◆ runMax

◆ runMin

◆ selBcidMax

◆ selBcidMin

◆ selFillMax

◆ selFillMin

◆ selRunMax

◆ selRunMin

◆ selTimeMax

◆ selTimeMin

◆ statusList

◆ timeMax

◆ timeMin

◆ treeName

◆ init()

◆ del()

◆ iter()

◆ next()

◆ del