Gabriele Cosmo
101 lines
3.1 KiB
Python
Executable File
101 lines
3.1 KiB
Python
Executable File
#!/usr/bin/python3
|
|
# ==================================================================
|
|
# python script for Geant4Py test
|
|
#
|
|
# gtest02
|
|
# - test for using site-module packages
|
|
# ==================================================================
|
|
from Geant4 import *
|
|
import g4py.NISTmaterials
|
|
import g4py.ezgeom
|
|
from g4py.ezgeom import G4EzVolume
|
|
import g4py.EMSTDpl
|
|
import g4py.ParticleGun
|
|
|
|
# ==================================================================
|
|
# intialize
|
|
# ==================================================================
|
|
def Configure():
|
|
# ------------------------------------------------------------------
|
|
# setup for materials
|
|
# ------------------------------------------------------------------
|
|
# simple materials for Qgeom
|
|
g4py.NISTmaterials.Construct()
|
|
|
|
# ------------------------------------------------------------------
|
|
# setup for geometry
|
|
# ------------------------------------------------------------------
|
|
#g4py.Qgeom.Construct()
|
|
g4py.ezgeom.Construct() # initialize
|
|
|
|
# ------------------------------------------------------------------
|
|
# setup for physics list
|
|
# ------------------------------------------------------------------
|
|
g4py.EMSTDpl.Construct()
|
|
|
|
# ------------------------------------------------------------------
|
|
# setup for primary generator action
|
|
# ------------------------------------------------------------------
|
|
g4py.ParticleGun.Construct()
|
|
gControlExecute("gun.mac")
|
|
|
|
# ==================================================================
|
|
# constructing geometry
|
|
# ==================================================================
|
|
def ConstructGeom():
|
|
print("* Constructing geometry...")
|
|
# reset world material
|
|
air= G4Material.GetMaterial("G4_AIR")
|
|
g4py.ezgeom.SetWorldMaterial(air)
|
|
|
|
# target
|
|
global target
|
|
target= G4EzVolume("Target")
|
|
au= G4Material.GetMaterial("G4_Au")
|
|
target.CreateTubeVolume(au, 0., 1.*cm, 1.*mm)
|
|
target.PlaceIt(G4ThreeVector(0.,0.,-10.*cm))
|
|
|
|
# dummy box
|
|
global detector_box, detector_box_pv
|
|
detector_box= G4EzVolume("DetectorBox")
|
|
detector_box.CreateBoxVolume(air, 20.*cm, 20.*cm, 40.*cm)
|
|
detector_box_pv= detector_box.PlaceIt(G4ThreeVector(0.,0.,20.*cm))
|
|
|
|
# calorimeter
|
|
global cal
|
|
cal= G4EzVolume("Calorimeter")
|
|
nai= G4Material.GetMaterial("G4_SODIUM_IODIDE")
|
|
cal.CreateBoxVolume(nai, 5.*cm, 5.*cm, 30.*cm)
|
|
dd= 5.*cm
|
|
for ical in range(-1, 2):
|
|
calPos= G4ThreeVector(dd*ical, 0., 0.)
|
|
print(calPos)
|
|
cal.PlaceIt(calPos, ical+1, detector_box)
|
|
|
|
|
|
# ==================================================================
|
|
# main
|
|
# ==================================================================
|
|
# ------------------------------------------------------------------
|
|
# randum number
|
|
# ------------------------------------------------------------------
|
|
rand_engine= Ranlux64Engine()
|
|
HepRandom.setTheEngine(rand_engine)
|
|
HepRandom.setTheSeed(20050830)
|
|
|
|
# setup...
|
|
Configure()
|
|
ConstructGeom()
|
|
|
|
# ------------------------------------------------------------------
|
|
# go...
|
|
# ------------------------------------------------------------------
|
|
gRunManager.Initialize()
|
|
|
|
# visualization
|
|
gControlExecute("vis.mac")
|
|
|
|
# beamOn
|
|
#gRunManager.BeamOn(3)
|
|
|