triangulation_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/distributed/tria.h>
#include <deal.II/grid/grid_tools_geometry.h>
#include <deal.II/grid/tria.h>
#include <deal.II/multigrid/mg_transfer_global_coarsening.h>
 
#include <prismspf/user_inputs/constraint_parameters.h>
#include <prismspf/user_inputs/spatial_discretization.h>
 
#include <prismspf/config.h>
 
#include <memory>
#include <string>
 
PRISMS_PF_BEGIN_NAMESPACE
 
template <unsigned int dim>
class TriangulationManager
{
public:
  explicit TriangulationManager(bool _has_multigrid = false);
 
  explicit TriangulationManager(const SpatialDiscretization<dim> &discretization_params,
                                bool                              _has_multigrid = false)
    : TriangulationManager(_has_multigrid)
  {
    generate_mesh(discretization_params);
    reinit();
  }
 
  void
  set_using_multigrid(bool _has_multigrid)
  {
    has_multigrid = _has_multigrid;
  }
 
  void
  reinit();
 
  [[nodiscard]] const Triangulation<dim> &
  get_triangulation() const;
 
  [[nodiscard]] const dealii::Triangulation<dim> &
  get_triangulation(unsigned int relative_level) const;
 
  [[nodiscard]] const std::vector<dealii::GridTools::PeriodicFacePair<
    typename dealii::Triangulation<dim>::cell_iterator>> &
  get_periodic_face_pairs() const;
 
  [[nodiscard]] double
  get_volume() const;
 
  void
  generate_mesh(const SpatialDiscretization<dim> &discretization_params);
 
  void
  export_triangulation_as_vtk(const std::string &filename) const;
 
  void
  prepare_for_grid_refinement()
  {
    triangulation.prepare_coarsening_and_refinement();
  }
 
  void
  execute_grid_refinement()
  {
    triangulation.execute_coarsening_and_refinement();
  }
 
  void
  clear_user_flags()
  {
    triangulation.clear_user_flags();
  }
 
private:
  bool has_multigrid = false;
 
  Triangulation<dim> triangulation;
 
  std::vector<std::shared_ptr<const dealii::Triangulation<dim>>> coarsened_triangulations;
 
  std::vector<dealii::GridTools::PeriodicFacePair<
    typename dealii::Triangulation<dim>::cell_iterator>>
    periodicity_vector;
 
  double volume = 0;
};
 
PRISMS_PF_END_NAMESPACE