diff options
Diffstat (limited to 'generator/g4gen.py')
| -rw-r--r-- | generator/g4gen.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/generator/g4gen.py b/generator/g4gen.py index 8dca086..3836897 100644 --- a/generator/g4gen.py +++ b/generator/g4gen.py @@ -120,7 +120,9 @@ class G4Generator(object): self.particle_gun.SetParticleEnergy(total_energy) self.particle_gun.SetParticlePosition(G4ThreeVector(*vertex.pos)*m) self.particle_gun.SetParticleMomentumDirection(G4ThreeVector(*vertex.dir).unit()) - + if vertex.pol is not None: + self.particle_gun.SetParticlePolarization(G4ThreeVector(*vertex.pol).unit()) + self.tracking_action.Clear() gRunManager.BeamOn(1) |