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_block.h>
#include <prismspf/core/triangulation_manager.h>
#include <prismspf/core/type_enums.h>
#include <prismspf/core/types.h>
 
#include <prismspf/config.h>
 
PRISMS_PF_BEGIN_NAMESPACE

template <unsigned int dim, unsigned int degree>
class DoFManager
{
public:
  DoFManager() = default;

  DoFManager(const DoFManager &)  = delete;
  DoFManager(const DoFManager &&) = delete;
  DoFManager
  operator=(const DoFManager &) = delete;
  DoFManager &
  operator=(DoFManager &&other) = delete;

  ~DoFManager() = default;

  void
  reinit(const TriangulationManager<dim> &triangulation_manager, bool init_mg = false);

  void
  reinit_mapping(const std::vector<FieldAttributes> &field_attributes);

  [[nodiscard]] const std::vector<const dealii::DoFHandler<dim> *> &
  get_field_dof_handlers() const;

  [[nodiscard]] const dealii::DoFHandler<dim> &
  get_field_dof_handler(Types::Index field_index) const;

  [[nodiscard]] std::vector<const dealii::DoFHandler<dim> *>
  get_block_dof_handlers(const std::set<unsigned int> &field_indices) const;

  [[nodiscard]] const std::array<dealii::DoFHandler<dim>, 2> &
  get_dof_handlers() const;

  [[nodiscard]] const dealii::DoFHandler<dim> &
  get_dof_handler(const unsigned int &rank) const;

  [[nodiscard]] bool
  has_mg() const;

  [[nodiscard]] unsigned int
  num_levels() const;

  [[nodiscard]] dealii::types::global_dof_index
  get_total_dofs() const;
 
private:
  std::vector<const dealii::DoFHandler<dim> *> field_dof_handlers;

  std::array<dealii::DoFHandler<dim>, 2> level_dof_handlers;
};
 
PRISMS_PF_END_NAMESPACE