Import Geant4 9.4.0 source tree

2016-06-09 16:25:56 +02:00
parent 74cad5e589
commit 89a9605df1
4440 changed files with 379508 additions and 189225 deletions
@@ -0,0 +1,126 @@
+#!/usr/bin/python3
+# ==================================================================
+# python script for Geant4Py test
+#
+#   gtest06
+#   - test for constructing/visualizing boolean geoemtries
+# ==================================================================
+from Geant4 import *
+import g4py.ExN01pl, g4py.ParticleGun
+
+# ==================================================================
+# user actions in python
+# ==================================================================
+class MyDetectorConstruction(G4VUserDetectorConstruction):
+  "My Detector Construction"
+
+  def __init__(self):
+    G4VUserDetectorConstruction.__init__(self)
+    self.air= gNistManager.FindOrBuildMaterial("G4_AIR")
+    self.lv_object= None
+    self.world= self.ConstructWorld()
+    
+    self.va_red= G4VisAttributes(G4Color(1.,0.,0.))
+    self.va_cyan= G4VisAttributes(G4Color(0.,1.,1.))
+    self.va_green= G4VisAttributes(G4Color(0.,1.,0.))
+    self.va_blue= G4VisAttributes(G4Color(0.,0.,1.))
+    self.va_magenta= G4VisAttributes(G4Color(1.,0.,1.))
+
+    self.sld_box= G4Box("box",20.*cm, 20.*cm, 20.*cm);
+    self.sld_cyl= G4Tubs("cylinder",0., 10.*cm, 30.*cm, 0., twopi)
+
+  # -----------------------------------------------------------------
+  def ConstructWorld(self):
+    # Python has automatic garbage collection system.
+    # Geometry objects must be defined as GLOBAL not to be deleted.
+    global sld_world, lv_world, pv_world, va_world
+
+    sld_world= G4Box("world", 1.*m, 1.*m, 1.*m)
+    lv_world= G4LogicalVolume(sld_world, self.air, "world")    
+    pv_world= G4PVPlacement(G4Transform3D(), lv_world, "world",
+                            None, False, 0)
+
+    va_world= G4VisAttributes()
+    va_world.SetVisibility(False)
+    lv_world.SetVisAttributes(va_world)
+
+    # solid object (dummy)
+    global sld_sld, lv_sld, pv_sld
+    sld_sld= G4Box("dummy", 10.*cm, 10.*cm, 10.*cm)
+    self.lv_object= lv_sld= G4LogicalVolume(sld_sld, self.air, "dummy")
+    pv_sld= G4PVPlacement(None, G4ThreeVector(), "dummy", lv_sld,
+                          pv_world, False, 0)
+
+    return pv_world
+
+  # -----------------------------------------------------------------
+  def ConstructUnion(self):
+    global sld_union
+    sld_union= G4UnionSolid("box+cylinder", self.sld_box, self.sld_cyl); 
+ 
+    self.lv_object.SetSolid(sld_union)
+    self.lv_object.SetVisAttributes(self.va_blue)
+    gRunManager.GeometryHasBeenModified()
+    
+  # -----------------------------------------------------------------
+  def ConstructIntersection(self):
+    offset= G4ThreeVector(20.*cm, 20.*cm, 0.)
+    global sld_intersection  
+    sld_intersection= G4IntersectionSolid("box*cylinder",
+                                          self.sld_box, self.sld_cyl,
+                                          None, offset)
+
+    self.lv_object.SetSolid(sld_intersection)
+    self.lv_object.SetVisAttributes(self.va_magenta)
+    gRunManager.GeometryHasBeenModified()
+
+  # -----------------------------------------------------------------
+  def ConstructSubtraction(self):
+    global sld_subtraction  
+    sld_subtraction= G4SubtractionSolid("box-cylinder",
+                                        self.sld_box, self.sld_cyl)
+
+    self.lv_object.SetSolid(sld_subtraction)
+    self.lv_object.SetVisAttributes(self.va_red)
+    gRunManager.GeometryHasBeenModified()
+            
+  # -----------------------------------------------------------------
+  def Construct(self): # return the world volume
+    return self.world  
+  
+# ==================================================================
+# main
+# ==================================================================
+# set geometry
+myDC= MyDetectorConstruction()
+gRunManager.SetUserInitialization(myDC)
+
+# minimal physics list
+g4py.ExN01pl.Construct()
+
+# set primary generator action
+g4py.ParticleGun.Construct()
+
+# initialize
+gRunManager.Initialize()
+
+# visualization
+gApplyUICommand("/vis/open RayTracer")
+gApplyUICommand("/vis/rayTracer/headAngle 40.")
+gApplyUICommand("/vis/rayTracer/eyePosition 100 100 150  cm")
+
+# create a vrml file for each solid type
+f_list= (
+  ("union",          myDC.ConstructUnion),
+  ("intersection",   myDC.ConstructIntersection),
+  ("subtraction",    myDC.ConstructSubtraction)
+  )
+
+
+for s,f in f_list:
+  f.__call__()
+  fname= "%s.jpg" % (s)
+  cmdstr= "/vis/rayTracer/trace " + fname
+  gApplyUICommand(cmdstr)
+  
+