warp-lang 1.1.0__py3-none-macosx_10_13_universal2.whl → 1.2.1__py3-none-macosx_10_13_universal2.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.

Potentially problematic release.


This version of warp-lang might be problematic. Click here for more details.

Files changed (218) hide show
  1. warp/bin/libwarp-clang.dylib +0 -0
  2. warp/bin/libwarp.dylib +0 -0
  3. warp/build.py +10 -37
  4. warp/build_dll.py +2 -2
  5. warp/builtins.py +274 -6
  6. warp/codegen.py +51 -4
  7. warp/config.py +2 -2
  8. warp/constants.py +4 -0
  9. warp/context.py +422 -203
  10. warp/examples/benchmarks/benchmark_api.py +0 -2
  11. warp/examples/benchmarks/benchmark_cloth_warp.py +0 -1
  12. warp/examples/benchmarks/benchmark_launches.py +0 -2
  13. warp/examples/core/example_dem.py +0 -2
  14. warp/examples/core/example_fluid.py +0 -2
  15. warp/examples/core/example_graph_capture.py +0 -2
  16. warp/examples/core/example_marching_cubes.py +0 -2
  17. warp/examples/core/example_mesh.py +0 -2
  18. warp/examples/core/example_mesh_intersect.py +0 -2
  19. warp/examples/core/example_nvdb.py +0 -2
  20. warp/examples/core/example_raycast.py +0 -2
  21. warp/examples/core/example_raymarch.py +0 -2
  22. warp/examples/core/example_render_opengl.py +0 -2
  23. warp/examples/core/example_sph.py +0 -2
  24. warp/examples/core/example_torch.py +0 -3
  25. warp/examples/core/example_wave.py +0 -2
  26. warp/examples/fem/example_apic_fluid.py +140 -115
  27. warp/examples/fem/example_burgers.py +262 -0
  28. warp/examples/fem/example_convection_diffusion.py +0 -2
  29. warp/examples/fem/example_convection_diffusion_dg.py +0 -2
  30. warp/examples/fem/example_deformed_geometry.py +0 -2
  31. warp/examples/fem/example_diffusion.py +0 -2
  32. warp/examples/fem/example_diffusion_3d.py +5 -4
  33. warp/examples/fem/example_diffusion_mgpu.py +0 -2
  34. warp/examples/fem/example_mixed_elasticity.py +0 -2
  35. warp/examples/fem/example_navier_stokes.py +0 -2
  36. warp/examples/fem/example_stokes.py +0 -2
  37. warp/examples/fem/example_stokes_transfer.py +0 -2
  38. warp/examples/optim/example_bounce.py +0 -2
  39. warp/examples/optim/example_cloth_throw.py +0 -2
  40. warp/examples/optim/example_diffray.py +0 -2
  41. warp/examples/optim/example_drone.py +0 -2
  42. warp/examples/optim/example_inverse_kinematics.py +0 -2
  43. warp/examples/optim/example_inverse_kinematics_torch.py +0 -2
  44. warp/examples/optim/example_spring_cage.py +0 -2
  45. warp/examples/optim/example_trajectory.py +0 -2
  46. warp/examples/optim/example_walker.py +0 -2
  47. warp/examples/sim/example_cartpole.py +0 -2
  48. warp/examples/sim/example_cloth.py +0 -2
  49. warp/examples/sim/example_granular.py +0 -2
  50. warp/examples/sim/example_granular_collision_sdf.py +0 -2
  51. warp/examples/sim/example_jacobian_ik.py +0 -2
  52. warp/examples/sim/example_particle_chain.py +0 -2
  53. warp/examples/sim/example_quadruped.py +0 -2
  54. warp/examples/sim/example_rigid_chain.py +0 -2
  55. warp/examples/sim/example_rigid_contact.py +0 -2
  56. warp/examples/sim/example_rigid_force.py +0 -2
  57. warp/examples/sim/example_rigid_gyroscopic.py +0 -2
  58. warp/examples/sim/example_rigid_soft_contact.py +0 -2
  59. warp/examples/sim/example_soft_body.py +0 -2
  60. warp/fem/__init__.py +1 -0
  61. warp/fem/cache.py +3 -1
  62. warp/fem/geometry/__init__.py +1 -0
  63. warp/fem/geometry/element.py +4 -0
  64. warp/fem/geometry/grid_3d.py +0 -4
  65. warp/fem/geometry/nanogrid.py +455 -0
  66. warp/fem/integrate.py +63 -9
  67. warp/fem/space/__init__.py +43 -158
  68. warp/fem/space/basis_space.py +34 -0
  69. warp/fem/space/collocated_function_space.py +1 -1
  70. warp/fem/space/grid_2d_function_space.py +13 -132
  71. warp/fem/space/grid_3d_function_space.py +16 -154
  72. warp/fem/space/hexmesh_function_space.py +37 -134
  73. warp/fem/space/nanogrid_function_space.py +202 -0
  74. warp/fem/space/quadmesh_2d_function_space.py +12 -119
  75. warp/fem/space/restriction.py +4 -1
  76. warp/fem/space/shape/__init__.py +77 -0
  77. warp/fem/space/shape/cube_shape_function.py +5 -15
  78. warp/fem/space/tetmesh_function_space.py +6 -76
  79. warp/fem/space/trimesh_2d_function_space.py +6 -76
  80. warp/native/array.h +12 -3
  81. warp/native/builtin.h +48 -5
  82. warp/native/bvh.cpp +14 -10
  83. warp/native/bvh.cu +23 -15
  84. warp/native/bvh.h +1 -0
  85. warp/native/clang/clang.cpp +2 -1
  86. warp/native/crt.cpp +11 -1
  87. warp/native/crt.h +18 -1
  88. warp/native/exports.h +187 -0
  89. warp/native/mat.h +47 -0
  90. warp/native/mesh.cpp +1 -1
  91. warp/native/mesh.cu +1 -2
  92. warp/native/nanovdb/GridHandle.h +366 -0
  93. warp/native/nanovdb/HostBuffer.h +590 -0
  94. warp/native/nanovdb/NanoVDB.h +3999 -2157
  95. warp/native/nanovdb/PNanoVDB.h +936 -99
  96. warp/native/quat.h +28 -1
  97. warp/native/rand.h +5 -1
  98. warp/native/vec.h +45 -1
  99. warp/native/volume.cpp +335 -103
  100. warp/native/volume.cu +39 -13
  101. warp/native/volume.h +725 -303
  102. warp/native/volume_builder.cu +381 -360
  103. warp/native/volume_builder.h +16 -1
  104. warp/native/volume_impl.h +61 -0
  105. warp/native/warp.cu +8 -2
  106. warp/native/warp.h +15 -7
  107. warp/render/render_opengl.py +191 -52
  108. warp/sim/integrator_featherstone.py +10 -3
  109. warp/sim/integrator_xpbd.py +16 -22
  110. warp/sparse.py +89 -27
  111. warp/stubs.py +83 -0
  112. warp/tests/assets/test_index_grid.nvdb +0 -0
  113. warp/tests/aux_test_dependent.py +0 -2
  114. warp/tests/aux_test_grad_customs.py +0 -2
  115. warp/tests/aux_test_reference.py +0 -2
  116. warp/tests/aux_test_reference_reference.py +0 -2
  117. warp/tests/aux_test_square.py +0 -2
  118. warp/tests/disabled_kinematics.py +0 -2
  119. warp/tests/test_adam.py +0 -2
  120. warp/tests/test_arithmetic.py +0 -36
  121. warp/tests/test_array.py +9 -11
  122. warp/tests/test_array_reduce.py +0 -2
  123. warp/tests/test_async.py +0 -2
  124. warp/tests/test_atomic.py +0 -2
  125. warp/tests/test_bool.py +58 -50
  126. warp/tests/test_builtins_resolution.py +0 -2
  127. warp/tests/test_bvh.py +0 -2
  128. warp/tests/test_closest_point_edge_edge.py +0 -1
  129. warp/tests/test_codegen.py +0 -4
  130. warp/tests/test_compile_consts.py +130 -10
  131. warp/tests/test_conditional.py +0 -2
  132. warp/tests/test_copy.py +0 -2
  133. warp/tests/test_ctypes.py +6 -8
  134. warp/tests/test_dense.py +0 -2
  135. warp/tests/test_devices.py +0 -2
  136. warp/tests/test_dlpack.py +9 -11
  137. warp/tests/test_examples.py +42 -39
  138. warp/tests/test_fabricarray.py +0 -3
  139. warp/tests/test_fast_math.py +0 -2
  140. warp/tests/test_fem.py +75 -54
  141. warp/tests/test_fp16.py +0 -2
  142. warp/tests/test_func.py +0 -2
  143. warp/tests/test_generics.py +27 -2
  144. warp/tests/test_grad.py +147 -8
  145. warp/tests/test_grad_customs.py +0 -2
  146. warp/tests/test_hash_grid.py +1 -3
  147. warp/tests/test_import.py +0 -2
  148. warp/tests/test_indexedarray.py +0 -2
  149. warp/tests/test_intersect.py +0 -2
  150. warp/tests/test_jax.py +0 -2
  151. warp/tests/test_large.py +11 -9
  152. warp/tests/test_launch.py +0 -2
  153. warp/tests/test_lerp.py +10 -54
  154. warp/tests/test_linear_solvers.py +3 -5
  155. warp/tests/test_lvalue.py +0 -2
  156. warp/tests/test_marching_cubes.py +0 -2
  157. warp/tests/test_mat.py +0 -2
  158. warp/tests/test_mat_lite.py +0 -2
  159. warp/tests/test_mat_scalar_ops.py +0 -2
  160. warp/tests/test_math.py +0 -2
  161. warp/tests/test_matmul.py +35 -37
  162. warp/tests/test_matmul_lite.py +29 -31
  163. warp/tests/test_mempool.py +0 -2
  164. warp/tests/test_mesh.py +0 -3
  165. warp/tests/test_mesh_query_aabb.py +0 -2
  166. warp/tests/test_mesh_query_point.py +0 -2
  167. warp/tests/test_mesh_query_ray.py +0 -2
  168. warp/tests/test_mlp.py +0 -2
  169. warp/tests/test_model.py +0 -2
  170. warp/tests/test_module_hashing.py +111 -0
  171. warp/tests/test_modules_lite.py +0 -3
  172. warp/tests/test_multigpu.py +0 -2
  173. warp/tests/test_noise.py +0 -4
  174. warp/tests/test_operators.py +0 -2
  175. warp/tests/test_options.py +0 -2
  176. warp/tests/test_peer.py +0 -2
  177. warp/tests/test_pinned.py +0 -2
  178. warp/tests/test_print.py +0 -2
  179. warp/tests/test_quat.py +0 -2
  180. warp/tests/test_rand.py +41 -5
  181. warp/tests/test_reload.py +0 -10
  182. warp/tests/test_rounding.py +0 -2
  183. warp/tests/test_runlength_encode.py +0 -2
  184. warp/tests/test_sim_grad.py +0 -2
  185. warp/tests/test_sim_kinematics.py +0 -2
  186. warp/tests/test_smoothstep.py +0 -2
  187. warp/tests/test_snippet.py +0 -2
  188. warp/tests/test_sparse.py +0 -2
  189. warp/tests/test_spatial.py +0 -2
  190. warp/tests/test_special_values.py +362 -0
  191. warp/tests/test_streams.py +0 -2
  192. warp/tests/test_struct.py +0 -2
  193. warp/tests/test_tape.py +0 -2
  194. warp/tests/test_torch.py +0 -2
  195. warp/tests/test_transient_module.py +0 -2
  196. warp/tests/test_types.py +0 -2
  197. warp/tests/test_utils.py +0 -2
  198. warp/tests/test_vec.py +0 -2
  199. warp/tests/test_vec_lite.py +0 -2
  200. warp/tests/test_vec_scalar_ops.py +0 -2
  201. warp/tests/test_verify_fp.py +0 -2
  202. warp/tests/test_volume.py +237 -13
  203. warp/tests/test_volume_write.py +86 -3
  204. warp/tests/unittest_serial.py +10 -9
  205. warp/tests/unittest_suites.py +6 -2
  206. warp/tests/unittest_utils.py +2 -171
  207. warp/tests/unused_test_misc.py +0 -2
  208. warp/tests/walkthrough_debug.py +1 -1
  209. warp/thirdparty/unittest_parallel.py +37 -40
  210. warp/types.py +526 -85
  211. {warp_lang-1.1.0.dist-info → warp_lang-1.2.1.dist-info}/METADATA +61 -31
  212. warp_lang-1.2.1.dist-info/RECORD +359 -0
  213. warp/examples/fem/example_convection_diffusion_dg0.py +0 -204
  214. warp/native/nanovdb/PNanoVDBWrite.h +0 -295
  215. warp_lang-1.1.0.dist-info/RECORD +0 -352
  216. {warp_lang-1.1.0.dist-info → warp_lang-1.2.1.dist-info}/LICENSE.md +0 -0
  217. {warp_lang-1.1.0.dist-info → warp_lang-1.2.1.dist-info}/WHEEL +0 -0
  218. {warp_lang-1.1.0.dist-info → warp_lang-1.2.1.dist-info}/top_level.txt +0 -0
