supporting 16bits images

pkg/core.go

@@ -15,6 +15,7 @@ type ImagePayload struct {
 	Height         int // Useful to decode the current payload
 	Width          int
 	Channels       int
+	BitDepth       int
 }
 
 type Sample struct {

pkg/serdes.go

@@ -82,6 +82,8 @@ func imageFromBuffer(buffer []byte, transform *ARAwareTransform, aspect_ratio fl
 	// If requested, re-encode the image to a jpg or png
 	var img_bytes []byte
 	var channels int
+	var bit_depth int
+
 	if pre_encode_image {
 		if err != nil {
 			return nil, -1., err
@@ -93,12 +95,15 @@ func imageFromBuffer(buffer []byte, transform *ARAwareTransform, aspect_ratio fl
 			if err != nil {
 				return nil, -1., err
 			}
+			bit_depth = 8
 		} else {
 			// Re-encode the image to a png
-			img_bytes, _, err = img.ExportPng(vips.NewPngExportParams())
+			png_export_params := vips.NewPngExportParams()
+			img_bytes, _, err = img.ExportPng(png_export_params)
 			if err != nil {
 				return nil, -1., err
 			}
+			bit_depth = png_export_params.Bitdepth
 		}
 		channels = -1 // Signal that we have encoded the image
 	} else {
@@ -107,6 +112,13 @@ func imageFromBuffer(buffer []byte, transform *ARAwareTransform, aspect_ratio fl
 			return nil, -1., err
 		}
 		channels = img.Bands()
+
+		// Define bit depth de facto, not exposed in the vips interface
+		bit_depth = len(img_bytes) / (width * height * channels)
+	}
+
+	if bit_depth == 0 && !pre_encode_image {
+		panic("Bit depth not set")
 	}
 
 	img_payload := ImagePayload{
@@ -116,6 +128,7 @@ func imageFromBuffer(buffer []byte, transform *ARAwareTransform, aspect_ratio fl
 		Height:         height,
 		Width:          width,
 		Channels:       channels,
+		BitDepth:       bit_depth,
 	}
 
 	return &img_payload, aspect_ratio, nil

python/go_types.py

@@ -12,8 +12,10 @@ def uint8_array_to_numpy(go_array):
     # We export them from Go with a Channels dimension of -1 to mark them as dimensionless.
     # Anything else is a valid number of channels and will thus lead to a reshape
     num_final_channels = max(go_array.Channels, 1)
+    bit_depth = getattr(go_array, "BitDepth", 8)
+
     length = (
-        go_array.Width * go_array.Height * num_final_channels
+        go_array.Width * go_array.Height * num_final_channels * bit_depth
         if go_array.Channels > 0
         else len(go_array.Data)
     )