stokes_matrix_free_global_coarsening.h

Go to the documentation of this file.

// ------------------------------------------------------
//
// SPDX-License-Identifier: GPL-2.0-or-later
//
// SPDX-FileCopyrightText: Copyright (C) 2011-2026 by the ASPECT authors.
//
// This file is part of ASPECT.
//
// Detailed license information governing the source code and
// contributions can be found in the folder LICENSES and in
// LICENSE and CONTRIBUTING.md at the top level directory.
//
// ------------------------------------------------------

#ifndef _aspect_simulator_solver_stokes_matrix_free_global_coarsening_h
#define _aspect_simulator_solver_stokes_matrix_free_global_coarsening_h

#include <aspect/global.h>
#include <aspect/simulator/solver/stokes_matrix_free.h>
#include <aspect/simulator/solver/matrix_free_operators.h>

#include <deal.II/matrix_free/matrix_free.h>
#include <deal.II/matrix_free/operators.h>
#include <deal.II/matrix_free/fe_evaluation.h>

#include <deal.II/multigrid/mg_constrained_dofs.h>
#include <deal.II/multigrid/multigrid.h>
#include <deal.II/multigrid/mg_transfer_matrix_free.h>
#include <deal.II/multigrid/mg_transfer_global_coarsening.templates.h>
#include <deal.II/multigrid/mg_tools.h>
#include <deal.II/multigrid/mg_coarse.h>
#include <deal.II/multigrid/mg_smoother.h>
#include <deal.II/multigrid/mg_matrix.h>

namespace aspect
{
  template <int dim, int velocity_degree>
  class StokesMatrixFreeHandlerGlobalCoarseningImplementation: public StokesMatrixFreeHandler<dim>
  {
    public:
      StokesMatrixFreeHandlerGlobalCoarseningImplementation(Simulator<dim> &simulator,
                                                            const Parameters<dim> &parameters);

      ~StokesMatrixFreeHandlerGlobalCoarseningImplementation() override = default;

      void initialize() override;

      std::string name() const override;

      StokesSolver::SolverOutputs
      solve(const LinearAlgebra::BlockSparseMatrix &system_matrix,
            const LinearAlgebra::BlockVector &system_rhs,
            const bool solve_newton_system,
            const double last_pressure_normalization_adjustment,
            LinearAlgebra::BlockVector &solution_vector) override;

      void setup_dofs() override;

      void setup_multigrid_hierarchy() override;

      const std::vector<std::shared_ptr<const Triangulation<dim, dim>>> &
      get_multigrid_triangulations() const override;

      void assemble(LinearAlgebra::BlockVector &system_rhs) override;

      void build_preconditioner() override;

      static
      void declare_parameters (ParameterHandler &prm);

      void parse_parameters (ParameterHandler &prm) override;

      std::size_t get_dof_handler_memory_consumption() const override;

      std::size_t get_mg_transfer_memory_consumption() const override;

      std::size_t get_constraint_memory_consumption() const override;

      std::size_t get_cell_data_memory_consumption() const override;

    private:
      void evaluate_material_model();

      void correct_stokes_rhs(LinearAlgebra::BlockVector &system_rhs);

      bool print_details;

      bool do_timings;

      double minimum_viscosity;
      double maximum_viscosity;

      FESystem<dim> fe_v;
      FESystem<dim> fe_p;
      FESystem<dim> fe_projection;

      MatrixFreeStokesOperators::OperatorCellData<dim, GMGNumberType> active_cell_data;

      MGLevelObject<MatrixFreeStokesOperators::OperatorCellData<dim, GMGNumberType>> level_cell_data;

      using StokesMatrixType = MatrixFreeStokesOperators::StokesOperator<dim,velocity_degree,double>;
      using SchurComplementMatrixType = MatrixFreeStokesOperators::MassMatrixOperator<dim,velocity_degree-1,double>;
      using ABlockMatrixType = MatrixFreeStokesOperators::ABlockOperator<dim,velocity_degree,double>;
      using BTBlockOperatorType = MatrixFreeStokesOperators::BTBlockOperator<dim,velocity_degree,double>;
      using GMGSchurComplementMatrixType = MatrixFreeStokesOperators::MassMatrixOperator<dim,velocity_degree-1,GMGNumberType>;
      using GMGABlockMatrixType = MatrixFreeStokesOperators::ABlockOperator<dim,velocity_degree,GMGNumberType>;

      StokesMatrixType stokes_matrix;
      ABlockMatrixType A_block_matrix;
      BTBlockOperatorType BT_block;
      SchurComplementMatrixType Schur_complement_block_matrix;

      MGLevelObject<GMGABlockMatrixType> mg_matrices_A_block;
      MGLevelObject<GMGSchurComplementMatrixType> mg_matrices_Schur_complement;

      MGConstrainedDoFs mg_constrained_dofs_A_block;
      MGConstrainedDoFs mg_constrained_dofs_Schur_complement;
      MGConstrainedDoFs mg_constrained_dofs_projection;

      unsigned int min_level;
      unsigned int max_level;

      std::vector<std::shared_ptr<MatrixFree<dim,double>>> matrix_free_objects;

      std::vector<std::shared_ptr<const Triangulation<dim, dim>>> trias;

      const Mapping<dim> &get_level_triangulation_mapping(const unsigned int level);

      std::unique_ptr<Mapping<dim>> level_triangulation_mapping;

      MGLevelObject<DoFHandler<dim>> dofhandlers_v;
      MGLevelObject<DoFHandler<dim>> dofhandlers_p;
      MGLevelObject<DoFHandler<dim>> dofhandlers_projection;

      MGLevelObject<AffineConstraints<double>> constraints_v;
      MGLevelObject<AffineConstraints<double>> constraints_p;

      MGLevelObject<MGTwoLevelTransfer<dim, ::LinearAlgebra::distributed::Vector<GMGNumberType>>> transfers_v;
      MGLevelObject<MGTwoLevelTransfer<dim, ::LinearAlgebra::distributed::Vector<GMGNumberType>>> transfers_p;
      std::unique_ptr<GCMGTransferType<dim,GMGNumberType>> mg_transfer_A_block;
      std::unique_ptr<GCMGTransferType<dim,GMGNumberType>> mg_transfer_Schur_complement;
  };
}

#endif