mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-04-30 07:44:10 +08:00
e5d301dc96
* 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
103 lines
3.9 KiB
C++
103 lines
3.9 KiB
C++
// 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
|