BelosDenseSolver.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//                 Belos: Block Linear Solvers Package
//
// Copyright 2004-2016 NTESS and the Belos contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER
//
#ifndef BELOS_DENSE_SOLVER_HPP
#define BELOS_DENSE_SOLVER_HPP
 
#include "Teuchos_RCP.hpp"
#include "Teuchos_BLAS_types.hpp"
 
namespace Belos {
 
  template<class ScalarType, class DM>
  class DenseSolver 
  {
  public:
 
 
 
    DenseSolver()
    : spd_(false),
      equilibrate_(false),
      TRANS_(Teuchos::NO_TRANS),
      newMatrix_(false)
    {}
 
    virtual ~DenseSolver() {}
 
 
 
    virtual int setMatrix(const Teuchos::RCP<DM>& A) { A_ = A; newMatrix_ = true; return 0; }
 
 
    virtual int setVectors(const Teuchos::RCP<DM>& X, const Teuchos::RCP<DM>& B) { X_ = X; B_ = B; return 0; }
 
 
 
    void setSPD(bool flag) {spd_ = flag; return;}
 
 
    void factorWithEquilibration(bool flag) {equilibrate_ = flag; return;}
 
 
    void solveWithTransposeFlag(Teuchos::ETransp trans) {TRANS_ = trans; return;}
 
 
 
 
    virtual int factor() = 0;
 
 
    virtual int solve() = 0;
 
  protected:
  
    bool spd_, equilibrate_;
    Teuchos::ETransp TRANS_;
 
    bool newMatrix_; 
    Teuchos::RCP<DM> A_, X_, B_;
  };
 
} // namespace Belos
 
#endif // end file BELOS_DENSE_SOLVER_HPP