feat(python): add Transform interface type

Author: thewtex

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

@@ -6,6 +6,7 @@
 from .image import Image, ImageType, ImageRegion
 from .pointset import PointSet, PointSetType
 from .mesh import Mesh, MeshType
+from .transform import Transform, TransformType, TransformParameterizations
 from .polydata import PolyData, PolyDataType
 from .binary_file import BinaryFile
 from .binary_stream import BinaryStream
@@ -42,6 +43,9 @@
     "MeshType",
     "PolyData",
     "PolyDataType",
+    "Transform",
+    "TransformType",
+    "TransformParameterizations",
     "BinaryFile",
     "BinaryStream",
     "TextFile",

packages/core/python/itkwasm/itkwasm/transform.py

@@ -0,0 +1,65 @@
+from dataclasses import dataclass, field
+from enum import Enum
+from typing import Sequence, Union, Dict, Optional
+
+try:
+    from numpy.typing import ArrayLike
+except ImportError:
+    from numpy import ndarray as ArrayLike
+import numpy as np
+
+class TransformParameterizations(str, Enum):
+    Composite = "Composite"
+    Identity = "Identity"
+    Translation = "Translation"
+    Euler2D = "Euler2D"
+    Euler3D = "Euler3D"
+    Rigid2D = "Rigid2D"
+    Rigid3D = "Rigid3D"
+    Rigid3DPerspective = "Rigid3DPerspective"
+    Versor = "Versor"
+    VersorRigid3D = "VersorRigid3D"
+    Scale = "Scale"
+    ScaleLogarithmic = "ScaleLogarithmic"
+    ScaleSkewVersor3D = "ScaleSkewVersor3D"
+    Similarity2D = "Similarity2D"
+    Similarity3D = "Similarity3D"
+    QuaternionRigid = "QuaternionRigid"
+    Affine = "Affine"
+    ScalableAffine = "ScalableAffine"
+    AzimuthElevationToCartesian = "AzimuthElevationToCartesian"
+    BSpline = "BSpline"
+    BSplineSmoothingOnUpdateDisplacementField = "BSplineSmoothingOnUpdateDisplacementField"
+    ConstantVelocityField = "ConstantVelocityField"
+    DisplacementField = "DisplacementField"
+    GaussianSmoothingOnUpdateDisplacementField = "GaussianSmoothingOnUpdateDisplacementField"
+    GaussianExponentialDiffeomorphic = "GaussianExponentialDiffeomorphic"
+    VelocityField = "VelocityField"
+    TimeVaryingVelocityField = "TimeVaryingVelocityField"
+    GaussianSmoothingOnUpdateTimeVaryingVelocityField = "GaussianSmoothingOnUpdateTimeVaryingVelocityField"
+
+@dataclass
+class TransformType:
+    inputDimension: int = 3
+    outputDimension: int = 3
+    transformParameterization: TransformParameterizations = TransformParameterizations.Identity
+
+@dataclass
+class Transform:
+    transformType: Union[TransformType, Dict] = field(default_factory=TransformType)
+    numberOfFixedParameters: int = 0
+    numberOfParameters: int = 0
+
+    name: str = "Transform"
+
+    inputSpaceName: str = ""
+    outputSpaceName: str = ""
+
+    fixedParameters: Optional[ArrayLike] = None
+    parameters: Optional[ArrayLike] = None
+
+    metadata: dict = field(default_factory=dict)
+
+    def __post_init__(self):
+        if isinstance(self.transformType, dict):
+            self.transformType = TransformType(**self.transformType)