devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Implement evaluator for CwiseUnaryView

Browse Source
This commit is contained in:
Jitse Niesen
2011-04-05 18:20:43 +01:00
parent cca7b146a2
commit 11ea81858a
2 changed files with 53 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
Eigen/src/Core/CoreEvaluators.h
View File

@@ -259,7 +259,7 @@ struct evaluator_impl<CwiseUnaryOp<UnaryOp, ArgType> >
}
protected:
const UnaryOpType& m_unaryOp;
const UnaryOpType m_unaryOp;
typename evaluator<ArgType>::type m_argImpl;
};
@@ -304,6 +304,47 @@ protected:
typename evaluator<Rhs>::type m_rhsImpl;
};
// -------------------- CwiseUnaryView --------------------
template<typename UnaryOp, typename ArgType>
struct evaluator_impl<CwiseUnaryView<UnaryOp, ArgType> >
{
typedef CwiseUnaryView<UnaryOp, ArgType> CwiseUnaryViewType;
evaluator_impl(const CwiseUnaryViewType& op)
: m_unaryOp(op.functor()),
m_argImpl(op.nestedExpression())
{ }
typedef typename CwiseUnaryViewType::Index Index;
typedef typename CwiseUnaryViewType::Scalar Scalar;
typedef typename CwiseUnaryViewType::CoeffReturnType CoeffReturnType;
CoeffReturnType coeff(Index row, Index col) const
{
return m_unaryOp(m_argImpl.coeff(row, col));
}
CoeffReturnType coeff(Index index) const
{
return m_unaryOp(m_argImpl.coeff(index));
}
Scalar& coeffRef(Index row, Index col)
{
return m_unaryOp(m_argImpl.coeffRef(row, col));
}
Scalar& coeffRef(Index index)
{
return m_unaryOp(m_argImpl.coeffRef(index));
}
protected:
const UnaryOp& m_unaryOp;
typename evaluator<ArgType>::type m_argImpl;
};
// -------------------- Product --------------------
template<typename Lhs, typename Rhs>