PETScKrylovSolver.h

// Copyright (C) 2004-2015 Johan Jansson and Garth N. Wells
//
// This file is part of DOLFINx (https://www.fenicsproject.org)
//
// SPDX-License-Identifier:    LGPL-3.0-or-later
 
#pragma once
 
#include <dolfinx/common/MPI.h>
#include <petscksp.h>
#include <petscmat.h>
#include <petscvec.h>
#include <string>
 
namespace dolfinx::fem
{
class PETScDMCollection;
}
 
namespace dolfinx::la
{
 
 
class PETScKrylovSolver
{
public:
  explicit PETScKrylovSolver(MPI_Comm comm);
 
  PETScKrylovSolver(KSP ksp, bool inc_ref_count);
 
  // Copy constructor (deleted)
  PETScKrylovSolver(const PETScKrylovSolver& solver) = delete;
 
  PETScKrylovSolver(PETScKrylovSolver&& solver);
 
  ~PETScKrylovSolver();
 
  // Assignment operator (deleted)
  PETScKrylovSolver& operator=(const PETScKrylovSolver&) = delete;
 
  PETScKrylovSolver& operator=(PETScKrylovSolver&& solver);
 
  void set_operator(const Mat A);
 
  void set_operators(const Mat A, const Mat P);
 
  int solve(Vec x, const Vec b, bool transpose = false) const;
 
  void set_options_prefix(std::string options_prefix);
 
  std::string get_options_prefix() const;
 
  void set_from_options() const;
 
  KSP ksp() const;
 
  void set_dm(DM dm);
 
  void set_dm_active(bool val);
 
private:
  // PETSc solver pointer
  KSP _ksp;
};
} // namespace dolfinx::la