diff options
| author | Stan Seibert <stan@mtrr.org> | 2011-10-21 16:50:15 -0400 |
|---|---|---|
| committer | tlatorre <tlatorre@uchicago.edu> | 2021-05-09 08:42:38 -0700 |
| commit | 03d3afd7aab2df0f2ffb5a81bf862c7fd17b3b11 (patch) | |
| tree | 2df2d8867ecdb207491cb2d2aadd00cf4abab130 | |
| parent | 536d16b98dbb808cbb8692bcd50ce5a9fb7bb00d (diff) | |
| download | chroma-03d3afd7aab2df0f2ffb5a81bf862c7fd17b3b11.tar.gz chroma-03d3afd7aab2df0f2ffb5a81bf862c7fd17b3b11.tar.bz2 chroma-03d3afd7aab2df0f2ffb5a81bf862c7fd17b3b11.zip | |
Use global per-thread ZeroMQ context rather than make a new one.
| -rw-r--r-- | chroma/generator/photon.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/chroma/generator/photon.py b/chroma/generator/photon.py index 39e8cf4..5ae1e93 100644 --- a/chroma/generator/photon.py +++ b/chroma/generator/photon.py @@ -18,7 +18,7 @@ class G4GeneratorProcess(multiprocessing.Process): def run(self): gen = g4gen.G4Generator(self.material, seed=self.seed) - context = zmq.Context() + context = zmq.Context.instance() vertex_socket = context.socket(zmq.PULL) vertex_socket.connect(self.vertex_socket_address) photon_socket = context.socket(zmq.PUSH) @@ -75,7 +75,7 @@ class G4ParallelGenerator(object): for p in self.processes: p.start() - self.zmq_context = zmq.Context() + self.zmq_context = zmq.Context.instance() self.vertex_socket = self.zmq_context.socket(zmq.PUSH) self.vertex_socket.bind(self.vertex_address) self.photon_socket = self.zmq_context.socket(zmq.PULL) |