Add methods for raw I/O with primitive arrays

ctrueden · ctrueden · commit 9f98e17edb89 · 2022-10-21T09:26:16.000-05:00
diff --git a/scijava/scijava-testutil/src/main/java/org/scijava/testutil/ArrayIO.java b/scijava/scijava-testutil/src/main/java/org/scijava/testutil/ArrayIO.java
@@ -0,0 +1,91 @@
+package org.scijava.testutil;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.nio.ByteBuffer;
+import java.nio.DoubleBuffer;
+import java.nio.FloatBuffer;
+import java.nio.IntBuffer;
+import java.nio.LongBuffer;
+import java.nio.ShortBuffer;
+
+/**
+ * Utility functions for reading and writing arrays simply.
+ *
+ * @author Curtis Rueden
+ */
+public final class ArrayIO {
+
+	private ArrayIO() {
+		// NB: Prevent instantiation of utility class.
+	}
+
+	public static byte[] bytes(final InputStream is) throws IOException {
+		return is.readAllBytes();
+	}
+
+	public static double[] doubles(final InputStream is) throws IOException {
+		final ByteBuffer bb = ByteBuffer.wrap(is.readAllBytes());
+		final DoubleBuffer view = bb.asDoubleBuffer();
+		return DoubleBuffer.allocate(view.limit()).put(view).array();
+	}
+
+	public static float[] floats(final InputStream is) throws IOException {
+		final ByteBuffer bb = ByteBuffer.wrap(is.readAllBytes());
+		final FloatBuffer view = bb.asFloatBuffer();
+		return FloatBuffer.allocate(view.limit()).put(view).array();
+	}
+
+	public static int[] ints(final InputStream is) throws IOException {
+		final ByteBuffer bb = ByteBuffer.wrap(is.readAllBytes());
+		final IntBuffer view = bb.asIntBuffer();
+		return IntBuffer.allocate(view.limit()).put(view).array();
+	}
+
+	public static long[] longs(final InputStream is) throws IOException {
+		final ByteBuffer bb = ByteBuffer.wrap(is.readAllBytes());
+		final LongBuffer view = bb.asLongBuffer();
+		return LongBuffer.allocate(view.limit()).put(view).array();
+	}
+
+	public static short[] shorts(final InputStream is) throws IOException {
+		final ByteBuffer bb = ByteBuffer.wrap(is.readAllBytes());
+		final ShortBuffer view = bb.asShortBuffer();
+		return ShortBuffer.allocate(view.limit()).put(view).array();
+	}
+
+	public static void writeBytes(final byte[] data, final OutputStream os) throws IOException {
+		os.write(data);
+	}
+
+	public static void writeDoubles(final double[] data, final OutputStream os) throws IOException {
+		final ByteBuffer bb = ByteBuffer.allocate(4 * data.length);
+		bb.asDoubleBuffer().put(DoubleBuffer.wrap(data));
+		os.write(bb.array());
+	}
+
+	public static void writeFloats(final float[] data, final OutputStream os) throws IOException {
+		final ByteBuffer bb = ByteBuffer.allocate(4 * data.length);
+		bb.asFloatBuffer().put(FloatBuffer.wrap(data));
+		os.write(bb.array());
+	}
+
+	public static void writeInts(final int[] data, final OutputStream os) throws IOException {
+		final ByteBuffer bb = ByteBuffer.allocate(4 * data.length);
+		bb.asIntBuffer().put(IntBuffer.wrap(data));
+		os.write(bb.array());
+	}
+
+	public static void writeLongs(final long[] data, final OutputStream os) throws IOException {
+		final ByteBuffer bb = ByteBuffer.allocate(8 * data.length);
+		bb.asLongBuffer().put(LongBuffer.wrap(data));
+		os.write(bb.array());
+	}
+
+	public static void writeShorts(final short[] data, final OutputStream os) throws IOException {
+		final ByteBuffer bb = ByteBuffer.allocate(4 * data.length);
+		bb.asShortBuffer().put(ShortBuffer.wrap(data));
+		os.write(bb.array());
+	}
+}
