warp.fem.ExplicitGeometryPartition#

class warp.fem.ExplicitGeometryPartition( geometry, cell_mask, max_cell_count=-1, max_side_count=-1, temporary_store=None, )[source]#

Geometry partition defined by an explicit cell selection mask.

Creates a partition where the included cells are specified by a binary mask array. This provides fine-grained control over which cells belong to the partition.

Parameters:

geometry (Geometry)
cell_mask (wp.array(dtype=int))
max_cell_count (int)
max_side_count (int)
temporary_store (TemporaryStore | None)

__init__( geometry, cell_mask, max_cell_count=-1, max_side_count=-1, temporary_store=None, )[source]#

Create a geometry partition from an active cell mask.

Parameters:

geometry (Geometry) – The geometry to partition.
cell_mask (array(ndim=1, dtype=int32)) – Warp array of length geometry.cell_count() indicating which cells are selected. Values must be 1 (selected) or 0 (not selected).
max_cell_count (int) – If positive, limits the number of cells to avoid device/host synchronization.
max_side_count (int) – If positive, limits the number of sides to avoid device/host synchronization.
temporary_store (TemporaryStore | None) – Optional temporary storage for intermediate arrays.

Methods

`__init__`(geometry, cell_mask[, ...])	Create a geometry partition from an active cell mask.
`boundary_side_count`()	Return the number of boundary sides in the partition.
`cell_arg_value`(device)
`cell_count`()	Return the number of cells in the partition.
`compute_side_indices_from_cells`(...[, ...])	Compute partition, boundary, and frontier side indices from a cell mask.
`fill_cell_arg`(args, device)	Fill the arguments to be passed to cell-related device functions.
`fill_side_arg`(args, device)	Fill the arguments to be passed to side-related device functions.
`frontier_side_count`()	Return the number of frontier sides in the partition.
`rebuild`(cell_mask[, temporary_store])	Rebuild the geometry partition from a new active cell mask.
`side_arg_value`(device)
`side_count`()	Return the number of sides in the partition.

Attributes

`CellArg`
`SideArg`	Argument structure for side-related device functions.
`boundary_side_index`
`cell_index`
`frontier_side_index`
`name`	Name of the geometry partition.
`partition_cell_index`
`side_index`
`side_to_cell_arg`
`geometry`	Underlying geometry.

rebuild(cell_mask, temporary_store=None)[source]#

Rebuild the geometry partition from a new active cell mask.

Parameters:

geometry – the geometry to partition
cell_mask (array(ndim=1, dtype=int32)) – warp array of length geometry.cell_count() indicating which cells are selected. Array values must be either 1 (selected) or 0 (not selected).
max_cell_count – if positive, will be used to limit the number of cells to avoid device/host synchronization
max_side_count – if positive, will be used to limit the number of sides to avoid device/host synchronization
temporary_store (TemporaryStore | None)

cell_count()[source]#

Return the number of cells in the partition.

Return type:: int

CellArg = <warp._src.codegen.Struct object>#

fill_cell_arg(args, device)[source]#

Fill the arguments to be passed to cell-related device functions.

Parameters:: args (CellArg)

cell_index = <Function ExplicitGeometryPartition__cell_index(args: warp._src.fem.geometry.partition.ExplicitGeometryPartition.CellArg, partition_cell_index: int)>#

Parameters:

args (CellArg)
partition_cell_index (int)

partition_cell_index = <Function ExplicitGeometryPartition__partition_cell_index(args: warp._src.fem.geometry.partition.ExplicitGeometryPartition.CellArg, cell_index: int)>#

Parameters:

args (CellArg)
cell_index (int)