hyperray 0.0.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (788) hide show
  1. hyperdrone/__init__.py +53 -0
  2. hyperdrone/_cuda/__init__.py +76 -0
  3. hyperdrone/_dlpack.py +30 -0
  4. hyperdrone/_jit/__init__.py +131 -0
  5. hyperdrone/_jit/_build.py +113 -0
  6. hyperdrone/_jit/_lock.py +23 -0
  7. hyperdrone/_jit/_patches.py +99 -0
  8. hyperdrone/_jit/_toolchain.py +17 -0
  9. hyperdrone/_jit/_workspace.py +94 -0
  10. hyperdrone/_native/cmake/hyperdrone_jit.cmake +65 -0
  11. hyperdrone/_native/common/bindings.h +39 -0
  12. hyperdrone/_native/common/cuda_staging.h +80 -0
  13. hyperdrone/_native/common/jit_host.h +69 -0
  14. hyperdrone/_native/common/models.h +24 -0
  15. hyperdrone/_native/common/verbosity.h +52 -0
  16. hyperdrone/_native/dynamics/CMakeLists.txt +63 -0
  17. hyperdrone/_native/dynamics/host.cpp +188 -0
  18. hyperdrone/_native/dynamics/iface.h +71 -0
  19. hyperdrone/_native/dynamics/impl.cpp +638 -0
  20. hyperdrone/_native/env/CMakeLists.txt +98 -0
  21. hyperdrone/_native/env/environment.cpp +190 -0
  22. hyperdrone/_native/env/environment.h +282 -0
  23. hyperdrone/_native/env/impl.cpp +779 -0
  24. hyperdrone/_native/env/rotorcraft_bindings.cpp +53 -0
  25. hyperdrone/_native/render/CMakeLists.txt +96 -0
  26. hyperdrone/_native/render/renderer.cpp +973 -0
  27. hyperdrone/_native/render/rig_bindings.h +152 -0
  28. hyperdrone/_native/render/scene_bindings.cpp +515 -0
  29. hyperdrone/_outputs.py +32 -0
  30. hyperdrone/_parameters.py +36 -0
  31. hyperdrone/_vendor/rl-tools/CMakeLists.txt +207 -0
  32. hyperdrone/_vendor/rl-tools/LICENSE +21 -0
  33. hyperdrone/_vendor/rl-tools/cmake/autodetect/all.cmake +34 -0
  34. hyperdrone/_vendor/rl-tools/cmake/autodetect/git-diff.cmake +100 -0
  35. hyperdrone/_vendor/rl-tools/cmake/autodetect/git-hash.cmake +24 -0
  36. hyperdrone/_vendor/rl-tools/cmake/autodetect/summary.cmake +65 -0
  37. hyperdrone/_vendor/rl-tools/cmake/autodetect/tier0-compiler.cmake +18 -0
  38. hyperdrone/_vendor/rl-tools/cmake/autodetect/tier1-blas.cmake +51 -0
  39. hyperdrone/_vendor/rl-tools/cmake/autodetect/tier2-json-hdf5-zlib.cmake +82 -0
  40. hyperdrone/_vendor/rl-tools/cmake/autodetect/tier3-tensorboard.cmake +25 -0
  41. hyperdrone/_vendor/rl-tools/cmake/autodetect/tier4-cuda.cmake +42 -0
  42. hyperdrone/_vendor/rl-tools/cmake/autodetect/tier5-cli11.cmake +24 -0
  43. hyperdrone/_vendor/rl-tools/cmake/dependencies/assimp.cmake +19 -0
  44. hyperdrone/_vendor/rl-tools/cmake/legacy_flags.cmake +33 -0
  45. hyperdrone/_vendor/rl-tools/cmake/optional/assimp.cmake +30 -0
  46. hyperdrone/_vendor/rl-tools/cmake/optional/googletest.cmake +20 -0
  47. hyperdrone/_vendor/rl-tools/cmake/optional/metal.cmake +25 -0
  48. hyperdrone/_vendor/rl-tools/cmake/optional/mujoco.cmake +89 -0
  49. hyperdrone/_vendor/rl-tools/cmake/optional/optix.cmake +25 -0
  50. hyperdrone/_vendor/rl-tools/cmake/optional/raytracing.cmake +105 -0
  51. hyperdrone/_vendor/rl-tools/cmake/optional/raytracing_generic.cmake +2 -0
  52. hyperdrone/_vendor/rl-tools/cmake/optional/vulkan.cmake +4 -0
  53. hyperdrone/_vendor/rl-tools/cmake/optional/webgpu.cmake +24 -0
  54. hyperdrone/_vendor/rl-tools/cmake/scripts/embed_text.cmake +7 -0
  55. hyperdrone/_vendor/rl-tools/cmake/scripts/generate-git-snapshot.cmake +138 -0
  56. hyperdrone/_vendor/rl-tools/include/conta/conta.h +1480 -0
  57. hyperdrone/_vendor/rl-tools/include/metra/metra.h +381 -0
  58. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/matrix/matrix.h +143 -0
  59. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/matrix/operations_arm.h +33 -0
  60. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/matrix/operations_cpu.h +314 -0
  61. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/matrix/operations_cpu_accelerate.h +37 -0
  62. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/matrix/operations_cpu_blas.h +67 -0
  63. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/matrix/operations_cpu_mkl.h +38 -0
  64. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/matrix/operations_cpu_openblas.h +34 -0
  65. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/matrix/operations_cuda.h +289 -0
  66. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/matrix/operations_dummy.h +20 -0
  67. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/matrix/operations_esp32.h +29 -0
  68. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/matrix/operations_generic.h +944 -0
  69. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/matrix/operations_wasm32.h +13 -0
  70. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/matrix/persist_code.h +136 -0
  71. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/tensor/operations_arm.h +1 -0
  72. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/tensor/operations_cpu.h +94 -0
  73. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/tensor/operations_cpu_accelerate.h +25 -0
  74. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/tensor/operations_cpu_blas.h +115 -0
  75. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/tensor/operations_cpu_mkl.h +25 -0
  76. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/tensor/operations_cpu_openblas.h +25 -0
  77. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/tensor/operations_cuda.h +713 -0
  78. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/tensor/operations_generic.h +1356 -0
  79. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/tensor/persist_code.h +130 -0
  80. hyperdrone/_vendor/rl-tools/include/rl_tools/containers/tensor/tensor.h +490 -0
  81. hyperdrone/_vendor/rl-tools/include/rl_tools/devices/arm.h +91 -0
  82. hyperdrone/_vendor/rl-tools/include/rl_tools/devices/cpu.h +210 -0
  83. hyperdrone/_vendor/rl-tools/include/rl_tools/devices/cpu_accelerate.h +23 -0
  84. hyperdrone/_vendor/rl-tools/include/rl_tools/devices/cpu_blas.h +21 -0
  85. hyperdrone/_vendor/rl-tools/include/rl_tools/devices/cpu_emscripten.h +20 -0
  86. hyperdrone/_vendor/rl-tools/include/rl_tools/devices/cpu_mkl.h +45 -0
  87. hyperdrone/_vendor/rl-tools/include/rl_tools/devices/cpu_openblas.h +56 -0
  88. hyperdrone/_vendor/rl-tools/include/rl_tools/devices/cpu_tensorboard.h +34 -0
  89. hyperdrone/_vendor/rl-tools/include/rl_tools/devices/cuda.h +464 -0
  90. hyperdrone/_vendor/rl-tools/include/rl_tools/devices/cuda_streams.h +23 -0
  91. hyperdrone/_vendor/rl-tools/include/rl_tools/devices/devices.h +105 -0
  92. hyperdrone/_vendor/rl-tools/include/rl_tools/devices/dummy.h +55 -0
  93. hyperdrone/_vendor/rl-tools/include/rl_tools/devices/esp32.h +85 -0
  94. hyperdrone/_vendor/rl-tools/include/rl_tools/devices/generic.h +0 -0
  95. hyperdrone/_vendor/rl-tools/include/rl_tools/devices/wasm32.h +59 -0
  96. hyperdrone/_vendor/rl-tools/include/rl_tools/dyn/model.h +218 -0
  97. hyperdrone/_vendor/rl-tools/include/rl_tools/dyn/operations_generic.h +769 -0
  98. hyperdrone/_vendor/rl-tools/include/rl_tools/dyn/persist.h +278 -0
  99. hyperdrone/_vendor/rl-tools/include/rl_tools/dyn/policy_adapter.h +100 -0
  100. hyperdrone/_vendor/rl-tools/include/rl_tools/dyn/policy_adapter_persist.h +52 -0
  101. hyperdrone/_vendor/rl-tools/include/rl_tools/dyn/tensor_operations_generic.h +55 -0
  102. hyperdrone/_vendor/rl-tools/include/rl_tools/inference/applications/l2f/c_backend.h +117 -0
  103. hyperdrone/_vendor/rl-tools/include/rl_tools/inference/applications/l2f/c_interface.h +36 -0
  104. hyperdrone/_vendor/rl-tools/include/rl_tools/inference/applications/l2f/l2f.h +98 -0
  105. hyperdrone/_vendor/rl-tools/include/rl_tools/inference/applications/l2f/operations_dyn.h +18 -0
  106. hyperdrone/_vendor/rl-tools/include/rl_tools/inference/applications/l2f/operations_generic.h +268 -0
  107. hyperdrone/_vendor/rl-tools/include/rl_tools/inference/debugging_pool/c_backend.h +31 -0
  108. hyperdrone/_vendor/rl-tools/include/rl_tools/inference/debugging_pool/c_interface.h +12 -0
  109. hyperdrone/_vendor/rl-tools/include/rl_tools/inference/executor/c_backend.h +98 -0
  110. hyperdrone/_vendor/rl-tools/include/rl_tools/inference/executor/c_interface.h +50 -0
  111. hyperdrone/_vendor/rl-tools/include/rl_tools/inference/executor/executor.h +133 -0
  112. hyperdrone/_vendor/rl-tools/include/rl_tools/inference/executor/helper.h +109 -0
  113. hyperdrone/_vendor/rl-tools/include/rl_tools/inference/executor/operations_generic.h +219 -0
  114. hyperdrone/_vendor/rl-tools/include/rl_tools/logging/operations_arduino.h +117 -0
  115. hyperdrone/_vendor/rl-tools/include/rl_tools/logging/operations_arm.h +56 -0
  116. hyperdrone/_vendor/rl-tools/include/rl_tools/logging/operations_cpu.h +57 -0
  117. hyperdrone/_vendor/rl-tools/include/rl_tools/logging/operations_cpu_tensorboard.h +151 -0
  118. hyperdrone/_vendor/rl-tools/include/rl_tools/logging/operations_cuda.h +65 -0
  119. hyperdrone/_vendor/rl-tools/include/rl_tools/logging/operations_dummy.h +45 -0
  120. hyperdrone/_vendor/rl-tools/include/rl_tools/logging/operations_wasm32.h +56 -0
  121. hyperdrone/_vendor/rl-tools/include/rl_tools/math/operations_arm.h +98 -0
  122. hyperdrone/_vendor/rl-tools/include/rl_tools/math/operations_cpu.h +138 -0
  123. hyperdrone/_vendor/rl-tools/include/rl_tools/math/operations_cuda.h +380 -0
  124. hyperdrone/_vendor/rl-tools/include/rl_tools/math/operations_dummy.h +93 -0
  125. hyperdrone/_vendor/rl-tools/include/rl_tools/math/operations_esp32.h +92 -0
  126. hyperdrone/_vendor/rl-tools/include/rl_tools/math/operations_generic.h +295 -0
  127. hyperdrone/_vendor/rl-tools/include/rl_tools/math/operations_wasm32.h +230 -0
  128. hyperdrone/_vendor/rl-tools/include/rl_tools/mode/mode.h +98 -0
  129. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/activation_functions.h +113 -0
  130. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/capability/capability.h +61 -0
  131. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/capability/persist_code.h +31 -0
  132. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/group.h +99 -0
  133. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/avg_pool2d/layer.h +122 -0
  134. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/avg_pool2d/operations_cuda.h +64 -0
  135. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/avg_pool2d/operations_generic.h +186 -0
  136. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/avg_pool2d/persist.h +33 -0
  137. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/conv2d/layer.h +300 -0
  138. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/conv2d/operations_cpu_mkl.h +590 -0
  139. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/conv2d/operations_cuda.h +809 -0
  140. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/conv2d/operations_generic.h +1219 -0
  141. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/conv2d/persist.h +99 -0
  142. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/conv2d/persist_code.h +170 -0
  143. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dense/layer.h +145 -0
  144. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dense/operations_arm/dsp.h +63 -0
  145. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dense/operations_arm/opt.h +173 -0
  146. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dense/operations_cpu.h +13 -0
  147. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dense/operations_cpu_accelerate.h +66 -0
  148. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dense/operations_cpu_blas.h +282 -0
  149. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dense/operations_cpu_mkl.h +66 -0
  150. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dense/operations_cpu_openblas.h +65 -0
  151. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dense/operations_cuda.h +519 -0
  152. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dense/operations_dummy.h +8 -0
  153. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dense/operations_esp32/dsp.h +58 -0
  154. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dense/operations_esp32/opt.h +167 -0
  155. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dense/operations_generic.h +429 -0
  156. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dense/persist.h +62 -0
  157. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dense/persist_code.h +178 -0
  158. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dense/persist_common.h +59 -0
  159. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dynamic_conv2d/layer.h +188 -0
  160. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dynamic_conv2d/operations_cuda.h +356 -0
  161. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dynamic_conv2d/operations_generic.h +536 -0
  162. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/dynamic_conv2d/persist.h +32 -0
  163. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/embedding/layer.h +119 -0
  164. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/embedding/operations_generic.h +223 -0
  165. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/embedding/persist.h +46 -0
  166. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/flatten/layer.h +121 -0
  167. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/flatten/operations_generic.h +154 -0
  168. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/flatten/persist.h +44 -0
  169. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/flatten/persist_code.h +106 -0
  170. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/gru/helper_operations_cuda.h +602 -0
  171. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/gru/helper_operations_generic.h +251 -0
  172. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/gru/layer.h +206 -0
  173. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/gru/operations_generic.h +889 -0
  174. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/gru/persist.h +75 -0
  175. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/gru/persist_code.h +199 -0
  176. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/layers.h +3 -0
  177. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/max_pool2d/layer.h +143 -0
  178. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/max_pool2d/operations_cuda.h +80 -0
  179. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/max_pool2d/operations_generic.h +216 -0
  180. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/max_pool2d/persist.h +39 -0
  181. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/operations_cpu.h +1 -0
  182. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/operations_cpu_accelerate.h +1 -0
  183. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/operations_cpu_blas.h +1 -0
  184. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/operations_cpu_mkl.h +3 -0
  185. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/operations_cpu_openblas.h +1 -0
  186. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/operations_cuda.h +9 -0
  187. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/operations_dummy.h +1 -0
  188. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/operations_generic.h +3 -0
  189. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/resnet_block/layer.h +223 -0
  190. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/resnet_block/operations_cuda.h +138 -0
  191. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/resnet_block/operations_generic.h +446 -0
  192. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/resnet_block/persist.h +70 -0
  193. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/sample_and_squash/layer.h +154 -0
  194. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/sample_and_squash/operations_cuda.h +119 -0
  195. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/sample_and_squash/operations_generic.h +502 -0
  196. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/sample_and_squash/persist.h +50 -0
  197. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/sample_and_squash/persist_code.h +173 -0
  198. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/standardize/layer.h +116 -0
  199. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/standardize/operations_cuda.h +156 -0
  200. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/standardize/operations_generic.h +295 -0
  201. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/standardize/persist.h +83 -0
  202. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/standardize/persist_code.h +163 -0
  203. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/td3_sampling/layer.h +123 -0
  204. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/td3_sampling/operations_generic.h +272 -0
  205. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/td3_sampling/persist.h +17 -0
  206. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/td3_sampling/persist_code.h +164 -0
  207. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/unflatten/layer.h +122 -0
  208. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/unflatten/operations_cuda.h +55 -0
  209. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/unflatten/operations_generic.h +160 -0
  210. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/unflatten/persist.h +47 -0
  211. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/unflatten/persist_code.h +106 -0
  212. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/upsample2d/layer.h +132 -0
  213. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/upsample2d/operations_cuda.h +141 -0
  214. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/layers/upsample2d/operations_generic.h +230 -0
  215. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/loss_functions/categorical_cross_entropy/operations_generic.h +102 -0
  216. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/loss_functions/mse/operations_cuda.h +97 -0
  217. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/loss_functions/mse/operations_generic.h +101 -0
  218. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/loss_functions/operations_generic.h +2 -0
  219. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/nn.h +14 -0
  220. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/operations_cpu.h +1 -0
  221. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/operations_cpu_accelerate.h +1 -0
  222. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/operations_cpu_blas.h +1 -0
  223. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/operations_cpu_mkl.h +1 -0
  224. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/operations_cpu_mux.h +16 -0
  225. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/operations_cpu_openblas.h +1 -0
  226. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/operations_cuda.h +2 -0
  227. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/operations_dummy.h +1 -0
  228. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/operations_generic.h +15 -0
  229. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/adam/adam.h +121 -0
  230. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/adam/instance/operations_cuda.h +98 -0
  231. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/adam/instance/operations_generic.h +150 -0
  232. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/adam/instance/persist.h +39 -0
  233. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/adam/instance/persist_code.h +80 -0
  234. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/adam/operations_cuda.h +85 -0
  235. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/adam/operations_generic.h +109 -0
  236. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/adam/persist.h +25 -0
  237. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/lamb/instance/operations_generic.h +107 -0
  238. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/lamb/lamb.h +40 -0
  239. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/lamb/operations_generic.h +68 -0
  240. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/optimizers.h +3 -0
  241. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/sgd/instance/operations_cuda.h +85 -0
  242. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/sgd/instance/operations_generic.h +140 -0
  243. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/sgd/instance/persist.h +37 -0
  244. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/sgd/instance/persist_code.h +72 -0
  245. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/sgd/operations_cuda.h +49 -0
  246. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/sgd/operations_generic.h +50 -0
  247. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/sgd/sgd.h +92 -0
  248. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/optimizers/update.h +54 -0
  249. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/parameters/operations_cuda.h +0 -0
  250. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/parameters/operations_generic.h +98 -0
  251. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/parameters/parameters.h +79 -0
  252. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/parameters/persist.h +34 -0
  253. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/parameters/persist_code.h +136 -0
  254. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/persist.h +2 -0
  255. hyperdrone/_vendor/rl-tools/include/rl_tools/nn/persist_code.h +27 -0
  256. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/behaviors/skip_parameter_gradients.h +65 -0
  257. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/group/README.md +107 -0
  258. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/group/model.h +19 -0
  259. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/group/persist_code.h +77 -0
  260. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/mlp/network.h +174 -0
  261. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/mlp/operations_cuda.h +10 -0
  262. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/mlp/operations_dummy.h +8 -0
  263. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/mlp/operations_generic.h +407 -0
  264. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/mlp/persist.h +56 -0
  265. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/mlp/persist_code.h +137 -0
  266. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/mlp_unconditional_stddev/network.h +55 -0
  267. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/mlp_unconditional_stddev/operations_generic.h +86 -0
  268. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/mlp_unconditional_stddev/persist.h +43 -0
  269. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/mlp_unconditional_stddev/persist_code.h +142 -0
  270. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/models.h +3 -0
  271. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/multi_agent_wrapper/model.h +148 -0
  272. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/multi_agent_wrapper/operations_generic.h +288 -0
  273. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/multi_agent_wrapper/persist.h +25 -0
  274. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/multi_agent_wrapper/persist_code.h +71 -0
  275. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/operations_cpu.h +0 -0
  276. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/operations_cuda.h +1 -0
  277. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/operations_dummy.h +1 -0
  278. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/operations_generic.h +4 -0
  279. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/parallel/model.h +290 -0
  280. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/parallel/operations_cuda.h +174 -0
  281. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/parallel/operations_generic.h +623 -0
  282. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/parallel/persist.h +80 -0
  283. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/parallel/persist_code.h +200 -0
  284. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/persist.h +1 -0
  285. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/persist_code.h +1 -0
  286. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/random_uniform/model.h +43 -0
  287. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/random_uniform/operations_generic.h +73 -0
  288. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/resnet/resnet.h +97 -0
  289. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/sequential/model.h +270 -0
  290. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/sequential/operations_generic.h +625 -0
  291. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/sequential/persist.h +99 -0
  292. hyperdrone/_vendor/rl-tools/include/rl_tools/nn_models/sequential/persist_code.h +115 -0
  293. hyperdrone/_vendor/rl-tools/include/rl_tools/numeric_types/bf16.h +28 -0
  294. hyperdrone/_vendor/rl-tools/include/rl_tools/numeric_types/categories.h +22 -0
  295. hyperdrone/_vendor/rl-tools/include/rl_tools/numeric_types/persist_code.h +54 -0
  296. hyperdrone/_vendor/rl-tools/include/rl_tools/numeric_types/policy.h +48 -0
  297. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/arm/group_1.h +18 -0
  298. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/arm/group_2.h +14 -0
  299. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/arm/group_3.h +14 -0
  300. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/arm.h +19 -0
  301. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu/group_1.h +17 -0
  302. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu/group_2.h +12 -0
  303. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu/group_3.h +15 -0
  304. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu.h +19 -0
  305. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_accelerate/group_1.h +16 -0
  306. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_accelerate/group_2.h +12 -0
  307. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_accelerate/group_3.h +15 -0
  308. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_accelerate.h +19 -0
  309. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_mkl/group_1.h +17 -0
  310. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_mkl/group_2.h +12 -0
  311. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_mkl/group_3.h +15 -0
  312. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_mkl.h +18 -0
  313. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_mux.h +158 -0
  314. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_openblas/group_1.h +16 -0
  315. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_openblas/group_2.h +12 -0
  316. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_openblas/group_3.h +15 -0
  317. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_openblas.h +18 -0
  318. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_tensorboard/group_1.h +16 -0
  319. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_tensorboard/group_2.h +12 -0
  320. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_tensorboard/group_3.h +15 -0
  321. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cpu_tensorboard.h +19 -0
  322. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cuda/group_1.h +18 -0
  323. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cuda/group_2.h +12 -0
  324. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cuda/group_3.h +12 -0
  325. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/cuda.h +19 -0
  326. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/dummy/group_1.h +15 -0
  327. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/dummy/group_2.h +12 -0
  328. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/dummy/group_3.h +12 -0
  329. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/dummy.h +19 -0
  330. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/esp32/group_1.h +16 -0
  331. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/esp32/group_2.h +12 -0
  332. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/esp32/group_3.h +13 -0
  333. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/esp32.h +19 -0
  334. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/wasm32/group_1.h +15 -0
  335. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/wasm32/group_2.h +12 -0
  336. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/wasm32/group_3.h +12 -0
  337. hyperdrone/_vendor/rl-tools/include/rl_tools/operations/wasm32.h +19 -0
  338. hyperdrone/_vendor/rl-tools/include/rl_tools/persist/backends/hdf5/hdf5.h +54 -0
  339. hyperdrone/_vendor/rl-tools/include/rl_tools/persist/backends/hdf5/operations_cpu.h +419 -0
  340. hyperdrone/_vendor/rl-tools/include/rl_tools/persist/backends/tar/io.h +55 -0
  341. hyperdrone/_vendor/rl-tools/include/rl_tools/persist/backends/tar/operations_cpu.h +58 -0
  342. hyperdrone/_vendor/rl-tools/include/rl_tools/persist/backends/tar/operations_generic.h +620 -0
  343. hyperdrone/_vendor/rl-tools/include/rl_tools/persist/backends/tar/operations_posix.h +63 -0
  344. hyperdrone/_vendor/rl-tools/include/rl_tools/persist/backends/tar/tar.h +74 -0
  345. hyperdrone/_vendor/rl-tools/include/rl_tools/persist/code.h +38 -0
  346. hyperdrone/_vendor/rl-tools/include/rl_tools/random/operations_arm.h +39 -0
  347. hyperdrone/_vendor/rl-tools/include/rl_tools/random/operations_cpu.h +81 -0
  348. hyperdrone/_vendor/rl-tools/include/rl_tools/random/operations_cuda.h +172 -0
  349. hyperdrone/_vendor/rl-tools/include/rl_tools/random/operations_dummy.h +72 -0
  350. hyperdrone/_vendor/rl-tools/include/rl_tools/random/operations_esp32.h +45 -0
  351. hyperdrone/_vendor/rl-tools/include/rl_tools/random/operations_generic.h +133 -0
  352. hyperdrone/_vendor/rl-tools/include/rl_tools/random/operations_generic_array.h +73 -0
  353. hyperdrone/_vendor/rl-tools/include/rl_tools/random/operations_wasm32.h +57 -0
  354. hyperdrone/_vendor/rl-tools/include/rl_tools/random/persist.h +77 -0
  355. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/camera.h +105 -0
  356. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/datasets/annotations/cache.h +26 -0
  357. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/datasets/annotations/free_space.h +60 -0
  358. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/datasets/annotations/operations_cpu.h +411 -0
  359. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/datasets/glb/glb.h +25 -0
  360. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/datasets/glb/operations_cpu.h +1288 -0
  361. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/datasets/operations_cpu.h +136 -0
  362. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/datasets/procthor/catalog.h +232 -0
  363. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/datasets/procthor/conversion.h +33 -0
  364. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/datasets/procthor/operations_cpu.h +123 -0
  365. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/datasets/procthor/procthor.h +39 -0
  366. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/generic/operations_cpu.h +717 -0
  367. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/generic/operations_generic.h +1283 -0
  368. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/metal/context.h +193 -0
  369. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/metal/device_source.h +17 -0
  370. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/metal/operations_cpu.h +1123 -0
  371. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/optix/device.h +207 -0
  372. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/optix/device_impl.h +1043 -0
  373. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/optix/operations_cuda.h +1747 -0
  374. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/optix/overlay_accel.h +64 -0
  375. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/specialization.h +37 -0
  376. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/vulkan/context.h +251 -0
  377. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/vulkan/device_source.h +48 -0
  378. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/vulkan/operations_cpu.h +2067 -0
  379. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/webgpu/bvh_sah.h +182 -0
  380. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/webgpu/context.h +280 -0
  381. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/webgpu/device_source.h +17 -0
  382. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/webgpu/operations_cpu.h +1405 -0
  383. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/operations_cpu_common.h +948 -0
  384. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/operations_cpu_mux.h +32 -0
  385. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/operations_cpu_post.h +96 -0
  386. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/renderer.h +484 -0
  387. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/save_cpu.h +314 -0
  388. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/scene.h +44 -0
  389. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/transforms_generic.h +25 -0
  390. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/types.h +45 -0
  391. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/rig/operations_cpu.h +119 -0
  392. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/rig/operations_generic.h +121 -0
  393. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/rig/rig.h +70 -0
  394. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/scene.h +132 -0
  395. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/segmentation_cpu.h +145 -0
  396. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/transforms.h +204 -0
  397. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/types.h +76 -0
  398. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/algorithms.h +6 -0
  399. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/operations_cpu.h +1 -0
  400. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/operations_cpu_mkl.h +1 -0
  401. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/operations_generic.h +1 -0
  402. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/loop/core/config.h +243 -0
  403. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/loop/core/operations_cuda.h +9 -0
  404. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/loop/core/operations_generic.h +205 -0
  405. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/loop/core/persist.h +109 -0
  406. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/loop/core/state.h +51 -0
  407. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/mixed_imitation.h +63 -0
  408. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/operations_cuda.h +190 -0
  409. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/operations_generic.h +332 -0
  410. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/operations_generic_mixed_imitation.h +81 -0
  411. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/persist.h +28 -0
  412. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/ppo.h +105 -0
  413. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/loop/core/approximators_gru.h +84 -0
  414. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/loop/core/approximators_mlp.h +61 -0
  415. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/loop/core/config.h +106 -0
  416. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/loop/core/operations_generic.h +193 -0
  417. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/loop/core/state.h +52 -0
  418. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/operations_cpu.h +9 -0
  419. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/operations_cpu_accelerate.h +2 -0
  420. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/operations_cpu_mkl.h +2 -0
  421. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/operations_cpu_mux.h +9 -0
  422. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/operations_cuda.h +131 -0
  423. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/operations_generic.h +580 -0
  424. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/sac.h +151 -0
  425. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/loop/core/approximators_gru.h +72 -0
  426. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/loop/core/approximators_mlp.h +54 -0
  427. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/loop/core/config.h +107 -0
  428. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/loop/core/operations_generic.h +174 -0
  429. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/loop/core/state.h +46 -0
  430. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/operations_cpu.h +9 -0
  431. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/operations_cpu_accelerate.h +2 -0
  432. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/operations_cpu_mkl.h +2 -0
  433. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/operations_cpu_mux.h +9 -0
  434. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/operations_cuda.h +117 -0
  435. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/operations_generic.h +359 -0
  436. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/td3.h +141 -0
  437. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/components.h +4 -0
  438. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/off_policy_runner/off_policy_runner.h +225 -0
  439. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/off_policy_runner/operations_cpu.h +95 -0
  440. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/off_policy_runner/operations_cpu_accelerate.h +21 -0
  441. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/off_policy_runner/operations_cpu_mkl.h +21 -0
  442. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/off_policy_runner/operations_cuda.h +237 -0
  443. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/off_policy_runner/operations_generic.h +498 -0
  444. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/off_policy_runner/operations_generic_per_env.h +112 -0
  445. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/collection.h +57 -0
  446. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/on_policy_runner.h +122 -0
  447. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/operations_cpu.h +2 -0
  448. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/operations_cpu_accelerate.h +8 -0
  449. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/operations_cpu_mkl.h +8 -0
  450. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/operations_cpu_mux.h +12 -0
  451. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/operations_cuda.h +183 -0
  452. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/operations_generic.h +366 -0
  453. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/operations_generic_common.h +74 -0
  454. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/persist.h +47 -0
  455. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/operations_cpu.h +2 -0
  456. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/operations_cpu_accelerate.h +1 -0
  457. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/operations_cpu_mkl.h +1 -0
  458. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/operations_generic.h +2 -0
  459. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/replay_buffer/operations_cpu.h +9 -0
  460. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/replay_buffer/operations_generic.h +123 -0
  461. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/replay_buffer/persist.h +57 -0
  462. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/replay_buffer/replay_buffer.h +93 -0
  463. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/running_normalizer/operations_generic.h +76 -0
  464. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/running_normalizer/persist.h +31 -0
  465. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/running_normalizer/running_normalizer.h +33 -0
  466. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environment_wrappers/operations_generic.h +58 -0
  467. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environment_wrappers/scale_observations/operations_generic.h +28 -0
  468. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environment_wrappers/scale_observations/wrapper.h +26 -0
  469. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environment_wrappers/wrappers.h +29 -0
  470. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/acrobot/acrobot.h +88 -0
  471. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/acrobot/operations_cpu.h +152 -0
  472. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/acrobot/operations_generic.h +202 -0
  473. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/batch/environment.h +40 -0
  474. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/batch/operations_cuda.h +204 -0
  475. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/batch/operations_generic.h +106 -0
  476. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/batch/operations_generic_common.h +26 -0
  477. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/batch/persist.h +33 -0
  478. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/car/car.h +119 -0
  479. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/car/operations_cpu.h +110 -0
  480. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/car/operations_generic.h +206 -0
  481. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/car/operations_json.h +59 -0
  482. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/car/track.h +23 -0
  483. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/car/ui.h +189 -0
  484. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/environments.h +28 -0
  485. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/flag/environment.h +87 -0
  486. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/flag/operations_cpu.h +212 -0
  487. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/flag/operations_generic.h +215 -0
  488. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/observation.h +75 -0
  489. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/operations_cpu.h +1063 -0
  490. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/operations_cuda.h +564 -0
  491. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/pose.h +253 -0
  492. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/presets.h +154 -0
  493. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/raptor/operations_cpu.h +26 -0
  494. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/raptor/operations_generic.h +91 -0
  495. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/raptor/raptor.h +71 -0
  496. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/rig/operations_cpu.h +190 -0
  497. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/rig/operations_generic.h +32 -0
  498. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/rig/rig.h +25 -0
  499. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/capture_cpu.h +86 -0
  500. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/capture_cuda.h +69 -0
  501. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/capture_optix.h +163 -0
  502. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/operations_cpu.h +223 -0
  503. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/operations_cpu_spectator.h +194 -0
  504. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/operations_cuda.h +1 -0
  505. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/operations_cuda_spectator.h +109 -0
  506. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/spectator.h +82 -0
  507. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/spy.h +107 -0
  508. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/ui.h +30 -0
  509. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/moving_gate/moving_gate.h +103 -0
  510. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/moving_gate/operations_cpu.h +279 -0
  511. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/moving_gate/operations_cuda.h +242 -0
  512. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/target_frame/operations_cpu.h +322 -0
  513. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/target_frame/operations_cuda.h +312 -0
  514. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/target_frame/operations_generic_position_hold.h +128 -0
  515. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/target_frame/position_hold.h +22 -0
  516. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/target_frame/target_frame.h +92 -0
  517. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/visual_inertial_localization/baseline.h +46 -0
  518. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/visual_inertial_localization/calibration.h +62 -0
  519. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/visual_inertial_localization/metrics.h +101 -0
  520. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/visual_inertial_localization/operations_cpu.h +374 -0
  521. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/visual_inertial_localization/visual_inertial_localization.h +126 -0
  522. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/world.h +333 -0
  523. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/inertial_velocity.h +27 -0
  524. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/metrics.h +266 -0
  525. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/multirotor.h +1138 -0
  526. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_cpu.h +2252 -0
  527. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/05_state_is_nan.h +148 -0
  528. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/10_sample_initial_parameters.h +228 -0
  529. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/20_initial_state.h +208 -0
  530. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/30_sample_initial_state.h +300 -0
  531. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/35_get_desired_state.h +63 -0
  532. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/40_observe.h +637 -0
  533. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/50_state_algebra.h +57 -0
  534. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/60_dynamics.h +134 -0
  535. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/70_post_integration.h +368 -0
  536. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/80_abs_diff.h +312 -0
  537. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic.h +367 -0
  538. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic_inertial_velocity.h +44 -0
  539. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_helper_generic.h +43 -0
  540. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_multitask_generic.h +70 -0
  541. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_multitask_generic_forward.h +24 -0
  542. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/default.h +209 -0
  543. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/arpl.h +112 -0
  544. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/crazyflie.h +127 -0
  545. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/crazyflie_openmv.h +119 -0
  546. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/flightmare.h +111 -0
  547. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/fs.h +113 -0
  548. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/mrs.h +110 -0
  549. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/soft.h +117 -0
  550. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/soft_rigid.h +118 -0
  551. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/x500.h +88 -0
  552. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/x500_real.h +88 -0
  553. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/x500_sim.h +114 -0
  554. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/init/default.h +56 -0
  555. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/registry.h +96 -0
  556. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/reward_functions/default.h +64 -0
  557. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/reward_functions/reward_functions.h +1 -0
  558. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/reward_functions/squared/operations_generic.h +202 -0
  559. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/reward_functions/squared/squared.h +52 -0
  560. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/termination/default.h +33 -0
  561. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/trajectories/lissajous.h +72 -0
  562. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/trajectories/trajectory.h +78 -0
  563. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/persist.h +21 -0
  564. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/persist_code.h +53 -0
  565. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/quaternion_helper.h +103 -0
  566. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/ui.h +180 -0
  567. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/memory/environment.h +69 -0
  568. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/memory/operations_cpu.h +46 -0
  569. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/memory/operations_generic.h +78 -0
  570. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/mujoco/ant/README.MD +5 -0
  571. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/mujoco/ant/ant.h +77 -0
  572. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/mujoco/ant/model.h +415 -0
  573. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/mujoco/ant/operations_cpu.h +162 -0
  574. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/mujoco/ant/persist.h +86 -0
  575. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/mujoco/ant/ui.h +141 -0
  576. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/mujoco/mujoco.h +1 -0
  577. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/multi_agent/bottleneck/bottleneck.h +113 -0
  578. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/multi_agent/bottleneck/operations_cpu.h +289 -0
  579. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/multi_agent/bottleneck/operations_generic.h +495 -0
  580. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/multi_agent/environments.h +16 -0
  581. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/observation.h +37 -0
  582. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/operations_cpu.h +2 -0
  583. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/operations_generic.h +33 -0
  584. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/pendulum/operations_cpu.h +146 -0
  585. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/pendulum/operations_generic.h +262 -0
  586. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/pendulum/pendulum.h +197 -0
  587. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/pendulum/ui.h +97 -0
  588. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/pendulum/ui_xeus.h +124 -0
  589. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/reacher/operations_cpu.h +426 -0
  590. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/reacher/operations_generic.h +424 -0
  591. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/reacher/reacher.h +137 -0
  592. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/loop.h +15 -0
  593. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/checkpoint/config.h +28 -0
  594. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/checkpoint/operations_cpu.h +233 -0
  595. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/checkpoint/persist.h +30 -0
  596. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/checkpoint/state.h +21 -0
  597. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/curriculum/config.h +25 -0
  598. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/curriculum/operations_generic.h +47 -0
  599. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/curriculum/persist.h +20 -0
  600. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/curriculum/state.h +17 -0
  601. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/evaluation/config.h +47 -0
  602. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/evaluation/operations_generic.h +106 -0
  603. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/evaluation/persist.h +40 -0
  604. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/evaluation/state.h +36 -0
  605. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/extrack/config.h +22 -0
  606. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/extrack/operations_cpu.h +44 -0
  607. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/extrack/persist.h +20 -0
  608. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/extrack/state.h +26 -0
  609. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/nn_analytics/config.h +37 -0
  610. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/nn_analytics/operations_cpu.h +113 -0
  611. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/nn_analytics/persist.h +20 -0
  612. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/nn_analytics/state.h +24 -0
  613. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/save_trajectories/config.h +43 -0
  614. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/save_trajectories/operations_cpu.h +190 -0
  615. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/save_trajectories/persist.h +30 -0
  616. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/save_trajectories/state.h +37 -0
  617. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/steps.h +4 -0
  618. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/timing/config.h +27 -0
  619. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/timing/operations_cpu.h +52 -0
  620. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/timing/persist.h +20 -0
  621. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/timing/state.h +25 -0
  622. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/operations_generic.h +3 -0
  623. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/rl.h +5 -0
  624. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/utils/evaluation/evaluation.h +106 -0
  625. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/utils/evaluation/operations_cpu.h +48 -0
  626. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/utils/evaluation/operations_generic.h +275 -0
  627. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/utils/validation.h +269 -0
  628. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/utils/validation_analysis.h +85 -0
  629. hyperdrone/_vendor/rl-tools/include/rl_tools/rl_tools.h +80 -0
  630. hyperdrone/_vendor/rl-tools/include/rl_tools/ui_server/client/client.h +50 -0
  631. hyperdrone/_vendor/rl-tools/include/rl_tools/ui_server/client/operations_boost.h +92 -0
  632. hyperdrone/_vendor/rl-tools/include/rl_tools/ui_server/client/operations_cpu.h +174 -0
  633. hyperdrone/_vendor/rl-tools/include/rl_tools/ui_server/client/operations_websocket.h +208 -0
  634. hyperdrone/_vendor/rl-tools/include/rl_tools/ui_server/server.h +448 -0
  635. hyperdrone/_vendor/rl-tools/include/rl_tools/universe.h +4 -0
  636. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/declarations_cpu.h +13 -0
  637. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/declarations_cpu_emscripten.h +15 -0
  638. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/operations_arm.h +29 -0
  639. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/operations_cpu.h +24 -0
  640. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/operations_cpu_emscripten.h +28 -0
  641. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/operations_cuda.h +33 -0
  642. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/operations_dummy.h +18 -0
  643. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/operations_esp32.h +20 -0
  644. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/operations_generic.h +19 -0
  645. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/operations_wasm32.h +18 -0
  646. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/env/operations_cpu.h +16 -0
  647. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/env/operations_generic.h +14 -0
  648. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/extrack/extrack.h +129 -0
  649. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/extrack/operations_cpu.h +588 -0
  650. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/generic/integrators.h +103 -0
  651. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/generic/memcpy.h +16 -0
  652. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/generic/tuple/operations_generic.h +48 -0
  653. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/generic/tuple/tuple.h +147 -0
  654. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/generic/typing.h +122 -0
  655. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/generic/vector_operations.h +172 -0
  656. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/polyak/operations_cuda.h +53 -0
  657. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/polyak/operations_generic.h +93 -0
  658. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/string/operations_generic.h +201 -0
  659. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/zlib/operations_cpu.h +45 -0
  660. hyperdrone/_vendor/rl-tools/include/rl_tools/version.h +28 -0
  661. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/CMakeLists.txt +15 -0
  662. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/README.md +214 -0
  663. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/CMakeLists.txt +11 -0
  664. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/generic/CMakeLists.txt +17 -0
  665. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/generic/freestanding_check.cpp +134 -0
  666. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/metal/CMakeLists.txt +31 -0
  667. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/metal/device.metal +912 -0
  668. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/metal/metal_cpp_impl.cpp +6 -0
  669. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/optix/CMakeLists.txt +51 -0
  670. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/optix/device.cu +1 -0
  671. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/optix/overlay_accel.cu +308 -0
  672. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/vulkan/CMakeLists.txt +37 -0
  673. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/vulkan/device.comp +970 -0
  674. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/vulkan/device_spirv.cpp +44 -0
  675. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/webgpu/CMakeLists.txt +31 -0
  676. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/webgpu/device.wgsl +1381 -0
  677. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/CMakeLists.txt +129 -0
  678. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/README.md +131 -0
  679. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/benchmark.cpp +180 -0
  680. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/benchmark_dynamic.cpp +491 -0
  681. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/download_paper_scene.py +74 -0
  682. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/ios_app/CMakeLists.txt +104 -0
  683. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/ios_app/Info.plist +36 -0
  684. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/ios_app/Sources/BenchmarkApp.swift +172 -0
  685. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/ios_app/benchmark.h +10 -0
  686. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/ios_app/benchmark.mm +46 -0
  687. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/ios_app/build.sh +17 -0
  688. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/rig.cpp +119 -0
  689. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix.cpp +1382 -0
  690. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix.h +5 -0
  691. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix_config.h.in +25 -0
  692. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix_identity.cmake +11 -0
  693. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix_main.cpp +5 -0
  694. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix_physics.cu +125 -0
  695. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix_physics.h +58 -0
  696. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix_physics_contract.h +179 -0
  697. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix_physics_cpu.cpp +77 -0
  698. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/camera_orbit.h +51 -0
  699. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/CMakeLists.txt +199 -0
  700. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/antialiasing.cpp +214 -0
  701. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/drone.cpp +320 -0
  702. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/drone_device.cu +359 -0
  703. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/environment/environment.h +96 -0
  704. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/environment/operations_cpu.h +263 -0
  705. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/example.cpp +273 -0
  706. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/fixed_pose.cpp +221 -0
  707. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/flow.cpp +120 -0
  708. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/interactive.cpp +1015 -0
  709. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/make_drone_glb.py +172 -0
  710. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/minimal.cpp +82 -0
  711. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/minimal_overlay.cpp +137 -0
  712. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/motion_blur.cpp +232 -0
  713. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/render_pose_trace.cpp +1829 -0
  714. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/render_pose_trace_motion_blur.cpp +647 -0
  715. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/CMakeLists.txt +147 -0
  716. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/README.md +15 -0
  717. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/job.sbatch +28 -0
  718. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/model.h +260 -0
  719. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/model_config.h +24 -0
  720. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/model_forward_cuda.h +397 -0
  721. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/model_operations.h +418 -0
  722. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/model_persist.h +45 -0
  723. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/model_student.h +70 -0
  724. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/scene.cpp +222 -0
  725. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/scene.h +68 -0
  726. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/.gitignore +4 -0
  727. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/CMakeLists.txt +4 -0
  728. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/Info.plist +24 -0
  729. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/Info_ios.plist +36 -0
  730. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/Sources/CameraManager.swift +120 -0
  731. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/Sources/ContentView.swift +703 -0
  732. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/Sources/DatasetRecorder.swift +285 -0
  733. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/Sources/YawPredictorApp.swift +25 -0
  734. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/YawPredictor.entitlements +8 -0
  735. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/build.sh +66 -0
  736. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/build_ios.sh +133 -0
  737. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/generate.sh +6 -0
  738. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/h5_to_tar.cpp +82 -0
  739. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/inference.cpp +166 -0
  740. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/inference.h +26 -0
  741. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/project.yml +30 -0
  742. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/test_tar_vs_hdf5.cpp +173 -0
  743. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/yaw_prediction_cuda.cu +699 -0
  744. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/yaw_prediction_dataset_overlay.cpp +475 -0
  745. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/yaw_prediction_distill_cuda.cu +995 -0
  746. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/yaw_prediction_eval.cu +457 -0
  747. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/yaw_prediction_real_world_eval.cpp +660 -0
  748. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/yaw_prediction_viewer.cu +443 -0
  749. hyperdrone/_vendor/rl-tools/src/rl/environments/hyperdrone/spy/CMakeLists.txt +63 -0
  750. hyperdrone/_vendor/rl-tools/src/rl/environments/hyperdrone/spy/example.cpp +101 -0
  751. hyperdrone/_vendor/rl-tools/src/rl/environments/hyperdrone/spy/ui.cpp +380 -0
  752. hyperdrone/dynamics/__init__.py +5 -0
  753. hyperdrone/dynamics/_component.py +31 -0
  754. hyperdrone/dynamics/_config.py +46 -0
  755. hyperdrone/dynamics/_sim.py +166 -0
  756. hyperdrone/env/__init__.py +23 -0
  757. hyperdrone/env/_multi_environment.py +566 -0
  758. hyperdrone/env/_rotorcraft.py +29 -0
  759. hyperdrone/env/_spy.py +39 -0
  760. hyperdrone/examples/.gitignore +3 -0
  761. hyperdrone/examples/Benchmark.ipynb +113 -0
  762. hyperdrone/examples/HyperDrone.ipynb +589 -0
  763. hyperdrone/examples/Orbiter.ipynb +496 -0
  764. hyperdrone/examples/Renderer.ipynb +218 -0
  765. hyperdrone/examples/__init__.py +16 -0
  766. hyperdrone/examples/articulation.py +55 -0
  767. hyperdrone/examples/benchmark.py +236 -0
  768. hyperdrone/examples/benchmark_minimal.py +65 -0
  769. hyperdrone/examples/data.py +71 -0
  770. hyperdrone/examples/drone_flythrough.py +169 -0
  771. hyperdrone/examples/hyperdrone_videos/my_world_preview.h +17 -0
  772. hyperdrone/examples/my_world.h +17 -0
  773. hyperdrone/examples/render_procthor.py +40 -0
  774. hyperdrone/examples/shots.py +108 -0
  775. hyperdrone/examples/spy.py +40 -0
  776. hyperdrone/examples/visual_inertial_localization.py +158 -0
  777. hyperdrone/gym/__init__.py +67 -0
  778. hyperdrone/render/__init__.py +57 -0
  779. hyperdrone/render/_component.py +51 -0
  780. hyperdrone/render/_config.py +76 -0
  781. hyperdrone/render/_renderer.py +498 -0
  782. hyperdrone/render/_rig.py +136 -0
  783. hyperdrone/render/_sampling.py +69 -0
  784. hyperdrone/render/_scene.py +38 -0
  785. hyperdrone/render/datasets.py +15 -0
  786. hyperray-0.0.1.dist-info/METADATA +519 -0
  787. hyperray-0.0.1.dist-info/RECORD +788 -0
  788. hyperray-0.0.1.dist-info/WHEEL +4 -0
