\(\newcommand{\B}[1]{ {\bf #1} }\) \(\newcommand{\R}[1]{ {\rm #1} }\)

sparse_rcv_xam.cpp

View page source

C++: Sparsity Patterns: Example and Test

# include <cstdio>
# include <cppad/py/cppad_py.hpp>

bool sparse_rcv_xam(void) {
   using cppad_py::vec_int;
   using cppad_py::vec_double;
   using cppad_py::sparse_rc;
   using cppad_py::sparse_rcv;
   //
   // initialize return variable
   bool ok = true;
   //------------------------------------------------------------------------
   //
   // create sparsity pattern for n by n identity matrix
   sparse_rc pattern;
   int n = 5;
   pattern.resize(n, n, n);
   for(int k = 0; k < n; k++) {
      pattern.put(k, k, k);
   }
   //
   // create n by n sparse representation of identity matrix
   sparse_rcv matrix(pattern);
   for(int k = 0; k < n; k++) {
      matrix.put(k, 1.0);
   }
   //
   // row, column indices
   vec_int row = matrix.row();
   vec_int col = matrix.col();
   vec_double val = matrix.val();
   //
   // check results
   for(int k = 0; k < n; k++) {
      ok = ok && row[k] == k;
      ok = ok && col[k] == k;
      ok = ok && val[k] == 1.0;
   }
   //
   // For this case, row_major and col_major order are same as original order
   vec_int row_major = matrix.row_major();
   vec_int col_major = matrix.col_major();
   for(int k = 0; k < n; k++) {
      ok = ok && row_major[k] == k;
      ok = ok && col_major[k] == k;
   }
   //
   return( ok );
}