Amesos2_SolverCore_decl.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//           Amesos2: Templated Direct Sparse Solver Package
//
// Copyright 2011 NTESS and the Amesos2 contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef AMESOS2_SOLVERCORE_DECL_HPP
#define AMESOS2_SOLVERCORE_DECL_HPP

#include <string>

#include <Teuchos_RCP.hpp>
#include <Teuchos_ParameterList.hpp>

#include "Amesos2_Solver.hpp"
#include "Amesos2_MatrixTraits.hpp"
#include "Amesos2_MatrixAdapter_decl.hpp"
#include "Amesos2_MultiVecAdapter_decl.hpp"
#include "Amesos2_TypeDecl.hpp"

#include "Amesos2_Control.hpp"
#include "Amesos2_Status.hpp"
#include "Amesos2_Timers.hpp"

namespace Amesos2 {


  /* This is the base class to be used in a *statically* polymorphic
   * way. E.g. for the Superlu solver:
   *
   * In Amesos2_Superlu.hpp:
   * class Superlu : SolverCore<Superlu> { ... }
   *
   * Each concrete solver will implement several private sub-functions
   * that will be called within the common code for each function.
   */

  template <template <class,class> class ConcreteSolver,
      class Matrix,
      class Vector >
  class SolverCore : public Amesos2::Solver<Matrix,Vector>
  {
  public:

    // Grant public access to contained types
    typedef SolverCore<ConcreteSolver,Matrix,Vector>                       type;
    typedef Solver<Matrix,Vector>                                    super_type;
    typedef ConcreteSolver<Matrix,Vector>                           solver_type;
    typedef Matrix                                                  matrix_type;
    typedef Vector                                                  vector_type;
    typedef typename MatrixAdapter<matrix_type>::scalar_t           scalar_type;
    typedef typename MatrixAdapter<matrix_type>::local_ordinal_t    local_ordinal_type;
    typedef typename MatrixAdapter<matrix_type>::global_ordinal_t   global_ordinal_type;
    typedef typename MatrixAdapter<matrix_type>::global_size_t      global_size_type;
    typedef typename MatrixAdapter<matrix_type>::node_t             node_type;



    SolverCore( Teuchos::RCP<const Matrix> A,
    Teuchos::RCP<Vector>       X,
    Teuchos::RCP<const Vector> B );


    ~SolverCore( );


    SolverCore(const solver_type& rhs);


    super_type& operator=(const solver_type* rhs);




    super_type& preOrdering() override;


    super_type& symbolicFactorization() override;


    super_type& numericFactorization() override;


    void solve() override;
    void solve(const Teuchos::Ptr<Vector> X, const Teuchos::Ptr<const Vector> B) const override;
    void solve(Vector* X, const Vector* B) const override;

    int solve_ir(const Teuchos::Ptr<Vector> X, const Teuchos::Ptr<const Vector> B,
                 const int maxNumIters, const bool verbose) const;
    int solve_ir(Vector* X, const Vector* B,
                 const int maxNumIters, const bool verbose) const;
    int solve_ir(const int maxNumIters, const bool verbose);



    bool matrixShapeOK() override;

    void setA( const Teuchos::RCP<const Matrix> a, EPhase keep_phase = CLEAN ) override;

    void setA( const Matrix* a, EPhase keep_phase = CLEAN ) override { setA(Teuchos::rcp(a), keep_phase); }

    void setX(const Teuchos::RCP<Vector> x) override { multiVecX_ = x; }

    void setX(Vector* x) override { multiVecX_ = Teuchos::rcp(x); }
  
    const Teuchos::RCP<Vector> getX() override { return( multiVecX_ ); }

    Vector* getXRaw() override { return multiVecX_.getRawPtr(); }

    void setB(const Teuchos::RCP<const Vector> b) override { multiVecB_ = b; }

    void setB(const Vector* b) override { multiVecB_ = Teuchos::rcp(b); }

    const Teuchos::RCP<const Vector> getB() override { return( multiVecB_ ); }

    const Vector* getBRaw() override { return multiVecB_.getRawPtr(); }




    super_type& setParameters(
            const Teuchos::RCP<Teuchos::ParameterList> & parameterList ) override;


    Teuchos::RCP<const Teuchos::ParameterList> getValidParameters() const override;


    void setParameterList(const Teuchos::RCP<Teuchos::ParameterList> & parameterList)
    {
      setParameters(parameterList);
    }


    Teuchos::RCP<Teuchos::ParameterList> getNonconstParameterList()
    {
      return Teuchos::null;
    }


    Teuchos::RCP<Teuchos::ParameterList> unsetParameterList()
    {
      return Teuchos::null;
    }






    Teuchos::RCP<const Teuchos::Comm<int> > getComm() const override
    {
      return matrixA_->getComm();
    }

  
    inline Status& getStatus() const override { return( status_ ); }




    std::string description() const override;


    void describe(Teuchos::FancyOStream &out,
      const Teuchos::EVerbosityLevel verbLevel=Teuchos::Describable::verbLevel_default) const override;


    void printTiming(
         Teuchos::FancyOStream &out,
         const Teuchos::EVerbosityLevel verbLevel) const override;


    void getTiming(Teuchos::ParameterList& timingParameterList) const override;


    std::string name() const override;

  private:

    void loadA(EPhase current_phase);

  protected:

    void setNnzLU(size_t nnz){ status_.lu_nnz_ = nnz; }

    Teuchos::RCP<const MatrixAdapter<Matrix> > matrixA_;

    bool matrix_loaded_;


    Teuchos::RCP<Vector> multiVecX_;

    Teuchos::RCP<const Vector> multiVecB_;

    global_size_type globalNumRows_;

    global_size_type globalNumCols_;

    global_size_type globalNumNonZeros_;

    global_size_type rowIndexBase_;

    global_size_type columnIndexBase_;

    mutable Status status_;

    Control control_;

    mutable Timers timers_;


    /* Useful MPI vars */

    int rank_;

    bool root_;

    int nprocs_;

  };        // End class Amesos2::SolverCore


} // end namespace Amesos2

#endif  // AMESOS2_SOLVERCORE_DECL_HPP