| 1 | #!/usr/bin/python
|
|---|
| 2 | # ==================================================================
|
|---|
| 3 | # python script for Geant4Py test
|
|---|
| 4 | #
|
|---|
| 5 | # gtest06
|
|---|
| 6 | # - test for constructing/visualizing boolean geoemtries
|
|---|
| 7 | # ==================================================================
|
|---|
| 8 | from Geant4 import *
|
|---|
| 9 | import g4py.ExN01pl, g4py.ParticleGun
|
|---|
| 10 |
|
|---|
| 11 | # ==================================================================
|
|---|
| 12 | # user actions in python
|
|---|
| 13 | # ==================================================================
|
|---|
| 14 | class MyDetectorConstruction(G4VUserDetectorConstruction):
|
|---|
| 15 | "My Detector Construction"
|
|---|
| 16 |
|
|---|
| 17 | def __init__(self):
|
|---|
| 18 | G4VUserDetectorConstruction.__init__(self)
|
|---|
| 19 | self.air= gNistManager.FindOrBuildMaterial("G4_AIR")
|
|---|
| 20 | self.lv_object= None
|
|---|
| 21 | self.world= self.ConstructWorld()
|
|---|
| 22 |
|
|---|
| 23 | self.va_red= G4VisAttributes(G4Color(1.,0.,0.))
|
|---|
| 24 | self.va_cyan= G4VisAttributes(G4Color(0.,1.,1.))
|
|---|
| 25 | self.va_green= G4VisAttributes(G4Color(0.,1.,0.))
|
|---|
| 26 | self.va_blue= G4VisAttributes(G4Color(0.,0.,1.))
|
|---|
| 27 | self.va_magenta= G4VisAttributes(G4Color(1.,0.,1.))
|
|---|
| 28 |
|
|---|
| 29 | self.sld_box= G4Box("box",20.*cm, 20.*cm, 20.*cm);
|
|---|
| 30 | self.sld_cyl= G4Tubs("cylinder",0., 10.*cm, 30.*cm, 0., twopi)
|
|---|
| 31 |
|
|---|
| 32 | # -----------------------------------------------------------------
|
|---|
| 33 | def ConstructWorld(self):
|
|---|
| 34 | # Python has automatic garbage collection system.
|
|---|
| 35 | # Geometry objects must be defined as GLOBAL not to be deleted.
|
|---|
| 36 | global sld_world, lv_world, pv_world, va_world
|
|---|
| 37 |
|
|---|
| 38 | sld_world= G4Box("world", 1.*m, 1.*m, 1.*m)
|
|---|
| 39 | lv_world= G4LogicalVolume(sld_world, self.air, "world")
|
|---|
| 40 | pv_world= G4PVPlacement(G4Transform3D(), lv_world, "world",
|
|---|
| 41 | None, False, 0)
|
|---|
| 42 |
|
|---|
| 43 | va_world= G4VisAttributes()
|
|---|
| 44 | va_world.SetVisibility(False)
|
|---|
| 45 | lv_world.SetVisAttributes(va_world)
|
|---|
| 46 |
|
|---|
| 47 | # solid object (dummy)
|
|---|
| 48 | global sld_sld, lv_sld, pv_sld
|
|---|
| 49 | sld_sld= G4Box("dummy", 10.*cm, 10.*cm, 10.*cm)
|
|---|
| 50 | self.lv_object= lv_sld= G4LogicalVolume(sld_sld, self.air, "dummy")
|
|---|
| 51 | pv_sld= G4PVPlacement(None, G4ThreeVector(), "dummy", lv_sld,
|
|---|
| 52 | pv_world, False, 0)
|
|---|
| 53 |
|
|---|
| 54 | return pv_world
|
|---|
| 55 |
|
|---|
| 56 | # -----------------------------------------------------------------
|
|---|
| 57 | def ConstructUnion(self):
|
|---|
| 58 | global sld_union
|
|---|
| 59 | sld_union= G4UnionSolid("box+cylinder", self.sld_box, self.sld_cyl);
|
|---|
| 60 |
|
|---|
| 61 | self.lv_object.SetSolid(sld_union)
|
|---|
| 62 | self.lv_object.SetVisAttributes(self.va_blue)
|
|---|
| 63 | gRunManager.GeometryHasBeenModified()
|
|---|
| 64 |
|
|---|
| 65 | # -----------------------------------------------------------------
|
|---|
| 66 | def ConstructIntersection(self):
|
|---|
| 67 | offset= G4ThreeVector(20.*cm, 20.*cm, 0.)
|
|---|
| 68 | global sld_intersection
|
|---|
| 69 | sld_intersection= G4IntersectionSolid("box*cylinder",
|
|---|
| 70 | self.sld_box, self.sld_cyl,
|
|---|
| 71 | None, offset)
|
|---|
| 72 |
|
|---|
| 73 | self.lv_object.SetSolid(sld_intersection)
|
|---|
| 74 | self.lv_object.SetVisAttributes(self.va_magenta)
|
|---|
| 75 | gRunManager.GeometryHasBeenModified()
|
|---|
| 76 |
|
|---|
| 77 | # -----------------------------------------------------------------
|
|---|
| 78 | def ConstructSubtraction(self):
|
|---|
| 79 | global sld_subtraction
|
|---|
| 80 | sld_subtraction= G4SubtractionSolid("box-cylinder",
|
|---|
| 81 | self.sld_box, self.sld_cyl)
|
|---|
| 82 |
|
|---|
| 83 | self.lv_object.SetSolid(sld_subtraction)
|
|---|
| 84 | self.lv_object.SetVisAttributes(self.va_red)
|
|---|
| 85 | gRunManager.GeometryHasBeenModified()
|
|---|
| 86 |
|
|---|
| 87 | # -----------------------------------------------------------------
|
|---|
| 88 | def Construct(self): # return the world volume
|
|---|
| 89 | return self.world
|
|---|
| 90 |
|
|---|
| 91 | # ==================================================================
|
|---|
| 92 | # main
|
|---|
| 93 | # ==================================================================
|
|---|
| 94 | # set geometry
|
|---|
| 95 | myDC= MyDetectorConstruction()
|
|---|
| 96 | gRunManager.SetUserInitialization(myDC)
|
|---|
| 97 |
|
|---|
| 98 | # minimal physics list
|
|---|
| 99 | g4py.ExN01pl.Construct()
|
|---|
| 100 |
|
|---|
| 101 | # set primary generator action
|
|---|
| 102 | g4py.ParticleGun.Construct()
|
|---|
| 103 |
|
|---|
| 104 | # initialize
|
|---|
| 105 | gRunManager.Initialize()
|
|---|
| 106 |
|
|---|
| 107 | # visualization
|
|---|
| 108 | gApplyUICommand("/vis/open RayTracer")
|
|---|
| 109 | gApplyUICommand("/vis/rayTracer/headAngle 40.")
|
|---|
| 110 | gApplyUICommand("/vis/rayTracer/eyePosition 100 100 150 cm")
|
|---|
| 111 |
|
|---|
| 112 | # create a vrml file for each solid type
|
|---|
| 113 | f_list= (
|
|---|
| 114 | ("union", myDC.ConstructUnion),
|
|---|
| 115 | ("intersection", myDC.ConstructIntersection),
|
|---|
| 116 | ("subtraction", myDC.ConstructSubtraction)
|
|---|
| 117 | )
|
|---|
| 118 |
|
|---|
| 119 |
|
|---|
| 120 | for s,f in f_list:
|
|---|
| 121 | f.__call__()
|
|---|
| 122 | fname= "%s.jpg" % (s)
|
|---|
| 123 | cmdstr= "/vis/rayTracer/trace " + fname
|
|---|
| 124 | gApplyUICommand(cmdstr)
|
|---|
| 125 |
|
|---|
| 126 |
|
|---|
