autopdex.models.mixed_reference_domain_residual_time

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

Constructs a multi-field ‘user residual’ for time-dependent weak forms.

Parameters:

  • integrand_fun (callable) – Function evaluating the integrand given: x_int, trial_ansatz (callable with args = (x, t)), test_ansatz (callable with args = (x,)), settings, static_settings, elem_number, set.

  • ansatz_fun (dict of callables) – Each function must have signature: f(x, t, xI, fI, settings, is_trial, n_dim)

  • ref_int_coor (jnp.ndarray) – Reference integration coordinates.

  • ref_int_weights (jnp.ndarray) – Integration weights.

  • mapping_key (str) – Key for the isoparametric mapping.

Returns:

user_residual(fI, xI, elem_number, settings, static_settings, set) that computes the element residual.

Return type:

A function with signature