summaryrefslogtreecommitdiff
path: root/generator/g4gen.py
diff options
context:
space:
mode:
authorStan Seibert <stan@mtrr.org>2011-09-13 16:10:04 -0400
committerStan Seibert <stan@mtrr.org>2011-09-13 16:10:04 -0400
commit4ab542a6f7e8dfbe6aeec2eb1b71176e2e851e75 (patch)
tree90a910177777115b6ef7ad2780b09ca69ff87109 /generator/g4gen.py
parent1ed35a8420b73afa76c12c82f4aebc31132650d9 (diff)
parent6f0703602270d03f4025221f13fee21aa842a863 (diff)
downloadchroma-4ab542a6f7e8dfbe6aeec2eb1b71176e2e851e75.tar.gz
chroma-4ab542a6f7e8dfbe6aeec2eb1b71176e2e851e75.tar.bz2
chroma-4ab542a6f7e8dfbe6aeec2eb1b71176e2e851e75.zip
merge
Diffstat (limited to 'generator/g4gen.py')
-rw-r--r--generator/g4gen.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/generator/g4gen.py b/generator/g4gen.py
index 7266456..91ddc18 100644
--- a/generator/g4gen.py
+++ b/generator/g4gen.py
@@ -122,6 +122,9 @@ class G4Generator(object):
self.particle_gun.SetParticleMomentumDirection(G4ThreeVector(*vertex.dir).unit())
self.particle_gun.SetParticleTime(vertex.t0*s)
+ if vertex.pol is not None:
+ self.particle_gun.SetParticlePolarization(G4ThreeVector(*vertex.pol).unit())
+
self.tracking_action.Clear()
gRunManager.BeamOn(1)