set_to_zero_test.cpp

来自「矩阵运算源码最新版本」· C++ 代码 · 共 75 行

// Software License for MTL
// 
// Copyright (c) 2007 The Trustees of Indiana University. All rights reserved.
// Authors: Peter Gottschling and Andrew Lumsdaine
// 
// This file is part of the Matrix Template Library
// 
// See also license.mtl.txt in the distribution.

#include <iostream>
#include <complex>
#include <cmath>
#include <boost/test/minimal.hpp>

#include <boost/numeric/mtl/matrix/dense2D.hpp>
#include <boost/numeric/mtl/matrix/morton_dense.hpp> 
#include <boost/numeric/mtl/matrix/compressed2D.hpp> 
#include <boost/numeric/mtl/matrix/transposed_view.hpp>

#include <boost/numeric/mtl/recursion/predefined_masks.hpp>
#include <boost/numeric/mtl/operation/print.hpp>
#include <boost/numeric/mtl/operation/set_to_zero.hpp>


using namespace mtl;
using namespace mtl::recursion;
using namespace std;  


template <typename Matrix>
void test(Matrix& matrix, const char* name)
{
    {
	matrix::inserter<Matrix> ins(matrix);
	for (int i= 0; i < matrix.num_rows(); i++)
	    for (int j= 0; j < matrix.num_cols(); j++)
		if ((i + j) & 1)
		    ins(i, j) << i + 2*j;
    }

    std::cout << "\n" << name << "\n" << matrix << "\n";
    set_to_zero(matrix);
    std::cout << "should be empty now:\n" << matrix << "\n";
    typename Matrix::value_type zero(0.0);
    if (matrix(0, 1) != zero)
	throw "not properly set to zero";
}


int test_main(int argc, char* argv[])
{
    dense2D<double>                                dr(5, 7);
    dense2D<double, matrix::parameters<col_major> > dc(5, 7);
    dense2D<std::complex<double> >                 cdr(5, 7);
    morton_dense<double,  morton_mask>             md(5, 7);
    morton_dense<double,  doppled_16_row_mask>     d16r(5, 7);
    compressed2D<double>                           comp(5, 7);
    compressed2D<std::complex<double> >            ccomp(5, 7);

    test(dr, "Dense row major");
    test(dc, "Dense column major");
    test(cdr, "Complex dense row major");
    test(md, "Morton N-order");
    test(d16r, "Hybrid 16 row-major");
    test(comp, "compressed2D");
    test(ccomp, "complex compressed2D");

    return 0;
}