objects.METAnalyzer Namespace Reference

Classes
class	METAnalyzer

Functions
def	makeGenTkMet (self, event)
def	makeMETNoEle (self, event)
def	makeMETNoMu (self, event)
def	makeMETNoPhoton (self, event)
def	makeMETs (self, event)
def	process (self, event)
def	sumXY (pxys)

Variables
	applyJetSmearing
	candidates
	candidatesTypes
	class_object
	collectionPostFix
	copyMETsByValue
	doMetNoEle
	doMetNoMu
	doMetNoPhoton
	doMetNoPU
	doTkMet
	dzMax
	includeTkMetCHS
	includeTkMetPVLoose
	includeTkMetPVTight
	jetAnalyzerPostFix
	makeShiftedMETs
	met
	met_raw
	met_sig
	met_sumet
	metCollection
	metNoEle
	metNoEleNoPU
	metNoMu
	metNoMuNoPU
	metNoPhoton
	metNoPhotonNoPU
	metNoPU
	noPUMetCollection
	old74XMiniAODs
	pvflag
	===> require the Track Candidate charge and with a minimum dz More...
	pxy
	recalibrate
	recalibrateMET
	sumEt

Function Documentation

makeGenTkMet()

def objects.METAnalyzer.makeGenTkMet	(		self,
			event
	)

Definition at line 109 of file METAnalyzer.py.

References funct.abs().

    def makeGenTkMet(self, event):
        genCharged = [ (x.px(),x.py()) for x in self.mchandles['packedGen'].product() if x.charge() != 0 and abs(x.eta()) < 2.4 ]
        px, py = sum(x[0] for x in genCharged), sum(x[1] for x in genCharged)
        setattr(event,"tkGenMet"+self.cfg_ana.collectionPostFix, ROOT.reco.Particle.LorentzVector(-px , -py, 0, hypot(px,py)))

makeMETNoEle()

def objects.METAnalyzer.makeMETNoEle	(		self,
			event
	)

Definition at line 134 of file METAnalyzer.py.

    def makeMETNoEle(self, event):
        self.metNoEle = copy.deepcopy(self.met)
        if self.cfg_ana.doMetNoPU: self.metNoEleNoPU = copy.deepcopy(self.metNoPU)

        elepx = 0
        elepy = 0
        #sum electron momentum
        for ele in event.selectedElectrons:
            elepx += ele.px()
            elepy += ele.py()

        #subtract electron momentum and construct met
        px,py = self.metNoEle.px()+elepx, self.metNoEle.py()+elepy
        self.metNoEle.setP4(ROOT.reco.Particle.LorentzVector(px,py, 0, hypot(px,py)))

        px,py = self.metNoEleNoPU.px()+elepx, self.metNoEleNoPU.py()+elepy
        self.metNoEleNoPU.setP4(ROOT.reco.Particle.LorentzVector(px,py, 0, hypot(px,py)))
        setattr(event, "metNoEle"+self.cfg_ana.collectionPostFix, self.metNoEle)
        if self.cfg_ana.doMetNoPU: setattr(event, "metNoEleNoPU"+self.cfg_ana.collectionPostFix, self.metNoEleNoPU)

makeMETNoMu()

def objects.METAnalyzer.makeMETNoMu	(		self,
			event
	)

Definition at line 114 of file METAnalyzer.py.

    def makeMETNoMu(self, event):
        self.metNoMu = copy.deepcopy(self.met)
        if self.cfg_ana.doMetNoPU: self.metNoMuNoPU = copy.deepcopy(self.metNoPU)

        mupx = 0
        mupy = 0
        #sum muon momentum
        for mu in event.selectedMuons:
            mupx += mu.px()
            mupy += mu.py()

        #subtract muon momentum and construct met
        px,py = self.metNoMu.px()+mupx, self.metNoMu.py()+mupy
        self.metNoMu.setP4(ROOT.reco.Particle.LorentzVector(px,py, 0, hypot(px,py)))
        px,py = self.metNoMuNoPU.px()+mupx, self.metNoMuNoPU.py()+mupy
        self.metNoMuNoPU.setP4(ROOT.reco.Particle.LorentzVector(px,py, 0, hypot(px,py)))
        setattr(event, "metNoMu"+self.cfg_ana.collectionPostFix, self.metNoMu)
        if self.cfg_ana.doMetNoPU: setattr(event, "metNoMuNoPU"+self.cfg_ana.collectionPostFix, self.metNoMuNoPU)

