warp.fem.PicQuadrature#

class warp.fem.PicQuadrature(
domain,
positions,
measures=None,
requires_grad=False,
max_dist=0.0,
temporary_store=None,
)[source]#

Particle-based quadrature formula, using a global set of points unevenly spread out over geometry elements.

Useful for Particle-In-Cell and derived methods.

Parameters:

  • domain (GeometryDomain) – Underlying domain for the quadrature

  • positions (wp.array(dtype=wp.vecXd) | tuple[wp.array(dtype=ElementIndex), wp.array(dtype=Coords)]) – Either an array containing the world positions of all particles, or a tuple of arrays containing the cell indices and coordinates for each particle.

  • measures (wp.array(dtype=float) | None) – Array containing the measure (area/volume) of each particle, used to defined the integration weights. If None, defaults to the cell measure divided by the number of particles in the cell.

  • max_dist (float) – When providing world positions that fall outside of the domain’s geometry partition, maximum distance to look up for embedding cells

  • requires_grad (bool) – Whether gradients should be allocated for the computed quantities

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

__init__(
domain,
positions,
measures=None,
requires_grad=False,
max_dist=0.0,
temporary_store=None,
)[source]#
Parameters:

  • domain (GeometryDomain)

  • positions (wp.array(dtype=wp.vecXd) | tuple[wp.array(dtype=ElementIndex), wp.array(dtype=Coords)])

  • measures (wp.array(dtype=float) | None)

  • requires_grad (bool)

  • max_dist (float)

  • temporary_store (TemporaryStore)

Methods

__init__(domain, positions[, measures, ...])

active_cell_count()

Number of cells containing at least one particle

arg_value(device)

element_index_arg_value(device)

evaluation_point_count()

Number of quadrature points that needs to be evaluated, mostly for internal purposes.

fill_arg(args, device)

Fill the quadrature argument structure for device functions.

fill_element_mask(mask)

Fills a mask array such that all non-empty elements are set to 1, all empty elements to zero.

max_points_per_element()

Maximum number of quadrature points per element.

total_point_count()

Total number of quadrature points.

Attributes

Arg

ElementIndexArg

Mapping from evaluation point indices to element indices.

domain

Domain over which this quadrature is defined

evaluation_point_element_index

name

Unique name of the quadrature rule.

point_coords

point_count

point_evaluation_index

point_index

point_weight

property name[source]#

Unique name of the quadrature rule.

property domain[source]#

Domain over which this quadrature is defined

Arg = <warp._src.codegen.Struct object>#
fill_arg(args, device)[source]#

Fill the quadrature argument structure for device functions.

Parameters:

args (Arg)

total_point_count()[source]#

Total number of quadrature points.

active_cell_count()[source]#

Number of cells containing at least one particle

max_points_per_element()[source]#

Maximum number of quadrature points per element.

point_count = <Function PicQuadrature__point_count(elt_arg: typing.Any, qp_arg: warp._src.fem.quadrature.pic_quadrature.PicQuadrature.Arg, domain_element_index: int, element_index: int)>#
Parameters:

  • elt_arg (Any)

  • qp_arg (Arg)

  • domain_element_index (int)

  • element_index (int)

point_coords = <Function PicQuadrature__point_coords(elt_arg: typing.Any, qp_arg: warp._src.fem.quadrature.pic_quadrature.PicQuadrature.Arg, domain_element_index: int, element_index: int, index: int)>#
Parameters:

  • elt_arg (Any)

  • qp_arg (Arg)

  • domain_element_index (int)

  • element_index (int)

  • index (int)

point_weight = <Function PicQuadrature__point_weight(elt_arg: typing.Any, qp_arg: warp._src.fem.quadrature.pic_quadrature.PicQuadrature.Arg, domain_element_index: int, element_index: int, index: int)>#
Parameters:

  • elt_arg (Any)

  • qp_arg (Arg)

  • domain_element_index (int)

  • element_index (int)

  • index (int)

point_index = <Function PicQuadrature__point_index(elt_arg: typing.Any, qp_arg: warp._src.fem.quadrature.pic_quadrature.PicQuadrature.Arg, domain_element_index: int, element_index: int, index: int)>#
Parameters:

  • elt_arg (Any)

  • qp_arg (Arg)

  • domain_element_index (int)

  • element_index (int)

  • index (int)

point_evaluation_index = <Function PicQuadrature__point_evaluation_index(elt_arg: typing.Any, qp_arg: warp._src.fem.quadrature.pic_quadrature.PicQuadrature.Arg, domain_element_index: int, element_index: int, index: int)>#
Parameters:

  • elt_arg (Any)

  • qp_arg (Arg)

  • domain_element_index (int)

  • element_index (int)

  • index (int)

fill_element_mask(mask)[source]#

Fills a mask array such that all non-empty elements are set to 1, all empty elements to zero.

Parameters:

mask (array(ndim=1, dtype=int32)) – Int warp array with size at least equal to self.domain.geometry_element_count()