warp-lang 1.1.0__py3-none-win_amd64.whl → 1.2.1__py3-none-win_amd64.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.
- warp/bin/warp-clang.dll +0 -0
- warp/bin/warp.dll +0 -0
- warp/build.py +10 -37
- warp/build_dll.py +2 -2
- warp/builtins.py +274 -6
- warp/codegen.py +51 -4
- warp/config.py +2 -2
- warp/constants.py +4 -0
- warp/context.py +422 -203
- warp/examples/benchmarks/benchmark_api.py +0 -2
- warp/examples/benchmarks/benchmark_cloth_warp.py +0 -1
- warp/examples/benchmarks/benchmark_launches.py +0 -2
- warp/examples/core/example_dem.py +0 -2
- warp/examples/core/example_fluid.py +0 -2
- warp/examples/core/example_graph_capture.py +0 -2
- warp/examples/core/example_marching_cubes.py +0 -2
- warp/examples/core/example_mesh.py +0 -2
- warp/examples/core/example_mesh_intersect.py +0 -2
- warp/examples/core/example_nvdb.py +0 -2
- warp/examples/core/example_raycast.py +0 -2
- warp/examples/core/example_raymarch.py +0 -2
- warp/examples/core/example_render_opengl.py +0 -2
- warp/examples/core/example_sph.py +0 -2
- warp/examples/core/example_torch.py +0 -3
- warp/examples/core/example_wave.py +0 -2
- warp/examples/fem/example_apic_fluid.py +140 -115
- warp/examples/fem/example_burgers.py +262 -0
- warp/examples/fem/example_convection_diffusion.py +0 -2
- warp/examples/fem/example_convection_diffusion_dg.py +0 -2
- warp/examples/fem/example_deformed_geometry.py +0 -2
- warp/examples/fem/example_diffusion.py +0 -2
- warp/examples/fem/example_diffusion_3d.py +5 -4
- warp/examples/fem/example_diffusion_mgpu.py +0 -2
- warp/examples/fem/example_mixed_elasticity.py +0 -2
- warp/examples/fem/example_navier_stokes.py +0 -2
- warp/examples/fem/example_stokes.py +0 -2
- warp/examples/fem/example_stokes_transfer.py +0 -2
- warp/examples/optim/example_bounce.py +0 -2
- warp/examples/optim/example_cloth_throw.py +0 -2
- warp/examples/optim/example_diffray.py +0 -2
- warp/examples/optim/example_drone.py +0 -2
- warp/examples/optim/example_inverse_kinematics.py +0 -2
- warp/examples/optim/example_inverse_kinematics_torch.py +0 -2
- warp/examples/optim/example_spring_cage.py +0 -2
- warp/examples/optim/example_trajectory.py +0 -2
- warp/examples/optim/example_walker.py +0 -2
- warp/examples/sim/example_cartpole.py +0 -2
- warp/examples/sim/example_cloth.py +0 -2
- warp/examples/sim/example_granular.py +0 -2
- warp/examples/sim/example_granular_collision_sdf.py +0 -2
- warp/examples/sim/example_jacobian_ik.py +0 -2
- warp/examples/sim/example_particle_chain.py +0 -2
- warp/examples/sim/example_quadruped.py +0 -2
- warp/examples/sim/example_rigid_chain.py +0 -2
- warp/examples/sim/example_rigid_contact.py +0 -2
- warp/examples/sim/example_rigid_force.py +0 -2
- warp/examples/sim/example_rigid_gyroscopic.py +0 -2
- warp/examples/sim/example_rigid_soft_contact.py +0 -2
- warp/examples/sim/example_soft_body.py +0 -2
- warp/fem/__init__.py +1 -0
- warp/fem/cache.py +3 -1
- warp/fem/geometry/__init__.py +1 -0
- warp/fem/geometry/element.py +4 -0
- warp/fem/geometry/grid_3d.py +0 -4
- warp/fem/geometry/nanogrid.py +455 -0
- warp/fem/integrate.py +63 -9
- warp/fem/space/__init__.py +43 -158
- warp/fem/space/basis_space.py +34 -0
- warp/fem/space/collocated_function_space.py +1 -1
- warp/fem/space/grid_2d_function_space.py +13 -132
- warp/fem/space/grid_3d_function_space.py +16 -154
- warp/fem/space/hexmesh_function_space.py +37 -134
- warp/fem/space/nanogrid_function_space.py +202 -0
- warp/fem/space/quadmesh_2d_function_space.py +12 -119
- warp/fem/space/restriction.py +4 -1
- warp/fem/space/shape/__init__.py +77 -0
- warp/fem/space/shape/cube_shape_function.py +5 -15
- warp/fem/space/tetmesh_function_space.py +6 -76
- warp/fem/space/trimesh_2d_function_space.py +6 -76
- warp/native/array.h +12 -3
- warp/native/builtin.h +48 -5
- warp/native/bvh.cpp +14 -10
- warp/native/bvh.cu +23 -15
- warp/native/bvh.h +1 -0
- warp/native/clang/clang.cpp +2 -1
- warp/native/crt.cpp +11 -1
- warp/native/crt.h +18 -1
- warp/native/exports.h +187 -0
- warp/native/mat.h +47 -0
- warp/native/mesh.cpp +1 -1
- warp/native/mesh.cu +1 -2
- warp/native/nanovdb/GridHandle.h +366 -0
- warp/native/nanovdb/HostBuffer.h +590 -0
- warp/native/nanovdb/NanoVDB.h +3999 -2157
- warp/native/nanovdb/PNanoVDB.h +936 -99
- warp/native/quat.h +28 -1
- warp/native/rand.h +5 -1
- warp/native/vec.h +45 -1
- warp/native/volume.cpp +335 -103
- warp/native/volume.cu +39 -13
- warp/native/volume.h +725 -303
- warp/native/volume_builder.cu +381 -360
- warp/native/volume_builder.h +16 -1
- warp/native/volume_impl.h +61 -0
- warp/native/warp.cu +8 -2
- warp/native/warp.h +15 -7
- warp/render/render_opengl.py +191 -52
- warp/sim/integrator_featherstone.py +10 -3
- warp/sim/integrator_xpbd.py +16 -22
- warp/sparse.py +89 -27
- warp/stubs.py +83 -0
- warp/tests/assets/test_index_grid.nvdb +0 -0
- warp/tests/aux_test_dependent.py +0 -2
- warp/tests/aux_test_grad_customs.py +0 -2
- warp/tests/aux_test_reference.py +0 -2
- warp/tests/aux_test_reference_reference.py +0 -2
- warp/tests/aux_test_square.py +0 -2
- warp/tests/disabled_kinematics.py +0 -2
- warp/tests/test_adam.py +0 -2
- warp/tests/test_arithmetic.py +0 -36
- warp/tests/test_array.py +9 -11
- warp/tests/test_array_reduce.py +0 -2
- warp/tests/test_async.py +0 -2
- warp/tests/test_atomic.py +0 -2
- warp/tests/test_bool.py +58 -50
- warp/tests/test_builtins_resolution.py +0 -2
- warp/tests/test_bvh.py +0 -2
- warp/tests/test_closest_point_edge_edge.py +0 -1
- warp/tests/test_codegen.py +0 -4
- warp/tests/test_compile_consts.py +130 -10
- warp/tests/test_conditional.py +0 -2
- warp/tests/test_copy.py +0 -2
- warp/tests/test_ctypes.py +6 -8
- warp/tests/test_dense.py +0 -2
- warp/tests/test_devices.py +0 -2
- warp/tests/test_dlpack.py +9 -11
- warp/tests/test_examples.py +42 -39
- warp/tests/test_fabricarray.py +0 -3
- warp/tests/test_fast_math.py +0 -2
- warp/tests/test_fem.py +75 -54
- warp/tests/test_fp16.py +0 -2
- warp/tests/test_func.py +0 -2
- warp/tests/test_generics.py +27 -2
- warp/tests/test_grad.py +147 -8
- warp/tests/test_grad_customs.py +0 -2
- warp/tests/test_hash_grid.py +1 -3
- warp/tests/test_import.py +0 -2
- warp/tests/test_indexedarray.py +0 -2
- warp/tests/test_intersect.py +0 -2
- warp/tests/test_jax.py +0 -2
- warp/tests/test_large.py +11 -9
- warp/tests/test_launch.py +0 -2
- warp/tests/test_lerp.py +10 -54
- warp/tests/test_linear_solvers.py +3 -5
- warp/tests/test_lvalue.py +0 -2
- warp/tests/test_marching_cubes.py +0 -2
- warp/tests/test_mat.py +0 -2
- warp/tests/test_mat_lite.py +0 -2
- warp/tests/test_mat_scalar_ops.py +0 -2
- warp/tests/test_math.py +0 -2
- warp/tests/test_matmul.py +35 -37
- warp/tests/test_matmul_lite.py +29 -31
- warp/tests/test_mempool.py +0 -2
- warp/tests/test_mesh.py +0 -3
- warp/tests/test_mesh_query_aabb.py +0 -2
- warp/tests/test_mesh_query_point.py +0 -2
- warp/tests/test_mesh_query_ray.py +0 -2
- warp/tests/test_mlp.py +0 -2
- warp/tests/test_model.py +0 -2
- warp/tests/test_module_hashing.py +111 -0
- warp/tests/test_modules_lite.py +0 -3
- warp/tests/test_multigpu.py +0 -2
- warp/tests/test_noise.py +0 -4
- warp/tests/test_operators.py +0 -2
- warp/tests/test_options.py +0 -2
- warp/tests/test_peer.py +0 -2
- warp/tests/test_pinned.py +0 -2
- warp/tests/test_print.py +0 -2
- warp/tests/test_quat.py +0 -2
- warp/tests/test_rand.py +41 -5
- warp/tests/test_reload.py +0 -10
- warp/tests/test_rounding.py +0 -2
- warp/tests/test_runlength_encode.py +0 -2
- warp/tests/test_sim_grad.py +0 -2
- warp/tests/test_sim_kinematics.py +0 -2
- warp/tests/test_smoothstep.py +0 -2
- warp/tests/test_snippet.py +0 -2
- warp/tests/test_sparse.py +0 -2
- warp/tests/test_spatial.py +0 -2
- warp/tests/test_special_values.py +362 -0
- warp/tests/test_streams.py +0 -2
- warp/tests/test_struct.py +0 -2
- warp/tests/test_tape.py +0 -2
- warp/tests/test_torch.py +0 -2
- warp/tests/test_transient_module.py +0 -2
- warp/tests/test_types.py +0 -2
- warp/tests/test_utils.py +0 -2
- warp/tests/test_vec.py +0 -2
- warp/tests/test_vec_lite.py +0 -2
- warp/tests/test_vec_scalar_ops.py +0 -2
- warp/tests/test_verify_fp.py +0 -2
- warp/tests/test_volume.py +237 -13
- warp/tests/test_volume_write.py +86 -3
- warp/tests/unittest_serial.py +10 -9
- warp/tests/unittest_suites.py +6 -2
- warp/tests/unittest_utils.py +2 -171
- warp/tests/unused_test_misc.py +0 -2
- warp/tests/walkthrough_debug.py +1 -1
- warp/thirdparty/unittest_parallel.py +37 -40
- warp/types.py +526 -85
- {warp_lang-1.1.0.dist-info → warp_lang-1.2.1.dist-info}/METADATA +61 -31
- warp_lang-1.2.1.dist-info/RECORD +359 -0
- warp/examples/fem/example_convection_diffusion_dg0.py +0 -204
- warp/native/nanovdb/PNanoVDBWrite.h +0 -295
- warp_lang-1.1.0.dist-info/RECORD +0 -352
- {warp_lang-1.1.0.dist-info → warp_lang-1.2.1.dist-info}/LICENSE.md +0 -0
- {warp_lang-1.1.0.dist-info → warp_lang-1.2.1.dist-info}/WHEEL +0 -0
- {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
|
-
|
|
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
|
-
|
|
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
|