TrilinosNOXSolver.h

Go to the documentation of this file.

#ifndef included_AMP_TrilinosNOXSolver
#define included_AMP_TrilinosNOXSolver
 
// AMP includes
#include "AMP/AMP_TPLs.h"
#include "AMP/solvers/SolverStrategy.h"
#include "AMP/solvers/trilinos/nox/TrilinosNOXSolverParameters.h"
 
 
// Trilinos includes
DISABLE_WARNINGS
#include "Teuchos_RCP.hpp"
ENABLE_WARNINGS
 
 
// Forward declare Trilinos classes (helps protect from nvcc_wrapper)
// clang-format off
namespace NOX::Solver { class Generic; }
namespace NOX::StatusTest { class Combo; }
namespace Teuchos { class ParameterList; }
namespace Thyra { template<class> class PreconditionerBase; }
namespace Thyra { template<class> class LinearOpWithSolveFactoryBase; }
// clang-format on
 
 
namespace AMP::Solver {
 
 
class TrilinosThyraModelEvaluator;
 
 
class TrilinosNOXSolver : public SolverStrategy
{
public:
    TrilinosNOXSolver();
 
    explicit TrilinosNOXSolver( std::shared_ptr<SolverStrategyParameters> parameters );
 
    virtual ~TrilinosNOXSolver();
 
    static std::unique_ptr<SolverStrategy>
    createSolver( std::shared_ptr<SolverStrategyParameters> solverStrategyParameters )
    {
        return std::make_unique<TrilinosNOXSolver>( solverStrategyParameters );
    }
 
    std::string type() const override { return "TrilinosNOXSolver"; }
 
    void reset( std::shared_ptr<SolverStrategyParameters> parameters ) override;
 
    void apply( std::shared_ptr<const AMP::LinearAlgebra::Vector> f,
                std::shared_ptr<AMP::LinearAlgebra::Vector> u ) override;
 
    void setInitialGuess( std::shared_ptr<AMP::LinearAlgebra::Vector> initialGuess ) override
    {
        d_initialGuess = initialGuess;
    }
 
 
protected:
    void initialize( std::shared_ptr<const SolverStrategyParameters> parameters ) override;
 
    std::shared_ptr<SolverStrategy>
    createPreconditioner( std::shared_ptr<AMP::Database> pc_solver_db );
 
    AMP_MPI d_comm;
 
    AMP::LinearAlgebra::Vector::shared_ptr d_initialGuess;
 
    Teuchos::RCP<NOX::Solver::Generic> d_solver;
 
    Teuchos::RCP<TrilinosThyraModelEvaluator> d_thyraModel;
    Teuchos::RCP<Teuchos::ParameterList> d_nlParams;
    Teuchos::RCP<Thyra::PreconditionerBase<double>> d_precOp;
    Teuchos::RCP<::Thyra::LinearOpWithSolveFactoryBase<double>> d_lowsFactory;
    Teuchos::RCP<NOX::StatusTest::Combo> d_status;
};
} // namespace AMP::Solver
 
#endif