View page source

warp.fem.Nanogrid#

class warp.fem.Nanogrid(
grid,
temporary_store=None,
scalar_type=warp.float32,
cell_env=None,
env_offsets=None,
)[source]#

Sparse grid geometry.

Parameters:
__init__(
grid,
temporary_store=None,
scalar_type=warp.float32,
cell_env=None,
env_offsets=None,
)[source]#

Construct a sparse grid geometry from an in-memory NanoVDB volume.

Parameters:

  • grid (Volume) – The NanoVDB volume. Any type is accepted, but for indexing efficiency an index grid is recommended. If grid is an "on" index grid, cells will be created for active voxels only, otherwise cells will be created for all leaf voxels.

  • temporary_store (TemporaryStore | None) – shared pool from which to allocate temporary arrays

  • scalar_type (type) – Scalar type for grid coordinates (warp.float32 or warp.float64)

  • cell_env (array | None)

  • env_offsets (array | None)

Methods

__init__(grid[, temporary_store, ...])

Construct a sparse grid geometry from an in-memory NanoVDB volume.

boundary_side_count()

build_bvh([device])

Rebuild the geometry's Bounding Volume Hierarchy (BVH) for device from scratch.

bvh_id(device)

Return the BVH identifier for the given device, or 0 if unavailable.

cell_arg_value(device)

cell_bvh_groups(device)

Return optional per-cell BVH group indices on device.

cell_count()

cell_env_arg_value(device)

Return per-cell environment indices for geometry argument structs.

edge_count()

environment_count()

fill_cell_arg(arg, device)

fill_side_arg(arg, device)

fill_side_index_arg(arg, device)

from_environment_voxels(cell_ijks[, ...])

Construct a sparse grid geometry from per-environment active cell coordinates.

make_filtered_cell_lookup([filter_func])

reference_cell()

reference_side()

side_arg_value(device)

side_count()

side_deformation_gradient(args, s)

Device function returning the gradient of world position with respect to reference side

side_index_arg_value(device)

side_normal(args, s)

Device function returning the element normal at a sample point

supports_cell_lookup(device)

update_bvh([device])

Refit the geometry's BVH if it exists on device, or build it from scratch otherwise.

vertex_count()

Attributes

FACE_AXIS_MASK

FACE_INNER_OFFSET_BIT

FACE_OUTER_OFFSET_BIT

GRID_AXIS_FLAG

SideIndexArg

Structure containing arguments to be passed to device functions for indexing sides.

base

Return the base geometry from which this geometry derives its topology.

boundary_side_index

cell_closest_point

cell_coordinates

cell_deformation_gradient

cell_dimension

Manifold dimension of the geometry cells

cell_env

cell_environment_index

cell_grid

cell_inverse_deformation_gradient

cell_lookup

Device function for looking up the closest cell to a position.

cell_measure

cell_measure_ratio

cell_normal

cell_position

compute_cell_bvh_groups

coords_type

Warp vector type for element coordinates matching this geometry's scalar precision.

dimension

Dimension of the embedding space.

edge_grid

env_offsets

face_grid

name

Name of the geometry, including environment and scalar type suffixes.

sample_type

Warp struct type for samples matching this geometry's scalar precision.

scalar_type

side_closest_point

side_coordinates

side_environment_index

side_from_cell_coords

side_inner_cell_coords

side_inner_cell_index

side_inner_inverse_deformation_gradient

side_measure

side_measure_ratio

side_outer_cell_coords

side_outer_cell_index

side_outer_inverse_deformation_gradient

side_position

side_to_cell_arg

transform

Transform matrix mapping index to world space.

vertex_grid

CellArg

Structure containing arguments to be passed to device functions evaluating cell-related quantities.

SideArg

Structure containing arguments to be passed to device functions evaluating side-related quantities.
classmethod from_environment_voxels(
cell_ijks,
env_offsets=None,
*,
guard_cells=3,
voxel_size=1.0,
translation=(0.0, 0.0, 0.0),
transform=None,
temporary_store=None,
scalar_type=warp.float32,
device=None,
)[source]#

Construct a sparse grid geometry from per-environment active cell coordinates.

