hyperlux 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 (794) 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 +26 -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 +191 -0
  22. hyperdrone/_native/env/environment.h +296 -0
  23. hyperdrone/_native/env/impl.cpp +783 -0
  24. hyperdrone/_native/env/rotorcraft_bindings.cpp +53 -0
  25. hyperdrone/_native/render/CMakeLists.txt +96 -0
  26. hyperdrone/_native/render/renderer.cpp +970 -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 +626 -0
  367. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/generic/operations_generic.h +1214 -0
  368. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/metal/context.h +195 -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 +1088 -0
  371. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/optix/device.h +215 -0
  372. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/optix/device_impl.h +1135 -0
  373. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/optix/operations_cuda.h +1701 -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 +35 -0
  376. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/backends/vulkan/context.h +234 -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 +2023 -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 +237 -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 +1277 -0
  383. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/operations_cpu_common.h +831 -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/outputs/operations_generic.h +225 -0
  387. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/outputs/outputs.h +228 -0
  388. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/renderer.h +388 -0
  389. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/save_cpu.h +319 -0
  390. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/scene.h +44 -0
  391. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/transforms_generic.h +25 -0
  392. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/raytracing/types.h +45 -0
  393. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/rig/operations_cpu.h +96 -0
  394. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/rig/operations_generic.h +154 -0
  395. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/rig/rig.h +70 -0
  396. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/scene.h +132 -0
  397. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/segmentation_cpu.h +145 -0
  398. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/transforms.h +204 -0
  399. hyperdrone/_vendor/rl-tools/include/rl_tools/rendering/types.h +76 -0
  400. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/algorithms.h +6 -0
  401. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/operations_cpu.h +1 -0
  402. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/operations_cpu_mkl.h +1 -0
  403. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/operations_generic.h +1 -0
  404. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/loop/core/config.h +243 -0
  405. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/loop/core/operations_cuda.h +9 -0
  406. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/loop/core/operations_generic.h +205 -0
  407. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/loop/core/persist.h +109 -0
  408. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/loop/core/state.h +51 -0
  409. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/mixed_imitation.h +63 -0
  410. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/operations_cuda.h +190 -0
  411. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/operations_generic.h +332 -0
  412. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/operations_generic_mixed_imitation.h +81 -0
  413. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/persist.h +28 -0
  414. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/ppo/ppo.h +105 -0
  415. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/loop/core/approximators_gru.h +84 -0
  416. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/loop/core/approximators_mlp.h +61 -0
  417. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/loop/core/config.h +106 -0
  418. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/loop/core/operations_generic.h +193 -0
  419. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/loop/core/state.h +52 -0
  420. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/operations_cpu.h +9 -0
  421. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/operations_cpu_accelerate.h +2 -0
  422. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/operations_cpu_mkl.h +2 -0
  423. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/operations_cpu_mux.h +9 -0
  424. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/operations_cuda.h +131 -0
  425. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/operations_generic.h +580 -0
  426. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/sac/sac.h +151 -0
  427. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/loop/core/approximators_gru.h +72 -0
  428. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/loop/core/approximators_mlp.h +54 -0
  429. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/loop/core/config.h +107 -0
  430. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/loop/core/operations_generic.h +174 -0
  431. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/loop/core/state.h +46 -0
  432. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/operations_cpu.h +9 -0
  433. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/operations_cpu_accelerate.h +2 -0
  434. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/operations_cpu_mkl.h +2 -0
  435. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/operations_cpu_mux.h +9 -0
  436. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/operations_cuda.h +117 -0
  437. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/operations_generic.h +359 -0
  438. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/algorithms/td3/td3.h +141 -0
  439. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/components.h +4 -0
  440. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/off_policy_runner/off_policy_runner.h +225 -0
  441. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/off_policy_runner/operations_cpu.h +95 -0
  442. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/off_policy_runner/operations_cpu_accelerate.h +21 -0
  443. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/off_policy_runner/operations_cpu_mkl.h +21 -0
  444. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/off_policy_runner/operations_cuda.h +237 -0
  445. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/off_policy_runner/operations_generic.h +498 -0
  446. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/off_policy_runner/operations_generic_per_env.h +112 -0
  447. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/collection.h +57 -0
  448. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/on_policy_runner.h +122 -0
  449. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/operations_cpu.h +2 -0
  450. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/operations_cpu_accelerate.h +8 -0
  451. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/operations_cpu_mkl.h +8 -0
  452. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/operations_cpu_mux.h +12 -0
  453. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/operations_cuda.h +183 -0
  454. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/operations_generic.h +366 -0
  455. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/operations_generic_common.h +74 -0
  456. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/on_policy_runner/persist.h +47 -0
  457. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/operations_cpu.h +2 -0
  458. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/operations_cpu_accelerate.h +1 -0
  459. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/operations_cpu_mkl.h +1 -0
  460. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/operations_generic.h +2 -0
  461. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/replay_buffer/operations_cpu.h +9 -0
  462. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/replay_buffer/operations_generic.h +123 -0
  463. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/replay_buffer/persist.h +57 -0
  464. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/replay_buffer/replay_buffer.h +93 -0
  465. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/running_normalizer/operations_generic.h +76 -0
  466. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/running_normalizer/persist.h +31 -0
  467. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/components/running_normalizer/running_normalizer.h +33 -0
  468. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environment_wrappers/operations_generic.h +58 -0
  469. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environment_wrappers/scale_observations/operations_generic.h +28 -0
  470. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environment_wrappers/scale_observations/wrapper.h +26 -0
  471. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environment_wrappers/wrappers.h +29 -0
  472. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/acrobot/acrobot.h +88 -0
  473. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/acrobot/operations_cpu.h +152 -0
  474. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/acrobot/operations_generic.h +202 -0
  475. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/batch/environment.h +40 -0
  476. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/batch/operations_cuda.h +204 -0
  477. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/batch/operations_generic.h +106 -0
  478. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/batch/operations_generic_common.h +26 -0
  479. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/batch/persist.h +33 -0
  480. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/car/car.h +119 -0
  481. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/car/operations_cpu.h +110 -0
  482. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/car/operations_generic.h +206 -0
  483. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/car/operations_json.h +59 -0
  484. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/car/track.h +23 -0
  485. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/car/ui.h +189 -0
  486. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/environments.h +28 -0
  487. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/flag/environment.h +87 -0
  488. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/flag/operations_cpu.h +212 -0
  489. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/flag/operations_generic.h +215 -0
  490. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/observation.h +75 -0
  491. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/operations_cpu.h +1079 -0
  492. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/operations_cuda.h +569 -0
  493. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/pose.h +253 -0
  494. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/presets.h +154 -0
  495. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/raptor/operations_cpu.h +26 -0
  496. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/raptor/operations_generic.h +91 -0
  497. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/raptor/raptor.h +71 -0
  498. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/rig/operations_cpu.h +225 -0
  499. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/rig/operations_generic.h +62 -0
  500. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/rig/rig.h +28 -0
  501. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/capture_cpu.h +86 -0
  502. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/capture_cuda.h +69 -0
  503. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/capture_optix.h +163 -0
  504. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/operations_cpu.h +223 -0
  505. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/operations_cpu_spectator.h +194 -0
  506. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/operations_cuda.h +1 -0
  507. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/operations_cuda_spectator.h +109 -0
  508. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/spectator.h +82 -0
  509. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/spy.h +108 -0
  510. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/spy/ui.h +30 -0
  511. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/moving_gate/moving_gate.h +103 -0
  512. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/moving_gate/operations_cpu.h +279 -0
  513. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/moving_gate/operations_cuda.h +242 -0
  514. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/target_frame/operations_cpu.h +324 -0
  515. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/target_frame/operations_cuda.h +314 -0
  516. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/target_frame/operations_generic_position_hold.h +128 -0
  517. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/target_frame/position_hold.h +22 -0
  518. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/target_frame/target_frame.h +94 -0
  519. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/target_frame/yaw/operations_cuda.h +43 -0
  520. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/target_frame/yaw/operations_generic.h +73 -0
  521. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/visual_inertial_localization/baseline.h +46 -0
  522. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/visual_inertial_localization/calibration.h +62 -0
  523. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/visual_inertial_localization/metrics.h +101 -0
  524. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/visual_inertial_localization/operations_cpu.h +374 -0
  525. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/tasks/visual_inertial_localization/visual_inertial_localization.h +126 -0
  526. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/hyperdrone/world.h +326 -0
  527. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/inertial_velocity.h +27 -0
  528. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/metrics.h +266 -0
  529. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/multirotor.h +1138 -0
  530. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_cpu.h +2252 -0
  531. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/05_state_is_nan.h +148 -0
  532. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/10_sample_initial_parameters.h +228 -0
  533. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/20_initial_state.h +208 -0
  534. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/30_sample_initial_state.h +300 -0
  535. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/35_get_desired_state.h +63 -0
  536. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/40_observe.h +637 -0
  537. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/50_state_algebra.h +57 -0
  538. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/60_dynamics.h +134 -0
  539. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/70_post_integration.h +368 -0
  540. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic/80_abs_diff.h +312 -0
  541. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic.h +367 -0
  542. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_generic_inertial_velocity.h +44 -0
  543. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_helper_generic.h +43 -0
  544. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_multitask_generic.h +70 -0
  545. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/operations_multitask_generic_forward.h +24 -0
  546. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/default.h +209 -0
  547. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/arpl.h +112 -0
  548. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/crazyflie.h +127 -0
  549. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/crazyflie_brushless.h +108 -0
  550. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/crazyflie_openmv.h +119 -0
  551. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/flightmare.h +111 -0
  552. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/fs.h +113 -0
  553. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/mrs.h +110 -0
  554. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/savagebee_pusher.h +108 -0
  555. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/soft.h +117 -0
  556. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/soft_rigid.h +118 -0
  557. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/x500.h +88 -0
  558. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/x500_real.h +88 -0
  559. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/dynamics/x500_sim.h +114 -0
  560. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/init/default.h +56 -0
  561. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/registry.h +108 -0
  562. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/reward_functions/default.h +64 -0
  563. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/reward_functions/reward_functions.h +1 -0
  564. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/reward_functions/squared/operations_generic.h +202 -0
  565. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/reward_functions/squared/squared.h +52 -0
  566. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/termination/default.h +33 -0
  567. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/trajectories/lissajous.h +72 -0
  568. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/parameters/trajectories/trajectory.h +78 -0
  569. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/persist.h +21 -0
  570. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/persist_code.h +53 -0
  571. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/quaternion_helper.h +103 -0
  572. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/l2f/ui.h +180 -0
  573. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/memory/environment.h +69 -0
  574. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/memory/operations_cpu.h +46 -0
  575. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/memory/operations_generic.h +78 -0
  576. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/mujoco/ant/README.MD +5 -0
  577. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/mujoco/ant/ant.h +77 -0
  578. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/mujoco/ant/model.h +415 -0
  579. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/mujoco/ant/operations_cpu.h +162 -0
  580. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/mujoco/ant/persist.h +86 -0
  581. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/mujoco/ant/ui.h +141 -0
  582. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/mujoco/mujoco.h +1 -0
  583. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/multi_agent/bottleneck/bottleneck.h +113 -0
  584. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/multi_agent/bottleneck/operations_cpu.h +289 -0
  585. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/multi_agent/bottleneck/operations_generic.h +495 -0
  586. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/multi_agent/environments.h +16 -0
  587. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/observation.h +37 -0
  588. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/operations_cpu.h +2 -0
  589. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/operations_generic.h +33 -0
  590. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/pendulum/operations_cpu.h +146 -0
  591. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/pendulum/operations_generic.h +262 -0
  592. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/pendulum/pendulum.h +197 -0
  593. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/pendulum/ui.h +97 -0
  594. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/pendulum/ui_xeus.h +124 -0
  595. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/reacher/operations_cpu.h +426 -0
  596. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/reacher/operations_generic.h +424 -0
  597. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/environments/reacher/reacher.h +137 -0
  598. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/loop.h +15 -0
  599. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/checkpoint/config.h +28 -0
  600. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/checkpoint/operations_cpu.h +233 -0
  601. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/checkpoint/persist.h +30 -0
  602. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/checkpoint/state.h +21 -0
  603. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/curriculum/config.h +25 -0
  604. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/curriculum/operations_generic.h +47 -0
  605. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/curriculum/persist.h +20 -0
  606. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/curriculum/state.h +17 -0
  607. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/evaluation/config.h +47 -0
  608. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/evaluation/operations_generic.h +106 -0
  609. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/evaluation/persist.h +40 -0
  610. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/evaluation/state.h +36 -0
  611. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/extrack/config.h +22 -0
  612. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/extrack/operations_cpu.h +44 -0
  613. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/extrack/persist.h +20 -0
  614. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/extrack/state.h +26 -0
  615. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/nn_analytics/config.h +37 -0
  616. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/nn_analytics/operations_cpu.h +113 -0
  617. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/nn_analytics/persist.h +20 -0
  618. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/nn_analytics/state.h +24 -0
  619. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/save_trajectories/config.h +43 -0
  620. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/save_trajectories/operations_cpu.h +190 -0
  621. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/save_trajectories/persist.h +30 -0
  622. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/save_trajectories/state.h +37 -0
  623. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/steps.h +4 -0
  624. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/timing/config.h +27 -0
  625. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/timing/operations_cpu.h +52 -0
  626. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/timing/persist.h +20 -0
  627. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/loop/steps/timing/state.h +25 -0
  628. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/operations_generic.h +3 -0
  629. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/rl.h +5 -0
  630. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/utils/evaluation/evaluation.h +106 -0
  631. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/utils/evaluation/operations_cpu.h +48 -0
  632. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/utils/evaluation/operations_generic.h +275 -0
  633. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/utils/validation.h +269 -0
  634. hyperdrone/_vendor/rl-tools/include/rl_tools/rl/utils/validation_analysis.h +85 -0
  635. hyperdrone/_vendor/rl-tools/include/rl_tools/rl_tools.h +80 -0
  636. hyperdrone/_vendor/rl-tools/include/rl_tools/ui_server/client/client.h +50 -0
  637. hyperdrone/_vendor/rl-tools/include/rl_tools/ui_server/client/operations_boost.h +92 -0
  638. hyperdrone/_vendor/rl-tools/include/rl_tools/ui_server/client/operations_cpu.h +174 -0
  639. hyperdrone/_vendor/rl-tools/include/rl_tools/ui_server/client/operations_websocket.h +208 -0
  640. hyperdrone/_vendor/rl-tools/include/rl_tools/ui_server/server.h +448 -0
  641. hyperdrone/_vendor/rl-tools/include/rl_tools/universe.h +4 -0
  642. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/declarations_cpu.h +13 -0
  643. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/declarations_cpu_emscripten.h +15 -0
  644. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/operations_arm.h +29 -0
  645. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/operations_cpu.h +24 -0
  646. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/operations_cpu_emscripten.h +28 -0
  647. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/operations_cuda.h +33 -0
  648. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/operations_dummy.h +18 -0
  649. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/operations_esp32.h +20 -0
  650. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/operations_generic.h +19 -0
  651. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/assert/operations_wasm32.h +18 -0
  652. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/env/operations_cpu.h +16 -0
  653. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/env/operations_generic.h +14 -0
  654. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/extrack/extrack.h +129 -0
  655. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/extrack/operations_cpu.h +588 -0
  656. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/generic/integrators.h +103 -0
  657. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/generic/memcpy.h +16 -0
  658. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/generic/tuple/operations_generic.h +48 -0
  659. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/generic/tuple/tuple.h +147 -0
  660. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/generic/typing.h +122 -0
  661. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/generic/vector_operations.h +172 -0
  662. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/polyak/operations_cuda.h +53 -0
  663. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/polyak/operations_generic.h +93 -0
  664. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/string/operations_generic.h +201 -0
  665. hyperdrone/_vendor/rl-tools/include/rl_tools/utils/zlib/operations_cpu.h +45 -0
  666. hyperdrone/_vendor/rl-tools/include/rl_tools/version.h +28 -0
  667. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/CMakeLists.txt +15 -0
  668. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/README.md +261 -0
  669. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/CMakeLists.txt +11 -0
  670. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/generic/CMakeLists.txt +17 -0
  671. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/generic/freestanding_check.cpp +125 -0
  672. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/metal/CMakeLists.txt +31 -0
  673. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/metal/device.metal +1053 -0
  674. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/metal/metal_cpp_impl.cpp +6 -0
  675. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/optix/CMakeLists.txt +51 -0
  676. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/optix/device.cu +1 -0
  677. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/optix/overlay_accel.cu +310 -0
  678. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/vulkan/CMakeLists.txt +37 -0
  679. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/vulkan/device.comp +1088 -0
  680. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/vulkan/device_spirv.cpp +44 -0
  681. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/webgpu/CMakeLists.txt +31 -0
  682. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/backends/webgpu/device.wgsl +1538 -0
  683. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/CMakeLists.txt +129 -0
  684. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/README.md +131 -0
  685. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/benchmark.cpp +184 -0
  686. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/benchmark_dynamic.cpp +497 -0
  687. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/download_paper_scene.py +74 -0
  688. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/ios_app/CMakeLists.txt +104 -0
  689. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/ios_app/Info.plist +36 -0
  690. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/ios_app/Sources/BenchmarkApp.swift +172 -0
  691. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/ios_app/benchmark.h +10 -0
  692. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/ios_app/benchmark.mm +46 -0
  693. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/ios_app/build.sh +17 -0
  694. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/rig.cpp +119 -0
  695. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix.cpp +1384 -0
  696. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix.h +5 -0
  697. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix_config.h.in +25 -0
  698. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix_identity.cmake +11 -0
  699. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix_main.cpp +5 -0
  700. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix_physics.cu +125 -0
  701. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix_physics.h +58 -0
  702. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix_physics_contract.h +179 -0
  703. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/benchmark/simulator_matrix_physics_cpu.cpp +77 -0
  704. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/camera_orbit.h +51 -0
  705. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/CMakeLists.txt +199 -0
  706. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/antialiasing.cpp +215 -0
  707. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/drone.cpp +321 -0
  708. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/drone_device.cu +359 -0
  709. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/environment/environment.h +92 -0
  710. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/environment/operations_cpu.h +263 -0
  711. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/example.cpp +274 -0
  712. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/fixed_pose.cpp +223 -0
  713. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/flow.cpp +120 -0
  714. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/interactive.cpp +1021 -0
  715. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/minimal.cpp +84 -0
  716. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/minimal_overlay.cpp +139 -0
  717. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/motion_blur.cpp +233 -0
  718. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/render_pose_trace.cpp +1829 -0
  719. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/example/render_pose_trace_motion_blur.cpp +648 -0
  720. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/CMakeLists.txt +147 -0
  721. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/README.md +15 -0
  722. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/job.sbatch +28 -0
  723. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/model.h +260 -0
  724. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/model_config.h +24 -0
  725. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/model_forward_cuda.h +397 -0
  726. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/model_operations.h +418 -0
  727. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/model_persist.h +45 -0
  728. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/model_student.h +70 -0
  729. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/scene.cpp +223 -0
  730. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/scene.h +68 -0
  731. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/.gitignore +5 -0
  732. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/CMakeLists.txt +4 -0
  733. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/Info.plist +24 -0
  734. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/Info_ios.plist +36 -0
  735. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/Sources/CameraManager.swift +120 -0
  736. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/Sources/ContentView.swift +703 -0
  737. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/Sources/DatasetRecorder.swift +285 -0
  738. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/Sources/YawPredictorApp.swift +25 -0
  739. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/YawPredictor.entitlements +8 -0
  740. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/build.sh +66 -0
  741. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/build_ios.sh +133 -0
  742. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/generate.sh +6 -0
  743. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/h5_to_tar.cpp +82 -0
  744. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/inference.cpp +166 -0
  745. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/inference.h +26 -0
  746. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/project.yml +30 -0
  747. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/swift_app/test_tar_vs_hdf5.cpp +173 -0
  748. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/yaw_prediction_cuda.cu +699 -0
  749. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/yaw_prediction_dataset_overlay.cpp +475 -0
  750. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/yaw_prediction_distill_cuda.cu +995 -0
  751. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/yaw_prediction_eval.cu +457 -0
  752. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/yaw_prediction_real_world_eval.cpp +660 -0
  753. hyperdrone/_vendor/rl-tools/src/rendering/raytracing/yaw_prediction/yaw_prediction_viewer.cu +443 -0
  754. hyperdrone/_vendor/rl-tools/src/rl/environments/hyperdrone/spy/CMakeLists.txt +63 -0
  755. hyperdrone/_vendor/rl-tools/src/rl/environments/hyperdrone/spy/example.cpp +102 -0
  756. hyperdrone/_vendor/rl-tools/src/rl/environments/hyperdrone/spy/ui.cpp +380 -0
  757. hyperdrone/dynamics/__init__.py +5 -0
  758. hyperdrone/dynamics/_component.py +31 -0
  759. hyperdrone/dynamics/_config.py +48 -0
  760. hyperdrone/dynamics/_sim.py +166 -0
  761. hyperdrone/env/__init__.py +23 -0
  762. hyperdrone/env/_multi_environment.py +570 -0
  763. hyperdrone/env/_rotorcraft.py +29 -0
  764. hyperdrone/env/_spy.py +39 -0
  765. hyperdrone/examples/.gitignore +3 -0
  766. hyperdrone/examples/Benchmark.ipynb +113 -0
  767. hyperdrone/examples/HyperDrone.ipynb +597 -0
  768. hyperdrone/examples/Orbiter.ipynb +497 -0
  769. hyperdrone/examples/Overview.ipynb +380 -0
  770. hyperdrone/examples/Renderer.ipynb +218 -0
  771. hyperdrone/examples/__init__.py +16 -0
  772. hyperdrone/examples/articulation.py +55 -0
  773. hyperdrone/examples/benchmark.py +236 -0
  774. hyperdrone/examples/benchmark_minimal.py +65 -0
  775. hyperdrone/examples/data.py +56 -0
  776. hyperdrone/examples/drone_flythrough.py +169 -0
  777. hyperdrone/examples/hyperdrone_videos/my_world_preview.h +17 -0
  778. hyperdrone/examples/my_world.h +17 -0
  779. hyperdrone/examples/render_procthor.py +40 -0
  780. hyperdrone/examples/shots.py +108 -0
  781. hyperdrone/examples/spy.py +40 -0
  782. hyperdrone/examples/visual_inertial_localization.py +158 -0
  783. hyperdrone/gym/__init__.py +67 -0
  784. hyperdrone/render/__init__.py +57 -0
  785. hyperdrone/render/_component.py +51 -0
  786. hyperdrone/render/_config.py +76 -0
  787. hyperdrone/render/_renderer.py +498 -0
  788. hyperdrone/render/_rig.py +136 -0
  789. hyperdrone/render/_sampling.py +72 -0
  790. hyperdrone/render/_scene.py +38 -0
  791. hyperdrone/render/datasets.py +15 -0
  792. hyperlux-0.0.1.dist-info/METADATA +518 -0
  793. hyperlux-0.0.1.dist-info/RECORD +794 -0
  794. hyperlux-0.0.1.dist-info/WHEEL +4 -0
