diff options
Diffstat (limited to 'unsupported/Eigen/src/NonLinearOptimization/r1mpyq.h')
-rw-r--r-- | unsupported/Eigen/src/NonLinearOptimization/r1mpyq.h | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/unsupported/Eigen/src/NonLinearOptimization/r1mpyq.h b/unsupported/Eigen/src/NonLinearOptimization/r1mpyq.h index 855cb7a1b..ad319d9eb 100644 --- a/unsupported/Eigen/src/NonLinearOptimization/r1mpyq.h +++ b/unsupported/Eigen/src/NonLinearOptimization/r1mpyq.h @@ -2,18 +2,20 @@ // TODO : move this to GivensQR once there's such a thing in Eigen template <typename Scalar> -void ei_r1mpyq(int m, int n, Scalar *a, const std::vector<PlanarRotation<Scalar> > &v_givens, const std::vector<PlanarRotation<Scalar> > &w_givens) +void ei_r1mpyq(DenseIndex m, DenseIndex n, Scalar *a, const std::vector<PlanarRotation<Scalar> > &v_givens, const std::vector<PlanarRotation<Scalar> > &w_givens) { + typedef DenseIndex Index; + /* apply the first set of givens rotations to a. */ - for (int j = n-2; j>=0; --j) - for (int i = 0; i<m; ++i) { + for (Index j = n-2; j>=0; --j) + for (Index i = 0; i<m; ++i) { Scalar temp = v_givens[j].c() * a[i+m*j] - v_givens[j].s() * a[i+m*(n-1)]; a[i+m*(n-1)] = v_givens[j].s() * a[i+m*j] + v_givens[j].c() * a[i+m*(n-1)]; a[i+m*j] = temp; } /* apply the second set of givens rotations to a. */ - for (int j = 0; j<n-1; ++j) - for (int i = 0; i<m; ++i) { + for (Index j = 0; j<n-1; ++j) + for (Index i = 0; i<m; ++i) { Scalar temp = w_givens[j].c() * a[i+m*j] + w_givens[j].s() * a[i+m*(n-1)]; a[i+m*(n-1)] = -w_givens[j].s() * a[i+m*j] + w_givens[j].c() * a[i+m*(n-1)]; a[i+m*j] = temp; |