petsc_solver.h

Go to the documentation of this file.

// -----------------------------------------------------------------------------
//
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception OR LGPL-2.1-or-later
// Copyright (C) 2004 - 2026 by the deal.II authors
//
// This file is part of the deal.II library.
//
// Detailed license information governing the source code and contributions
// can be found in LICENSE.md and CONTRIBUTING.md at the top level directory.
//
// -----------------------------------------------------------------------------
 
#ifndef dealii_petsc_solver_h
#define dealii_petsc_solver_h
 
 
#include <deal.II/base/config.h>
 
#ifdef DEAL_II_WITH_PETSC
 
#  include <deal.II/base/mpi_stub.h>
#  include <deal.II/base/observer_pointer.h>
 
#  include <deal.II/lac/exceptions.h>
#  include <deal.II/lac/solver_control.h>
 
#  include <petscksp.h>
 
#endif // DEAL_II_WITH_PETSC
 
DEAL_II_NAMESPACE_OPEN
 
#ifdef DEAL_II_WITH_PETSC
// Forward declarations
#  ifndef DOXYGEN
#    ifdef DEAL_II_WITH_SLEPC
namespace SLEPcWrappers
{
  // forward declarations
  class TransformationBase;
} // namespace SLEPcWrappers
#    endif
#  endif
 
namespace PETScWrappers
{
  // forward declarations
#  ifndef DOXYGEN
  class MatrixBase;
  class VectorBase;
  class PreconditionBase;
#  endif
 
 
  class SolverBase
  {
  public:
    SolverBase();
 
    SolverBase(SolverControl &cn);
 
    virtual ~SolverBase();
 
    void
    solve(const MatrixBase       &A,
          VectorBase             &x,
          const VectorBase       &b,
          const PreconditionBase &preconditioner);
 
    virtual void
    reset();
 
    void
    set_prefix(const std::string &prefix);
 
    SolverControl &
    control() const;
 
    void
    initialize(const PreconditionBase &preconditioner);
 
    KSP
    petsc_ksp();
 
    operator KSP() const;
 
  protected:
    KSP ksp;
 
    ObserverPointer<SolverControl, SolverBase> solver_control;
 
    void
    initialize_ksp_with_comm(const MPI_Comm comm);
 
    virtual void
    set_solver_type(KSP &ksp) const;
 
    void
    perhaps_set_convergence_test() const;
 
    std::string prefix_name;
 
  private:
    static PetscErrorCode
    convergence_test(KSP                 ksp,
                     const PetscInt      iteration,
                     const PetscReal     residual_norm,
                     KSPConvergedReason *reason,
                     void               *solver_control);
 
 
#  ifdef DEAL_II_WITH_SLEPC
    // Make the transformation class a friend, since it needs to set the KSP
    // solver.
    friend class SLEPcWrappers::TransformationBase;
#  endif
  };
 
 
 
  class SolverRichardson : public SolverBase
  {
  public:
    struct AdditionalData
    {
      explicit AdditionalData(const double omega = 1);
 
      double omega;
    };
 
    SolverRichardson(SolverControl        &cn,
                     const AdditionalData &data = AdditionalData());
 
  protected:
    const AdditionalData additional_data;
 
    virtual void
    set_solver_type(KSP &ksp) const override;
  };
 
 
 
  class SolverChebychev : public SolverBase
  {
  public:
    struct AdditionalData
    {};
 
    SolverChebychev(SolverControl        &cn,
                    const AdditionalData &data = AdditionalData());
 
  protected:
    const AdditionalData additional_data;
 
    virtual void
    set_solver_type(KSP &ksp) const override;
  };
 
 
 
  class SolverCG : public SolverBase
  {
  public:
    struct AdditionalData
    {};
 
    SolverCG(SolverControl &cn, const AdditionalData &data = AdditionalData());
 
  protected:
    const AdditionalData additional_data;
 
    virtual void
    set_solver_type(KSP &ksp) const override;
  };
 
 
 
