liblaf-melon 0.2.3__tar.gz → 0.2.4__tar.gz

{liblaf_melon-0.2.3 → liblaf_melon-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: liblaf-melon
-Version: 0.2.3
+Version: 0.2.4
 Summary: Add your description here
 Project-URL: Changelog, https://github.com/liblaf/melon/blob/main/CHANGELOG.md
 Project-URL: Documentation, https://liblaf.github.io/melon/

{liblaf_melon-0.2.3 → liblaf_melon-0.2.4}/src/liblaf/melon/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.2.3'
-__version_tuple__ = version_tuple = (0, 2, 3)
+__version__ = version = '0.2.4'
+__version_tuple__ = version_tuple = (0, 2, 4)

liblaf_melon-0.2.4/src/liblaf/melon/barycentric/_points.py

@@ -0,0 +1,15 @@
+import einops
+import numpy as np
+from jaxtyping import Float
+from numpy.typing import ArrayLike
+
+
+def barycentric_to_points(
+    cells: Float[ArrayLike, "*N B D"], barycentric: Float[ArrayLike, "*N B"]
+) -> Float[np.ndarray, "*N D"]:
+    cells: Float[np.ndarray, "*N B D"] = np.asarray(cells)
+    barycentric: Float[np.ndarray, "*N B"] = np.asarray(barycentric)
+    points: Float[np.ndarray, "*N D"] = einops.einsum(
+        barycentric, cells, "... B, ... B D -> ... D"
+    )
+    return points

{liblaf_melon-0.2.3 → liblaf_melon-0.2.4}/src/liblaf/melon/io/paraview/_series_writer.py

@@ -99,7 +99,6 @@ class SeriesWriter(Sequence[File], contextlib.AbstractContextManager):
         self, data: Any, *, time: float | None = None, timestep: float | None = None
     ) -> None:
         filename: str = f"{self.name}_{len(self):06d}{self.ext}"
-        ic(filename, self.ext, self.file)
         save(self.folder / filename, data)
         if time is None:
             if timestep is None:

{liblaf_melon-0.2.3 → liblaf_melon-0.2.4}/src/liblaf/melon/io/pyvista/point_set/convert/_array.py

@@ -8,7 +8,7 @@ from liblaf.melon.io.abc import AbstractConverter
 
 class ArrayToPointSet(AbstractConverter):
     @override
-    def match_from(self, data: Any) -> bool:
+    def match_from(self, data: Any, /) -> bool:
         return utils.is_array_like(data)
 
     @override

{liblaf_melon-0.2.3 → liblaf_melon-0.2.4}/src/liblaf/melon/io/pyvista/point_set/convert/_convert.py

@@ -2,7 +2,9 @@ import pyvista as pv
 
 from liblaf.melon.io.abc import ConverterDispatcher
 
+from ._array import ArrayToPointSet
 from ._poly_data import PolyDataToPointSet
 
 as_point_set = ConverterDispatcher(pv.PointSet)
+as_point_set.register(ArrayToPointSet())
 as_point_set.register(PolyDataToPointSet())

{liblaf_melon-0.2.3 → liblaf_melon-0.2.4}/src/liblaf/melon/io/pyvista/point_set/convert/_poly_data.py

@@ -8,7 +8,7 @@ from liblaf.melon.io.abc import AbstractConverter
 
 class PolyDataToPointSet(AbstractConverter):
     @override
-    def match_from(self, data: Any) -> bool:
+    def match_from(self, data: Any, /) -> bool:
         return utils.is_poly_data(data)
 
     @override

{liblaf_melon-0.2.3 → liblaf_melon-0.2.4}/src/liblaf/melon/utils/_typing.py

@@ -12,12 +12,32 @@ def is_subclass(cls: type, prefix: str, name: str) -> bool:
 
 def is_array_like(obj: Any) -> bool:
     # TODO: Implement a more robust check for array-like objects.
-    return isinstance(obj, Sequence)
+    return (
+        is_jax(obj)
+        or is_numpy(obj)
+        or is_torch(obj)
+        or is_warp(obj)
+        or isinstance(obj, Sequence)
+    )
 
 
 # ----------------------------------- numpy ---------------------------------- #
+
+
+def is_jax(obj: Any) -> bool:
+    return is_instance(obj, "jax", "Array")
+
+
 def is_numpy(obj: Any) -> bool:
-    return is_subclass(obj, "numpy", "ndarray")
+    return is_instance(obj, "numpy", "ndarray")
+
+
+def is_torch(obj: Any) -> bool:
+    return is_instance(obj, "torch", "Tensor")
+
+
+def is_warp(obj: Any) -> bool:
+    return is_instance(obj, "warp", "array")
 
 
 # ---------------------------------- pyvista --------------------------------- #

liblaf_melon-0.2.3/src/liblaf/melon/barycentric/_points.py

@@ -1,15 +0,0 @@
-import einops
-import numpy as np
-from jaxtyping import Float
-from numpy.typing import ArrayLike
-
-
-def barycentric_to_points(
-    cells: Float[ArrayLike, "N B D"], barycentric: Float[ArrayLike, "N B"]
-) -> Float[np.ndarray, "N D"]:
-    cells: Float[np.ndarray, "N B D"] = np.asarray(cells)
-    barycentric: Float[np.ndarray, "N B"] = np.asarray(barycentric)
-    points: Float[np.ndarray, "N D"] = einops.einsum(
-        barycentric, cells, "N B, N B D -> N D"
-    )
-    return points