Gabriele Cosmo
134 lines
4.0 KiB
Python
Executable File
134 lines
4.0 KiB
Python
Executable File
#!/usr/bin/python3
|
|
# ==================================================================
|
|
# python script for Geant4Py test
|
|
#
|
|
# gtest02
|
|
# - test for using site-module packages
|
|
# ==================================================================
|
|
from Geant4 import *
|
|
import g4pytest.Qmaterials, g4pytest.NISTmaterials
|
|
import g4pytest.Qgeom, g4pytest.ExN01geom, g4pytest.ExN03geom
|
|
import g4pytest.ExN01pl, g4pytest.EMSTDpl
|
|
import g4pytest.ParticleGun, g4pytest.MedicalBeam
|
|
|
|
# ==================================================================
|
|
# user setup
|
|
# ==================================================================
|
|
|
|
# ------------------------------------------------------------------
|
|
# Setup-0 (Q)
|
|
# ------------------------------------------------------------------
|
|
def Setup0():
|
|
# simple materials for Qgeom
|
|
g4pytest.Qmaterials.Construct()
|
|
|
|
# NIST materials
|
|
#g4pytest.NISTmaterials.Construct()
|
|
|
|
# normal way for constructing user geometry
|
|
#qDC= g4pytest.Qgeom.QDetectorConstruction()
|
|
#gRunManager.SetUserInitialization(qDC)
|
|
|
|
# 2nd way, short-cut way
|
|
g4pytest.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():
|
|
g4pytest.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= g4pytest.ExN03geom.ExN03DetectorConstruction()
|
|
#gRunManager.SetUserInitialization(exN03geom)
|
|
|
|
g4pytest.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(20050830)
|
|
|
|
# ------------------------------------------------------------------
|
|
# 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
|
|
#g4pytest.ExN01pl.Construct()
|
|
|
|
# electron/gamma standard EM
|
|
g4pytest.EMSTDpl.Construct()
|
|
|
|
# ------------------------------------------------------------------
|
|
# setup for primary generator action
|
|
# ------------------------------------------------------------------
|
|
# ------------
|
|
# Particle Gun
|
|
# ------------
|
|
# normal way for constructing user physics list
|
|
#pgPGA= g4pytest.ParticleGun.ParticleGunAction()
|
|
#gRunManager.SetUserAction(pgPGA)
|
|
#pg= pgPGA.GetParticleGun()
|
|
|
|
# 2nd way, short-cut way
|
|
pg= g4pytest.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= g4pytest.MedicalBeam.Construct()
|
|
|
|
# ------------------------------------------------------------------
|
|
# go...
|
|
# ------------------------------------------------------------------
|
|
gRunManager.Initialize()
|
|
|
|
# visualization
|
|
gApplyUICommand("/control/execute vis.mac")
|
|
|
|
# beamOn
|
|
#gRunManager.BeamOn(3)
|
|
|