initial implementation

Author: KarsKnook

firedrake/bcs.py

@@ -488,6 +488,7 @@ class EquationBC(object):
     def __init__(self, *args, bcs=None, J=None, Jp=None, V=None, is_linear=False, Jp_eq_J=False):
         from firedrake.variational_solver import check_pde_args, is_form_consistent
         if isinstance(args[0], ufl.classes.Equation):
+            self.function_arg = firedrake.Constant(0) # so line 286 of mg/ufl_utils.py runs
             # initial construction from equation
             eq = args[0]
             u = args[1]

firedrake/mg/ufl_utils.py

@@ -126,6 +126,16 @@ def coarsen_bc(bc, self, coefficient_mapping=None):
     return type(bc)(V, val, subdomain)
 
 
+@coarsen.register(firedrake.EquationBC)
+def coarsen_equation_bc(ebc, self, coefficient_mapping=None):
+    eq = self(ebc._F.f, self, coefficient_mapping=coefficient_mapping)
+    u = self(ebc._F.u, self, coefficient_mapping=coefficient_mapping)
+    sub_domain = ebc._F.sub_domain
+    V = self(ebc._F.function_space(), self, coefficient_mapping=coefficient_mapping)
+
+    return type(ebc)(eq == 0, u, sub_domain, V=V)
+
+
 @coarsen.register(firedrake.functionspaceimpl.WithGeometryBase)
 def coarsen_function_space(V, self, coefficient_mapping=None):
     if hasattr(V, "_coarse"):
@@ -311,7 +321,8 @@ def mult(self, mat, x, y, inc=False):
             x.copy(v)
         self.manager.prolong(self.cprimal, self.fprimal)
         for bc in self.fbcs:
-            bc.zero(self.fprimal)
+            if isinstance(bc, firedrake.DirichletBC):
+                bc.zero(self.fprimal)
         with self.fprimal.dat.vec_ro as v:
             if inc:
                 y.axpy(1.0, v)
@@ -330,7 +341,8 @@ def multTranspose(self, mat, x, y, inc=False):
             x.copy(v)
         self.manager.restrict(self.fdual, self.cdual)
         for bc in self.cbcs:
-            bc.zero(self.cdual)
+            if isinstance(bc, firedrake.DirichletBC):
+                bc.zero(self.cdual)
         with self.cdual.dat.vec_ro as v:
             if inc:
                 y.axpy(1.0, v)