warp.fem.SpaceRestriction#

class warp.fem.SpaceRestriction( space_partition, domain, device=None, temporary_store=None, )[source]#

Restriction of a space partition to a given GeometryDomain

Parameters:

space_partition (SpacePartition)
domain (GeometryDomain)
temporary_store (TemporaryStore | None)

__init__( space_partition, domain, device=None, temporary_store=None, )[source]#

Parameters:

space_partition (SpacePartition)
domain (GeometryDomain)
temporary_store (TemporaryStore | None)

Methods

`__init__`(space_partition, domain[, device, ...])
`fill_node_arg`(arg, device)
`node_arg_value`(device)
`node_count`()	Upper bound for the node count, use node_count_sync to get the actual value
`node_count_sync`()	Ensures that the node count is synchronized with the device and returns it
`node_partition_indices`()
`partition_element_offsets`()
`rebuild`([device, temporary_store])
`total_node_element_count`()

Attributes

`NodeArg`
`node_element_index`
`node_element_range`
`node_partition_index`
`node_partition_index_from_element_offset`
`space_partition`	Partition of the function space nodes that will be restricted
`space_topology`	Topology of the full function space
`domain`	Domain to which the space partition is being restricted

space_partition: SpacePartition#: Partition of the function space nodes that will be restricted

space_topology: SpaceTopology#: Topology of the full function space

domain: GeometryDomain#: Domain to which the space partition is being restricted

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

Parameters:

device (Device | str | None)
temporary_store (TemporaryStore | None)

node_count_sync()[source]#

Ensures that the node count is synchronized with the device and returns it

Return type:: int

node_count()[source]#

Upper bound for the node count, use node_count_sync to get the actual value

Return type:: int

partition_element_offsets()[source]#

node_partition_indices()[source]#

total_node_element_count()[source]#

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

node_arg_value(device)[source]#

fill_node_arg(arg, device)[source]#

Parameters:: arg (NodeArg)

node_partition_index = <Function SpaceRestriction__node_partition_index(args: warp._src.fem.space.restriction.SpaceRestriction.NodeArg, restriction_node_index: int)>#

Parameters:

args (NodeArg)
restriction_node_index (int)

node_partition_index_from_element_offset = <Function SpaceRestriction__node_partition_index_from_element_offset(args: warp._src.fem.space.restriction.SpaceRestriction.NodeArg, element_offset: int)>#

Parameters:

args (NodeArg)
element_offset (int)

node_element_range = <Function SpaceRestriction__node_element_range(args: warp._src.fem.space.restriction.SpaceRestriction.NodeArg, partition_node_index: int)>#

Parameters:

args (NodeArg)
partition_node_index (int)

node_element_index = <Function SpaceRestriction__node_element_index(args: warp._src.fem.space.restriction.SpaceRestriction.NodeArg, node_element_offset: int)>#

Parameters:

args (NodeArg)
node_element_offset (int)