unitfield 0.1.0__py3-none-any.whl

unitfield/__init__.py

@@ -0,0 +1,21 @@
+from .core import *
+
+Unit2DEndo = Unit2DMappedEndomorphism
+UnitEndo2D = Unit2DMappedEndomorphism
+U2DE = Unit2DMappedEndomorphism
+UEndo2D = Unit2DMappedEndomorphism
+UNDField = UnitNdimField
+MUnitField = MappedUnitField
+
+
+__version__ = "1.0.0"
+__all__ = [
+    'InterpMethod',
+    'UnitArray',
+    'UnitSpaceVector',
+    'UnitNdimField',
+    'MappedUnitField',
+    'UnitMappedEndomorphism',
+    'Unit2DMappedEndomorphism',
+    'remap_tensor_cv2'
+]

unitfield/core/__init__.py

@@ -0,0 +1,23 @@
+"""Unit field transformation module for coordinate mappings."""
+
+from .enums import InterpMethod
+from .types import UnitArray, UnitSpaceVector
+from .unitfield import (
+    UnitNdimField,
+    MappedUnitField,
+    UnitMappedEndomorphism,
+    Unit2DMappedEndomorphism,
+    remap_tensor_cv2
+)
+
+__version__ = "1.0.0"
+__all__ = [
+    'InterpMethod',
+    'UnitArray',
+    'UnitSpaceVector',
+    'UnitNdimField',
+    'MappedUnitField',
+    'UnitMappedEndomorphism',
+    'Unit2DMappedEndomorphism',
+    'remap_tensor_cv2'
+]

unitfield/core/enums.py

@@ -0,0 +1,43 @@
+#UnitFied\unitfield\core\enums.py
+"""
+Interpolation method enumerations for unit field transformations.
+"""
+
+from enum import Enum, unique
+from typing import Dict, Set
+
+
+
+class InterpMethod(Enum):
+    """Interpolation methods for unit field sampling."""
+    
+    NEAREST_MANHATTAN = "nearest"
+    LINEAR = "linear"
+    NEAREST_EUCLIDEAN = "euclidean"
+    LANCZOS4 = "lanczos4"
+    CUBIC = "cubic"
+    
+    # Backward compatibility alias
+    EUCLIDEAN = "euclidean"
+    
+    @classmethod
+    def get_cv2_methods(cls) -> Set['InterpMethod']:
+        """Get interpolation methods supported by OpenCV backend."""
+        return {
+            cls.NEAREST_MANHATTAN,
+            cls.LINEAR,
+            cls.NEAREST_EUCLIDEAN,
+            cls.CUBIC,
+            cls.LANCZOS4
+        }
+    
+    @classmethod
+    def get_numpy_methods(cls) -> Dict['InterpMethod', str]:
+        """Get mapping of interpolation methods to numpy backend names."""
+        return {
+            cls.NEAREST_MANHATTAN: "nearest_manhattan",
+            cls.LINEAR: "linear",
+            cls.NEAREST_EUCLIDEAN: "nearest_euclidean",
+            cls.CUBIC: "cubic",
+            cls.LANCZOS4: "lanczos4"
+        }

unitfield/core/types.py

@@ -0,0 +1,31 @@
+#UnitFied\unitfield\core\types.py
+"""
+Type definitions for unit field transformations.
+"""
+
+from typing import Annotated, Tuple, List, Union, TypeAlias, Final
+from typing_extensions import TypeAlias as TypeAliasExt
+import numpy as np
+from numpy.typing import NDArray
+from boundednumbers import UnitFloat
+
+
+# Unit array type (values in [0, 1])
+UnitArray: TypeAlias = Annotated[NDArray[np.floating], "values in [0, 1]"]
+
+# Unit space vector types
+UnitSpaceVector: TypeAlias = Union[
+    UnitArray,
+    Tuple[UnitFloat, ...],
+    List[UnitFloat],
+]
+
+# Type aliases for clarity
+Coordinate: TypeAlias = Tuple[UnitFloat, ...]
+PixelCoordinate: TypeAlias = Tuple[float, ...]
+ImageShape: TypeAlias = Tuple[int, int]
+
+# Constants
+MAX_CACHE_SIZE: Final[int] = 1024
+DEFAULT_CACHE_SIZE: Final[int] = 128
+DEFAULT_DTYPE: Final[type] = np.float32