ufl.algorithms package¶
Submodules¶
ufl.algorithms.ad module¶
Front-end for AD routines.
-
ufl.algorithms.ad.
expand_derivatives
(form, **kwargs)¶ Expand all derivatives of expr.
In the returned expression g which is mathematically equivalent to expr, there are no VariableDerivative or CoefficientDerivative objects left, and Grad objects have been propagated to Terminal nodes.
ufl.algorithms.analysis module¶
Utility algorithms for inspection of and information extraction from UFL objects in various ways.
-
ufl.algorithms.analysis.
extract_arguments
(a)¶ Build a sorted list of all arguments in a, which can be a Form, Integral or Expr.
-
ufl.algorithms.analysis.
extract_arguments_and_coefficients
(a)¶ Build two sorted lists of all arguments and coefficients in a, which can be a Form, Integral or Expr.
-
ufl.algorithms.analysis.
extract_coefficients
(a)¶ Build a sorted list of all coefficients in a, which can be a Form, Integral or Expr.
-
ufl.algorithms.analysis.
extract_constants
(a)¶ Build a sorted list of all constants in a
-
ufl.algorithms.analysis.
extract_elements
(form)¶ Build sorted tuple of all elements used in form.
-
ufl.algorithms.analysis.
extract_sub_elements
(elements)¶ Build sorted tuple of all sub elements (including parent element).
-
ufl.algorithms.analysis.
extract_type
(a, ufl_type)¶ Build a set of all objects of class ufl_type found in a. The argument a can be a Form, Integral or Expr.
-
ufl.algorithms.analysis.
extract_unique_elements
(form)¶ Build sorted tuple of all unique elements used in form.
-
ufl.algorithms.analysis.
has_exact_type
(a, ufl_type)¶ Return if an object of class ufl_type can be found in a. The argument a can be a Form, Integral or Expr.
-
ufl.algorithms.analysis.
has_type
(a, ufl_type)¶ Return if an object of class ufl_type can be found in a. The argument a can be a Form, Integral or Expr.
-
ufl.algorithms.analysis.
sort_elements
(elements)¶ Sort elements so that any sub elements appear before the corresponding mixed elements. This is useful when sub elements need to be defined before the corresponding mixed elements.
The ordering is based on sorting a directed acyclic graph.
-
ufl.algorithms.analysis.
unique_tuple
(objects)¶ Return tuple of unique objects, preserving initial ordering.
ufl.algorithms.apply_algebra_lowering module¶
Algorithm for expanding compound expressions into equivalent representations using basic operators.
-
class
ufl.algorithms.apply_algebra_lowering.
LowerCompoundAlgebra
¶ Bases:
ufl.corealg.multifunction.MultiFunction
Expands high level compound operators (e.g. inner) to equivalent representations using basic operators (e.g. index notation).
-
altenative_dot
(o, a, b)¶
-
alternative_inner
(o, a, b)¶
-
cofactor
(o, A)¶
-
cross
(o, a, b)¶
-
curl
(o, a)¶
-
determinant
(o, A)¶
-
deviatoric
(o, A)¶
-
div
(o, a)¶
-
dot
(o, a, b)¶
-
expr
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouched
as a default rule.
-
inner
(o, a, b)¶
-
inverse
(o, A)¶
-
nabla_div
(o, a)¶
-
nabla_grad
(o, a)¶
-
outer
(o, a, b)¶
-
skew
(o, A)¶
-
sym
(o, A)¶
-
trace
(o, A)¶
-
transposed
(o, A)¶
-
-
ufl.algorithms.apply_algebra_lowering.
apply_algebra_lowering
(expr)¶ Expands high level compound operators (e.g. inner) to equivalent representations using basic operators (e.g. index notation).
ufl.algorithms.apply_derivatives module¶
This module contains the apply_derivatives algorithm which computes the derivatives of a form of expression.
-
class
ufl.algorithms.apply_derivatives.
CoordinateDerivativeRuleDispatcher
¶ Bases:
ufl.corealg.multifunction.MultiFunction
-
coefficient_derivative
(o)¶
-
coordinate_derivative
(o)¶
-
derivative
(o)¶
-
expr
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouched
as a default rule.
-
grad
(o)¶
-
reference_grad
(o)¶
-
terminal
(o)¶
-
-
class
ufl.algorithms.apply_derivatives.
CoordinateDerivativeRuleset
(coefficients, arguments, coefficient_derivatives)¶ Bases:
ufl.algorithms.apply_derivatives.GenericDerivativeRuleset
Apply AFD (Automatic Functional Differentiation) to expression.
Implements rules for the Gateaux derivative D_w[v](…) defined as
D_w[v](e) = d/dtau e(w+tau v)|tau=0where ‘e’ is a ufl form after pullback and w is a SpatialCoordinate.
-
argument
(o)¶ Return a zero with the right shape for terminals independent of differentiation variable.
-
coefficient
(o)¶
-
geometric_quantity
(o)¶ Return a zero with the right shape for terminals independent of differentiation variable.
-
grad
(o)¶
-
jacobian
(o)¶
-
reference_grad
(g)¶
-
reference_value
(o)¶
-
spatial_coordinate
(o)¶
-
-
class
ufl.algorithms.apply_derivatives.
DerivativeRuleDispatcher
¶ Bases:
ufl.corealg.multifunction.MultiFunction
-
coefficient_derivative
(o, f, dummy_w, dummy_v, dummy_cd)¶
-
coordinate_derivative
(o, f, dummy_w, dummy_v, dummy_cd)¶
-
derivative
(o)¶
-
expr
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouched
as a default rule.
-
grad
(o, f)¶
-
indexed
(o, Ap, ii)¶
-
reference_grad
(o, f)¶
-
terminal
(o)¶
-
variable_derivative
(o, f, dummy_v)¶
-
-
class
ufl.algorithms.apply_derivatives.
GateauxDerivativeRuleset
(coefficients, arguments, coefficient_derivatives)¶ Bases:
ufl.algorithms.apply_derivatives.GenericDerivativeRuleset
Apply AFD (Automatic Functional Differentiation) to expression.
Implements rules for the Gateaux derivative D_w[v](…) defined as
D_w[v](e) = d/dtau e(w+tau v)|tau=0-
argument
(o)¶ Return a zero with the right shape for terminals independent of differentiation variable.
-
cell_avg
(o, fp)¶
-
coefficient
(o)¶
-
coordinate_derivative
(o)¶
-
facet_avg
(o, fp)¶
-
geometric_quantity
(o)¶ Return a zero with the right shape for terminals independent of differentiation variable.
-
grad
(g)¶
-
reference_grad
(o)¶
-
reference_value
(o)¶
-
-
class
ufl.algorithms.apply_derivatives.
GenericDerivativeRuleset
(var_shape)¶ Bases:
ufl.corealg.multifunction.MultiFunction
-
abs
(o, df)¶
-
acos
(o, fp)¶
-
asin
(o, fp)¶
-
atan
(o, fp)¶
-
atan_2
(o, fp, gp)¶
-
bessel_i
(o, nup, fp)¶
-
bessel_j
(o, nup, fp)¶
-
bessel_k
(o, nup, fp)¶
-
bessel_y
(o, nup, fp)¶
-
binary_condition
(o, dl, dr)¶
-
cell_avg
(o)¶
-
component_tensor
(o, Ap, ii)¶
-
conditional
(o, unused_dc, dt, df)¶
-
conj
(o, df)¶
-
constant
(o)¶ Return a zero with the right shape for terminals independent of differentiation variable.
-
constant_value
(o)¶ Return a zero with the right shape for terminals independent of differentiation variable.
-
cos
(o, fp)¶
-
cosh
(o, fp)¶
-
derivative
(o)¶
-
division
(o, fp, gp)¶
-
erf
(o, fp)¶
-
exp
(o, fp)¶
-
expr
(o)¶ Trigger error for types with missing handlers.
-
facet_avg
(o)¶
-
fixme
(o)¶
-
form_argument
(o)¶
-
geometric_quantity
(o)¶
-
grad
(o)¶
-
imag
(o, df)¶
-
independent_operator
(o)¶ Return a zero with the right shape and indices for operators independent of differentiation variable.
-
independent_terminal
(o)¶ Return a zero with the right shape for terminals independent of differentiation variable.
-
index_sum
(o, Ap, i)¶
-
indexed
(o, Ap, ii)¶
-
label
(o)¶ Labels and indices are not differentiable. It’s convenient to return the non-differentiated object.
-
list_tensor
(o, *dops)¶
-
ln
(o, fp)¶
-
math_function
(o, df)¶
-
max_value
(o, df, dg)¶
-
min_value
(o, df, dg)¶
-
multi_index
(o)¶ Labels and indices are not differentiable. It’s convenient to return the non-differentiated object.
-
non_differentiable_terminal
(o)¶ Labels and indices are not differentiable. It’s convenient to return the non-differentiated object.
-
not_condition
(o, c)¶
-
override
(o)¶
-
power
(o, fp, gp)¶
-
product
(o, da, db)¶
-
real
(o, df)¶
-
restricted
(o, fp)¶
-
sin
(o, fp)¶
-
sinh
(o, fp)¶
-
sqrt
(o, fp)¶
-
sum
(o, da, db)¶
-
tan
(o, fp)¶
-
tanh
(o, fp)¶
-
unexpected
(o)¶
-
variable
(o, df, unused_l)¶
-
-
class
ufl.algorithms.apply_derivatives.
GradRuleset
(geometric_dimension)¶ Bases:
ufl.algorithms.apply_derivatives.GenericDerivativeRuleset
-
argument
(o)¶
-
cell_avg
(o)¶ Return a zero with the right shape and indices for operators independent of differentiation variable.
-
cell_coordinate
(o)¶ dX/dx = inv(dx/dX) = inv(J) = K
-
coefficient
(o)¶
-
facet_avg
(o)¶ Return a zero with the right shape and indices for operators independent of differentiation variable.
-
geometric_quantity
(o)¶ Default for geometric quantities is dg/dx = 0 if piecewise constant, otherwise keep Grad(g). Override for specific types if other behaviour is needed.
-
grad
(o)¶ Represent grad(grad(f)) as Grad(Grad(f)).
-
jacobian_inverse
(o)¶
-
reference_grad
(o)¶
-
reference_value
(o)¶
-
spatial_coordinate
(o)¶ dx/dx = I
-
-
class
ufl.algorithms.apply_derivatives.
ReferenceGradRuleset
(topological_dimension)¶ Bases:
ufl.algorithms.apply_derivatives.GenericDerivativeRuleset
-
argument
(o)¶
-
cell_avg
(o)¶ Return a zero with the right shape and indices for operators independent of differentiation variable.
-
cell_coordinate
(o)¶ dX/dX = I
-
coefficient
(o)¶
-
facet_avg
(o)¶ Return a zero with the right shape and indices for operators independent of differentiation variable.
-
geometric_quantity
(o)¶ dg/dX = 0 if piecewise constant, otherwise ReferenceGrad(g)
-
grad
(o)¶
-
reference_grad
(o)¶ Represent ref_grad(ref_grad(f)) as RefGrad(RefGrad(f)).
-
reference_value
(o)¶
-
spatial_coordinate
(o)¶ dx/dX = J
-
-
class
ufl.algorithms.apply_derivatives.
VariableRuleset
(var)¶ Bases:
ufl.algorithms.apply_derivatives.GenericDerivativeRuleset
-
argument
(o)¶ Return a zero with the right shape for terminals independent of differentiation variable.
-
cell_avg
(o)¶ Return a zero with the right shape and indices for operators independent of differentiation variable.
-
coefficient
(o)¶ df/dv = Id if v is f else 0.
Note that if v = variable(f), df/dv is still 0, but if v == f, i.e. isinstance(v, Coefficient) == True, then df/dv == df/df = Id.
-
facet_avg
(o)¶ Return a zero with the right shape and indices for operators independent of differentiation variable.
-
geometric_quantity
(o)¶ Return a zero with the right shape for terminals independent of differentiation variable.
-
grad
(o)¶ Variable derivative of a gradient of a terminal must be 0.
-
reference_grad
(o)¶ Variable derivative of a gradient of a terminal must be 0.
-
reference_value
(o)¶
-
variable
(o, df, l)¶
-
-
ufl.algorithms.apply_derivatives.
apply_coordinate_derivatives
(expression)¶
-
ufl.algorithms.apply_derivatives.
apply_derivatives
(expression)¶
ufl.algorithms.apply_function_pullbacks module¶
Algorithm for replacing gradients in an expression with reference gradients and coordinate mappings.
-
class
ufl.algorithms.apply_function_pullbacks.
FunctionPullbackApplier
¶ Bases:
ufl.corealg.multifunction.MultiFunction
-
expr
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouched
as a default rule.
-
form_argument
(o)¶
-
terminal
(t)¶
-
-
ufl.algorithms.apply_function_pullbacks.
apply_function_pullbacks
(expr)¶ Change representation of coefficients and arguments in expression by applying Piola mappings where applicable and representing all form arguments in reference value.
- @param expr:
- An Expr.
-
ufl.algorithms.apply_function_pullbacks.
apply_single_function_pullbacks
(g)¶
-
ufl.algorithms.apply_function_pullbacks.
create_nested_lists
(shape)¶
-
ufl.algorithms.apply_function_pullbacks.
reshape_to_nested_list
(components, shape)¶
-
ufl.algorithms.apply_function_pullbacks.
sub_elements_with_mappings
(element)¶ Return an ordered list of the largest subelements that have a defined mapping.
ufl.algorithms.apply_geometry_lowering module¶
Algorithm for lowering abstractions of geometric types.
This means replacing high-level types with expressions of mostly the Jacobian and reference cell data.
-
class
ufl.algorithms.apply_geometry_lowering.
GeometryLoweringApplier
(preserve_types=())¶ Bases:
ufl.corealg.multifunction.MultiFunction
-
cell_coordinate
(o)¶
-
cell_diameter
(o)¶
-
cell_normal
(o)¶
-
cell_volume
(o)¶
-
circumradius
(o)¶
-
expr
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouched
as a default rule.
-
facet_area
(o)¶
-
facet_cell_coordinate
(o)¶
-
facet_jacobian
(o)¶
-
facet_jacobian_determinant
(o)¶
-
facet_jacobian_inverse
(o)¶
-
facet_normal
(o)¶
-
jacobian
(o)¶
-
jacobian_determinant
(o)¶
-
jacobian_inverse
(o)¶
-
max_cell_edge_length
(o)¶
-
max_facet_edge_length
(o)¶
-
min_cell_edge_length
(o)¶
-
min_facet_edge_length
(o)¶
-
spatial_coordinate
(o)¶
-
terminal
(t)¶
-
-
ufl.algorithms.apply_geometry_lowering.
apply_geometry_lowering
(form, preserve_types=())¶ Change GeometricQuantity objects in expression to the lowest level GeometricQuantity objects.
Assumes the expression is preprocessed or at least that derivatives have been expanded.
- @param form:
- An Expr or Form.
ufl.algorithms.apply_integral_scaling module¶
Algorithm for replacing gradients in an expression with reference gradients and coordinate mappings.
-
ufl.algorithms.apply_integral_scaling.
apply_integral_scaling
(form)¶ Multiply integrands by a factor to scale the integral to reference frame.
-
ufl.algorithms.apply_integral_scaling.
compute_integrand_scaling_factor
(integral)¶ Change integrand geometry to the right representations.
ufl.algorithms.apply_restrictions module¶
This module contains the apply_restrictions algorithm which propagates restrictions in a form towards the terminals.
-
class
ufl.algorithms.apply_restrictions.
DefaultRestrictionApplier
(side=None)¶ Bases:
ufl.corealg.multifunction.MultiFunction
-
derivative
(o)¶
-
facet_area
(o)¶ Restrict a continuous quantity to default side if no current restriction is set.
-
facet_jacobian
(o)¶ Restrict a continuous quantity to default side if no current restriction is set.
-
facet_jacobian_determinant
(o)¶ Restrict a continuous quantity to default side if no current restriction is set.
-
facet_jacobian_inverse
(o)¶ Restrict a continuous quantity to default side if no current restriction is set.
-
facet_origin
(o)¶ Restrict a continuous quantity to default side if no current restriction is set.
-
max_facet_edge_length
(o)¶ Restrict a continuous quantity to default side if no current restriction is set.
-
min_facet_edge_length
(o)¶ Restrict a continuous quantity to default side if no current restriction is set.
-
operator
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouched
as a default rule.
-
restricted
(o)¶
-
spatial_coordinate
(o)¶ Restrict a continuous quantity to default side if no current restriction is set.
-
terminal
(o)¶
-
-
class
ufl.algorithms.apply_restrictions.
RestrictionPropagator
(side=None)¶ Bases:
ufl.corealg.multifunction.MultiFunction
-
argument
(o)¶ Restrict a discontinuous quantity to current side, require a side to be set.
-
coefficient
(o)¶ Allow coefficients to be unrestricted (apply default if so) if the values are fully continuous across the facet.
-
constant
(o)¶ Ignore current restriction, quantity is independent of side also from a computational point of view.
-
constant_value
(o)¶ Ignore current restriction, quantity is independent of side also from a computational point of view.
-
facet_coordinate
(o)¶ Ignore current restriction, quantity is independent of side also from a computational point of view.
-
facet_normal
(o)¶
-
geometric_cell_quantity
(o)¶ Restrict a discontinuous quantity to current side, require a side to be set.
-
geometric_facet_quantity
(o)¶ Restrict a discontinuous quantity to current side, require a side to be set.
-
grad
(o)¶ Restrict a discontinuous quantity to current side, require a side to be set.
-
label
(o)¶ Ignore current restriction, quantity is independent of side also from a computational point of view.
-
multi_index
(o)¶ Ignore current restriction, quantity is independent of side also from a computational point of view.
-
operator
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouched
as a default rule.
-
quadrature_weight
(o)¶ Ignore current restriction, quantity is independent of side also from a computational point of view.
-
reference_cell_volume
(o)¶ Ignore current restriction, quantity is independent of side also from a computational point of view.
-
reference_facet_volume
(o)¶ Ignore current restriction, quantity is independent of side also from a computational point of view.
-
reference_value
(o)¶ Reference value of something follows same restriction rule as the underlying object.
-
restricted
(o)¶ When hitting a restricted quantity, visit child with a separate restriction algorithm.
-
terminal
(o)¶
-
variable
(o, op, label)¶ Strip variable.
-
-
ufl.algorithms.apply_restrictions.
apply_default_restrictions
(expression)¶ Some terminals can be restricted from either side.
This applies a default restriction to such terminals if unrestricted.
-
ufl.algorithms.apply_restrictions.
apply_restrictions
(expression)¶ Propagate restriction nodes to wrap differential terminals directly.
ufl.algorithms.balancing module¶
-
class
ufl.algorithms.balancing.
BalanceModifiers
¶ Bases:
ufl.corealg.multifunction.MultiFunction
-
cell_avg
(expr, *ops)¶
-
expr
(expr, *ops)¶ Trigger error for types with missing handlers.
-
facet_avg
(expr, *ops)¶
-
grad
(expr, *ops)¶
-
negative_restricted
(expr, *ops)¶
-
positive_restricted
(expr, *ops)¶
-
reference_grad
(expr, *ops)¶
-
reference_value
(expr, *ops)¶
-
terminal
(expr)¶
-
-
ufl.algorithms.balancing.
balance_modified_terminal
(expr)¶
-
ufl.algorithms.balancing.
balance_modifiers
(expr)¶
ufl.algorithms.change_to_reference module¶
Algorithm for replacing gradients in an expression with reference gradients and coordinate mappings.
-
class
ufl.algorithms.change_to_reference.
NEWChangeToReferenceGrad
¶ Bases:
ufl.corealg.multifunction.MultiFunction
-
cell_avg
(o, *dummy_ops)¶
-
coefficient_derivative
(o, *dummy_ops)¶
-
expr
(o, *ops)¶ Trigger error for types with missing handlers.
-
facet_avg
(o, *dummy_ops)¶
-
form_argument
(t)¶
-
geometric_quantity
(t)¶
-
grad
(o, *dummy_ops)¶ Store modifier state.
-
reference_grad
(o, *dummy_ops)¶
-
restricted
(o, *dummy_ops)¶ Store modifier state.
-
terminal
(o)¶
-
-
class
ufl.algorithms.change_to_reference.
OLDChangeToReferenceGrad
¶ Bases:
ufl.corealg.multifunction.MultiFunction
-
coefficient_derivative
(o)¶
-
expr
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouched
as a default rule.
-
grad
(o)¶
-
reference_grad
(o)¶
-
terminal
(o)¶
-
-
ufl.algorithms.change_to_reference.
change_integrand_geometry_representation
(integrand, scale, integral_type)¶ Change integrand geometry to the right representations.
-
ufl.algorithms.change_to_reference.
change_to_reference_grad
(e)¶ Change Grad objects in expression to products of JacobianInverse and ReferenceGrad.
Assumes the expression is preprocessed or at least that derivatives have been expanded.
- @param e:
- An Expr or Form.
ufl.algorithms.check_arities module¶
-
class
ufl.algorithms.check_arities.
ArityChecker
(arguments)¶ Bases:
ufl.corealg.multifunction.MultiFunction
-
argument
(o)¶
-
cell_avg
(o, a)¶
-
component_tensor
(o, a, i)¶
-
conditional
(o, c, a, b)¶
-
conj
(o, a)¶
-
division
(o, a, b)¶
-
dot
(o, a, b)¶
-
expr
(o)¶
-
facet_avg
(o, a)¶
-
grad
(o, a)¶
-
index_sum
(o, a, i)¶
-
indexed
(o, a, i)¶
-
inner
(o, a, b)¶
-
linear_indexed_type
(o, a, i)¶
-
linear_operator
(o, a)¶
-
list_tensor
(o, *ops)¶
-
negative_restricted
(o, a)¶
-
nonlinear_operator
(o)¶
-
outer
(o, a, b)¶
-
positive_restricted
(o, a)¶
-
product
(o, a, b)¶
-
reference_grad
(o, a)¶
-
reference_value
(o, a)¶
-
sum
(o, a, b)¶
-
terminal
(o)¶
-
variable
(o, f, l)¶
-
-
exception
ufl.algorithms.check_arities.
ArityMismatch
¶ Bases:
ufl.log.UFLException
-
ufl.algorithms.check_arities.
check_form_arity
(form, arguments, complex_mode=False)¶
-
ufl.algorithms.check_arities.
check_integrand_arity
(expr, arguments, complex_mode=False)¶
ufl.algorithms.check_restrictions module¶
Algorithms related to restrictions.
-
class
ufl.algorithms.check_restrictions.
RestrictionChecker
(require_restriction)¶ Bases:
ufl.corealg.multifunction.MultiFunction
-
expr
(o)¶ Trigger error for types with missing handlers.
-
facet_normal
(o)¶
-
form_argument
(o)¶
-
restricted
(o)¶
-
-
ufl.algorithms.check_restrictions.
check_restrictions
(expression, require_restriction)¶ Check that types that must be restricted are restricted in expression.
ufl.algorithms.checks module¶
Functions to check the validity of forms.
-
ufl.algorithms.checks.
validate_form
(form)¶ Performs all implemented validations on a form. Raises exception if something fails.
ufl.algorithms.comparison_checker module¶
Algorithm to check for ‘comparison’ nodes in a form when the user is in ‘complex mode’
-
class
ufl.algorithms.comparison_checker.
CheckComparisons
¶ Bases:
ufl.corealg.multifunction.MultiFunction
Raises an error if comparisons are done with complex quantities.
If quantities are real, adds the Real operator to the compared quantities.
Terminals that are real are RealValue, Zero, and Argument (even in complex FEM, the basis functions are real) Operations that produce reals are Abs, Real, Imag. Terminals default to complex, and Sqrt, Pow (defensively) imply complex. Otherwise, operators preserve the type of their operands.
-
abs
(o, *ops)¶
-
compare
(o, *ops)¶
-
expr
(o, *ops)¶ Defaults expressions to complex unless they only act on real quantities. Overridden for specific operators.
Rebuilds objects if necessary.
-
ge
(o, *ops)¶
-
gt
(o, *ops)¶
-
imag
(o, *ops)¶
-
indexed
(o, expr, multiindex)¶
-
le
(o, *ops)¶
-
lt
(o, *ops)¶
-
max_value
(o, *ops)¶
-
min_value
(o, *ops)¶
-
power
(o, base, exponent)¶
-
real
(o, *ops)¶
-
sign
(o, *ops)¶
-
sqrt
(o, *ops)¶
-
terminal
(term, *ops)¶
-
-
exception
ufl.algorithms.comparison_checker.
ComplexComparisonError
¶ Bases:
Exception
-
ufl.algorithms.comparison_checker.
do_comparison_check
(form)¶ Raises an error if invalid comparison nodes exist
ufl.algorithms.compute_form_data module¶
This module provides the compute_form_data function which form compilers will typically call prior to code generation to preprocess/simplify a raw input form given by a user.
-
ufl.algorithms.compute_form_data.
attach_estimated_degrees
(form)¶ Attach estimated polynomial degree to a form’s integrals.
Parameters: form – The Form
to inspect.Returns: A new Form with estimate degrees attached.
-
ufl.algorithms.compute_form_data.
compute_form_data
(form, do_apply_function_pullbacks=False, do_apply_integral_scaling=False, do_apply_geometry_lowering=False, preserve_geometry_types=(), do_apply_default_restrictions=True, do_apply_restrictions=True, do_estimate_degrees=True, do_append_everywhere_integrals=True, complex_mode=False)¶
ufl.algorithms.coordinate_derivative_helpers module¶
This module provides the necessary tools to strip away and then reattach the coordinate derivatives at the right time point in compute_form_data.
-
class
ufl.algorithms.coordinate_derivative_helpers.
CoordinateDerivativeIsOutermostChecker
¶ Bases:
ufl.corealg.multifunction.MultiFunction
Traverses the tree to make sure that CoordinateDerivatives are only on the outside. The visitor returns False as long as no CoordinateDerivative has been seen.
-
coordinate_derivative
(o, expr, *_)¶
-
expr
(o, *operands)¶ If we have already seen a CoordinateDerivative, then no other expressions apart from more CoordinateDerivatives are allowed to wrap around it.
-
multi_index
(o)¶
-
terminal
(o)¶
-
-
ufl.algorithms.coordinate_derivative_helpers.
attach_coordinate_derivatives
(integral, coordinate_derivatives)¶
-
ufl.algorithms.coordinate_derivative_helpers.
strip_coordinate_derivatives
(integrals)¶
ufl.algorithms.domain_analysis module¶
Algorithms for building canonical data structure for integrals over subdomains.
-
class
ufl.algorithms.domain_analysis.
IntegralData
(domain, integral_type, subdomain_id, integrals, metadata)¶ Bases:
object
- Utility class with the members (domain, integral_type,
- subdomain_id, integrals, metadata)
where metadata is an empty dictionary that may be used for associating metadata with each object.
-
domain
¶
-
enabled_coefficients
¶
-
integral_coefficients
¶
-
integral_type
¶
-
integrals
¶
-
metadata
¶
-
subdomain_id
¶
-
ufl.algorithms.domain_analysis.
accumulate_integrands_with_same_metadata
(integrals)¶ - Taking input on the form:
- integrals = [integral0, integral1, …]
- Return result on the form:
- integrands_by_id = [(integrand0, metadata0),
- (integrand1, metadata1), …]
where integrand0 < integrand1 by the canonical ufl expression ordering criteria.
-
ufl.algorithms.domain_analysis.
build_integral_data
(integrals)¶ Build integral data given a list of integrals.
Parameters: integrals – An iterable of Integral
objects.Returns: A tuple of IntegralData
objects.The integrals you pass in here must have been rearranged and gathered (removing the “everywhere” subdomain_id. To do this, you should call
group_form_integrals()
.
-
ufl.algorithms.domain_analysis.
dicts_lt
(a, b)¶
-
ufl.algorithms.domain_analysis.
group_form_integrals
(form, domains, do_append_everywhere_integrals=True)¶ Group integrals by domain and type, performing canonical simplification.
Parameters: Returns: A new
Form
with gathered integrands.
-
ufl.algorithms.domain_analysis.
group_integrals_by_domain_and_type
(integrals, domains)¶ - Input:
- integrals: list of Integral objects domains: list of AbstractDomain objects from the parent Form
- Output:
- integrals_by_domain_and_type: dict: (domain, integral_type) -> list(Integral)
-
ufl.algorithms.domain_analysis.
integral_subdomain_ids
(integral)¶ Get a tuple of integer subdomains or a valid string subdomain from integral.
-
ufl.algorithms.domain_analysis.
rearrange_integrals_by_single_subdomains
(integrals, do_append_everywhere_integrals)¶ Rearrange integrals over multiple subdomains to single subdomain integrals.
- Input:
- integrals: list(Integral)
- Output:
- integrals: dict: subdomain_id -> list(Integral) (reconstructed with single subdomain_id)
-
ufl.algorithms.domain_analysis.
reconstruct_form_from_integral_data
(integral_data)¶
ufl.algorithms.elementtransformations module¶
This module provides helper functions to - FFC/DOLFIN adaptive chain, - UFL algorithms taking care of underspecified DOLFIN expressions.
-
ufl.algorithms.elementtransformations.
increase_order
(element)¶ Return element of same family, but a polynomial degree higher.
-
ufl.algorithms.elementtransformations.
tear
(element)¶ For a finite element, return the corresponding discontinuous element.
ufl.algorithms.estimate_degrees module¶
Algorithms for estimating polynomial degrees of expressions.
-
class
ufl.algorithms.estimate_degrees.
IrreducibleInt
¶ Bases:
int
Degree type used by quadrilaterals.
Unlike int, values of this type are not decremeneted by _reduce_degree.
-
class
ufl.algorithms.estimate_degrees.
SumDegreeEstimator
(default_degree, element_replace_map)¶ Bases:
ufl.corealg.multifunction.MultiFunction
This algorithm is exact for a few operators and heuristic for many.
-
abs
(v, a)¶ This is a heuristic, correct if there is no
-
argument
(v)¶ A form argument provides a degree depending on the element, or the default degree if the element has no degree.
-
atan_2
(v, a, b)¶ Using the heuristic degree(atan2(const,const)) == 0 degree(atan2(a,b)) == max(degree(a),degree(b))+2 which can be wildly inaccurate but at least gives a somewhat high integration degree.
-
bessel_function
(v, nu, x)¶ Using the heuristic degree(bessel_*(const)) == 0 degree(bessel_*(x)) == degree(x)+2 which can be wildly inaccurate but at least gives a somewhat high integration degree.
-
cell_avg
(v, a)¶ Cell average of a function is always cellwise constant.
-
cell_coordinate
(v)¶ A coordinate provides one additional degree.
-
coefficient
(v)¶ A form argument provides a degree depending on the element, or the default degree if the element has no degree.
-
cofactor
(v, *args)¶
-
component_tensor
(v, A, ii)¶
-
compound_derivative
(v, *args)¶
-
compound_tensor_operator
(v, *args)¶
-
condition
(v, *args)¶
-
conditional
(v, c, t, f)¶ Degree of condition does not influence degree of values which conditional takes. So heuristicaly taking max of true degree and false degree. This will be exact in cells where condition takes single value. For improving accuracy of quadrature near condition transition surface quadrature order must be adjusted manually.
-
conj
(v, a)¶
-
constant
(v)¶
-
constant_value
(v)¶ Constant values are constant.
-
coordinate_derivative
(v, integrand_degree, b, direction_degree, d)¶ We use the heuristic that a shape derivative in direction V introduces terms V and grad(V) into the integrand. Hence we add the degree of the deformation to the estimate.
-
cross
(v, *ops)¶
-
curl
(v, f)¶ Reduces the estimated degree by one; used when derivatives are taken. Does not reduce the degree when TensorProduct elements or quadrilateral elements are involved.
-
derivative
(v, *args)¶
-
determinant
(v, *args)¶
-
deviatoric
(v, *args)¶
-
div
(v, f)¶ Reduces the estimated degree by one; used when derivatives are taken. Does not reduce the degree when TensorProduct elements or quadrilateral elements are involved.
-
division
(v, *ops)¶ Using the sum here is a heuristic. Consider e.g. (x+1)/(x-1).
-
dot
(v, *ops)¶
-
expr
(v, *ops)¶ For most operators we take the max degree of its operands.
-
expr_list
(v, *o)¶
-
expr_mapping
(v, *o)¶
-
facet_avg
(v, a)¶ Facet average of a function is always cellwise constant.
-
geometric_quantity
(v)¶ Some geometric quantities are cellwise constant. Others are nonpolynomial and thus hard to estimate.
-
grad
(v, f)¶ Reduces the estimated degree by one; used when derivatives are taken. Does not reduce the degree when TensorProduct elements or quadrilateral elements are involved.
-
imag
(v, a)¶
-
index_sum
(v, A, ii)¶
-
indexed
(v, A, ii)¶
-
inner
(v, *ops)¶
-
inverse
(v, *args)¶
-
label
(v)¶
-
list_tensor
(v, *ops)¶
-
math_function
(v, a)¶ Using the heuristic degree(sin(const)) == 0 degree(sin(a)) == degree(a)+2 which can be wildly inaccurate but at least gives a somewhat high integration degree.
-
max_value
(v, l, r)¶ Same as conditional.
-
min_value
(v, l, r)¶ Same as conditional.
-
multi_index
(v)¶
-
nabla_div
(v, f)¶ Reduces the estimated degree by one; used when derivatives are taken. Does not reduce the degree when TensorProduct elements or quadrilateral elements are involved.
-
nabla_grad
(v, f)¶ Reduces the estimated degree by one; used when derivatives are taken. Does not reduce the degree when TensorProduct elements or quadrilateral elements are involved.
-
negative_restricted
(v, a)¶
-
outer
(v, *ops)¶
-
positive_restricted
(v, a)¶
-
power
(v, a, b)¶ If b is a positive integer: degree(a**b) == degree(a)*b otherwise use the heuristic degree(a**b) == degree(a) + 2
-
product
(v, *ops)¶
-
real
(v, a)¶
-
reference_curl
(v, f)¶ Reduces the estimated degree by one; used when derivatives are taken. Does not reduce the degree when TensorProduct elements or quadrilateral elements are involved.
-
reference_div
(v, f)¶ Reduces the estimated degree by one; used when derivatives are taken. Does not reduce the degree when TensorProduct elements or quadrilateral elements are involved.
-
reference_grad
(v, f)¶ Reduces the estimated degree by one; used when derivatives are taken. Does not reduce the degree when TensorProduct elements or quadrilateral elements are involved.
-
reference_value
(rv, f)¶
-
skew
(v, *args)¶
-
spatial_coordinate
(v)¶ A coordinate provides additional degrees depending on coordinate field of domain.
-
sum
(v, *ops)¶
-
sym
(v, *args)¶
-
trace
(v, *args)¶
-
transposed
(v, A)¶
-
variable
(v, e, l)¶
-
variable_derivative
(v, *args)¶
-
-
ufl.algorithms.estimate_degrees.
estimate_total_polynomial_degree
(e, default_degree=1, element_replace_map={})¶ Estimate total polynomial degree of integrand.
NB! Although some compound types are supported here, some derivatives and compounds must be preprocessed prior to degree estimation. In generic code, this algorithm should only be applied after preprocessing.
For coefficients defined on an element with unspecified degree (None), the degree is set to the given default degree.
ufl.algorithms.expand_compounds module¶
Algorithm for expanding compound expressions into equivalent representations using basic operators.
-
ufl.algorithms.expand_compounds.
expand_compounds
(e)¶
ufl.algorithms.expand_indices module¶
This module defines expression transformation utilities, for expanding free indices in expressions to explicit fixed indices only.
-
class
ufl.algorithms.expand_indices.
IndexExpander
¶ Bases:
ufl.algorithms.transformer.ReuseTransformer
…
-
component
()¶ Return current component tuple.
-
component_tensor
(x)¶
-
conditional
(x)¶
-
division
(x)¶
-
form_argument
(x)¶
-
grad
(x)¶
-
index_sum
(x)¶
-
indexed
(x)¶
-
list_tensor
(x)¶
-
multi_index
(x)¶
-
scalar_value
(x)¶
-
terminal
(x)¶ Always reuse Expr (ignore children)
-
zero
(x)¶
-
-
ufl.algorithms.expand_indices.
expand_indices
(e)¶
-
ufl.algorithms.expand_indices.
purge_list_tensors
(expr)¶ Get rid of all ListTensor instances by expanding expressions to use their components directly. Will usually increase the size of the expression.
ufl.algorithms.formdata module¶
FormData class easy for collecting of various data about a form.
-
class
ufl.algorithms.formdata.
ExprData
¶ Bases:
object
Class collecting various information extracted from a Expr by calling preprocess.
-
class
ufl.algorithms.formdata.
FormData
¶ Bases:
object
Class collecting various information extracted from a Form by calling preprocess.
ufl.algorithms.formfiles module¶
A collection of utility algorithms for handling UFL files.
-
class
ufl.algorithms.formfiles.
FileData
¶ Bases:
object
-
ufl.algorithms.formfiles.
execute_ufl_code
(uflcode, filename)¶
-
ufl.algorithms.formfiles.
interpret_ufl_namespace
(namespace)¶ Takes a namespace dict from an executed ufl file and converts it to a FileData object.
-
ufl.algorithms.formfiles.
load_forms
(filename)¶ Return a list of all forms in a file.
-
ufl.algorithms.formfiles.
load_ufl_file
(filename)¶ Load a .ufl file with elements, coefficients and forms.
-
ufl.algorithms.formfiles.
read_lines_decoded
(fn)¶
-
ufl.algorithms.formfiles.
read_ufl_file
(filename)¶ Read a .ufl file, handling file extension, file existance, and #include replacement.
-
ufl.algorithms.formfiles.
replace_include_statements
(lines)¶ Replace ‘#include foo.ufl’ statements with contents of foo.ufl.
ufl.algorithms.formsplitter module¶
Extract part of a form in a mixed FunctionSpace.
-
class
ufl.algorithms.formsplitter.
FormSplitter
¶ Bases:
ufl.corealg.multifunction.MultiFunction
-
argument
(obj)¶
-
expr
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouched
as a default rule.
-
multi_index
(obj)¶
-
split
(form, ix, iy=0)¶
-
-
ufl.algorithms.formsplitter.
extract_blocks
(form, i=None, j=None)¶
ufl.algorithms.formtransformations module¶
This module defines utilities for transforming complete Forms into new related Forms.
-
class
ufl.algorithms.formtransformations.
PartExtracter
(arguments)¶ Bases:
ufl.algorithms.transformer.Transformer
PartExtracter extracts those parts of a form that contain the given argument(s).
-
argument
(x)¶ Return itself unless itself provides too much.
-
cell_avg
(x, arg)¶ A linear operator with a single operand accepting arity > 0, providing whatever Argument its operand does.
-
component_tensor
(x)¶ Return parts of expression belonging to this indexed expression.
-
conj
(x, arg)¶ A linear operator with a single operand accepting arity > 0, providing whatever Argument its operand does.
-
division
(x)¶ Return parts_of_numerator/denominator.
-
dot
(x, *ops)¶ Note: Product is a visit-children-first handler. ops are the visited factors.
-
expr
(x)¶ The default is a nonlinear operator not accepting any Arguments among its children.
-
facet_avg
(x, arg)¶ A linear operator with a single operand accepting arity > 0, providing whatever Argument its operand does.
-
grad
(x, arg)¶ A linear operator with a single operand accepting arity > 0, providing whatever Argument its operand does.
-
imag
(x, arg)¶ A linear operator with a single operand accepting arity > 0, providing whatever Argument its operand does.
-
index_sum
(x)¶ Return parts of expression belonging to this indexed expression.
-
indexed
(x)¶ Return parts of expression belonging to this indexed expression.
-
inner
(x, *ops)¶ Note: Product is a visit-children-first handler. ops are the visited factors.
-
linear_indexed_type
(x)¶ Return parts of expression belonging to this indexed expression.
-
linear_operator
(x, arg)¶ A linear operator with a single operand accepting arity > 0, providing whatever Argument its operand does.
-
list_tensor
(x, *ops)¶
-
negative_restricted
(x, arg)¶ A linear operator with a single operand accepting arity > 0, providing whatever Argument its operand does.
-
outer
(x, *ops)¶ Note: Product is a visit-children-first handler. ops are the visited factors.
-
positive_restricted
(x, arg)¶ A linear operator with a single operand accepting arity > 0, providing whatever Argument its operand does.
-
product
(x, *ops)¶ Note: Product is a visit-children-first handler. ops are the visited factors.
-
real
(x, arg)¶ A linear operator with a single operand accepting arity > 0, providing whatever Argument its operand does.
-
sum
(x)¶ Return the terms that might eventually yield the correct parts(!)
The logic required for sums is a bit elaborate:
A sum may contain terms providing different arguments. We should return (a sum of) a suitable subset of these terms. Those should all provide the same arguments.
For each term in a sum, there are 2 simple possibilities:
1a) The relevant part of the term is zero -> skip. 1b) The term provides more arguments than we want -> skip
2) If all terms fall into the above category, we can just return zero.
Any remaining terms may provide exactly the arguments we want, or fewer. This is where things start getting interesting.
3) Bottom-line: if there are terms with providing different arguments – provide terms that contain the most arguments. If there are terms providing different sets of same size -> throw error (e.g. Argument(-1) + Argument(-2))
-
terminal
(x)¶ The default is a nonlinear operator not accepting any Arguments among its children.
-
variable
(x)¶ Return relevant parts of this variable.
-
-
ufl.algorithms.formtransformations.
compute_energy_norm
(form, coefficient)¶ Compute the a-norm of a Coefficient given a form a.
This works simply by replacing the two Arguments with a Coefficient on the same function space (element). The Form returned will thus be a functional with no Arguments, and one additional Coefficient at the end if no coefficient has been provided.
-
ufl.algorithms.formtransformations.
compute_form_action
(form, coefficient)¶ Compute the action of a form on a Coefficient.
This works simply by replacing the last Argument with a Coefficient on the same function space (element). The form returned will thus have one Argument less and one additional Coefficient at the end if no Coefficient has been provided.
-
ufl.algorithms.formtransformations.
compute_form_adjoint
(form, reordered_arguments=None)¶ Compute the adjoint of a bilinear form.
This works simply by swapping the number and part of the two arguments, but keeping their elements and places in the integrand expressions.
-
ufl.algorithms.formtransformations.
compute_form_arities
(form)¶ Return set of arities of terms present in form.
-
ufl.algorithms.formtransformations.
compute_form_functional
(form)¶ Compute the functional part of a form, that is the terms independent of Arguments.
(Used for testing, not sure if it’s useful for anything?)
-
ufl.algorithms.formtransformations.
compute_form_lhs
(form)¶ Compute the left hand side of a form.
a = u*v*dx + f*v*dx a = lhs(a) -> u*v*dx
-
ufl.algorithms.formtransformations.
compute_form_rhs
(form)¶ Compute the right hand side of a form.
a = u*v*dx + f*v*dx L = rhs(a) -> -f*v*dx
-
ufl.algorithms.formtransformations.
compute_form_with_arity
(form, arity, arguments=None)¶ Compute parts of form of given arity.
-
ufl.algorithms.formtransformations.
zero_expr
(e)¶
ufl.algorithms.map_integrands module¶
Basic algorithms for applying functions to subexpressions.
-
ufl.algorithms.map_integrands.
map_integrand_dags
(function, form, only_integral_type=None, compress=True)¶
-
ufl.algorithms.map_integrands.
map_integrands
(function, form, only_integral_type=None)¶ Apply transform(expression) to each integrand expression in form, or to form if it is an Expr.
ufl.algorithms.multifunction module¶
ufl.algorithms.remove_complex_nodes module¶
Algorithm for removing conj, real, and imag nodes from a form for when the user is in ‘real mode’
-
class
ufl.algorithms.remove_complex_nodes.
ComplexNodeRemoval
¶ Bases:
ufl.corealg.multifunction.MultiFunction
Replaces complex operator nodes with their children
-
conj
(o, a)¶
-
expr
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouched
as a default rule.
-
imag
(o, a)¶
-
real
(o, a)¶
-
terminal
(t, *ops)¶
-
-
ufl.algorithms.remove_complex_nodes.
remove_complex_nodes
(expr)¶ Replaces complex operator nodes with their children. This is called during compute_form_data if the compiler wishes to compile real-valued forms. In essence this strips all trace of complex support from the preprocessed form.
ufl.algorithms.renumbering module¶
Algorithms for renumbering of counted objects, currently variables and indices.
-
class
ufl.algorithms.renumbering.
IndexRenumberingTransformer
¶ Bases:
ufl.algorithms.renumbering.VariableRenumberingTransformer
This is a poorly designed algorithm. It is used in some tests, please do not use for anything else.
-
index
(o)¶
-
multi_index
(o)¶
-
zero
(o)¶
-
-
class
ufl.algorithms.renumbering.
VariableRenumberingTransformer
¶ Bases:
ufl.algorithms.transformer.ReuseTransformer
-
variable
(o)¶
-
-
ufl.algorithms.renumbering.
renumber_indices
(expr)¶
ufl.algorithms.replace module¶
Algorithm for replacing terminals in an expression.
-
class
ufl.algorithms.replace.
Replacer
(mapping)¶ Bases:
ufl.corealg.multifunction.MultiFunction
-
coefficient_derivative
(o)¶
-
expr
(o, *args)¶ Trigger error for types with missing handlers.
-
-
ufl.algorithms.replace.
replace
(e, mapping)¶ Replace subexpressions in expression.
- @param e:
- An Expr or Form.
- @param mapping:
- A dict with from:to replacements to perform.
ufl.algorithms.signature module¶
Signature computation for forms.
-
ufl.algorithms.signature.
compute_expression_hashdata
(expression, terminal_hashdata)¶
-
ufl.algorithms.signature.
compute_expression_signature
(expr, renumbering)¶
-
ufl.algorithms.signature.
compute_form_signature
(form, renumbering)¶
-
ufl.algorithms.signature.
compute_multiindex_hashdata
(expr, index_numbering)¶
-
ufl.algorithms.signature.
compute_terminal_hashdata
(expressions, renumbering)¶
ufl.algorithms.transformer module¶
This module defines the Transformer base class and some basic specializations to further base other algorithms upon, as well as some utilities for easier application of such algorithms.
-
class
ufl.algorithms.transformer.
CopyTransformer
(variable_cache=None)¶ Bases:
ufl.algorithms.transformer.Transformer
-
expr
(o, *operands)¶ Always reconstruct expr.
-
terminal
(o)¶ Always reuse Expr (ignore children)
-
variable
(o)¶
-
-
class
ufl.algorithms.transformer.
ReuseTransformer
(variable_cache=None)¶ Bases:
ufl.algorithms.transformer.Transformer
-
expr
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouchedas a default rule.
-
terminal
(o)¶ Always reuse Expr (ignore children)
-
variable
(o)¶
-
-
class
ufl.algorithms.transformer.
Transformer
(variable_cache=None)¶ Bases:
object
Base class for a visitor-like algorithm design pattern used to transform expression trees from one representation to another.
-
always_reconstruct
(o, *operands)¶ Always reconstruct expr.
-
expr
(o)¶ Trigger error.
-
print_visit_stack
()¶
-
reconstruct_variable
(o)¶
-
reuse
(o)¶ Always reuse Expr (ignore children)
-
reuse_if_possible
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouchedas a default rule.
-
reuse_if_untouched
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouchedas a default rule.
-
reuse_variable
(o)¶
-
terminal
(o)¶ Always reuse Expr (ignore children)
-
undefined
(o)¶ Trigger error.
-
visit
(o)¶
-
-
class
ufl.algorithms.transformer.
VariableStripper
¶ Bases:
ufl.algorithms.transformer.ReuseTransformer
-
variable
(o)¶
-
-
ufl.algorithms.transformer.
apply_transformer
(e, transformer, integral_type=None)¶ Apply transformer.visit(expression) to each integrand expression in form, or to form if it is an Expr.
-
ufl.algorithms.transformer.
is_post_handler
(function)¶ Is this a handler that expects transformed children as input?
-
ufl.algorithms.transformer.
strip_variables
(e)¶ Replace all Variable instances with the expression they represent.
-
ufl.algorithms.transformer.
ufl2ufl
(e)¶ Convert an UFL expression to a new UFL expression, with no changes. This is used for testing that objects in the expression behave as expected.
-
ufl.algorithms.transformer.
ufl2uflcopy
(e)¶ Convert an UFL expression to a new UFL expression. All nonterminal object instances are replaced with identical copies, while terminal objects are kept. This is used for testing that objects in the expression behave as expected.
ufl.algorithms.traversal module¶
This module contains algorithms for traversing expression trees in different ways.
-
ufl.algorithms.traversal.
iter_expressions
(a)¶ Utility function to handle Form, Integral and any Expr the same way when inspecting expressions. Returns an iterable over Expr instances: - a is an Expr: (a,) - a is an Integral: the integrand expression of a - a is a Form: all integrand expressions of all integrals
Module contents¶
This module collects algorithms and utility functions operating on UFL objects.
-
ufl.algorithms.
estimate_total_polynomial_degree
(e, default_degree=1, element_replace_map={})¶ Estimate total polynomial degree of integrand.
NB! Although some compound types are supported here, some derivatives and compounds must be preprocessed prior to degree estimation. In generic code, this algorithm should only be applied after preprocessing.
For coefficients defined on an element with unspecified degree (None), the degree is set to the given default degree.
-
ufl.algorithms.
sort_elements
(elements)¶ Sort elements so that any sub elements appear before the corresponding mixed elements. This is useful when sub elements need to be defined before the corresponding mixed elements.
The ordering is based on sorting a directed acyclic graph.
-
ufl.algorithms.
compute_form_data
(form, do_apply_function_pullbacks=False, do_apply_integral_scaling=False, do_apply_geometry_lowering=False, preserve_geometry_types=(), do_apply_default_restrictions=True, do_apply_restrictions=True, do_estimate_degrees=True, do_append_everywhere_integrals=True, complex_mode=False)¶
-
ufl.algorithms.
purge_list_tensors
(expr)¶ Get rid of all ListTensor instances by expanding expressions to use their components directly. Will usually increase the size of the expression.
-
ufl.algorithms.
apply_transformer
(e, transformer, integral_type=None)¶ Apply transformer.visit(expression) to each integrand expression in form, or to form if it is an Expr.
-
class
ufl.algorithms.
ReuseTransformer
(variable_cache=None)¶ Bases:
ufl.algorithms.transformer.Transformer
-
expr
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouchedas a default rule.
-
terminal
(o)¶ Always reuse Expr (ignore children)
-
variable
(o)¶
-
-
ufl.algorithms.
load_ufl_file
(filename)¶ Load a .ufl file with elements, coefficients and forms.
-
class
ufl.algorithms.
Transformer
(variable_cache=None)¶ Bases:
object
Base class for a visitor-like algorithm design pattern used to transform expression trees from one representation to another.
-
always_reconstruct
(o, *operands)¶ Always reconstruct expr.
-
expr
(o)¶ Trigger error.
-
print_visit_stack
()¶
-
reconstruct_variable
(o)¶
-
reuse
(o)¶ Always reuse Expr (ignore children)
-
reuse_if_possible
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouchedas a default rule.
-
reuse_if_untouched
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouchedas a default rule.
-
reuse_variable
(o)¶
-
terminal
(o)¶ Always reuse Expr (ignore children)
-
undefined
(o)¶ Trigger error.
-
visit
(o)¶
-
-
class
ufl.algorithms.
MultiFunction
¶ Bases:
object
Base class for collections of non-recursive expression node handlers.
Subclass this (remember to call the
__init__
method of this class), and implement handler functions for eachExpr
type, using the lower case handler name of the type (exprtype._ufl_handler_name_
).This class is optimized for efficient type based dispatch in the
__call__
operator via typecode based lookup of the handler function bound to the algorithm object. Of course Python’s function call overhead still applies.-
expr
(o, *args)¶ Trigger error for types with missing handlers.
-
reuse_if_untouched
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouched
as a default rule.
-
undefined
(o, *args)¶ Trigger error for types with missing handlers.
-
-
ufl.algorithms.
extract_unique_elements
(form)¶ Build sorted tuple of all unique elements used in form.
-
ufl.algorithms.
extract_type
(a, ufl_type)¶ Build a set of all objects of class ufl_type found in a. The argument a can be a Form, Integral or Expr.
-
ufl.algorithms.
extract_elements
(form)¶ Build sorted tuple of all elements used in form.
-
ufl.algorithms.
extract_sub_elements
(elements)¶ Build sorted tuple of all sub elements (including parent element).
-
ufl.algorithms.
expand_indices
(e)¶
-
ufl.algorithms.
replace
(e, mapping)¶ Replace subexpressions in expression.
- @param e:
- An Expr or Form.
- @param mapping:
- A dict with from:to replacements to perform.
-
ufl.algorithms.
expand_derivatives
(form, **kwargs)¶ Expand all derivatives of expr.
In the returned expression g which is mathematically equivalent to expr, there are no VariableDerivative or CoefficientDerivative objects left, and Grad objects have been propagated to Terminal nodes.
-
ufl.algorithms.
extract_coefficients
(a)¶ Build a sorted list of all coefficients in a, which can be a Form, Integral or Expr.
-
ufl.algorithms.
strip_variables
(e)¶ Replace all Variable instances with the expression they represent.
-
ufl.algorithms.
post_traversal
(expr)¶ Yield
o
for each nodeo
in expr, child before parent.
-
ufl.algorithms.
change_to_reference_grad
(e)¶ Change Grad objects in expression to products of JacobianInverse and ReferenceGrad.
Assumes the expression is preprocessed or at least that derivatives have been expanded.
- @param e:
- An Expr or Form.
-
ufl.algorithms.
expand_compounds
(e)¶
-
ufl.algorithms.
validate_form
(form)¶ Performs all implemented validations on a form. Raises exception if something fails.
-
class
ufl.algorithms.
FormSplitter
¶ Bases:
ufl.corealg.multifunction.MultiFunction
-
argument
(obj)¶
-
expr
(o, *ops)¶ Reuse object if operands are the same objects.
Use in your own subclass by setting e.g.
expr = MultiFunction.reuse_if_untouched
as a default rule.
-
multi_index
(obj)¶
-
split
(form, ix, iy=0)¶
-
-
ufl.algorithms.
extract_arguments
(a)¶ Build a sorted list of all arguments in a, which can be a Form, Integral or Expr.
-
ufl.algorithms.
compute_form_adjoint
(form, reordered_arguments=None)¶ Compute the adjoint of a bilinear form.
This works simply by swapping the number and part of the two arguments, but keeping their elements and places in the integrand expressions.
-
ufl.algorithms.
compute_form_action
(form, coefficient)¶ Compute the action of a form on a Coefficient.
This works simply by replacing the last Argument with a Coefficient on the same function space (element). The form returned will thus have one Argument less and one additional Coefficient at the end if no Coefficient has been provided.
-
ufl.algorithms.
compute_energy_norm
(form, coefficient)¶ Compute the a-norm of a Coefficient given a form a.
This works simply by replacing the two Arguments with a Coefficient on the same function space (element). The Form returned will thus be a functional with no Arguments, and one additional Coefficient at the end if no coefficient has been provided.
-
ufl.algorithms.
compute_form_lhs
(form)¶ Compute the left hand side of a form.
a = u*v*dx + f*v*dx a = lhs(a) -> u*v*dx
-
ufl.algorithms.
compute_form_rhs
(form)¶ Compute the right hand side of a form.
a = u*v*dx + f*v*dx L = rhs(a) -> -f*v*dx
-
ufl.algorithms.
compute_form_functional
(form)¶ Compute the functional part of a form, that is the terms independent of Arguments.
(Used for testing, not sure if it’s useful for anything?)
-
ufl.algorithms.
compute_form_signature
(form, renumbering)¶
-
ufl.algorithms.
tree_format
(expression, indentation=0, parentheses=True)¶