def facet_avg(self, o):
if self.context.integral_type == "cell":
raise ValueError("Can't take FacetAvg in cell integral")
integrand, = o.ufl_operands
domain = o.ufl_domain()
measure = ufl.Measure(self.context.integral_type, domain=domain)
integrand, degree, argument_multiindices = entity_avg(integrand / CellVolume(domain), measure, self.context.argument_multiindices)
config = {name: getattr(self.context, name)
for name in ["ufl_cell", "precision", "index_cache"]},
config.update(quadrature_degree=degree, interface=self.context,
argument_multiindices=argument_multiindices)
expr, = compile_ufl(integrand, point_sum=True, **config)
return expr
def cell_avg(self, o):
if self.context.integral_type != "cell":
# Need to create a cell-based quadrature rule and
# translate the expression using that (c.f. CellVolume
# below).
raise NotImplementedError("CellAvg on non-cell integrals not yet implemented")
integrand, = o.ufl_operands
domain = o.ufl_domain()
measure = ufl.Measure(self.context.integral_type, domain=domain)
integrand, degree, argument_multiindices = entity_avg(integrand / CellVolume(domain), measure, self.context.argument_multiindices)
config = {name: getattr(self.context, name)
for name in ["ufl_cell", "precision", "index_cache"]}
config.update(quadrature_degree=degree, interface=self.context,
argument_multiindices=argument_multiindices)
expr, = compile_ufl(integrand, point_sum=True, **config)
return expr