@@ -0,0 +1,1021 @@
1
+
2
+ #define RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_RGB 0
3
+ #define RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_RGBD 1
4
+ #define RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_DEPTH 2
5
+ #define RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_SEGMENTATION 3
6
+ #define RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_FLOW 4
7
+
8
+ #ifndef RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE
9
+ #define RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_RGB
10
+ #endif
11
+
12
+ #include <rl_tools/operations/cpu_mux.h>
13
+
14
+ #include "environment/environment.h"
15
+ #include "environment/operations_cpu.h"
16
+
17
+ #include <GLFW/glfw3.h>
18
+
19
+ #include <conta/conta.h>
20
+
21
+ #include <iostream>
22
+ #include <fstream>
23
+ #include <algorithm>
24
+ #include <cerrno>
25
+ #include <cmath>
26
+ #include <cstring>
27
+ #include <chrono>
28
+ #include <cstdio>
29
+ #include <cstdlib>
30
+ #include <iomanip>
31
+ #include <string>
32
+ #include <sys/stat.h>
33
+ #include <vector>
34
+
35
+ namespace rlt = rl_tools;
36
+ namespace rro = rlt::rendering::raytracing::outputs;
37
+
38
+ static constexpr bool OUTPUT_FLOW = RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_FLOW;
39
+ static constexpr bool OUTPUT_RGB = RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_RGB || RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_RGBD || OUTPUT_FLOW;
40
+ static constexpr bool OUTPUT_DEPTH = RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_RGBD || RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_DEPTH;
41
+ static constexpr bool OUTPUT_SEGMENTATION = RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_SEGMENTATION;
42
+ // flow mode renders at half the window width and shows RGB on the left and the selected flow view
43
+ // on the right; one orbiting, spinning cube overlay in front of the start pose adds object flow, a
44
+ // disocclusion band and a forward-direction target next to the camera's ego-motion flow
45
+ static constexpr unsigned int FLOW_NUM_OVERLAYS = OUTPUT_FLOW ? 1 : 0;
46
+ static constexpr unsigned int FLOW_HALVES = OUTPUT_FLOW ? 2 : 1;
47
+ static constexpr float FLOW_CUBE_HALF_EXTENT = 0.25f;
48
+ static constexpr float FLOW_CUBE_DISTANCE = 2.5f;
49
+ static constexpr float FLOW_CUBE_ORBIT_RADIUS = 0.6f;
50
+ static constexpr float FLOW_CUBE_ORBIT_RATE = 0.8f; // rad/s
51
+ static constexpr float FLOW_CUBE_SPIN_RATE = 1.5f; // rad/s about z
52
+ static constexpr float FLOW_DISPLAY_SCALE_PX = 32.0f; // flow magnitude that saturates the hue display
53
+
54
+ // Minimal 5x7 bitmap font for overlay text
55
+ struct FontGlyph {
56
+ char ch;
57
+ uint8_t rows[7];
58
+ };
59
+
60
+ static const FontGlyph FONT_GLYPHS[] = {
61
+ {'0', {0x0E,0x11,0x13,0x15,0x19,0x11,0x0E}},
62
+ {'1', {0x04,0x0C,0x04,0x04,0x04,0x04,0x0E}},
63
+ {'2', {0x0E,0x11,0x01,0x06,0x08,0x10,0x1F}},
64
+ {'3', {0x0E,0x11,0x01,0x06,0x01,0x11,0x0E}},
65
+ {'4', {0x02,0x06,0x0A,0x12,0x1F,0x02,0x02}},
66
+ {'5', {0x1F,0x10,0x1E,0x01,0x01,0x11,0x0E}},
67
+ {'6', {0x06,0x08,0x10,0x1E,0x11,0x11,0x0E}},
68
+ {'7', {0x1F,0x01,0x02,0x04,0x08,0x08,0x08}},
69
+ {'8', {0x0E,0x11,0x11,0x0E,0x11,0x11,0x0E}},
70
+ {'9', {0x0E,0x11,0x11,0x0F,0x01,0x02,0x0C}},
71
+ {'.', {0x00,0x00,0x00,0x00,0x00,0x00,0x04}},
72
+ {'-', {0x00,0x00,0x00,0x0E,0x00,0x00,0x00}},
73
+ {'+', {0x00,0x04,0x04,0x1F,0x04,0x04,0x00}},
74
+ {' ', {0x00,0x00,0x00,0x00,0x00,0x00,0x00}},
75
+ {':', {0x00,0x00,0x04,0x00,0x04,0x00,0x00}},
76
+ {'(', {0x02,0x04,0x08,0x08,0x08,0x04,0x02}},
77
+ {')', {0x08,0x04,0x02,0x02,0x02,0x04,0x08}},
78
+ {',', {0x00,0x00,0x00,0x00,0x00,0x04,0x08}},
79
+ {'=', {0x00,0x00,0x1F,0x00,0x1F,0x00,0x00}},
80
+ {'P', {0x1E,0x11,0x11,0x1E,0x10,0x10,0x10}},
81
+ {'Q', {0x0E,0x11,0x11,0x11,0x15,0x12,0x0D}},
82
+ {'o', {0x00,0x00,0x0E,0x11,0x11,0x11,0x0E}},
83
+ {'s', {0x00,0x00,0x0E,0x10,0x0E,0x01,0x1E}},
84
+ {'u', {0x00,0x00,0x11,0x11,0x11,0x11,0x0F}},
85
+ {'a', {0x00,0x00,0x0E,0x01,0x0F,0x11,0x0F}},
86
+ {'t', {0x08,0x08,0x1C,0x08,0x08,0x09,0x06}},
87
+ {'F', {0x1F,0x10,0x10,0x1E,0x10,0x10,0x10}},
88
+ {'l', {0x0C,0x04,0x04,0x04,0x04,0x04,0x0E}},
89
+ {'w', {0x00,0x00,0x11,0x11,0x15,0x15,0x0A}},
90
+ {'b', {0x10,0x10,0x16,0x19,0x11,0x11,0x1E}},
91
+ {'c', {0x00,0x00,0x0E,0x10,0x10,0x11,0x0E}},
92
+ {'k', {0x10,0x10,0x12,0x14,0x18,0x14,0x12}},
93
+ {'r', {0x00,0x00,0x16,0x19,0x10,0x10,0x10}},
94
+ {'d', {0x01,0x01,0x0D,0x13,0x11,0x11,0x0F}},
95
+ {'f', {0x06,0x09,0x08,0x1C,0x08,0x08,0x08}},
96
+ {'g', {0x00,0x00,0x0F,0x11,0x0F,0x01,0x0E}},
97
+ };
98
+
99
+ static const uint8_t* font_lookup(char ch) {
100
+ static const uint8_t empty[7] = {};
101
+ for (const auto& g : FONT_GLYPHS) {
102
+ if (g.ch == ch) return g.rows;
103
+ }
104
+ return empty;
105
+ }
106
+
107
+ static void draw_char(uint32_t* pixels, int width, int height, int x0, int y0, char ch, uint32_t color) {
108
+ const uint8_t* glyph = font_lookup(ch);
109
+ for (int row = 0; row < 7; row++) {
110
+ for (int col = 0; col < 5; col++) {
111
+ if (glyph[row] & (0x10 >> col)) {
112
+ int px = x0 + col;
113
+ int py = y0 + row;
114
+ if (px >= 0 && px < width && py >= 0 && py < height) {
115
+ pixels[py * width + px] = color;
116
+ }
117
+ }
118
+ }
119
+ }
120
+ }
121
+
122
+ static void draw_string(uint32_t* pixels, int width, int height, int x0, int y0, const char* str, uint32_t color) {
123
+ int x = x0;
124
+ for (int i = 0; str[i] != '\0'; i++) {
125
+ draw_char(pixels, width, height, x, y0, str[i], color);
126
+ x += 6;
127
+ }
128
+ }
129
+
130
+ #if RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_RGBD || RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_DEPTH
131
+ static void depth_to_rgba(const float* depth, uint32_t* pixels, int count, float max_depth) {
132
+ for (int i = 0; i < count; i++) {
133
+ float normalized = std::fmin(std::fmax(depth[i] / max_depth, 0.0f), 1.0f);
134
+ uint8_t value = static_cast<uint8_t>(normalized * 255.0f);
135
+ pixels[i] = (0xFFu << 24) | (uint32_t(value) << 16) | (uint32_t(value) << 8) | uint32_t(value);
136
+ }
137
+ }
138
+ #endif
139
+
140
+ static void draw_overlay_background(uint32_t* pixels, int width, int /*height*/, int x0, int y0, int w, int h) {
141
+ for (int row = y0; row < y0 + h; row++) {
142
+ for (int col = x0; col < x0 + w; col++) {
143
+ if (row >= 0 && col >= 0) {
144
+ uint32_t pixel = pixels[row * width + col];
145
+ uint8_t r = (pixel >> 0) & 0xFF;
146
+ uint8_t g = (pixel >> 8) & 0xFF;
147
+ uint8_t b = (pixel >> 16) & 0xFF;
148
+ r = r / 3;
149
+ g = g / 3;
150
+ b = b / 3;
151
+ pixels[row * width + col] = (0xFF << 24) | (b << 16) | (g << 8) | r;
152
+ }
153
+ }
154
+ }
155
+ }
156
+
157
+ struct Quaternion {
158
+ float w, x, y, z;
159
+ };
160
+
161
+ static void cross3(const float a[3], const float b[3], float out[3]) {
162
+ out[0] = a[1] * b[2] - a[2] * b[1];
163
+ out[1] = a[2] * b[0] - a[0] * b[2];
164
+ out[2] = a[0] * b[1] - a[1] * b[0];
165
+ }
166
+
167
+ static bool normalize3(float v[3]) {
168
+ const float norm = std::sqrt(v[0] * v[0] + v[1] * v[1] + v[2] * v[2]);
169
+ if(norm <= 1e-6f) {
170
+ return false;
171
+ }
172
+ v[0] /= norm;
173
+ v[1] /= norm;
174
+ v[2] /= norm;
175
+ return true;
176
+ }
177
+
178
+ static Quaternion normalize_quaternion(Quaternion q) {
179
+ const float norm = std::sqrt(q.w * q.w + q.x * q.x + q.y * q.y + q.z * q.z);
180
+ if(norm <= 1e-6f) {
181
+ return {1.0f, 0.0f, 0.0f, 0.0f};
182
+ }
183
+ return {q.w / norm, q.x / norm, q.y / norm, q.z / norm};
184
+ }
185
+
186
+ static Quaternion quaternion_from_rotation_columns(const float x_axis[3], const float y_axis[3], const float z_axis[3]) {
187
+ const float m00 = x_axis[0], m01 = y_axis[0], m02 = z_axis[0];
188
+ const float m10 = x_axis[1], m11 = y_axis[1], m12 = z_axis[1];
189
+ const float m20 = x_axis[2], m21 = y_axis[2], m22 = z_axis[2];
190
+ Quaternion q;
191
+ const float trace = m00 + m11 + m22;
192
+ if(trace > 0.0f) {
193
+ const float s = std::sqrt(trace + 1.0f) * 2.0f;
194
+ q.w = 0.25f * s;
195
+ q.x = (m21 - m12) / s;
196
+ q.y = (m02 - m20) / s;
197
+ q.z = (m10 - m01) / s;
198
+ }
199
+ else if(m00 > m11 && m00 > m22) {
200
+ const float s = std::sqrt(1.0f + m00 - m11 - m22) * 2.0f;
201
+ q.w = (m21 - m12) / s;
202
+ q.x = 0.25f * s;
203
+ q.y = (m01 + m10) / s;
204
+ q.z = (m02 + m20) / s;
205
+ }
206
+ else if(m11 > m22) {
207
+ const float s = std::sqrt(1.0f + m11 - m00 - m22) * 2.0f;
208
+ q.w = (m02 - m20) / s;
209
+ q.x = (m01 + m10) / s;
210
+ q.y = 0.25f * s;
211
+ q.z = (m12 + m21) / s;
212
+ }
213
+ else {
214
+ const float s = std::sqrt(1.0f + m22 - m00 - m11) * 2.0f;
215
+ q.w = (m10 - m01) / s;
216
+ q.x = (m02 + m20) / s;
217
+ q.y = (m12 + m21) / s;
218
+ q.z = 0.25f * s;
219
+ }
220
+ return normalize_quaternion(q);
221
+ }
222
+
223
+ static Quaternion l2f_camera_quaternion(const float forward_in[3], const float up_reference_in[3]) {
224
+ float x_axis[3] = {forward_in[0], forward_in[1], forward_in[2]};
225
+ if(!normalize3(x_axis)) {
226
+ x_axis[0] = 1.0f;
227
+ x_axis[1] = 0.0f;
228
+ x_axis[2] = 0.0f;
229
+ }
230
+ float up_reference[3] = {up_reference_in[0], up_reference_in[1], up_reference_in[2]};
231
+ if(!normalize3(up_reference)) {
232
+ up_reference[0] = 0.0f;
233
+ up_reference[1] = 0.0f;
234
+ up_reference[2] = 1.0f;
235
+ }
236
+ float y_axis[3];
237
+ cross3(up_reference, x_axis, y_axis);
238
+ if(!normalize3(y_axis)) {
239
+ const float fallback_up[3] = {0.0f, 1.0f, 0.0f};
240
+ cross3(fallback_up, x_axis, y_axis);
241
+ if(!normalize3(y_axis)) {
242
+ y_axis[0] = 0.0f;
243
+ y_axis[1] = 1.0f;
244
+ y_axis[2] = 0.0f;
245
+ }
246
+ }
247
+ float z_axis[3];
248
+ cross3(x_axis, y_axis, z_axis);
249
+ normalize3(z_axis);
250
+ return quaternion_from_rotation_columns(x_axis, y_axis, z_axis);
251
+ }
252
+
253
+ struct InputState {
254
+ bool forward = false, backward = false, left = false, right = false;
255
+ bool up = false, down = false;
256
+ double mouse_x = 0, mouse_y = 0;
257
+ double last_mouse_x = 0, last_mouse_y = 0;
258
+ bool first_mouse = true;
259
+ bool cursor_captured = true;
260
+ float yaw = 0;
261
+ float pitch = 0;
262
+ };
263
+
264
+ static InputState g_input;
265
+ static bool g_capture_pose_requested = false;
266
+ #if RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_RGBD
267
+ static bool g_show_depth = true;
268
+ #endif
269
+ #if RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_FLOW
270
+ enum class FlowView : int {
271
+ BACKWARD = 0,
272
+ BACKWARD_STATUS,
273
+ FORWARD,
274
+ FORWARD_STATUS,
275
+ COUNT
276
+ };
277
+ static FlowView g_flow_view = FlowView::BACKWARD;
278
+ static const char* flow_view_label(FlowView view) {
279
+ switch(view) {
280
+ case FlowView::BACKWARD: return "Flow: backward";
281
+ case FlowView::BACKWARD_STATUS: return "Flow: backward status";
282
+ case FlowView::FORWARD: return "Flow: forward";
283
+ default: return "Flow: forward status";
284
+ }
285
+ }
286
+ static rlt::rendering::raytracing::Mesh make_flow_cube() {
287
+ using namespace rlt::rendering::raytracing::constants;
288
+ rlt::rendering::raytracing::Mesh mesh;
289
+ mesh.color[0] = 0.85f;
290
+ mesh.color[1] = 0.3f;
291
+ mesh.color[2] = 0.1f;
292
+ for(int vertex_i = 0; vertex_i < NUM_VERTICES; vertex_i++) {
293
+ for(int dim_i = 0; dim_i < 3; dim_i++) {
294
+ mesh.vertices.push_back(default_vertices[vertex_i][dim_i] * FLOW_CUBE_HALF_EXTENT);
295
+ }
296
+ }
297
+ for(int triangle_i = 0; triangle_i < NUM_INDICES; triangle_i++) {
298
+ for(int corner_i = 0; corner_i < 3; corner_i++) {
299
+ mesh.indices.push_back(default_indices[triangle_i][corner_i]);
300
+ }
301
+ }
302
+ return mesh;
303
+ }
304
+ static void flow_cube_transform(const float centre[3], float time_s, float transform[12]) {
305
+ const float orbit = FLOW_CUBE_ORBIT_RATE * time_s;
306
+ const float position[3] = {centre[0] + FLOW_CUBE_ORBIT_RADIUS * std::cos(orbit), centre[1] + FLOW_CUBE_ORBIT_RADIUS * std::sin(orbit), centre[2]};
307
+ const float half_spin = FLOW_CUBE_SPIN_RATE * time_s / 2.0f;
308
+ const float quaternion_wxyz[4] = {std::cos(half_spin), 0, 0, std::sin(half_spin)};
309
+ rlt::make_transform(position, quaternion_wxyz, transform);
310
+ }
311
+ #endif
312
+
313
+ static constexpr char CAPTURED_POSE_PATH[] = "camera_poses.txt";
314
+
315
+ struct InteractiveOptions {
316
+ std::string scene_arg;
317
+ std::string record_camera_pose_path;
318
+ float linear_velocity = 3.0f;
319
+ bool help = false;
320
+ };
321
+
322
+ struct RecordedCameraPose {
323
+ double timestamp_s;
324
+ size_t frame_index;
325
+ float position[3];
326
+ float look_at[3];
327
+ float up[3];
328
+ float yaw;
329
+ float pitch;
330
+ };
331
+
332
+ static void print_usage(const char* argv0) {
333
+ std::cerr << "Usage: " << argv0 << " [conta:HASH | scene.glb] [--record-camera-pose trace.json] [--linear-velocity m/s]" << std::endl;
334
+ std::cerr << " Without arguments the default scene is fetched via conta (downloaded into the cache if required)" << std::endl;
335
+ }
336
+
337
+ static bool parse_float_arg(const std::string& value, float& out) {
338
+ char* end = nullptr;
339
+ const float parsed = std::strtof(value.c_str(), &end);
340
+ if(end == value.c_str() || *end != '\0' || !std::isfinite(parsed)) {
341
+ return false;
342
+ }
343
+ out = parsed;
344
+ return true;
345
+ }
346
+
347
+ static bool parse_options(int argc, char** argv, InteractiveOptions& options) {
348
+ for(int i = 1; i < argc; i++) {
349
+ std::string arg = argv[i];
350
+ if(arg == "-h" || arg == "--help") {
351
+ options.help = true;
352
+ return true;
353
+ }
354
+ const std::string record_prefix = "--record-camera-pose=";
355
+ const std::string record_trace_prefix = "--record-camera-trace=";
356
+ const std::string linear_velocity_prefix = "--linear-velocity=";
357
+ if(arg == "--record-camera-pose" || arg == "--record-camera-trace") {
358
+ if(i + 1 >= argc) {
359
+ std::cerr << "Missing value for " << arg << std::endl;
360
+ return false;
361
+ }
362
+ options.record_camera_pose_path = argv[++i];
363
+ }
364
+ else if(arg.compare(0, record_prefix.size(), record_prefix) == 0) {
365
+ options.record_camera_pose_path = arg.substr(record_prefix.size());
366
+ }
367
+ else if(arg.compare(0, record_trace_prefix.size(), record_trace_prefix) == 0) {
368
+ options.record_camera_pose_path = arg.substr(record_trace_prefix.size());
369
+ }
370
+ else if(arg == "--linear-velocity") {
371
+ if(i + 1 >= argc) {
372
+ std::cerr << "Missing value for " << arg << std::endl;
373
+ return false;
374
+ }
375
+ if(!parse_float_arg(argv[++i], options.linear_velocity) || options.linear_velocity < 0.0f) {
376
+ std::cerr << "Invalid --linear-velocity" << std::endl;
377
+ return false;
378
+ }
379
+ }
380
+ else if(arg.compare(0, linear_velocity_prefix.size(), linear_velocity_prefix) == 0) {
381
+ if(!parse_float_arg(arg.substr(linear_velocity_prefix.size()), options.linear_velocity) || options.linear_velocity < 0.0f) {
382
+ std::cerr << "Invalid --linear-velocity" << std::endl;
383
+ return false;
384
+ }
385
+ }
386
+ else if(!arg.empty() && arg[0] == '-') {
387
+ std::cerr << "Unknown argument: " << arg << std::endl;
388
+ return false;
389
+ }
390
+ else if(options.scene_arg.empty()) {
391
+ options.scene_arg = arg;
392
+ }
393
+ else {
394
+ std::cerr << "Multiple scene arguments provided: " << options.scene_arg << " and " << arg << std::endl;
395
+ return false;
396
+ }
397
+ }
398
+ return true;
399
+ }
400
+
401
+ static std::string camera_cache_path(const std::string& scene_path) {
402
+ const char* home = std::getenv("HOME");
403
+ if (!home) return "";
404
+ std::string cache_dir = std::string(home) + "/.cache/rl_tools_viewer";
405
+ mkdir(cache_dir.c_str(), 0755);
406
+ std::string key;
407
+ for (char c : scene_path) key += (c == '/' || c == '\\') ? '_' : c;
408
+ return cache_dir + "/" + key + ".cam";
409
+ }
410
+
411
+ static void save_camera_state(const std::string& path, float x, float y, float z, float yaw, float pitch) {
412
+ if (path.empty()) return;
413
+ std::ofstream f(path, std::ios::binary);
414
+ if (f) f.write(reinterpret_cast<const char*>(&x), 4)
415
+ .write(reinterpret_cast<const char*>(&y), 4)
416
+ .write(reinterpret_cast<const char*>(&z), 4)
417
+ .write(reinterpret_cast<const char*>(&yaw), 4)
418
+ .write(reinterpret_cast<const char*>(&pitch), 4);
419
+ }
420
+
421
+ static bool load_camera_state(const std::string& path, float& x, float& y, float& z, float& yaw, float& pitch) {
422
+ if (path.empty()) return false;
423
+ std::ifstream f(path, std::ios::binary);
424
+ if (!f) return false;
425
+ return f.read(reinterpret_cast<char*>(&x), 4)
426
+ .read(reinterpret_cast<char*>(&y), 4)
427
+ .read(reinterpret_cast<char*>(&z), 4)
428
+ .read(reinterpret_cast<char*>(&yaw), 4)
429
+ .read(reinterpret_cast<char*>(&pitch), 4)
430
+ .good();
431
+ }
432
+
433
+ static std::string json_escape(const std::string& value) {
434
+ std::string escaped;
435
+ escaped.reserve(value.size());
436
+ for(char c : value) {
437
+ switch(c) {
438
+ case '"': escaped += "\\\""; break;
439
+ case '\\': escaped += "\\\\"; break;
440
+ case '\b': escaped += "\\b"; break;
441
+ case '\f': escaped += "\\f"; break;
442
+ case '\n': escaped += "\\n"; break;
443
+ case '\r': escaped += "\\r"; break;
444
+ case '\t': escaped += "\\t"; break;
445
+ default:
446
+ if(static_cast<unsigned char>(c) < 0x20) {
447
+ char buffer[7];
448
+ std::snprintf(buffer, sizeof(buffer), "\\u%04x", static_cast<unsigned char>(c));
449
+ escaped += buffer;
450
+ }
451
+ else {
452
+ escaped += c;
453
+ }
454
+ }
455
+ }
456
+ return escaped;
457
+ }
458
+
459
+ static void write_json_number(std::ostream& out, float value) {
460
+ if(std::isfinite(value)) {
461
+ out << value;
462
+ }
463
+ else {
464
+ out << "null";
465
+ }
466
+ }
467
+
468
+ static void write_json_vec3(std::ostream& out, const float v[3]) {
469
+ out << "[";
470
+ write_json_number(out, v[0]);
471
+ out << ", ";
472
+ write_json_number(out, v[1]);
473
+ out << ", ";
474
+ write_json_number(out, v[2]);
475
+ out << "]";
476
+ }
477
+
478
+ static void write_json_quaternion(std::ostream& out, const Quaternion& q) {
479
+ out << "[";
480
+ write_json_number(out, q.w);
481
+ out << ", ";
482
+ write_json_number(out, q.x);
483
+ out << ", ";
484
+ write_json_number(out, q.y);
485
+ out << ", ";
486
+ write_json_number(out, q.z);
487
+ out << "]";
488
+ }
489
+
490
+ static bool write_camera_trace_json(const std::string& path, const std::string& scene_path, const std::vector<RecordedCameraPose>& trace) {
491
+ if(path.empty()) return true;
492
+ const std::string tmp_path = path + ".tmp";
493
+
494
+ std::ofstream f(tmp_path);
495
+ if(!f) {
496
+ std::cerr << "Failed to write camera pose trace JSON: " << tmp_path << std::endl;
497
+ return false;
498
+ }
499
+
500
+ f << std::setprecision(9);
501
+ f << "{\n";
502
+ f << " \"scene_path\": \"" << json_escape(scene_path) << "\",\n";
503
+ f << " \"yaw_pitch_units\": \"radians\",\n";
504
+ f << " \"frame_count\": " << trace.size() << ",\n";
505
+ f << " \"poses\": [\n";
506
+ for(size_t i = 0; i < trace.size(); i++) {
507
+ const RecordedCameraPose& pose = trace[i];
508
+ const float forward[3] = {
509
+ pose.look_at[0] - pose.position[0],
510
+ pose.look_at[1] - pose.position[1],
511
+ pose.look_at[2] - pose.position[2]
512
+ };
513
+ const Quaternion q = l2f_camera_quaternion(forward, pose.up);
514
+
515
+ f << " {\n";
516
+ f << " \"timestamp_s\": " << pose.timestamp_s << ",\n";
517
+ f << " \"frame_index\": " << pose.frame_index << ",\n";
518
+ f << " \"renderer\": {\n";
519
+ f << " \"frame\": \"interactive_camera_input\",\n";
520
+ f << " \"position\": ";
521
+ write_json_vec3(f, pose.position);
522
+ f << ",\n";
523
+ f << " \"yaw\": ";
524
+ write_json_number(f, pose.yaw);
525
+ f << ",\n";
526
+ f << " \"pitch\": ";
527
+ write_json_number(f, pose.pitch);
528
+ f << ",\n";
529
+ f << " \"quaternion_wxyz\": ";
530
+ write_json_quaternion(f, q);
531
+ f << ",\n";
532
+ f << " \"forward\": ";
533
+ write_json_vec3(f, forward);
534
+ f << ",\n";
535
+ f << " \"up\": ";
536
+ write_json_vec3(f, pose.up);
537
+ f << ",\n";
538
+ f << " \"look_at\": ";
539
+ write_json_vec3(f, pose.look_at);
540
+ f << "\n";
541
+ f << " }\n";
542
+ f << " }" << (i + 1 < trace.size() ? "," : "") << "\n";
543
+ }
544
+ f << " ]\n";
545
+ f << "}\n";
546
+ f.close();
547
+ if(!f) {
548
+ std::cerr << "Failed while writing camera pose trace JSON: " << tmp_path << std::endl;
549
+ return false;
550
+ }
551
+ if(std::rename(tmp_path.c_str(), path.c_str()) != 0) {
552
+ std::cerr << "Failed to move camera pose trace JSON into place: " << std::strerror(errno) << std::endl;
553
+ return false;
554
+ }
555
+ return true;
556
+ }
557
+
558
+ static bool append_captured_pose(const char* path, const float position[3], const float look_at[3], const float up[3], float yaw, float pitch) {
559
+ std::ofstream f(path, std::ios::app);
560
+ if(!f) {
561
+ return false;
562
+ }
563
+ const float forward[3] = {
564
+ look_at[0] - position[0],
565
+ look_at[1] - position[1],
566
+ look_at[2] - position[2]
567
+ };
568
+ const Quaternion q = l2f_camera_quaternion(forward, up);
569
+ f << std::setprecision(9);
570
+ f << "position=" << position[0] << "," << position[1] << "," << position[2];
571
+ f << " yaw=" << yaw;
572
+ f << " pitch=" << pitch;
573
+ f << " quaternion_wxyz=" << q.w << "," << q.x << "," << q.y << "," << q.z;
574
+ f << "\n";
575
+ f.close();
576
+ return static_cast<bool>(f);
577
+ }
578
+
579
+ static void key_callback(GLFWwindow* window, int key, int /*scancode*/, int action, int /*mods*/) {
580
+ if (key == GLFW_KEY_ESCAPE && action == GLFW_PRESS) {
581
+ glfwSetWindowShouldClose(window, GLFW_TRUE);
582
+ return;
583
+ }
584
+ if (key == GLFW_KEY_C && action == GLFW_PRESS) {
585
+ g_capture_pose_requested = true;
586
+ return;
587
+ }
588
+ #if RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_RGBD
589
+ if (key == GLFW_KEY_TAB && action == GLFW_PRESS) {
590
+ g_show_depth = !g_show_depth;
591
+ return;
592
+ }
593
+ #endif
594
+ #if RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_FLOW
595
+ if (key == GLFW_KEY_TAB && action == GLFW_PRESS) {
596
+ g_flow_view = static_cast<FlowView>((static_cast<int>(g_flow_view) + 1) % static_cast<int>(FlowView::COUNT));
597
+ std::cout << flow_view_label(g_flow_view) << std::endl;
598
+ return;
599
+ }
600
+ #endif
601
+ bool pressed = (action == GLFW_PRESS || action == GLFW_REPEAT);
602
+ if (key == GLFW_KEY_W || key == GLFW_KEY_UP) g_input.forward = pressed;
603
+ if (key == GLFW_KEY_S || key == GLFW_KEY_DOWN) g_input.backward = pressed;
604
+ if (key == GLFW_KEY_A || key == GLFW_KEY_LEFT) g_input.left = pressed;
605
+ if (key == GLFW_KEY_D || key == GLFW_KEY_RIGHT) g_input.right = pressed;
606
+ if (key == GLFW_KEY_SPACE) g_input.up = pressed;
607
+ if (key == GLFW_KEY_LEFT_SHIFT) g_input.down = pressed;
608
+ }
609
+
610
+ static void cursor_pos_callback(GLFWwindow* /*window*/, double xpos, double ypos) {
611
+ if (!g_input.cursor_captured) return;
612
+ if (g_input.first_mouse) {
613
+ g_input.last_mouse_x = xpos;
614
+ g_input.last_mouse_y = ypos;
615
+ g_input.first_mouse = false;
616
+ return;
617
+ }
618
+ float dx = static_cast<float>(xpos - g_input.last_mouse_x);
619
+ float dy = static_cast<float>(ypos - g_input.last_mouse_y);
620
+ g_input.last_mouse_x = xpos;
621
+ g_input.last_mouse_y = ypos;
622
+
623
+ constexpr float sensitivity = 0.002f;
624
+ g_input.yaw -= dx * sensitivity;
625
+ g_input.pitch -= dy * sensitivity;
626
+ constexpr float max_pitch = static_cast<float>(M_PI) * 0.49f;
627
+ if (g_input.pitch > max_pitch) g_input.pitch = max_pitch;
628
+ if (g_input.pitch < -max_pitch) g_input.pitch = -max_pitch;
629
+ }
630
+
631
+ static void mouse_button_callback(GLFWwindow* window, int button, int action, int /*mods*/) {
632
+ if (button == GLFW_MOUSE_BUTTON_LEFT && action == GLFW_PRESS) {
633
+ if (g_input.cursor_captured) {
634
+ glfwSetInputMode(window, GLFW_CURSOR, GLFW_CURSOR_NORMAL);
635
+ g_input.cursor_captured = false;
636
+ } else {
637
+ glfwSetInputMode(window, GLFW_CURSOR, GLFW_CURSOR_DISABLED);
638
+ g_input.cursor_captured = true;
639
+ g_input.first_mouse = true;
640
+ }
641
+ }
642
+ }
643
+
644
+ int main(int argc, char** argv) {
645
+ using T = float;
646
+ using TI = unsigned int;
647
+ constexpr TI CAM_WIDTH = 1280;
648
+ constexpr TI CAM_HEIGHT = 960;
649
+ constexpr TI RENDER_WIDTH = CAM_WIDTH / FLOW_HALVES; // flow mode: each half of the window shows one full render
650
+ constexpr TI NUM_ENVS = 1;
651
+ using OUTPUTS = typename rro::Concat<
652
+ rro::If<OUTPUT_RGB, rro::RGB<>>, rro::If<OUTPUT_DEPTH, rro::Depth<>>,
653
+ rro::If<OUTPUT_SEGMENTATION, rro::Segmentation<>>,
654
+ rro::If<OUTPUT_FLOW, rro::Flow<>>, rro::If<OUTPUT_FLOW, rro::FlowStatus<>>,
655
+ rro::If<OUTPUT_FLOW, rro::Flow<rro::direction::Forward>>,
656
+ rro::If<OUTPUT_FLOW, rro::FlowStatus<rro::direction::Forward>>, rro::List<rro::Probes>>::type;
657
+ using SPEC = rlt::rl::environments::raytracing_example::Specification<T, TI, NUM_ENVS, RENDER_WIDTH, CAM_HEIGHT, 64, rlt::rendering::raytracing::High, false, 1, true, 2, OUTPUTS, FLOW_NUM_OVERLAYS, FLOW_NUM_OVERLAYS, FLOW_NUM_OVERLAYS>;
658
+ // FOV is horizontal: a half-width render keeps the full view's vertical field and pixel density
659
+ // by halving the tangent of the horizontal half-angle
660
+ const T fov = OUTPUT_FLOW ? static_cast<T>(2) * std::atan(std::tan(rlt::rendering::degrees_to_radians(SPEC::FOV) / static_cast<T>(2)) / static_cast<T>(FLOW_HALVES)) * static_cast<T>(180.0 / M_PI) : SPEC::FOV;
661
+ const T aspect = static_cast<T>(RENDER_WIDTH) / static_cast<T>(CAM_HEIGHT);
662
+ using DEVICE = rlt::devices::DEVICE_FACTORY<>;
663
+
664
+ InteractiveOptions options;
665
+ if(!parse_options(argc, argv, options)) {
666
+ print_usage(argv[0]);
667
+ return 1;
668
+ }
669
+ if(options.help) {
670
+ print_usage(argv[0]);
671
+ return 0;
672
+ }
673
+
674
+ DEVICE device;
675
+ rlt::init(device);
676
+
677
+ static constexpr char DEFAULT_CONTA_HASH[] = "75d1d67a17ecd16bc3a214baa895240da8060df7";
678
+ std::string resolved_scene_path;
679
+ std::string conta_error;
680
+ if (!options.scene_arg.empty()) {
681
+ const char* scene_arg = options.scene_arg.c_str();
682
+ if (std::strncmp(scene_arg, "conta:", 6) == 0) {
683
+ if (!conta::resolve(options.scene_arg.substr(6), resolved_scene_path, conta_error)) {
684
+ std::cerr << conta_error << std::endl;
685
+ return 1;
686
+ }
687
+ } else {
688
+ resolved_scene_path = scene_arg;
689
+ }
690
+ } else {
691
+ std::cout << "No scene argument given, using default: conta:" << DEFAULT_CONTA_HASH << std::endl;
692
+ if (!conta::resolve(DEFAULT_CONTA_HASH, resolved_scene_path, conta_error)) {
693
+ std::cerr << conta_error << std::endl;
694
+ return 1;
695
+ }
696
+ }
697
+ rlt::rl::environments::raytracing_example::Environment<SPEC> env;
698
+ env.scene_path = resolved_scene_path.c_str();
699
+
700
+ rlt::malloc(device, env);
701
+ #if RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_FLOW
702
+ const auto flow_cube_asset = rlt::add(device, env.pool, make_flow_cube());
703
+ #endif
704
+ rlt::init(device, env);
705
+ #if RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_RGBD
706
+ std::cout << "RGBD target enabled. Press Tab to toggle RGB/depth display." << std::endl;
707
+ #elif RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_DEPTH
708
+ std::cout << "Depth target enabled." << std::endl;
709
+ #elif RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_SEGMENTATION
710
+ std::cout << "Segmentation target enabled." << std::endl;
711
+ #elif RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_FLOW
712
+ std::cout << "Flow target enabled: RGB on the left, flow on the right. Press Tab to cycle backward flow / backward status / forward flow / forward status (hue = direction, saturation = magnitude up to " << FLOW_DISPLAY_SCALE_PX << " px; status: white VALID, black MISS, magenta BEHIND, blue OUT_OF_FRAME, orange OCCLUDED)." << std::endl;
713
+ #endif
714
+
715
+ rlt::rl::environments::raytracing_example::State<SPEC> state{};
716
+ if (env.num_indoor_initial_states > 0) {
717
+ state = env.indoor_initial_states[0];
718
+ }
719
+
720
+ std::string cam_cache = camera_cache_path(resolved_scene_path);
721
+ float cached_yaw = state.yaw, cached_pitch = 0;
722
+ if (load_camera_state(cam_cache, state.position[0], state.position[1], state.position[2], cached_yaw, cached_pitch)) {
723
+ std::cout << "Restored camera from " << cam_cache << std::endl;
724
+ }
725
+
726
+ if (!glfwInit()) {
727
+ std::cerr << "Failed to initialize GLFW" << std::endl;
728
+ return 1;
729
+ }
730
+ glfwWindowHint(GLFW_RESIZABLE, GLFW_FALSE);
731
+ GLFWwindow* window = glfwCreateWindow(CAM_WIDTH, CAM_HEIGHT, "RLtools Interactive Viewer", nullptr, nullptr);
732
+ if (!window) {
733
+ std::cerr << "Failed to create GLFW window" << std::endl;
734
+ glfwTerminate();
735
+ return 1;
736
+ }
737
+ glfwMakeContextCurrent(window);
738
+ glfwSwapInterval(0);
739
+ glfwSetKeyCallback(window, key_callback);
740
+ glfwSetCursorPosCallback(window, cursor_pos_callback);
741
+ glfwSetMouseButtonCallback(window, mouse_button_callback);
742
+ glfwSetInputMode(window, GLFW_CURSOR, GLFW_CURSOR_DISABLED);
743
+
744
+ g_input.yaw = cached_yaw;
745
+ g_input.pitch = cached_pitch;
746
+ #if RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_FLOW
747
+ const float flow_cube_centre[3] = {state.position[0] + FLOW_CUBE_DISTANCE * std::cos(cached_yaw), state.position[1] + FLOW_CUBE_DISTANCE * std::sin(cached_yaw), state.position[2]};
748
+ rlt::attach(device, *env.renderer, (TI)0, rlt::rendering::raytracing::OverlayIndex{0});
749
+ float flow_cube_pose[12];
750
+ flow_cube_transform(flow_cube_centre, 0.0f, flow_cube_pose);
751
+ const auto flow_cube_placement = rlt::spawn(device, *env.renderer, rlt::rendering::raytracing::OverlayIndex{0}, flow_cube_asset, flow_cube_pose);
752
+ rlt::update(device, *env.renderer);
753
+ #endif
754
+
755
+ GLuint texture;
756
+ glGenTextures(1, &texture);
757
+ glBindTexture(GL_TEXTURE_2D, texture);
758
+ glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
759
+ glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
760
+ glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, CAM_WIDTH, CAM_HEIGHT, 0, GL_RGBA, GL_UNSIGNED_BYTE, nullptr);
761
+
762
+ std::vector<uint32_t> pixels(CAM_WIDTH * CAM_HEIGHT);
763
+ std::vector<RecordedCameraPose> camera_trace;
764
+ rlt::Tensor<typename decltype(env.renderer->cameras)::SPEC> camera_staging;
765
+ rlt::malloc(device, camera_staging);
766
+ #if RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_FLOW
767
+ {
768
+ // publish the start pose as the current instant so the first advance_flow has a valid previous camera
769
+ const T eye[3] = {state.position[0], state.position[1], state.position[2]};
770
+ const T look_at[3] = {eye[0] + std::cos(g_input.yaw) * std::cos(g_input.pitch), eye[1] + std::sin(g_input.yaw) * std::cos(g_input.pitch), eye[2] + std::sin(g_input.pitch)};
771
+ const T up[3] = {0, 0, 1};
772
+ rlt::set(device, camera_staging, rlt::make_camera_data(eye, look_at, up, fov, aspect), 0);
773
+ rlt::copy(device, env.renderer->device, camera_staging, rlt::cameras(device, *env.renderer));
774
+ }
775
+ #endif
776
+
777
+ constexpr float LINEAR_VELOCITY_RAMP_S = 0.5f;
778
+ float linear_velocity_ramp_elapsed_s = 0.0f;
779
+ auto last_time = std::chrono::steady_clock::now();
780
+ const auto trace_start_time = last_time;
781
+ size_t frame_index = 0;
782
+
783
+ while (!glfwWindowShouldClose(window)) {
784
+ auto now = std::chrono::steady_clock::now();
785
+ float dt = std::chrono::duration<float>(now - last_time).count();
786
+ last_time = now;
787
+ if (dt > 0.1f) dt = 0.1f;
788
+
789
+ glfwPollEvents();
790
+
791
+ float cos_yaw = std::cos(g_input.yaw);
792
+ float sin_yaw = std::sin(g_input.yaw);
793
+ float dx = 0, dy = 0, dz = 0;
794
+ if (g_input.forward) { dx += cos_yaw; dy += sin_yaw; }
795
+ if (g_input.backward) { dx -= cos_yaw; dy -= sin_yaw; }
796
+ if (g_input.left) { dx -= sin_yaw; dy += cos_yaw; }
797
+ if (g_input.right) { dx += sin_yaw; dy -= cos_yaw; }
798
+ if (g_input.up) { dz += 1; }
799
+ if (g_input.down) { dz -= 1; }
800
+ float move_len = std::sqrt(dx * dx + dy * dy + dz * dz);
801
+ if (move_len > 0) {
802
+ linear_velocity_ramp_elapsed_s = std::min(linear_velocity_ramp_elapsed_s + dt, LINEAR_VELOCITY_RAMP_S);
803
+ const float ramp = LINEAR_VELOCITY_RAMP_S > 0.0f ? linear_velocity_ramp_elapsed_s / LINEAR_VELOCITY_RAMP_S : 1.0f;
804
+ float speed = options.linear_velocity * ramp * dt / move_len;
805
+ state.position[0] += dx * speed;
806
+ state.position[1] += dy * speed;
807
+ state.position[2] += dz * speed;
808
+ }
809
+ else {
810
+ linear_velocity_ramp_elapsed_s = 0.0f;
811
+ }
812
+ state.yaw = g_input.yaw;
813
+
814
+ T eye[3] = {state.position[0], state.position[1], state.position[2]};
815
+ T look_at[3] = {
816
+ eye[0] + std::cos(g_input.yaw) * std::cos(g_input.pitch),
817
+ eye[1] + std::sin(g_input.yaw) * std::cos(g_input.pitch),
818
+ eye[2] + std::sin(g_input.pitch)
819
+ };
820
+ T up[3] = {0, 0, 1};
821
+ const auto camera = rlt::make_camera_data(eye, look_at, up, fov, aspect);
822
+
823
+ if (g_capture_pose_requested) {
824
+ g_capture_pose_requested = false;
825
+ if (append_captured_pose(CAPTURED_POSE_PATH, eye, look_at, up, g_input.yaw, g_input.pitch)) {
826
+ std::cout << "Appended camera pose to " << CAPTURED_POSE_PATH << std::endl;
827
+ }
828
+ else {
829
+ std::cerr << "Failed to append camera pose to " << CAPTURED_POSE_PATH << std::endl;
830
+ }
831
+ }
832
+
833
+ if(!options.record_camera_pose_path.empty()) {
834
+ RecordedCameraPose pose;
835
+ pose.timestamp_s = std::chrono::duration<double>(now - trace_start_time).count();
836
+ pose.frame_index = frame_index;
837
+ pose.position[0] = eye[0];
838
+ pose.position[1] = eye[1];
839
+ pose.position[2] = eye[2];
840
+ pose.look_at[0] = look_at[0];
841
+ pose.look_at[1] = look_at[1];
842
+ pose.look_at[2] = look_at[2];
843
+ pose.up[0] = up[0];
844
+ pose.up[1] = up[1];
845
+ pose.up[2] = up[2];
846
+ pose.yaw = g_input.yaw;
847
+ pose.pitch = g_input.pitch;
848
+ camera_trace.push_back(pose);
849
+ }
850
+ frame_index++;
851
+
852
+ #if RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_FLOW
853
+ rlt::advance_flow(device, *env.renderer); // last frame's camera and cube pose become the previous instant
854
+ #endif
855
+ rlt::set(device, camera_staging, camera, 0);
856
+ rlt::copy(device, env.renderer->device, camera_staging, rlt::cameras(device, *env.renderer));
857
+ #if RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_FLOW
858
+ flow_cube_transform(flow_cube_centre, std::chrono::duration<float>(now - trace_start_time).count(), flow_cube_pose);
859
+ rlt::set_transform(device, *env.renderer, rlt::rendering::raytracing::OverlayIndex{0}, flow_cube_placement, flow_cube_pose);
860
+ rlt::update(device, *env.renderer);
861
+ rlt::render(device, *env.renderer);
862
+ {
863
+ constexpr size_t RENDER_PIXELS = (size_t)RENDER_WIDTH * CAM_HEIGHT;
864
+ std::vector<uint32_t> rgb_staging(RENDER_PIXELS);
865
+ {
866
+ rlt::Tensor<typename rlt::utils::typing::remove_reference_t<decltype(rlt::output(device, *env.renderer, rro::RGB<>{}))>::SPEC> frame_alias;
867
+ frame_alias._data = rgb_staging.data();
868
+ rlt::copy(env.renderer->device, device, rlt::output(device, *env.renderer, rro::RGB<>{}), frame_alias);
869
+ }
870
+ std::vector<uint32_t> view_staging(RENDER_PIXELS);
871
+ if (g_flow_view == FlowView::BACKWARD || g_flow_view == FlowView::FORWARD) {
872
+ std::vector<float> flow_staging(RENDER_PIXELS * 2);
873
+ rlt::Tensor<typename rlt::utils::typing::remove_reference_t<decltype(rlt::output(device, *env.renderer, rro::Flow<>{}))>::SPEC> flow_alias;
874
+ flow_alias._data = flow_staging.data();
875
+ if (g_flow_view == FlowView::BACKWARD) {
876
+ rlt::copy(env.renderer->device, device, rlt::output(device, *env.renderer, rro::Flow<>{}), flow_alias);
877
+ }
878
+ else {
879
+ rlt::copy(env.renderer->device, device, rlt::output(device, *env.renderer, rro::Flow<rro::direction::Forward>{}), flow_alias);
880
+ }
881
+ for (size_t pixel_i = 0; pixel_i < RENDER_PIXELS; pixel_i++) {
882
+ view_staging[pixel_i] = rlt::flow_color(device, flow_staging[pixel_i * 2 + 0], flow_staging[pixel_i * 2 + 1], FLOW_DISPLAY_SCALE_PX);
883
+ }
884
+ }
885
+ else {
886
+ std::vector<uint32_t> status_staging(RENDER_PIXELS);
887
+ rlt::Tensor<typename rlt::utils::typing::remove_reference_t<decltype(rlt::output(device, *env.renderer, rro::FlowStatus<>{}))>::SPEC> status_alias;
888
+ status_alias._data = status_staging.data();
889
+ if (g_flow_view == FlowView::BACKWARD_STATUS) {
890
+ rlt::copy(env.renderer->device, device, rlt::output(device, *env.renderer, rro::FlowStatus<>{}), status_alias);
891
+ }
892
+ else {
893
+ rlt::copy(env.renderer->device, device, rlt::output(device, *env.renderer, rro::FlowStatus<rro::direction::Forward>{}), status_alias);
894
+ }
895
+ for (size_t pixel_i = 0; pixel_i < RENDER_PIXELS; pixel_i++) {
896
+ view_staging[pixel_i] = rlt::flow_status_color(device, status_staging[pixel_i]);
897
+ }
898
+ }
899
+ for (size_t row = 0; row < CAM_HEIGHT; row++) {
900
+ std::memcpy(pixels.data() + row * CAM_WIDTH, rgb_staging.data() + row * RENDER_WIDTH, RENDER_WIDTH * sizeof(uint32_t));
901
+ std::memcpy(pixels.data() + row * CAM_WIDTH + RENDER_WIDTH, view_staging.data() + row * RENDER_WIDTH, RENDER_WIDTH * sizeof(uint32_t));
902
+ }
903
+ }
904
+ #elif RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_DEPTH
905
+ rlt::render(device, *env.renderer);
906
+ {
907
+ std::vector<float> depth_staging(pixels.size());
908
+ {
909
+ rlt::Tensor<typename rlt::utils::typing::remove_reference_t<decltype(rlt::output(device, *env.renderer, rro::Depth<>{}))>::SPEC> depth_alias;
910
+ depth_alias._data = depth_staging.data();
911
+ rlt::copy(env.renderer->device, device, rlt::output(device, *env.renderer, rro::Depth<>{}), depth_alias);
912
+ }
913
+ const float max_depth = env.renderer->max_ray_length > 0 ? env.renderer->max_ray_length : 1e30f;
914
+ depth_to_rgba(depth_staging.data(), pixels.data(), static_cast<int>(pixels.size()), max_depth);
915
+ }
916
+ #elif RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_RGBD
917
+ if (g_show_depth) {
918
+ rlt::render(device, *env.renderer);
919
+ std::vector<float> depth_staging(pixels.size());
920
+ {
921
+ rlt::Tensor<typename rlt::utils::typing::remove_reference_t<decltype(rlt::output(device, *env.renderer, rro::Depth<>{}))>::SPEC> depth_alias;
922
+ depth_alias._data = depth_staging.data();
923
+ rlt::copy(env.renderer->device, device, rlt::output(device, *env.renderer, rro::Depth<>{}), depth_alias);
924
+ }
925
+ const float max_depth = env.renderer->max_ray_length > 0 ? env.renderer->max_ray_length : 1e30f;
926
+ depth_to_rgba(depth_staging.data(), pixels.data(), static_cast<int>(pixels.size()), max_depth);
927
+ }
928
+ else {
929
+ rlt::render(device, *env.renderer);
930
+ {
931
+ rlt::Tensor<typename rlt::utils::typing::remove_reference_t<decltype(rlt::output(device, *env.renderer, rro::RGB<>{}))>::SPEC> frame_alias;
932
+ frame_alias._data = pixels.data();
933
+ rlt::copy(env.renderer->device, device, rlt::output(device, *env.renderer, rro::RGB<>{}), frame_alias);
934
+ }
935
+ }
936
+ #elif RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_SEGMENTATION
937
+ rlt::render(device, *env.renderer);
938
+ {
939
+ std::vector<uint32_t> segmentation_staging(pixels.size());
940
+ {
941
+ rlt::Tensor<typename rlt::utils::typing::remove_reference_t<decltype(rlt::output(device, *env.renderer, rro::Segmentation<>{}))>::SPEC> segmentation_alias;
942
+ segmentation_alias._data = segmentation_staging.data();
943
+ rlt::copy(env.renderer->device, device, rlt::output(device, *env.renderer, rro::Segmentation<>{}), segmentation_alias);
944
+ }
945
+ for (size_t pixel_i = 0; pixel_i < pixels.size(); pixel_i++) {
946
+ pixels[pixel_i] = rlt::segmentation_color(device, segmentation_staging[pixel_i]);
947
+ }
948
+ }
949
+ #else
950
+ rlt::render(device, *env.renderer);
951
+ {
952
+ rlt::Tensor<typename rlt::utils::typing::remove_reference_t<decltype(rlt::output(device, *env.renderer, rro::RGB<>{}))>::SPEC> frame_alias;
953
+ frame_alias._data = pixels.data();
954
+ rlt::copy(env.renderer->device, device, rlt::output(device, *env.renderer, rro::RGB<>{}), frame_alias);
955
+ }
956
+ #endif
957
+
958
+ {
959
+ const float forward[3] = {
960
+ look_at[0] - eye[0],
961
+ look_at[1] - eye[1],
962
+ look_at[2] - eye[2]
963
+ };
964
+ const Quaternion q = l2f_camera_quaternion(forward, up);
965
+ char line_flu_pos[128];
966
+ char line_flu_quat[128];
967
+ std::snprintf(line_flu_pos, sizeof(line_flu_pos), "Pos: (%.2f, %.2f, %.2f)", state.position[0], state.position[1], state.position[2]);
968
+ std::snprintf(line_flu_quat, sizeof(line_flu_quat), "Quat: (%.3f, %.3f, %.3f, %.3f)", q.w, q.x, q.y, q.z);
969
+ int overlay_x = 4;
970
+ int overlay_y = 4;
971
+ int max_len = std::max(std::strlen(line_flu_pos), std::strlen(line_flu_quat));
972
+ int text_width = static_cast<int>(max_len) * 6 + 4;
973
+ int text_height = 20;
974
+ #if RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_FLOW
975
+ const char* line_flow_view = flow_view_label(g_flow_view);
976
+ text_width = std::max(text_width, static_cast<int>(std::strlen(line_flow_view)) * 6 + 4);
977
+ text_height = 29;
978
+ #endif
979
+ draw_overlay_background(pixels.data(), CAM_WIDTH, CAM_HEIGHT, overlay_x, overlay_y, text_width, text_height);
980
+ draw_string(pixels.data(), CAM_WIDTH, CAM_HEIGHT, overlay_x + 2, overlay_y + 2, line_flu_pos, 0xFF00FF00);
981
+ draw_string(pixels.data(), CAM_WIDTH, CAM_HEIGHT, overlay_x + 2, overlay_y + 11, line_flu_quat, 0xFF00FF00);
982
+ #if RL_TOOLS_RENDERING_RAYTRACING_INTERACTIVE_OUTPUT_MODE == RL_TOOLS_RENDERING_RAYTRACING_OUTPUT_FLOW
983
+ draw_string(pixels.data(), CAM_WIDTH, CAM_HEIGHT, overlay_x + 2, overlay_y + 20, line_flow_view, 0xFF00FF00);
984
+ // label the right half as well so a screenshot of either half is self-describing
985
+ draw_overlay_background(pixels.data(), CAM_WIDTH, CAM_HEIGHT, RENDER_WIDTH + overlay_x, overlay_y, static_cast<int>(std::strlen(line_flow_view)) * 6 + 4, 11);
986
+ draw_string(pixels.data(), CAM_WIDTH, CAM_HEIGHT, RENDER_WIDTH + overlay_x + 2, overlay_y + 2, line_flow_view, 0xFF00FF00);
987
+ #endif
988
+ }
989
+
990
+ glBindTexture(GL_TEXTURE_2D, texture);
991
+ glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, CAM_WIDTH, CAM_HEIGHT, GL_RGBA, GL_UNSIGNED_BYTE, pixels.data());
992
+
993
+ glEnable(GL_TEXTURE_2D);
994
+ glBegin(GL_QUADS);
995
+ glTexCoord2f(0, 1); glVertex2f(-1, -1);
996
+ glTexCoord2f(1, 1); glVertex2f( 1, -1);
997
+ glTexCoord2f(1, 0); glVertex2f( 1, 1);
998
+ glTexCoord2f(0, 0); glVertex2f(-1, 1);
999
+ glEnd();
1000
+
1001
+ glfwSwapBuffers(window);
1002
+ }
1003
+
1004
+ save_camera_state(cam_cache, state.position[0], state.position[1], state.position[2], g_input.yaw, g_input.pitch);
1005
+ const bool pose_recorded = write_camera_trace_json(
1006
+ options.record_camera_pose_path,
1007
+ resolved_scene_path,
1008
+ camera_trace
1009
+ );
1010
+ if(pose_recorded && !options.record_camera_pose_path.empty()) {
1011
+ std::cout << "Wrote camera pose trace JSON to " << options.record_camera_pose_path << std::endl;
1012
+ }
1013
+
1014
+ glDeleteTextures(1, &texture);
1015
+ glfwDestroyWindow(window);
1016
+ glfwTerminate();
1017
+
1018
+ rlt::free(device, camera_staging);
1019
+ rlt::free(device, env);
1020
+ return pose_recorded ? 0 : 1;
1021
+ }