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 7 of file ParticleDecayDrawer.py.

Constructor & Destructor Documentation

def python.ParticleDecayDrawer.ParticleDecayDrawer.__init__

(

self

)

Definition at line 10 of file ParticleDecayDrawer.py.

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

Member Function Documentation

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

private

Definition at line 14 of file ParticleDecayDrawer.py.

References python.ParticleDecayDrawer.ParticleDecayDrawer._select().

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

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

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

private

Definition at line 30 of file ParticleDecayDrawer.py.

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

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

    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 xrange(nOfDaughters):
              if self._accept(candidate.listChildren()[i], skipList): validDau+=1
          if validDau == 0: return out
    
          out += " ->"
    
          for i in xrange(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

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

private

Definition at line 21 of file ParticleDecayDrawer.py.

References python.ParticleDecayDrawer.ParticleDecayDrawer._select().

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

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

private

Definition at line 27 of file ParticleDecayDrawer.py.

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

    def _printP4(self, candidate):
        return " "

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

private

Definition at line 18 of file ParticleDecayDrawer.py.

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

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

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

draw decay tree from list(HepMC.GenParticles)

Definition at line 57 of file ParticleDecayDrawer.py.

References python.ParticleDecayDrawer.ParticleDecayDrawer._decay(), and python.ParticleDecayDrawer.ParticleDecayDrawer._select().

    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 xrange(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 xrange(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
    

---

Generated for CMSSW Reference Manual by  1.8.11