mcscorerootio Namespace Reference

Classes
class	MCScoreROOTIO

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

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
# ==================================================================

Here is the caller graph for this function:

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
# ==================================================================

Here is the call graph for this function:

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

# ------------------------------------------------------------------

Here is the call graph for this function:

Here is the caller graph for this function:

Variable Documentation

__all__

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

private

Definition at line 20 of file mcscorerootio.py.