Use multithreaded Loopbuilder for SACA

elevans · elevans · commit 449d2bc95680 · 2024-04-29T15:26:59.000-05:00
diff --git a/scijava-ops-image/src/main/java/org/scijava/ops/image/coloc/saca/AdaptiveSmoothedKendallTau.java b/scijava-ops-image/src/main/java/org/scijava/ops/image/coloc/saca/AdaptiveSmoothedKendallTau.java
@@ -90,7 +90,8 @@ private AdaptiveSmoothedKendallTau() {}
 		for (int s = 0; s < 3; s++)
 			stop.add(factory.create(image1));
 
-		LoopBuilder.setImages(oldsqrtN).forEachPixel(t -> t.setOne());
+		LoopBuilder.setImages(oldsqrtN).multiThreaded().forEachPixel(t -> t
+			.setOne());
 
 		for (int s = 0; s < TU; s++) {
 			intSize = (int) Math.floor(size);
@@ -100,7 +101,7 @@ private AdaptiveSmoothedKendallTau() {}
 			if (s == TL) {
 				isCheck = true;
 				LoopBuilder.setImages(stop.get(1), stop.get(2), newtau, newsqrtN)
-					.forEachPixel((ts1, ts2, tTau, tSqrtN) -> {
+					.multiThreaded().forEachPixel((ts1, ts2, tTau, tSqrtN) -> {
 						ts1.set(tTau);
 						ts2.set(tSqrtN);
 					});
@@ -197,11 +198,11 @@ private static <I extends RealType<I>> void singleiteration(
 
 		// TODO: instead of copying pixels here, swap oldTau and newTau every time.
 		// :-)
-		LoopBuilder.setImages(oldtau, newtau, oldsqrtN, newsqrtN).forEachPixel((
-			tOldTau, tNewTau, tOldSqrtN, tNewSqrtN) -> {
-			tOldTau.set(tNewTau);
-			tOldSqrtN.set(tNewSqrtN);
-		});
+		LoopBuilder.setImages(oldtau, newtau, oldsqrtN, newsqrtN).multiThreaded()
+			.forEachPixel((tOldTau, tNewTau, tOldSqrtN, tNewSqrtN) -> {
+				tOldTau.set(tNewTau);
+				tOldSqrtN.set(tNewSqrtN);
+			});
 	}
 
 	private static <I extends RealType<I>, T extends RealType<T>> void getData(
