sparse_view_test.cpp 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
  1. // Copyright (c) 2009-2011 Gunter Winkler, David Bellot
  2. //
  3. // Distributed under the Boost Software License, Version 1.0. (See
  4. // accompanying file LICENSE_1_0.txt or copy at
  5. // http://www.boost.org/LICENSE_1_0.txt)
  6. // ublas headers
  7. #include <boost/numeric/ublas/experimental/sparse_view.hpp>
  8. #include <boost/numeric/ublas/matrix.hpp>
  9. #include <boost/numeric/ublas/matrix_sparse.hpp>
  10. #include <boost/numeric/ublas/io.hpp>
  11. #include <boost/numeric/ublas/traits/c_array.hpp>
  12. // other boost headers
  13. // headers for testcase
  14. #define BOOST_TEST_MODULE SparseMatrixErasureTest
  15. #include <boost/test/included/unit_test.hpp>
  16. // standard and system headers
  17. #include <iostream>
  18. #include <string>
  19. namespace ublas = boost::numeric::ublas;
  20. /*
  21. sparse input matrix:
  22. 1 2 0 0
  23. 0 3 9 0
  24. 0 1 4 0
  25. */
  26. static const std::string inputMatrix = "[3,4]((1,2,0,0),(0,3,9,0),(0,1,4,0))\n";
  27. const unsigned int NNZ = 6;
  28. const unsigned int IB = 1;
  29. const double VA[] = { 1.0, 2.0, 3.0, 9.0, 1.0, 4.0 };
  30. const unsigned int IA[] = { 1, 3, 5, 7 };
  31. const unsigned int JA[] = { 1, 2, 2, 3, 2, 3 };
  32. BOOST_AUTO_TEST_CASE( test_construction_and_basic_operations )
  33. {
  34. typedef ublas::matrix<double> DENSE_MATRIX;
  35. // prepare data
  36. DENSE_MATRIX A;
  37. std::istringstream iss(inputMatrix);
  38. iss >> A;
  39. std::cout << A << std::endl;
  40. std::cout << ( ublas::make_compressed_matrix_view<ublas::row_major,IB>(3,4,NNZ,IA,JA,VA) ) << std::endl;
  41. typedef ublas::compressed_matrix_view<ublas::row_major, IB, unsigned int [4], unsigned int [NNZ], double[NNZ]> COMPMATVIEW;
  42. COMPMATVIEW viewA(3,4,NNZ,IA,JA,VA);
  43. std::cout << viewA << std::endl;
  44. }
  45. BOOST_AUTO_TEST_CASE( test_construction_from_pointers )
  46. {
  47. std::cout << ( ublas::make_compressed_matrix_view<ublas::column_major,IB>(4,3,NNZ
  48. , ublas::c_array_view<const unsigned int>(4,&(IA[0]))
  49. , ublas::c_array_view<const unsigned int>(6,&(JA[0]))
  50. , ublas::c_array_view<const double>(6,&(VA[0]))) ) << std::endl;
  51. unsigned int * ia = new unsigned int[4]();
  52. unsigned int * ja = new unsigned int[6]();
  53. double * va = new double[6]();
  54. std::copy(&(IA[0]),&(IA[4]),ia);
  55. std::copy(&(JA[0]),&(JA[6]),ja);
  56. std::copy(&(VA[0]),&(VA[6]),va);
  57. typedef ublas::compressed_matrix_view<ublas::column_major
  58. , IB
  59. , ublas::c_array_view<unsigned int>
  60. , ublas::c_array_view<unsigned int>
  61. , ublas::c_array_view<double> > COMPMATVIEW;
  62. COMPMATVIEW viewA(4,3,NNZ
  63. , ublas::c_array_view<unsigned int>(4,ia)
  64. , ublas::c_array_view<unsigned int>(6,ja)
  65. , ublas::c_array_view<double>(6,va));
  66. std::cout << viewA << std::endl;
  67. delete[] va;
  68. delete[] ja;
  69. delete[] ia;
  70. }