warp/native/exports.h CHANGED
@@ -298,6 +298,192 @@ WP_API void builtin_ceil_float64(float64 x, float64* ret) { *ret = wp::ceil(x);
298
298
  WP_API void builtin_frac_float16(float16 x, float16* ret) { *ret = wp::frac(x); }
299
299
  WP_API void builtin_frac_float32(float32 x, float32* ret) { *ret = wp::frac(x); }
300
300
  WP_API void builtin_frac_float64(float64 x, float64* ret) { *ret = wp::frac(x); }
301
+ WP_API void builtin_isfinite_float16(float16 x, bool* ret) { *ret = wp::isfinite(x); }
302
+ WP_API void builtin_isfinite_float32(float32 x, bool* ret) { *ret = wp::isfinite(x); }
303
+ WP_API void builtin_isfinite_float64(float64 x, bool* ret) { *ret = wp::isfinite(x); }
304
+ WP_API void builtin_isfinite_int16(int16 x, bool* ret) { *ret = wp::isfinite(x); }
305
+ WP_API void builtin_isfinite_int32(int32 x, bool* ret) { *ret = wp::isfinite(x); }
306
+ WP_API void builtin_isfinite_int64(int64 x, bool* ret) { *ret = wp::isfinite(x); }
307
+ WP_API void builtin_isfinite_int8(int8 x, bool* ret) { *ret = wp::isfinite(x); }
308
+ WP_API void builtin_isfinite_uint16(uint16 x, bool* ret) { *ret = wp::isfinite(x); }
309
+ WP_API void builtin_isfinite_uint32(uint32 x, bool* ret) { *ret = wp::isfinite(x); }
310
+ WP_API void builtin_isfinite_uint64(uint64 x, bool* ret) { *ret = wp::isfinite(x); }
311
+ WP_API void builtin_isfinite_uint8(uint8 x, bool* ret) { *ret = wp::isfinite(x); }
312
+ WP_API void builtin_isfinite_vec2h(vec2h& x, bool* ret) { *ret = wp::isfinite(x); }
313
+ WP_API void builtin_isfinite_vec3h(vec3h& x, bool* ret) { *ret = wp::isfinite(x); }
314
+ WP_API void builtin_isfinite_vec4h(vec4h& x, bool* ret) { *ret = wp::isfinite(x); }
315
+ WP_API void builtin_isfinite_spatial_vectorh(spatial_vectorh& x, bool* ret) { *ret = wp::isfinite(x); }
316
+ WP_API void builtin_isfinite_vec2f(vec2f& x, bool* ret) { *ret = wp::isfinite(x); }
317
+ WP_API void builtin_isfinite_vec3f(vec3f& x, bool* ret) { *ret = wp::isfinite(x); }
318
+ WP_API void builtin_isfinite_vec4f(vec4f& x, bool* ret) { *ret = wp::isfinite(x); }
319
+ WP_API void builtin_isfinite_spatial_vectorf(spatial_vectorf& x, bool* ret) { *ret = wp::isfinite(x); }
320
+ WP_API void builtin_isfinite_vec2d(vec2d& x, bool* ret) { *ret = wp::isfinite(x); }
321
+ WP_API void builtin_isfinite_vec3d(vec3d& x, bool* ret) { *ret = wp::isfinite(x); }
322
+ WP_API void builtin_isfinite_vec4d(vec4d& x, bool* ret) { *ret = wp::isfinite(x); }
323
+ WP_API void builtin_isfinite_spatial_vectord(spatial_vectord& x, bool* ret) { *ret = wp::isfinite(x); }
324
+ WP_API void builtin_isfinite_vec2s(vec2s& x, bool* ret) { *ret = wp::isfinite(x); }
325
+ WP_API void builtin_isfinite_vec3s(vec3s& x, bool* ret) { *ret = wp::isfinite(x); }
326
+ WP_API void builtin_isfinite_vec4s(vec4s& x, bool* ret) { *ret = wp::isfinite(x); }
327
+ WP_API void builtin_isfinite_vec2i(vec2i& x, bool* ret) { *ret = wp::isfinite(x); }
328
+ WP_API void builtin_isfinite_vec3i(vec3i& x, bool* ret) { *ret = wp::isfinite(x); }
329
+ WP_API void builtin_isfinite_vec4i(vec4i& x, bool* ret) { *ret = wp::isfinite(x); }
330
+ WP_API void builtin_isfinite_vec2l(vec2l& x, bool* ret) { *ret = wp::isfinite(x); }
331
+ WP_API void builtin_isfinite_vec3l(vec3l& x, bool* ret) { *ret = wp::isfinite(x); }
332
+ WP_API void builtin_isfinite_vec4l(vec4l& x, bool* ret) { *ret = wp::isfinite(x); }
333
+ WP_API void builtin_isfinite_vec2b(vec2b& x, bool* ret) { *ret = wp::isfinite(x); }
334
+ WP_API void builtin_isfinite_vec3b(vec3b& x, bool* ret) { *ret = wp::isfinite(x); }
335
+ WP_API void builtin_isfinite_vec4b(vec4b& x, bool* ret) { *ret = wp::isfinite(x); }
336
+ WP_API void builtin_isfinite_vec2us(vec2us& x, bool* ret) { *ret = wp::isfinite(x); }
337
+ WP_API void builtin_isfinite_vec3us(vec3us& x, bool* ret) { *ret = wp::isfinite(x); }
338
+ WP_API void builtin_isfinite_vec4us(vec4us& x, bool* ret) { *ret = wp::isfinite(x); }
339
+ WP_API void builtin_isfinite_vec2ui(vec2ui& x, bool* ret) { *ret = wp::isfinite(x); }
340
+ WP_API void builtin_isfinite_vec3ui(vec3ui& x, bool* ret) { *ret = wp::isfinite(x); }
341
+ WP_API void builtin_isfinite_vec4ui(vec4ui& x, bool* ret) { *ret = wp::isfinite(x); }
342
+ WP_API void builtin_isfinite_vec2ul(vec2ul& x, bool* ret) { *ret = wp::isfinite(x); }
343
+ WP_API void builtin_isfinite_vec3ul(vec3ul& x, bool* ret) { *ret = wp::isfinite(x); }
344
+ WP_API void builtin_isfinite_vec4ul(vec4ul& x, bool* ret) { *ret = wp::isfinite(x); }
345
+ WP_API void builtin_isfinite_vec2ub(vec2ub& x, bool* ret) { *ret = wp::isfinite(x); }
346
+ WP_API void builtin_isfinite_vec3ub(vec3ub& x, bool* ret) { *ret = wp::isfinite(x); }
347
+ WP_API void builtin_isfinite_vec4ub(vec4ub& x, bool* ret) { *ret = wp::isfinite(x); }
348
+ WP_API void builtin_isfinite_quath(quath& x, bool* ret) { *ret = wp::isfinite(x); }
349
+ WP_API void builtin_isfinite_quatf(quatf& x, bool* ret) { *ret = wp::isfinite(x); }
350
+ WP_API void builtin_isfinite_quatd(quatd& x, bool* ret) { *ret = wp::isfinite(x); }
351
+ WP_API void builtin_isfinite_mat22h(mat22h& m, bool* ret) { *ret = wp::isfinite(m); }
352
+ WP_API void builtin_isfinite_mat33h(mat33h& m, bool* ret) { *ret = wp::isfinite(m); }
353
+ WP_API void builtin_isfinite_mat44h(mat44h& m, bool* ret) { *ret = wp::isfinite(m); }
354
+ WP_API void builtin_isfinite_spatial_matrixh(spatial_matrixh& m, bool* ret) { *ret = wp::isfinite(m); }
355
+ WP_API void builtin_isfinite_mat22f(mat22f& m, bool* ret) { *ret = wp::isfinite(m); }
356
+ WP_API void builtin_isfinite_mat33f(mat33f& m, bool* ret) { *ret = wp::isfinite(m); }
357
+ WP_API void builtin_isfinite_mat44f(mat44f& m, bool* ret) { *ret = wp::isfinite(m); }
358
+ WP_API void builtin_isfinite_spatial_matrixf(spatial_matrixf& m, bool* ret) { *ret = wp::isfinite(m); }
359
+ WP_API void builtin_isfinite_mat22d(mat22d& m, bool* ret) { *ret = wp::isfinite(m); }
360
+ WP_API void builtin_isfinite_mat33d(mat33d& m, bool* ret) { *ret = wp::isfinite(m); }
361
+ WP_API void builtin_isfinite_mat44d(mat44d& m, bool* ret) { *ret = wp::isfinite(m); }
362
+ WP_API void builtin_isfinite_spatial_matrixd(spatial_matrixd& m, bool* ret) { *ret = wp::isfinite(m); }
363
+ WP_API void builtin_isnan_float16(float16 x, bool* ret) { *ret = wp::isnan(x); }
364
+ WP_API void builtin_isnan_float32(float32 x, bool* ret) { *ret = wp::isnan(x); }
365
+ WP_API void builtin_isnan_float64(float64 x, bool* ret) { *ret = wp::isnan(x); }
366
+ WP_API void builtin_isnan_int16(int16 x, bool* ret) { *ret = wp::isnan(x); }
367
+ WP_API void builtin_isnan_int32(int32 x, bool* ret) { *ret = wp::isnan(x); }
368
+ WP_API void builtin_isnan_int64(int64 x, bool* ret) { *ret = wp::isnan(x); }
369
+ WP_API void builtin_isnan_int8(int8 x, bool* ret) { *ret = wp::isnan(x); }
370
+ WP_API void builtin_isnan_uint16(uint16 x, bool* ret) { *ret = wp::isnan(x); }
371
+ WP_API void builtin_isnan_uint32(uint32 x, bool* ret) { *ret = wp::isnan(x); }
372
+ WP_API void builtin_isnan_uint64(uint64 x, bool* ret) { *ret = wp::isnan(x); }
373
+ WP_API void builtin_isnan_uint8(uint8 x, bool* ret) { *ret = wp::isnan(x); }
374
+ WP_API void builtin_isnan_vec2h(vec2h& x, bool* ret) { *ret = wp::isnan(x); }
375
+ WP_API void builtin_isnan_vec3h(vec3h& x, bool* ret) { *ret = wp::isnan(x); }
376
+ WP_API void builtin_isnan_vec4h(vec4h& x, bool* ret) { *ret = wp::isnan(x); }
377
+ WP_API void builtin_isnan_spatial_vectorh(spatial_vectorh& x, bool* ret) { *ret = wp::isnan(x); }
378
+ WP_API void builtin_isnan_vec2f(vec2f& x, bool* ret) { *ret = wp::isnan(x); }
379
+ WP_API void builtin_isnan_vec3f(vec3f& x, bool* ret) { *ret = wp::isnan(x); }
380
+ WP_API void builtin_isnan_vec4f(vec4f& x, bool* ret) { *ret = wp::isnan(x); }
381
+ WP_API void builtin_isnan_spatial_vectorf(spatial_vectorf& x, bool* ret) { *ret = wp::isnan(x); }
382
+ WP_API void builtin_isnan_vec2d(vec2d& x, bool* ret) { *ret = wp::isnan(x); }
383
+ WP_API void builtin_isnan_vec3d(vec3d& x, bool* ret) { *ret = wp::isnan(x); }
384
+ WP_API void builtin_isnan_vec4d(vec4d& x, bool* ret) { *ret = wp::isnan(x); }
385
+ WP_API void builtin_isnan_spatial_vectord(spatial_vectord& x, bool* ret) { *ret = wp::isnan(x); }
386
+ WP_API void builtin_isnan_vec2s(vec2s& x, bool* ret) { *ret = wp::isnan(x); }
387
+ WP_API void builtin_isnan_vec3s(vec3s& x, bool* ret) { *ret = wp::isnan(x); }
388
+ WP_API void builtin_isnan_vec4s(vec4s& x, bool* ret) { *ret = wp::isnan(x); }
389
+ WP_API void builtin_isnan_vec2i(vec2i& x, bool* ret) { *ret = wp::isnan(x); }
390
+ WP_API void builtin_isnan_vec3i(vec3i& x, bool* ret) { *ret = wp::isnan(x); }
391
+ WP_API void builtin_isnan_vec4i(vec4i& x, bool* ret) { *ret = wp::isnan(x); }
392
+ WP_API void builtin_isnan_vec2l(vec2l& x, bool* ret) { *ret = wp::isnan(x); }
393
+ WP_API void builtin_isnan_vec3l(vec3l& x, bool* ret) { *ret = wp::isnan(x); }
394
+ WP_API void builtin_isnan_vec4l(vec4l& x, bool* ret) { *ret = wp::isnan(x); }
395
+ WP_API void builtin_isnan_vec2b(vec2b& x, bool* ret) { *ret = wp::isnan(x); }
396
+ WP_API void builtin_isnan_vec3b(vec3b& x, bool* ret) { *ret = wp::isnan(x); }
397
+ WP_API void builtin_isnan_vec4b(vec4b& x, bool* ret) { *ret = wp::isnan(x); }
398
+ WP_API void builtin_isnan_vec2us(vec2us& x, bool* ret) { *ret = wp::isnan(x); }
399
+ WP_API void builtin_isnan_vec3us(vec3us& x, bool* ret) { *ret = wp::isnan(x); }
400
+ WP_API void builtin_isnan_vec4us(vec4us& x, bool* ret) { *ret = wp::isnan(x); }
401
+ WP_API void builtin_isnan_vec2ui(vec2ui& x, bool* ret) { *ret = wp::isnan(x); }
402
+ WP_API void builtin_isnan_vec3ui(vec3ui& x, bool* ret) { *ret = wp::isnan(x); }
403
+ WP_API void builtin_isnan_vec4ui(vec4ui& x, bool* ret) { *ret = wp::isnan(x); }
404
+ WP_API void builtin_isnan_vec2ul(vec2ul& x, bool* ret) { *ret = wp::isnan(x); }
405
+ WP_API void builtin_isnan_vec3ul(vec3ul& x, bool* ret) { *ret = wp::isnan(x); }
406
+ WP_API void builtin_isnan_vec4ul(vec4ul& x, bool* ret) { *ret = wp::isnan(x); }
407
+ WP_API void builtin_isnan_vec2ub(vec2ub& x, bool* ret) { *ret = wp::isnan(x); }
408
+ WP_API void builtin_isnan_vec3ub(vec3ub& x, bool* ret) { *ret = wp::isnan(x); }
409
+ WP_API void builtin_isnan_vec4ub(vec4ub& x, bool* ret) { *ret = wp::isnan(x); }
410
+ WP_API void builtin_isnan_quath(quath& x, bool* ret) { *ret = wp::isnan(x); }
411
+ WP_API void builtin_isnan_quatf(quatf& x, bool* ret) { *ret = wp::isnan(x); }
412
+ WP_API void builtin_isnan_quatd(quatd& x, bool* ret) { *ret = wp::isnan(x); }
413
+ WP_API void builtin_isnan_mat22h(mat22h& m, bool* ret) { *ret = wp::isnan(m); }
414
+ WP_API void builtin_isnan_mat33h(mat33h& m, bool* ret) { *ret = wp::isnan(m); }
415
+ WP_API void builtin_isnan_mat44h(mat44h& m, bool* ret) { *ret = wp::isnan(m); }
416
+ WP_API void builtin_isnan_spatial_matrixh(spatial_matrixh& m, bool* ret) { *ret = wp::isnan(m); }
417
+ WP_API void builtin_isnan_mat22f(mat22f& m, bool* ret) { *ret = wp::isnan(m); }
418
+ WP_API void builtin_isnan_mat33f(mat33f& m, bool* ret) { *ret = wp::isnan(m); }
419
+ WP_API void builtin_isnan_mat44f(mat44f& m, bool* ret) { *ret = wp::isnan(m); }
420
+ WP_API void builtin_isnan_spatial_matrixf(spatial_matrixf& m, bool* ret) { *ret = wp::isnan(m); }
421
+ WP_API void builtin_isnan_mat22d(mat22d& m, bool* ret) { *ret = wp::isnan(m); }
422
+ WP_API void builtin_isnan_mat33d(mat33d& m, bool* ret) { *ret = wp::isnan(m); }
423
+ WP_API void builtin_isnan_mat44d(mat44d& m, bool* ret) { *ret = wp::isnan(m); }
424
+ WP_API void builtin_isnan_spatial_matrixd(spatial_matrixd& m, bool* ret) { *ret = wp::isnan(m); }
425
+ WP_API void builtin_isinf_float16(float16 x, bool* ret) { *ret = wp::isinf(x); }
426
+ WP_API void builtin_isinf_float32(float32 x, bool* ret) { *ret = wp::isinf(x); }
427
+ WP_API void builtin_isinf_float64(float64 x, bool* ret) { *ret = wp::isinf(x); }
428
+ WP_API void builtin_isinf_int16(int16 x, bool* ret) { *ret = wp::isinf(x); }
429
+ WP_API void builtin_isinf_int32(int32 x, bool* ret) { *ret = wp::isinf(x); }
430
+ WP_API void builtin_isinf_int64(int64 x, bool* ret) { *ret = wp::isinf(x); }
431
+ WP_API void builtin_isinf_int8(int8 x, bool* ret) { *ret = wp::isinf(x); }
432
+ WP_API void builtin_isinf_uint16(uint16 x, bool* ret) { *ret = wp::isinf(x); }
433
+ WP_API void builtin_isinf_uint32(uint32 x, bool* ret) { *ret = wp::isinf(x); }
434
+ WP_API void builtin_isinf_uint64(uint64 x, bool* ret) { *ret = wp::isinf(x); }
435
+ WP_API void builtin_isinf_uint8(uint8 x, bool* ret) { *ret = wp::isinf(x); }
436
+ WP_API void builtin_isinf_vec2h(vec2h& x, bool* ret) { *ret = wp::isinf(x); }
437
+ WP_API void builtin_isinf_vec3h(vec3h& x, bool* ret) { *ret = wp::isinf(x); }
438
+ WP_API void builtin_isinf_vec4h(vec4h& x, bool* ret) { *ret = wp::isinf(x); }
439
+ WP_API void builtin_isinf_spatial_vectorh(spatial_vectorh& x, bool* ret) { *ret = wp::isinf(x); }
440
+ WP_API void builtin_isinf_vec2f(vec2f& x, bool* ret) { *ret = wp::isinf(x); }
441
+ WP_API void builtin_isinf_vec3f(vec3f& x, bool* ret) { *ret = wp::isinf(x); }
442
+ WP_API void builtin_isinf_vec4f(vec4f& x, bool* ret) { *ret = wp::isinf(x); }
443
+ WP_API void builtin_isinf_spatial_vectorf(spatial_vectorf& x, bool* ret) { *ret = wp::isinf(x); }
444
+ WP_API void builtin_isinf_vec2d(vec2d& x, bool* ret) { *ret = wp::isinf(x); }
445
+ WP_API void builtin_isinf_vec3d(vec3d& x, bool* ret) { *ret = wp::isinf(x); }
446
+ WP_API void builtin_isinf_vec4d(vec4d& x, bool* ret) { *ret = wp::isinf(x); }
447
+ WP_API void builtin_isinf_spatial_vectord(spatial_vectord& x, bool* ret) { *ret = wp::isinf(x); }
448
+ WP_API void builtin_isinf_vec2s(vec2s& x, bool* ret) { *ret = wp::isinf(x); }
449
+ WP_API void builtin_isinf_vec3s(vec3s& x, bool* ret) { *ret = wp::isinf(x); }
450
+ WP_API void builtin_isinf_vec4s(vec4s& x, bool* ret) { *ret = wp::isinf(x); }
451
+ WP_API void builtin_isinf_vec2i(vec2i& x, bool* ret) { *ret = wp::isinf(x); }
452
+ WP_API void builtin_isinf_vec3i(vec3i& x, bool* ret) { *ret = wp::isinf(x); }
453
+ WP_API void builtin_isinf_vec4i(vec4i& x, bool* ret) { *ret = wp::isinf(x); }
454
+ WP_API void builtin_isinf_vec2l(vec2l& x, bool* ret) { *ret = wp::isinf(x); }
455
+ WP_API void builtin_isinf_vec3l(vec3l& x, bool* ret) { *ret = wp::isinf(x); }
456
+ WP_API void builtin_isinf_vec4l(vec4l& x, bool* ret) { *ret = wp::isinf(x); }
457
+ WP_API void builtin_isinf_vec2b(vec2b& x, bool* ret) { *ret = wp::isinf(x); }
458
+ WP_API void builtin_isinf_vec3b(vec3b& x, bool* ret) { *ret = wp::isinf(x); }
459
+ WP_API void builtin_isinf_vec4b(vec4b& x, bool* ret) { *ret = wp::isinf(x); }
460
+ WP_API void builtin_isinf_vec2us(vec2us& x, bool* ret) { *ret = wp::isinf(x); }
461
+ WP_API void builtin_isinf_vec3us(vec3us& x, bool* ret) { *ret = wp::isinf(x); }
462
+ WP_API void builtin_isinf_vec4us(vec4us& x, bool* ret) { *ret = wp::isinf(x); }
463
+ WP_API void builtin_isinf_vec2ui(vec2ui& x, bool* ret) { *ret = wp::isinf(x); }
464
+ WP_API void builtin_isinf_vec3ui(vec3ui& x, bool* ret) { *ret = wp::isinf(x); }
465
+ WP_API void builtin_isinf_vec4ui(vec4ui& x, bool* ret) { *ret = wp::isinf(x); }
466
+ WP_API void builtin_isinf_vec2ul(vec2ul& x, bool* ret) { *ret = wp::isinf(x); }
467
+ WP_API void builtin_isinf_vec3ul(vec3ul& x, bool* ret) { *ret = wp::isinf(x); }
468
+ WP_API void builtin_isinf_vec4ul(vec4ul& x, bool* ret) { *ret = wp::isinf(x); }
469
+ WP_API void builtin_isinf_vec2ub(vec2ub& x, bool* ret) { *ret = wp::isinf(x); }
470
+ WP_API void builtin_isinf_vec3ub(vec3ub& x, bool* ret) { *ret = wp::isinf(x); }
471
+ WP_API void builtin_isinf_vec4ub(vec4ub& x, bool* ret) { *ret = wp::isinf(x); }
472
+ WP_API void builtin_isinf_quath(quath& x, bool* ret) { *ret = wp::isinf(x); }
473
+ WP_API void builtin_isinf_quatf(quatf& x, bool* ret) { *ret = wp::isinf(x); }
474
+ WP_API void builtin_isinf_quatd(quatd& x, bool* ret) { *ret = wp::isinf(x); }
475
+ WP_API void builtin_isinf_mat22h(mat22h& m, bool* ret) { *ret = wp::isinf(m); }
476
+ WP_API void builtin_isinf_mat33h(mat33h& m, bool* ret) { *ret = wp::isinf(m); }
477
+ WP_API void builtin_isinf_mat44h(mat44h& m, bool* ret) { *ret = wp::isinf(m); }
478
+ WP_API void builtin_isinf_spatial_matrixh(spatial_matrixh& m, bool* ret) { *ret = wp::isinf(m); }
479
+ WP_API void builtin_isinf_mat22f(mat22f& m, bool* ret) { *ret = wp::isinf(m); }
480
+ WP_API void builtin_isinf_mat33f(mat33f& m, bool* ret) { *ret = wp::isinf(m); }
481
+ WP_API void builtin_isinf_mat44f(mat44f& m, bool* ret) { *ret = wp::isinf(m); }
482
+ WP_API void builtin_isinf_spatial_matrixf(spatial_matrixf& m, bool* ret) { *ret = wp::isinf(m); }
483
+ WP_API void builtin_isinf_mat22d(mat22d& m, bool* ret) { *ret = wp::isinf(m); }
484
+ WP_API void builtin_isinf_mat33d(mat33d& m, bool* ret) { *ret = wp::isinf(m); }
485
+ WP_API void builtin_isinf_mat44d(mat44d& m, bool* ret) { *ret = wp::isinf(m); }
486
+ WP_API void builtin_isinf_spatial_matrixd(spatial_matrixd& m, bool* ret) { *ret = wp::isinf(m); }
301
487
  WP_API void builtin_dot_vec2h_vec2h(vec2h& x, vec2h& y, float16* ret) { *ret = wp::dot(x, y); }
302
488
  WP_API void builtin_dot_vec3h_vec3h(vec3h& x, vec3h& y, float16* ret) { *ret = wp::dot(x, y); }
303
489
  WP_API void builtin_dot_vec4h_vec4h(vec4h& x, vec4h& y, float16* ret) { *ret = wp::dot(x, y); }
@@ -782,6 +968,7 @@ WP_API void builtin_volume_sample_v_uint64_vec3f_int32(uint64 id, vec3f& uvw, in
782
968
  WP_API void builtin_volume_lookup_v_uint64_int32_int32_int32(uint64 id, int32 i, int32 j, int32 k, vec3f* ret) { *ret = wp::volume_lookup_v(id, i, j, k); }
783
969
  WP_API void builtin_volume_sample_i_uint64_vec3f(uint64 id, vec3f& uvw, int* ret) { *ret = wp::volume_sample_i(id, uvw); }
784
970
  WP_API void builtin_volume_lookup_i_uint64_int32_int32_int32(uint64 id, int32 i, int32 j, int32 k, int* ret) { *ret = wp::volume_lookup_i(id, i, j, k); }
971
+ WP_API void builtin_volume_lookup_index_uint64_int32_int32_int32(uint64 id, int32 i, int32 j, int32 k, int32* ret) { *ret = wp::volume_lookup_index(id, i, j, k); }
785
972
  WP_API void builtin_volume_index_to_world_uint64_vec3f(uint64 id, vec3f& uvw, vec3f* ret) { *ret = wp::volume_index_to_world(id, uvw); }
786
973
  WP_API void builtin_volume_world_to_index_uint64_vec3f(uint64 id, vec3f& xyz, vec3f* ret) { *ret = wp::volume_world_to_index(id, xyz); }
787
974
  WP_API void builtin_volume_index_to_world_dir_uint64_vec3f(uint64 id, vec3f& uvw, vec3f* ret) { *ret = wp::volume_index_to_world_dir(id, uvw); }
warp/native/mat.h CHANGED
@@ -397,6 +397,41 @@ inline bool CUDA_CALLABLE isfinite(const mat_t<Rows,Cols,Type>& m)
397
397
  return true;
398
398
  }
399
399
 
400
+ template<unsigned Rows, unsigned Cols, typename Type>
401
+ inline void CUDA_CALLABLE adj_isfinite(const mat_t<Rows,Cols,Type>& m, mat_t<Rows,Cols,Type>& adj_m, const bool &adj_ret)
402
+ {
403
+ }
404
+
405
+ template<unsigned Rows, unsigned Cols, typename Type>
406
+ inline bool CUDA_CALLABLE isnan(const mat_t<Rows,Cols,Type>& m)
407
+ {
408
+ for (unsigned i=0; i < Rows; ++i)
409
+ for (unsigned j=0; j < Cols; ++j)
410
+ if (isnan(m.data[i][j]))
411
+ return true;
412
+ return false;
413
+ }
414
+
415
+ template<unsigned Rows, unsigned Cols, typename Type>
416
+ inline void CUDA_CALLABLE adj_isnan(const mat_t<Rows,Cols,Type>& m, mat_t<Rows,Cols,Type>& adj_m, const bool &adj_ret)
417
+ {
418
+ }
419
+
420
+ template<unsigned Rows, unsigned Cols, typename Type>
421
+ inline bool CUDA_CALLABLE isinf(const mat_t<Rows,Cols,Type>& m)
422
+ {
423
+ for (unsigned i=0; i < Rows; ++i)
424
+ for (unsigned j=0; j < Cols; ++j)
425
+ if (isinf(m.data[i][j]))
426
+ return true;
427
+ return false;
428
+ }
429
+
430
+ template<unsigned Rows, unsigned Cols, typename Type>
431
+ inline void CUDA_CALLABLE adj_isinf(const mat_t<Rows,Cols,Type>& m, mat_t<Rows,Cols,Type>& adj_m, const bool &adj_ret)
432
+ {
433
+ }
434
+
400
435
  template<unsigned Rows, unsigned Cols, typename Type>
401
436
  inline CUDA_CALLABLE mat_t<Rows,Cols,Type> add(const mat_t<Rows,Cols,Type>& a, const mat_t<Rows,Cols,Type>& b)
402
437
  {
@@ -842,6 +877,18 @@ inline CUDA_CALLABLE mat_t<Rows,Cols,Type> outer(const vec_t<Rows,Type>& a, cons
842
877
  return ret;
843
878
  }
844
879
 
880
+ template<unsigned Cols,typename Type>
881
+ inline CUDA_CALLABLE vec_t<Cols,Type> outer(Type a, const vec_t<Cols,Type>& b)
882
+ {
883
+ return mul(a, b);
884
+ }
885
+
886
+ template<unsigned Rows,typename Type>
887
+ inline CUDA_CALLABLE vec_t<Rows,Type> outer(const vec_t<Rows,Type>& a, Type b)
888
+ {
889
+ return mul(a, b);
890
+ }
891
+
845
892
  template<typename Type>
846
893
  inline CUDA_CALLABLE mat_t<3,3,Type> skew(const vec_t<3,Type>& a)
847
894
  {
warp/native/mesh.cpp CHANGED
@@ -127,7 +127,7 @@ uint64_t mesh_create_host(array_t<wp::vec3> points, array_t<wp::vec3> velocities
127
127
  }
128
128
  m->average_edge_length = sum / (num_tris*3);
129
129
 
130
- m->bvh = *(wp::BVH*)bvh_create_host(m->lowers, m->uppers, num_tris);
130
+ wp::bvh_create_host(m->lowers, m->uppers, num_tris, m->bvh);
131
131
 
132
132
  if (support_winding_number)
133
133
  {
warp/native/mesh.cu CHANGED
@@ -208,8 +208,7 @@ uint64_t mesh_create_device(void* context, wp::array_t<wp::vec3> points, wp::arr
208
208
 
209
209
  wp_launch_device(WP_CURRENT_CONTEXT, wp::compute_triangle_bounds, num_tris, (num_tris, points.data, indices.data, mesh.lowers, mesh.uppers));
210
210
 
211
- uint64_t bvh_id = bvh_create_device(mesh.context, mesh.lowers, mesh.uppers, num_tris);
212
- wp::bvh_get_descriptor(bvh_id, mesh.bvh);
211
+ wp::bvh_create_device(mesh.context, mesh.lowers, mesh.uppers, num_tris, mesh.bvh);
213
212
 
214
213
  if (support_winding_number)
215
214
  {
@@ -0,0 +1,366 @@
1
+ // Copyright Contributors to the OpenVDB Project
2
+ // SPDX-License-Identifier: MPL-2.0
3
+
4
+ /*!
5
+ \file nanovdb/GridHandle.h
6
+
7
+ \author Ken Museth
8
+
9
+ \date January 8, 2020
10
+
11
+ \brief Defines GridHandle, which manages a host, and possibly a device,
12
+ memory buffer containing one or more NanoVDB grids.
13
+ */
14
+
15
+ #ifndef NANOVDB_GRID_HANDLE_H_HAS_BEEN_INCLUDED
16
+ #define NANOVDB_GRID_HANDLE_H_HAS_BEEN_INCLUDED
17
+
18
+ #include <fstream> // for std::ifstream
19
+ #include <iostream> // for std::cerr/cout
20
+ #include <vector>
21
+ #include <initializer_list>
22
+
23
+ #include <nanovdb/NanoVDB.h>// for toGridType
24
+ #include <nanovdb/HostBuffer.h>
25
+
26
+ namespace nanovdb {
27
+
28
+ // --------------------------> GridHandle <------------------------------------
29
+
30
+ struct GridHandleMetaData {uint64_t offset, size; GridType gridType;};
31
+
32
+ /// @brief This class serves to manage a buffer containing one or more NanoVDB Grids.
33
+ ///
34
+ /// @note It is important to note that this class does NOT depend on OpenVDB.
35
+ template<typename BufferT = HostBuffer>
36
+ class GridHandle
37
+ {
38
+ std::vector<GridHandleMetaData> mMetaData;
39
+ BufferT mBuffer;
40
+
41
+ template <typename T>
42
+ static T* no_const(const T* ptr) { return const_cast<T*>(ptr); }
43
+
44
+ public:
45
+ using BufferType = BufferT;
46
+
47
+ /// @brief Move constructor from a host buffer
48
+ /// @param buffer buffer containing one or more NanoGrids that will be moved into this GridHandle
49
+ /// @throw Will throw and error with the buffer does not contain a valid NanoGrid!
50
+ template<typename T = BufferT, typename util::enable_if<BufferTraits<T>::hasDeviceDual, int>::type = 0>
51
+ GridHandle(T&& buffer);
52
+
53
+ /// @brief Move constructor from a dual host-device buffer
54
+ /// @param buffer buffer containing one or more NanoGrids that will be moved into this GridHandle
55
+ /// @throw Will throw and error with the buffer does not contain a valid NanoGrid!
56
+ template<typename T = BufferT, typename util::disable_if<BufferTraits<T>::hasDeviceDual, int>::type = 0>
57
+ GridHandle(T&& buffer);
58
+
59
+ /// @brief Constructs an empty GridHandle
60
+ GridHandle() = default;
61
+
62
+ /// @brief Disallow copy-construction
63
+ GridHandle(const GridHandle&) = delete;
64
+
65
+ /// @brief Move copy-constructor
66
+ GridHandle(GridHandle&& other) noexcept {
67
+ mBuffer = std::move(other.mBuffer);
68
+ mMetaData = std::move(other.mMetaData);
69
+ }
70
+
71
+ /// @brief clear this GridHandle to an empty handle
72
+ void reset() {
73
+ mBuffer.clear();
74
+ mMetaData.clear();
75
+ }
76
+
77
+ /// @brief Disallow copy assignment operation
78
+ GridHandle& operator=(const GridHandle&) = delete;
79
+
80
+ /// @brief Move copy assignment operation
81
+ GridHandle& operator=(GridHandle&& other) noexcept {
82
+ mBuffer = std::move(other.mBuffer);
83
+ mMetaData = std::move(other.mMetaData);
84
+ return *this;
85
+ }
86
+
87
+ /// @brief Performs a deep copy of the GridHandle, possibly templated on a different buffer type
88
+ /// @tparam OtherBufferT Buffer type of the deep copy
89
+ /// @param buffer optional buffer used for allocation
90
+ /// @return A new handle of the specified buffer type that contains a deep copy of the current handle
91
+ template <typename OtherBufferT = HostBuffer>
92
+ GridHandle<OtherBufferT> copy(const OtherBufferT& buffer = OtherBufferT()) const;
93
+
94
+ /// @brief Return a reference to the buffer
95
+ BufferT& buffer() { return mBuffer; }
96
+
97
+ /// @brief Return a const reference to the buffer
98
+ const BufferT& buffer() const { return mBuffer; }
99
+
100
+ /// @brief Returns a non-const pointer to the data.
101
+ /// @warning Note that the return pointer can be NULL if the GridHandle was not initialized
102
+ void* data() { return mBuffer.data(); }
103
+
104
+ /// @brief Returns a const pointer to the data.
105
+ /// @warning Note that the return pointer can be NULL if the GridHandle was not initialized
106
+ const void* data() const { return mBuffer.data(); }
107
+
108
+ template<typename U = BufferT>
109
+ typename util::enable_if<BufferTraits<U>::hasDeviceDual, const void*>::type
110
+ deviceData() const { return mBuffer.deviceData(); }
111
+ template<typename U = BufferT>
112
+ typename util::enable_if<BufferTraits<U>::hasDeviceDual, void*>::type
113
+ deviceData() { return mBuffer.deviceData(); }
114
+
115
+ /// @brief Returns the size in bytes of the raw memory buffer managed by this GridHandle.
116
+ uint64_t size() const { return mBuffer.size(); }
117
+
118
+ //@{
119
+ /// @brief Return true if this handle is empty, i.e. has no allocated memory
120
+ bool empty() const { return this->size() == 0; }
121
+ bool isEmpty() const { return this->size() == 0; }
122
+ //@}
123
+
124
+ /// @brief Return true if this handle contains any grids
125
+ operator bool() const { return !this->empty(); }
126
+
127
+ /// @brief Returns a const host pointer to the @a n'th NanoVDB grid encoded in this GridHandle.
128
+ /// @tparam ValueT Value type of the grid point to be returned
129
+ /// @param n Index of the (host) grid pointer to be returned
130
+ /// @warning Note that the return pointer can be NULL if the GridHandle no host grid, @a n is invalid
131
+ /// or if the template parameter does not match the specified grid!
132
+ template<typename ValueT>
133
+ const NanoGrid<ValueT>* grid(uint32_t n = 0) const;
134
+
135
+ /// @brief Returns a host pointer to the @a n'th NanoVDB grid encoded in this GridHandle.
136
+ /// @tparam ValueT Value type of the grid point to be returned
137
+ /// @param n Index of the (host) grid pointer to be returned
138
+ /// @warning Note that the return pointer can be NULL if the GridHandle no host grid, @a n is invalid
139
+ /// or if the template parameter does not match the specified grid!
140
+ template<typename ValueT>
141
+ NanoGrid<ValueT>* grid(uint32_t n = 0) {return const_cast<NanoGrid<ValueT>*>(static_cast<const GridHandle*>(this)->template grid<ValueT>(n));}
142
+
143
+ /// @brief Return a const pointer to the @a n'th grid encoded in this GridHandle on the device, e.g. GPU
144
+ /// @tparam ValueT Value type of the grid point to be returned
145
+ /// @param n Index of the (device) grid pointer to be returned
146
+ /// @warning Note that the return pointer can be NULL if the GridHandle has no device grid, @a n is invalid,
147
+ /// or if the template parameter does not match the specified grid.
148
+ template<typename ValueT, typename U = BufferT>
149
+ typename util::enable_if<BufferTraits<U>::hasDeviceDual, const NanoGrid<ValueT>*>::type
150
+ deviceGrid(uint32_t n=0) const;
151
+
152
+ /// @brief Return a const pointer to the @a n'th grid encoded in this GridHandle on the device, e.g. GPU
153
+ /// @tparam ValueT Value type of the grid point to be returned
154
+ /// @param n Index if of the grid pointer to be returned
155
+ /// @param verbose if non-zero error messages will be printed in case something failed
156
+ /// @warning Note that the return pointer can be NULL if the GridHandle was not initialized, @a n is invalid,
157
+ /// or if the template parameter does not match the specified grid.
158
+ template<typename ValueT, typename U = BufferT>
159
+ typename util::enable_if<BufferTraits<U>::hasDeviceDual, NanoGrid<ValueT>*>::type
160
+ deviceGrid(uint32_t n=0){return const_cast<NanoGrid<ValueT>*>(static_cast<const GridHandle*>(this)->template deviceGrid<ValueT>(n));}
161
+
162
+ /// @brief Upload the grid to the device, e.g. from CPU to GPU
163
+ /// @note This method is only available if the buffer supports devices
164
+ template<typename U = BufferT>
165
+ typename util::enable_if<BufferTraits<U>::hasDeviceDual, void>::type
166
+ deviceUpload(void* stream = nullptr, bool sync = true) { mBuffer.deviceUpload(stream, sync); }
167
+
168
+ /// @brief Download the grid to from the device, e.g. from GPU to CPU
169
+ /// @note This method is only available if the buffer supports devices
170
+ template<typename U = BufferT>
171
+ typename util::enable_if<BufferTraits<U>::hasDeviceDual, void>::type
172
+ deviceDownload(void* stream = nullptr, bool sync = true) { mBuffer.deviceDownload(stream, sync); }
173
+
174
+ /// @brief Check if the buffer is this handle has any padding, i.e. if the buffer is larger than the combined size of all its grids
175
+ /// @return true is the combined size of all grid is smaller than the buffer size
176
+ bool isPadded() const {return mMetaData.empty() ? false : mMetaData.back().offset + mMetaData.back().size != mBuffer.size();}
177
+
178
+ /// @brief Return the total number of grids contained in this buffer
179
+ uint32_t gridCount() const {return static_cast<uint32_t>(mMetaData.size());}
180
+
181
+ /// @brief Return the grid size of the @a n'th grid in this GridHandle
182
+ /// @param n index of the grid (assumed to be less than gridCount())
183
+ /// @return Return the byte size of the specified grid
184
+ uint64_t gridSize(uint32_t n = 0) const {return mMetaData[n].size; }
185
+
186
+ /// @brief Return the GridType of the @a n'th grid in this GridHandle
187
+ /// @param n index of the grid (assumed to be less than gridCount())
188
+ /// @return Return the GridType of the specified grid
189
+ GridType gridType(uint32_t n = 0) const {return mMetaData[n].gridType; }
190
+
191
+ /// @brief Access to the GridData of the n'th grid in the current handle
192
+ /// @param n zero-based ID of the grid
193
+ /// @return Const pointer to the n'th GridData in the current handle
194
+ const GridData* gridData(uint32_t n = 0) const;
195
+
196
+ /// @brief Returns a const point to the @a n'th grid meta data
197
+ /// @param n zero-based ID of the grid
198
+ /// @warning Note that the return pointer can be NULL if the GridHandle was not initialized
199
+ const GridMetaData* gridMetaData(uint32_t n = 0) const;
200
+
201
+ /// @brief Write a specific grid in this buffer to an output stream
202
+ /// @param os output stream that the buffer will be written to
203
+ /// @param n zero-based index of the grid to be written to stream
204
+ void write(std::ostream& os, uint32_t n) const {
205
+ if (const GridData* data = this->gridData(n)) {
206
+ os.write((const char*)data, data->mGridSize);
207
+ } else {
208
+ throw std::runtime_error("GridHandle does not contain a #" + std::to_string(n) + " grid");
209
+ }
210
+ }
211
+
212
+ /// @brief Write the entire grid buffer to an output stream
213
+ /// @param os output stream that the buffer will be written to
214
+ void write(std::ostream& os) const {
215
+ for (uint32_t n=0; n<this->gridCount(); ++n) this->write(os, n);
216
+ }
217
+
218
+ /// @brief Write this entire grid buffer to a file
219
+ /// @param fileName string name of the output file
220
+ void write(const std::string &fileName) const {
221
+ std::ofstream os(fileName, std::ios::out | std::ios::binary | std::ios::trunc);
222
+ if (!os.is_open()) throw std::ios_base::failure("Unable to open file named \"" + fileName + "\" for output");
223
+ this->write(os);
224
+ }
225
+
226
+ /// @brief Write a specific grid to file
227
+ /// @param fileName string name of the output file
228
+ /// @param n zero-based index of the grid to be written to file
229
+ void write(const std::string &fileName, uint32_t n) const {
230
+ std::ofstream os(fileName, std::ios::out | std::ios::binary | std::ios::trunc);
231
+ if (!os.is_open()) throw std::ios_base::failure("Unable to open file named \"" + fileName + "\" for output");
232
+ this->write(os, n);
233
+ }
234
+
235
+ /// @brief Read an entire raw grid buffer from an input stream
236
+ /// @param is input stream containing a raw grid buffer
237
+ /// @param pool optional pool from which to allocate the new grid buffer
238
+ /// @throw Will throw a std::logic_error if the stream does not contain a valid raw grid
239
+ void read(std::istream& is, const BufferT& pool = BufferT());
240
+
241
+ /// @brief Read a specific grid from an input stream containing a raw grid buffer
242
+ /// @param is input stream containing a raw grid buffer
243
+ /// @param n zero-based index of the grid to be read
244
+ /// @param pool optional pool from which to allocate the new grid buffer
245
+ /// @throw Will throw a std::logic_error if the stream does not contain a valid raw grid
246
+ void read(std::istream& is, uint32_t n, const BufferT& pool = BufferT());
247
+
248
+ /// @brief Read a specific grid from an input stream containing a raw grid buffer
249
+ /// @param is input stream containing a raw grid buffer
250
+ /// @param gridName string name of the grid to be read
251
+ /// @param pool optional pool from which to allocate the new grid buffer
252
+ /// @throw Will throw a std::logic_error if the stream does not contain a valid raw grid with the speficied name
253
+ void read(std::istream& is, const std::string &gridName, const BufferT& pool = BufferT());
254
+
255
+ /// @brief Read a raw grid buffer from a file
256
+ /// @param filename string name of the input file containing a raw grid buffer
257
+ /// @param pool optional pool from which to allocate the new grid buffe
258
+ void read(const std::string &fileName, const BufferT& pool = BufferT()) {
259
+ std::ifstream is(fileName, std::ios::in | std::ios::binary);
260
+ if (!is.is_open()) throw std::ios_base::failure("Unable to open file named \"" + fileName + "\" for input");
261
+ this->read(is, pool);
262
+ }
263
+
264
+ /// @brief Read a specific grid from a file containing a raw grid buffer
265
+ /// @param filename string name of the input file containing a raw grid buffer
266
+ /// @param n zero-based index of the grid to be read
267
+ /// @param pool optional pool from which to allocate the new grid buffer
268
+ /// @throw Will throw a std::ios_base::failure if the file does not exist and a
269
+ /// std::logic_error if the files does not contain a valid raw grid
270
+ void read(const std::string &fileName, uint32_t n, const BufferT& pool = BufferT()) {
271
+ std::ifstream is(fileName, std::ios::in | std::ios::binary);
272
+ if (!is.is_open()) throw std::ios_base::failure("Unable to open file named \"" + fileName + "\" for input");
273
+ this->read(is, n, pool);
274
+ }
275
+
276
+ /// @brief Read a specific grid from a file containing a raw grid buffer
277
+ /// @param filename string name of the input file containing a raw grid buffer
278
+ /// @param gridName string name of the grid to be read
279
+ /// @param pool optional pool from which to allocate the new grid buffer
280
+ /// @throw Will throw a std::ios_base::failure if the file does not exist and a
281
+ /// std::logic_error if the files does not contain a valid raw grid withe the specified name
282
+ void read(const std::string &fileName, const std::string &gridName, const BufferT& pool = BufferT()) {
283
+ std::ifstream is(fileName, std::ios::in | std::ios::binary);
284
+ if (!is.is_open()) throw std::ios_base::failure("Unable to open file named \"" + fileName + "\" for input");
285
+ this->read(is, gridName, pool);
286
+ }
287
+ }; // GridHandle
288
+
289
+ // --------------------------> Implementation of private methods in GridHandle <------------------------------------
290
+
291
+ template<typename BufferT>
292
+ inline const GridData* GridHandle<BufferT>::gridData(uint32_t n) const
293
+ {
294
+ const void *data = this->data();
295
+ if (data == nullptr || n >= mMetaData.size()) return nullptr;
296
+ return util::PtrAdd<GridData>(data, mMetaData[n].offset);
297
+ }// const GridData* GridHandle<BufferT>::gridData(uint32_t n) const
298
+
299
+ template<typename BufferT>
300
+ inline const GridMetaData* GridHandle<BufferT>::gridMetaData(uint32_t n) const
301
+ {
302
+ const auto *data = this->data();
303
+ if (data == nullptr || n >= mMetaData.size()) return nullptr;
304
+ return util::PtrAdd<GridMetaData>(data, mMetaData[n].offset);
305
+ }// const GridMetaData* GridHandle<BufferT>::gridMetaData(uint32_t n) const
306
+
307
+ inline __hostdev__ void cpyGridHandleMeta(const GridData *data, GridHandleMetaData *meta)
308
+ {
309
+ uint64_t offset = 0;
310
+ for (auto *p=meta, *q=p+data->mGridCount; p!=q; ++p) {
311
+ *p = {offset, data->mGridSize, data->mGridType};
312
+ offset += p->size;
313
+ data = util::PtrAdd<GridData>(data, p->size);
314
+ }
315
+ }// void cpyGridHandleMeta(const GridData *data, GridHandleMetaData *meta)
316
+
317
+ template<typename BufferT>
318
+ template<typename T, typename util::disable_if<BufferTraits<T>::hasDeviceDual, int>::type>
319
+ GridHandle<BufferT>::GridHandle(T&& buffer)
320
+ {
321
+ static_assert(util::is_same<T,BufferT>::value, "Expected U==BufferT");
322
+ mBuffer = std::move(buffer);
323
+ if (auto *data = reinterpret_cast<const GridData*>(mBuffer.data())) {
324
+ if (!data->isValid()) throw std::runtime_error("GridHandle was constructed with an invalid host buffer");
325
+ mMetaData.resize(data->mGridCount);
326
+ cpyGridHandleMeta(data, mMetaData.data());
327
+ }
328
+ }// GridHandle<BufferT>::GridHandle(T&& buffer)
329
+
330
+ template<typename BufferT>
331
+ template <typename OtherBufferT>
332
+ inline GridHandle<OtherBufferT> GridHandle<BufferT>::copy(const OtherBufferT& other) const
333
+ {
334
+ if (mBuffer.isEmpty()) return GridHandle<OtherBufferT>();// return an empty handle
335
+ auto buffer = OtherBufferT::create(mBuffer.size(), &other);
336
+ std::memcpy(buffer.data(), mBuffer.data(), mBuffer.size());// deep copy of buffer
337
+ return GridHandle<OtherBufferT>(std::move(buffer));
338
+ }// GridHandle<OtherBufferT> GridHandle<BufferT>::copy(const OtherBufferT& other) const
339
+
340
+ template<typename BufferT>
341
+ template<typename ValueT>
342
+ inline const NanoGrid<ValueT>* GridHandle<BufferT>::grid(uint32_t n) const
343
+ {
344
+ const void *data = mBuffer.data();
345
+ if (data == nullptr || n >= mMetaData.size() || mMetaData[n].gridType != toGridType<ValueT>()) return nullptr;
346
+ return util::PtrAdd<NanoGrid<ValueT>>(data, mMetaData[n].offset);
347
+ }// const NanoGrid<ValueT>* GridHandle<BufferT>::grid(uint32_t n) const
348
+
349
+ template<typename BufferT>
350
+ template<typename ValueT, typename U>
351
+ inline typename util::enable_if<BufferTraits<U>::hasDeviceDual, const NanoGrid<ValueT>*>::type
352
+ GridHandle<BufferT>::deviceGrid(uint32_t n) const
353
+ {
354
+ const void *data = mBuffer.deviceData();
355
+ if (data == nullptr || n >= mMetaData.size() || mMetaData[n].gridType != toGridType<ValueT>()) return nullptr;
356
+ return util::PtrAdd<NanoGrid<ValueT>>(data, mMetaData[n].offset);
357
+ }// GridHandle<BufferT>::deviceGrid(uint32_t n) cons
358
+
359
+
360
+ } // namespace nanovdb
361
+
362
+ #if defined(__CUDACC__)
363
+ #include <nanovdb/cuda/GridHandle.cuh>
364
+ #endif// defined(__CUDACC__)
365
+
366
+ #endif // NANOVDB_GRID_HANDLE_H_HAS_BEEN_INCLUDED