01_Utilize_Thyra.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"
 
 
using namespace std;
using Teuchos::RCP;
 
int main(int  /*argc*/, char * /*argv*/[])
{
  bool verbose = true;
  bool success = false;
  try {
    // Solution and its time-derivative.
    int vectorLength = 2;  // number state unknowns
    RCP<const Thyra::VectorSpaceBase<double> >  xSpace =
      Thyra::defaultSpmdVectorSpace<double>(vectorLength);
 
    RCP<Thyra::VectorBase<double> > x_n    = Thyra::createMember(xSpace);
    RCP<Thyra::VectorBase<double> > xDot_n = Thyra::createMember(xSpace);
 
    // Initial Conditions
    int n = 0;
    double time = 0.0;
    double epsilon = 1.0e-1;
    bool passed = true;   // ICs are considered passed.
    { // Scope to delete DetachedVectorViews
      Thyra::DetachedVectorView<double> x_n_view(*x_n);
      x_n_view[0] = 2.0;
      x_n_view[1] = 0.0;
      Thyra::DetachedVectorView<double> xDot_n_view(*xDot_n);
      xDot_n_view[0] = 0.0;
      xDot_n_view[1] = -2.0/epsilon;
    }
 
    // Timestep size
    double finalTime = 2.0;
    int nTimeSteps = 2000;
    const double constDT = finalTime/nTimeSteps;
 
    // Output
    cout << std::fixed;
    cout << std::setw(8)  << "index"
         << std::setw(10) << "time"
         << std::setw(12) << "x_0"
         << std::setw(12) << "x_1" << endl;
 
    cout << std::setw(8)  << n
         << std::setw(10) << std::setprecision(3) << time
         << std::setw(12) << std::setprecision(4) << get_ele(*x_n, 0)
         << std::setw(12) << std::setprecision(4) << get_ele(*x_n, 1) << endl;
 
    // Advance the solution to the next timestep.
    while (passed && time < finalTime && n < nTimeSteps) {
 
      // Initialize next time step
      RCP<Thyra::VectorBase<double> > x_np1 = x_n->clone_v(); // at time index n+1
 
      // Set the timestep and time.
      double dt = constDT;
      time = (n+1)*dt;
 
      // Righthand side evaluation and time-derivative at n.
      {
        Thyra::ConstDetachedVectorView<double> x_n_view(*x_n);
        Thyra::DetachedVectorView<double> xDot_n_view(*xDot_n);
        xDot_n_view[0] = x_n_view[1];
        xDot_n_view[1] =
          ((1.0-x_n_view[0]*x_n_view[0])*x_n_view[1]-x_n_view[0])/epsilon;
      }
 
      // Take the timestep - Forward Euler
      Thyra::V_VpStV(x_np1.ptr(), *x_n, dt, *xDot_n);
 
      // Test if solution has passed.
      if ( std::isnan(Thyra::norm(*x_np1)) ) {
        passed = false;
      } else {
        // Promote to next step (n <- n+1).
        Thyra::V_V(x_n.ptr(), *x_np1);
        n++;
      }
 
      // Output
      if (n % 100 == 0)
        cout << std::setw(8)  << n
             << std::setw(10) << std::setprecision(3) << time
             << std::setw(12) << std::setprecision(4) << get_ele(*x_n, 0)
             << std::setw(12) << std::setprecision(4) << get_ele(*x_n, 1) << endl;
    }
 
    // Test for regression.
    bool regressionPassed = tutorialRegressionTest(x_n);
 
    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 );
}