06_Intro_TimeStepControl.cpp

Go to the documentation of this file.

//@HEADER
// *****************************************************************************
//          Tempus: Time Integration and Sensitivity Analysis Package
//
// Copyright 2017 NTESS and the Tempus contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
//@HEADER
 
#include <iomanip>
#include <iostream>
#include <stdlib.h>
#include <math.h>
#include "Teuchos_StandardCatchMacros.hpp"
 
#include "../00_Basic_Problem/Tutorial_Regression_Tester.hpp"
 
#include "Thyra_VectorStdOps.hpp"
#include "Thyra_DefaultSpmdVectorSpace.hpp"
#include "Thyra_DetachedVectorView.hpp"
 
#include "../02_Use_ModelEvaluator/VanDerPol_ModelEvaluator_02.hpp"
 
#include "Tempus_SolutionState.hpp"
#include "Tempus_SolutionHistory.hpp"
#include "Tempus_Stepper.hpp"
#include "Tempus_StepperForwardEuler.hpp"
#include "Tempus_TimeStepControl.hpp"
 
using namespace std;
using Teuchos::RCP;
 
int main(int argc, char *argv[])
{
  bool verbose = true;
  bool success = false;
  try {
    // Construct ModelEvaluator
    Teuchos::RCP<const Thyra::ModelEvaluator<double> >
      model = Teuchos::rcp(new VanDerPol_ModelEvaluator_02<double>());
 
    // Setup initial condition SolutionState
    auto solState = Tempus::createSolutionStateX(
                      model->getNominalValues().get_x()->clone_v());
    solState->setIndex   (0);
    solState->setTime    (0.0);
    solState->setTimeStep(0.0);  // By convention, the IC has dt = 0.
    solState->setSolutionStatus(Tempus::Status::PASSED);  // ICs are considered passed.
 
    // Create SolutionHistory
    auto solHistory = Tempus::createSolutionHistoryState<double>(solState);
 
    // Create and initialize StepperForwardEuler
    auto stepper = Teuchos::rcp(new Tempus::StepperForwardEuler<double>());
    stepper->setModel(model);
    stepper->initialize();
    stepper->setInitialConditions(solHistory);
 
    // Create and initialize TimeStepControl
    auto timeStepControl = Teuchos::rcp(new Tempus::TimeStepControl<double>());
    timeStepControl->setFinalTime(2.0);
    timeStepControl->setNumTimeSteps(2000);
    timeStepControl->initialize();
 
    Tempus::Status integratorStatus = Tempus::Status::WORKING;
 
    // Output
    cout << std::fixed;
    cout << std::setw(8)  << "index"
         << std::setw(10) << "time"
         << std::setw(12) << "x_0"
         << std::setw(12) << "x_1" << endl;
 
    auto currentState = solHistory->getCurrentState();
    cout << std::setw(8)  << currentState->getIndex()
         << std::setw(10) << std::setprecision(3) << currentState->getTime()
         << std::setw(12) << std::setprecision(4) << Thyra::get_ele(*(currentState->getX()), 0)
         << std::setw(12) << std::setprecision(4) << Thyra::get_ele(*(currentState->getX()), 1)
         << endl;
 
    // Advance the solution to the next timestep.
    while (solHistory->getCurrentState()->getSolutionStatus() == Tempus::Status::PASSED &&
           timeStepControl->timeInRange(solHistory->getCurrentTime()) &&
           timeStepControl->indexInRange(solHistory->getCurrentIndex())) {
 
      // Initialize next time step using SolutionHistory
      solHistory->initWorkingState();
 
      // Let TimeStepControl determine the next time-step metadata.
      timeStepControl->setNextTimeStep(solHistory, integratorStatus);
 
      // Take one Forward Euler step through Tempus::StepperForwardEuler
      stepper->takeStep(solHistory);
 
      // Promote working state to current state
      solHistory->promoteWorkingState();
 
      // Output
      if (solHistory->getCurrentState()->getIndex() % 100 == 0) {
        currentState = solHistory->getCurrentState();
        cout << std::setw(8)  << currentState->getIndex()
             << std::setw(10) << std::setprecision(3) << currentState->getTime()
             << std::setw(12) << std::setprecision(4) << Thyra::get_ele(*(currentState->getX()), 0)
             << std::setw(12) << std::setprecision(4) << Thyra::get_ele(*(currentState->getX()), 1)
             << endl;
      }
    }
 
    // Test for regression.
    auto finalState = solHistory->getCurrentState();
    bool passed = (finalState->getSolutionStatus() == Tempus::Status::PASSED);
    bool regressionPassed = tutorialRegressionTest(finalState);
 
    if (passed && regressionPassed) success = true;
  }
  TEUCHOS_STANDARD_CATCH_STATEMENTS(verbose, std::cerr, success);
 
  if(success)
    cout << "\nEnd Result: Test Passed!" << std::endl;
 
  return ( success ? EXIT_SUCCESS : EXIT_FAILURE );
}