Import Geant4 8.1.0 source tree

environments/g4py/tests/gtest02/test.py

@@ -0,0 +1,134 @@
+#!/usr/bin/python
+# ==================================================================
+# python script for Geant4Py test
+#
+#   gtest02
+#   - test for using site-module packages
+# ==================================================================
+from Geant4 import *
+import Qmaterials, NISTmaterials
+import Qgeom, ExN01geom, ExN03geom
+import ExN01pl, EMSTDpl
+import ParticleGun, MedicalBeam
+
+
+# ==================================================================
+# user setup
+# ==================================================================
+
+# ------------------------------------------------------------------
+# Setup-0 (Q)
+# ------------------------------------------------------------------
+def Setup0():
+  # simple materials for Qgeom
+  Qmaterials.Construct()
+
+  # NIST materials
+  #NISTmaterials.Construct()
+  
+  # normal way for constructing user geometry
+  #qDC= Qgeom.QDetectorConstruction()
+  #gRunManager.SetUserInitialization(qDC)
+
+  # 2nd way, short-cut way
+  Qgeom.Construct()
+
+  # primary
+  global primary_position, primary_direction
+  primary_position= G4ThreeVector(0.,0., -14.9*cm)
+  primary_direction= G4ThreeVector(0.2, 0., 1.)
+
+
+# ------------------------------------------------------------------
+# Setup-1 (ExampleN01)
+# ------------------------------------------------------------------
+def Setup1():
+  ExN01geom.Construct()
+
+  global primary_position, primary_direction
+  primary_position= G4ThreeVector(-2.5*m, 0., 0.)
+  primary_direction= G4ThreeVector(1., 0., 0.)
+
+
+# ------------------------------------------------------------------
+# Setup-3 (ExampleN03)
+# ------------------------------------------------------------------
+def Setup3():
+  #exN03geom= ExN03geom.ExN03DetectorConstruction()
+  #gRunManager.SetUserInitialization(exN03geom)
+
+  ExN03geom.Construct()
+
+  global primary_position, primary_direction
+  primary_position= G4ThreeVector(-1.*m, 0., 0.)
+  primary_direction= G4ThreeVector(1., 0., 0.)
+
+
+# ==================================================================
+# main
+# ==================================================================
+# ------------------------------------------------------------------
+# randum number
+# ------------------------------------------------------------------
+rand_engine= Ranlux64Engine()
+HepRandom.setTheEngine(rand_engine)
+HepRandom.setTheSeed(20050830L)
+
+# ------------------------------------------------------------------
+# user setup
+# ------------------------------------------------------------------
+Setup0()
+#Setup1()
+#Setup3()
+
+
+# ------------------------------------------------------------------
+# setup for physics list
+# ------------------------------------------------------------------
+# normal way for constructing user physics list
+#exN01PL= ExN01PhysicsList.ExN01PhysicsList()
+#gRunManager.SetUserInitialization(exN01PL)
+
+# 2nd way, short-cut way
+# geantino + transportation
+#ExN01pl.Construct()
+
+# electron/gamma standard EM
+EMSTDpl.Construct()
+
+# ------------------------------------------------------------------
+# setup for primary generator action
+# ------------------------------------------------------------------
+# ------------
+# Particle Gun
+# ------------
+# normal way for constructing user physics list
+#pgPGA= ParticleGun.ParticleGunAction()
+#gRunManager.SetUserAction(pgPGA)
+#pg= pgPGA.GetParticleGun()
+
+# 2nd way, short-cut way
+pg= ParticleGun.Construct()
+
+# set parameters of particle gun
+pg.SetParticleByName("e-")
+pg.SetParticleEnergy(300.*MeV)
+pg.SetParticlePosition(primary_position)
+pg.SetParticleMomentumDirection(primary_direction)
+
+# ------------
+# Medical Beam
+# ------------
+#beam= MedicalBeam.Construct()
+
+# ------------------------------------------------------------------
+# go...
+# ------------------------------------------------------------------
+gRunManager.Initialize()
+
+# visualization
+gApplyUICommand("/control/execute vis.mac")
+
+# beamOn
+#gRunManager.BeamOn(3)
+