constraint_handler.h

// SPDX-FileCopyrightText: © 2025 PRISMS Center at the University of Michigan
// SPDX-License-Identifier: GNU Lesser General Public Version 2.1
 
#ifndef constraint_handler_h
#define constraint_handler_h
 
#include <deal.II/base/mg_level_object.h>
#include <deal.II/dofs/dof_handler.h>
#include <deal.II/fe/mapping_q1.h>
#include <deal.II/lac/affine_constraints.h>
 
#include <prismspf/config.h>
#include <prismspf/user_inputs/user_input_parameters.h>
 
#include <vector>
 
PRISMS_PF_BEGIN_NAMESPACE
 
template <int dim>
class constraintHandler
{
public:
  explicit constraintHandler(const userInputParameters<dim> &_user_inputs);
 
  [[nodiscard]] std::vector<const dealii::AffineConstraints<double> *>
  get_constraints();
 
  [[nodiscard]] const dealii::AffineConstraints<double> &
  get_constraint(const unsigned int &index) const;
 
  [[nodiscard]] const dealii::MGLevelObject<dealii::AffineConstraints<float>> &
  get_mg_constraint(unsigned int index) const;
 
  [[nodiscard]] const dealii::AffineConstraints<float> &
  get_mg_constraint(unsigned int index, unsigned int level) const;
 
  void
  make_constraints(const dealii::Mapping<dim>                         &mapping,
                   const std::vector<const dealii::DoFHandler<dim> *> &dof_handlers);
 
  void
  make_mg_constraints(
    const dealii::Mapping<dim> &mapping,
    const std::map<unsigned int, dealii::MGLevelObject<dealii::DoFHandler<dim>>>
      &mg_dof_handlers);
 
private:
  void
  make_constraint(const dealii::Mapping<dim>    &mapping,
                  const dealii::DoFHandler<dim> &dof_handler,
                  const unsigned int            &index);
 
  void
  make_mg_constraint(const dealii::Mapping<dim>                           &mapping,
                     const dealii::MGLevelObject<dealii::DoFHandler<dim>> &dof_handler,
                     const unsigned int                                   &index);
 
  void
  set_pinned_point(const dealii::DoFHandler<dim> &dof_handler, const unsigned int &index);
 
  const userInputParameters<dim> *user_inputs;
 
  std::vector<dealii::AffineConstraints<double>> constraints;
 
  std::map<unsigned int, dealii::MGLevelObject<dealii::AffineConstraints<float>>>
    mg_constraints;
};
class constraintHandler {…};
 
PRISMS_PF_END_NAMESPACE
 
#endif