python.ParticleDecayDrawer.ParticleDecayDrawer Class Reference

Inheritance diagram for python.ParticleDecayDrawer.ParticleDecayDrawer:

Public Member Functions
def	__init__ (self)
def	draw (self, particles)

Private Member Functions
def	_accept (self, candidate, skipList)
def	_decay (self, candidate, skipList)
def	_hasValidDaughters (self, candidate)
def	_printP4 (self, candidate)
def	_select (self, candidate)

Detailed Description

Draws particle decay tree 

Definition at line 9 of file ParticleDecayDrawer.py.

Constructor & Destructor Documentation

__init__()

def python.ParticleDecayDrawer.ParticleDecayDrawer.__init__

(

self

)

Definition at line 12 of file ParticleDecayDrawer.py.

    def __init__(self):
        print("Init particleDecayDrawer")
        #  booleans: printP4 printPtEtaPhi printVertex   
        

References edm.print().

Member Function Documentation

_accept()

def python.ParticleDecayDrawer.ParticleDecayDrawer._accept (   self,   candidate,   skipList  )

private

Definition at line 16 of file ParticleDecayDrawer.py.

    def _accept(self, candidate, skipList):
        if candidate in skipList: return False;
        return self._select(candidate)
        

References python.ParticleDecayDrawer.ParticleDecayDrawer._select().

Referenced by python.ParticleDecayDrawer.ParticleDecayDrawer._decay().

_decay()

def python.ParticleDecayDrawer.ParticleDecayDrawer._decay (   self,   candidate,   skipList  )

private

Definition at line 32 of file ParticleDecayDrawer.py.

    def _decay(self, candidate, skipList):
 
          out = str()
          if candidate in skipList:
              return ""
          skipList.append(candidate)
    
          id = candidate.pdg_id()
          # here the part about the names :-(
          out += str(id) + self._printP4(candidate)
    
          validDau = 0
          nOfDaughters = candidate.numChildren()
          for i in range(nOfDaughters):
              if self._accept(candidate.listChildren()[i], skipList): validDau+=1
          if validDau == 0: return out
    
          out += " ->"
    
          for i in range(nOfDaughters):
              d = candidate.listChildren()[i]
              if self._accept(d, skipList):
                  decString = self._decay(d, skipList)
                  if ("->" in decString):  out += " ( %s ) " %decString
                  else:  out += " %s" %decString
          return out
 

References python.ParticleDecayDrawer.ParticleDecayDrawer._accept(), python.ParticleDecayDrawer.ParticleDecayDrawer._decay(), python.ParticleDecayDrawer.ParticleDecayDrawer._printP4(), FastTimerService_cff.range, and str.

Referenced by python.ParticleDecayDrawer.ParticleDecayDrawer._decay(), and python.ParticleDecayDrawer.ParticleDecayDrawer.draw().

_hasValidDaughters()

def python.ParticleDecayDrawer.ParticleDecayDrawer._hasValidDaughters (   self,   candidate  )

private

Definition at line 23 of file ParticleDecayDrawer.py.

    def _hasValidDaughters(self, candidate):
        nDaughters = candidate.numChildren()
        for i in range(nDaughters):
            if self._select(candidate.listChildren()[i]): return True
        return False        
 

References python.ParticleDecayDrawer.ParticleDecayDrawer._select(), and FastTimerService_cff.range.

_printP4()

def python.ParticleDecayDrawer.ParticleDecayDrawer._printP4 (   self,   candidate  )

private

Definition at line 29 of file ParticleDecayDrawer.py.

    def _printP4(self, candidate):
        return " "
 

Referenced by python.ParticleDecayDrawer.ParticleDecayDrawer._decay().

_select()

def python.ParticleDecayDrawer.ParticleDecayDrawer._select (   self,   candidate  )

private

Definition at line 20 of file ParticleDecayDrawer.py.

    def _select(self, candidate):
        return candidate.status() == 3
         

Referenced by python.ParticleDecayDrawer.ParticleDecayDrawer._accept(), python.ParticleDecayDrawer.ParticleDecayDrawer._hasValidDaughters(), and python.ParticleDecayDrawer.ParticleDecayDrawer.draw().

draw()

def python.ParticleDecayDrawer.ParticleDecayDrawer.draw	(		self,
			particles
	)

draw decay tree from list(HepMC.GenParticles)

Definition at line 59 of file ParticleDecayDrawer.py.

    def draw(self, particles): 
        """ draw decay tree from list(HepMC.GenParticles)""" 
        skipList = []
        nodesList = []
        momsList = []  
        for particle in particles:
            if particle.numParents() > 1:
                if self._select(particle):
                    skipList.append(particle)
                    nodesList.append(particle)
                    for j in range(particle.numParents()):
                        mom = particle.listParents()[j]
                        while (mom.mother()):# != None ):
                            mom = mom.mother()
                        if self._select(mom):
                            momsList.append(mom)
 
        print("-- decay --")  
        if len(momsList) > 0:
            if len(momsList) > 1:
                for m in range(len(momsList)):
                    decString = self._decay( momsList[m], skipList)
                    if len(decString) > 0:
                       print("{ %s } " %decString)
            else:
                print(self._decay(momsList[0], skipList))   
  
        if len(nodesList) > 0:
            print("-> ")
            if len(nodesList) > 1:
                for node in nodesList:
                   skipList.remove(node)
                   decString = self._decay(node, skipList)
                   if len(decString) > 0:
                       if "->" in decString:  print(" ( %s ) " %decString)
                       else:  print(" " + decString)
            else:
                skipList.remove(nodesList[0])
                print(self._decay(nodesList[0], skipList))
          
        print()
    

References python.ParticleDecayDrawer.ParticleDecayDrawer._decay(), python.ParticleDecayDrawer.ParticleDecayDrawer._select(), edm.print(), and FastTimerService_cff.range.