Import Geant4 8.2.0 source tree

environments/g4py/site-modules/utils/GNUmakefile

@@ -0,0 +1,19 @@
+# $Id: GNUmakefile,v 1.1 2006/07/12 01:57:46 kmura Exp $
+# $Name: geant4-08-02 $
+# ===========================================================
+#   Makefile for building Geant4Py site-modules
+# ===========================================================
+
+SUBDIR := MCScore
+
+.PHONY: all install clean cleanlib
+
+all:
+	@for dir in $(SUBDIR); do (cd $$dir && $(MAKE)); done;:
+
+install:
+	@for dir in $(SUBDIR); do (cd $$dir && $(MAKE) install); done;:
+
+clean:
+	@for dir in $(SUBDIR); do (cd $$dir && $(MAKE) clean); done;:
+

environments/g4py/site-modules/utils/MCScore/GNUmakefile

@@ -0,0 +1,29 @@
+# $Id: GNUmakefile,v 1.1 2006/07/12 02:01:07 kmura Exp $
+# $Name: geant4-08-02 $
+# ===========================================================
+#   Makefile for building Geant4Py modules
+# ===========================================================
+include ../../../config/config.gmk
+
+include $(G4PY_INSTALL)/config/g4py.gmk
+include $(G4PY_INSTALL)/config/sys/$(Q_SYSTEM).gmk
+
+install_dir := $(Q_LIBDIR)
+
+pys  := $(wildcard *.py)
+pycs := $(patsubst %.py, %.pyc, $(pys))
+pyos := $(patsubst %.py, %.pyo, $(pys))
+
+.PHONY: clean
+
+all: $(pycs) $(pyos)
+
+
+install: $(pycs) $(pyos)
+	@echo ... intall "*.py" into $(install_dir)
+	@if [ ! -d $(install_dir) ]; then install -d $(install_dir); fi
+	@install -m 644 $(pys) $(pycs) $(pyos) $(install_dir)
+
+clean:
+	@-\rm -f $(pycs) $(pyos)
+

environments/g4py/site-modules/utils/MCScore/MCScore.py

@@ -0,0 +1,132 @@
+# ==================================================================
+"""
+Python module
+
+This module provides classes and functions for scoring reactions
+
+  [C] MCVertex:
+  [C] MCParticle:
+  [f] read_next_vertex(stream):
+
+                                              Q, 2006
+"""
+# ==================================================================
+from Geant4.HEPUnit import *
+import string
+
+# ==================================================================
+#   class definition
+# ==================================================================
+
+# ------------------------------------------------------------------
+# MCParticle
+# ------------------------------------------------------------------
+class MCParticle:
+  "MC particle"
+  def __init__(self, aname, aZ, aA, akE, apx, apy, apz):
+    self.name= aname
+    self.Z= aZ
+    self.A= aA
+    self.kineticE= akE
+    self.px= apx
+    self.py= apy
+    self.pz= apz
+
+  def printout(self):
+    print "--- particle: %s, Z=%2d, A=%2d, kE=%g" % \
+          (self.name, self.Z, self.A, self.kineticE/MeV)
+
+# ------------------------------------------------------------------
+# MCVertex
+# ------------------------------------------------------------------
+class MCVertex :
+  "MC vertex"
+
+  def __init__(self, ax, ay, az):
+    self.x= ax
+    self.y= ay
+    self.z= az
+    self.nparticle= 0
+    self.particle_list= []
+
+  def append_particle(self, aparticle):
+    self.particle_list.append(aparticle)
+    self.nparticle= self.nparticle+1
+
+  def printout(self):
+    print "@@@ vertex: x=(%g,%g,%g) Nsec=%3d" % \
+          (self.x/cm, self.y/cm, self.z/cm, self.nparticle)
+    for p in self.particle_list:
+      p.printout()
+  
+  def dump_vertex(self, stream):
+    aline= "%g %g %g %d\n" % \
+           (self.x/m, self.y/m, self.z/m, self.nparticle)
+    stream.write(aline)
+    for p in self.particle_list:
+      aline= " %s %d %d %g %g %g %g\n" % \
+             (p.name, p.Z, p.A, p.kineticE/MeV, p.px/MeV, p.py/MeV, p.pz/MeV)
+      stream.write(aline)
+
+  def __del__(self):
+    np= len(self.particle_list)
+    del self.particle_list[0:np]
+    
+
+# ==================================================================
+#   I/O interface
+# ==================================================================
+# ------------------------------------------------------------------
+def read_next_vertex(stream):
+# ------------------------------------------------------------------
+  line= stream.readline()
+  if (line == "") :  # EOF
+    return 0
+
+  # reading vertex
+  data= line.split()
+  x=  string.atof(data[0])*m
+  y=  string.atof(data[1])*m
+  z=  string.atof(data[2])*m
+  nsec= string.atoi(data[3])
+
+  vertex= MCVertex(x,y,z)
+
+  # reading particles
+  for p in range(0, nsec):
+    data= stream.readline().split()
+    pname= data[0]
+    Z= string.atoi(data[1])
+    A= string.atoi(data[2])    
+    kE= string.atof(data[3])*MeV
+    px= string.atof(data[4])*MeV
+    py= string.atof(data[5])*MeV
+    pz= string.atof(data[6])*MeV
+
+    particle= MCParticle(pname, Z, A, kE, px, py, pz)
+    vertex.append_particle(particle)
+
+  return vertex
+
+# ==================================================================
+# test
+# ==================================================================
+def test():
+  f= open("reaction.dat")
+  f.seek(0)
+
+  while(1):
+    vertex= read_next_vertex(f)
+    if(vertex==0):
+      break
+    vertex.printout()
+    del vertex
+  f.close()
+  print ">>> EOF"
+
+# ==================================================================
+# main
+# ==================================================================
+if (__name__ == "__main__") :
+  test()
+  

environments/g4py/site-modules/utils/MCScore/MCScoreROOTIO.py

@@ -0,0 +1,190 @@
+# ==================================================================
+"""
+Python module
+
+This module provides ROOT IO interface for MCScore data
+
+  [C] MCScoreROOTIO
+  [f] loop_tree(tfile, analyze_vertex):
+
+                                              Q, 2006
+"""
+# ==================================================================
+from Geant4.HEPUnit import *
+from array import array
+import string
+import MCScore
+import ROOT
+
+# ==================================================================
+from Geant4.HEPUnit import *
+import string
+
+# ==================================================================
+#   class definition
+# ==================================================================
+# ------------------------------------------------------------------
+# MCScoreROOTIO
+# ------------------------------------------------------------------
+class MCScoreROOTIO:
+  "ROOT IO interface for MCScore"
+  def __init__(self, bsize=250):
+    self.maxparticle= bsize # buffer size for #particles/vertex
+
+  # ------------------------------------------------------------------
+  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')
+
+  # ------------------------------------------------------------------
+  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()
+                        
+# ==================================================================
+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
+# ==================================================================
+# ------------------------------------------------------------------
+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()
+
+# ------------------------------------------------------------------
+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
+# ==================================================================
+if (__name__ == "__main__") :
+  test_t2root()
+  test_loop()
+