Inheritance diagram for python.BeamSpotData.BeamSpotCOOL:

Collaboration diagram for python.BeamSpotData.BeamSpotCOOL:

Public Member Functions
def	__init__ (self, tag, database='COOLOFL_INDET/CONDBR2', folder='/Indet/Beampos', fullCorrelations=False)

def	__del__ (self)

def	allData (self)

def	summary (self)

def	initStatistics (self)

def	__iter__ (self)

def	next (self)

def	__next__ (self)

def	selectedData (self)

def	getDataCache (self)

def	cutSummary (self)

Public Attributes
	database

	tag

	folder

	cooldb

	runMin

	runMax

	runList

	runListExclude

	fillMin

	fillMax

	bcidMin

	bcidMax

	lbMin

	lbMax

	timeMin

	timeMax

	statusList

	acqFlag

	grl

	grlIOVs

	previousGRLIndex

	iter

	nTot

	nSel

	selRunMin

	selRunMax

	selFillMin

	selFillMax

	selBcidMin

	selBcidMax

	selTimeMin

	selTimeMax

Detailed Description

BeamSpotContainer for beam spot information stored in COOL.

Definition at line 1128 of file BeamSpotData.py.

Constructor & Destructor Documentation

◆ init()

def python.BeamSpotData.BeamSpotCOOL.__init__	(	self,
		tag,
		database = `'COOLOFL_INDET/CONDBR2'`,
		folder = `'/Indet/Beampos'`,
		fullCorrelations = `False`
	)

Definition at line 1131 of file BeamSpotData.py.

     def __init__(self, tag, database='COOLOFL_INDET/CONDBR2', folder='/Indet/Beampos', fullCorrelations=False):
         BeamSpotContainer.__init__(self)
         self.database = database
         self.tag = tag
         self.folder = folder
         from CoolConvUtilities import AtlCoolLib
         self.cooldb = AtlCoolLib.indirectOpen(database, True, True)
  

◆ del()

def python.BeamSpotData.BeamSpotCOOL.__del__ ( self )

Definition at line 1139 of file BeamSpotData.py.

     def __del__(self):
         self.cooldb.closeDatabase()
  

Member Function Documentation

◆ iter()

def 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()

def 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()

def python.BeamSpotData.BeamSpotCOOL.allData ( self )

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

Reimplemented from python.BeamSpotData.BeamSpotContainer.

