Tests work for MultiField+GMG

src/PatchBasedSmoothers/seq/PatchBasedLinearSolvers.jl

@@ -70,7 +70,7 @@ end
 function Gridap.Algebra.solve!(x::AbstractVector,ns::PatchBasedSmootherNumericalSetup,r::AbstractVector)
   Ap_ns, weights, caches = ns.local_ns, ns.weights, ns.caches
 
-  Ph = ns.solver.Ph
+  Ph = ns.solver.patch_space
   #w, w_sums = weights
   rp, dxp = caches
 

test/LinearSolvers/GMGTests.jl

@@ -30,7 +30,7 @@ function get_patch_smoothers(tests,patch_spaces,patch_decompositions,biform,qdeg
       Vh = get_fe_space(tests,lev)
       Ω  = Triangulation(PD)
       dΩ = Measure(Ω,qdegree)
-      local_solver = LUSolver() # IS_ConjugateGradientSolver(;reltol=1.e-6)
+      local_solver = LUSolver()
       patch_smoother = PatchBasedLinearSolver(biform,Ph,Vh,dΩ,local_solver)
       smoothers[lev] = RichardsonSmoother(patch_smoother,10,0.2)
     end
@@ -49,7 +49,8 @@ function gmg_driver(t,parts,mh,spaces,qdegree,smoothers,biform,liform,u)
   coarse_solver = LUSolver()
   restrictions, prolongations = setup_transfer_operators(trials,
                                                          qdegree;
-                                                         mode=:residual)
+                                                         mode=:residual,
+                                                         solver=IS_ConjugateGradientSolver(;reltol=1.e-6))
   gmg = GMGLinearSolver(mh,
                         smatrices,
                         prolongations,
@@ -73,17 +74,18 @@ function gmg_driver(t,parts,mh,spaces,qdegree,smoothers,biform,liform,u)
   toc!(t,"Solver")
 
   # Error
-  model = get_model(mh,1)
-  Uh    = get_fe_space(trials,1)
-  Ω     = Triangulation(model)
-  dΩ    = Measure(Ω,qdegree)
-  uh    = FEFunction(Uh,x)
-  eh    = u-uh
-  e_l2  = sum(∫(eh⋅eh)dΩ)
-  if i_am_main(parts)
-    println("L2 error = ", e_l2)
+  if !isa(u,Nothing)
+    model = get_model(mh,1)
+    Uh    = get_fe_space(trials,1)
+    Ω     = Triangulation(model)
+    dΩ    = Measure(Ω,qdegree)
+    uh    = FEFunction(Uh,x)
+    eh    = u-uh
+    e_l2  = sum(∫(eh⋅eh)dΩ)
+    if i_am_main(parts)
+      println("L2 error = ", e_l2)
+    end
   end
-  return e_l2
 end
 
 function gmg_poisson_driver(t,parts,mh,order)
@@ -168,6 +170,41 @@ function gmg_hdiv_driver(t,parts,mh,order)
   return gmg_driver(t,parts,mh,spaces,qdegree,smoothers,biform,liform,u)
 end
 
+function gmg_multifield_driver(t,parts,mh,order)
+  tic!(t;barrier=true)
+  Dc = num_cell_dims(get_model(mh,1))
+  @assert Dc == 3
+
+  β = 1.0
+  γ = 1.0
+  B = VectorValue(0.0,0.0,1.0)
+  f = VectorValue(fill(1.0,Dc)...)
+  biform((u,j),(v_u,v_j),dΩ) = ∫(β*∇(u)⊙∇(v_u) -γ*(j×B)⋅v_u + j⋅v_j - (u×B)⋅v_j)dΩ
+  liform((v_u,v_j),dΩ) = ∫(v_u⋅f)dΩ
+
+  reffe_u  = ReferenceFE(lagrangian,VectorValue{Dc,Float64},order)
+  tests_u  = FESpace(mh,reffe_u;dirichlet_tags="boundary");
+  trials_u = TrialFESpace(tests_u);
+
+  reffe_j  = ReferenceFE(raviart_thomas,Float64,order-1)
+  tests_j  = FESpace(mh,reffe_j;dirichlet_tags="boundary");
+  trials_j = TrialFESpace(tests_j);
+
+  trials = MultiFieldFESpace([trials_u,trials_j]);
+  tests  = MultiFieldFESpace([tests_u,tests_j]);
+  spaces = tests, trials
+  toc!(t,"FESpaces")
+
+  tic!(t;barrier=true)
+  qdegree = 2*(order+1)
+  patch_decompositions = PatchDecomposition(mh)
+  patch_spaces = PatchFESpace(tests,patch_decompositions)
+  smoothers = get_patch_smoothers(tests,patch_spaces,patch_decompositions,biform,qdegree)
+  toc!(t,"Patch Decomposition")
+
+  return gmg_driver(t,parts,mh,spaces,qdegree,smoothers,biform,liform,nothing)
+end
+
 function main_gmg_driver(parts,mh,order,pde)
   t = PTimer(parts,verbose=true)
   if     pde == :poisson
@@ -178,6 +215,10 @@ function main_gmg_driver(parts,mh,order,pde)
     gmg_vector_laplace_driver(t,parts,mh,order)
   elseif pde == :hdiv
     gmg_hdiv_driver(t,parts,mh,order)
+  elseif pde == :multifield
+    gmg_multifield_driver(t,parts,mh,order)
+  else
+    error("Unknown PDE")
   end
 end
 
@@ -197,14 +238,17 @@ end
 function main(distribute,np::Integer,nc::Tuple,np_per_level::Vector)
   parts = distribute(LinearIndices((np,)))
   mh = get_mesh_hierarchy(parts,nc,np_per_level)
+  Dc = length(nc)
 
-  for pde in [:poisson,:laplace,:vector_laplace,:hdiv]
-    if i_am_main(parts)
-      println(repeat("=",80))
-      println("Testing GMG with Dc=$(length(nc)), PDE=$pde")
+  for pde in [:poisson,:laplace,:vector_laplace,:hdiv,:multifield]
+    if (pde != :multifield) || (Dc == 3)
+      if i_am_main(parts)
+        println(repeat("=",80))
+        println("Testing GMG with Dc=$(length(nc)), PDE=$pde")
+      end
+      order = (pde !== :hdiv) ? 1 : 0
+      main_gmg_driver(parts,mh,order,pde)
     end
-    order = (pde !== :hdiv) ? 1 : 0
-    main_gmg_driver(parts,mh,order,pde)
   end
 end