Amesos2LUSolver.h

// Copyright (C) 2015 Chris Richardson
//
// This file is part of DOLFIN.
//
// DOLFIN is free software: you can 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 3 of the License, or
// (at your option) any later version.
//
// DOLFIN 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 Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with DOLFIN. If not, see <http://www.gnu.org/licenses/>.
//

#ifndef __DOLFIN_AMESOS2_LU_SOLVER_H
#define __DOLFIN_AMESOS2_LU_SOLVER_H

#ifdef HAS_TRILINOS

#include "GenericLinearSolver.h"
#include <Amesos2_Factory.hpp>

namespace dolfin
{
  class GenericLinearOperator;
  class GenericVector;
  class TpetraMatrix;
  class TpetraVector;


  class Amesos2LUSolver : public GenericLinearSolver
  {
  public:

    Amesos2LUSolver(std::string method="default");

    Amesos2LUSolver(std::shared_ptr<const TpetraMatrix> A,
                    std::string method="default");

    ~Amesos2LUSolver();

    void set_operator(std::shared_ptr<const GenericLinearOperator> A);

    void set_operator(std::shared_ptr<const TpetraMatrix> A);

    const GenericLinearOperator& get_operator() const;

    std::size_t solve(GenericVector& x, const GenericVector& b);

    std::size_t solve(const GenericLinearOperator& A,
                      GenericVector& x,
                      const GenericVector& b);

    std::size_t solve(const TpetraMatrix& A,
                      TpetraVector& x,
                      const TpetraVector& b);

    std::string str(bool verbose) const;

    static std::map<std::string, std::string> methods();

    static Parameters default_parameters();

    std::string parameter_type() const
    { return "lu_solver"; }

  private:

    void init_solver(std::string& method);

    // Reference counted pointer (RCP) to solver
    Teuchos::RCP<Amesos2::Solver<TpetraMatrix::matrix_type,
                                 TpetraVector::vector_type>> _solver;

    // Operator (the matrix)
    std::shared_ptr<const TpetraMatrix> _matA;

    // Method name
    std::string _method_name;
  };

}

#endif

#endif