python::ParticleDecayDrawer::ParticleDecayDrawer Class Reference

List of all members.

Public Member Functions
def	__init__
def	draw
Private Member Functions
def	_accept
def	_decay
def	_hasValidDaughters
def	_printP4
def	_select

---

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.

00011                       :
00012         print "Init particleDecayDrawer"
00013         #  booleans: printP4 printPtEtaPhi printVertex   
        

---

Member Function Documentation

def python::ParticleDecayDrawer::ParticleDecayDrawer::_accept	(		self,
			candidate,
			skipList
	)		[private]

Definition at line 14 of file ParticleDecayDrawer.py.

00015                                           :
00016         if candidate in skipList: return False;
00017         return self._select(candidate)
        

def python::ParticleDecayDrawer::ParticleDecayDrawer::_decay	(		self,
			candidate,
			skipList
	)		[private]

Definition at line 30 of file ParticleDecayDrawer.py.

                                         :

          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.

                                           :
        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.

                                 :
        return " "

def python::ParticleDecayDrawer::ParticleDecayDrawer::_select	(		self,
			candidate
	)		[private]

Definition at line 18 of file ParticleDecayDrawer.py.

00019                                 :
00020         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.

                             : 
        """ 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