From 6f0703602270d03f4025221f13fee21aa842a863 Mon Sep 17 00:00:00 2001 From: Anthony LaTorre Date: Mon, 12 Sep 2011 16:41:17 -0400 Subject: add chroma-sim script. get rid of GPU class; contexts should now be created with create_cuda_context(). --- generator/g4gen.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'generator/g4gen.py') 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) -- cgit