Rythmos_StepControlInfo.hpp

#ifndef RYTHMOS_STEP_CONTROL_INFO_HPP
#define RYTHMOS_STEP_CONTROL_INFO_HPP

// disable clang warnings
#ifdef __clang__
#pragma clang system_header
#endif

#include "Rythmos_StepperSupportTypes.hpp"

namespace Rythmos {


template<class Scalar>
struct StepControlInfo {
  Scalar stepSize;
  StepSizeType stepType;
  bool limitedByBreakPoint;
  EBreakPointType breakPointType;
  StepControlInfo()
    :stepSize(-1.0), stepType(STEP_TYPE_FIXED),
     limitedByBreakPoint(false),
     breakPointType(BREAK_POINT_TYPE_SOFT)
    {}
};


template<class Scalar>
std::ostream& operator<<(
  std::ostream &out, const StepControlInfo<Scalar> &stepCtrlInfo
  )
{
  using std::endl;
  out
    << "stepType = " << toString(stepCtrlInfo.stepType) << endl
    << "stepSize = " << stepCtrlInfo.stepSize << endl
    << "limitedByBreakPoint = " << stepCtrlInfo.limitedByBreakPoint << endl
    << "breakPointType = " << toString(stepCtrlInfo.breakPointType) << endl;
  return out;
}


template<class Scalar>
StepControlInfo<Scalar>
stepCtrlInfoTaken(
  const StepControlInfo<Scalar> &trialStepCtrlInfo,
  const Scalar &stepSizeTaken
  )
{
  typedef Teuchos::ScalarTraits<Scalar> ST;
  const Scalar zero = ST::zero();
  StepControlInfo<Scalar> stepCtrlInfo = trialStepCtrlInfo;
  stepCtrlInfo.stepSize = stepSizeTaken;
  if ( trialStepCtrlInfo.stepSize > zero && stepSizeTaken > zero ) {
    if (stepSizeTaken < trialStepCtrlInfo.stepSize) {
      stepCtrlInfo.limitedByBreakPoint = false;
    }
  }
  else {
    stepCtrlInfo.limitedByBreakPoint = false;
  }
  return stepCtrlInfo;
}


// 2007/09/14: rabartl: ToDo: Above: Move this function into
// Rythmos_IntegratorBaseHelpers.hpp once created!


} // namespace Rythmos


#endif // RYTHMOS_STEP_CONTROL_INFO_HPP