Collaboration diagram for python.BeamSpotData.BeamSpotAverage:

Public Member Functions
	__init__ (self, varList=None, weightedAverage=True)
	readLumiData (self, lumiCalcNtupleName)
	add (self, b)
	average (self)
	getIndex (self, varName)

Public Attributes
list	varList = varList
	weightedAverage = weightedAverage
	sumw = numpy.zeros(len(self.varList))
	sumwx = numpy.zeros(len(self.varList))
	sumwxx = numpy.zeros(len(self.varList))
	sumwwee = numpy.zeros(len(self.varList))
	sum = numpy.zeros(len(self.varList))
	sumx = numpy.zeros(len(self.varList))
	sumxx = numpy.zeros(len(self.varList))
int	nWarnings = 0
dict	lumiData = None
	rms = numpy.sqrt(self.sumxx/self.sum - self.sumx*self.sumx/self.sum/self.sum)
	ave = self.sumwx/self.sumw
int	err = 1./numpy.sqrt(self.sumw)

Detailed Description

A utility class for averaging beam spot data.

Definition at line 590 of file BeamSpotData.py.

Constructor & Destructor Documentation

◆ init()

python.BeamSpotData.BeamSpotAverage.__init__	(	self,
		varList = None,
		weightedAverage = True )

Definition at line 593 of file BeamSpotData.py.

    def __init__(self,varList=None,weightedAverage=True):
        if varList:
            self.varList = varList
        else:
            self.varList = ['posX','posY','posZ','sigmaX','sigmaY','sigmaZ','tiltX','tiltY','rhoXY','k']
        self.weightedAverage = weightedAverage
        self.sumw = numpy.zeros(len(self.varList))
        self.sumwx = numpy.zeros(len(self.varList))
        self.sumwxx = numpy.zeros(len(self.varList))
        self.sumwwee = numpy.zeros(len(self.varList))
        # Unweighted quantities always needed for RMS
        self.sum = numpy.zeros(len(self.varList))
        self.sumx = numpy.zeros(len(self.varList))
        self.sumxx = numpy.zeros(len(self.varList))
        self.nWarnings = 0
        self.lumiData = None   # luminosity data in ubarn, indexed by [run][LB]
 

Member Function Documentation

◆ add()

python.BeamSpotData.BeamSpotAverage.add	(		self,
			b )

Add elements of BeamSpotValue b to average.

Definition at line 633 of file BeamSpotData.py.

    def add(self,b):
        """Add elements of BeamSpotValue b to average."""
        if self.lumiData is not None:
            lumi = 0
            if b.lbEnd <= b.lbStart:
                print ('ERROR: Illegal luminosity block range: [%i,%i]' % (b.lbStart,b.lbEnd))
                self.nWarnings += 1
            for lb in range(b.lbStart,b.lbEnd):
                try:
                    lumi += self.lumiData[b.run][lb]
                except Exception:
                    print ('ERROR: missing luminosity information for run %i LB %i (in [%i,%i]) - weight set to zero!!' % (b.run,lb,b.lbStart,b.lbEnd))
                    self.nWarnings += 1
        for i in range(len(self.varList)):
            parName = self.varList[i]
            val = getattr(b,parName)
            valErr = getattr(b,parName+'Err')
            self.sum[i] += 1.
            self.sumx[i] += val
            self.sumxx[i] += val*val
            if self.lumiData is None:
                if valErr != 0. :
                    w = 1./valErr/valErr
                else:
                    w = 1e-16
                    print ('WARNING: Divison by zero for parameter %s   (val = %f  valErr = %f)\n' % (parName,val,valErr))
                    self.nWarnings += 1
            else:
                w = lumi
            self.sumw[i] += w
            self.sumwx[i] += w*val
            self.sumwxx[i] += w*val*val
            self.sumwwee[i] += w*w*valErr*valErr
 

◆ average()

python.BeamSpotData.BeamSpotAverage.average ( self )

Definition at line 667 of file BeamSpotData.py.

    def average(self):
        self.rms = numpy.sqrt(self.sumxx/self.sum - self.sumx*self.sumx/self.sum/self.sum)   # factor n/n-1 ?
        if self.weightedAverage:
            if self.lumiData is None:
                self.ave = self.sumwx/self.sumw
                self.err = 1./numpy.sqrt(self.sumw)
            else:
                self.ave = self.sumwx/self.sumw
                self.err = numpy.sqrt(self.sumwwee)/self.sumw
        else:
            self.ave = self.sumx/self.sum
            self.err = self.rms/numpy.sqrt(self.sum)
 

