diff options
author | Gael Guennebaud <g.gael@free.fr> | 2017-01-06 21:53:32 +0100 |
---|---|---|
committer | Gael Guennebaud <g.gael@free.fr> | 2017-01-06 21:53:32 +0100 |
commit | 3264d3c761e6b08101e7577b4278119dea42ec09 (patch) | |
tree | c3ccee6cab6e28143dd61e5b0661d1580c2d0557 /Eigen/src/Core | |
parent | a875167d99cffa76a662de5475627d60238f0f36 (diff) |
Add support for plain-array as indices, e.g., mat({1,2,3,4})
Diffstat (limited to 'Eigen/src/Core')
-rw-r--r-- | Eigen/src/Core/ArithmeticSequence.h | 6 | ||||
-rw-r--r-- | Eigen/src/Core/DenseBase.h | 7 | ||||
-rw-r--r-- | Eigen/src/Core/IndexedView.h | 4 |
3 files changed, 15 insertions, 2 deletions
diff --git a/Eigen/src/Core/ArithmeticSequence.h b/Eigen/src/Core/ArithmeticSequence.h index 1585c1dbf..bfc586a61 100644 --- a/Eigen/src/Core/ArithmeticSequence.h +++ b/Eigen/src/Core/ArithmeticSequence.h @@ -159,6 +159,12 @@ span(FirstType first, SizeType size) { namespace internal { +template<typename T> +Index size(const T& x) { return x.size(); } + +template<typename T,std::size_t N> +Index size(const T (&x) [N]) { return N; } + template<typename T, int XprSize, typename EnableIf = void> struct get_compile_time_size { enum { value = Dynamic }; }; diff --git a/Eigen/src/Core/DenseBase.h b/Eigen/src/Core/DenseBase.h index f8a6b2625..13aa3854a 100644 --- a/Eigen/src/Core/DenseBase.h +++ b/Eigen/src/Core/DenseBase.h @@ -566,6 +566,13 @@ template<typename Derived> class DenseBase derived(), internal::make_indexing(rowIndices,derived().rows()), internal::make_indexing(colIndices,derived().cols())); } + template<typename RowIndicesT, std::size_t RowIndicesN, typename ColIndices> + IndexedView<const Derived,const RowIndicesT (&)[RowIndicesN],typename internal::MakeIndexing<ColIndices>::type> + operator()(const RowIndicesT (&rowIndices)[RowIndicesN], const ColIndices& colIndices) const { + return IndexedView<const Derived,const RowIndicesT (&) [RowIndicesN],typename internal::MakeIndexing<ColIndices>::type>( + derived(), rowIndices, internal::make_indexing(colIndices,derived().cols())); + } + #define EIGEN_CURRENT_STORAGE_BASE_CLASS Eigen::DenseBase #define EIGEN_DOC_BLOCK_ADDONS_NOT_INNER_PANEL #define EIGEN_DOC_BLOCK_ADDONS_INNER_PANEL_IF(COND) diff --git a/Eigen/src/Core/IndexedView.h b/Eigen/src/Core/IndexedView.h index 26c048584..5ff1c1837 100644 --- a/Eigen/src/Core/IndexedView.h +++ b/Eigen/src/Core/IndexedView.h @@ -70,8 +70,8 @@ public: IndexedView(XprType& xpr, const T0& rowIndices, const T1& colIndices) : m_xpr(xpr), m_rowIndices(rowIndices), m_colIndices(colIndices) {} - Index rows() const { return m_rowIndices.size(); } - Index cols() const { return m_colIndices.size(); } + Index rows() const { return internal::size(m_rowIndices); } + Index cols() const { return internal::size(m_colIndices); } /** \returns the nested expression */ const typename internal::remove_all<XprType>::type& |