support show bool result

bioimage-io · Nov 14, 2023 · e8da21c · e8da21c
1 parent 90a0915
commit e8da21c
Showing 1 changed file with 21 additions and 11 deletions.
diff --git a/src/components/TestRun.vue b/src/components/TestRun.vue
@@ -129,6 +129,16 @@ function insertCharAtPosition(originalString, charToInsert, position) {
 }
 
 function getConstructor(tpstr) {
+  /**
+Int8Array	int8	int8
+Int16Array	int16	int16
+Int32Array	int32	int32
+Uint8Array	uint8	uint8
+Uint16Array	uint16	uint16
+Uint32Array	uint32	uint32
+Float32Array	float32	float32
+Float64Array	float64	float64
+   */
   let Constructor;
   if (tpstr == "uint8") {
     Constructor = Uint8Array;
@@ -146,6 +156,8 @@ function getConstructor(tpstr) {
     Constructor = Float32Array;
   } else if (tpstr == "float64") {
     Constructor = Float64Array;
+  } else if (tpstr == "bool") {
+    Constructor = Uint8Array;
   } else {
     throw new Error("Unsupported dtype: " + tpstr);
   }
@@ -167,24 +179,22 @@ function getConstructor(tpstr) {
 //}
 
 function toTfJs(arr) {
-  /**
-Int8Array	int8	int8
-Int16Array	int16	int16
-Int32Array	int32	int32
-Uint8Array	uint8	uint8
-Uint16Array	uint16	uint16
-Uint32Array	uint32	uint32
-Float32Array	float32	float32
-Float64Array	float64	float64
-   */
   if (arr._rvalue instanceof ArrayBuffer) {
     arr._rvalue = new Uint8Array(arr._rvalue);
   }
   let buffer = new ArrayBuffer(arr._rvalue.length);
   let bufferView = new Uint8Array(buffer);
   bufferView.set(arr._rvalue);
   const Constructor = getConstructor(arr._rdtype);
-  const tarr = new Constructor(buffer);
+  let tarr = new Constructor(buffer);
+  if (arr._rdtype === "bool") {
+    // convert 1 to 255
+    for (let i = 0; i < tarr.length; i++) {
+      if (tarr[i] === 1) {
+        tarr[i] = 255;
+      }
+    }
+  }
   //const Constructor = getConstructor(arr._rdtype);
   //const bLen =  multiplyArrayElements(arr._rshape) * Constructor.BYTES_PER_ELEMENT;
   //const bytes = arr._rvalue.buffer.slice(0, bLen)