stokes_matrix_free_local_smoothing.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_local_smoothing_h
#define _aspect_simulator_solver_stokes_matrix_free_local_smoothing_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 StokesMatrixFreeHandlerLocalSmoothingImplementation: public StokesMatrixFreeHandler<dim>
  {
    public:
      StokesMatrixFreeHandlerLocalSmoothingImplementation(Simulator<dim> &simulator,
                                                          const Parameters<dim> &parameters);

      ~StokesMatrixFreeHandlerLocalSmoothingImplementation() 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;

      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;

      const DoFHandler<dim> &get_velocity_multigrid_dof_handler() const
      {
        return dof_handler_v;
      }

      const MGConstrainedDoFs &get_velocity_multigrid_constraints() const
      {
        return mg_constrained_dofs_A_block;
      }

      const MGTransferType<dim,GMGNumberType> &get_velocity_multigrid_transfer() const
      {
        return mg_transfer_A_block;
      }

    private:
      void evaluate_material_model();

      void correct_stokes_rhs(LinearAlgebra::BlockVector &system_rhs);


      Simulator<dim> &sim;

      bool print_details;

      bool do_timings;

      double minimum_viscosity;
      double maximum_viscosity;

      DoFHandler<dim> dof_handler_v;
      DoFHandler<dim> dof_handler_p;
      DoFHandler<dim> dof_handler_projection;

      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;

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

      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;

      MGTransferType<dim,GMGNumberType> mg_transfer_A_block;
      MGTransferType<dim,GMGNumberType> mg_transfer_Schur_complement;

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

#endif