Gabriele Cosmo
135 lines
3.8 KiB
Python
Executable File
135 lines
3.8 KiB
Python
Executable File
#!/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)
|
|
|