Silence annoying GEANT4 warnings and banners during operation.

1 files changed, 13 insertions, 4 deletions

diff --git a/chroma/generator/g4gen.py b/chroma/generator/g4gen.py
index 7f55e56..0ed5433 100644
--- a/chroma/generator/g4gen.py
+++ b/chroma/generator/g4gen.py
@@ -1,12 +1,20 @@
+from chroma.generator.mute import *
+
+import pyublas
+import numpy as np
+from chroma.event import Photons, Vertex
+
+g4mute()
 from Geant4 import *
+g4unmute()
 import g4py.ezgeom
 import g4py.NISTmaterials
 import g4py.ParticleGun
-import pyublas
-import numpy as np
-from chroma.event import Photons, Vertex
 from chroma.generator import _g4chroma
 
+#cppmute()
+#cppunmute()
+
 class G4Generator(object):
     def __init__(self, material, seed=None):
         """Create generator to produce photons inside the specified material.
@@ -40,8 +48,9 @@ class G4Generator(object):
 
         self.tracking_action = _g4chroma.PhotonTrackingAction()
         gRunManager.SetUserAction(self.tracking_action)
+        g4mute()
         gRunManager.Initialize()
-        
+        g4unmute()
         # preinitialize the process by running a simple event
         self.generate_photons([Vertex('e-', (0,0,0), (1,0,0), 0, 1.0)])