Public Member Functions
def	__init__ (self, bsize=250)

def	define_tree (self)

def	fill_tree (self, vertex)

Data Fields
	a_A

	a_ke

	a_namelist

	a_np

	a_px

	a_py

	a_pz

	a_x

	a_y

	a_z

	a_Z

	maxparticle

	ptree

	vtree

Detailed Description

Definition at line 29 of file mcscorerootio.py.

Constructor & Destructor Documentation

◆ init()

def mcscorerootio.MCScoreROOTIO.__init__	(	self,
		bsize = `250`
	)

Definition at line 31 of file mcscorerootio.py.

  def __init__(self, bsize=250):
    self.maxparticle = bsize # buffer size for #particles/vertex
 

Member Function Documentation

◆ define_tree()

def mcscorerootio.MCScoreROOTIO.define_tree ( self )

Definition at line 35 of file mcscorerootio.py.

  def define_tree(self):
    "define ROOT tree"
    # defining tree header...
    self.vtree = ROOT.TTree('vertex',   'mc vertex')
    self.ptree = ROOT.TTree('particle', 'mc particle')
 
    # vertex...  
    self.a_x = array('d', [0.]); self.vtree.Branch('x', self.a_x, 'x/d')
    self.a_y = array('d', [0.]); self.vtree.Branch('y', self.a_y, 'y/d')
    self.a_z = array('d', [0.]); self.vtree.Branch('z', self.a_z, 'z/d')
    
    #global a_np, a_namelist, a_Z, a_A, a_ke, a_px, a_py, a_pz
    self.a_np = array('i', [0]); self.ptree.Branch('np', self.a_np, 'np/i')
    
    self.a_namelist = array('c', self.maxparticle*10*['\0'])
      # 10 characters/particle
    self.ptree.Branch('namelist', self.a_namelist, 'namelist/C')
    
    self.a_Z = array('i', self.maxparticle*[0])
    self.ptree.Branch('Z',  self.a_Z,  'Z[np]/I')
 
    self.a_A = array('i', self.maxparticle*[0])
    self.ptree.Branch('A',  self.a_A,  'A[np]/i')
 
    self.a_ke = array('d', self.maxparticle*[0.])
    self.ptree.Branch('kE', self.a_ke, 'kE[np]/d')
 
    self.a_px = array('d', self.maxparticle*[0.])
    self.ptree.Branch('px', self.a_px, 'px[np]/d')
 
    self.a_py = array('d', self.maxparticle*[0.])
    self.ptree.Branch('py', self.a_py, 'py[np]/d')
 
    self.a_pz = array('d', self.maxparticle*[0.])
    self.ptree.Branch('pz', self.a_pz, 'pz[np]/d')
 

◆ fill_tree()

def mcscorerootio.MCScoreROOTIO.fill_tree	(	self,
		vertex
	)

Definition at line 72 of file mcscorerootio.py.

  def fill_tree(self, vertex):
    "fill vertex information to ROOT tree"
    # ------------------------------------------------------------------
    def push_pname(i0, pname): # local function
      n = len(pname)
      for i in xrange(n):
        self.a_namelist[i0+i] = pname[i]
        self.a_namelist[i0+n] = ' '
 
    self.a_x[0] = vertex.x
    self.a_y[0] = vertex.y
    self.a_z[0] = vertex.z
    self.vtree.Fill()
          
    if vertex.nparticle > self.maxparticle:
      raise """
      *** buffer overflow in #particles/vertex.
      *** please increment buffersize in MCScoreROOTIO(bsize).
      """, self.maxparticle
    
    idx_namelist = 0
    self.a_np[0] = vertex.nparticle
    for ip in range(vertex.nparticle):
      particle = vertex.particle_list[ip]
      push_pname(idx_namelist, particle.name)
      idx_namelist += (len(particle.name)+1)
      self.a_Z[ip] = particle.Z
      self.a_A[ip] = particle.A
      self.a_ke[ip] = particle.kineticE
      self.a_px[ip] = particle.px
      self.a_py[ip] = particle.py
      self.a_pz[ip] = particle.pz
 
    self.a_namelist[idx_namelist] = '\0'
    self.ptree.Fill()
      
                  
# ==================================================================
# functions
# ==================================================================