def restrict(self, gf, gc): """Restrict a dual function. :arg gf: The source (fine grid) dual function. :arg gc: The target (coarse grid) dual function. """ Vc = gc.function_space() Vf = gf.function_space() dgf = self.DG_work(Vf) dgc = self.DG_work(Vc) VDGf = dgf.function_space() VDGc = dgc.function_space() work = self.work_vec(Vf) dgwork = self.work_vec(VDGc) # g \in Vf^* -> g \in VDGf^* with gf.dat.vec_ro as gfv, dgf.dat.vec_wo as dgscratch: if self.use_fortin_interpolation: work.pointwiseDivide(gfv, self.V_dof_weights(Vf)) self.V_approx_inv_mass(Vf, VDGf).multTranspose(work, dgscratch) else: self.V_inv_mass_ksp(Vf).solve(gfv, work) self.V_DG_mass(Vf, VDGf).mult(work, dgscratch) # g \in VDGf^* -> g \in VDGc^* firedrake.restrict(dgf, dgc) # g \in VDGc^* -> g \in Vc^* with dgc.dat.vec_ro as dgscratch, gc.dat.vec_wo as gcv: self.DG_inv_mass(VDGc).mult(dgscratch, dgwork) self.V_DG_mass(Vc, VDGc).multTranspose(dgwork, gcv)
def restrict(self, gf, gc): """Restrict a dual function. :arg gf: The source (fine grid) dual function. :arg gc: The target (coarse grid) dual function. """ Vc = gc.function_space() Vf = gf.function_space() dgf = self.DG_work(Vf) dgc = self.DG_work(Vc) VDGf = dgf.function_space() VDGc = dgc.function_space() work = self.work_vec(Vf) dgwork = self.work_vec(VDGc) # g \in Vf^* -> g \in VDGf^* with gf.dat.vec_ro as gfv, dgf.dat.vec_wo as dgscratch: if self.use_fortin_interpolation: work.pointwiseDivide(gfv, self.V_dof_weights(Vf)) self.V_approx_inv_mass(Vf, VDGf).multTranspose(work, dgscratch) else: self.V_inv_mass_ksp(Vf).solve(gfv, work) self.V_DG_mass(Vf, VDGf).mult(work, dgscratch) # g \in VDGf^* -> g \in VDGc^* firedrake.restrict(dgf, dgc) # g \in VDGc^* -> g \in Vc^* with dgc.dat.vec_ro as dgscratch, gc.dat.vec_wo as gcv: self.DG_inv_mass(VDGc).mult(dgscratch, dgwork) self.V_DG_mass(Vc, VDGc).multTranspose(dgwork, gcv)
def multTranspose(self, mat, x, y, inc=False): with self.ffn.dat.vec as v: x.copy(v) firedrake.restrict(self.ffn, self.cfn) for bc in self.cbcs: bc.zero(self.cfn) with self.cfn.dat.vec_ro as v: if inc: y.axpy(1.0, v) else: v.copy(y)
def multTranspose(self, mat, x, y, inc=False): with self.ffn.dat.vec as v: x.copy(v) firedrake.restrict(self.ffn, self.cfn) for bc in self.cbcs: bc.zero(self.cfn) with self.cfn.dat.vec_ro as v: if inc: y.axpy(1.0, v) else: v.copy(y)
def restrict(f, c): return firedrake.restrict(f, c)
def restrict(self, residual, out): Tf = residual for Tinter in reversed(self.intermediate_Ts): fd.restrict(Tf, Tinter) Tf = Tinter fd.restrict(Tf, out.fun)
def restrict(f, c): return firedrake.restrict(f, c)