makeMETNoPhoton()

def objects.METAnalyzer.makeMETNoPhoton	(		self,
			event
	)

Definition at line 154 of file METAnalyzer.py.

    def makeMETNoPhoton(self, event):
        self.metNoPhoton = copy.deepcopy(self.met)

        phopx = 0
        phopy = 0
        #sum photon momentum
        for pho in event.selectedPhotons:
            phopx += pho.px()
            phopy += pho.py()

        #subtract photon momentum and construct met
        px,py = self.metNoPhoton.px()+phopx, self.metNoPhoton.py()+phopy
        self.metNoPhoton.setP4(ROOT.reco.Particle.LorentzVector(px,py, 0, hypot(px,py)))
        setattr(event, "metNoPhoton"+self.cfg_ana.collectionPostFix, self.metNoPhoton)
        if self.cfg_ana.doMetNoPU: 
          self.metNoPhotonNoPU = copy.deepcopy(self.metNoPU)
          px,py = self.metNoPhotonNoPU.px()+phopx, self.metNoPhotonNoPU.py()+phopy
          self.metNoPhotonNoPU.setP4(ROOT.reco.Particle.LorentzVector(px,py, 0, hypot(px,py)))
          setattr(event, "metNoPhotonNoPU"+self.cfg_ana.collectionPostFix, self.metNoPhotonNoPU)

makeMETs()

def objects.METAnalyzer.makeMETs	(		self,
			event
	)

Definition at line 175 of file METAnalyzer.py.

    def makeMETs(self, event):
        import ROOT
        if self.cfg_ana.copyMETsByValue:
          self.met = ROOT.pat.MET(self.handles['met'].product()[0])
          if self.cfg_ana.doMetNoPU: self.metNoPU = ROOT.pat.MET(self.handles['nopumet'].product()[0])
        else:
          self.met = self.handles['met'].product()[0]
          if self.cfg_ana.doMetNoPU: self.metNoPU = self.handles['nopumet'].product()[0]

        if self.recalibrateMET == "type1":
          type1METCorr = getattr(event, 'type1METCorr'+self.jetAnalyzerPostFix)
          self.type1METCorrector.correct(self.met, type1METCorr)
        elif self.recalibrateMET == True:
          deltaMetJEC = getattr(event, 'deltaMetFromJEC'+self.jetAnalyzerPostFix)
          self.applyDeltaMet(self.met, deltaMetJEC)
        if self.applyJetSmearing:
          deltaMetSmear = getattr(event, 'deltaMetFromJetSmearing'+self.jetAnalyzerPostFix)
          self.applyDeltaMet(self.met, deltaMetSmear)


        if (not self.cfg_ana.copyMETsByValue) and getattr(self.cfg_ana, 'makeShiftedMETs', True):
          shifts = [] 
          for obj in 'JetEn', 'JetRes', 'MuonEn', 'ElectronEn', 'PhotonEn', 'TauEn', 'UnclusteredEn':
            for sh in 'Up', 'Down':
                shifts.append( (obj+sh, getattr(self.met,obj+sh)) )
          shifts.append( ('NoShift', self.met.NoShift) )
          for name,i in shifts:
               key = i
               m = ROOT.pat.MET(self.met)
               if self.old74XMiniAODs:
                    if key > 12:   key = 12
                    elif key <= 3: key = { 'JetEnUp':0, 'JetEnDown':1, 'JetResUp':2, 'JetResDown':3 }[name]
               m.setP4(self.met.shiftedP4(key))
               setattr(event, "met{0}_shifted_{1}".format(self.cfg_ana.collectionPostFix, i),m)
               setattr(event, "met{0}_shifted_{1}".format(self.cfg_ana.collectionPostFix, name),m)

        self.met_sig = self.met.significance()
        self.met_sumet = self.met.sumEt()

        if self.old74XMiniAODs and self.recalibrateMET != "type1":
           oldraw = self.met.shiftedP2_74x(12,0);
           setFakeRawMETOnOldMiniAODs( self.met, oldraw.px, oldraw.py, self.met.shiftedSumEt_74x(12,0) )
           px, py = oldraw.px, oldraw.py
        else:
           px, py = self.met.uncorPx(), self.met.uncorPy()
        self.met_raw = ROOT.reco.Particle.LorentzVector(px,py,0,hypot(px,py))

        if hasattr(event,'zll_p4'):
            self.adduParaPerp(self.met,event.zll_p4,"_zll")
            self.adduParaPerp(self.met_raw, event.zll_p4,"_zll")
            setattr(event,"met_raw"+self.cfg_ana.collectionPostFix, self.met_raw)
            setattr(event,"met_raw.upara_zll"+self.cfg_ana.collectionPostFix, self.met_raw.upara_zll)
            setattr(event,"met_raw.uperp_zll"+self.cfg_ana.collectionPostFix, self.met_raw.uperp_zll)

        if hasattr(event,'gamma_p4'):
            self.adduParaPerp(self.met,event.gamma_p4,"_gamma")
            self.adduParaPerp(self.met_raw, event.gamma_p4,"_gamma")
            setattr(event,"met_raw"+self.cfg_ana.collectionPostFix, self.met_raw)
            setattr(event,"met_raw.upara_gamma"+self.cfg_ana.collectionPostFix, self.met_raw.upara_gamma)
            setattr(event,"met_raw.uperp_gamma"+self.cfg_ana.collectionPostFix, self.met_raw.uperp_gamma)

        if hasattr(event,"met"+self.cfg_ana.collectionPostFix): raise RuntimeError("Event already contains met with the following postfix: "+self.cfg_ana.collectionPostFix)
        setattr(event, "met"+self.cfg_ana.collectionPostFix, self.met)
        if self.cfg_ana.doMetNoPU: setattr(event, "metNoPU"+self.cfg_ana.collectionPostFix, self.metNoPU)
        setattr(event, "met_sig"+self.cfg_ana.collectionPostFix, self.met_sig)
        setattr(event, "met_sumet"+self.cfg_ana.collectionPostFix, self.met_sumet)

        genMET = self.met.genMET()
        if genMET:
          setattr(event, "met_genPt"+self.cfg_ana.collectionPostFix, genMET.pt())
          setattr(event, "met_genPhi"+self.cfg_ana.collectionPostFix, genMET.phi())
        else:
          setattr(event, "met_genPt"+self.cfg_ana.collectionPostFix, float('nan'))
          setattr(event, "met_genPhi"+self.cfg_ana.collectionPostFix, float('nan'))

        if self.cfg_ana.doMetNoMu and hasattr(event, 'selectedMuons'):
            self.makeMETNoMu(event)

        if self.cfg_ana.doMetNoEle and hasattr(event, 'selectedElectrons'):
            self.makeMETNoEle(event)

        if self.cfg_ana.doMetNoPhoton and hasattr(event, 'selectedPhotons'):
            self.makeMETNoPhoton(event)

