gen.GenHeavyFlavourAnalyzer.GenHeavyFlavourAnalyzer Class Reference

Inheritance diagram for gen.GenHeavyFlavourAnalyzer.GenHeavyFlavourAnalyzer:

Public Member Functions
def	__init__ (self, cfg_ana, cfg_comp, looperName)
def	beginLoop (self, setup)
def	declareHandles (self)
def	makeBPartons (self, event)
def	process (self, event)

Public Attributes
	bquarkPtCut
	status2Only

Detailed Description

   Makes:
      event.genallbquarks, event.genallcquarks:
            list of all b and c quarks (without doublecounting of b->b and c->c chains).
            if status2Only == True, only status 2 ones are included.
      event.allBPartons
            all status 2 b-quarks, sorted by pt decreasingly
      event.bPartons:
            status 2 b-quarks passing a pt cut bquarkPtCut (default: 15)

   Requires:
      event.genParticles      

Definition at line 4 of file GenHeavyFlavourAnalyzer.py.

Constructor & Destructor Documentation

__init__()

def gen.GenHeavyFlavourAnalyzer.GenHeavyFlavourAnalyzer.__init__	(		self,
			cfg_ana,
			cfg_comp,
			looperName
	)

Definition at line 19 of file GenHeavyFlavourAnalyzer.py.

    def __init__(self, cfg_ana, cfg_comp, looperName ):
        super(GenHeavyFlavourAnalyzer,self).__init__(cfg_ana,cfg_comp,looperName)
        self.status2Only = cfg_ana.status2Only 
        self.bquarkPtCut = cfg_ana.bquarkPtCut
 

Member Function Documentation

beginLoop()

def gen.GenHeavyFlavourAnalyzer.GenHeavyFlavourAnalyzer.beginLoop	(		self,
			setup
	)

Definition at line 27 of file GenHeavyFlavourAnalyzer.py.

    def beginLoop(self,setup):
        super(GenHeavyFlavourAnalyzer,self).beginLoop(setup)

declareHandles()

def gen.GenHeavyFlavourAnalyzer.GenHeavyFlavourAnalyzer.declareHandles

(

self

)

Definition at line 24 of file GenHeavyFlavourAnalyzer.py.

    def declareHandles(self):
        super(GenHeavyFlavourAnalyzer, self).declareHandles()

makeBPartons()

def gen.GenHeavyFlavourAnalyzer.GenHeavyFlavourAnalyzer.makeBPartons	(		self,
			event
	)

Definition at line 30 of file GenHeavyFlavourAnalyzer.py.

References funct.abs(), gen.GenHeavyFlavourAnalyzer.GenHeavyFlavourAnalyzer.bquarkPtCut, and PbPb_ZMuSkimMuonDPG_cff.deltaR.

Referenced by gen.GenHeavyFlavourAnalyzer.GenHeavyFlavourAnalyzer.process().

    def makeBPartons(self,event):
        event.allBPartons = [ q for q in event.genParticles if abs(q.pdgId()) == 5 and abs(q.status()) == 2 and abs(q.pt()) > self.bquarkPtCut ]
        event.allBPartons.sort(key = lambda q : q.pt(), reverse = True)
        event.bPartons = []
        for q in event.allBPartons:
            duplicate = False
            for q2 in event.bPartons:
                if deltaR(q.eta(),q.phi(),q2.eta(),q2.phi()) < 0.5:
                    duplicate = True
            continue
            if not duplicate: event.bPartons.append(q)

process()

def gen.GenHeavyFlavourAnalyzer.GenHeavyFlavourAnalyzer.process	(		self,
			event
	)

Definition at line 42 of file GenHeavyFlavourAnalyzer.py.

References funct.abs(), analyzer.Analyzer.cfg_comp, looper.Looper.cfg_comp, gen.GenHeavyFlavourAnalyzer.GenHeavyFlavourAnalyzer.makeBPartons(), EgammaValidation_cff.pdgId, core.Analyzer.Analyzer.readCollections(), and gen.GenHeavyFlavourAnalyzer.GenHeavyFlavourAnalyzer.status2Only.

    def process(self, event):
        self.readCollections( event.input )

        # if not MC, nothing to do
        if not self.cfg_comp.isMC: 
            return True

        status2f = (lambda p : p.status() == 2) if self.status2Only else (lambda p : True)
        event.genallcquarks = [ p for p in event.genParticles if abs(p.pdgId()) == 5 and ( p.numberOfDaughters() == 0 or abs(p.daughter(0).pdgId()) != 5) and status2f(p) ]
        event.genallbquarks = [ p for p in event.genParticles if abs(p.pdgId()) == 4 and ( p.numberOfDaughters() == 0 or abs(p.daughter(0).pdgId()) != 4) and status2f(p) ]

        self.makeBPartons(event)

        return True

Member Data Documentation

bquarkPtCut

gen.GenHeavyFlavourAnalyzer.GenHeavyFlavourAnalyzer.bquarkPtCut

Definition at line 22 of file GenHeavyFlavourAnalyzer.py.

Referenced by gen.GenHeavyFlavourAnalyzer.GenHeavyFlavourAnalyzer.makeBPartons().

status2Only

gen.GenHeavyFlavourAnalyzer.GenHeavyFlavourAnalyzer.status2Only

Definition at line 21 of file GenHeavyFlavourAnalyzer.py.

Referenced by gen.GenHeavyFlavourAnalyzer.GenHeavyFlavourAnalyzer.process().