perf: faster statistics calculation

seung-lab · Nov 16, 2024 · df66ad9 · df66ad9
1 parent 454d9cd
commit df66ad9
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/cc3d/fastcc3d.pyx b/cc3d/fastcc3d.pyx
@@ -802,6 +802,8 @@ def _statistics_helper2d(
   if no_slice_conversion:
     return output
 
+  cdef BBOX_T xs = 0, xe = 0, ys = 0, ye = 0
+
   slices = []
   for xs, xe, ys, ye in bbxes:
     if xs < voxels and ys < voxels:
@@ -891,6 +893,11 @@ def _statistics_helper3d(
   if no_slice_conversion:
     return output
 
+  cdef:
+    BBOX_T xs = 0, xe = 0
+    BBOX_T ys = 0, ye = 0
+    BBOX_T zs = 0, ze = 0
+
   slices = []
   for xs, xe, ys, ye, zs, ze in bbxes:
     if xs < voxels and ys < voxels and zs < voxels: