From 8fb1678f0f174e85f6550e14f349841e406c8f53 Mon Sep 17 00:00:00 2001 From: Daniel Gomez Ferro Date: Tue, 2 Sep 2008 15:28:49 +0000 Subject: [PATCH] Extended sparse unit-test: nested blocks and InnerIterators. Block specialization for sparse matrices. InnerIterators for Blocks and fixes in CoreIterators. --- Eigen/Sparse | 1 + Eigen/src/Core/Block.h | 5 + Eigen/src/Core/MatrixBase.h | 2 +- Eigen/src/Core/util/Constants.h | 3 +- Eigen/src/Core/util/ForwardDeclarations.h | 3 +- Eigen/src/Sparse/CoreIterators.h | 80 +++++++++++++- Eigen/src/Sparse/SparseBlock.h | 122 ++++++++++++++++++++++ Eigen/src/Sparse/SparseMatrix.h | 4 +- Eigen/src/Sparse/SparseMatrixBase.h | 19 +++- test/sparse.cpp | 50 ++++++++- 10 files changed, 276 insertions(+), 13 deletions(-) create mode 100644 Eigen/src/Sparse/SparseBlock.h diff --git a/Eigen/Sparse b/Eigen/Sparse index 53bc4eccc..89fa387ba 100644 --- a/Eigen/Sparse +++ b/Eigen/Sparse @@ -13,6 +13,7 @@ namespace Eigen { #include "src/Sparse/SparseUtil.h" #include "src/Sparse/SparseMatrixBase.h" #include "src/Sparse/SparseArray.h" +#include "src/Sparse/SparseBlock.h" #include "src/Sparse/SparseMatrix.h" #include "src/Sparse/HashMatrix.h" #include "src/Sparse/LinkedVectorMatrix.h" diff --git a/Eigen/src/Core/Block.h b/Eigen/src/Core/Block.h index 11583d042..18fcdbb2b 100644 --- a/Eigen/src/Core/Block.h +++ b/Eigen/src/Core/Block.h @@ -65,6 +65,8 @@ template > { typedef typename MatrixType::Scalar Scalar; + typedef typename MatrixType::Nested MatrixTypeNested; + typedef typename ei_unref::type _MatrixTypeNested; enum{ RowsAtCompileTime = MatrixType::RowsAtCompileTime == 1 ? 1 : BlockRows, ColsAtCompileTime = MatrixType::ColsAtCompileTime == 1 ? 1 : BlockCols, @@ -94,6 +96,8 @@ template _EIGEN_GENERIC_PUBLIC_INTERFACE(Block, MapBase) + class InnerIterator; typedef typename ei_traits::AlignedDerivedType AlignedDerivedType; EIGEN_INHERIT_ASSIGNMENT_OPERATORS(Block) diff --git a/Eigen/src/Core/MatrixBase.h b/Eigen/src/Core/MatrixBase.h index dbb481914..4248ae523 100644 --- a/Eigen/src/Core/MatrixBase.h +++ b/Eigen/src/Core/MatrixBase.h @@ -155,7 +155,7 @@ template class MatrixBase /** \returns the number of rows. \sa cols(), RowsAtCompileTime */ inline int rows() const { return derived().rows(); } - /** \returns the number of columns. \sa row(), ColsAtCompileTime*/ + /** \returns the number of columns. \sa rows(), ColsAtCompileTime*/ inline int cols() const { return derived().cols(); } /** \returns the number of coefficients, which is \a rows()*cols(). * \sa rows(), cols(), SizeAtCompileTime. */ diff --git a/Eigen/src/Core/util/Constants.h b/Eigen/src/Core/util/Constants.h index 9576ac6d8..d08af60e0 100644 --- a/Eigen/src/Core/util/Constants.h +++ b/Eigen/src/Core/util/Constants.h @@ -211,7 +211,8 @@ enum { enum { NoDirectAccess = 0, - HasDirectAccess = DirectAccessBit + HasDirectAccess = DirectAccessBit, + IsSparse = SparseBit }; const int FullyCoherentAccessPattern = 0x1; diff --git a/Eigen/src/Core/util/ForwardDeclarations.h b/Eigen/src/Core/util/ForwardDeclarations.h index fca47bad4..92852cbfa 100644 --- a/Eigen/src/Core/util/ForwardDeclarations.h +++ b/Eigen/src/Core/util/ForwardDeclarations.h @@ -36,7 +36,8 @@ template class NestByValue; template class SwapWrapper; template class Minor; template::Flags&DirectAccessBit> class Block; + int _DirectAccessStatus = ei_traits::Flags&DirectAccessBit ? DirectAccessBit + : ei_traits::Flags&SparseBit> class Block; template class Transpose; template class Conjugate; template class CwiseNullaryOp; diff --git a/Eigen/src/Sparse/CoreIterators.h b/Eigen/src/Sparse/CoreIterators.h index f44db0c1f..ebe0bf8bd 100644 --- a/Eigen/src/Sparse/CoreIterators.h +++ b/Eigen/src/Sparse/CoreIterators.h @@ -37,7 +37,7 @@ class MatrixBase::InnerIterator : m_matrix(mat), m_inner(0), m_outer(outer), m_end(mat.rows()) {} - Scalar value() + Scalar value() const { return (Derived::Flags&RowMajorBit) ? m_matrix.coeff(m_outer, m_inner) : m_matrix.coeff(m_inner, m_outer); @@ -66,6 +66,80 @@ class Transpose::InnerIterator : public MatrixType::InnerIterator {} }; +template +class Block::InnerIterator +{ + typedef typename Block::Scalar Scalar; + typedef typename ei_traits::_MatrixTypeNested _MatrixTypeNested; + typedef typename _MatrixTypeNested::InnerIterator MatrixTypeIterator; + public: + + InnerIterator(const Block& block, int outer) + : m_iter(block.m_matrix,(Block::Flags&RowMajor) ? block.m_startRow.value() + outer : block.m_startCol.value() + outer), + m_start( (Block::Flags&RowMajor) ? block.m_startCol.value() : block.m_startRow.value()), + m_end(m_start + ((Block::Flags&RowMajor) ? block.m_blockCols.value() : block.m_blockRows.value())), + m_offset( (Block::Flags&RowMajor) ? block.m_startCol.value() : block.m_startRow.value()) + { + while (m_iter.index()>=0 && m_iter.index() +class Block::InnerIterator +{ + typedef typename Block::Scalar Scalar; + typedef typename ei_traits::_MatrixTypeNested _MatrixTypeNested; + typedef typename _MatrixTypeNested::InnerIterator MatrixTypeIterator; + public: + + InnerIterator(const Block& block, int outer) + : m_iter(block.m_matrix,(Block::Flags&RowMajor) ? block.m_startRow.value() + outer : block.m_startCol.value() + outer), + m_start( (Block::Flags&RowMajor) ? block.m_startCol.value() : block.m_startRow.value()), + m_end(m_start + ((Block::Flags&RowMajor) ? block.m_blockCols.value() : block.m_blockRows.value())), + m_offset( (Block::Flags&RowMajor) ? block.m_startCol.value() : block.m_startRow.value()) + { + while (m_iter.index()>=0 && m_iter.index() class CwiseUnaryOp::InnerIterator { @@ -133,13 +207,13 @@ class CwiseBinaryOp::InnerIterator ++m_lhsIter; ++m_rhsIter; } - else if (m_lhsIter && ((!m_rhsIter) || m_lhsIter.index() < m_rhsIter.index())) + else if (m_lhsIter && (!m_rhsIter || (m_lhsIter.index() < m_rhsIter.index()))) { m_id = m_lhsIter.index(); m_value = m_functor(m_lhsIter.value(), Scalar(0)); ++m_lhsIter; } - else if (m_rhsIter && ((!m_lhsIter) || m_lhsIter.index() > m_rhsIter.index())) + else if (m_rhsIter && (!m_lhsIter || (m_lhsIter.index() > m_rhsIter.index()))) { m_id = m_rhsIter.index(); m_value = m_functor(Scalar(0), m_rhsIter.value()); diff --git a/Eigen/src/Sparse/SparseBlock.h b/Eigen/src/Sparse/SparseBlock.h new file mode 100644 index 000000000..dd952714d --- /dev/null +++ b/Eigen/src/Sparse/SparseBlock.h @@ -0,0 +1,122 @@ +// This file is part of Eigen, a lightweight C++ template library +// for linear algebra. Eigen itself is part of the KDE project. +// +// Copyright (C) 2008 Gael Guennebaud +// Copyright (C) 2008 Daniel Gomez Ferro +// +// Eigen is free software; you can redistribute it and/or +// modify it under the terms of the GNU Lesser General Public +// License as published by the Free Software Foundation; either +// version 3 of the License, or (at your option) any later version. +// +// Alternatively, you can redistribute it and/or +// modify it under the terms of the GNU General Public License as +// published by the Free Software Foundation; either version 2 of +// the License, or (at your option) any later version. +// +// Eigen is distributed in the hope that it will be useful, but WITHOUT ANY +// WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS +// FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License or the +// GNU General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public +// License and a copy of the GNU General Public License along with +// Eigen. If not, see . + +#ifndef EIGEN_SPARSEBLOCK_H +#define EIGEN_SPARSEBLOCK_H + +template +class Block + : public SparseMatrixBase > +{ +public: + + _EIGEN_GENERIC_PUBLIC_INTERFACE(Block, SparseMatrixBase) + class InnerIterator; + + /** Column or Row constructor + */ + inline Block(const MatrixType& matrix, int i) + : m_matrix(matrix), + // It is a row if and only if BlockRows==1 and BlockCols==MatrixType::ColsAtCompileTime, + // and it is a column if and only if BlockRows==MatrixType::RowsAtCompileTime and BlockCols==1, + // all other cases are invalid. + // The case a 1x1 matrix seems ambiguous, but the result is the same anyway. + m_startRow( (BlockRows==1) && (BlockCols==MatrixType::ColsAtCompileTime) ? i : 0), + m_startCol( (BlockRows==MatrixType::RowsAtCompileTime) && (BlockCols==1) ? i : 0), + m_blockRows(matrix.rows()), // if it is a row, then m_blockRows has a fixed-size of 1, so no pb to try to overwrite it + m_blockCols(matrix.cols()) // same for m_blockCols + { + ei_assert( (i>=0) && ( + ((BlockRows==1) && (BlockCols==MatrixType::ColsAtCompileTime) && i= 0 && BlockRows >= 1 && startRow + BlockRows <= matrix.rows() + && startCol >= 0 && BlockCols >= 1 && startCol + BlockCols <= matrix.cols()); + } + + /** Dynamic-size constructor + */ + inline Block(const MatrixType& matrix, + int startRow, int startCol, + int blockRows, int blockCols) + : m_matrix(matrix), m_startRow(startRow), m_startCol(startCol), + m_blockRows(blockRows), m_blockCols(blockCols) + { + ei_assert((RowsAtCompileTime==Dynamic || RowsAtCompileTime==blockRows) + && (ColsAtCompileTime==Dynamic || ColsAtCompileTime==blockCols)); + ei_assert(startRow >= 0 && blockRows >= 1 && startRow + blockRows <= matrix.rows() + && startCol >= 0 && blockCols >= 1 && startCol + blockCols <= matrix.cols()); + } + + inline int rows() const { return m_blockRows.value(); } + inline int cols() const { return m_blockCols.value(); } + + inline int stride(void) const { return m_matrix.stride(); } + + inline Scalar& coeffRef(int row, int col) + { + return m_matrix.const_cast_derived() + .coeffRef(row + m_startRow.value(), col + m_startCol.value()); + } + + inline const Scalar coeff(int row, int col) const + { + return m_matrix.coeff(row + m_startRow.value(), col + m_startCol.value()); + } + + inline Scalar& coeffRef(int index) + { + return m_matrix.const_cast_derived() + .coeffRef(m_startRow.value() + (RowsAtCompileTime == 1 ? 0 : index), + m_startCol.value() + (RowsAtCompileTime == 1 ? index : 0)); + } + + inline const Scalar coeff(int index) const + { + return m_matrix + .coeff(m_startRow.value() + (RowsAtCompileTime == 1 ? 0 : index), + m_startCol.value() + (RowsAtCompileTime == 1 ? index : 0)); + } + + protected: + + const typename MatrixType::Nested m_matrix; + const ei_int_if_dynamic m_startRow; + const ei_int_if_dynamic m_startCol; + const ei_int_if_dynamic m_blockRows; + const ei_int_if_dynamic m_blockCols; + +}; + + +#endif // EIGEN_SPARSEBLOCK_H diff --git a/Eigen/src/Sparse/SparseMatrix.h b/Eigen/src/Sparse/SparseMatrix.h index 9d4f325a3..d62718491 100644 --- a/Eigen/src/Sparse/SparseMatrix.h +++ b/Eigen/src/Sparse/SparseMatrix.h @@ -94,7 +94,7 @@ class SparseMatrix // ^^ optimization: let's first check if it is the last coefficient // (very common in high level algorithms) - const int* r = std::lower_bound(&m_data.index(start),&m_data.index(end),inner); + const int* r = std::lower_bound(&m_data.index(start),&m_data.index(end-1),inner); const int id = r-&m_data.index(0); return ((*r==inner) && (id::InnerIterator InnerIterator& operator++() { m_id++; return *this; } - Scalar value() { return m_matrix.m_data.value(m_id); } + Scalar value() const { return m_matrix.m_data.value(m_id); } int index() const { return m_matrix.m_data.index(m_id); } diff --git a/Eigen/src/Sparse/SparseMatrixBase.h b/Eigen/src/Sparse/SparseMatrixBase.h index 051d1cbc2..83dcefe06 100644 --- a/Eigen/src/Sparse/SparseMatrixBase.h +++ b/Eigen/src/Sparse/SparseMatrixBase.h @@ -139,8 +139,23 @@ class SparseMatrixBase : public MatrixBase } else { - LinkedVectorMatrix trans = m.derived(); - s << trans; + if (m.cols() == 1) { + int row = 0; + for (typename Derived::InnerIterator it(m.derived(), 0); it; ++it) + { + for ( ; row trans = m.derived(); + s << trans; + } } return s; } diff --git a/test/sparse.cpp b/test/sparse.cpp index bac1091ce..7decb4111 100644 --- a/test/sparse.cpp +++ b/test/sparse.cpp @@ -25,9 +25,8 @@ #include "main.h" #include -template void sparse() +template void sparse(int rows, int cols) { - int rows = 8, cols = 8; double density = std::max(8./(rows*cols), 0.01); typedef Matrix DenseMatrix; Scalar eps = 1e-6; @@ -73,6 +72,49 @@ template void sparse() VERIFY_IS_APPROX(m, refMat); + // test InnerIterators and Block expressions + for(int j=0; j void sparse() } } VERIFY_IS_APPROX(m, refMat); + } void test_sparse() { - sparse(); + sparse(8, 8); + sparse(16, 16); }