WIP

python/examples/play.py

@@ -13,36 +13,66 @@ from aare import File, ClusterFinder, VarClusterFinder
 base = Path('/mnt/sls_det_storage/matterhorn_data/aare_test_data/')
 
 f = File(base/'Moench03new/cu_half_speed_master_4.json')
-cf = ClusterFinder((400,400), (3,3))
+
+
+from aare._aare import ClusterFinderMT, ClusterCollector
+
+
+cf = ClusterFinderMT((400,400), (3,3), n_threads = 3)
+collector = ClusterCollector(cf)
+
 for i in range(1000):
-    cf.push_pedestal_frame(f.read_frame())
+    img = f.read_frame()
+    cf.push_pedestal_frame(img)
+print('Pedestal done')
+cf.sync()
 
-fig, ax = plt.subplots()
-im = ax.imshow(cf.pedestal())
-cf.pedestal()
-cf.noise()
+for i in range(100):
+    img = f.read_frame()
+    cf.find_clusters(img)
+
+
+# time.sleep(1)
+cf.stop()  
+collector.stop()
+cv = collector.steal_clusters()
+print(f'Processed {len(cv)} frames')
+
+print('Done')
 
 
 
-N = 500
-t0 = time.perf_counter()
-hist1 = bh.Histogram(bh.axis.Regular(40, -2, 4000))
-f.seek(0)
 
-t0 = time.perf_counter()
-data = f.read_n(N)
-t_elapsed = time.perf_counter()-t0
+# cf = ClusterFinder((400,400), (3,3))
+# for i in range(1000):
+#     cf.push_pedestal_frame(f.read_frame())
+
+# fig, ax = plt.subplots()
+# im = ax.imshow(cf.pedestal())
+# cf.pedestal()
+# cf.noise()
 
 
-n_bytes = data.itemsize*data.size
 
-print(f'Reading {N} frames took {t_elapsed:.3f}s {N/t_elapsed:.0f} FPS, {n_bytes/1024**2:.4f} GB/s')
+# N = 500
+# t0 = time.perf_counter()
+# hist1 = bh.Histogram(bh.axis.Regular(40, -2, 4000))
+# f.seek(0)
+
+# t0 = time.perf_counter()
+# data = f.read_n(N)
+# t_elapsed = time.perf_counter()-t0
 
 
-for frame in data:
-    a = cf.find_clusters(frame)
+# n_bytes = data.itemsize*data.size
 
-clusters = cf.steal_clusters()
+# print(f'Reading {N} frames took {t_elapsed:.3f}s {N/t_elapsed:.0f} FPS, {n_bytes/1024**2:.4f} GB/s')
+
+
+# for frame in data:
+#     a = cf.find_clusters(frame)
+
+# clusters = cf.steal_clusters()
 
 # t_elapsed = time.perf_counter()-t0
 # print(f'Clustering {N} frames took {t_elapsed:.2f}s  {N/t_elapsed:.0f} FPS')