differt-core 0.0.27__cp39-cp39-win32.whl

differt_core/__init__.py

@@ -0,0 +1,18 @@
+"""
+Core package written in Rust and re-exported here.
+
+The present package only provides lower-level utilities with
+a focus on performances.
+
+Currently, Rust uses NumPy's C API to communicate between Python
+and Rust, hence casting NumPy arrays to JAX arrays in the process,
+thus making it impossible to trace variables used in Rust code.
+
+In the future, we plan on providing XLA-compatible functions,
+so that one can use :mod:`differt_core` and still be able to differentiate
+its code. We welcome any contribution on that topic!
+"""
+
+__all__ = ("__version__",)
+
+from differt_core._lowlevel import __version__

differt_core/_lowlevel/__init__.pyi

@@ -0,0 +1,7 @@
+from types import ModuleType
+
+__version__: str
+
+geometry: ModuleType
+rt: ModuleType
+scene: ModuleType

differt_core/_lowlevel/geometry/triangle_mesh.pyi

@@ -0,0 +1,16 @@
+import numpy as np
+from jaxtyping import Float, Int, UInt
+
+class TriangleMesh:
+    vertices: Float[np.ndarray, "num_vertices 3"]
+    triangles: UInt[np.ndarray, "num_triangles 3"]
+    face_colors: Float[np.ndarray, "num_triangles 3"] | None
+    face_materials: Int[np.ndarray, " num_triangles"] | None
+    material_names: list[str]
+    object_bounds: UInt[np.ndarray, "num_objects 2"] | None
+
+    def append(self, other: TriangleMesh) -> None: ...
+    @classmethod
+    def load_obj(cls, file: str) -> TriangleMesh: ...
+    @classmethod
+    def load_ply(cls, file: str) -> TriangleMesh: ...

differt_core/_lowlevel/rt/graph.pyi

@@ -0,0 +1,100 @@
+from collections.abc import Iterator, Sized
+
+import numpy as np
+from jaxtyping import Bool, UInt
+
+class CompleteGraph:
+    @property
+    def num_nodes(self) -> int: ...
+    def __init__(self, num_nodes: int) -> None: ...
+    def all_paths(
+        self,
+        from_: int,
+        to: int,
+        depth: int,
+        *,
+        include_from_and_to: bool = True,
+    ) -> AllPathsFromCompleteGraphIter: ...
+    def all_paths_array(
+        self,
+        from_: int,
+        to: int,
+        depth: int,
+        *,
+        include_from_and_to: bool = True,
+    ) -> UInt[np.ndarray, "num_paths path_depth"]: ...
+    def all_paths_array_chunks(
+        self,
+        from_: int,
+        to: int,
+        depth: int,
+        *,
+        include_from_and_to: bool = True,
+        chunk_size: int,
+    ) -> AllPathsFromCompleteGraphChunksIter: ...
+
+class DiGraph:
+    @property
+    def num_nodes(self) -> int: ...
+    @classmethod
+    def from_adjacency_matrix(
+        cls,
+        adjacency_matrix: Bool[np.ndarray, "num_nodes num_nodes"],
+    ) -> DiGraph: ...
+    @classmethod
+    def from_complete_graph(cls, graph: CompleteGraph) -> DiGraph: ...
+    def insert_from_and_to_nodes(
+        self,
+        *,
+        direct_path: bool = True,
+        from_adjacency: Bool[np.ndarray, " num_nodes"],
+        to_adjacency: Bool[np.ndarray, " num_nodes"],
+    ) -> tuple[int, int]: ...
+    def disconnect_nodes(self, *nodes: int, fast_mode: bool = True) -> None: ...
+    def all_paths(
+        self,
+        from_: int,
+        to: int,
+        depth: int,
+        *,
+        include_from_and_to: bool = True,
+    ) -> AllPathsFromDiGraphIter: ...
+    def all_paths_array(
+        self,
+        from_: int,
+        to: int,
+        depth: int,
+        *,
+        include_from_and_to: bool = True,
+    ) -> UInt[np.ndarray, "num_paths path_depth"]: ...
+    def all_paths_array_chunks(
+        self,
+        from_: int,
+        to: int,
+        depth: int,
+        *,
+        include_from_and_to: bool = True,
+        chunk_size: int,
+    ) -> AllPathsFromDiGraphChunksIter: ...
+
+class AllPathsFromCompleteGraphIter(Iterator, Sized):
+    def __iter__(self) -> AllPathsFromCompleteGraphIter: ...
+    def __next__(self) -> UInt[np.ndarray, " path_depth"]: ...
+    def __len__(self) -> int: ...
+    def count(self) -> int: ...
+
+class AllPathsFromCompleteGraphChunksIter(Iterator, Sized):
+    def __iter__(self) -> AllPathsFromCompleteGraphChunksIter: ...
+    def __next__(self) -> UInt[np.ndarray, "chunk_size path_depth"]: ...
+    def __len__(self) -> int: ...
+    def count(self) -> int: ...
+
+class AllPathsFromDiGraphIter(Iterator):
+    def __iter__(self) -> AllPathsFromDiGraphIter: ...
+    def __next__(self) -> UInt[np.ndarray, " path_depth"]: ...
+    def count(self) -> int: ...
+
+class AllPathsFromDiGraphChunksIter(Iterator):
+    def __iter__(self) -> AllPathsFromDiGraphChunksIter: ...
+    def __next__(self) -> UInt[np.ndarray, "chunk_size path_depth"]: ...
+    def count(self) -> int: ...

