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

Add templated subVector<Vertical/Horizonal>(Index) aliases to col/row(Index) methods (plus subVectors<>() to retrieve the number of rows/columns)

Browse Source
This commit is contained in:
Gael Guennebaud
2018-10-02 14:02:34 +02:00
parent 37e29fc893
commit 12487531ce
3 changed files with 37 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
Eigen/src/plugins/BlockMethods.h
View File

@@ -1399,3 +1399,32 @@ innerVectors(Index outerStart, Index outerSize) const
IsRowMajor ? outerSize : rows(), IsRowMajor ? cols() : outerSize);
}
/** \returns the i-th subvector (column or vector) according to the \c Direction
* \sa subVectors()
*/
EIGEN_DEVICE_FUNC
template<DirectionType Direction>
typename internal::conditional<Direction==Vertical,ColXpr,RowXpr>::type
subVector(Index i)
{
return typename internal::conditional<Direction==Vertical,ColXpr,RowXpr>::type(derived(),i);
}
/** This is the const version of subVector(Index) */
EIGEN_DEVICE_FUNC
template<DirectionType Direction>
typename internal::conditional<Direction==Vertical,ConstColXpr,ConstRowXpr>::type
subVector(Index i) const
{
return typename internal::conditional<Direction==Vertical,ConstColXpr,ConstRowXpr>::type(derived(),i);
}
/** \returns the number of subvectors (rows or columns) in the direction \c Direction
* \sa subVector(Index)
*/
EIGEN_DEVICE_FUNC
template<DirectionType Direction>
Index subVectors() const
{ return (Direction==Vertical)?cols():rows(); }