More DofMapBase APIs

Also, set n components to 0 for DofObjects for the reduced "system"

Author: lindsayad

include/base/dof_map.h

@@ -661,10 +661,7 @@ class DofMap : public DofMapBase,
     return (this->has_blocked_representation() ? this->n_variables() : 1);
   }
 
-  /**
-   * \returns The total number of degrees of freedom in the problem.
-   */
-  dof_id_type n_dofs() const { return _n_dfs; }
+  dof_id_type n_dofs() const override { return _n_dfs; }
 
   /**
    * \returns The total number of degrees of freedom for a particular
@@ -682,10 +679,7 @@ class DofMap : public DofMapBase,
    */
   dof_id_type n_SCALAR_dofs() const { return _n_SCALAR_dofs; }
 
-  /**
-   * \returns The number of degrees of freedom on this processor.
-   */
-  dof_id_type n_local_dofs () const
+  dof_id_type n_local_dofs () const override
   { return this->n_dofs_on_processor (this->processor_id()); }
 
   /**

include/base/dof_map_base.h

@@ -74,6 +74,15 @@ class DofMapBase
   virtual void dof_indices(const Node * const node,
                            std::vector<dof_id_type> & di,
                            const unsigned int vn) const = 0;
+  /**
+   * \returns The total number of degrees of freedom in the problem.
+   */
+  virtual dof_id_type n_dofs() const = 0;
+
+  /**
+   * \returns The number of degrees of freedom on this processor.
+   */
+  virtual dof_id_type n_local_dofs() const = 0;
 };
 
 }

include/numerics/static_condensation.h

@@ -64,7 +64,7 @@ typedef Eigen::VectorXd EigenVector;
 class StaticCondensation : public PetscMatrixShellMatrix<Number>, public DofMapBase
 {
 public:
-  StaticCondensation(const MeshBase & mesh, const System & system, const DofMap & dof_map);
+  StaticCondensation(const MeshBase & mesh, System & system, const DofMap & dof_map);
   virtual ~StaticCondensation();
 
   //
@@ -191,10 +191,6 @@ class StaticCondensation : public PetscMatrixShellMatrix<Number>, public DofMapB
 
   virtual const Variable & variable(const unsigned int c) const override;
 
-  virtual dof_id_type first_dof() const override { return this->row_start(); }
-
-  virtual dof_id_type end_dof() const override { return this->row_stop(); }
-
   virtual void dof_indices(const Elem * const elem,
                            std::vector<dof_id_type> & di,
                            const unsigned int vn,
@@ -204,11 +200,21 @@ class StaticCondensation : public PetscMatrixShellMatrix<Number>, public DofMapB
                            std::vector<dof_id_type> & di,
                            const unsigned int vn) const override;
 
+  virtual dof_id_type first_dof() const override;
+  virtual dof_id_type end_dof() const override;
+  virtual dof_id_type n_dofs() const override;
+  virtual dof_id_type n_local_dofs() const override;
+
   /**
    * @returns The reduced system linear solver
    */
   LinearSolver<Number> & reduced_system_solver();
 
+  /*
+   * @returns The dummyish reduced system
+   */
+  const System & reduced_system() const;
+
 private:
   /**
    * Retrieves the degree of freedom values from \p global_vector corresponding to \p
@@ -279,7 +285,7 @@ class StaticCondensation : public PetscMatrixShellMatrix<Number>, public DofMapB
   std::vector<dof_id_type> _local_uncondensed_dofs;
 
   const MeshBase & _mesh;
-  const System & _system;
+  System & _system;
   const DofMap & _dof_map;
 
   /// global sparse matrix for the uncondensed degrees of freedom
@@ -318,6 +324,9 @@ class StaticCondensation : public PetscMatrixShellMatrix<Number>, public DofMapB
 
   /// The variables in the reduced system
   std::vector<Variable> _reduced_vars;
+
+  /// A dummyish system to help with DofObjects
+  System * _reduced_system;
 };
 
 inline const SparseMatrix<Number> & StaticCondensation::get_condensed_mat() const
@@ -337,6 +346,12 @@ inline LinearSolver<Number> & StaticCondensation::reduced_system_solver()
   return *_reduced_solver;
 }
 
+inline const System & StaticCondensation::reduced_system() const
+{
+  libmesh_assert(_reduced_system);
+  return *_reduced_system;
+}
+
 }
 
 #else

src/numerics/static_condensation.C

@@ -24,18 +24,18 @@
 #include "libmesh/elem.h"
 #include "libmesh/int_range.h"
 #include "libmesh/numeric_vector.h"
-#include "libmesh/petsc_matrix_base.h"
 #include "libmesh/linear_solver.h"
 #include "libmesh/static_condensation_preconditioner.h"
 #include "libmesh/system.h"
 #include "libmesh/petsc_matrix.h"
+#include "libmesh/equation_systems.h"
 #include "timpi/parallel_sync.h"
 #include <unordered_set>
 
 namespace libMesh
 {
 StaticCondensation::StaticCondensation(const MeshBase & mesh,
-                                       const System & system,
+                                       System & system,
                                        const DofMap & dof_map)
   : PetscMatrixShellMatrix<Number>(dof_map.comm()),
     _mesh(mesh),
@@ -384,6 +384,20 @@ void StaticCondensation::init()
   // Build ghosted full solution vector. Note that this is, in general, *not equal* to the system
   // solution, e.g. this may correspond to the solution for the Newton *update*
   _ghosted_full_sol = _system.current_local_solution->clone();
+
+  // Prevent querying Nodes for dof indices
+  std::vector<unsigned int> nvpg(_reduced_vars.size());
+  for (auto & elem : nvpg)
+    elem = 1;
+
+  _reduced_system = &_system.get_equation_systems().add_system("Basic", "reduced");
+  _reduced_system->init();
+  for (auto * const nd : _mesh.active_node_ptr_range())
+    {
+      nd->set_n_vars_per_group(_reduced_system->number(), nvpg);
+      for (const auto g : index_range(nvpg))
+        nd->set_n_comp_group(_reduced_system->number(), g, 0);
+    }
   _sc_is_initialized = true;
 }
 
@@ -673,6 +687,14 @@ void StaticCondensation::dof_indices(const Node * const,
       "StaticCondensation dof indices are only meant to be queried with elements, not nodes");
 }
 
+dof_id_type StaticCondensation::first_dof() const { return _reduced_sys_mat->row_start(); }
+
+dof_id_type StaticCondensation::end_dof() const { return _reduced_sys_mat->row_stop(); }
+
+dof_id_type StaticCondensation::n_dofs() const { return _reduced_sys_mat->m(); }
+
+dof_id_type StaticCondensation::n_local_dofs() const { return _reduced_sys_mat->local_m(); }
+
 }
 #else