Definition at line 1142 of file BeamSpotData.py.

     def allData(self):
         from PyCool import cool
         
         folder = self.cooldb.getFolder(self.folder)
  
         iov1 = self.runMin << 32
         iov2 = (self.runMax+1) << 32
         if (iov2>cool.ValidityKeyMax):
             iov2=cool.ValidityKeyMax
  
         itr = folder.browseObjects(iov1, iov2, cool.ChannelSelection.all(), self.tag)
  
         while itr.goToNext():
  
             obj = itr.currentRef()
             
             since = obj.since()
             runBegin = since >> 32
             lumiBegin = since & 0xFFFFFFFF
  
             until = obj.until()
             lumiUntil = until & 0xFFFFFFFF
  
             bs            = BeamSpotValue()                            
             bs.run        = int(runBegin)
             bs.lbStart    = int(lumiBegin)
             bs.lbEnd      = int(lumiUntil)
             bs.posX       = float(obj.payloadValue('posX'))
             bs.posY       = float(obj.payloadValue('posY'))
             bs.posZ       = float(obj.payloadValue('posZ'))
             bs.sigmaX     = float(obj.payloadValue('sigmaX'))
             bs.sigmaY     = float(obj.payloadValue('sigmaY'))
             bs.sigmaZ     = float(obj.payloadValue('sigmaZ'))
             bs.tiltX      = float(obj.payloadValue('tiltX'))
             bs.tiltY      = float(obj.payloadValue('tiltY'))
             bs.status     = int(obj.payloadValue('status'))
             bs.posXErr    = float(obj.payloadValue('posXErr'))
             bs.posYErr    = float(obj.payloadValue('posYErr'))
             bs.posZErr    = float(obj.payloadValue('posZErr'))
             bs.sigmaXErr  = float(obj.payloadValue('sigmaXErr'))
             bs.sigmaYErr  = float(obj.payloadValue('sigmaYErr'))
             bs.sigmaZErr  = float(obj.payloadValue('sigmaZErr'))
             bs.tiltXErr   = float(obj.payloadValue('tiltXErr'))
             bs.tiltYErr   = float(obj.payloadValue('tiltYErr'))
  
             # COOL stores sigmaXY but BeamSpotData wants rho
             sigmaXYtmp    = float(obj.payloadValue('sigmaXY'))
             sigmaXtmp     = float(obj.payloadValue('sigmaX'))
             sigmaYtmp     = float(obj.payloadValue('sigmaY'))
             sigmaXYErrtmp = float(obj.payloadValue('sigmaXYErr'))
             sigmaXErrtmp  = float(obj.payloadValue('sigmaXErr'))
             sigmaYErrtmp  = float(obj.payloadValue('sigmaYErr'))
  
             try:
                 rhoXYtmp = sigmaXYtmp / sigmaXtmp / sigmaYtmp
             except Exception:
                 rhoXYtmp = 0
             bs.rhoXY      = rhoXYtmp
  
             try:
                 sumtmp = sigmaXYErrtmp * sigmaXYErrtmp / sigmaXYtmp / sigmaXYtmp
                 sumtmp += sigmaXErrtmp * sigmaXErrtmp / sigmaXtmp / sigmaXtmp
                 sumtmp += sigmaYErrtmp * sigmaYErrtmp / sigmaYtmp / sigmaYtmp
                 rhoXYErrtmp = sqrt(rhoXYtmp * rhoXYtmp * sumtmp)
             except Exception:
                 rhoXYErrtmp = 0
  
             bs.rhoXYErr = rhoXYErrtmp
  
             yield bs
  

◆ cutSummary()

def 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
  
  
  
 ROOT.gROOT.ProcessLine(BeamSpotValue(fullCorrelations=True,addScanVars=True).getROOTStruct())

◆ getDataCache()

def 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()

def 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()

def 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()

def 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()

def python.BeamSpotData.BeamSpotCOOL.summary ( self )

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

Reimplemented from python.BeamSpotData.BeamSpotContainer.

Definition at line 1213 of file BeamSpotData.py.

     def summary(self):
         s = 'COOL database %s, tag %s:\n' % (self.database,self.tag)
         s += '    %s ntuple, COOL folder %s\n' % (self.__class__.__name__,self.folder)
         return s
  

Member Data Documentation

◆ acqFlag

python.BeamSpotData.BeamSpotContainer.acqFlag

inherited

Definition at line 765 of file BeamSpotData.py.

◆ bcidMax

python.BeamSpotData.BeamSpotContainer.bcidMax

inherited

Definition at line 759 of file BeamSpotData.py.

◆ bcidMin

python.BeamSpotData.BeamSpotContainer.bcidMin

inherited

Definition at line 758 of file BeamSpotData.py.

◆ cooldb

python.BeamSpotData.BeamSpotCOOL.cooldb

Definition at line 1137 of file BeamSpotData.py.

◆ database

python.BeamSpotData.BeamSpotCOOL.database

Definition at line 1133 of file BeamSpotData.py.

◆ fillMax

python.BeamSpotData.BeamSpotContainer.fillMax

inherited

Definition at line 757 of file BeamSpotData.py.

◆ fillMin

python.BeamSpotData.BeamSpotContainer.fillMin

inherited

Definition at line 756 of file BeamSpotData.py.

◆ folder

python.BeamSpotData.BeamSpotCOOL.folder

Definition at line 1135 of file BeamSpotData.py.

◆ grl

python.BeamSpotData.BeamSpotContainer.grl

inherited

Definition at line 766 of file BeamSpotData.py.

◆ grlIOVs

python.BeamSpotData.BeamSpotContainer.grlIOVs

inherited

Definition at line 767 of file BeamSpotData.py.

◆ iter

python.BeamSpotData.BeamSpotContainer.iter

inherited

Definition at line 771 of file BeamSpotData.py.

◆ lbMax

python.BeamSpotData.BeamSpotContainer.lbMax

