diff options
Diffstat (limited to 'generator/g4gen.py')
| -rw-r--r-- | generator/g4gen.py | 3 |
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) |