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.

     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)
  

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

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

◆ process()

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

Definition at line 42 of file GenHeavyFlavourAnalyzer.py.

     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
  