differt_core/_lowlevel/scene/sionna.pyi

@@ -0,0 +1,16 @@
+class SionnaScene:
+    shapes: dict[str, Shape]
+    materials: dict[str, Material]
+
+    @classmethod
+    def load_xml(cls, file: str) -> SionnaScene: ...
+
+class Material:
+    id: str
+    rgb: tuple[float, float, float]
+
+class Shape:
+    type: str
+    id: str
+    file: str
+    material_id: str

differt_core/_lowlevel/scene/triangle_scene.pyi

@@ -0,0 +1,7 @@
+from differt_core.geometry import TriangleMesh
+
+class TriangleScene:
+    mesh: list[TriangleMesh]
+
+    @classmethod
+    def load_xml(cls, file: str) -> TriangleScene: ...

differt_core/geometry/__init__.py

@@ -0,0 +1,5 @@
+"""Geometry utilities used by :mod:`differt.geometry`."""
+
+__all__ = ("TriangleMesh",)
+
+from ._triangle_mesh import TriangleMesh

differt_core/geometry/_triangle_mesh.py

@@ -0,0 +1,5 @@
+__all__ = ("TriangleMesh",)
+
+from differt_core import _lowlevel
+
+TriangleMesh = _lowlevel.geometry.triangle_mesh.TriangleMesh

differt_core/rt/__init__.py

@@ -0,0 +1,19 @@
+"""Ray Tracing utilities used by :mod:`differt.rt`."""
+
+__all__ = (
+    "AllPathsFromCompleteGraphChunksIter",
+    "AllPathsFromCompleteGraphIter",
+    "AllPathsFromDiGraphChunksIter",
+    "AllPathsFromDiGraphIter",
+    "CompleteGraph",
+    "DiGraph",
+)
+
+from ._graph import (
+    AllPathsFromCompleteGraphChunksIter,
+    AllPathsFromCompleteGraphIter,
+    AllPathsFromDiGraphChunksIter,
+    AllPathsFromDiGraphIter,
+    CompleteGraph,
+    DiGraph,
+)

differt_core/rt/_graph.py

@@ -0,0 +1,19 @@
+__all__ = (
+    "AllPathsFromCompleteGraphChunksIter",
+    "AllPathsFromCompleteGraphIter",
+    "AllPathsFromDiGraphChunksIter",
+    "AllPathsFromDiGraphIter",
+    "CompleteGraph",
+    "DiGraph",
+)
+
+from differt_core import _lowlevel
+
+AllPathsFromCompleteGraphChunksIter = (
+    _lowlevel.rt.graph.AllPathsFromCompleteGraphChunksIter
+)
+AllPathsFromCompleteGraphIter = _lowlevel.rt.graph.AllPathsFromCompleteGraphIter
+AllPathsFromDiGraphChunksIter = _lowlevel.rt.graph.AllPathsFromDiGraphChunksIter
+AllPathsFromDiGraphIter = _lowlevel.rt.graph.AllPathsFromDiGraphIter
+CompleteGraph = _lowlevel.rt.graph.CompleteGraph
+DiGraph = _lowlevel.rt.graph.DiGraph

differt_core/scene/__init__.py

@@ -0,0 +1,6 @@
+"""Scene utilities used by :mod:`differt.scene`."""
+
+__all__ = ("Material", "Shape", "SionnaScene", "TriangleScene")
+
+from ._sionna import Material, Shape, SionnaScene
+from ._triangle_scene import TriangleScene

differt_core/scene/_sionna.py

@@ -0,0 +1,7 @@
+__all__ = ("Material", "Shape", "SionnaScene")
+
+from differt_core import _lowlevel
+
+Material = _lowlevel.scene.sionna.Material
+Shape = _lowlevel.scene.sionna.Shape
+SionnaScene = _lowlevel.scene.sionna.SionnaScene

differt_core/scene/_triangle_scene.py

@@ -0,0 +1,5 @@
+__all__ = ("TriangleScene",)
+
+from differt_core import _lowlevel
+
+TriangleScene = _lowlevel.scene.triangle_scene.TriangleScene

differt_core-0.0.27.dist-info/METADATA

