ROL
function/sketching/test_01.cpp
Go to the documentation of this file.
1// @HEADER
2// ************************************************************************
3//
4// Rapid Optimization Library (ROL) Package
5// Copyright (2014) Sandia Corporation
6//
7// Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
8// license for use of this work by or on behalf of the U.S. Government.
9//
10// Redistribution and use in source and binary forms, with or without
11// modification, are permitted provided that the following conditions are
12// met:
13//
14// 1. Redistributions of source code must retain the above copyright
15// notice, this list of conditions and the following disclaimer.
16//
17// 2. Redistributions in binary form must reproduce the above copyright
18// notice, this list of conditions and the following disclaimer in the
19// documentation and/or other materials provided with the distribution.
20//
21// 3. Neither the name of the Corporation nor the names of the
22// contributors may be used to endorse or promote products derived from
23// this software without specific prior written permission.
24//
25// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
26// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
28// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
29// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
30// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
31// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
32// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
33// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
34// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
35// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
36//
37// Questions? Contact lead developers:
38// Drew Kouri (dpkouri@sandia.gov) and
39// Denis Ridzal (dridzal@sandia.gov)
40//
41// ************************************************************************
42// @HEADER
43
49#include "ROL_StdVector.hpp"
50#include "ROL_Stream.hpp"
51#include "ROL_Sketch.hpp"
52#include "Teuchos_GlobalMPISession.hpp"
53
54#include <iostream>
55
56typedef double RealT;
57
58int main(int argc, char *argv[]) {
59
60 Teuchos::GlobalMPISession mpiSession(&argc, &argv);
61
62 // This little trick lets us print to std::cout only if a (dummy) command-line argument is provided.
63 int iprint = argc - 1;
64 ROL::Ptr<std::ostream> outStream;
65 ROL::nullstream bhs; // outputs nothing
66 if (iprint > 0)
67 outStream = ROL::makePtrFromRef(std::cout);
68 else
69 outStream = ROL::makePtrFromRef(bhs);
70
71 // Save the format state of the original std::cout.
72 ROL::nullstream oldFormatState;
73 oldFormatState.copyfmt(std::cout);
74
75 int errorFlag = 0;
76
77 // *** Test body.
78 try {
79
80 int nrow = 128, ncol = 100, rank = 1, testrank = 6;
81 ROL::Ptr<std::vector<RealT> > x_ptr = ROL::makePtr<std::vector<RealT>>(nrow, 0.0);
82 ROL::StdVector<RealT> x(x_ptr);
83
84 ROL::Sketch<RealT> sketch(x,ncol,rank,ROL::ROL_EPSILON<RealT>());
85 bool flag = true;
86 *outStream << std::endl;
87 for (int i = 0; i < testrank; ++i) {
88 *outStream << "Rank = " << i+1 << " Test Rank = " << testrank << std::endl;
89 sketch.setRank(i+1);
90 flag = sketch.test(testrank,*outStream,1);
91 }
92 errorFlag += (flag ? 0 : 1);
93 }
94 catch (std::logic_error& err) {
95 *outStream << err.what() << "\n";
96 errorFlag = -1000;
97 }; // end try
98
99 if (errorFlag != 0)
100 std::cout << "End Result: TEST FAILED\n";
101 else
102 std::cout << "End Result: TEST PASSED\n";
103
104 // reset format state of std::cout
105 std::cout.copyfmt(oldFormatState);
106
107 return 0;
108
109}
110
Defines a no-output stream class ROL::NullStream and a function makeStreamPtr which either wraps a re...
Provides an interface for randomized sketching.
void setRank(const int rank)
bool test(const int rank, std::ostream &outStream=std::cout, const int verbosity=0)
Provides the ROL::Vector interface for scalar values, to be used, for example, with scalar constraint...
int main(int argc, char *argv[])