RayD/python/rayd/torch/__init__.py at main · Asixa/RayD · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
# Copyright Xingyu Chen.
# Exposes the public Torch Python package.

from __future__ import annotations

from typing import TYPE_CHECKING

from rayd._impl import runtime as _runtime

_NATIVE_AVAILABLE = _runtime.NATIVE_AVAILABLE
_C = _runtime.C
_EXTENSION_IMPORT_ERROR = _runtime.EXTENSION_IMPORT_ERROR

from rayd._impl.camera import Camera
from rayd._impl.capabilities import api_manifest, backend_capabilities
from rayd._impl.geometry import (
    AccumGrid,
    AccumOptions,
    AccumResult,
    AxialEdgeVisibility,
    DfrAccum,
    DfrCoherentAccum,
    DfrGrid,
    DfrMaterial,
    DfrPathLayout,
    DfrPaths,
    DfrStates,
    Intersection,
    NearestEdgesTopK,
    NearestPointEdge,
    NearestRayEdge,
    Ray,
    RayFlags,
    ReflEpc,
    ReflEpcField,
    ReflEpcOptions,
    ReflMaterial,
    ReflectionChain,
    SceneGlobalGeometry,
    SdfIntersection,
    SegmentChainVisibility,
    SegmentPairVisibility,
    WedgeEvents,
)
from rayd._impl.mixed import MixedScene
from rayd._impl.scene import Mesh
from rayd._impl.scene import Scene
from rayd._impl.sdf import SdfGrid, SdfGridBatch, SdfTraceOptions, sdf_intersect
from rayd._impl.surfel import SurfelCloud, SurfelComposite, SurfelIntersection, SurfelScene, SurfelTraceOptions

if TYPE_CHECKING:
    from rayd._impl.multi import MultiDeviceOptions as MultiDeviceOptions

if not TYPE_CHECKING:

    def __getattr__(name: str):
        if name == "MultiDeviceOptions":
            from rayd._impl.multi import MultiDeviceOptions as implementation

            globals()["MultiDeviceOptions"] = implementation
            return implementation
        raise AttributeError(f"module {__name__!r} has no attribute {name!r}")


__all__ = [
    "AccumGrid",
    "AccumOptions",
    "AccumResult",
    "DfrAccum",
    "DfrCoherentAccum",
    "DfrGrid",
    "DfrMaterial",
    "DfrPathLayout",
    "DfrPaths",
    "DfrStates",
    "Camera",
    "Intersection",
    "AxialEdgeVisibility",
    "Mesh",
    "MixedScene",
    "MultiDeviceOptions",
    "NearestPointEdge",
    "NearestEdgesTopK",
    "NearestRayEdge",
    "Ray",
    "RayFlags",
    "ReflEpc",
    "ReflEpcField",
    "ReflEpcOptions",
    "ReflMaterial",
    "ReflectionChain",
    "Scene",
    "SceneGlobalGeometry",
    "SdfGrid",
    "SdfGridBatch",
    "SdfIntersection",
    "SdfTraceOptions",
    "SurfelCloud",
    "SurfelComposite",
    "SurfelIntersection",
    "SurfelScene",
    "SurfelTraceOptions",
    "SegmentChainVisibility",
    "SegmentPairVisibility",
    "WedgeEvents",
    "api_manifest",
    "backend_capabilities",
    "sdf_intersect",
]