format, extract num rows in ml schwarz

Author: greole

include/OGL/Preconditioner/Jacobi.hpp

@@ -49,20 +49,13 @@ class BlockJacobi  // : public PreconditionerWrapper
                                   .on(exec_));
         };
 
-        auto wrapper = [this](auto f) {
-            if (multi_level_schwarz_) {
-                return wrap_multi_level_schwarz(
-                    mtx_, exec_, f, d_.subDict("multiLevelConfig"), verbose_);
-            } else {
-                return wrap_schwarz(mtx_, exec_, std::move(f));
-            }
-        };
-
         if (precision_ == "double") {
-            return wrapper(builder(dbj::build()));
+            return dispatch_schwarz(mtx_, exec_, builder(dbj::build()),
+                                    d_, verbose);
         }
         if (precision_ == "float") {
-            return wrapper(builder(fbj::build()));
+            return dispatch_schwarz(mtx_, exec_, builder(fbj::build()),
+                                    d_, verbose);
         }
 
         return {};

include/OGL/Preconditioner/Schwarz.hpp

@@ -116,4 +116,18 @@ std::shared_ptr<gko::LinOp> wrap_multi_level_schwarz(
     }
 }
 
+
+template <typename PrecondFactory>
+std::shared_ptr<gko::LinOp> dispatch_schwarz(
+    std::shared_ptr<const gko::LinOp> mtx, std::shared_ptr<gko::Executor> exec,
+    std::shared_ptr<PrecondFactory> precond, const dictionary &d, label verbose)
+{
+    if (d.lookupOrDefault<Switch>("multiLevelSchwarz", false)) {
+        return wrap_multi_level_schwarz(mtx, exec, precond,
+                                        d.subDict("multiLevelConfig"), verbose);
+    } else {
+        return wrap_schwarz(mtx, exec, std::move(precond));
+    }
+}
+
 }  // namespace Foam