  class SolverBiCG : public SolverBase
  {
  public:
    struct AdditionalData
    {};
 
    SolverBiCG(SolverControl        &cn,
               const AdditionalData &data = AdditionalData());
 
  protected:
    const AdditionalData additional_data;
 
    virtual void
    set_solver_type(KSP &ksp) const override;
  };
 
 
 
  class SolverGMRES : public SolverBase
  {
  public:
    struct AdditionalData
    {
      AdditionalData(const unsigned int restart_parameter     = 30,
                     const bool         right_preconditioning = false);
 
      unsigned int restart_parameter;
 
      bool right_preconditioning;
    };
 
    SolverGMRES(SolverControl        &cn,
                const AdditionalData &data = AdditionalData());
 
  protected:
    const AdditionalData additional_data;
 
    virtual void
    set_solver_type(KSP &ksp) const override;
  };
 
 
 
  class SolverBicgstab : public SolverBase
  {
  public:
    struct AdditionalData
    {};
 
    SolverBicgstab(SolverControl        &cn,
                   const AdditionalData &data = AdditionalData());
 
  protected:
    const AdditionalData additional_data;
 
    virtual void
    set_solver_type(KSP &ksp) const override;
  };
 
 
 
  class SolverCGS : public SolverBase
  {
  public:
    struct AdditionalData
    {};
 
    SolverCGS(SolverControl &cn, const AdditionalData &data = AdditionalData());
 
  protected:
    const AdditionalData additional_data;
 
    virtual void
    set_solver_type(KSP &ksp) const override;
  };
 
 
 
  class SolverTFQMR : public SolverBase
  {
  public:
    struct AdditionalData
    {};
 
    SolverTFQMR(SolverControl        &cn,
                const AdditionalData &data = AdditionalData());
 
  protected:
    const AdditionalData additional_data;
 
    virtual void
    set_solver_type(KSP &ksp) const override;
  };
 
 
 
  class SolverTCQMR : public SolverBase
  {
  public:
    struct AdditionalData
    {};
 
    SolverTCQMR(SolverControl        &cn,
                const AdditionalData &data = AdditionalData());
 
  protected:
    const AdditionalData additional_data;
 
    virtual void
    set_solver_type(KSP &ksp) const override;
  };
 
 
 
  class SolverCR : public SolverBase
  {
  public:
    struct AdditionalData
    {};
 
    SolverCR(SolverControl &cn, const AdditionalData &data = AdditionalData());
 
  protected:
    const AdditionalData additional_data;
 
    virtual void
    set_solver_type(KSP &ksp) const override;
  };
 
 
 
  class SolverLSQR : public SolverBase
  {
  public:
    struct AdditionalData
    {};
 
    SolverLSQR(SolverControl        &cn,
               const AdditionalData &data = AdditionalData());
 
  protected:
    const AdditionalData additional_data;
 
    virtual void
    set_solver_type(KSP &ksp) const override;
  };
 
 
  class SolverPreOnly : public SolverBase
  {
  public:
    struct AdditionalData
    {};
 
    SolverPreOnly(SolverControl        &cn,
                  const AdditionalData &data = AdditionalData());
 
  protected:
    const AdditionalData additional_data;
 
    virtual void
    set_solver_type(KSP &ksp) const override;
  };
 
  class SparseDirectMUMPS : public SolverBase
  {
  public:
    struct AdditionalData
    {};
 
    SparseDirectMUMPS(SolverControl        &cn,
                      const AdditionalData &data = AdditionalData());
 
    void
    solve(const MatrixBase &A, VectorBase &x, const VectorBase &b);
 
    void
    set_symmetric_mode(const bool matrix_is_symmetric);
 
  protected:
    const AdditionalData additional_data;
 
    virtual void
    set_solver_type(KSP &ksp) const override;
 
    bool symmetric_mode;
  };
} // namespace PETScWrappers
 
#endif // DEAL_II_WITH_PETSC
 
DEAL_II_NAMESPACE_CLOSE
 
#endif