mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-05-03 01:04:23 +08:00
286 lines
9.1 KiB
C++
286 lines
9.1 KiB
C++
// This file is part of Eigen, a lightweight C++ template library
|
|
// for linear algebra.
|
|
//
|
|
// Copyright (C) 2009-2014 Gael Guennebaud <gael.guennebaud@inria.fr>
|
|
// Copyright (C) 2012 Désiré Nuentsa-Wakam <desire.nuentsa_wakam@inria.fr>
|
|
//
|
|
// 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
|
|
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
#ifndef EIGEN_SPARSE_TRIANGULARVIEW_H
|
|
#define EIGEN_SPARSE_TRIANGULARVIEW_H
|
|
|
|
namespace Eigen {
|
|
|
|
template<typename MatrixType, unsigned int Mode> class TriangularViewImpl<MatrixType,Mode,Sparse>
|
|
: public SparseMatrixBase<TriangularView<MatrixType,Mode> >
|
|
{
|
|
enum { SkipFirst = ((Mode&Lower) && !(MatrixType::Flags&RowMajorBit))
|
|
|| ((Mode&Upper) && (MatrixType::Flags&RowMajorBit)),
|
|
SkipLast = !SkipFirst,
|
|
SkipDiag = (Mode&ZeroDiag) ? 1 : 0,
|
|
HasUnitDiag = (Mode&UnitDiag) ? 1 : 0
|
|
};
|
|
|
|
typedef TriangularView<MatrixType,Mode> TriangularViewType;
|
|
|
|
protected:
|
|
// dummy solve function to make TriangularView happy.
|
|
void solve() const;
|
|
|
|
public:
|
|
|
|
EIGEN_SPARSE_PUBLIC_INTERFACE(TriangularViewType)
|
|
|
|
class InnerIterator;
|
|
class ReverseInnerIterator;
|
|
|
|
typedef typename MatrixType::Nested MatrixTypeNested;
|
|
typedef typename internal::remove_reference<MatrixTypeNested>::type MatrixTypeNestedNonRef;
|
|
typedef typename internal::remove_all<MatrixTypeNested>::type MatrixTypeNestedCleaned;
|
|
|
|
#ifndef EIGEN_TEST_EVALUATORS
|
|
template<typename OtherDerived>
|
|
typename internal::plain_matrix_type_column_major<OtherDerived>::type
|
|
solve(const MatrixBase<OtherDerived>& other) const;
|
|
#else // EIGEN_TEST_EVALUATORS
|
|
template<typename RhsType, typename DstType>
|
|
EIGEN_DEVICE_FUNC
|
|
EIGEN_STRONG_INLINE void _solve_impl(const RhsType &rhs, DstType &dst) const {
|
|
if(!(internal::is_same<RhsType,DstType>::value && internal::extract_data(dst) == internal::extract_data(rhs)))
|
|
dst = rhs;
|
|
this->solveInPlace(dst);
|
|
}
|
|
#endif // EIGEN_TEST_EVALUATORS
|
|
|
|
template<typename OtherDerived> void solveInPlace(MatrixBase<OtherDerived>& other) const;
|
|
template<typename OtherDerived> void solveInPlace(SparseMatrixBase<OtherDerived>& other) const;
|
|
|
|
};
|
|
|
|
template<typename MatrixType, unsigned int Mode>
|
|
class TriangularViewImpl<MatrixType,Mode,Sparse>::InnerIterator : public MatrixTypeNestedCleaned::InnerIterator
|
|
{
|
|
typedef typename MatrixTypeNestedCleaned::InnerIterator Base;
|
|
typedef typename TriangularViewType::Index Index;
|
|
public:
|
|
|
|
EIGEN_STRONG_INLINE InnerIterator(const TriangularViewImpl& view, Index outer)
|
|
: Base(view.derived().nestedExpression(), outer), m_returnOne(false)
|
|
{
|
|
if(SkipFirst)
|
|
{
|
|
while((*this) && ((HasUnitDiag||SkipDiag) ? this->index()<=outer : this->index()<outer))
|
|
Base::operator++();
|
|
if(HasUnitDiag)
|
|
m_returnOne = true;
|
|
}
|
|
else if(HasUnitDiag && ((!Base::operator bool()) || Base::index()>=Base::outer()))
|
|
{
|
|
if((!SkipFirst) && Base::operator bool())
|
|
Base::operator++();
|
|
m_returnOne = true;
|
|
}
|
|
}
|
|
|
|
EIGEN_STRONG_INLINE InnerIterator& operator++()
|
|
{
|
|
if(HasUnitDiag && m_returnOne)
|
|
m_returnOne = false;
|
|
else
|
|
{
|
|
Base::operator++();
|
|
if(HasUnitDiag && (!SkipFirst) && ((!Base::operator bool()) || Base::index()>=Base::outer()))
|
|
{
|
|
if((!SkipFirst) && Base::operator bool())
|
|
Base::operator++();
|
|
m_returnOne = true;
|
|
}
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
inline Index row() const { return (MatrixType::Flags&RowMajorBit ? Base::outer() : this->index()); }
|
|
inline Index col() const { return (MatrixType::Flags&RowMajorBit ? this->index() : Base::outer()); }
|
|
inline Index index() const
|
|
{
|
|
if(HasUnitDiag && m_returnOne) return Base::outer();
|
|
else return Base::index();
|
|
}
|
|
inline Scalar value() const
|
|
{
|
|
if(HasUnitDiag && m_returnOne) return Scalar(1);
|
|
else return Base::value();
|
|
}
|
|
|
|
EIGEN_STRONG_INLINE operator bool() const
|
|
{
|
|
if(HasUnitDiag && m_returnOne)
|
|
return true;
|
|
if(SkipFirst) return Base::operator bool();
|
|
else
|
|
{
|
|
if (SkipDiag) return (Base::operator bool() && this->index() < this->outer());
|
|
else return (Base::operator bool() && this->index() <= this->outer());
|
|
}
|
|
}
|
|
protected:
|
|
bool m_returnOne;
|
|
};
|
|
|
|
template<typename MatrixType, unsigned int Mode>
|
|
class TriangularViewImpl<MatrixType,Mode,Sparse>::ReverseInnerIterator : public MatrixTypeNestedCleaned::ReverseInnerIterator
|
|
{
|
|
typedef typename MatrixTypeNestedCleaned::ReverseInnerIterator Base;
|
|
typedef typename TriangularViewImpl::Index Index;
|
|
public:
|
|
|
|
EIGEN_STRONG_INLINE ReverseInnerIterator(const TriangularViewType& view, Index outer)
|
|
: Base(view.derived().nestedExpression(), outer)
|
|
{
|
|
eigen_assert((!HasUnitDiag) && "ReverseInnerIterator does not support yet triangular views with a unit diagonal");
|
|
if(SkipLast) {
|
|
while((*this) && (SkipDiag ? this->index()>=outer : this->index()>outer))
|
|
--(*this);
|
|
}
|
|
}
|
|
|
|
EIGEN_STRONG_INLINE ReverseInnerIterator& operator--()
|
|
{ Base::operator--(); return *this; }
|
|
|
|
inline Index row() const { return Base::row(); }
|
|
inline Index col() const { return Base::col(); }
|
|
|
|
EIGEN_STRONG_INLINE operator bool() const
|
|
{
|
|
if (SkipLast) return Base::operator bool() ;
|
|
else
|
|
{
|
|
if(SkipDiag) return (Base::operator bool() && this->index() > this->outer());
|
|
else return (Base::operator bool() && this->index() >= this->outer());
|
|
}
|
|
}
|
|
};
|
|
|
|
#ifdef EIGEN_TEST_EVALUATORS
|
|
namespace internal {
|
|
|
|
template<typename ArgType, unsigned int Mode>
|
|
struct unary_evaluator<TriangularView<ArgType,Mode>, IteratorBased>
|
|
: evaluator_base<TriangularView<ArgType,Mode> >
|
|
{
|
|
typedef TriangularView<ArgType,Mode> XprType;
|
|
|
|
protected:
|
|
|
|
typedef typename XprType::Scalar Scalar;
|
|
typedef typename XprType::Index Index;
|
|
typedef typename evaluator<ArgType>::InnerIterator EvalIterator;
|
|
|
|
enum { SkipFirst = ((Mode&Lower) && !(ArgType::Flags&RowMajorBit))
|
|
|| ((Mode&Upper) && (ArgType::Flags&RowMajorBit)),
|
|
SkipLast = !SkipFirst,
|
|
SkipDiag = (Mode&ZeroDiag) ? 1 : 0,
|
|
HasUnitDiag = (Mode&UnitDiag) ? 1 : 0
|
|
};
|
|
|
|
public:
|
|
|
|
enum {
|
|
CoeffReadCost = evaluator<ArgType>::CoeffReadCost,
|
|
Flags = XprType::Flags
|
|
};
|
|
|
|
unary_evaluator(const XprType &xpr) : m_argImpl(xpr.nestedExpression()) {}
|
|
|
|
class InnerIterator : public EvalIterator
|
|
{
|
|
typedef EvalIterator Base;
|
|
public:
|
|
|
|
EIGEN_STRONG_INLINE InnerIterator(const unary_evaluator& xprEval, Index outer)
|
|
: Base(xprEval.m_argImpl,outer), m_returnOne(false)
|
|
{
|
|
if(SkipFirst)
|
|
{
|
|
while((*this) && ((HasUnitDiag||SkipDiag) ? this->index()<=outer : this->index()<outer))
|
|
Base::operator++();
|
|
if(HasUnitDiag)
|
|
m_returnOne = true;
|
|
}
|
|
else if(HasUnitDiag && ((!Base::operator bool()) || Base::index()>=Base::outer()))
|
|
{
|
|
if((!SkipFirst) && Base::operator bool())
|
|
Base::operator++();
|
|
m_returnOne = true;
|
|
}
|
|
}
|
|
|
|
EIGEN_STRONG_INLINE InnerIterator& operator++()
|
|
{
|
|
if(HasUnitDiag && m_returnOne)
|
|
m_returnOne = false;
|
|
else
|
|
{
|
|
Base::operator++();
|
|
if(HasUnitDiag && (!SkipFirst) && ((!Base::operator bool()) || Base::index()>=Base::outer()))
|
|
{
|
|
if((!SkipFirst) && Base::operator bool())
|
|
Base::operator++();
|
|
m_returnOne = true;
|
|
}
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
EIGEN_STRONG_INLINE operator bool() const
|
|
{
|
|
if(HasUnitDiag && m_returnOne)
|
|
return true;
|
|
if(SkipFirst) return Base::operator bool();
|
|
else
|
|
{
|
|
if (SkipDiag) return (Base::operator bool() && this->index() < this->outer());
|
|
else return (Base::operator bool() && this->index() <= this->outer());
|
|
}
|
|
}
|
|
|
|
// inline Index row() const { return (ArgType::Flags&RowMajorBit ? Base::outer() : this->index()); }
|
|
// inline Index col() const { return (ArgType::Flags&RowMajorBit ? this->index() : Base::outer()); }
|
|
inline Index index() const
|
|
{
|
|
if(HasUnitDiag && m_returnOne) return Base::outer();
|
|
else return Base::index();
|
|
}
|
|
inline Scalar value() const
|
|
{
|
|
if(HasUnitDiag && m_returnOne) return Scalar(1);
|
|
else return Base::value();
|
|
}
|
|
|
|
protected:
|
|
bool m_returnOne;
|
|
private:
|
|
Scalar& valueRef();
|
|
};
|
|
|
|
protected:
|
|
typename evaluator<ArgType>::type m_argImpl;
|
|
};
|
|
|
|
} // end namespace internal
|
|
#endif
|
|
|
|
template<typename Derived>
|
|
template<int Mode>
|
|
inline const TriangularView<Derived, Mode>
|
|
SparseMatrixBase<Derived>::triangularView() const
|
|
{
|
|
return derived();
|
|
}
|
|
|
|
} // end namespace Eigen
|
|
|
|
#endif // EIGEN_SPARSE_TRIANGULARVIEW_H
|