@@ -0,0 +1,1123 @@
1
+ #include "../../../../version.h"
2
+ #include "../../../../rl_tools.h"
3
+ #if (defined(RL_TOOLS_DISABLE_INCLUDE_GUARDS) || !defined(RL_TOOLS_RENDERING_RAYTRACING_BACKENDS_METAL_OPERATIONS_CPU_H)) && (RL_TOOLS_USE_THIS_VERSION == 1)
4
+ #pragma once
5
+ #define RL_TOOLS_RENDERING_RAYTRACING_BACKENDS_METAL_OPERATIONS_CPU_H
6
+
7
+ #include "../../renderer.h"
8
+ #include "../../operations_cpu_common.h"
9
+ #include "../specialization.h"
10
+ #include "context.h"
11
+ #include "device_source.h"
12
+
13
+ #include <vector>
14
+ #include <cstring>
15
+ #include <string>
16
+
17
+ RL_TOOLS_NAMESPACE_WRAPPER_START
18
+ namespace rl_tools {
19
+ namespace rendering::raytracing::backends {
20
+ template <typename SPEC>
21
+ struct RendererState<rendering::raytracing::backends::Metal, SPEC>: metal::Context {};
22
+
23
+ template <typename SPEC>
24
+ struct LibraryState<rendering::raytracing::backends::Metal, SPEC> {};
25
+
26
+ template <typename SPEC>
27
+ struct SceneState<rendering::raytracing::backends::Metal, SPEC> {};
28
+ }
29
+
30
+ namespace rendering::raytracing::backends::metal{
31
+ template <typename SPEC>
32
+ Context& context(rendering::raytracing::Renderer<SPEC, rendering::raytracing::backends::Metal>& renderer){
33
+ return *renderer.backend;
34
+ }
35
+
36
+ inline NS::SharedPtr<MTL::Texture> make_texture(MTL::Device* device, const uint8_t* pixels, int width, int height, bool srgb){
37
+ auto descriptor = MTL::TextureDescriptor::texture2DDescriptor(srgb ? MTL::PixelFormatRGBA8Unorm_sRGB : MTL::PixelFormatRGBA8Unorm, width, height, false);
38
+ descriptor->setUsage(MTL::TextureUsageShaderRead);
39
+ descriptor->setStorageMode(MTL::StorageModeShared);
40
+ auto texture = NS::TransferPtr(device->newTexture(descriptor));
41
+ texture->replaceRegion(MTL::Region::Make2D(0, 0, width, height), 0, pixels, (NS::UInteger)width * 4);
42
+ return texture;
43
+ }
44
+
45
+ inline void wait_in_flight(Context& ctx){
46
+ if(ctx.in_flight.get() != nullptr){
47
+ ctx.in_flight->waitUntilCompleted();
48
+ ctx.in_flight.reset();
49
+ }
50
+ if(ctx.in_flight_collision.get() != nullptr){
51
+ ctx.in_flight_collision->waitUntilCompleted();
52
+ ctx.in_flight_collision.reset();
53
+ }
54
+ if(ctx.in_flight_update.get() != nullptr){
55
+ ctx.in_flight_update->waitUntilCompleted();
56
+ ctx.in_flight_update.reset();
57
+ }
58
+ }
59
+
60
+ // secondary_output: the flow passes' status buffer next to their flow OUTPUT
61
+ template <typename SPEC>
62
+ void encode_fullscreen_pass(Context& ctx, MTL::CommandBuffer* command_buffer, MTL::ComputePipelineState* pipeline, MTL::Buffer* output, float shutter_t = 0.f, MTL::Buffer* secondary_output = nullptr){
63
+ MTL::ComputeCommandEncoder* encoder = command_buffer->computeCommandEncoder();
64
+ encoder->setComputePipelineState(pipeline);
65
+ encoder->setBuffer(ctx.launch_params.get(), 0, bindings::LAUNCH_PARAMS);
66
+ encoder->setBuffer(ctx.cameras.get(), 0, bindings::CAMERAS_CLOSE);
67
+ encoder->setBuffer(ctx.cameras_open.get() != nullptr ? ctx.cameras_open.get() : ctx.cameras.get(), 0, bindings::CAMERAS_OPEN);
68
+ encoder->setBuffer(ctx.cameras_previous.get() != nullptr ? ctx.cameras_previous.get() : ctx.cameras.get(), 0, bindings::CAMERAS_PREVIOUS);
69
+ encoder->setBuffer(output, 0, bindings::OUTPUT);
70
+ if(secondary_output != nullptr){
71
+ encoder->setBuffer(secondary_output, 0, bindings::FLOW_STATUS_OUTPUT);
72
+ }
73
+ if constexpr (SPEC::ENABLE_DYNAMIC_MOTION_BLUR){
74
+ if(ctx.rgb_accumulator.get() != nullptr){
75
+ encoder->setBuffer(ctx.rgb_accumulator.get(), 0, bindings::RGB_ACCUMULATOR);
76
+ }
77
+ encoder->setBytes(&shutter_t, sizeof(float), bindings::SHUTTER);
78
+ }
79
+ encoder->setBuffer(ctx.mesh_records.get(), 0, bindings::MESH_RECORDS);
80
+ if(ctx.scene_lights.get() != nullptr){
81
+ encoder->setBuffer(ctx.scene_lights.get(), 0, bindings::SCENE_LIGHTS);
82
+ }
83
+ encoder->setAccelerationStructure(ctx.acceleration_structure.get(), bindings::ACCELERATION_STRUCTURE);
84
+ encoder->setBuffer(ctx.instance_record_base.get(), 0, bindings::INSTANCE_RECORD_BASE);
85
+ encoder->setBuffer(ctx.instance_data.get(), 0, bindings::INSTANCE_DATA);
86
+ encoder->setBuffer(ctx.overlay_structures.get(), 0, bindings::OVERLAY_STRUCTURES);
87
+ encoder->setBuffer(ctx.overlay_structures_previous.get() != nullptr ? ctx.overlay_structures_previous.get() : ctx.overlay_structures.get(), 0, bindings::OVERLAY_STRUCTURES_PREVIOUS);
88
+ encoder->setBuffer(ctx.overlay_attachments.get(), 0, bindings::OVERLAY_ATTACHMENTS);
89
+ encoder->setBuffer(ctx.instance_classes.get(), 0, bindings::INSTANCE_CLASSES);
90
+ if(ctx.observation.get() != nullptr){
91
+ encoder->setBuffer(ctx.observation.get(), 0, bindings::OBSERVATION);
92
+ }
93
+ if(ctx.flow_deltas.get() != nullptr){
94
+ encoder->setBuffer(ctx.flow_deltas.get(), 0, bindings::FLOW_DELTAS);
95
+ }
96
+ for(auto& object_acceleration_structure : ctx.object_acceleration_structures){
97
+ encoder->useResource(object_acceleration_structure.get(), MTL::ResourceUsageRead);
98
+ }
99
+ for(auto& overlay_acceleration_structure : ctx.overlay_acceleration_structures){
100
+ encoder->useResource(overlay_acceleration_structure.get(), MTL::ResourceUsageRead);
101
+ }
102
+ for(auto& overlay_acceleration_structure : ctx.overlay_acceleration_structures_previous){
103
+ encoder->useResource(overlay_acceleration_structure.get(), MTL::ResourceUsageRead);
104
+ }
105
+ for(auto& buffer : ctx.mesh_buffers){
106
+ encoder->useResource(buffer.get(), MTL::ResourceUsageRead);
107
+ }
108
+ for(auto& texture : ctx.mesh_textures){
109
+ encoder->useResource(texture.get(), MTL::ResourceUsageRead);
110
+ }
111
+ if(ctx.dummy_texture.get() != nullptr){
112
+ encoder->useResource(ctx.dummy_texture.get(), MTL::ResourceUsageRead);
113
+ }
114
+ encoder->dispatchThreads(MTL::Size::Make(SPEC::FB_WIDTH, SPEC::FB_HEIGHT, 1), MTL::Size::Make(8, 8, 1));
115
+ encoder->endEncoding();
116
+ }
117
+
118
+ template <typename SPEC>
119
+ void encode_resolve_pass(Context& ctx, MTL::CommandBuffer* command_buffer){
120
+ MTL::ComputeCommandEncoder* encoder = command_buffer->computeCommandEncoder();
121
+ encoder->setComputePipelineState(ctx.resolve_pipeline.get());
122
+ encoder->setBuffer(ctx.launch_params.get(), 0, bindings::LAUNCH_PARAMS);
123
+ if constexpr (SPEC::HAS_RGB){
124
+ encoder->setBuffer(ctx.frame_buffer.get(), 0, bindings::OUTPUT);
125
+ encoder->setBuffer(ctx.rgb_accumulator.get(), 0, bindings::RGB_ACCUMULATOR);
126
+ }
127
+ if constexpr (SPEC::HAS_OBSERVATION){
128
+ encoder->setBuffer(ctx.observation.get(), 0, bindings::OBSERVATION);
129
+ }
130
+ encoder->dispatchThreads(MTL::Size::Make(SPEC::FB_WIDTH, SPEC::FB_HEIGHT, 1), MTL::Size::Make(8, 8, 1));
131
+ encoder->endEncoding();
132
+ }
133
+
134
+ template <typename SPEC>
135
+ void encode_collision_pass(Context& ctx, MTL::CommandBuffer* command_buffer){
136
+ MTL::ComputeCommandEncoder* encoder = command_buffer->computeCommandEncoder();
137
+ encoder->setComputePipelineState(ctx.collision_pipeline.get());
138
+ encoder->setBuffer(ctx.launch_params.get(), 0, bindings::LAUNCH_PARAMS);
139
+ encoder->setBuffer(ctx.cameras.get(), 0, bindings::CAMERAS_CLOSE);
140
+ encoder->setBuffer(ctx.probe_directions.get(), 0, bindings::PROBE_DIRECTIONS);
141
+ encoder->setBuffer(ctx.collision_results.get(), 0, bindings::COLLISION_RESULTS);
142
+ encoder->setAccelerationStructure(ctx.acceleration_structure.get(), bindings::ACCELERATION_STRUCTURE);
143
+ encoder->setBuffer(ctx.overlay_structures.get(), 0, bindings::OVERLAY_STRUCTURES);
144
+ encoder->setBuffer(ctx.overlay_attachments.get(), 0, bindings::OVERLAY_ATTACHMENTS);
145
+ for(auto& object_acceleration_structure : ctx.object_acceleration_structures){
146
+ encoder->useResource(object_acceleration_structure.get(), MTL::ResourceUsageRead);
147
+ }
148
+ for(auto& overlay_acceleration_structure : ctx.overlay_acceleration_structures){
149
+ encoder->useResource(overlay_acceleration_structure.get(), MTL::ResourceUsageRead);
150
+ }
151
+ encoder->dispatchThreads(MTL::Size::Make(SPEC::NUM_CAMERAS, SPEC::NUM_PROBES, 1), MTL::Size::Make(8, 8, 1));
152
+ encoder->endEncoding();
153
+ }
154
+ }
155
+
156
+ template <typename DEVICE, typename SPEC>
157
+ void malloc(DEVICE& device, rendering::raytracing::Renderer<SPEC, rendering::raytracing::backends::Metal>& renderer){
158
+ namespace metal = rendering::raytracing::backends::metal;
159
+ static_assert(utils::typing::is_same_v<typename SPEC::T, float>, "The Metal raytracing backend requires T = float");
160
+
161
+ if constexpr (SPEC::ENABLE_OVERLAYS) {
162
+ malloc(device, renderer.transforms);
163
+ rendering::raytracing::detail::malloc_overlay_staging(renderer);
164
+ }
165
+ if constexpr (SPEC::HAS_FLOW_OVERLAYS) {
166
+ malloc(device, renderer.transforms_previous);
167
+ std::memset(data(renderer.transforms_previous), 0, decltype(renderer.transforms_previous)::SPEC::SIZE_BYTES);
168
+ }
169
+ if constexpr (SPEC::ENABLE_DYNAMIC_MOTION_BLUR) {
170
+ malloc(device, renderer.transforms_motion);
171
+ std::memset(data(renderer.transforms_motion), 0, decltype(renderer.transforms_motion)::SPEC::SIZE_BYTES);
172
+ }
173
+
174
+ renderer.backend = new rendering::raytracing::backends::RendererState<rendering::raytracing::backends::Metal, SPEC>{};
175
+ renderer.device.context = renderer.backend;
176
+ auto* ctx = renderer.backend;
177
+ ctx->device = NS::TransferPtr(MTL::CreateSystemDefaultDevice());
178
+ utils::assert_exit(device, ctx->device.get() != nullptr, "Metal: no default device available");
179
+ utils::assert_exit(device, ctx->device->supportsRaytracing(), "Metal: device does not support raytracing");
180
+ ctx->queue = NS::TransferPtr(ctx->device->newCommandQueue());
181
+
182
+ auto compile_options = NS::TransferPtr(MTL::CompileOptions::alloc()->init());
183
+ compile_options->setLanguageVersion(MTL::LanguageVersion3_0);
184
+ auto source = NS::RetainPtr(NS::String::string(rendering::raytracing::backends::metal::device_source(), NS::UTF8StringEncoding));
185
+ NS::Error* error = nullptr;
186
+ ctx->library = NS::TransferPtr(ctx->device->newLibrary(source.get(), compile_options.get(), &error));
187
+ if(ctx->library.get() == nullptr){
188
+ log(device, device.logger, devices::logging::Verbosity::error, rendering::raytracing::LOG_PREFIX, "Metal device source compilation failed: ", (error != nullptr ? error->localizedDescription()->utf8String() : "unknown error"));
189
+ utils::assert_exit(device, false, "Metal device source compilation failed");
190
+ }
191
+
192
+ constexpr typename SPEC::TI cam_pixels = SPEC::CAM_PIXELS;
193
+ // the output tensors alias the shared MTLBuffers the kernels write (see frame_buffer/
194
+ // depth_buffer/segmentation_buffer/collision_results/observation accessors)
195
+ if constexpr (SPEC::HAS_RGB) {
196
+ ctx->frame_buffer = NS::TransferPtr(ctx->device->newBuffer((size_t)SPEC::NUM_CAMERAS * cam_pixels * sizeof(uint32_t), MTL::ResourceStorageModeShared));
197
+ renderer.frame_buffer._data = (uint32_t*)ctx->frame_buffer->contents();
198
+ }
199
+ if constexpr (SPEC::HAS_DEPTH) {
200
+ ctx->depth_buffer = NS::TransferPtr(ctx->device->newBuffer((size_t)SPEC::NUM_CAMERAS * cam_pixels * sizeof(float), MTL::ResourceStorageModeShared));
201
+ renderer.depth_buffer._data = (float*)ctx->depth_buffer->contents();
202
+ }
203
+ if constexpr (SPEC::HAS_SEGMENTATION) {
204
+ ctx->segmentation_buffer = NS::TransferPtr(ctx->device->newBuffer((size_t)SPEC::NUM_CAMERAS * cam_pixels * sizeof(uint32_t), MTL::ResourceStorageModeShared));
205
+ renderer.segmentation_buffer._data = (uint32_t*)ctx->segmentation_buffer->contents();
206
+ }
207
+ if constexpr (SPEC::HAS_NORMALS) {
208
+ ctx->normals_buffer = NS::TransferPtr(ctx->device->newBuffer((size_t)SPEC::NUM_CAMERAS * cam_pixels * 3 * sizeof(float), MTL::ResourceStorageModeShared));
209
+ renderer.normals_buffer._data = (float*)ctx->normals_buffer->contents();
210
+ }
211
+ if constexpr (SPEC::HAS_FLOW) {
212
+ ctx->flow_buffer = NS::TransferPtr(ctx->device->newBuffer((size_t)SPEC::NUM_CAMERAS * cam_pixels * 2 * sizeof(float), MTL::ResourceStorageModeShared));
213
+ renderer.flow_buffer._data = (float*)ctx->flow_buffer->contents();
214
+ ctx->flow_status_buffer = NS::TransferPtr(ctx->device->newBuffer((size_t)SPEC::NUM_CAMERAS * cam_pixels * sizeof(uint32_t), MTL::ResourceStorageModeShared));
215
+ renderer.flow_status_buffer._data = (uint32_t*)ctx->flow_status_buffer->contents();
216
+ if constexpr (SPEC::HAS_FLOW_OVERLAYS) {
217
+ ctx->flow_deltas = NS::TransferPtr(ctx->device->newBuffer((size_t)2 * SPEC::NUM_OVERLAYS * SPEC::MAX_OVERLAY_INSTANCES * 12 * sizeof(float), MTL::ResourceStorageModeShared));
218
+ }
219
+ }
220
+ if constexpr (SPEC::HAS_FLOW_FORWARD) {
221
+ ctx->flow_forward_buffer = NS::TransferPtr(ctx->device->newBuffer((size_t)SPEC::NUM_CAMERAS * cam_pixels * 2 * sizeof(float), MTL::ResourceStorageModeShared));
222
+ renderer.flow_forward_buffer._data = (float*)ctx->flow_forward_buffer->contents();
223
+ ctx->flow_forward_status_buffer = NS::TransferPtr(ctx->device->newBuffer((size_t)SPEC::NUM_CAMERAS * cam_pixels * sizeof(uint32_t), MTL::ResourceStorageModeShared));
224
+ renderer.flow_forward_status_buffer._data = (uint32_t*)ctx->flow_forward_status_buffer->contents();
225
+ }
226
+ if constexpr (SPEC::HAS_OBSERVATION) {
227
+ static_assert(utils::typing::is_same_v<typename SPEC::OBSERVATION_T, float>, "The Metal raytracing backend requires OBSERVATION_T = float");
228
+ ctx->observation = NS::TransferPtr(ctx->device->newBuffer((size_t)SPEC::NUM_CAMERAS * cam_pixels * SPEC::OBSERVATION_CHANNELS * sizeof(float), MTL::ResourceStorageModeShared));
229
+ renderer.observation._data = (float*)ctx->observation->contents();
230
+ }
231
+ if constexpr (SPEC::ENABLE_DYNAMIC_MOTION_BLUR) {
232
+ // linear-radiance accumulators for the launch-level motion loop, resolved into the
233
+ // packed frame buffer / depth buffer by resolve_outputs after the last sample pass
234
+ if constexpr (SPEC::HAS_RGB) {
235
+ ctx->rgb_accumulator = NS::TransferPtr(ctx->device->newBuffer((size_t)SPEC::NUM_CAMERAS * cam_pixels * 3 * sizeof(float), MTL::ResourceStorageModeShared));
236
+ renderer.rgb_accumulator._data = (float*)ctx->rgb_accumulator->contents();
237
+ }
238
+ }
239
+ #if !RL_TOOLS_RENDERING_RAYTRACING_DISABLE_PROBE_RAYS
240
+ ctx->collision_results = NS::TransferPtr(ctx->device->newBuffer((size_t)SPEC::NUM_CAMERAS * SPEC::NUM_PROBES * sizeof(rendering::raytracing::CollisionResult), MTL::ResourceStorageModeShared));
241
+ renderer.collision_results._data = (rendering::raytracing::CollisionResult*)ctx->collision_results->contents();
242
+ #endif
243
+ ctx->launch_params = NS::TransferPtr(ctx->device->newBuffer(sizeof(metal::LaunchParams), MTL::ResourceStorageModeShared));
244
+
245
+ // camera tensors alias the shared MTLBuffers the encoders bind (see cameras(device,
246
+ // renderer)): host writes are consumed by the next launch with no staging copy
247
+ constexpr size_t camera_bytes = (size_t)SPEC::NUM_CAMERAS * sizeof(rendering::raytracing::Camera<typename SPEC::T>);
248
+ ctx->cameras = NS::TransferPtr(ctx->device->newBuffer(camera_bytes, MTL::ResourceStorageModeShared));
249
+ renderer.cameras._data = (rendering::raytracing::Camera<typename SPEC::T>*)ctx->cameras->contents();
250
+ if constexpr (SPEC::HAS_CAMERA_PAIR) {
251
+ ctx->cameras_open = NS::TransferPtr(ctx->device->newBuffer(camera_bytes, MTL::ResourceStorageModeShared));
252
+ renderer.cameras_open._data = (rendering::raytracing::Camera<typename SPEC::T>*)ctx->cameras_open->contents();
253
+ }
254
+ if constexpr (SPEC::HAS_FLOW) {
255
+ ctx->cameras_previous = NS::TransferPtr(ctx->device->newBuffer(camera_bytes, MTL::ResourceStorageModeShared));
256
+ renderer.cameras_previous._data = (rendering::raytracing::Camera<typename SPEC::T>*)ctx->cameras_previous->contents();
257
+ }
258
+ rendering::raytracing::detail::announce_backend(device, renderer);
259
+ }
260
+
261
+ template <typename DEVICE, typename SPEC, typename METADATA_T>
262
+ void init(DEVICE& device, rendering::raytracing::Renderer<SPEC, rendering::raytracing::backends::Metal>& renderer, const rendering::raytracing::Scene& scene, const rendering::raytracing::AssetPool& pool, const rendering::SceneMetadata<METADATA_T>& metadata){
263
+ validate_semantic_segmentation<SPEC>(device, scene, pool);
264
+ renderer.max_ray_length = (typename SPEC::T)metadata.max_ray_length;
265
+ rendering::raytracing::detail::announce_configuration<SPEC>(device);
266
+ rendering::raytracing::detail::resolve_environment<SPEC>(scene.environment, renderer.ambient_color, renderer.miss_color_0, renderer.miss_color_1);
267
+ namespace metal = rendering::raytracing::backends::metal;
268
+ auto& ctx = metal::context(renderer);
269
+ NS::AutoreleasePool* autorelease_pool = NS::AutoreleasePool::alloc()->init();
270
+
271
+ log(device, device.logger, devices::logging::Verbosity::debug, rendering::raytracing::LOG_PREFIX, "building ", scene.objects.size(), " object(s), ", scene.instances.size(), " instance(s) ...");
272
+
273
+ // scene-dependent resources from a previous init are released here; the compiled library
274
+ // and pipelines below are spec-dependent and survive re-init
275
+ ctx.mesh_buffers.clear();
276
+ ctx.mesh_textures.clear();
277
+ ctx.object_acceleration_structures.clear();
278
+ ctx.scene_lights = NS::SharedPtr<MTL::Buffer>{};
279
+
280
+ const uint8_t dummy_pixel[4] = {255, 255, 255, 255};
281
+ ctx.dummy_texture = metal::make_texture(ctx.device.get(), dummy_pixel, 1, 1, false);
282
+ const MTL::ResourceID dummy_texture_id = ctx.dummy_texture->gpuResourceID();
283
+
284
+ std::vector<const rendering::raytracing::Object*> all_objects;
285
+ for(const auto& object : scene.objects){
286
+ all_objects.push_back(&object);
287
+ }
288
+ if constexpr (SPEC::ENABLE_OVERLAYS){
289
+ rendering::raytracing::detail::register_pool_assets(device, renderer, pool, all_objects);
290
+ }
291
+
292
+ size_t total_meshes = 0;
293
+ for(const auto* object : all_objects){
294
+ total_meshes += object->meshes.size();
295
+ }
296
+ std::vector<metal::MeshRecord> mesh_records(total_meshes);
297
+ std::vector<uint32_t> object_record_base;
298
+ std::vector<std::vector<NS::Object*>> object_geometry_descriptors(all_objects.size());
299
+
300
+ size_t m = 0;
301
+ for(size_t object_i = 0; object_i < all_objects.size(); object_i++){
302
+ object_record_base.push_back((uint32_t)m);
303
+ for(const auto& md : all_objects[object_i]->meshes){
304
+ auto& geometry_descriptors = object_geometry_descriptors[object_i];
305
+ metal::MeshRecord& record = mesh_records[m];
306
+ record = {};
307
+ record.texture = dummy_texture_id;
308
+ record.normal_map = dummy_texture_id;
309
+ record.metallic_roughness_map = dummy_texture_id;
310
+ record.emissive_map = dummy_texture_id;
311
+ record.occlusion_map = dummy_texture_id;
312
+
313
+ size_t num_triangles = md.indices.size() / 3;
314
+
315
+ auto vertex_buffer = NS::TransferPtr(ctx.device->newBuffer(md.vertices.data(), md.vertices.size() * sizeof(float), MTL::ResourceStorageModeShared));
316
+ auto index_buffer = NS::TransferPtr(ctx.device->newBuffer(md.indices.data(), md.indices.size() * sizeof(int), MTL::ResourceStorageModeShared));
317
+ record.vertices = vertex_buffer->gpuAddress();
318
+ record.index = index_buffer->gpuAddress();
319
+
320
+ MTL::AccelerationStructureTriangleGeometryDescriptor* geometry = MTL::AccelerationStructureTriangleGeometryDescriptor::descriptor();
321
+ geometry->setVertexBuffer(vertex_buffer.get());
322
+ geometry->setVertexStride(3 * sizeof(float));
323
+ geometry->setVertexFormat(MTL::AttributeFormatFloat3);
324
+ geometry->setIndexBuffer(index_buffer.get());
325
+ geometry->setIndexType(MTL::IndexTypeUInt32);
326
+ geometry->setTriangleCount(num_triangles);
327
+ geometry->setOpaque(true);
328
+ geometry_descriptors.push_back(geometry);
329
+
330
+ ctx.mesh_buffers.push_back(vertex_buffer);
331
+ ctx.mesh_buffers.push_back(index_buffer);
332
+
333
+ if(!md.tex_coords.empty()){
334
+ auto tex_coord_buffer = NS::TransferPtr(ctx.device->newBuffer(md.tex_coords.data(), md.tex_coords.size() * sizeof(float), MTL::ResourceStorageModeShared));
335
+ record.tex_coord = tex_coord_buffer->gpuAddress();
336
+ ctx.mesh_buffers.push_back(tex_coord_buffer);
337
+ }
338
+ if(!md.normals.empty()){
339
+ auto normal_buffer = NS::TransferPtr(ctx.device->newBuffer(md.normals.data(), md.normals.size() * sizeof(float), MTL::ResourceStorageModeShared));
340
+ record.normal = normal_buffer->gpuAddress();
341
+ ctx.mesh_buffers.push_back(normal_buffer);
342
+ }
343
+
344
+ record.color[0] = md.color[0];
345
+ record.color[1] = md.color[1];
346
+ record.color[2] = md.color[2];
347
+ record.metallic = md.metallic;
348
+ record.roughness = md.roughness;
349
+ record.opacity = md.opacity;
350
+ record.emissive[0] = md.emissive[0];
351
+ record.emissive[1] = md.emissive[1];
352
+ record.emissive[2] = md.emissive[2];
353
+ record.alpha_cutoff = md.alpha_cutoff;
354
+ record.alpha_mode = md.alpha_mode;
355
+ record.unlit = md.unlit;
356
+
357
+ if(md.texture.present()){
358
+ auto texture = metal::make_texture(ctx.device.get(), md.texture.pixels->data(), md.texture.width, md.texture.height, true);
359
+ record.texture = texture->gpuResourceID();
360
+ record.has_texture = 1;
361
+ ctx.mesh_textures.push_back(texture);
362
+ }
363
+ if(md.normal_map.present()){
364
+ auto texture = metal::make_texture(ctx.device.get(), md.normal_map.pixels->data(), md.normal_map.width, md.normal_map.height, false);
365
+ record.normal_map = texture->gpuResourceID();
366
+ record.has_normal_map = 1;
367
+ ctx.mesh_textures.push_back(texture);
368
+ }
369
+ if(md.metallic_roughness_map.present()){
370
+ auto texture = metal::make_texture(ctx.device.get(), md.metallic_roughness_map.pixels->data(), md.metallic_roughness_map.width, md.metallic_roughness_map.height, false);
371
+ record.metallic_roughness_map = texture->gpuResourceID();
372
+ record.has_metallic_roughness_map = 1;
373
+ ctx.mesh_textures.push_back(texture);
374
+ }
375
+ if(md.emissive_map.present()){
376
+ auto texture = metal::make_texture(ctx.device.get(), md.emissive_map.pixels->data(), md.emissive_map.width, md.emissive_map.height, true);
377
+ record.emissive_map = texture->gpuResourceID();
378
+ record.has_emissive_map = 1;
379
+ ctx.mesh_textures.push_back(texture);
380
+ }
381
+ if(md.occlusion_map.present()){
382
+ auto texture = metal::make_texture(ctx.device.get(), md.occlusion_map.pixels->data(), md.occlusion_map.width, md.occlusion_map.height, false);
383
+ record.occlusion_map = texture->gpuResourceID();
384
+ record.has_occlusion_map = 1;
385
+ ctx.mesh_textures.push_back(texture);
386
+ }
387
+ m++;
388
+ }
389
+ }
390
+
391
+ // one primitive (bottom-level) acceleration structure per object, shared by its instances
392
+ {
393
+ MTL::CommandBuffer* command_buffer = ctx.queue->commandBuffer();
394
+ MTL::AccelerationStructureCommandEncoder* encoder = command_buffer->accelerationStructureCommandEncoder();
395
+ std::vector<NS::SharedPtr<MTL::Buffer>> scratch_buffers;
396
+ for(size_t object_i = 0; object_i < all_objects.size(); object_i++){
397
+ auto& geometry_descriptors = object_geometry_descriptors[object_i];
398
+ NS::Array* geometry_array = NS::Array::array((const NS::Object* const*)geometry_descriptors.data(), geometry_descriptors.size());
399
+ MTL::PrimitiveAccelerationStructureDescriptor* accel_descriptor = MTL::PrimitiveAccelerationStructureDescriptor::descriptor();
400
+ accel_descriptor->setGeometryDescriptors(geometry_array);
401
+ MTL::AccelerationStructureSizes sizes = ctx.device->accelerationStructureSizes(accel_descriptor);
402
+ auto object_acceleration_structure = NS::TransferPtr(ctx.device->newAccelerationStructure(sizes.accelerationStructureSize));
403
+ auto scratch_buffer = NS::TransferPtr(ctx.device->newBuffer(sizes.buildScratchBufferSize, MTL::ResourceStorageModePrivate));
404
+ encoder->buildAccelerationStructure(object_acceleration_structure.get(), accel_descriptor, scratch_buffer.get(), 0);
405
+ scratch_buffers.push_back(scratch_buffer);
406
+ ctx.object_acceleration_structures.push_back(object_acceleration_structure);
407
+ }
408
+ encoder->endEncoding();
409
+ command_buffer->commit();
410
+ command_buffer->waitUntilCompleted();
411
+ }
412
+
413
+ // instance (top-level) acceleration structure over the placed objects; overlay slots get
414
+ // their own tiny instance structures but share the global instance-id-indexed data arrays
415
+ ctx.object_record_base = object_record_base;
416
+ ctx.object_classes.clear();
417
+ for(const auto* object_pointer : all_objects){
418
+ ctx.object_classes.push_back(object_pointer->segmentation_class);
419
+ }
420
+ ctx.num_scene_instances = (uint32_t)scene.instances.size();
421
+ {
422
+ size_t total_instances = scene.instances.size();
423
+ if constexpr (SPEC::ENABLE_OVERLAYS){
424
+ // dynamic motion blur appends one extra overlay id range per motion sample:
425
+ // sample s's instances carry shifted user ids (num_scene + num_overlay_slots *
426
+ // (1 + s) + slot) so each pass's hits index a private slice of the instance side
427
+ // tables with no shader changes; the contract ids [num_scene, num_scene +
428
+ // num_overlay_slots) stay exclusive to the shutter-close state segmentation reads
429
+ // flow appends one more range after the blur samples for the previous instant's
430
+ // structures (visibility rays and forward primaries only: no side tables read)
431
+ constexpr size_t num_overlay_slots = (size_t)SPEC::NUM_OVERLAYS * SPEC::MAX_OVERLAY_INSTANCES;
432
+ constexpr size_t overlay_id_ranges = (SPEC::ENABLE_DYNAMIC_MOTION_BLUR ? (size_t)1 + SPEC::MOTION_BLUR_SAMPLES : 1) + (SPEC::HAS_FLOW_OVERLAYS ? 1 : 0);
433
+ total_instances += num_overlay_slots * overlay_id_ranges;
434
+ }
435
+ std::vector<MTL::AccelerationStructureUserIDInstanceDescriptor> instance_descriptors(scene.instances.size());
436
+ std::vector<metal::InstanceData> instance_data(total_instances, metal::InstanceData{});
437
+ std::vector<uint32_t> instance_record_base(total_instances, 0);
438
+ for(size_t instance_i = 0; instance_i < scene.instances.size(); instance_i++){
439
+ const auto& instance = scene.instances[instance_i];
440
+ auto& descriptor = instance_descriptors[instance_i];
441
+ descriptor = {};
442
+ for(int column = 0; column < 4; column++){
443
+ descriptor.transformationMatrix.columns[column] = MTL::PackedFloat3{instance.transform[0 + column], instance.transform[4 + column], instance.transform[8 + column]};
444
+ }
445
+ descriptor.options = MTL::AccelerationStructureInstanceOptionOpaque;
446
+ descriptor.mask = 0xFFFFFFFFu;
447
+ descriptor.intersectionFunctionTableOffset = 0;
448
+ descriptor.accelerationStructureIndex = (uint32_t)instance.object;
449
+ descriptor.userID = (uint32_t)instance_i; // global instance id (contract: segmentation_object, operations_cpu_common.h)
450
+
451
+ auto& data = instance_data[instance_i];
452
+ data = {};
453
+ for(int element = 0; element < 12; element++){
454
+ data.object_to_world[element] = instance.transform[element];
455
+ }
456
+ if(instance.identity){
457
+ std::memcpy(data.world_to_object, data.object_to_world, sizeof(data.world_to_object));
458
+ }
459
+ else{
460
+ rendering::raytracing::detail::invert_transform(instance.transform, data.world_to_object);
461
+ }
462
+ data.identity = instance.identity ? 1 : 0;
463
+ instance_record_base[instance_i] = object_record_base[instance.object];
464
+ }
465
+ ctx.instance_descriptors = NS::TransferPtr(ctx.device->newBuffer(instance_descriptors.data(), instance_descriptors.size() * sizeof(MTL::AccelerationStructureUserIDInstanceDescriptor), MTL::ResourceStorageModeShared));
466
+ ctx.instance_data = NS::TransferPtr(ctx.device->newBuffer(instance_data.data(), instance_data.size() * sizeof(metal::InstanceData), MTL::ResourceStorageModeShared));
467
+ ctx.instance_record_base = NS::TransferPtr(ctx.device->newBuffer(instance_record_base.data(), instance_record_base.size() * sizeof(uint32_t), MTL::ResourceStorageModeShared));
468
+ std::vector<uint32_t> instance_classes(total_instances > 0 ? total_instances : 1, 0);
469
+ for(size_t instance_i = 0; instance_i < scene.instances.size(); instance_i++){
470
+ instance_classes[instance_i] = ctx.object_classes[scene.instances[instance_i].object];
471
+ }
472
+ ctx.instance_classes = NS::TransferPtr(ctx.device->newBuffer(instance_classes.data(), instance_classes.size() * sizeof(uint32_t), MTL::ResourceStorageModeShared));
473
+
474
+ std::vector<NS::Object*> object_acceleration_structure_pointers;
475
+ for(auto& object_acceleration_structure : ctx.object_acceleration_structures){
476
+ object_acceleration_structure_pointers.push_back(object_acceleration_structure.get());
477
+ }
478
+ NS::Array* object_array = NS::Array::array((const NS::Object* const*)object_acceleration_structure_pointers.data(), object_acceleration_structure_pointers.size());
479
+ MTL::InstanceAccelerationStructureDescriptor* instance_accel_descriptor = MTL::InstanceAccelerationStructureDescriptor::descriptor();
480
+ instance_accel_descriptor->setInstancedAccelerationStructures(object_array);
481
+ instance_accel_descriptor->setInstanceCount(scene.instances.size());
482
+ instance_accel_descriptor->setInstanceDescriptorBuffer(ctx.instance_descriptors.get());
483
+ instance_accel_descriptor->setInstanceDescriptorType(MTL::AccelerationStructureInstanceDescriptorTypeUserID);
484
+ MTL::AccelerationStructureSizes sizes = ctx.device->accelerationStructureSizes(instance_accel_descriptor);
485
+ ctx.acceleration_structure = NS::TransferPtr(ctx.device->newAccelerationStructure(sizes.accelerationStructureSize));
486
+ auto scratch_buffer = NS::TransferPtr(ctx.device->newBuffer(sizes.buildScratchBufferSize, MTL::ResourceStorageModePrivate));
487
+ MTL::CommandBuffer* command_buffer = ctx.queue->commandBuffer();
488
+ MTL::AccelerationStructureCommandEncoder* encoder = command_buffer->accelerationStructureCommandEncoder();
489
+ encoder->buildAccelerationStructure(ctx.acceleration_structure.get(), instance_accel_descriptor, scratch_buffer.get(), 0);
490
+ encoder->endEncoding();
491
+ command_buffer->commit();
492
+ command_buffer->waitUntilCompleted();
493
+ }
494
+ ctx.overlay_acceleration_structures.clear();
495
+ ctx.overlay_instance_descriptors.clear();
496
+ ctx.overlay_sample_instance_descriptors.clear();
497
+ ctx.overlay_scratch_buffers.clear();
498
+ ctx.overlay_acceleration_structures_previous.clear();
499
+ ctx.overlay_instance_descriptors_previous.clear();
500
+ ctx.overlay_scratch_buffers_previous.clear();
501
+ ctx.overlay_structures_previous = NS::SharedPtr<MTL::Buffer>{};
502
+ if constexpr (SPEC::ENABLE_OVERLAYS){
503
+ std::vector<NS::Object*> object_acceleration_structure_pointers;
504
+ for(auto& object_acceleration_structure : ctx.object_acceleration_structures){
505
+ object_acceleration_structure_pointers.push_back(object_acceleration_structure.get());
506
+ }
507
+ NS::Array* object_array = NS::Array::array((const NS::Object* const*)object_acceleration_structure_pointers.data(), object_acceleration_structure_pointers.size());
508
+ // one overlay structure set per instant: the current one, plus the previous one under flow
509
+ const auto create_overlay_structure_set = [&](std::vector<NS::SharedPtr<MTL::Buffer>>& instance_descriptors, std::vector<NS::SharedPtr<MTL::AccelerationStructure>>& acceleration_structures, std::vector<NS::SharedPtr<MTL::Buffer>>& scratch_buffers) -> NS::SharedPtr<MTL::Buffer> {
510
+ std::vector<metal::OverlayStructureEntry> overlay_entries(SPEC::NUM_OVERLAYS, metal::OverlayStructureEntry{});
511
+ for(size_t overlay = 0; overlay < SPEC::NUM_OVERLAYS; overlay++){
512
+ auto descriptor_buffer = NS::TransferPtr(ctx.device->newBuffer((size_t)SPEC::MAX_OVERLAY_INSTANCES * sizeof(MTL::AccelerationStructureUserIDInstanceDescriptor), MTL::ResourceStorageModeShared));
513
+ MTL::InstanceAccelerationStructureDescriptor* overlay_descriptor = MTL::InstanceAccelerationStructureDescriptor::descriptor();
514
+ overlay_descriptor->setInstancedAccelerationStructures(object_array);
515
+ overlay_descriptor->setInstanceCount(SPEC::MAX_OVERLAY_INSTANCES);
516
+ overlay_descriptor->setInstanceDescriptorBuffer(descriptor_buffer.get());
517
+ overlay_descriptor->setInstanceDescriptorType(MTL::AccelerationStructureInstanceDescriptorTypeUserID);
518
+ MTL::AccelerationStructureSizes overlay_sizes = ctx.device->accelerationStructureSizes(overlay_descriptor);
519
+ auto overlay_acceleration_structure = NS::TransferPtr(ctx.device->newAccelerationStructure(overlay_sizes.accelerationStructureSize));
520
+ auto overlay_scratch = NS::TransferPtr(ctx.device->newBuffer(overlay_sizes.buildScratchBufferSize, MTL::ResourceStorageModePrivate));
521
+ instance_descriptors.push_back(descriptor_buffer);
522
+ acceleration_structures.push_back(overlay_acceleration_structure);
523
+ scratch_buffers.push_back(overlay_scratch);
524
+ overlay_entries[overlay].structure = overlay_acceleration_structure->gpuResourceID();
525
+ overlay_entries[overlay].num_active = 0;
526
+ }
527
+ return NS::TransferPtr(ctx.device->newBuffer(overlay_entries.data(), overlay_entries.size() * sizeof(metal::OverlayStructureEntry), MTL::ResourceStorageModeShared));
528
+ };
529
+ ctx.overlay_structures = create_overlay_structure_set(ctx.overlay_instance_descriptors, ctx.overlay_acceleration_structures, ctx.overlay_scratch_buffers);
530
+ if constexpr (SPEC::HAS_FLOW_OVERLAYS){
531
+ ctx.overlay_structures_previous = create_overlay_structure_set(ctx.overlay_instance_descriptors_previous, ctx.overlay_acceleration_structures_previous, ctx.overlay_scratch_buffers_previous);
532
+ }
533
+ if constexpr (SPEC::ENABLE_DYNAMIC_MOTION_BLUR){
534
+ for(size_t sample = 0; sample < SPEC::MOTION_BLUR_SAMPLES; sample++){
535
+ for(size_t overlay = 0; overlay < SPEC::NUM_OVERLAYS; overlay++){
536
+ ctx.overlay_sample_instance_descriptors.push_back(NS::TransferPtr(ctx.device->newBuffer((size_t)SPEC::MAX_OVERLAY_INSTANCES * sizeof(MTL::AccelerationStructureUserIDInstanceDescriptor), MTL::ResourceStorageModeShared)));
537
+ }
538
+ }
539
+ }
540
+ std::vector<uint32_t> attachments_init((size_t)SPEC::NUM_CAMERAS * SPEC::MAX_OVERLAYS_PER_CAMERA, 0xFFFFFFFFu);
541
+ ctx.overlay_attachments = NS::TransferPtr(ctx.device->newBuffer(attachments_init.data(), attachments_init.size() * sizeof(uint32_t), MTL::ResourceStorageModeShared));
542
+ rendering::raytracing::detail::reset_overlay_state(renderer);
543
+ }
544
+ else if(ctx.overlay_structures.get() == nullptr){
545
+ // never dereferenced (fc_overlay_count == 0) but keeps the kernel bindings valid
546
+ ctx.overlay_structures = NS::TransferPtr(ctx.device->newBuffer(sizeof(metal::OverlayStructureEntry), MTL::ResourceStorageModeShared));
547
+ ctx.overlay_attachments = NS::TransferPtr(ctx.device->newBuffer(sizeof(uint32_t), MTL::ResourceStorageModeShared));
548
+ }
549
+
550
+ ctx.mesh_records = NS::TransferPtr(ctx.device->newBuffer(mesh_records.data(), mesh_records.size() * sizeof(metal::MeshRecord), MTL::ResourceStorageModeShared));
551
+
552
+ const auto scene_lights = rendering::raytracing::detail::effective_scene_lights<SPEC::HAS_RGB && SPEC::SHADING::PBR_SHADING>(device, scene);
553
+ if(!scene_lights.empty()){
554
+ ctx.scene_lights = NS::TransferPtr(ctx.device->newBuffer(scene_lights.data(), scene_lights.size() * sizeof(rendering::raytracing::SceneLight), MTL::ResourceStorageModeShared));
555
+ }
556
+ else{
557
+ ctx.scene_lights = NS::TransferPtr(ctx.device->newBuffer(sizeof(rendering::raytracing::SceneLight), MTL::ResourceStorageModeShared));
558
+ }
559
+
560
+ auto* params = (metal::LaunchParams*)ctx.launch_params->contents();
561
+ *params = {};
562
+ params->fb_width = SPEC::FB_WIDTH;
563
+ params->fb_height = SPEC::FB_HEIGHT;
564
+ params->cam_width = SPEC::CAM_WIDTH;
565
+ params->cam_height = SPEC::CAM_HEIGHT;
566
+ params->grid_cols = SPEC::GRID_COLS;
567
+ params->num_cameras = SPEC::NUM_CAMERAS;
568
+ params->num_probes = SPEC::NUM_PROBES;
569
+ params->num_scene_lights = (uint32_t)scene_lights.size();
570
+ params->max_dist = renderer.max_ray_length;
571
+ for(int component = 0; component < 3; component++){
572
+ params->ambient_color[component] = (float)renderer.ambient_color[component];
573
+ params->miss_color_0[component] = (float)renderer.miss_color_0[component];
574
+ params->miss_color_1[component] = (float)renderer.miss_color_1[component];
575
+ }
576
+ params->first_overlay_instance = (uint32_t)scene.instances.size();
577
+ params->num_overlay_slots = SPEC::ENABLE_OVERLAYS ? (uint32_t)SPEC::NUM_OVERLAYS * SPEC::MAX_OVERLAY_INSTANCES : 0u;
578
+
579
+ if(!ctx.pipelines_built){
580
+ auto constants = NS::TransferPtr(MTL::FunctionConstantValues::alloc()->init());
581
+ using CONSTANTS = rendering::raytracing::backends::SpecializationConstants<SPEC>;
582
+ bool srgb_output = CONSTANTS::SRGB_OUTPUT;
583
+ bool motion_blur = CONSTANTS::MOTION_BLUR;
584
+ int motion_samples = CONSTANTS::MOTION_SAMPLES;
585
+ int aa_grid = CONSTANTS::AA_GRID;
586
+ bool checker_background = CONSTANTS::CHECKER_BACKGROUND;
587
+ bool load_textures = CONSTANTS::LOAD_TEXTURES;
588
+ bool normal_shading = CONSTANTS::NORMAL_SHADING;
589
+ bool metallic_reflections = CONSTANTS::METALLIC_REFLECTIONS;
590
+ bool pbr_shading = CONSTANTS::PBR_SHADING;
591
+ bool punctual_light_shadows = CONSTANTS::PUNCTUAL_LIGHT_SHADOWS;
592
+ int overlay_count = CONSTANTS::OVERLAY_COUNT;
593
+ bool semantic_segmentation = CONSTANTS::SEMANTIC_SEGMENTATION;
594
+ bool has_observation = CONSTANTS::HAS_OBSERVATION;
595
+ bool dynamic_motion_blur = CONSTANTS::DYNAMIC_MOTION_BLUR;
596
+ bool resolve_rgb = CONSTANTS::RESOLVE_RGB;
597
+ constants->setConstantValue(&srgb_output, MTL::DataTypeBool, (NS::UInteger)metal::function_constants::SRGB_OUTPUT);
598
+ constants->setConstantValue(&motion_blur, MTL::DataTypeBool, (NS::UInteger)metal::function_constants::MOTION_BLUR);
599
+ constants->setConstantValue(&motion_samples, MTL::DataTypeInt, (NS::UInteger)metal::function_constants::MOTION_SAMPLES);
600
+ constants->setConstantValue(&aa_grid, MTL::DataTypeInt, (NS::UInteger)metal::function_constants::AA_GRID);
601
+ constants->setConstantValue(&checker_background, MTL::DataTypeBool, (NS::UInteger)metal::function_constants::CHECKER_BACKGROUND);
602
+ constants->setConstantValue(&load_textures, MTL::DataTypeBool, (NS::UInteger)metal::function_constants::LOAD_TEXTURES);
603
+ constants->setConstantValue(&normal_shading, MTL::DataTypeBool, (NS::UInteger)metal::function_constants::NORMAL_SHADING);
604
+ constants->setConstantValue(&metallic_reflections, MTL::DataTypeBool, (NS::UInteger)metal::function_constants::METALLIC_REFLECTIONS);
605
+ constants->setConstantValue(&pbr_shading, MTL::DataTypeBool, (NS::UInteger)metal::function_constants::PBR_SHADING);
606
+ constants->setConstantValue(&punctual_light_shadows, MTL::DataTypeBool, (NS::UInteger)metal::function_constants::PUNCTUAL_LIGHT_SHADOWS);
607
+ constants->setConstantValue(&overlay_count, MTL::DataTypeInt, (NS::UInteger)metal::function_constants::OVERLAY_COUNT);
608
+ constants->setConstantValue(&semantic_segmentation, MTL::DataTypeBool, (NS::UInteger)metal::function_constants::SEMANTIC_SEGMENTATION);
609
+ constants->setConstantValue(&has_observation, MTL::DataTypeBool, (NS::UInteger)metal::function_constants::HAS_OBSERVATION);
610
+ constants->setConstantValue(&dynamic_motion_blur, MTL::DataTypeBool, (NS::UInteger)metal::function_constants::DYNAMIC_MOTION_BLUR);
611
+ constants->setConstantValue(&resolve_rgb, MTL::DataTypeBool, (NS::UInteger)metal::function_constants::RESOLVE_RGB);
612
+
613
+ auto make_pipeline = [&](const char* name) -> NS::SharedPtr<MTL::ComputePipelineState> {
614
+ NS::Error* error = nullptr;
615
+ auto function = NS::TransferPtr(ctx.library->newFunction(NS::String::string(name, NS::UTF8StringEncoding), constants.get(), &error));
616
+ if(function.get() == nullptr){
617
+ log(device, device.logger, devices::logging::Verbosity::error, rendering::raytracing::LOG_PREFIX, "Metal function specialization failed for ", name, ": ", (error != nullptr ? error->localizedDescription()->utf8String() : "unknown error"));
618
+ utils::assert_exit(device, false, "Metal function specialization failed");
619
+ }
620
+ error = nullptr;
621
+ auto pipeline = NS::TransferPtr(ctx.device->newComputePipelineState(function.get(), &error));
622
+ if(pipeline.get() == nullptr){
623
+ log(device, device.logger, devices::logging::Verbosity::error, rendering::raytracing::LOG_PREFIX, "Metal pipeline creation failed for ", name, ": ", (error != nullptr ? error->localizedDescription()->utf8String() : "unknown error"));
624
+ utils::assert_exit(device, false, "Metal pipeline creation failed");
625
+ }
626
+ return pipeline;
627
+ };
628
+
629
+ if constexpr (SPEC::HAS_RGB) {
630
+ ctx.rgb_pipeline = make_pipeline("render_rgb");
631
+ }
632
+ if constexpr (SPEC::HAS_DEPTH) {
633
+ ctx.depth_pipeline = make_pipeline("render_depth");
634
+ }
635
+ if constexpr (SPEC::HAS_SEGMENTATION) {
636
+ ctx.segmentation_pipeline = make_pipeline("render_segmentation");
637
+ }
638
+ if constexpr (SPEC::HAS_NORMALS) {
639
+ ctx.normals_pipeline = make_pipeline("render_normals");
640
+ }
641
+ if constexpr (SPEC::HAS_FLOW) {
642
+ ctx.flow_pipeline = make_pipeline("render_flow");
643
+ }
644
+ if constexpr (SPEC::HAS_FLOW_FORWARD) {
645
+ ctx.flow_forward_pipeline = make_pipeline("render_flow_forward");
646
+ }
647
+ if constexpr (SPEC::ENABLE_DYNAMIC_MOTION_BLUR) {
648
+ ctx.resolve_pipeline = make_pipeline("resolve_outputs");
649
+ }
650
+ #if !RL_TOOLS_RENDERING_RAYTRACING_DISABLE_PROBE_RAYS
651
+ ctx.collision_pipeline = make_pipeline("render_collision");
652
+ #endif
653
+ ctx.pipelines_built = true;
654
+ }
655
+
656
+ if constexpr (SPEC::ENABLE_OVERLAYS){
657
+ update(device, renderer); // publish the (empty) overlays and the attachment table
658
+ }
659
+
660
+ autorelease_pool->release();
661
+ }
662
+
663
+
664
+ // the host must not mutate the shared instance/descriptor buffers while renders or builds
665
+ // are in flight, hence the wait at the top; the commit is not waited on — command buffers on
666
+ // one queue execute in commit order and Metal's hazard tracking orders the acceleration
667
+ // structure writes before any subsequent render pass that reads them
668
+ template <typename DEVICE, typename SPEC>
669
+ void update_launch(DEVICE& device, rendering::raytracing::Renderer<SPEC, rendering::raytracing::backends::Metal>& renderer){
670
+ static_assert(SPEC::ENABLE_OVERLAYS, "update requires an overlay-enabled renderer specification");
671
+ namespace metal = rendering::raytracing::backends::metal;
672
+ using TI = typename SPEC::TI;
673
+ auto& ctx = metal::context(renderer);
674
+ metal::wait_in_flight(ctx);
675
+ NS::AutoreleasePool* autorelease_pool = NS::AutoreleasePool::alloc()->init();
676
+
677
+ auto* overlay_entries = (metal::OverlayStructureEntry*)ctx.overlay_structures->contents();
678
+ auto* instance_data = (metal::InstanceData*)ctx.instance_data->contents();
679
+ auto* instance_record_base = (uint32_t*)ctx.instance_record_base->contents();
680
+
681
+ MTL::CommandBuffer* command_buffer = nullptr;
682
+ MTL::AccelerationStructureCommandEncoder* encoder = nullptr;
683
+ std::vector<NS::Object*> object_acceleration_structure_pointers;
684
+ for(auto& object_acceleration_structure : ctx.object_acceleration_structures){
685
+ object_acceleration_structure_pointers.push_back(object_acceleration_structure.get());
686
+ }
687
+ NS::Array* object_array = NS::Array::array((const NS::Object* const*)object_acceleration_structure_pointers.data(), object_acceleration_structure_pointers.size());
688
+
689
+ rendering::raytracing::detail::publish_overlay_transforms(renderer);
690
+ // rebuilt unconditionally: producers may write the transforms tensors directly, which
691
+ // leaves no host-observable pending row
692
+ for(TI overlay = 0; overlay < SPEC::NUM_OVERLAYS; overlay++){
693
+ auto& overlay_state = renderer.overlays[overlay];
694
+ const size_t base = (size_t)ctx.num_scene_instances + (size_t)overlay * SPEC::MAX_OVERLAY_INSTANCES;
695
+ auto* descriptors = (MTL::AccelerationStructureUserIDInstanceDescriptor*)ctx.overlay_instance_descriptors[overlay]->contents();
696
+ uint32_t num_active = 0;
697
+ for(TI slot = 0; slot < SPEC::MAX_OVERLAY_INSTANCES; slot++){
698
+ const auto& host_slot = overlay_state.slots[slot];
699
+ if(!host_slot.active) continue;
700
+ const size_t global = base + slot;
701
+ const float* world = data(renderer.transforms) + ((size_t)overlay * SPEC::MAX_OVERLAY_INSTANCES + slot) * 12;
702
+ auto& descriptor = descriptors[num_active];
703
+ descriptor = {};
704
+ for(int column = 0; column < 4; column++){
705
+ descriptor.transformationMatrix.columns[column] = MTL::PackedFloat3{world[0 + column], world[4 + column], world[8 + column]};
706
+ }
707
+ descriptor.options = MTL::AccelerationStructureInstanceOptionOpaque;
708
+ descriptor.mask = 0xFFFFFFFFu;
709
+ descriptor.intersectionFunctionTableOffset = 0;
710
+ descriptor.accelerationStructureIndex = (uint32_t)host_slot.object;
711
+ descriptor.userID = (uint32_t)global; // global instance id (contract: segmentation_object, operations_cpu_common.h)
712
+
713
+ auto& data = instance_data[global];
714
+ data = {};
715
+ for(int element = 0; element < 12; element++){
716
+ data.object_to_world[element] = world[element];
717
+ }
718
+ const bool identity = rendering::raytracing::detail::transform_is_identity(world);
719
+ if(identity){
720
+ std::memcpy(data.world_to_object, data.object_to_world, sizeof(data.world_to_object));
721
+ }
722
+ else{
723
+ rendering::raytracing::detail::invert_transform(world, data.world_to_object);
724
+ }
725
+ data.identity = identity ? 1 : 0;
726
+ instance_record_base[global] = ctx.object_record_base[host_slot.object];
727
+ ((uint32_t*)ctx.instance_classes->contents())[global] = ctx.object_classes[host_slot.object];
728
+ num_active++;
729
+ }
730
+ overlay_entries[overlay].num_active = num_active;
731
+ if(num_active > 0){
732
+ if(command_buffer == nullptr){
733
+ command_buffer = ctx.queue->commandBuffer();
734
+ encoder = command_buffer->accelerationStructureCommandEncoder();
735
+ }
736
+ MTL::InstanceAccelerationStructureDescriptor* overlay_descriptor = MTL::InstanceAccelerationStructureDescriptor::descriptor();
737
+ overlay_descriptor->setInstancedAccelerationStructures(object_array);
738
+ overlay_descriptor->setInstanceCount(num_active);
739
+ overlay_descriptor->setInstanceDescriptorBuffer(ctx.overlay_instance_descriptors[overlay].get());
740
+ overlay_descriptor->setInstanceDescriptorType(MTL::AccelerationStructureInstanceDescriptorTypeUserID);
741
+ encoder->buildAccelerationStructure(ctx.overlay_acceleration_structures[overlay].get(), overlay_descriptor, ctx.overlay_scratch_buffers[overlay].get(), 0);
742
+ }
743
+ }
744
+ if constexpr (SPEC::HAS_FLOW_OVERLAYS){
745
+ // the previous instant: descriptors from transforms_previous over the shifted id range
746
+ // after the blur samples (no side tables: the flow passes only need hit distances and
747
+ // ids), built into the resident second structure set; plus the per-slot delta table
748
+ constexpr size_t num_overlay_slots = (size_t)SPEC::NUM_OVERLAYS * SPEC::MAX_OVERLAY_INSTANCES;
749
+ constexpr size_t previous_range = 1 + (SPEC::ENABLE_DYNAMIC_MOTION_BLUR ? (size_t)SPEC::MOTION_BLUR_SAMPLES : 0);
750
+ auto* overlay_entries_previous = (metal::OverlayStructureEntry*)ctx.overlay_structures_previous->contents();
751
+ for(TI overlay = 0; overlay < SPEC::NUM_OVERLAYS; overlay++){
752
+ auto& overlay_state = renderer.overlays[overlay];
753
+ const size_t base = (size_t)ctx.num_scene_instances + num_overlay_slots * previous_range + (size_t)overlay * SPEC::MAX_OVERLAY_INSTANCES;
754
+ auto* descriptors = (MTL::AccelerationStructureUserIDInstanceDescriptor*)ctx.overlay_instance_descriptors_previous[overlay]->contents();
755
+ uint32_t num_active = 0;
756
+ for(TI slot = 0; slot < SPEC::MAX_OVERLAY_INSTANCES; slot++){
757
+ const auto& host_slot = overlay_state.slots[slot];
758
+ if(!host_slot.active) continue;
759
+ const float* world = data(renderer.transforms_previous) + ((size_t)overlay * SPEC::MAX_OVERLAY_INSTANCES + slot) * 12;
760
+ auto& descriptor = descriptors[num_active];
761
+ descriptor = {};
762
+ for(int column = 0; column < 4; column++){
763
+ descriptor.transformationMatrix.columns[column] = MTL::PackedFloat3{world[0 + column], world[4 + column], world[8 + column]};
764
+ }
765
+ descriptor.options = MTL::AccelerationStructureInstanceOptionOpaque;
766
+ descriptor.mask = 0xFFFFFFFFu;
767
+ descriptor.intersectionFunctionTableOffset = 0;
768
+ descriptor.accelerationStructureIndex = (uint32_t)host_slot.object;
769
+ descriptor.userID = (uint32_t)(base + slot);
770
+ num_active++;
771
+ }
772
+ overlay_entries_previous[overlay].num_active = num_active;
773
+ if(num_active > 0){
774
+ if(command_buffer == nullptr){
775
+ command_buffer = ctx.queue->commandBuffer();
776
+ encoder = command_buffer->accelerationStructureCommandEncoder();
777
+ }
778
+ MTL::InstanceAccelerationStructureDescriptor* overlay_descriptor = MTL::InstanceAccelerationStructureDescriptor::descriptor();
779
+ overlay_descriptor->setInstancedAccelerationStructures(object_array);
780
+ overlay_descriptor->setInstanceCount(num_active);
781
+ overlay_descriptor->setInstanceDescriptorBuffer(ctx.overlay_instance_descriptors_previous[overlay].get());
782
+ overlay_descriptor->setInstanceDescriptorType(MTL::AccelerationStructureInstanceDescriptorTypeUserID);
783
+ encoder->buildAccelerationStructure(ctx.overlay_acceleration_structures_previous[overlay].get(), overlay_descriptor, ctx.overlay_scratch_buffers_previous[overlay].get(), 0);
784
+ }
785
+ }
786
+ rendering::raytracing::detail::compose_flow_deltas(renderer, data(renderer.transforms), data(renderer.transforms_previous), (float*)ctx.flow_deltas->contents());
787
+ }
788
+ if constexpr (SPEC::ENABLE_DYNAMIC_MOTION_BLUR){
789
+ // per-sample overlay instance descriptors + shifted-id side-table slices consumed by
790
+ // render_launch's per-sample builds; layout mirrors the shutter-close path above
791
+ constexpr size_t SLAB = (size_t)SPEC::NUM_OVERLAYS * SPEC::MAX_OVERLAY_INSTANCES * 12;
792
+ constexpr size_t num_overlay_slots = (size_t)SPEC::NUM_OVERLAYS * SPEC::MAX_OVERLAY_INSTANCES;
793
+ for(TI sample = 0; sample < SPEC::MOTION_BLUR_SAMPLES; sample++){
794
+ const float* transforms_base = data(renderer.transforms_motion) + sample * SLAB;
795
+ for(TI overlay = 0; overlay < SPEC::NUM_OVERLAYS; overlay++){
796
+ auto& overlay_state = renderer.overlays[overlay];
797
+ auto* descriptors = (MTL::AccelerationStructureUserIDInstanceDescriptor*)ctx.overlay_sample_instance_descriptors[(size_t)sample * SPEC::NUM_OVERLAYS + overlay]->contents();
798
+ uint32_t num_active = 0;
799
+ for(TI slot = 0; slot < SPEC::MAX_OVERLAY_INSTANCES; slot++){
800
+ const auto& host_slot = overlay_state.slots[slot];
801
+ if(!host_slot.active) continue;
802
+ const size_t global = (size_t)ctx.num_scene_instances + num_overlay_slots * (1 + (size_t)sample) + (size_t)overlay * SPEC::MAX_OVERLAY_INSTANCES + slot;
803
+ const float* world = transforms_base + ((size_t)overlay * SPEC::MAX_OVERLAY_INSTANCES + slot) * 12;
804
+ auto& descriptor = descriptors[num_active];
805
+ descriptor = {};
806
+ for(int column = 0; column < 4; column++){
807
+ descriptor.transformationMatrix.columns[column] = MTL::PackedFloat3{world[0 + column], world[4 + column], world[8 + column]};
808
+ }
809
+ descriptor.options = MTL::AccelerationStructureInstanceOptionOpaque;
810
+ descriptor.mask = 0xFFFFFFFFu;
811
+ descriptor.intersectionFunctionTableOffset = 0;
812
+ descriptor.accelerationStructureIndex = (uint32_t)host_slot.object;
813
+ descriptor.userID = (uint32_t)global;
814
+
815
+ auto& data_entry = instance_data[global];
816
+ data_entry = {};
817
+ for(int element = 0; element < 12; element++){
818
+ data_entry.object_to_world[element] = world[element];
819
+ }
820
+ const bool identity = rendering::raytracing::detail::transform_is_identity(world);
821
+ if(identity){
822
+ std::memcpy(data_entry.world_to_object, data_entry.object_to_world, sizeof(data_entry.world_to_object));
823
+ }
824
+ else{
825
+ rendering::raytracing::detail::invert_transform(world, data_entry.world_to_object);
826
+ }
827
+ data_entry.identity = identity ? 1 : 0;
828
+ instance_record_base[global] = ctx.object_record_base[host_slot.object];
829
+ ((uint32_t*)ctx.instance_classes->contents())[global] = ctx.object_classes[host_slot.object];
830
+ num_active++;
831
+ }
832
+ }
833
+ }
834
+ }
835
+ if(renderer.attachments_dirty){
836
+ auto* attachments = (uint32_t*)ctx.overlay_attachments->contents();
837
+ for(size_t index = 0; index < (size_t)SPEC::NUM_CAMERAS * SPEC::MAX_OVERLAYS_PER_CAMERA; index++){
838
+ attachments[index] = (uint32_t)renderer.attachments[index];
839
+ }
840
+ renderer.attachments_dirty = false;
841
+ }
842
+ if(command_buffer != nullptr){
843
+ encoder->endEncoding();
844
+ command_buffer->commit();
845
+ ctx.in_flight_update = NS::RetainPtr(command_buffer);
846
+ }
847
+ autorelease_pool->release();
848
+ }
849
+
850
+ template <typename DEVICE, typename SPEC>
851
+ void update_sync(DEVICE& device, rendering::raytracing::Renderer<SPEC, rendering::raytracing::backends::Metal>& renderer){
852
+ static_assert(SPEC::ENABLE_OVERLAYS, "update requires an overlay-enabled renderer specification");
853
+ namespace metal = rendering::raytracing::backends::metal;
854
+ auto& ctx = metal::context(renderer);
855
+ if(ctx.in_flight_update.get() != nullptr){
856
+ ctx.in_flight_update->waitUntilCompleted();
857
+ ctx.in_flight_update.reset();
858
+ }
859
+ }
860
+
861
+ // the previous instant becomes the current one: shared-storage buffers and host tensors, so
862
+ // plain copies after the in-flight wait; rows the host has not yet published reach
863
+ // transforms_previous through their staging mirror at the next update
864
+ template <typename DEVICE, typename SPEC>
865
+ void advance_flow_launch(DEVICE& device, rendering::raytracing::Renderer<SPEC, rendering::raytracing::backends::Metal>& renderer){
866
+ static_assert(SPEC::HAS_FLOW, "advance_flow requires a flow-capable renderer specification");
867
+ namespace metal = rendering::raytracing::backends::metal;
868
+ metal::wait_in_flight(metal::context(renderer));
869
+ std::memcpy(data(renderer.cameras_previous), data(renderer.cameras), decltype(renderer.cameras)::SPEC::SIZE_BYTES);
870
+ if constexpr (SPEC::HAS_FLOW_OVERLAYS){
871
+ rendering::raytracing::detail::advance_flow_staging(renderer);
872
+ std::memcpy(data(renderer.transforms_previous), data(renderer.transforms), decltype(renderer.transforms)::SPEC::SIZE_BYTES);
873
+ }
874
+ }
875
+
876
+ template <typename DEVICE, typename SPEC>
877
+ void advance_flow_sync(DEVICE& device, rendering::raytracing::Renderer<SPEC, rendering::raytracing::backends::Metal>& renderer){
878
+ static_assert(SPEC::HAS_FLOW, "advance_flow requires a flow-capable renderer specification");
879
+ }
880
+
881
+ template <typename DEVICE, typename SPEC>
882
+ void advance_flow(DEVICE& device, rendering::raytracing::Renderer<SPEC, rendering::raytracing::backends::Metal>& renderer){
883
+ advance_flow_launch(device, renderer);
884
+ advance_flow_sync(device, renderer);
885
+ }
886
+
887
+
888
+ // renderer memory-domain copies: shared-storage buffers are host-addressable after the
889
+ // in-flight wait, so the transfer delegates to the host-device tensor copy
890
+ template <typename TO_DEVICE, typename FROM_SPEC, typename TO_SPEC>
891
+ void copy(rendering::raytracing::backends::Device<rendering::raytracing::backends::Metal>& from_device, TO_DEVICE& to_device, const Tensor<FROM_SPEC>& from, Tensor<TO_SPEC>& to){
892
+ rendering::raytracing::backends::metal::wait_in_flight(*from_device.context);
893
+ copy(to_device, to_device, from, to);
894
+ }
895
+ template <typename FROM_DEVICE, typename FROM_SPEC, typename TO_SPEC>
896
+ void copy(FROM_DEVICE& from_device, rendering::raytracing::backends::Device<rendering::raytracing::backends::Metal>& to_device, const Tensor<FROM_SPEC>& from, Tensor<TO_SPEC>& to){
897
+ rendering::raytracing::backends::metal::wait_in_flight(*to_device.context);
898
+ copy(from_device, from_device, from, to);
899
+ }
900
+
901
+ template <typename DEVICE, typename SPEC>
902
+ void generate_probe_directions(DEVICE& device, rendering::raytracing::Renderer<SPEC, rendering::raytracing::backends::Metal>& renderer){
903
+ #if RL_TOOLS_RENDERING_RAYTRACING_DISABLE_PROBE_RAYS
904
+ log(device, device.logger, devices::logging::Verbosity::info, rendering::raytracing::LOG_PREFIX, "Probe rays disabled (RL_TOOLS_RENDERING_RAYTRACING_DISABLE_PROBE_RAYS=1)");
905
+ return;
906
+ #else
907
+ namespace metal = rendering::raytracing::backends::metal;
908
+ auto& ctx = metal::context(renderer);
909
+ std::vector<float> dirs = rendering::raytracing::detail::generate_probe_direction_vectors<SPEC>(device);
910
+ ctx.probe_directions = NS::TransferPtr(ctx.device->newBuffer(dirs.data(), dirs.size() * sizeof(float), MTL::ResourceStorageModeShared));
911
+ #endif
912
+ }
913
+
914
+ // render produces the image outputs the spec declares; the collision-probe pass is the
915
+ // separate probe verb so it can be scheduled independently (e.g. alongside update)
916
+ template <typename DEVICE, typename SPEC>
917
+ void render_launch(DEVICE& device, rendering::raytracing::Renderer<SPEC, rendering::raytracing::backends::Metal>& renderer){
918
+ namespace metal = rendering::raytracing::backends::metal;
919
+ using TI = typename SPEC::TI;
920
+ auto& ctx = metal::context(renderer);
921
+ NS::AutoreleasePool* pool = NS::AutoreleasePool::alloc()->init();
922
+ MTL::CommandBuffer* command_buffer = ctx.queue->commandBuffer();
923
+ if constexpr (SPEC::ENABLE_DYNAMIC_MOTION_BLUR){
924
+ // one command buffer per frame: zero the accumulators, then per motion sample build
925
+ // the overlay structures in place from that sample's descriptors and run the
926
+ // accumulate dispatches, then restore the shutter-close state (segmentation, probes,
927
+ // steady state) and resolve — encoder order plus Metal hazard tracking on the
928
+ // acceleration structures and accumulators serializes builds against dispatches
929
+ uint32_t num_active_per_overlay[SPEC::NUM_OVERLAYS];
930
+ uint32_t total_active = 0;
931
+ for(TI overlay = 0; overlay < SPEC::NUM_OVERLAYS; overlay++){
932
+ uint32_t num_active = 0;
933
+ for(TI slot = 0; slot < SPEC::MAX_OVERLAY_INSTANCES; slot++){
934
+ num_active += renderer.overlays[overlay].slots[slot].active ? 1 : 0;
935
+ }
936
+ num_active_per_overlay[overlay] = num_active;
937
+ total_active += num_active;
938
+ }
939
+ {
940
+ MTL::BlitCommandEncoder* blit = command_buffer->blitCommandEncoder();
941
+ if constexpr (SPEC::HAS_RGB){
942
+ blit->fillBuffer(ctx.rgb_accumulator.get(), NS::Range::Make(0, ctx.rgb_accumulator->length()), 0);
943
+ }
944
+ blit->endEncoding();
945
+ }
946
+ std::vector<NS::Object*> object_acceleration_structure_pointers;
947
+ for(auto& object_acceleration_structure : ctx.object_acceleration_structures){
948
+ object_acceleration_structure_pointers.push_back(object_acceleration_structure.get());
949
+ }
950
+ NS::Array* object_array = NS::Array::array((const NS::Object* const*)object_acceleration_structure_pointers.data(), object_acceleration_structure_pointers.size());
951
+ const auto encode_overlay_builds = [&](auto&& descriptor_buffer_for_overlay){
952
+ if(total_active == 0){
953
+ return;
954
+ }
955
+ MTL::AccelerationStructureCommandEncoder* encoder = command_buffer->accelerationStructureCommandEncoder();
956
+ for(TI overlay = 0; overlay < SPEC::NUM_OVERLAYS; overlay++){
957
+ if(num_active_per_overlay[overlay] == 0) continue;
958
+ MTL::InstanceAccelerationStructureDescriptor* overlay_descriptor = MTL::InstanceAccelerationStructureDescriptor::descriptor();
959
+ overlay_descriptor->setInstancedAccelerationStructures(object_array);
960
+ overlay_descriptor->setInstanceCount(num_active_per_overlay[overlay]);
961
+ overlay_descriptor->setInstanceDescriptorBuffer(descriptor_buffer_for_overlay(overlay));
962
+ overlay_descriptor->setInstanceDescriptorType(MTL::AccelerationStructureInstanceDescriptorTypeUserID);
963
+ encoder->buildAccelerationStructure(ctx.overlay_acceleration_structures[overlay].get(), overlay_descriptor, ctx.overlay_scratch_buffers[overlay].get(), 0);
964
+ }
965
+ encoder->endEncoding();
966
+ };
967
+ for(TI sample = 0; sample < SPEC::MOTION_BLUR_SAMPLES; sample++){
968
+ encode_overlay_builds([&](TI overlay){ return ctx.overlay_sample_instance_descriptors[(size_t)sample * SPEC::NUM_OVERLAYS + overlay].get(); });
969
+ const float shutter_t = ((float)sample + 0.5f) / (float)SPEC::MOTION_BLUR_SAMPLES;
970
+ if constexpr (SPEC::HAS_RGB) {
971
+ metal::encode_fullscreen_pass<SPEC>(ctx, command_buffer, ctx.rgb_pipeline.get(), ctx.frame_buffer.get(), shutter_t);
972
+ }
973
+ }
974
+ encode_overlay_builds([&](TI overlay){ return ctx.overlay_instance_descriptors[overlay].get(); });
975
+ if constexpr (SPEC::HAS_DEPTH) {
976
+ metal::encode_fullscreen_pass<SPEC>(ctx, command_buffer, ctx.depth_pipeline.get(), ctx.depth_buffer.get());
977
+ }
978
+ if constexpr (SPEC::HAS_SEGMENTATION) {
979
+ metal::encode_fullscreen_pass<SPEC>(ctx, command_buffer, ctx.segmentation_pipeline.get(), ctx.segmentation_buffer.get());
980
+ }
981
+ if constexpr (SPEC::HAS_NORMALS) {
982
+ metal::encode_fullscreen_pass<SPEC>(ctx, command_buffer, ctx.normals_pipeline.get(), ctx.normals_buffer.get());
983
+ }
984
+ if constexpr (SPEC::HAS_FLOW) {
985
+ metal::encode_fullscreen_pass<SPEC>(ctx, command_buffer, ctx.flow_pipeline.get(), ctx.flow_buffer.get(), 0.f, ctx.flow_status_buffer.get());
986
+ }
987
+ if constexpr (SPEC::HAS_FLOW_FORWARD) {
988
+ metal::encode_fullscreen_pass<SPEC>(ctx, command_buffer, ctx.flow_forward_pipeline.get(), ctx.flow_forward_buffer.get(), 0.f, ctx.flow_forward_status_buffer.get());
989
+ }
990
+ metal::encode_resolve_pass<SPEC>(ctx, command_buffer);
991
+ command_buffer->commit();
992
+ ctx.in_flight = NS::RetainPtr(command_buffer);
993
+ pool->release();
994
+ return;
995
+ }
996
+ if constexpr (SPEC::HAS_RGB) {
997
+ metal::encode_fullscreen_pass<SPEC>(ctx, command_buffer, ctx.rgb_pipeline.get(), ctx.frame_buffer.get());
998
+ }
999
+ if constexpr (SPEC::HAS_DEPTH) {
1000
+ metal::encode_fullscreen_pass<SPEC>(ctx, command_buffer, ctx.depth_pipeline.get(), ctx.depth_buffer.get());
1001
+ }
1002
+ if constexpr (SPEC::HAS_SEGMENTATION) {
1003
+ metal::encode_fullscreen_pass<SPEC>(ctx, command_buffer, ctx.segmentation_pipeline.get(), ctx.segmentation_buffer.get());
1004
+ }
1005
+ if constexpr (SPEC::HAS_NORMALS) {
1006
+ metal::encode_fullscreen_pass<SPEC>(ctx, command_buffer, ctx.normals_pipeline.get(), ctx.normals_buffer.get());
1007
+ }
1008
+ if constexpr (SPEC::HAS_FLOW) {
1009
+ metal::encode_fullscreen_pass<SPEC>(ctx, command_buffer, ctx.flow_pipeline.get(), ctx.flow_buffer.get(), 0.f, ctx.flow_status_buffer.get());
1010
+ }
1011
+ if constexpr (SPEC::HAS_FLOW_FORWARD) {
1012
+ metal::encode_fullscreen_pass<SPEC>(ctx, command_buffer, ctx.flow_forward_pipeline.get(), ctx.flow_forward_buffer.get(), 0.f, ctx.flow_forward_status_buffer.get());
1013
+ }
1014
+ command_buffer->commit();
1015
+ ctx.in_flight = NS::RetainPtr(command_buffer);
1016
+ pool->release();
1017
+ }
1018
+
1019
+ template <typename DEVICE, typename SPEC>
1020
+ void render_sync(DEVICE& device, rendering::raytracing::Renderer<SPEC, rendering::raytracing::backends::Metal>& renderer){
1021
+ namespace metal = rendering::raytracing::backends::metal;
1022
+ auto& ctx = metal::context(renderer);
1023
+ if(ctx.in_flight.get() != nullptr){
1024
+ ctx.in_flight->waitUntilCompleted();
1025
+ ctx.in_flight.reset();
1026
+ }
1027
+ }
1028
+
1029
+ template <typename DEVICE, typename SPEC>
1030
+ void probe_launch(DEVICE& device, rendering::raytracing::Renderer<SPEC, rendering::raytracing::backends::Metal>& renderer){
1031
+ namespace metal = rendering::raytracing::backends::metal;
1032
+ auto& ctx = metal::context(renderer);
1033
+ if(ctx.collision_pipeline.get() != nullptr){
1034
+ NS::AutoreleasePool* pool = NS::AutoreleasePool::alloc()->init();
1035
+ MTL::CommandBuffer* command_buffer = ctx.queue->commandBuffer();
1036
+ metal::encode_collision_pass<SPEC>(ctx, command_buffer);
1037
+ command_buffer->commit();
1038
+ ctx.in_flight_collision = NS::RetainPtr(command_buffer);
1039
+ pool->release();
1040
+ }
1041
+ }
1042
+
1043
+ template <typename DEVICE, typename SPEC>
1044
+ void probe_sync(DEVICE& device, rendering::raytracing::Renderer<SPEC, rendering::raytracing::backends::Metal>& renderer){
1045
+ namespace metal = rendering::raytracing::backends::metal;
1046
+ auto& ctx = metal::context(renderer);
1047
+ if(ctx.in_flight_collision.get() != nullptr){
1048
+ ctx.in_flight_collision->waitUntilCompleted();
1049
+ ctx.in_flight_collision.reset();
1050
+ }
1051
+ }
1052
+
1053
+ template <typename DEVICE, typename SPEC>
1054
+ std::string device_name(DEVICE& device, rendering::raytracing::Renderer<SPEC, rendering::raytracing::backends::Metal>& renderer){
1055
+ return renderer.backend->device->name()->utf8String();
1056
+ }
1057
+
1058
+ template <typename DEVICE, typename SPEC>
1059
+ void synchronize(DEVICE& device, rendering::raytracing::Renderer<SPEC, rendering::raytracing::backends::Metal>& renderer){
1060
+ namespace metal = rendering::raytracing::backends::metal;
1061
+ metal::wait_in_flight(metal::context(renderer));
1062
+ }
1063
+
1064
+ template <typename DEVICE, typename SPEC>
1065
+ void free(DEVICE& device, rendering::raytracing::Renderer<SPEC, rendering::raytracing::backends::Metal>& renderer){
1066
+ namespace metal = rendering::raytracing::backends::metal;
1067
+ if(renderer.backend != nullptr){
1068
+ metal::wait_in_flight(metal::context(renderer));
1069
+ delete renderer.backend;
1070
+ renderer.backend = nullptr;
1071
+ renderer.device.context = nullptr;
1072
+ }
1073
+ // the input and output tensors alias shared MTLBuffers destroyed with the context
1074
+ renderer.cameras._data = nullptr;
1075
+ if constexpr (SPEC::HAS_CAMERA_PAIR) {
1076
+ renderer.cameras_open._data = nullptr;
1077
+ }
1078
+ if constexpr (SPEC::HAS_RGB) {
1079
+ renderer.frame_buffer._data = nullptr;
1080
+ }
1081
+ if constexpr (SPEC::HAS_DEPTH) {
1082
+ renderer.depth_buffer._data = nullptr;
1083
+ }
1084
+ if constexpr (SPEC::HAS_SEGMENTATION) {
1085
+ renderer.segmentation_buffer._data = nullptr;
1086
+ }
1087
+ if constexpr (SPEC::HAS_NORMALS) {
1088
+ renderer.normals_buffer._data = nullptr;
1089
+ }
1090
+ if constexpr (SPEC::HAS_FLOW) {
1091
+ renderer.cameras_previous._data = nullptr;
1092
+ renderer.flow_buffer._data = nullptr;
1093
+ renderer.flow_status_buffer._data = nullptr;
1094
+ }
1095
+ if constexpr (SPEC::HAS_FLOW_FORWARD) {
1096
+ renderer.flow_forward_buffer._data = nullptr;
1097
+ renderer.flow_forward_status_buffer._data = nullptr;
1098
+ }
1099
+ if constexpr (SPEC::HAS_OBSERVATION) {
1100
+ renderer.observation._data = nullptr;
1101
+ }
1102
+ renderer.collision_results._data = nullptr;
1103
+ if constexpr (SPEC::ENABLE_OVERLAYS) {
1104
+ free(device, renderer.transforms);
1105
+ }
1106
+ if constexpr (SPEC::HAS_FLOW_OVERLAYS) {
1107
+ free(device, renderer.transforms_previous);
1108
+ }
1109
+ if constexpr (SPEC::ENABLE_DYNAMIC_MOTION_BLUR) {
1110
+ free(device, renderer.transforms_motion);
1111
+ if constexpr (SPEC::HAS_RGB) {
1112
+ renderer.rgb_accumulator._data = nullptr;
1113
+ }
1114
+ }
1115
+ }
1116
+
1117
+ }
1118
+ RL_TOOLS_NAMESPACE_WRAPPER_END
1119
+
1120
+
1121
+ #include "../../operations_cpu_post.h"
1122
+
1123
+ #endif