The per-environment coordinates are interpreted in local FEM index space. They are packed into a single NanoVDB index grid by adding an environment offset, while FEM positions subtract the offset so environments may remain colocated in world space.

Parameters:

  • cell_ijks (Sequence[array]) – Sequence of warp.vec3i arrays, one array per environment.

  • env_offsets (array | Sequence[Sequence[int]] | None) – Optional packed-grid offsets, one warp.vec3i per environment. If omitted, offsets are generated along the x axis with at least one guard region between consecutive environments. Custom offsets are an advanced override for callers that need deterministic packed NanoVDB coordinates, for example to match an externally built volume. They must still keep active cells from different environments from sharing packed-grid faces.

  • guard_cells (int) – Number of empty packed cells between generated environment tiles. The default isolates padded B-spline node grids up to degree 3.

  • voxel_size (int | float | Sequence[float] | None) – Voxel size for the packed NanoVDB volume. Ignored if transform is provided.

  • translation – Translation between packed index and world spaces.

  • transform – Linear transform between packed index and world spaces.

  • temporary_store (TemporaryStore | None) – Shared pool from which to allocate temporary arrays.

  • scalar_type (type) – Scalar type for grid coordinates (warp.float32 or warp.float64).

  • device – CUDA device on which to build the packed volume.

property edge_grid: Volume[source]#
edge_count()[source]#
fill_cell_arg(arg, device)[source]#
fill_side_arg(arg, device)[source]#
supports_cell_lookup(device)[source]#
cell_position = <Function Nanogrid__cell_position(args: Any, s: Any)>#
Parameters:
cell_deformation_gradient = <Function Nanogrid__cell_deformation_gradient(args: Any, s: Any)>#
Parameters:
cell_inverse_deformation_gradient = <Function Nanogrid__cell_inverse_deformation_gradient(args: Any, s: Any)>#
Parameters:
cell_measure = <Function Nanogrid__cell_measure(args: Any, s: Any)>#
Parameters:
cell_normal = <Function Nanogrid__cell_normal(args: Any, s: Any)>#
Parameters:
cell_coordinates = <Function Nanogrid__cell_coordinates(args: Any, cell_index: int, pos: Any)>#
Parameters:
cell_closest_point = <Function Nanogrid__cell_closest_point(args: Any, cell_index: int, pos: Any)>#
Parameters:
side_position = <Function Nanogrid__side_position(args: Any, s: Any)>#
Parameters:
side_inner_inverse_deformation_gradient = <Function Nanogrid__side_inner_inverse_deformation_gradient(args: Any, s: Any)>#
Parameters:
side_outer_inverse_deformation_gradient = <Function Nanogrid__side_outer_inverse_deformation_gradient(args: Any, s: Any)>#
Parameters:
side_measure = <Function Nanogrid__side_measure(args: Any, s: Any)>#
Parameters:
side_measure_ratio = <Function Nanogrid__side_measure_ratio(args: Any, s: Any)>#
Parameters:
side_inner_cell_index = <Function Nanogrid__side_inner_cell_index(args: Any, side_index: int)>#
Parameters:
side_outer_cell_index = <Function Nanogrid__side_outer_cell_index(args: Any, side_index: int)>#
Parameters:
side_inner_cell_coords = <Function Nanogrid__side_inner_cell_coords(args: Any, side_index: int, side_coords: Any)>#
Parameters:

  • args (Any)

  • side_index (int)

  • side_coords (Any)

side_outer_cell_coords = <Function Nanogrid__side_outer_cell_coords(args: Any, side_index: int, side_coords: Any)>#
Parameters:

  • args (Any)

  • side_index (int)

  • side_coords (Any)

side_from_cell_coords = <Function Nanogrid__side_from_cell_coords(args: Any, side_index: int, element_index: int, element_coords: Any)>#
Parameters:

  • args (Any)

  • side_index (int)

  • element_index (int)

  • element_coords (Any)

side_to_cell_arg = <Function Nanogrid__side_to_cell_arg(side_arg: Any)>#
Parameters:

side_arg (Any)

side_coordinates = <Function Nanogrid__side_coordinates(args: Any, side_index: int, pos: Any)>#
Parameters:
side_closest_point = <Function Nanogrid__side_closest_point(args: Any, side_index: int, pos: Any)>#
Parameters: