matrix_free_operators.h

Go to the documentation of this file.

/*
  Copyright (C) 2018 - 2024 by the authors of the ASPECT code.

  This file is part of ASPECT.

  ASPECT is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2, or (at your option)
  any later version.

  ASPECT is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with ASPECT; see the file LICENSE.  If not see
  <http://www.gnu.org/licenses/>.
*/

#ifndef _aspect_simulator_stokes_matrix_free_operators_h
#define _aspect_simulator_stokes_matrix_free_operators_h

#include <aspect/global.h>
#include <aspect/simulator/solver/interface.h>
#include <aspect/simulator.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>

#include <deal.II/lac/vector.h>
#include <deal.II/lac/block_vector.h>
#include <deal.II/lac/la_parallel_vector.h>
#include <deal.II/lac/la_parallel_block_vector.h>

namespace aspect
{
  namespace internal
  {
    namespace ChangeVectorTypes
    {
      void import(TrilinosWrappers::MPI::Vector &out,
                  const ::LinearAlgebra::ReadWriteVector<double> &rwv,
                  const VectorOperation::values                 operation);

      void copy(TrilinosWrappers::MPI::Vector &out,
                const ::LinearAlgebra::distributed::Vector<double> &in);

      void copy(::LinearAlgebra::distributed::Vector<double> &out,
                const TrilinosWrappers::MPI::Vector &in);

      void copy(TrilinosWrappers::MPI::BlockVector &out,
                const ::LinearAlgebra::distributed::BlockVector<double> &in);

      void copy(::LinearAlgebra::distributed::BlockVector<double> &out,
                const TrilinosWrappers::MPI::BlockVector &in);
    }
  }

  namespace MatrixFreeStokesOperators
  {

    template <int dim, typename number>
    struct OperatorCellData
    {
      bool is_compressible;

      double pressure_scaling;

      bool enable_newton_derivatives;

      bool enable_prescribed_dilation;

      bool symmetrize_newton_system;

      bool apply_stabilization_free_surface_faces;

      Table<2, VectorizedArray<number>> viscosity;

      Table<2, SymmetricTensor<2, dim, VectorizedArray<number>>> strain_rate_table;

      Table<2, VectorizedArray<number>> newton_factor_wrt_pressure_table;

      Table<2, SymmetricTensor<2, dim, VectorizedArray<number>>>
      newton_factor_wrt_strain_rate_table;

      Table<2, VectorizedArray<number>> dilation_lhs_term_table;

      Table<2, VectorizedArray<number>>
      dilation_derivative_wrt_pressure_table;

      Table<2, SymmetricTensor<2, dim, VectorizedArray<number>>>
      dilation_derivative_wrt_strain_rate_table;

      Table<2, Tensor<1, dim, VectorizedArray<number>>> free_surface_stabilization_term_table;

      std::set<types::boundary_id> free_surface_boundary_indicators;

      std::size_t
      memory_consumption() const;

      void clear();
    };

    template <int dim, int degree_v, typename number>
    class StokesOperator
      : public MatrixFreeOperators::Base<dim, ::LinearAlgebra::distributed::BlockVector<number>>
    {
      public:

        StokesOperator ();

        void clear () override;

        void set_cell_data (const OperatorCellData<dim,number> &data);

        void compute_diagonal () override;

      private:

        void apply_add (::LinearAlgebra::distributed::BlockVector<number> &dst,
                        const ::LinearAlgebra::distributed::BlockVector<number> &src) const override;

        void local_apply (const ::MatrixFree<dim, number> &data,
                          ::LinearAlgebra::distributed::BlockVector<number> &dst,
                          const ::LinearAlgebra::distributed::BlockVector<number> &src,
                          const std::pair<unsigned int, unsigned int> &cell_range) const;

        void local_apply_face (const ::MatrixFree<dim, number> &data,
                               ::LinearAlgebra::distributed::BlockVector<number> &dst,
                               const ::LinearAlgebra::distributed::BlockVector<number> &src,
                               const std::pair<unsigned int, unsigned int> &face_range) const;

        void local_apply_boundary_face (const ::MatrixFree<dim, number> &data,
                                        ::LinearAlgebra::distributed::BlockVector<number> &dst,
                                        const ::LinearAlgebra::distributed::BlockVector<number> &src,
                                        const std::pair<unsigned int, unsigned int> &face_range) const;

        const OperatorCellData<dim,number> *cell_data;
    };



    template <int dim, int degree_v, typename number>
    class BTBlockOperator
      : public MatrixFreeOperators::Base<dim, ::LinearAlgebra::distributed::BlockVector<number>>
    {
      public:

        BTBlockOperator ();

        void clear () override;

        void set_cell_data (const OperatorCellData<dim,number> &data);

        void compute_diagonal () override;

      private:

        void apply_add (::LinearAlgebra::distributed::BlockVector<number> &dst,
                        const ::LinearAlgebra::distributed::BlockVector<number> &src) const override;

        void local_apply (const ::MatrixFree<dim, number> &data,
                          ::LinearAlgebra::distributed::BlockVector<number> &dst,
                          const ::LinearAlgebra::distributed::BlockVector<number> &src,
                          const std::pair<unsigned int, unsigned int> &cell_range) const;

        void local_apply_face (const ::MatrixFree<dim, number> &data,
                               ::LinearAlgebra::distributed::BlockVector<number> &dst,
                               const ::LinearAlgebra::distributed::BlockVector<number> &src,
                               const std::pair<unsigned int, unsigned int> &face_range) const;

        const OperatorCellData<dim,number> *cell_data;
    };



    template <int dim, int degree_p, typename number>
    class MassMatrixOperator
      : public MatrixFreeOperators::Base<dim, ::LinearAlgebra::distributed::Vector<number>>
    {
      public:

        MassMatrixOperator ();

        void clear () override;

        void reinit(const Mapping<dim>              &mapping,
                    const DoFHandler<dim>           &dof_handler_v,
                    const DoFHandler<dim>           &dof_handler_p,
                    const AffineConstraints<number> &constraints_v,
                    const AffineConstraints<number> &constraints_p,
                    std::shared_ptr<MatrixFree<dim,double>> mf_storage,
                    const unsigned int level = numbers::invalid_unsigned_int);

        void set_cell_data (const OperatorCellData<dim,number> &data);

        void compute_diagonal () override;

      private:

        void apply_add (::LinearAlgebra::distributed::Vector<number> &dst,
                        const ::LinearAlgebra::distributed::Vector<number> &src) const override;

        void local_apply (const ::MatrixFree<dim, number> &data,
                          ::LinearAlgebra::distributed::Vector<number> &dst,
                          const ::LinearAlgebra::distributed::Vector<number> &src,
                          const std::pair<unsigned int, unsigned int> &cell_range) const;

        void inner_cell_operation(FEEvaluation<dim,
                                  degree_p,
                                  degree_p+2,
                                  1,
                                  number> &pressure) const;

        const OperatorCellData<dim,number> *cell_data;
    };

    template <int dim, int degree_v, typename number>
    class ABlockOperator
      : public MatrixFreeOperators::Base<dim, ::LinearAlgebra::distributed::Vector<number>>
    {
      public:

        ABlockOperator ();

        void clear () override;

        void reinit(const Mapping<dim>              &mapping,
                    const DoFHandler<dim>           &dof_handler_v,
                    const DoFHandler<dim>           &dof_handler_p,
                    const AffineConstraints<number> &constraints_v,
                    const AffineConstraints<number> &constraints_p,
                    std::shared_ptr<MatrixFree<dim,double>> mf_storage,
                    const unsigned int level = numbers::invalid_unsigned_int);
        void set_cell_data (const OperatorCellData<dim,number> &data);

        void compute_diagonal () override;

        void set_diagonal (const ::LinearAlgebra::distributed::Vector<number> &diag);

      private:
        void inner_cell_operation(FEEvaluation<dim,
                                  degree_v,
                                  degree_v+1,
                                  dim,
                                  number> &velocity) const;

        void cell_operation(FEEvaluation<dim,
                            degree_v,
                            degree_v+1,
                            dim,
                            number> &velocity) const;

        void apply_add (::LinearAlgebra::distributed::Vector<number> &dst,
                        const ::LinearAlgebra::distributed::Vector<number> &src) const override;

        void local_apply (const ::MatrixFree<dim, number> &data,
                          ::LinearAlgebra::distributed::Vector<number> &dst,
                          const ::LinearAlgebra::distributed::Vector<number> &src,
                          const std::pair<unsigned int, unsigned int> &cell_range) const;

        const OperatorCellData<dim,number> *cell_data;
    };
  }

}

#endif