GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-04-19 08:09:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

add compile time info to reverse in place

Browse Source
This commit is contained in:
Charles Schlosser 2024-10-13 17:55:56 +00:00
parent b55dab7f21
commit 820e8a45fb
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
Eigen/src/Core/Reverse.h
View File

@ -127,19 +127,25 @@ EIGEN_DEVICE_FUNC inline typename DenseBase<Derived>::ReverseReturnType DenseBas
* \sa VectorwiseOp::reverseInPlace(), reverse() */ * \sa VectorwiseOp::reverseInPlace(), reverse() */
template <typename Derived> template <typename Derived>
EIGEN_DEVICE_FUNC inline void DenseBase<Derived>::reverseInPlace() { EIGEN_DEVICE_FUNC inline void DenseBase<Derived>::reverseInPlace() {
constexpr int HalfRowsAtCompileTime = RowsAtCompileTime == Dynamic ? Dynamic : RowsAtCompileTime / 2;
constexpr int HalfColsAtCompileTime = ColsAtCompileTime == Dynamic ? Dynamic : ColsAtCompileTime / 2;
if (cols() > rows()) { if (cols() > rows()) {
Index half = cols() / 2; Index half = cols() / 2;
leftCols(half).swap(rightCols(half).reverse()); this->template leftCols<HalfColsAtCompileTime>(half).swap(
this->template rightCols<HalfColsAtCompileTime>(half).reverse());
if ((cols() % 2) == 1) { if ((cols() % 2) == 1) {
Index half2 = rows() / 2; Index half2 = rows() / 2;
col(half).head(half2).swap(col(half).tail(half2).reverse()); col(half).template head<HalfRowsAtCompileTime>(half2).swap(
col(half).template tail<HalfRowsAtCompileTime>(half2).reverse());
} }
} else { } else {
Index half = rows() / 2; Index half = rows() / 2;
topRows(half).swap(bottomRows(half).reverse()); this->template topRows<HalfRowsAtCompileTime>(half).swap(
this->template bottomRows<HalfRowsAtCompileTime>(half).reverse());
if ((rows() % 2) == 1) { if ((rows() % 2) == 1) {
Index half2 = cols() / 2; Index half2 = cols() / 2;
row(half).head(half2).swap(row(half).tail(half2).reverse()); row(half).template head<HalfColsAtCompileTime>(half2).swap(
row(half).template tail<HalfColsAtCompileTime>(half2).reverse());
} }
} }
} }