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 std::vector<FieldAttributes> &field_attributes,
             const TriangulationManager<dim>    &triangulation_manager);
  DoFManager(const DoFManager &)  = delete;
  DoFManager(const DoFManager &&) = delete;
  DoFManager
  operator=(const DoFManager &) = delete;
  DoFManager &
  operator=(DoFManager &&other) = delete;

  ~DoFManager() = default;

  void
  init(unsigned int num_levels);

  void
  reinit(const TriangulationManager<dim> &triangulation_manager);

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

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

  [[nodiscard]] const std::vector<const dealii::DoFHandler<dim> *> &
  get_field_dof_handlers(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;

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

  [[nodiscard]] const std::array<dealii::DoFHandler<dim>, 2> &
  get_dof_handlers(unsigned int relative_level = 0) const;

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

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

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