Thyra_GeneralSolveCriteriaBelosStatusTest_decl.hpp

// @HEADER
// *****************************************************************************
//         Stratimikos: Thyra-based strategies for linear solvers
//
// Copyright 2006 NTESS and the Stratimikos contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER
 
#ifndef THYRA_GENERAL_SOLVE_CRITERIA_BELOS_STATUS_TEST_DECL_HPP
#define THYRA_GENERAL_SOLVE_CRITERIA_BELOS_STATUS_TEST_DECL_HPP
 
#include "Thyra_SolveSupportTypes.hpp"
#include "BelosStatusTest.hpp"
#include "BelosLinearProblem.hpp"
#include "BelosMultiVecTraits.hpp"
#include "Teuchos_VerboseObject.hpp"
 
 
namespace Thyra {
 

template<class Scalar>
class GeneralSolveCriteriaBelosStatusTest
  : public Belos::StatusTest<Scalar, MultiVectorBase<Scalar>, LinearOpBase<Scalar> >,
    public Teuchos::VerboseObject<GeneralSolveCriteriaBelosStatusTest<Scalar> >
{
public:


  typedef MultiVectorBase<Scalar> MV;
  typedef LinearOpBase<Scalar> OP;
  typedef typename ScalarTraits<Scalar>::magnitudeType ScalarMag;


  GeneralSolveCriteriaBelosStatusTest();

  void setSolveCriteria(const SolveCriteria<Scalar> &solveCriteria,
    const int convergenceTestFrequency);

  ArrayView<const ScalarMag> achievedTol() const;

  

  virtual Belos::StatusType checkStatus(Belos::Iteration<Scalar,MV,OP> *iSolver);
  virtual Belos::StatusType getStatus() const;
  virtual void reset();
  virtual void print(std::ostream& os, int indent) const;
 
private:
 
  SolveCriteria<Scalar> solveCriteria_;
  int convergenceTestFrequency_;
 
  bool compute_x_;
  bool compute_r_;
 
  Array<ScalarMag> r0_nrm_;
  Array<ScalarMag> b_nrm_;
  Array<ScalarMag> lastNumerator_;
  Array<ScalarMag> lastDenominator_;
  Array<ScalarMag> lastAchievedTol_;
  int lastCurrIter_;
  Belos::StatusType lastRtnStatus_;
 
  // Private member functions
 
  ScalarMag computeReductionFunctional(ESolveMeasureNormType measureType,
    const Ptr<const ReductionFunctional<Scalar> > &reductFunc,
    const Ptr<const VectorBase<Scalar> > &x,
    const Ptr<const VectorBase<Scalar> > &r
    ) const;
 
  void printRhsStatus(const int currIter, const int j, std::ostream &out,
    int indent = 0) const;
 
};
 

template<class Scalar>
RCP<GeneralSolveCriteriaBelosStatusTest<Scalar> >
createGeneralSolveCriteriaBelosStatusTest(
  const SolveCriteria<Scalar> &solveCriteria,
  const int convergenceTestFrequency
  )
{
  RCP<GeneralSolveCriteriaBelosStatusTest<Scalar> >
    gscbst = Teuchos::rcp(new GeneralSolveCriteriaBelosStatusTest<Scalar>);
  gscbst->setSolveCriteria(solveCriteria, convergenceTestFrequency);
  return gscbst;
}
 
 
} // namespace Thyra
 
 
#endif  // THYRA_GENERAL_SOLVE_CRITERIA_BELOS_STATUS_TEST_DECL_HPP