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