RayD/python/rayd/_impl/runtime_jit.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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
# Copyright Xingyu Chen.
# Implements shared Python support for runtime Dr.Jit.

import os as _os
import pathlib as _pathlib
import sys as _sys
from typing import TYPE_CHECKING as _TYPE_CHECKING

_drjit_dll_directory = None
if not _TYPE_CHECKING:
    import drjit as _drjit

    if _sys.platform == "win32" and hasattr(_os, "add_dll_directory"):
        _drjit_dll_directory = _os.add_dll_directory(str(_pathlib.Path(_drjit.__file__).resolve().parent))

from rayd.drjit._C import *  # noqa: F401,F403
from rayd._impl.capabilities_jit import api_manifest, backend_capabilities

# Explicit public surface. `from ._C import *` above is the import mechanism;
# this list is the contract. It equals every name bound by NB_MODULE(_C, m) in
# src/bindings/module_jit.cpp plus the two `_capabilities` helpers, and is
# recomputed from that source by tests/test_public_api_manifest.py and
# cross-checked against the built extension by tests/test_namespace_isolation.py.
__all__ = [
    "AccumGrid",
    "AccumOptions",
    "AccumResult",
    "AccumResultAD",
    "AxialEdgeVisibility",
    "AxialEdgeVisibilityAD",
    "DfrAccum",
    "DfrAccumAD",
    "DfrCoherentAccum",
    "DfrCoherentAccumAD",
    "DfrCoherentCandidatePairs",
    "DfrCoherentCandidatePairsAD",
    "DfrCoherentEdge",
    "DfrCoherentEdgeAD",
    "DfrCoherentOptions",
    "DfrCoherentUtdStates",
    "DfrCoherentUtdStatesAD",
    "DfrGrid",
    "DfrMaterial",
    "DfrMaterialAD",
    "DfrOptions",
    "DfrPathOptions",
    "DfrPaths",
    "DfrPathsAD",
    "DfrStates",
    "DfrStatesAD",
    "Intersection",
    "IntersectionAD",
    "Material",
    "MaterialAD",
    "Mesh",
    "MixedScene",
    "NearestEdgesTopK",
    "NearestEdgesTopKAD",
    "NearestPointEdge",
    "NearestPointEdgeAD",
    "NearestRayEdge",
    "NearestRayEdgeAD",
    "RAYD_DFR_DIRECT",
    "RAYD_DFR_HASH",
    "RAYD_DFR_KELLER",
    "RAYD_DFR_MATCHED_ISO",
    "RAYD_DFR_SOBOL",
    "RAYD_DFR_SUFFIX_REFL",
    "REFLECTION_EXPORT_COUNT_ONLY",
    "REFLECTION_EXPORT_FULL",
    "REFLECTION_EXPORT_MINIMAL",
    "Ray",
    "RayAD",
    "RayFlags",
    "ReflEpc",
    "ReflEpcAD",
    "ReflEpcField",
    "ReflEpcFieldAD",
    "ReflEpcFieldOptions",
    "ReflEpcFieldOptionsAD",
    "ReflEpcOptions",
    "ReflectionBounce",
    "ReflectionBounceAD",
    "ReflectionChain",
    "ReflectionChainAD",
    "ReflectionTrace",
    "ReflectionTraceAD",
    "ReflectionTraceOptions",
    "Scene",
    "SceneEdgeBVHStats",
    "SceneEdgeInfo",
    "SceneEdgeTopology",
    "SceneGeometry",
    "SceneSyncProfile",
    "SecondaryEdgeInfo",
    "SegmentChainVisibility",
    "SegmentChainVisibilityAD",
    "SegmentPairVisibility",
    "SegmentPairVisibilityAD",
    "SegmentVisibility",
    "SegmentVisibilityAD",
    "SdfGrid",
    "SdfIntersection",
    "SdfIntersectionAD",
    "SdfTraceOptions",
    "SurfelAppearance",
    "SurfelCloud",
    "SurfelColorModel",
    "SurfelComposite",
    "SurfelCompositeAD",
    "SurfelGeometry",
    "SurfelIntersection",
    "SurfelIntersectionAD",
    "SurfelPrimitiveMode",
    "SurfelRender",
    "SurfelRenderAD",
    "SurfelRenderMode",
    "SurfelRenderOptions",
    "SurfelScene",
    "SurfelTraceOptions",
    "WedgeEvents",
    "WedgeEventsAD",
    "api_manifest",
    "backend_capabilities",
    "current_device",
    "device_count",
    "native_launch_audit",
    "native_launch_audit_clear",
    "optix_available",
    "set_device",
]