python.BeamSpotData.BeamSpotAverage Class Reference

Collaboration diagram for python.BeamSpotData.BeamSpotAverage:

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

Public Attributes
	varList
	weightedAverage
	sumw
	sumwx
	sumwxx
	sumwwee
	sum
	sumx
	sumxx
	nWarnings
	lumiData
	rms
	ave
	err

Detailed Description

A utility class for averaging beam spot data.

Definition at line 590 of file BeamSpotData.py.

Constructor & Destructor Documentation

__init__()

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

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

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

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

def 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

Definition at line 671 of file BeamSpotData.py.

err

python.BeamSpotData.BeamSpotAverage.err

Definition at line 672 of file BeamSpotData.py.

lumiData

python.BeamSpotData.BeamSpotAverage.lumiData

Definition at line 608 of file BeamSpotData.py.

nWarnings

python.BeamSpotData.BeamSpotAverage.nWarnings

Definition at line 607 of file BeamSpotData.py.

rms

python.BeamSpotData.BeamSpotAverage.rms

Definition at line 668 of file BeamSpotData.py.

sum

python.BeamSpotData.BeamSpotAverage.sum

Definition at line 604 of file BeamSpotData.py.

sumw

python.BeamSpotData.BeamSpotAverage.sumw

Definition at line 599 of file BeamSpotData.py.

sumwwee

python.BeamSpotData.BeamSpotAverage.sumwwee

Definition at line 602 of file BeamSpotData.py.

sumwx

python.BeamSpotData.BeamSpotAverage.sumwx

Definition at line 600 of file BeamSpotData.py.

sumwxx

python.BeamSpotData.BeamSpotAverage.sumwxx

Definition at line 601 of file BeamSpotData.py.

sumx

python.BeamSpotData.BeamSpotAverage.sumx

Definition at line 605 of file BeamSpotData.py.

sumxx

python.BeamSpotData.BeamSpotAverage.sumxx

Definition at line 606 of file BeamSpotData.py.

varList

python.BeamSpotData.BeamSpotAverage.varList

Definition at line 595 of file BeamSpotData.py.

weightedAverage

python.BeamSpotData.BeamSpotAverage.weightedAverage

Definition at line 598 of file BeamSpotData.py.

---

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

• BeamSpotData.py