- Added generalized functions for horizontal and vertical down sampling

Author: lessthanoptimal

main/boofcv-ip/src/main/java/boofcv/alg/filter/misc/AverageDownSampleOps.java

@@ -21,7 +21,9 @@
 import boofcv.alg.filter.misc.impl.*;
 import boofcv.concurrency.BoofConcurrency;
 import boofcv.struct.image.*;
+import org.ddogleg.struct.DogArray_F32;
 import org.jetbrains.annotations.Nullable;
+import pabeles.concurrency.GrowArray;
 
 ///
 /// Operations related to down sampling image by computing the average within square regions. The first square region is
@@ -180,6 +182,78 @@ void down( T input, boolean centered, T output, @Nullable ImageGray middle ) {
 		}
 	}
 
+	/// Downsample the horizontal axis only. Output must be [GrayF32] or [GrayF64], depending on image type.
+	public static void horizontal( ImageGray input, boolean centered, ImageGray output ) {
+		if (ImageGray.class.isAssignableFrom(input.getClass())) {
+			if (BoofConcurrency.USE_CONCURRENT) {
+				if (input instanceof GrayU8) {
+					ImplAverageDownSample_MT.horizontal((GrayU8)input, centered, (GrayF32)output);
+				} else if (input instanceof GrayU16) {
+					ImplAverageDownSample_MT.horizontal((GrayU16)input, centered, (GrayF32)output);
+				} else if (input instanceof GrayF32) {
+					ImplAverageDownSample_MT.horizontal((GrayF32)input, centered, (GrayF32)output);
+				} else if (input instanceof GrayF64) {
+					ImplAverageDownSample_MT.horizontal((GrayF64)input, centered, (GrayF64)output);
+				} else {
+					throw new IllegalArgumentException("Unknown image type");
+				}
+			} else {
+				if (input instanceof GrayU8) {
+					ImplAverageDownSample.horizontal((GrayU8)input, centered, (GrayF32)output);
+				} else if (input instanceof GrayU16) {
+					ImplAverageDownSample.horizontal((GrayU16)input, centered, (GrayF32)output);
+				} else if (input instanceof GrayF32) {
+					ImplAverageDownSample.horizontal((GrayF32)input, centered, (GrayF32)output);
+				} else if (input instanceof GrayF64) {
+					ImplAverageDownSample.horizontal((GrayF64)input, centered, (GrayF64)output);
+				} else {
+					throw new IllegalArgumentException("Unknown image type");
+				}
+			}
+		} else if (Planar.class.isAssignableFrom(input.getClass())) {
+			throw new IllegalArgumentException("Image type not supported yet");
+		}
+	}
+
+	/// Downsample vertical axis only. Input must be [GrayF32] or [GrayF64]. The workspace is only
+	/// used with integer output image types
+	public static void vertical( ImageBase input, boolean centered, ImageBase output,
+								 @Nullable GrowArray<DogArray_F32> workspaces ) {
+		if (workspaces == null && output.getImageType().getDataType().isInteger()) {
+			workspaces = new GrowArray<>(DogArray_F32::new);
+		}
+
+		if (ImageGray.class.isAssignableFrom(input.getClass())) {
+			if (BoofConcurrency.USE_CONCURRENT) {
+				if (input instanceof GrayU8) {
+					ImplAverageDownSample_MT.vertical((GrayF32)input, centered, (GrayI8)output, workspaces);
+				} else if (input instanceof GrayU16) {
+					ImplAverageDownSample_MT.vertical((GrayF32)input, centered, (GrayU16)output, workspaces);
+				} else if (input instanceof GrayF32) {
+					ImplAverageDownSample_MT.vertical((GrayF32)input, centered, (GrayF32)output);
+				} else if (input instanceof GrayF64) {
+					ImplAverageDownSample_MT.vertical((GrayF64)input, centered, (GrayF64)output);
+				} else {
+					throw new IllegalArgumentException("Unknown image type");
+				}
+			} else {
+				if (input instanceof GrayU8) {
+					ImplAverageDownSample.vertical((GrayF32)input, centered, (GrayI8)output, workspaces);
+				} else if (input instanceof GrayU16) {
+					ImplAverageDownSample.vertical((GrayF32)input, centered, (GrayU16)output, workspaces);
+				} else if (input instanceof GrayF32) {
+					ImplAverageDownSample.vertical((GrayF32)input, centered, (GrayF32)output);
+				} else if (input instanceof GrayF64) {
+					ImplAverageDownSample.vertical((GrayF64)input, centered, (GrayF64)output);
+				} else {
+					throw new IllegalArgumentException("Unknown image type");
+				}
+			}
+		} else if (Planar.class.isAssignableFrom(input.getClass())) {
+			throw new IllegalArgumentException("Image type not supported yet");
+		}
+	}
+
 	/// Down samples a planar image. Type checking is done at runtime.
 	///
 	/// @param input Input image. Not modified.