288 lines
8.1 KiB
C++
288 lines
8.1 KiB
C++
|
|
#include <iostream>
|
|
#include <boost/numeric/ublas/banded.hpp>
|
|
#include <boost/numeric/ublas/io.hpp>
|
|
#include <boost/numeric/ublas/operation.hpp>
|
|
#include <iomanip>
|
|
|
|
#include "utils.hpp"
|
|
|
|
using namespace boost::numeric::ublas;
|
|
|
|
int expected_index( size_t index, column_major ) {
|
|
// this is the data shown on http://www.netlib.org/lapack/lug/node124.html
|
|
// read column-by-column, aka column_major
|
|
int mapping[] = { 0, 11, 21, 31, 12, 22, 32, 42, 23, 33, 43, 53, 34, 44, 54, 0, 45, 55, 0, 0 };
|
|
return mapping[ index ];
|
|
}
|
|
|
|
|
|
int expected_index( size_t index, row_major ) {
|
|
// this is the data shown on http://www.netlib.org/lapack/lug/node124.html
|
|
// read row-by-row, aka row_major
|
|
int mapping[] = { 0, 0, 11, 12, 0, 21, 22, 23, 31, 32, 33, 34, 42, 43, 44, 45, 53, 54, 55, 0 };
|
|
return mapping[ index ];
|
|
}
|
|
|
|
int expected_index_6_by_5( size_t index, column_major ) {
|
|
// read column-by-column, aka column_major
|
|
int mapping[] = { 0, 11, 21, 31, 12, 22, 32, 42, 23, 33, 43, 53, 34, 44, 54, 64, 45, 55, 65, 0 };
|
|
return mapping[ index ];
|
|
}
|
|
|
|
int expected_index_6_by_5( size_t index, row_major ) {
|
|
// read row-by-row, aka row_major
|
|
int mapping[] = { 0, 0, 11, 12, 0, 21, 22, 23, 31, 32, 33, 34, 42, 43, 44, 45, 53, 54, 55, 0, 64, 65, 0, 0 };
|
|
return mapping[ index ];
|
|
}
|
|
|
|
int expected_index_5_by_6( size_t index, column_major ) {
|
|
// read column-by-column, aka column_major
|
|
int mapping[] = { 0, 11, 21, 31, 12, 22, 32, 42, 23, 33, 43, 53, 34, 44, 54, 0, 45, 55, 0, 0, 56, 0, 0, 0 };
|
|
return mapping[ index ];
|
|
}
|
|
|
|
int expected_index_5_by_6( size_t index, row_major ) {
|
|
// read row-by-row, aka row_major
|
|
int mapping[] = { 0, 0, 11, 12, 0, 21, 22, 23, 31, 32, 33, 34, 42, 43, 44, 45, 53, 54, 55, 56};
|
|
return mapping[ index ];
|
|
}
|
|
|
|
template< typename Orientation >
|
|
bool test_band_storage() {
|
|
|
|
int m = 5;
|
|
int n = 5;
|
|
int kl = 2;
|
|
int ku = 1;
|
|
|
|
banded_matrix< int, Orientation > test_matrix( m, n, kl, ku );
|
|
test_matrix.clear();
|
|
size_t band_storage_size = test_matrix.data().size();
|
|
|
|
test_matrix( 0, 0 ) = 11;
|
|
test_matrix( 0, 1 ) = 12;
|
|
test_matrix( 1, 0 ) = 21;
|
|
test_matrix( 1, 1 ) = 22;
|
|
test_matrix( 1, 2 ) = 23;
|
|
test_matrix( 2, 0 ) = 31;
|
|
test_matrix( 2, 1 ) = 32;
|
|
test_matrix( 2, 2 ) = 33;
|
|
test_matrix( 2, 3 ) = 34;
|
|
test_matrix( 3, 1 ) = 42;
|
|
test_matrix( 3, 2 ) = 43;
|
|
test_matrix( 3, 3 ) = 44;
|
|
test_matrix( 3, 4 ) = 45;
|
|
test_matrix( 4, 2 ) = 53;
|
|
test_matrix( 4, 3 ) = 54;
|
|
test_matrix( 4, 4 ) = 55;
|
|
|
|
BOOST_UBLAS_TEST_TRACE( "Full matrix" );
|
|
BOOST_UBLAS_TEST_TRACE( std::setw( 3 ) << test_matrix );
|
|
|
|
BOOST_UBLAS_TEST_TRACE( "data() of matrix" );
|
|
for ( size_t i = 0; i < band_storage_size; ++i ) {
|
|
std::cerr << test_matrix.data()[ i ] << " ";
|
|
}
|
|
std::cerr << std::endl;
|
|
|
|
BOOST_UBLAS_TEST_TRACE( "Expected data() of matrix" );
|
|
for ( size_t i = 0; i < band_storage_size; ++i ) {
|
|
std::cerr << expected_index( i, Orientation() ) << " ";
|
|
}
|
|
std::cerr << std::endl;
|
|
|
|
size_t mismatch = 0;
|
|
|
|
for ( size_t i = 0; i < band_storage_size; ++i ) {
|
|
if ( test_matrix.data()[ i ] != expected_index( i, Orientation() ) ) {
|
|
++mismatch;
|
|
}
|
|
}
|
|
|
|
return 0 == mismatch;
|
|
}
|
|
|
|
template< typename Orientation >
|
|
bool test_band_storage_6_by_5() {
|
|
|
|
int m = 6;
|
|
int n = 5;
|
|
int kl = 2;
|
|
int ku = 1;
|
|
|
|
|
|
banded_matrix< int, Orientation > test_matrix( m, n, kl, ku );
|
|
test_matrix.clear();
|
|
size_t band_storage_size = test_matrix.data().size();
|
|
|
|
test_matrix( 0, 0 ) = 11;
|
|
test_matrix( 0, 1 ) = 12;
|
|
test_matrix( 1, 0 ) = 21;
|
|
test_matrix( 1, 1 ) = 22;
|
|
test_matrix( 1, 2 ) = 23;
|
|
test_matrix( 2, 0 ) = 31;
|
|
test_matrix( 2, 1 ) = 32;
|
|
test_matrix( 2, 2 ) = 33;
|
|
test_matrix( 2, 3 ) = 34;
|
|
test_matrix( 3, 1 ) = 42;
|
|
test_matrix( 3, 2 ) = 43;
|
|
test_matrix( 3, 3 ) = 44;
|
|
test_matrix( 3, 4 ) = 45;
|
|
test_matrix( 4, 2 ) = 53;
|
|
test_matrix( 4, 3 ) = 54;
|
|
test_matrix( 4, 4 ) = 55;
|
|
test_matrix( 5, 3 ) = 64;
|
|
test_matrix( 5, 4 ) = 65;
|
|
|
|
BOOST_UBLAS_TEST_TRACE( "Full matrix" );
|
|
BOOST_UBLAS_TEST_TRACE( std::setw( 3 ) << test_matrix );
|
|
|
|
BOOST_UBLAS_TEST_TRACE( "data() of matrix" );
|
|
for ( size_t i = 0; i < band_storage_size; ++i ) {
|
|
std::cerr << test_matrix.data()[ i ] << " ";
|
|
}
|
|
std::cerr << std::endl;
|
|
|
|
BOOST_UBLAS_TEST_TRACE( "Expected data() of matrix" );
|
|
for ( size_t i = 0; i < band_storage_size; ++i ) {
|
|
std::cerr << expected_index_6_by_5( i, Orientation() ) << " ";
|
|
}
|
|
std::cerr << std::endl;
|
|
|
|
size_t mismatch = 0;
|
|
|
|
for ( size_t i = 0; i < band_storage_size; ++i ) {
|
|
if ( test_matrix.data()[ i ] != expected_index_6_by_5( i, Orientation() ) ) {
|
|
++mismatch;
|
|
}
|
|
}
|
|
|
|
return 0 == mismatch;
|
|
}
|
|
|
|
template< typename Orientation >
|
|
bool test_band_storage_5_by_6() {
|
|
|
|
int m = 5;
|
|
int n = 6;
|
|
int kl = 2;
|
|
int ku = 1;
|
|
|
|
banded_matrix< int, Orientation > test_matrix( m, n, kl, ku );
|
|
test_matrix.clear();
|
|
size_t band_storage_size = test_matrix.data().size();
|
|
|
|
test_matrix( 0, 0 ) = 11;
|
|
test_matrix( 0, 1 ) = 12;
|
|
test_matrix( 1, 0 ) = 21;
|
|
test_matrix( 1, 1 ) = 22;
|
|
test_matrix( 1, 2 ) = 23;
|
|
test_matrix( 2, 0 ) = 31;
|
|
test_matrix( 2, 1 ) = 32;
|
|
test_matrix( 2, 2 ) = 33;
|
|
test_matrix( 2, 3 ) = 34;
|
|
test_matrix( 3, 1 ) = 42;
|
|
test_matrix( 3, 2 ) = 43;
|
|
test_matrix( 3, 3 ) = 44;
|
|
test_matrix( 3, 4 ) = 45;
|
|
test_matrix( 4, 2 ) = 53;
|
|
test_matrix( 4, 3 ) = 54;
|
|
test_matrix( 4, 4 ) = 55;
|
|
test_matrix( 4, 5 ) = 56;
|
|
|
|
BOOST_UBLAS_TEST_TRACE( "Full matrix" );
|
|
BOOST_UBLAS_TEST_TRACE( std::setw( 3 ) << test_matrix );
|
|
|
|
BOOST_UBLAS_TEST_TRACE( "data() of matrix" );
|
|
for ( size_t i = 0; i < band_storage_size; ++i ) {
|
|
std::cerr << test_matrix.data()[ i ] << " ";
|
|
}
|
|
std::cerr << std::endl;
|
|
|
|
BOOST_UBLAS_TEST_TRACE( "Expected data() of matrix" );
|
|
for ( size_t i = 0; i < band_storage_size; ++i ) {
|
|
std::cerr << expected_index_5_by_6( i, Orientation() ) << " ";
|
|
}
|
|
std::cerr << std::endl;
|
|
|
|
size_t mismatch = 0;
|
|
|
|
for ( size_t i = 0; i < band_storage_size; ++i ) {
|
|
if ( test_matrix.data()[ i ] != expected_index_5_by_6( i, Orientation() ) ) {
|
|
++mismatch;
|
|
}
|
|
}
|
|
|
|
return 0 == mismatch;
|
|
}
|
|
|
|
|
|
|
|
|
|
BOOST_UBLAS_TEST_DEF( banded_matrix_column_major )
|
|
{
|
|
BOOST_UBLAS_TEST_TRACE( "Test case: storage layout banded_matrix < column_major >" );
|
|
|
|
BOOST_UBLAS_TEST_CHECK( test_band_storage< column_major >() );
|
|
}
|
|
|
|
BOOST_UBLAS_TEST_DEF( banded_matrix_row_major )
|
|
{
|
|
BOOST_UBLAS_TEST_TRACE( "Test case: storage layout banded_matrix < row_major >" );
|
|
|
|
BOOST_UBLAS_TEST_CHECK( test_band_storage< row_major >() );
|
|
}
|
|
|
|
BOOST_UBLAS_TEST_DEF( banded_matrix_column_major_6_by_5 )
|
|
{
|
|
BOOST_UBLAS_TEST_TRACE( "Test case: storage layout banded_matrix < column_major > 6x5" );
|
|
|
|
BOOST_UBLAS_TEST_CHECK( test_band_storage_6_by_5< column_major >() );
|
|
}
|
|
|
|
BOOST_UBLAS_TEST_DEF( banded_matrix_row_major_6_by_5 )
|
|
{
|
|
BOOST_UBLAS_TEST_TRACE( "Test case: storage layout banded_matrix < row_major > 6x5" );
|
|
|
|
BOOST_UBLAS_TEST_CHECK( test_band_storage_6_by_5< row_major >() );
|
|
}
|
|
|
|
BOOST_UBLAS_TEST_DEF( banded_matrix_column_major_5_by_6 )
|
|
{
|
|
BOOST_UBLAS_TEST_TRACE( "Test case: storage layout banded_matrix < column_major > 5x6" );
|
|
|
|
BOOST_UBLAS_TEST_CHECK( test_band_storage_5_by_6< column_major >() );
|
|
}
|
|
|
|
BOOST_UBLAS_TEST_DEF( banded_matrix_row_major_5_by_6 )
|
|
{
|
|
BOOST_UBLAS_TEST_TRACE( "Test case: storage layout banded_matrix < row_major > 5x6" );
|
|
|
|
BOOST_UBLAS_TEST_CHECK( test_band_storage_5_by_6< row_major >() );
|
|
}
|
|
|
|
int main()
|
|
{
|
|
|
|
BOOST_UBLAS_TEST_SUITE( "Test storage layout of banded matrix type" );
|
|
|
|
BOOST_UBLAS_TEST_TRACE( "Example data taken from http://www.netlib.org/lapack/lug/node124.html" );
|
|
|
|
BOOST_UBLAS_TEST_BEGIN();
|
|
|
|
BOOST_UBLAS_TEST_DO( banded_matrix_column_major );
|
|
|
|
BOOST_UBLAS_TEST_DO( banded_matrix_row_major );
|
|
|
|
BOOST_UBLAS_TEST_DO( banded_matrix_column_major_6_by_5 );
|
|
|
|
BOOST_UBLAS_TEST_DO( banded_matrix_row_major_6_by_5 );
|
|
|
|
BOOST_UBLAS_TEST_DO( banded_matrix_column_major_5_by_6 );
|
|
|
|
BOOST_UBLAS_TEST_DO( banded_matrix_row_major_5_by_6 );
|
|
|
|
BOOST_UBLAS_TEST_END();
|
|
}
|