inherited

Definition at line 761 of file BeamSpotData.py.

◆ lbMin

python.BeamSpotData.BeamSpotContainer.lbMin

inherited

Definition at line 760 of file BeamSpotData.py.

◆ nSel

python.BeamSpotData.BeamSpotContainer.nSel

inherited

Definition at line 778 of file BeamSpotData.py.

◆ nTot

python.BeamSpotData.BeamSpotContainer.nTot

inherited

Definition at line 777 of file BeamSpotData.py.

◆ previousGRLIndex

python.BeamSpotData.BeamSpotContainer.previousGRLIndex

inherited

Definition at line 768 of file BeamSpotData.py.

◆ runList

python.BeamSpotData.BeamSpotContainer.runList

inherited

Definition at line 754 of file BeamSpotData.py.

◆ runListExclude

python.BeamSpotData.BeamSpotContainer.runListExclude

inherited

Definition at line 755 of file BeamSpotData.py.

◆ runMax

python.BeamSpotData.BeamSpotContainer.runMax

inherited

Definition at line 753 of file BeamSpotData.py.

◆ runMin

python.BeamSpotData.BeamSpotContainer.runMin

inherited

Definition at line 752 of file BeamSpotData.py.

◆ selBcidMax

python.BeamSpotData.BeamSpotContainer.selBcidMax

inherited

Definition at line 784 of file BeamSpotData.py.

◆ selBcidMin

python.BeamSpotData.BeamSpotContainer.selBcidMin

inherited

Definition at line 783 of file BeamSpotData.py.

◆ selFillMax

python.BeamSpotData.BeamSpotContainer.selFillMax

inherited

Definition at line 782 of file BeamSpotData.py.

◆ selFillMin

python.BeamSpotData.BeamSpotContainer.selFillMin

inherited

Definition at line 781 of file BeamSpotData.py.

◆ selRunMax

python.BeamSpotData.BeamSpotContainer.selRunMax

inherited

Definition at line 780 of file BeamSpotData.py.

◆ selRunMin

python.BeamSpotData.BeamSpotContainer.selRunMin

inherited

Definition at line 779 of file BeamSpotData.py.

◆ selTimeMax

python.BeamSpotData.BeamSpotContainer.selTimeMax

inherited

Definition at line 786 of file BeamSpotData.py.

◆ selTimeMin

python.BeamSpotData.BeamSpotContainer.selTimeMin

inherited

Definition at line 785 of file BeamSpotData.py.

◆ statusList

python.BeamSpotData.BeamSpotContainer.statusList

inherited

Definition at line 764 of file BeamSpotData.py.

◆ tag

python.BeamSpotData.BeamSpotCOOL.tag

Definition at line 1134 of file BeamSpotData.py.

◆ timeMax

python.BeamSpotData.BeamSpotContainer.timeMax

inherited

Definition at line 763 of file BeamSpotData.py.

◆ timeMin

python.BeamSpotData.BeamSpotContainer.timeMin

inherited

Definition at line 762 of file BeamSpotData.py.

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

BeamSpotData.py

Public Member Functions

Public Attributes

Detailed Description

Constructor & Destructor Documentation

◆ __init__()

◆ __del__()

Member Function Documentation

◆ __iter__()

◆ __next__()

◆ allData()

◆ cutSummary()

◆ getDataCache()

◆ initStatistics()

◆ next()

◆ selectedData()

◆ summary()

Member Data Documentation

◆ acqFlag

◆ bcidMax

◆ bcidMin

◆ cooldb

◆ database

◆ fillMax

◆ fillMin

◆ folder

◆ grl

◆ grlIOVs

◆ iter

◆ lbMax

◆ lbMin

◆ nSel

◆ nTot

◆ previousGRLIndex

◆ runList

◆ runListExclude

◆ runMax

◆ runMin

◆ selBcidMax

◆ selBcidMin

◆ selFillMax

◆ selFillMin

◆ selRunMax

◆ selRunMin

◆ selTimeMax

◆ selTimeMin

◆ statusList

◆ tag

◆ timeMax

◆ timeMin

◆ init()

◆ del()

◆ iter()

◆ next()