Inheritance diagram for python.BeamSpotData.BeamSpotNt:

Collaboration diagram for python.BeamSpotData.BeamSpotNt:

Public Member Functions
	__init__ (self, fileName, update=False, fullCorrelations=False, addScanVars=False)
	__del__ (self)
	allData (self)
	fill (self, bs)
	summary (self)
	initStatistics (self)
	__iter__ (self)
	next (self)
	__next__ (self)
	selectedData (self)
	getDataCache (self)
	cutSummary (self)

Public Attributes
str	treeName = 'BeamSpotNt'
	fileName = fileName
	update = update
	fullCorrelations = fullCorrelations
	addScanVars = addScanVars
	rootFile = ROOT.TFile(fileName,'UPDATE')
	ntbuf = BeamSpotNtBuf()
	nt = self.rootFile.Get(self.treeName)
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

Private Attributes
	__del__

Detailed Description

BeamSpotContainer for master beam spot ntuple.

Definition at line 917 of file BeamSpotData.py.

Constructor & Destructor Documentation

◆ init()

python.BeamSpotData.BeamSpotNt.__init__	(	self,
		fileName,
		update = False,
		fullCorrelations = False,
		addScanVars = False )

Definition at line 920 of file BeamSpotData.py.

    def __init__(self,fileName,update=False,fullCorrelations=False,addScanVars=False):
        BeamSpotContainer.__init__(self)
        self.treeName = 'BeamSpotNt'
        self.fileName = fileName
        self.update = update
        self.fullCorrelations = fullCorrelations
        self.addScanVars = addScanVars
 
        if update:
            bs = BeamSpotValue(self.fullCorrelations,self.addScanVars)
            self.rootFile = ROOT.TFile(fileName,'UPDATE')
            self.ntbuf = BeamSpotNtBuf()   # Not intialized
            self.nt = self.rootFile.Get(self.treeName)
            if not self.nt:
                self.nt = ROOT.TTree(self.treeName,'Master beam spot ntuple')
                for v in bs.varList():
                    varType = bs.getROOTType(v)
                    self.nt.Branch(v,ROOT.addressof( self.ntbuf,v), v+varType)
            else:
                for v in bs.varList():
                    self.nt.SetBranchAddress(v,cast['void*'](ROOT.addressof(self.ntbuf,v)))
        else:
            self.rootFile = ROOT.TFile(fileName)
            self.nt = self.rootFile.Get(self.treeName)
            if not self.nt:
                raise ValueError ('Tree %s not found in ntuple file %s' % (self.treeName,self.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.BeamSpotNt.__del__ ( self )

Definition at line 952 of file BeamSpotData.py.

    def __del__(self):
        # Prevent caling more than one (atexit/explicit del)
        if self.rootFile is not None:
            if self.update:
                self.rootFile.Write('',2)   # don't keep old versions
            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.BeamSpotNt.allData ( self )

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

Reimplemented from python.BeamSpotData.BeamSpotContainer.

Definition at line 960 of file BeamSpotData.py.

    def allData(self):
        varList = BeamSpotValue(self.fullCorrelations).varList()
        for j in range(self.nt.GetEntries()):
            self.nt.GetEntry(j)
            bs = BeamSpotValue(self.fullCorrelations)
            for v in varList:
                setattr(bs,v,getattr(self.nt,v,0))
            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
 
 
 

◆ fill()

python.BeamSpotData.BeamSpotNt.fill	(		self,
			bs )

Definition at line 969 of file BeamSpotData.py.

    def fill(self,bs):
        for v in bs.varList():
            setattr(self.ntbuf,v,getattr(bs,v))
        self.nt.Fill()
 

◆ 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.BeamSpotNt.summary ( self )

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

Reimplemented from python.BeamSpotData.BeamSpotContainer.

Definition at line 974 of file BeamSpotData.py.

    def summary(self):
        s = '%s:\n' % (self.fileName)
        if self.update:
            s += '    UPDATE MODE enabled\n'
        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.BeamSpotNt.__del__

private

Definition at line 950 of file BeamSpotData.py.

◆ acqFlag

python.BeamSpotData.BeamSpotContainer.acqFlag = None

inherited

Definition at line 765 of file BeamSpotData.py.

◆ addScanVars

python.BeamSpotData.BeamSpotNt.addScanVars = addScanVars

Definition at line 926 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.BeamSpotNt.fileName = fileName

Definition at line 923 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.

◆ fullCorrelations

python.BeamSpotData.BeamSpotNt.fullCorrelations = fullCorrelations

Definition at line 925 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.BeamSpotNt.nt = self.rootFile.Get(self.treeName)

Definition at line 932 of file BeamSpotData.py.

◆ ntbuf

python.BeamSpotData.BeamSpotNt.ntbuf = BeamSpotNtBuf()

Definition at line 931 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.BeamSpotNt.rootFile = ROOT.TFile(fileName,'UPDATE')

Definition at line 930 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.BeamSpotNt.treeName = 'BeamSpotNt'

Definition at line 922 of file BeamSpotData.py.

◆ update

python.BeamSpotData.BeamSpotNt.update = update

Definition at line 924 of file BeamSpotData.py.

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

BeamSpotData.py

Public Member Functions

Public Attributes

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ __init__()

◆ __del__()

Member Function Documentation

◆ __iter__()

◆ __next__()

◆ allData()

◆ cutSummary()

◆ fill()

◆ getDataCache()

◆ initStatistics()

◆ next()

◆ selectedData()

◆ summary()

Member Data Documentation

◆ __del__

◆ acqFlag

◆ addScanVars

◆ bcidMax

◆ bcidMin

◆ fileName

◆ fillMax

◆ fillMin

◆ fullCorrelations

◆ grl

◆ grlIOVs

◆ iter

◆ lbMax

◆ lbMin

◆ nSel

◆ nt

◆ ntbuf

◆ nTot

◆ previousGRLIndex

◆ rootFile

◆ runList

◆ runListExclude

◆ runMax

◆ runMin

◆ selBcidMax

◆ selBcidMin

◆ selFillMax

◆ selFillMin

◆ selRunMax

◆ selRunMin

◆ selTimeMax

◆ selTimeMin

◆ statusList

◆ timeMax

◆ timeMin

◆ treeName

◆ update

◆ init()

◆ del()

◆ iter()

◆ next()

◆ del