mcscorerootio Namespace Reference

Data Structures
class	MCScoreROOTIO

Functions
def	loop_tree (tfile, analyze_vertex)
def	test_loop ()
def	test_t2root ()

Variables
list	__all__ = [ 'MCScoreROOTIO', 'loop_tree' ]

Detailed Description

Python module

This module provides ROOT IO interface for MCScore data

  [C] MCScoreROOTIO
  [f] loop_tree(tfile, analyze_vertex):

                                              Q, 2006

Function Documentation

loop_tree()

def mcscorerootio.loop_tree	(		tfile,
			analyze_vertex
	)

loop ROOT tree in a ROOT file.
  * analyze_vertex: user function : analyze_vertex(MCVertex)

Definition at line 112 of file mcscorerootio.py.

def loop_tree(tfile, analyze_vertex):
  """
  loop ROOT tree in a ROOT file.
    * analyze_vertex: user function : analyze_vertex(MCVertex)
  """
  avtree = tfile.Get("vertex")
  aptree = tfile.Get("particle")
 
  # reading vertex...
  n_vertex = avtree.GetEntries()
  for ivtx in xrange(n_vertex):
    avtree.GetEntry(ivtx)
    aptree.GetEntry(ivtx)
 
    # vertex
    vertex = MCScore.MCVertex(avtree.x, avtree.y, avtree.z)    
 
    # reading secondary particles...
    nsec = aptree.np
    namelist = aptree.namelist
    pname = namelist.split()
    for ip in xrange(nsec):
      particle = MCScore.MCParticle(pname[ip], aptree.Z[ip], aptree.A[ip],
                                    aptree.kE[ip], aptree.px[ip],
                                    aptree.py[ip], aptree.pz[ip])
      vertex.append_particle(particle)
 
    analyze_vertex(vertex)
 
  return n_vertex
 
 
# ==================================================================
# test
# ==================================================================

Referenced by test_loop().

test_loop()

def mcscorerootio.test_loop

(

)

Definition at line 174 of file mcscorerootio.py.

def test_loop():
  def my_analysis(vertex):
    vertex.printout()
 
  f = ROOT.TFile("reaction.root", 'read')  
  nv = loop_tree(f, my_analysis)
  print "*** # of vertex= ", nv
  
 
# ==================================================================
# main
# ==================================================================

References loop_tree(), test_loop(), and test_t2root().

Referenced by test_loop().

test_t2root()

def mcscorerootio.test_t2root

(

)

Definition at line 147 of file mcscorerootio.py.

def test_t2root():
  g = ROOT.TFile("reaction.root", 'recreate')
 
  rootio = MCScoreROOTIO()
  rootio.define_tree()
 
  f = open("reaction.dat")
  f.seek(0)
 
  while(1):
    vertex = MCScore.read_next_vertex(f)
    if vertex == 0:
      break
 
    # filling ...
    rootio.fill_tree(vertex)
 
    del vertex
 
  print ">>> EOF"
  f.close()  
 
  # closing ROOT file
  g.Write()
  g.Close()
 
# ------------------------------------------------------------------

Referenced by test_loop().

Variable Documentation

__all__

list mcscorerootio.__all__ = [ 'MCScoreROOTIO', 'loop_tree' ]

private

Definition at line 20 of file mcscorerootio.py.