[1337] | 1 | #!/usr/bin/python |
---|
| 2 | # ================================================================== |
---|
| 3 | # python script for Geant4Py test |
---|
| 4 | # |
---|
| 5 | # gtest02 |
---|
| 6 | # - test for using site-module packages |
---|
| 7 | # ================================================================== |
---|
| 8 | from Geant4 import * |
---|
| 9 | import g4py.NISTmaterials |
---|
| 10 | import g4py.ezgeom |
---|
| 11 | from g4py.ezgeom import G4EzVolume |
---|
| 12 | import g4py.EMSTDpl |
---|
| 13 | import g4py.ParticleGun |
---|
| 14 | |
---|
| 15 | # ================================================================== |
---|
| 16 | # intialize |
---|
| 17 | # ================================================================== |
---|
| 18 | def Configure(): |
---|
| 19 | # ------------------------------------------------------------------ |
---|
| 20 | # setup for materials |
---|
| 21 | # ------------------------------------------------------------------ |
---|
| 22 | # simple materials for Qgeom |
---|
| 23 | g4py.NISTmaterials.Construct() |
---|
| 24 | |
---|
| 25 | # ------------------------------------------------------------------ |
---|
| 26 | # setup for geometry |
---|
| 27 | # ------------------------------------------------------------------ |
---|
| 28 | #g4py.Qgeom.Construct() |
---|
| 29 | g4py.ezgeom.Construct() # initialize |
---|
| 30 | |
---|
| 31 | # ------------------------------------------------------------------ |
---|
| 32 | # setup for physics list |
---|
| 33 | # ------------------------------------------------------------------ |
---|
| 34 | g4py.EMSTDpl.Construct() |
---|
| 35 | |
---|
| 36 | # ------------------------------------------------------------------ |
---|
| 37 | # setup for primary generator action |
---|
| 38 | # ------------------------------------------------------------------ |
---|
| 39 | g4py.ParticleGun.Construct() |
---|
| 40 | gControlExecute("gun.mac") |
---|
| 41 | |
---|
| 42 | # ================================================================== |
---|
| 43 | # constructing geometry |
---|
| 44 | # ================================================================== |
---|
| 45 | def ConstructGeom(): |
---|
| 46 | print "* Constructing geometry..." |
---|
| 47 | # reset world material |
---|
| 48 | air= G4Material.GetMaterial("G4_AIR") |
---|
| 49 | g4py.ezgeom.SetWorldMaterial(air) |
---|
| 50 | |
---|
| 51 | # target |
---|
| 52 | global target |
---|
| 53 | target= G4EzVolume("Target") |
---|
| 54 | au= G4Material.GetMaterial("G4_Au") |
---|
| 55 | target.CreateTubeVolume(au, 0., 1.*cm, 1.*mm) |
---|
| 56 | target.PlaceIt(G4ThreeVector(0.,0.,-10.*cm)) |
---|
| 57 | |
---|
| 58 | # dummy box |
---|
| 59 | global detector_box, detector_box_pv |
---|
| 60 | detector_box= G4EzVolume("DetectorBox") |
---|
| 61 | detector_box.CreateBoxVolume(air, 20.*cm, 20.*cm, 40.*cm) |
---|
| 62 | detector_box_pv= detector_box.PlaceIt(G4ThreeVector(0.,0.,20.*cm)) |
---|
| 63 | |
---|
| 64 | # calorimeter |
---|
| 65 | global cal |
---|
| 66 | cal= G4EzVolume("Calorimeter") |
---|
| 67 | nai= G4Material.GetMaterial("G4_SODIUM_IODIDE") |
---|
| 68 | cal.CreateBoxVolume(nai, 5.*cm, 5.*cm, 30.*cm) |
---|
| 69 | dd= 5.*cm |
---|
| 70 | for ical in range(-1, 2): |
---|
| 71 | calPos= G4ThreeVector(dd*ical, 0., 0.) |
---|
| 72 | print calPos |
---|
| 73 | cal.PlaceIt(calPos, ical+1, detector_box) |
---|
| 74 | |
---|
| 75 | |
---|
| 76 | # ================================================================== |
---|
| 77 | # main |
---|
| 78 | # ================================================================== |
---|
| 79 | # ------------------------------------------------------------------ |
---|
| 80 | # randum number |
---|
| 81 | # ------------------------------------------------------------------ |
---|
| 82 | rand_engine= Ranlux64Engine() |
---|
| 83 | HepRandom.setTheEngine(rand_engine) |
---|
| 84 | HepRandom.setTheSeed(20050830L) |
---|
| 85 | |
---|
| 86 | # setup... |
---|
| 87 | Configure() |
---|
| 88 | ConstructGeom() |
---|
| 89 | |
---|
| 90 | # ------------------------------------------------------------------ |
---|
| 91 | # go... |
---|
| 92 | # ------------------------------------------------------------------ |
---|
| 93 | gRunManager.Initialize() |
---|
| 94 | |
---|
| 95 | # visualization |
---|
| 96 | gControlExecute("vis.mac") |
---|
| 97 | |
---|
| 98 | # beamOn |
---|
| 99 | #gRunManager.BeamOn(3) |
---|
| 100 | |
---|