vector/test_10.cpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//               Rapid Optimization Library (ROL) Package
//
// Copyright 2014 NTESS and the ROL contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#include "Teuchos_GlobalMPISession.hpp"

#include "ROL_Objective.hpp"

#include "ROL_StdVector.hpp"
#include "ROL_PartitionedVector.hpp"
#include "ROL_RandomVector.hpp"
#include "ROL_VectorWorkspace.hpp"
#include "ROL_Stream.hpp"

#include <iomanip>

using RealT = double;
using size_type = typename std::vector<RealT>::size_type;

namespace details {

using namespace ROL;

// An alternative, more expensive way to compute the dot product!
template<typename Real> 
class PolarizationIdentity {
private:

  mutable VectorWorkspace<Real> workspace_;

public:

  Real dot( const Vector<Real>& x, const Vector<Real>& y ) {

    auto w = workspace_.copy(x);
    auto z = workspace_.copy(x);
    w->plus(y);
    z->axpy(-1.0,y);
    auto result = 0.25*( w->dot(*w) - z->dot(*z) );
    return result;
  } // w and z should go out of scope - decrement ref counts

  void status( std::ostream& os ) const { workspace_.status(os); }
};




} // namespace details

using details::PolarizationIdentity;


int main( int argc, char* argv[] ) {

  using namespace ROL;

  Teuchos::GlobalMPISession mpiSession(&argc, &argv);

  // This little trick lets us print to std::cout only if a (dummy) command-line argument is provided.
  auto outStream = makeStreamPtr( std::cout, argc > 1 );
  
  int errorFlag  = 0;
  RealT errtol = std::sqrt(ROL_EPSILON<RealT>());

  try {

    PolarizationIdentity<RealT> polar;

    size_type N = 20;

    auto xp = makePtr<std::vector<RealT>>(N);
    auto x = makePtr<StdVector<RealT>>(xp);
    auto y = x->clone();

    auto xx = PartitionedVector<RealT>::create({x,x});
    auto yy = PartitionedVector<RealT>::create({y,y});

    RandomizeVector( *x );
    RandomizeVector( *y );

    auto result = 0.5*polar.dot(*x,*y);
    result     += 0.5*polar.dot(*y,*x);

    auto x_dot_y = x->dot(*y);
    errorFlag += ( std::abs( x_dot_y - result ) > errtol );

    *outStream << std::setprecision(16) <<  x_dot_y << std::endl;
    *outStream << std::setprecision(16) <<  result  << std::endl;
 

    polar.dot(*xx,*yy);

    polar.status(*outStream);
    
  }
  catch (std::logic_error& err) {
    *outStream << err.what() << "\n";
    errorFlag = -1000;
  }; // end try

  if (errorFlag != 0)
    std::cout << "End Result: TEST FAILED\n";
  else
    std::cout << "End Result: TEST PASSED\n";

  return 0;
}