Poisson.hh

Go to the documentation of this file.

/* Copyright (C) 2019, 2020, 2022 PISM Authors
 *
 * This file is part of PISM.
 *
 * PISM 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 3 of the License, or (at your option) any later
 * version.
 *
 * PISM 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 PISM; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 */
 
#include "pism/util/Logger.hh"
#include "pism/util/array/Scalar.hh"
#include "pism/util/petscwrappers/KSP.hh"
#include "pism/util/petscwrappers/Mat.hh"
 
namespace pism {
 
class Grid;
 
class Poisson {
public:
  Poisson(std::shared_ptr<const Grid> grid);
 
  int solve(const array::Scalar& mask, const array::Scalar& bc, double rhs,
            bool reuse_matrix = false);
 
  const array::Scalar &solution() const;
private:
  void assemble_matrix(const array::Scalar1 &mask, Mat A);
  void assemble_rhs(double rhs,
                    const array::Scalar &mask,
                    const array::Scalar &bc,
                    array::Scalar &b);
 
  std::shared_ptr<const Grid> m_grid;
  Logger::ConstPtr m_log;
  std::shared_ptr<petsc::DM> m_da;         // dof=1 DA used by the KSP solver
  petsc::KSP m_KSP;
  petsc::Mat m_A;
  array::Scalar m_b;
  array::Scalar m_x;
  array::Scalar1 m_mask;
};
 
} // end of namespace pism