mirror of
https://gitlab.com/libeigen/eigen.git
synced 2026-04-10 11:34:33 +08:00
various work on the Sparse module:
* added some glue to Eigen/Core (SparseBit, ei_eval, Matrix)
* add two new sparse matrix types:
HashMatrix: based on std::map (for random writes)
LinkedVectorMatrix: array of linked vectors
(for outer coherent writes, e.g. to transpose a matrix)
* add a SparseSetter class to easily set/update any kind of matrices, e.g.:
{ SparseSetter<MatrixType,RandomAccessPattern> wrapper(mymatrix);
for (...) wrapper->coeffRef(rand(),rand()) = rand(); }
* automatic shallow copy for RValue
* and a lot of mess !
plus:
* remove the remaining ArrayBit related stuff
* don't use alloca in product for very large memory allocation
This commit is contained in:
102
Eigen/src/Sparse/SparseSetter.h
Normal file
102
Eigen/src/Sparse/SparseSetter.h
Normal file
@@ -0,0 +1,102 @@
|
||||
// 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 <g.gael@free.fr>
|
||||
//
|
||||
// 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 <http://www.gnu.org/licenses/>.
|
||||
|
||||
#ifndef EIGEN_SPARSESETTER_H
|
||||
#define EIGEN_SPARSESETTER_H
|
||||
|
||||
template<typename MatrixType, int AccessPattern,
|
||||
int IsSupported = ei_support_access_pattern<MatrixType,AccessPattern>::ret>
|
||||
struct ei_sparse_setter_selector;
|
||||
|
||||
template<typename MatrixType,
|
||||
int AccessPattern,
|
||||
typename WrapperType = typename ei_sparse_setter_selector<MatrixType,AccessPattern>::type>
|
||||
class SparseSetter
|
||||
{
|
||||
typedef typename ei_unref<WrapperType>::type _WrapperType;
|
||||
public:
|
||||
|
||||
inline SparseSetter(MatrixType& matrix) : m_wrapper(matrix), mp_matrix(&matrix) {}
|
||||
|
||||
~SparseSetter()
|
||||
{ *mp_matrix = m_wrapper; }
|
||||
|
||||
inline _WrapperType* operator->() { return &m_wrapper; }
|
||||
|
||||
inline _WrapperType& operator*() { return m_wrapper; }
|
||||
|
||||
protected:
|
||||
|
||||
WrapperType m_wrapper;
|
||||
MatrixType* mp_matrix;
|
||||
};
|
||||
|
||||
template<typename MatrixType, int AccessPattern>
|
||||
struct ei_sparse_setter_selector<MatrixType, AccessPattern, AccessPatternSupported>
|
||||
{
|
||||
typedef MatrixType& type;
|
||||
};
|
||||
|
||||
// forward each derived of SparseMatrixBase to the generic SparseMatrixBase specializations
|
||||
template<typename Scalar, int Flags, int AccessPattern>
|
||||
struct ei_sparse_setter_selector<SparseMatrix<Scalar,Flags>, AccessPattern, AccessPatternNotSupported>
|
||||
: public ei_sparse_setter_selector<SparseMatrixBase<SparseMatrix<Scalar,Flags> >,AccessPattern, AccessPatternNotSupported>
|
||||
{};
|
||||
|
||||
template<typename Scalar, int Flags, int AccessPattern>
|
||||
struct ei_sparse_setter_selector<LinkedVectorMatrix<Scalar,Flags>, AccessPattern, AccessPatternNotSupported>
|
||||
: public ei_sparse_setter_selector<LinkedVectorMatrix<SparseMatrix<Scalar,Flags> >,AccessPattern, AccessPatternNotSupported>
|
||||
{};
|
||||
|
||||
template<typename Scalar, int Flags, int AccessPattern>
|
||||
struct ei_sparse_setter_selector<HashMatrix<Scalar,Flags>, AccessPattern, AccessPatternNotSupported>
|
||||
: public ei_sparse_setter_selector<HashMatrix<SparseMatrix<Scalar,Flags> >,AccessPattern, AccessPatternNotSupported>
|
||||
{};
|
||||
|
||||
// generic SparseMatrixBase specializations
|
||||
template<typename Derived>
|
||||
struct ei_sparse_setter_selector<SparseMatrixBase<Derived>, RandomAccessPattern, AccessPatternNotSupported>
|
||||
{
|
||||
typedef HashMatrix<typename Derived::Scalar, Derived::Flags> type;
|
||||
};
|
||||
|
||||
template<typename Derived>
|
||||
struct ei_sparse_setter_selector<SparseMatrixBase<Derived>, OuterCoherentAccessPattern, AccessPatternNotSupported>
|
||||
{
|
||||
typedef HashMatrix<typename Derived::Scalar, Derived::Flags> type;
|
||||
};
|
||||
|
||||
template<typename Derived>
|
||||
struct ei_sparse_setter_selector<SparseMatrixBase<Derived>, InnerCoherentAccessPattern, AccessPatternNotSupported>
|
||||
{
|
||||
typedef LinkedVectorMatrix<typename Derived::Scalar, Derived::Flags> type;
|
||||
};
|
||||
|
||||
template<typename Derived>
|
||||
struct ei_sparse_setter_selector<SparseMatrixBase<Derived>, FullyCoherentAccessPattern, AccessPatternNotSupported>
|
||||
{
|
||||
typedef SparseMatrix<typename Derived::Scalar, Derived::Flags> type;
|
||||
};
|
||||
|
||||
#endif // EIGEN_SPARSESETTER_H
|
||||
Reference in New Issue
Block a user