dof_manager.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: © 2025 PRISMS Center at the University of Michigan
// SPDX-License-Identifier: GNU Lesser General Public Version 2.1
 
#pragma once
 
#include <deal.II/base/mg_level_object.h>
#include <deal.II/dofs/dof_handler.h>
#include <deal.II/fe/fe_system.h>
 
#include <prismspf/core/field_attributes.h>
#include <prismspf/core/solve_group.h>
#include <prismspf/core/triangulation_manager.h>
#include <prismspf/core/type_enums.h>
#include <prismspf/core/types.h>
 
#include <prismspf/config.h>
 
#include <memory>
 
PRISMS_PF_BEGIN_NAMESPACE
 
template <unsigned int dim, unsigned int degree>
class DoFManager
{
public:
  explicit DoFManager(const std::vector<FieldAttributes> &field_attributes);
 
  DoFManager(const std::vector<FieldAttributes> &field_attributes,
             const TriangulationManager<dim>    &triangulation_handler);
 
  void
  init(const TriangulationManager<dim> &triangulation_handler);
 
  void
  reinit(const TriangulationManager<dim> &triangulation_handler);
 
  [[nodiscard]] std::vector<const dealii::DoFHandler<dim> *>
  get_field_dof_handlers(const std::set<unsigned int> &field_indices,
                         unsigned int                  relative_level = 0) const;
 
  [[nodiscard]] const dealii::DoFHandler<dim> &
  get_field_dof_handler(Types::Index field_index, unsigned int relative_level = 0) const
  {
    return *dof_handlers[field_index][relative_level];
  }
 
  const std::vector<std::array<dealii::DoFHandler<dim>, 2>> &
  get_dof_handlers() const
  {
    return level_dof_handlers;
  }
 
  [[nodiscard]] const dealii::DoFHandler<dim> &
  get_dof_handler(const unsigned int &rank, unsigned int relative_level = 0) const
  {
    return level_dof_handlers[relative_level][rank];
  }
 
  [[nodiscard]] dealii::types::global_dof_index
  get_total_dofs() const
  {
    dealii::types::global_dof_index n_dofs = 0;
    for (const auto &dof_handler_set : dof_handlers)
      {
        n_dofs += dof_handler_set[0]->n_dofs();
      }
    return n_dofs;
  }
 
private:
  std::vector<std::vector<const dealii::DoFHandler<dim> *>> dof_handlers;
 
  std::vector<std::array<dealii::DoFHandler<dim>, 2>> level_dof_handlers;
};
 
PRISMS_PF_END_NAMESPACE