add missing data types

clEsperanto · Aug 12, 2024 · 026a407 · 026a407
1 parent fef3529
commit 026a407
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 2 deletions.
diff --git a/src/main/java/net/clesperanto/imagej/ImageJConverters.java b/src/main/java/net/clesperanto/imagej/ImageJConverters.java
@@ -3,6 +3,7 @@
 import java.nio.ByteBuffer;
 import java.nio.ByteOrder;
 import java.nio.FloatBuffer;
+import java.nio.IntBuffer;
 import java.nio.ShortBuffer;
 import java.util.LinkedHashMap;
 import java.util.Map;
@@ -106,10 +107,25 @@ private static ImagePlus fromBuffer(ByteBuffer byteBuffer, ImageJDataType type,
 	        case UINT8:
 	            fillImage(im, dimensions, byteBuffer::get);
 	            break;
+	        case INT8:
+	            fillImage(im, dimensions, byteBuffer::get);
+	            break;
 	        case UINT16:
+	            ShortBuffer uShortBuff = byteBuffer.asShortBuffer();
+	            fillImage(im, dimensions, uShortBuff::get);
+	            break;
+	        case INT16:
 	            ShortBuffer shortBuff = byteBuffer.asShortBuffer();
 	            fillImage(im, dimensions, shortBuff::get);
 	            break;
+	        case UINT32:
+	        	IntBuffer uIntBuff = byteBuffer.asIntBuffer();
+	            fillImage(im, dimensions, uIntBuff::get);
+	            break;
+	        case INT32:
+	            IntBuffer intBuff = byteBuffer.asIntBuffer();
+	            fillImage(im, dimensions, intBuff::get);
+	            break;
 	        default:
 	            throw new IllegalArgumentException("Data type not supported.");
 	    }

diff --git a/src/main/java/net/clesperanto/imagej/ImageJDataType.java b/src/main/java/net/clesperanto/imagej/ImageJDataType.java
@@ -7,11 +7,14 @@
 import net.clesperanto.core.DataType;
 import net.clesperanto.core.DeviceJ;
 
-// TODO add all types ImagePlus.GRAY8, ImagePlus.GRAY16, ImagePlus.GRAY32, ImagePlus.COLOR_256 or ImagePlus.COLOR_RGB)
 public enum ImageJDataType {
     FLOAT32(DataType.fromString("float"), ImagePlus.GRAY32, float[].class, 32),
+    INT32(DataType.fromString("int"), ImagePlus.GRAY32, int[].class, 32),
+    UINT32(DataType.fromString("uint"), ImagePlus.GRAY32, int[].class, 32),
     UINT16(DataType.fromString("ushort"), ImagePlus.GRAY16, short[].class, 16),
-    UINT8(DataType.fromString("uchar"), ImagePlus.GRAY8, byte[].class, 8);
+    INT16(DataType.fromString("short"), ImagePlus.GRAY16, short[].class, 16),
+    UINT8(DataType.fromString("uchar"), ImagePlus.GRAY8, byte[].class, 8),
+    INT8(DataType.fromString("char"), ImagePlus.GRAY8, byte[].class, 8);
 
 	private final DataType dt;
     private final int imgDtype;
@@ -25,6 +28,10 @@ public enum ImageJDataType {
         this.bitDepth = bitDepth;
     }
 
+    public static ImageJDataType fromImagePlus(ImagePlus imp) {
+        return fromImgPlusDataType(imp.getType());
+    }
+
     public static ImageJDataType fromString(String dType) {
         for (ImageJDataType type : values()) {
             if (type.dt.getName().equals(dType)) {