mirror/eigen
2
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2024-12-15 07:10:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make SparseTranspose inherit SparseCompressBase when possible

Browse Source
This commit is contained in:
Gael Guennebaud 2015-02-07 22:02:14 +01:00
parent 7838fda82c
commit 08081f8293
1 changed files with 31 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
Eigen/src/SparseCore/SparseTranspose.h
View File

@ -1,7 +1,7 @@
// This file is part of Eigen, a lightweight C++ template library // This file is part of Eigen, a lightweight C++ template library
// for linear algebra. // for linear algebra.
// //
// Copyright (C) 2008-2014 Gael Guennebaud <gael.guennebaud@inria.fr> // Copyright (C) 2008-2015 Gael Guennebaud <gael.guennebaud@inria.fr>
// //
// This Source Code Form is subject to the terms of the Mozilla // This Source Code Form is subject to the terms of the Mozilla
// Public License v. 2.0. If a copy of the MPL was not distributed // Public License v. 2.0. If a copy of the MPL was not distributed
@ -12,13 +12,41 @@
namespace Eigen { namespace Eigen {
namespace internal {
template<typename MatrixType,int CompressedAccess=int(MatrixType::Flags&CompressedAccessBit)>
class SparseTransposeImpl
: public SparseMatrixBase<Transpose<MatrixType> >
{};
template<typename MatrixType>
class SparseTransposeImpl<MatrixType,CompressedAccessBit>
: public SparseCompressedBase<Transpose<MatrixType> >
{
typedef SparseCompressedBase<Transpose<MatrixType> > Base;
public:
using Base::derived;
typedef typename Base::Scalar Scalar;
typedef typename Base::Index Index;
inline const Scalar* valuePtr() const { return derived().nestedExpression().valuePtr(); }
inline const Index* innerIndexPtr() const { return derived().nestedExpression().innerIndexPtr(); }
inline const Index* outerIndexPtr() const { return derived().nestedExpression().outerIndexPtr(); }
inline const Index* innerNonZeroPtr() const { return derived().nestedExpression().innerNonZeroPtr(); }
inline Scalar* valuePtr() { return derived().nestedExpression().valuePtr(); }
inline Index* innerIndexPtr() { return derived().nestedExpression().innerIndexPtr(); }
inline Index* outerIndexPtr() { return derived().nestedExpression().outerIndexPtr(); }
inline Index* innerNonZeroPtr() { return derived().nestedExpression().innerNonZeroPtr(); }
};
}
// Implement nonZeros() for transpose. I'm not sure that's the best approach for that. // Implement nonZeros() for transpose. I'm not sure that's the best approach for that.
// Perhaps it should be implemented in Transpose<> itself. // Perhaps it should be implemented in Transpose<> itself.
template<typename MatrixType> class TransposeImpl<MatrixType,Sparse> template<typename MatrixType> class TransposeImpl<MatrixType,Sparse>
: public SparseMatrixBase<Transpose<MatrixType> > : public internal::SparseTransposeImpl<MatrixType>
{ {
protected: protected:
typedef SparseMatrixBase<Transpose<MatrixType> > Base; typedef internal::SparseTransposeImpl<MatrixType> Base;
public: public:
inline typename MatrixType::Index nonZeros() const { return Base::derived().nestedExpression().nonZeros(); } inline typename MatrixType::Index nonZeros() const { return Base::derived().nestedExpression().nonZeros(); }
}; };