sunlinsol_wrapper.h

Go to the documentation of this file.

//-----------------------------------------------------------
//
//    Copyright (C) 2021 by the deal.II authors
//
//    This file is part of the deal.II library.
//
//    The deal.II library is free software; you can use it, redistribute
//    it, and/or modify it under the terms of the GNU Lesser General
//    Public License as published by the Free Software Foundation; either
//    version 2.1 of the License, or (at your option) any later version.
//    The full text of the license can be found in the file LICENSE.md at
//    the top level directory of deal.II.
//
//-----------------------------------------------------------
 
#ifndef dealii_sundials_sunlinsol_wrapper_h
#define dealii_sundials_sunlinsol_wrapper_h
 
#include <deal.II/base/config.h>
 
#ifdef DEAL_II_WITH_SUNDIALS
#  if DEAL_II_SUNDIALS_VERSION_GTE(4, 0, 0)
 
#    include <sundials/sundials_linearsolver.h>
 
#    include <functional>
#    include <memory>
 
DEAL_II_NAMESPACE_OPEN
 
namespace SUNDIALS
{
#    ifndef DOXYGEN
  // forward declarations
  namespace internal
  {
    template <typename VectorType>
    struct LinearSolverContent;
  }
#    endif
 
  template <typename VectorType>
  struct SundialsOperator
  {
    void
    vmult(VectorType &dst, const VectorType &src) const;
 
    SundialsOperator(void *A_data, ATimesFn a_times_fn);
 
  private:
    void *A_data;
 
    ATimesFn a_times_fn;
  };
 
 
 
  template <typename VectorType>
  struct SundialsPreconditioner
  {
    void
    vmult(VectorType &dst, const VectorType &src) const;
 
    SundialsPreconditioner(void *P_data, PSolveFn p_solve_fn, double tol);
 
  private:
    void *P_data;
 
    PSolveFn p_solve_fn;
 
    double tol;
  };
 
  template <typename VectorType>
  using LinearSolveFunction =
    std::function<int(SundialsOperator<VectorType> &      op,
                      SundialsPreconditioner<VectorType> &prec,
                      VectorType &                        x,
                      const VectorType &                  b,
                      double                              tol)>;
 
  namespace internal
  {
    template <typename VectorType>
    class LinearSolverWrapper
    {
    public:
      explicit LinearSolverWrapper(LinearSolveFunction<VectorType> lsolve);
 
      ~LinearSolverWrapper();
 
      operator SUNLinearSolver();
 
    private:
      SUNLinearSolver                                  sun_linear_solver;
      std::unique_ptr<LinearSolverContent<VectorType>> content;
    };
  } // namespace internal
} // namespace SUNDIALS
 
DEAL_II_NAMESPACE_CLOSE
 
#  endif
#endif
#endif