Merge pull request #1246 from thewtex/point-set-io

point set io

Author: thewtex

include/itkWasmMeshIO.h

@@ -23,6 +23,7 @@
 #include <fstream>
 
 #include "itkMeshJSON.h"
+#include "itkPointSetJSON.h"
 #include "cbor.h"
 
 namespace itk
@@ -72,6 +73,9 @@ class WebAssemblyInterface_EXPORT WasmMeshIO: public MeshIOBase
   /** Set the JSON representation of the image information. */
   void SetJSON(const MeshJSON & json);
 
+  /** Set the JSON representation of the image information. */
+  void SetJSON(const PointSetJSON & json);
+
   /*-------- This part of the interfaces deals with writing data ----- */
 
   /** Writes the data to disk from the memory buffer provided. Make sure
@@ -93,6 +97,9 @@ class WebAssemblyInterface_EXPORT WasmMeshIO: public MeshIOBase
 #if !defined(ITK_WRAPPING_PARSER)
   /** Get the JSON representation of the mesh information. */
   auto GetJSON() -> MeshJSON;
+
+  /** Get the JSON representation of the point set information. */
+  auto GetPointSetJSON() -> PointSetJSON;
 #endif
 
 protected:

packages/compare-images/python/itkwasm-compare-images-emscripten/itkwasm_compare_images_emscripten/js_package.py

@@ -1,10 +1,10 @@
 """itkwasm: Python interface to itk-wasm WebAssembly modules."""
 
-__version__ = "1.0b178"
+__version__ = "1.0b179"
 
 from .interface_types import InterfaceTypes
 from .image import Image, ImageType, ImageRegion
-from .pointset import PointSet, PointSetType
+from .point_set import PointSet, PointSetType
 from .mesh import Mesh, MeshType
 from .transform import Transform, TransformType, TransformParameterizations
 from .polydata import PolyData, PolyDataType

packages/compress-stringify/python/itkwasm-compress-stringify-emscripten/itkwasm_compress_stringify_emscripten/js_package.py

@@ -7,6 +7,7 @@ class InterfaceTypes(str, Enum):
     BinaryStream = "BinaryStream"
     Image = "Image"
     Mesh = "Mesh"
+    PointSet = "PointSet"
     PolyData = "PolyData"
     Transform = "Transform"
     JsonCompatible = "JsonCompatible"

packages/core/python/itkwasm/itkwasm/__init__.py

@@ -22,6 +22,7 @@
 from .binary_file import BinaryFile
 from .image import Image
 from .mesh import Mesh
+from .point_set import PointSet
 from .polydata import PolyData
 from .json_compatible import JsonCompatible
 from .int_types import IntTypes
@@ -277,6 +278,27 @@ def run(
                     "cellData": f"data:application/vnd.itk.address,0:{cell_data_ptr}",
                 }
                 ri.set_input_json(mesh_json, index)
+            elif input_.type == InterfaceTypes.PointSet:
+                point_set = input_.data
+                if point_set.numberOfPoints:
+                    pv = array_like_to_bytes(point_set.points)
+                else:
+                    pv = bytes([])
+                points_ptr = ri.set_input_array(pv, index, 0)
+                if point_set.numberOfPointPixels:
+                    pdv = array_like_to_bytes(point_set.pointData)
+                else:
+                    pdv = bytes([])
+                point_data_ptr = ri.set_input_array(pdv, index, 1)
+                point_set_json = {
+                    "pointSetType": asdict(point_set.pointSetType),
+                    "name": point_set.name,
+                    "numberOfPoints": point_set.numberOfPoints,
+                    "points": f"data:application/vnd.itk.address,0:{points_ptr}",
+                    "numberOfPointPixels": point_set.numberOfPointPixels,
+                    "pointData": f"data:application/vnd.itk.address,0:{point_data_ptr}",
+                }
+                ri.set_input_json(point_set_json, index)
             elif input_.type == InterfaceTypes.PolyData:
                 polydata = input_.data
                 if polydata.numberOfPoints:
@@ -442,6 +464,31 @@ def run(
                         mesh.cellData = buffer_to_numpy_array(mesh.meshType.cellPixelComponentType, bytes([]))
 
                     output_data = PipelineOutput(InterfaceTypes.Mesh, mesh)
+                elif output.type == InterfaceTypes.PointSet:
+                    point_set_json = ri.get_output_json(index)
+                    point_set = PointSet(**point_set_json)
+
+                    if point_set.numberOfPoints > 0:
+                        data_ptr = ri.get_output_array_address(0, index, 0)
+                        data_size = ri.get_output_array_size(0, index, 0)
+                        point_set.points = buffer_to_numpy_array(
+                            point_set.pointSetType.pointComponentType,
+                            ri.wasmtime_lift(data_ptr, data_size),
+                        )
+                    else:
+                        point_set.points = buffer_to_numpy_array(point_set.pointSetType.pointComponentType, bytes([]))
+
+                    if point_set.numberOfPointPixels > 0:
+                        data_ptr = ri.get_output_array_address(0, index, 1)
+                        data_size = ri.get_output_array_size(0, index, 1)
+                        point_set.pointData = buffer_to_numpy_array(
+                            point_set.pointSetType.pointPixelComponentType,
+                            ri.wasmtime_lift(data_ptr, data_size),
+                        )
+                    else:
+                        point_set.pointData = buffer_to_numpy_array(point_set.pointSetType.pointPixelComponentType, bytes([]))
+
+                    output_data = PipelineOutput(InterfaceTypes.PointSet, point_set)
                 elif output.type == InterfaceTypes.PolyData:
                     polydata_json = ri.get_output_json(index)
                     polydata = PolyData(**polydata_json)

packages/core/python/itkwasm/itkwasm/interface_types.py

@@ -2,7 +2,7 @@
 from typing import Optional
 
 from .image import Image, ImageType
-from .pointset import PointSet, PointSetType
+from .point_set import PointSet, PointSetType
 from .mesh import Mesh, MeshType
 from .polydata import PolyData, PolyDataType
 from .binary_file import BinaryFile

packages/core/python/itkwasm/itkwasm/pipeline.py

@@ -47,6 +47,7 @@ dependencies = [
     "itk>=5.4.0",
     "pytest >=2.7.3",
     "pytest-pyodide",
+    "pyodide-py",
     "dask[array]",
     "test_accelerator @ {root:uri}/test/test-accelerator",
 ]
@@ -57,7 +58,7 @@ test = [
     "pytest --dist-dir=./dist --rt=chrome",
 ]
 download-pyodide = [
-  "curl -L https://github.com/pyodide/pyodide/releases/download/0.25.1/pyodide-0.25.1.tar.bz2 -o pyodide.tar.bz2",
+  "curl -L https://github.com/pyodide/pyodide/releases/download/0.26.2/pyodide-0.26.2.tar.bz2 -o pyodide.tar.bz2",
   "tar xjf pyodide.tar.bz2",
   "rm -rf dist pyodide.tar.bz2",
   "mv pyodide dist",

packages/core/python/itkwasm/itkwasm/pointset.py renamed to packages/core/python/itkwasm/itkwasm/point_set.py

@@ -1,6 +1,6 @@
 {
   "name": "itk-wasm",
-  "version": "1.0.0-b.178",
+  "version": "1.0.0-b.179",
   "packageManager": "[email protected]",
   "description": "High-performance spatial analysis in a web browser, Node.js, and reproducible execution across programming languages and hardware architectures.",
   "type": "module",