diff options
Diffstat (limited to 'threadtest.py')
| -rw-r--r-- | threadtest.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/threadtest.py b/threadtest.py index 81b6d62..91a1857 100644 --- a/threadtest.py +++ b/threadtest.py @@ -11,7 +11,7 @@ from histogram import Histogram jobs = Queue() output = Queue() -def create_job(position=(0,0,0), nphotons=10000, max_steps=10): +def create_job(position=(0,0,0), nphotons=1000, max_steps=10): positions = np.tile(position, nphotons).reshape((nphotons, 3)) directions = uniform_sphere(nphotons) polarizations = uniform_sphere(nphotons) @@ -23,7 +23,7 @@ def create_job(position=(0,0,0), nphotons=10000, max_steps=10): return Job(positions, directions, wavelengths, polarizations, times, states, last_hit_triangles, max_steps) -def generate_event(position=(0,0,0), nphotons=10000): +def generate_event(position=(0,0,0), nphotons=1000): jobs.put(create_job(position, nphotons)) jobs.join() @@ -34,6 +34,9 @@ def generate_event(position=(0,0,0), nphotons=10000): solids[last_hit_triangles == -1] = -1 surface_absorbed = job.states == 2 + for i in np.unique(job.states): + print 'state %2i %i' % (i, len(job.states[job.states == i])) + event_times = [] for i in lbne.pmtids: photons = np.logical_and(solids == i, surface_absorbed) @@ -43,12 +46,11 @@ def generate_event(position=(0,0,0), nphotons=10000): if len(hit_times) > 0: event_times.append((i, np.min(hit_times))) - print len(event_times) - print event_times + print '%i hit pmts' % len(event_times) return event_times -def likelihood(event_times, position=(0,0,0), nphotons=10000, neval=100): +def likelihood(event_times, position=(0,0,0), nphotons=1000, neval=100): for i in range(neval): jobs.put(create_job(position, nphotons)) jobs.join() |