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 | |
---|