◆ getIndex()

python.BeamSpotData.BeamSpotAverage.getIndex	(		self,
			varName )

Definition at line 680 of file BeamSpotData.py.

    def getIndex(self, varName):
        for i in range(len(self.varList)):
            if self.varList[i]==varName:
                return i
        return None
 
 
 

◆ readLumiData()

python.BeamSpotData.BeamSpotAverage.readLumiData	(		self,
			lumiCalcNtupleName )

Read iLumiCalc.exe ntuple with luminosity data, store data in self.lumiData dict.

Definition at line 610 of file BeamSpotData.py.

    def readLumiData(self,lumiCalcNtupleName):
        """Read iLumiCalc.exe ntuple with luminosity data, store data in self.lumiData dict."""
        lumiFile = ROOT.TFile(lumiCalcNtupleName)
        lumiNt = lumiFile.Get('LumiMetaData')
        print ('Reading',lumiNt.GetEntries(), 'entries from luminosity ntuple',lumiCalcNtupleName)
        self.lumiData = {}
        for j in range(lumiNt.GetEntries()):
            lumiNt.GetEntry(j)
            run = lumiNt.RunNbr
            lb = lumiNt.LBStart
            lumi = lumiNt.IntLumi
            try:
                self.lumiData[run][lb] = lumi
            except Exception:
                self.lumiData[run] = {}
                self.lumiData[run][lb] = lumi
        lumiSum = 0.
        for (run,lbdict) in self.lumiData.items():
            for (lb,lumi) in lbdict.items():
                lumiSum += lumi
        print ('... total luminosity = %6.1f / pb' % (lumiSum/1.E6))
        return self.lumiData
 

Member Data Documentation

◆ ave

python.BeamSpotData.BeamSpotAverage.ave = self.sumwx/self.sumw

Definition at line 671 of file BeamSpotData.py.

◆ err

int python.BeamSpotData.BeamSpotAverage.err = 1./numpy.sqrt(self.sumw)

Definition at line 672 of file BeamSpotData.py.

◆ lumiData

dict python.BeamSpotData.BeamSpotAverage.lumiData = None

Definition at line 608 of file BeamSpotData.py.

◆ nWarnings

int python.BeamSpotData.BeamSpotAverage.nWarnings = 0

Definition at line 607 of file BeamSpotData.py.

◆ rms

python.BeamSpotData.BeamSpotAverage.rms = numpy.sqrt(self.sumxx/self.sum - self.sumx*self.sumx/self.sum/self.sum)

Definition at line 668 of file BeamSpotData.py.

◆ sum

python.BeamSpotData.BeamSpotAverage.sum = numpy.zeros(len(self.varList))

Definition at line 604 of file BeamSpotData.py.

◆ sumw

python.BeamSpotData.BeamSpotAverage.sumw = numpy.zeros(len(self.varList))

Definition at line 599 of file BeamSpotData.py.

◆ sumwwee

python.BeamSpotData.BeamSpotAverage.sumwwee = numpy.zeros(len(self.varList))

Definition at line 602 of file BeamSpotData.py.

◆ sumwx

python.BeamSpotData.BeamSpotAverage.sumwx = numpy.zeros(len(self.varList))

Definition at line 600 of file BeamSpotData.py.

◆ sumwxx

python.BeamSpotData.BeamSpotAverage.sumwxx = numpy.zeros(len(self.varList))

Definition at line 601 of file BeamSpotData.py.

◆ sumx

python.BeamSpotData.BeamSpotAverage.sumx = numpy.zeros(len(self.varList))

Definition at line 605 of file BeamSpotData.py.

◆ sumxx

python.BeamSpotData.BeamSpotAverage.sumxx = numpy.zeros(len(self.varList))

Definition at line 606 of file BeamSpotData.py.

◆ varList

python.BeamSpotData.BeamSpotAverage.varList = varList

Definition at line 595 of file BeamSpotData.py.

◆ weightedAverage

python.BeamSpotData.BeamSpotAverage.weightedAverage = weightedAverage

Definition at line 598 of file BeamSpotData.py.

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

BeamSpotData.py

Public Member Functions

Public Attributes