11 #ifndef EIGEN_JACOBISVD_H
12 #define EIGEN_JACOBISVD_H
19 template<
typename MatrixType,
int QRPreconditioner,
20 bool IsComplex = NumTraits<typename MatrixType::Scalar>::IsComplex>
30 enum { PreconditionIfMoreColsThanRows, PreconditionIfMoreRowsThanCols };
32 template<
typename MatrixType,
int QRPreconditioner,
int Case>
35 enum { a = MatrixType::RowsAtCompileTime !=
Dynamic &&
36 MatrixType::ColsAtCompileTime !=
Dynamic &&
37 MatrixType::ColsAtCompileTime <= MatrixType::RowsAtCompileTime,
38 b = MatrixType::RowsAtCompileTime !=
Dynamic &&
39 MatrixType::ColsAtCompileTime !=
Dynamic &&
40 MatrixType::RowsAtCompileTime <= MatrixType::ColsAtCompileTime,
42 (Case == PreconditionIfMoreColsThanRows &&
bool(a)) ||
43 (Case == PreconditionIfMoreRowsThanCols &&
bool(b)) )
47 template<
typename MatrixType,
int QRPreconditioner,
int Case,
51 template<
typename MatrixType,
int QRPreconditioner,
int Case>
64 template<
typename MatrixType>
68 typedef typename MatrixType::Scalar Scalar;
71 RowsAtCompileTime = MatrixType::RowsAtCompileTime,
72 MaxRowsAtCompileTime = MatrixType::MaxRowsAtCompileTime
78 if (svd.rows() != m_qr.rows() || svd.cols() != m_qr.cols())
81 ::new (&m_qr)
QRType(svd.rows(), svd.cols());
83 if (svd.m_computeFullU) m_workspace.resize(svd.rows());
88 if(matrix.rows() > matrix.cols())
91 svd.m_workMatrix = m_qr.matrixQR().block(0,0,matrix.cols(),matrix.cols()).template triangularView<Upper>();
92 if(svd.m_computeFullU) m_qr.matrixQ().evalTo(svd.m_matrixU, m_workspace);
93 if(svd.
computeV()) svd.m_matrixV = m_qr.colsPermutation();
104 template<
typename MatrixType>
108 typedef typename MatrixType::Scalar Scalar;
111 RowsAtCompileTime = MatrixType::RowsAtCompileTime,
112 ColsAtCompileTime = MatrixType::ColsAtCompileTime,
113 MaxRowsAtCompileTime = MatrixType::MaxRowsAtCompileTime,
114 MaxColsAtCompileTime = MatrixType::MaxColsAtCompileTime,
115 TrOptions = RowsAtCompileTime==1 ? (MatrixType::Options & ~(
RowMajor))
116 : ColsAtCompileTime==1 ? (MatrixType::Options |
RowMajor)
117 : MatrixType::Options
124 if (svd.cols() != m_qr.rows() || svd.rows() != m_qr.cols())
127 ::new (&m_qr)
QRType(svd.cols(), svd.rows());
129 m_adjoint.resize(svd.cols(), svd.rows());
130 if (svd.m_computeFullV) m_workspace.resize(svd.cols());
135 if(matrix.cols() > matrix.rows())
137 m_adjoint = matrix.adjoint();
138 m_qr.compute(m_adjoint);
139 svd.m_workMatrix = m_qr.matrixQR().block(0,0,matrix.rows(),matrix.rows()).template triangularView<Upper>().adjoint();
140 if(svd.m_computeFullV) m_qr.matrixQ().evalTo(svd.m_matrixV, m_workspace);
141 if(svd.
computeU()) svd.m_matrixU = m_qr.colsPermutation();
155 template<
typename MatrixType>
161 if (svd.rows() != m_qr.rows() || svd.cols() != m_qr.cols())
164 ::new (&m_qr)
QRType(svd.rows(), svd.cols());
166 if (svd.m_computeFullU) m_workspace.resize(svd.rows());
167 else if (svd.m_computeThinU) m_workspace.resize(svd.cols());
172 if(matrix.rows() > matrix.cols())
174 m_qr.compute(matrix);
175 svd.m_workMatrix = m_qr.matrixQR().block(0,0,matrix.cols(),matrix.cols()).template triangularView<Upper>();
176 if(svd.m_computeFullU) m_qr.householderQ().evalTo(svd.m_matrixU, m_workspace);
177 else if(svd.m_computeThinU)
179 svd.m_matrixU.setIdentity(matrix.rows(), matrix.cols());
180 m_qr.householderQ().applyThisOnTheLeft(svd.m_matrixU, m_workspace);
182 if(svd.
computeV()) svd.m_matrixV = m_qr.colsPermutation();
194 template<
typename MatrixType>
198 typedef typename MatrixType::Scalar Scalar;
201 RowsAtCompileTime = MatrixType::RowsAtCompileTime,
202 ColsAtCompileTime = MatrixType::ColsAtCompileTime,
203 MaxRowsAtCompileTime = MatrixType::MaxRowsAtCompileTime,
204 MaxColsAtCompileTime = MatrixType::MaxColsAtCompileTime,
205 TrOptions = RowsAtCompileTime==1 ? (MatrixType::Options & ~(
RowMajor))
206 : ColsAtCompileTime==1 ? (MatrixType::Options |
RowMajor)
207 : MatrixType::Options
215 if (svd.cols() != m_qr.rows() || svd.rows() != m_qr.cols())
218 ::new (&m_qr)
QRType(svd.cols(), svd.rows());
220 if (svd.m_computeFullV) m_workspace.resize(svd.cols());
221 else if (svd.m_computeThinV) m_workspace.resize(svd.rows());
222 m_adjoint.resize(svd.cols(), svd.rows());
227 if(matrix.cols() > matrix.rows())
229 m_adjoint = matrix.adjoint();
230 m_qr.compute(m_adjoint);
232 svd.m_workMatrix = m_qr.matrixQR().block(0,0,matrix.rows(),matrix.rows()).template triangularView<Upper>().adjoint();
233 if(svd.m_computeFullV) m_qr.householderQ().evalTo(svd.m_matrixV, m_workspace);
234 else if(svd.m_computeThinV)
236 svd.m_matrixV.setIdentity(matrix.cols(), matrix.rows());
237 m_qr.householderQ().applyThisOnTheLeft(svd.m_matrixV, m_workspace);
239 if(svd.
computeU()) svd.m_matrixU = m_qr.colsPermutation();
254 template<
typename MatrixType>
260 if (svd.rows() != m_qr.rows() || svd.cols() != m_qr.cols())
263 ::new (&m_qr)
QRType(svd.rows(), svd.cols());
265 if (svd.m_computeFullU) m_workspace.resize(svd.rows());
266 else if (svd.m_computeThinU) m_workspace.resize(svd.cols());
271 if(matrix.rows() > matrix.cols())
273 m_qr.compute(matrix);
274 svd.m_workMatrix = m_qr.matrixQR().block(0,0,matrix.cols(),matrix.cols()).template triangularView<Upper>();
275 if(svd.m_computeFullU) m_qr.householderQ().evalTo(svd.m_matrixU, m_workspace);
276 else if(svd.m_computeThinU)
278 svd.m_matrixU.setIdentity(matrix.rows(), matrix.cols());
279 m_qr.householderQ().applyThisOnTheLeft(svd.m_matrixU, m_workspace);
281 if(svd.
computeV()) svd.m_matrixV.setIdentity(matrix.cols(), matrix.cols());
292 template<
typename MatrixType>
296 typedef typename MatrixType::Scalar Scalar;
299 RowsAtCompileTime = MatrixType::RowsAtCompileTime,
300 ColsAtCompileTime = MatrixType::ColsAtCompileTime,
301 MaxRowsAtCompileTime = MatrixType::MaxRowsAtCompileTime,
302 MaxColsAtCompileTime = MatrixType::MaxColsAtCompileTime,
303 Options = MatrixType::Options
311 if (svd.cols() != m_qr.rows() || svd.rows() != m_qr.cols())
314 ::new (&m_qr)
QRType(svd.cols(), svd.rows());
316 if (svd.m_computeFullV) m_workspace.resize(svd.cols());
317 else if (svd.m_computeThinV) m_workspace.resize(svd.rows());
318 m_adjoint.resize(svd.cols(), svd.rows());
323 if(matrix.cols() > matrix.rows())
325 m_adjoint = matrix.adjoint();
326 m_qr.compute(m_adjoint);
328 svd.m_workMatrix = m_qr.matrixQR().block(0,0,matrix.rows(),matrix.rows()).template triangularView<Upper>().adjoint();
329 if(svd.m_computeFullV) m_qr.householderQ().evalTo(svd.m_matrixV, m_workspace);
330 else if(svd.m_computeThinV)
332 svd.m_matrixV.setIdentity(matrix.cols(), matrix.rows());
333 m_qr.householderQ().applyThisOnTheLeft(svd.m_matrixV, m_workspace);
335 if(svd.
computeU()) svd.m_matrixU.setIdentity(matrix.rows(), matrix.rows());
353 template<
typename MatrixType,
int QRPreconditioner>
357 typedef typename MatrixType::RealScalar RealScalar;
361 template<
typename MatrixType,
int QRPreconditioner>
365 typedef typename MatrixType::Scalar Scalar;
366 typedef typename MatrixType::RealScalar RealScalar;
373 RealScalar n = sqrt(numext::abs2(work_matrix.
coeff(p,p)) + numext::abs2(work_matrix.
coeff(q,p)));
375 const RealScalar considerAsZero = (std::numeric_limits<RealScalar>::min)();
383 if(abs(numext::imag(work_matrix.
coeff(p,q)))>considerAsZero)
386 z = abs(work_matrix.
coeff(p,q)) / work_matrix.
coeff(p,q);
387 work_matrix.row(p) *= z;
388 if(svd.
computeU()) svd.m_matrixU.col(p) *= conj(z);
390 if(abs(numext::imag(work_matrix.
coeff(q,q)))>considerAsZero)
392 z = abs(work_matrix.
coeff(q,q)) / work_matrix.
coeff(q,q);
393 work_matrix.row(q) *= z;
394 if(svd.
computeU()) svd.m_matrixU.col(q) *= conj(z);
400 rot.c() = conj(work_matrix.
coeff(p,p)) / n;
401 rot.s() = work_matrix.
coeff(q,p) / n;
402 work_matrix.applyOnTheLeft(p,q,rot);
404 if(abs(numext::imag(work_matrix.
coeff(p,q)))>considerAsZero)
406 z = abs(work_matrix.
coeff(p,q)) / work_matrix.
coeff(p,q);
407 work_matrix.col(q) *= z;
408 if(svd.
computeV()) svd.m_matrixV.col(q) *= z;
410 if(abs(numext::imag(work_matrix.
coeff(q,q)))>considerAsZero)
412 z = abs(work_matrix.
coeff(q,q)) / work_matrix.
coeff(q,q);
413 work_matrix.row(q) *= z;
414 if(svd.
computeU()) svd.m_matrixU.col(q) *= conj(z);
419 maxDiagEntry = numext::maxi<RealScalar>(maxDiagEntry,numext::maxi<RealScalar>(abs(work_matrix.
coeff(p,p)), abs(work_matrix.
coeff(q,q))));
421 RealScalar threshold = numext::maxi<RealScalar>(considerAsZero, precision * maxDiagEntry);
422 return abs(work_matrix.
coeff(p,q))>threshold || abs(work_matrix.
coeff(q,p)) > threshold;
426 template<
typename _MatrixType,
int QRPreconditioner>
430 typedef _MatrixType MatrixType;
488 template<
typename _MatrixType,
int QRPreconditioner>
class JacobiSVD
489 :
public SVDBase<JacobiSVD<_MatrixType,QRPreconditioner> >
494 typedef _MatrixType MatrixType;
495 typedef typename MatrixType::Scalar Scalar;
498 RowsAtCompileTime = MatrixType::RowsAtCompileTime,
499 ColsAtCompileTime = MatrixType::ColsAtCompileTime,
500 DiagSizeAtCompileTime = EIGEN_SIZE_MIN_PREFER_DYNAMIC(RowsAtCompileTime,ColsAtCompileTime),
501 MaxRowsAtCompileTime = MatrixType::MaxRowsAtCompileTime,
502 MaxColsAtCompileTime = MatrixType::MaxColsAtCompileTime,
503 MaxDiagSizeAtCompileTime = EIGEN_SIZE_MIN_PREFER_FIXED(MaxRowsAtCompileTime,MaxColsAtCompileTime),
504 MatrixOptions = MatrixType::Options
513 typedef Matrix<Scalar, DiagSizeAtCompileTime, DiagSizeAtCompileTime,
514 MatrixOptions, MaxDiagSizeAtCompileTime, MaxDiagSizeAtCompileTime>
534 allocate(rows, cols, computationOptions);
547 explicit JacobiSVD(
const MatrixType& matrix,
unsigned int computationOptions = 0)
549 compute(matrix, computationOptions);
562 JacobiSVD&
compute(
const MatrixType& matrix,
unsigned int computationOptions);
572 return compute(matrix, m_computationOptions);
582 void allocate(
Index rows,
Index cols,
unsigned int computationOptions);
585 using Base::m_matrixU;
586 using Base::m_matrixV;
587 using Base::m_singularValues;
588 using Base::m_isInitialized;
589 using Base::m_isAllocated;
590 using Base::m_usePrescribedThreshold;
591 using Base::m_computeFullU;
592 using Base::m_computeThinU;
593 using Base::m_computeFullV;
594 using Base::m_computeThinV;
595 using Base::m_computationOptions;
596 using Base::m_nonzeroSingularValues;
599 using Base::m_diagSize;
600 using Base::m_prescribedThreshold;
601 WorkMatrixType m_workMatrix;
603 template<
typename __MatrixType,
int _QRPreconditioner,
bool _IsComplex>
605 template<
typename __MatrixType,
int _QRPreconditioner,
int _Case,
bool _DoAnything>
610 MatrixType m_scaledMatrix;
613 template<
typename MatrixType,
int QRPreconditioner>
614 void JacobiSVD<MatrixType, QRPreconditioner>::allocate(
Eigen::Index rows,
Eigen::Index cols,
unsigned int computationOptions)
616 eigen_assert(rows >= 0 && cols >= 0);
621 computationOptions == m_computationOptions)
628 m_isInitialized =
false;
629 m_isAllocated =
true;
630 m_computationOptions = computationOptions;
631 m_computeFullU = (computationOptions &
ComputeFullU) != 0;
632 m_computeThinU = (computationOptions &
ComputeThinU) != 0;
633 m_computeFullV = (computationOptions &
ComputeFullV) != 0;
634 m_computeThinV = (computationOptions &
ComputeThinV) != 0;
635 eigen_assert(!(m_computeFullU && m_computeThinU) &&
"JacobiSVD: you can't ask for both full and thin U");
636 eigen_assert(!(m_computeFullV && m_computeThinV) &&
"JacobiSVD: you can't ask for both full and thin V");
637 eigen_assert(EIGEN_IMPLIES(m_computeThinU || m_computeThinV, MatrixType::ColsAtCompileTime==
Dynamic) &&
638 "JacobiSVD: thin U and V are only available when your matrix has a dynamic number of columns.");
641 eigen_assert(!(m_computeThinU || m_computeThinV) &&
642 "JacobiSVD: can't compute thin U or thin V with the FullPivHouseholderQR preconditioner. "
643 "Use the ColPivHouseholderQR preconditioner instead.");
645 m_diagSize = (std::min)(m_rows, m_cols);
646 m_singularValues.resize(m_diagSize);
648 m_matrixU.resize(m_rows, m_computeFullU ? m_rows
649 : m_computeThinU ? m_diagSize
652 m_matrixV.resize(m_cols, m_computeFullV ? m_cols
653 : m_computeThinV ? m_diagSize
655 m_workMatrix.resize(m_diagSize, m_diagSize);
657 if(m_cols>m_rows) m_qr_precond_morecols.allocate(*
this);
658 if(m_rows>m_cols) m_qr_precond_morerows.allocate(*
this);
659 if(m_rows!=m_cols) m_scaledMatrix.resize(rows,cols);
662 template<
typename MatrixType,
int QRPreconditioner>
663 JacobiSVD<MatrixType, QRPreconditioner>&
667 allocate(matrix.rows(), matrix.cols(), computationOptions);
674 const RealScalar considerAsZero = (std::numeric_limits<RealScalar>::min)();
677 RealScalar scale = matrix.cwiseAbs().maxCoeff();
678 if(scale==RealScalar(0)) scale = RealScalar(1);
684 m_scaledMatrix = matrix / scale;
685 m_qr_precond_morecols.run(*
this, m_scaledMatrix);
686 m_qr_precond_morerows.run(*
this, m_scaledMatrix);
690 m_workMatrix = matrix.block(0,0,m_diagSize,m_diagSize) / scale;
691 if(m_computeFullU) m_matrixU.setIdentity(m_rows,m_rows);
692 if(m_computeThinU) m_matrixU.setIdentity(m_rows,m_diagSize);
693 if(m_computeFullV) m_matrixV.setIdentity(m_cols,m_cols);
694 if(m_computeThinV) m_matrixV.setIdentity(m_cols, m_diagSize);
698 RealScalar maxDiagEntry = m_workMatrix.cwiseAbs().diagonal().maxCoeff();
700 bool finished =
false;
707 for(
Index p = 1; p < m_diagSize; ++p)
709 for(
Index q = 0; q < p; ++q)
714 RealScalar threshold = numext::maxi<RealScalar>(considerAsZero, precision * maxDiagEntry);
715 if(abs(m_workMatrix.coeff(p,q))>threshold || abs(m_workMatrix.coeff(q,p)) > threshold)
723 internal::real_2x2_jacobi_svd(m_workMatrix, p, q, &j_left, &j_right);
726 m_workMatrix.applyOnTheLeft(p,q,j_left);
727 if(computeU()) m_matrixU.applyOnTheRight(p,q,j_left.
transpose());
729 m_workMatrix.applyOnTheRight(p,q,j_right);
730 if(computeV()) m_matrixV.applyOnTheRight(p,q,j_right);
733 maxDiagEntry = numext::maxi<RealScalar>(maxDiagEntry,numext::maxi<RealScalar>(abs(m_workMatrix.coeff(p,p)), abs(m_workMatrix.coeff(q,q))));
742 for(
Index i = 0; i < m_diagSize; ++i)
749 RealScalar a = abs(m_workMatrix.coeff(i,i));
750 m_singularValues.coeffRef(i) = abs(a);
751 if(computeU()) m_matrixU.col(i) *= m_workMatrix.coeff(i,i)/a;
756 RealScalar a = numext::real(m_workMatrix.coeff(i,i));
757 m_singularValues.coeffRef(i) = abs(a);
758 if(computeU() && (a<RealScalar(0))) m_matrixU.col(i) = -m_matrixU.col(i);
762 m_singularValues *= scale;
766 m_nonzeroSingularValues = m_diagSize;
767 for(
Index i = 0; i < m_diagSize; i++)
770 RealScalar maxRemainingSingularValue = m_singularValues.tail(m_diagSize-i).maxCoeff(&pos);
771 if(maxRemainingSingularValue == RealScalar(0))
773 m_nonzeroSingularValues = i;
779 std::swap(m_singularValues.coeffRef(i), m_singularValues.coeffRef(pos));
780 if(computeU()) m_matrixU.col(pos).swap(m_matrixU.col(i));
781 if(computeV()) m_matrixV.col(pos).swap(m_matrixV.col(i));
785 m_isInitialized =
true;
796 template<
typename Derived>
805 #endif // EIGEN_JACOBISVD_H