process()

def objects.METAnalyzer.process	(		self,
			event
	)

Definition at line 259 of file METAnalyzer.py.

    def process(self, event):
        self.readCollections( event.input)
        self.counters.counter('events').inc('all events')

        self.makeMETs(event)

        if self.cfg_ana.doTkMet: 
            self.makeTkMETs(event);

        if getattr(self.cfg_ana,"doTkGenMet",self.cfg_ana.doTkMet) and self.cfg_comp.isMC and hasattr(event, 'genParticles'):
            self.makeGenTkMet(event)

        return True


setattr(METAnalyzer,"defaultConfig", cfg.Analyzer(

sumXY()

def objects.METAnalyzer.sumXY

(

pxys

)

Definition at line 90 of file METAnalyzer.py.

        def sumXY(pxys):
            px, py = sum(x[0] for x in pxys), sum(x[1] for x in pxys)
            return ROOT.reco.Particle.LorentzVector(-px, -py, 0, hypot(px,py))

Variable Documentation

applyJetSmearing

objects.METAnalyzer.applyJetSmearing

Definition at line 280 of file METAnalyzer.py.

candidates

objects.METAnalyzer.candidates

Definition at line 292 of file METAnalyzer.py.

candidatesTypes

objects.METAnalyzer.candidatesTypes

Definition at line 293 of file METAnalyzer.py.

class_object

objects.METAnalyzer.class_object

Definition at line 275 of file METAnalyzer.py.

collectionPostFix

objects.METAnalyzer.collectionPostFix

Definition at line 295 of file METAnalyzer.py.

copyMETsByValue

objects.METAnalyzer.copyMETsByValue

Definition at line 278 of file METAnalyzer.py.

doMetNoEle

objects.METAnalyzer.doMetNoEle

Definition at line 290 of file METAnalyzer.py.

doMetNoMu

objects.METAnalyzer.doMetNoMu

Definition at line 289 of file METAnalyzer.py.

doMetNoPhoton

objects.METAnalyzer.doMetNoPhoton

Definition at line 291 of file METAnalyzer.py.

doMetNoPU

objects.METAnalyzer.doMetNoPU

Definition at line 288 of file METAnalyzer.py.

doTkMet

objects.METAnalyzer.doTkMet

Definition at line 284 of file METAnalyzer.py.

dzMax

objects.METAnalyzer.dzMax

Definition at line 294 of file METAnalyzer.py.

includeTkMetCHS

objects.METAnalyzer.includeTkMetCHS

Definition at line 285 of file METAnalyzer.py.

includeTkMetPVLoose

objects.METAnalyzer.includeTkMetPVLoose

Definition at line 286 of file METAnalyzer.py.

includeTkMetPVTight

objects.METAnalyzer.includeTkMetPVTight

Definition at line 287 of file METAnalyzer.py.

jetAnalyzerPostFix

objects.METAnalyzer.jetAnalyzerPostFix

Definition at line 281 of file METAnalyzer.py.

makeShiftedMETs

objects.METAnalyzer.makeShiftedMETs

Definition at line 283 of file METAnalyzer.py.

met

objects.METAnalyzer.met

Definition at line 178 of file METAnalyzer.py.

met_raw

objects.METAnalyzer.met_raw

Definition at line 220 of file METAnalyzer.py.

met_sig

objects.METAnalyzer.met_sig

Definition at line 211 of file METAnalyzer.py.

met_sumet

objects.METAnalyzer.met_sumet

Definition at line 212 of file METAnalyzer.py.

metCollection

objects.METAnalyzer.metCollection

Definition at line 276 of file METAnalyzer.py.

metNoEle

objects.METAnalyzer.metNoEle

Definition at line 135 of file METAnalyzer.py.

metNoEleNoPU

objects.METAnalyzer.metNoEleNoPU

Definition at line 136 of file METAnalyzer.py.

metNoMu

objects.METAnalyzer.metNoMu

Definition at line 115 of file METAnalyzer.py.

metNoMuNoPU

objects.METAnalyzer.metNoMuNoPU

Definition at line 116 of file METAnalyzer.py.

metNoPhoton

objects.METAnalyzer.metNoPhoton

Definition at line 155 of file METAnalyzer.py.

metNoPhotonNoPU

objects.METAnalyzer.metNoPhotonNoPU

Definition at line 169 of file METAnalyzer.py.

metNoPU

objects.METAnalyzer.metNoPU

Definition at line 179 of file METAnalyzer.py.

noPUMetCollection

objects.METAnalyzer.noPUMetCollection

Definition at line 277 of file METAnalyzer.py.

old74XMiniAODs

objects.METAnalyzer.old74XMiniAODs

Definition at line 282 of file METAnalyzer.py.

pvflag

objects.METAnalyzer.pvflag

===> require the Track Candidate charge and with a minimum dz

Definition at line 75 of file METAnalyzer.py.

pxy

objects.METAnalyzer.pxy

Definition at line 76 of file METAnalyzer.py.

Referenced by L1MetPfProducer.Project().

recalibrate

objects.METAnalyzer.recalibrate

Definition at line 279 of file METAnalyzer.py.

recalibrateMET

objects.METAnalyzer.recalibrateMET

Definition at line 184 of file METAnalyzer.py.

sumEt

objects.METAnalyzer.sumEt

Definition at line 97 of file METAnalyzer.py.

Referenced by MBUEandQCDValidation.analyze(), l1t::Stage1Layer2EtSumAlgorithmImpHW.doSumAndMET(), l1t::Stage1Layer2EtSumAlgorithmImpHI.doSumAndMET(), TriggerSummaryProducerAOD.fillTriggerObject(), TCMETAlgo.initialize_MET_with_CaloMET(), pf2pat::PFMETAlgo.produce(), SysShiftMETcorrInputProducer.produce(), l1t::GenToInputProducer.produce(), CATopJetAlgorithm.run(), PrimaryVertexSorting.score(), and FWRPZView.voteCaloMaxVal().