autopdex.models.mixed_reference_domain_residual

autopdex.models.mixed_reference_domain_residual(integrand_fun, ansatz_fun, ref_int_coor, ref_int_weights, mapping_key)[source]

Constructs a multi-field ‘user residual’ for integration of a weak form in the reference configuration of finite elements. Works for DOFs as a dict of jnp.ndarrays.

Parameters:

  • integrand_fun (callable) – Function that evaluates the integrand given the integration point, trial ansatz, settings, static settings, and element number.

  • ansatz_fun (dict of callables) – Functions that constructs the ansatz functions for each field. Keywords have to match with those defined in the DOFs dictionary.

  • ref_int_coor (jnp.ndarray) – Reference integration coordinates for the isoparametric element.

  • ref_int_weights (jnp.ndarray) – Reference integration weights for the isoparametric element.

  • mapping_key (string) – Which solution space to use for the isoparametric mapping.

Returns:

function

A function that evaluates the integrand for the isoparametric element.

The returned function has the following parameters:

  • fI (dict of jnp.ndarray): Element nodal values.

  • xI (dict of jnp.ndarray): Element nodal coordinates.

  • elem_number (int): Element number.

  • settings (dict): Settings for the computation.

  • static_settings (dict): Static settings for the computation.

Notes

This models works for DOFs as a dict of jnp.ndarrays.