perf(connected_components_stack): major speedup

william-silversmith · william-silversmith · commit 4146715119c9 · 2025-10-03T17:10:59.000-04:00
diff --git a/cc3d/__init__.py b/cc3d/__init__.py
@@ -410,7 +410,7 @@ def connected_components_stack(
       return_N=True, out_dtype=np.uint64,
       binary_image=bool(binary_image),
     )
-    cc_labels[cc_labels != 0] += offset
+    np.add(cc_labels, offset, out=cc_labels, where=(cc_labels != 0))
     offset += N
     binary = crackle.compress(cc_labels)
 

Original file line number	Diff line number	Diff line change
`@@ -410,7 +410,7 @@ def connected_components_stack(`
`410`	`410`	`return_N=True, out_dtype=np.uint64,`
`411`	`411`	`binary_image=bool(binary_image),`
`412`	`412`	`)`
`413`		`- cc_labels[cc_labels != 0] += offset`
	`413`	`+ np.add(cc_labels, offset, out=cc_labels, where=(cc_labels != 0))`
`414`	`414`	`offset += N`
`415`	`415`	`binary = crackle.compress(cc_labels)`
`416`	`416`