@@ -0,0 +1,57 @@
+Metadata-Version: 2.4
+Name: differt-core
+Version: 0.0.27
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Topic :: Scientific/Engineering
+Classifier: Programming Language :: Rust
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Programming Language :: Python :: Implementation :: PyPy
+Requires-Dist: numpy >=1.20
+License-File: LICENSE.md
+Summary: Core backend of DifffeRT implemented in Rust
+Keywords: ray tracing,differentiable,propagation,radio,jax
+Author-email: Jérome Eertmans <jeertmans@icloud.com>
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
+
+<div align="center">
+<img src="https://raw.githubusercontent.com/jeertmans/DiffeRT/main/static/logo_250px.png" alt="DiffeRT logo"></img>
+</div>
+
+<div align="center">
+
+# DiffeRT-core
+
+[![Latest Release][pypi-version-badge]][pypi-version-url]
+[![Python version][pypi-python-version-badge]][pypi-version-url]
+[![Documentation][documentation-badge]][documentation-url]
+
+</div>
+
+This package contains the core backend of
+[DiffeRT](https://pypi.org/project/DiffeRT/),
+implemented in Rust for performances.
+
+As a result, both `differt` and `differt-core` will
+share the same version, and `differt` directly depends on `differt-core`.
+
+However, you can decide to only install `differt-core`
+if you want to use features that are specific to this package.
+
+The installation procedure, contributing guidelines, and documentation,
+are shared with the
+[main DiffeRT package](https://github.com/jeertmans/DiffeRT).
+
+[pypi-version-badge]: https://img.shields.io/pypi/v/DiffeRT-core?label=DiffeRT-core&color=blueviolet
+[pypi-version-url]: https://pypi.org/project/DiffeRT-core/
+[pypi-python-version-badge]: https://img.shields.io/pypi/pyversions/DiffeRT-core?color=orange
+[documentation-badge]: https://readthedocs.org/projects/differt/badge/?version=latest
+[documentation-url]: https://differt.readthedocs.io/latest/?badge=latest
+

differt_core-0.0.27.dist-info/RECORD

@@ -0,0 +1,20 @@
+differt_core-0.0.27.dist-info/METADATA,sha256=RAYsJBFJh3XNoMyO08LZZBPx5KA3NDLW_r75efXDdxg,2344
+differt_core-0.0.27.dist-info/WHEEL,sha256=NURljkXZ8ggkKSIhkO4o3o0rGa4Zpk_geyDZjUnUrBA,90
+differt_core-0.0.27.dist-info/licenses/LICENSE.md,sha256=IYiRvvMH922nW-enFUx36Wp1OaGDVd8B9EV9ickkRG4,1099
+differt_core/geometry/_triangle_mesh.py,sha256=Z3hnUcVaXwdFACIq0k4iBx1LU2hdKTdxvpLZJcsH_fI,131
+differt_core/geometry/__init__.py,sha256=4wErUp7LhmeLX07bacsjxMCgYsIegRSm-mSKsrfO-6k,134
+differt_core/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+differt_core/rt/_graph.py,sha256=7Xv2gsrS-OdjnBYoSF0lRhj2bCgBon3-ekyeRujYjYQ,672
+differt_core/rt/__init__.py,sha256=ZqStqeyQiaNH0V7MEfuq2BOo7CIqtsUNW7NW50DTMbk,470
+differt_core/scene/_sionna.py,sha256=JsY4d2m3hepomwMFjtU78UYRGJQUNUSj4Xl7HEFR-DA,220
+differt_core/scene/_triangle_scene.py,sha256=WvvbYeNMylWSveMrOOwMcSIcnW_CEAgjkBsGA6yrVYU,132
+differt_core/scene/__init__.py,sha256=Lt9sU9N4bFCTThk2MFl5RnP2hlI4posOp0tH-abDm-Y,217
+differt_core/_lowlevel/geometry/triangle_mesh.pyi,sha256=yVXlmkjKYq20eP_Hk02kX80QNVtDdMxVyXpmbbzca2s,606
+differt_core/_lowlevel/rt/graph.pyi,sha256=PfcDJoJ9V1N8pZeWfFFM8AG8rmdKinC0YOC1fdERg_k,3089
+differt_core/_lowlevel/rt/__init__.pyi,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+differt_core/_lowlevel/scene/sionna.pyi,sha256=tUX3Cf6z3Ie2JB_BlqrKvGAkW2RvYDdZ2Mdp7p4N4OY,310
+differt_core/_lowlevel/scene/triangle_scene.pyi,sha256=Arp_KWBw2W7YAoccLUHbYOu17I6u3IERz7x7Mr4kTTU,178
+differt_core/_lowlevel/__init__.pyi,sha256=-cviOkUXZ-cSCDZmRz8XBK9M3AmGAwICl7muOYIryRE,109
+differt_core/__init__.py,sha256=U_wgb4hzzl7DDL_uTuqPkzlnYnFrZP7KZQRyonyoRm0,628
+differt_core/_lowlevel.cp39-win32.pyd,sha256=JQXfXO7yfUcuhxuRyc3v5kLRfaRMtVDp4dveP4rVmDw,873984
+differt_core-0.0.27.dist-info/RECORD,,

differt_core-0.0.27.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: maturin (1.7.6)
+Root-Is-Purelib: false
+Tag: cp39-cp39-win32

differt_core-0.0.27.dist-info/licenses/LICENSE.md

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023-2024 Jérome Eertmans
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.