2009-08-31 22:26:15 -04:00
|
|
|
// This file is part of Eigen, a lightweight C++ template library
|
|
|
|
|
// for linear algebra.
|
|
|
|
|
//
|
2010-05-30 16:00:58 -04:00
|
|
|
// Copyright (C) 2009-2010 Benoit Jacob <jacob.benoit.1@gmail.com>
|
2009-08-31 22:26:15 -04:00
|
|
|
//
|
|
|
|
|
// Eigen is free software; you can redistribute it and/or
|
|
|
|
|
// modify it under the terms of the GNU Lesser General Public
|
|
|
|
|
// License as published by the Free Software Foundation; either
|
|
|
|
|
// version 3 of the License, or (at your option) any later version.
|
|
|
|
|
//
|
|
|
|
|
// Alternatively, you can redistribute it and/or
|
|
|
|
|
// modify it under the terms of the GNU General Public License as
|
|
|
|
|
// published by the Free Software Foundation; either version 2 of
|
|
|
|
|
// the License, or (at your option) any later version.
|
|
|
|
|
//
|
|
|
|
|
// Eigen is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
|
|
|
// WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
|
|
|
// FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License or the
|
|
|
|
|
// GNU General Public License for more details.
|
|
|
|
|
//
|
|
|
|
|
// You should have received a copy of the GNU Lesser General Public
|
|
|
|
|
// License and a copy of the GNU General Public License along with
|
|
|
|
|
// Eigen. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
|
|
|
|
#ifndef EIGEN_JACOBISVD_H
|
|
|
|
|
#define EIGEN_JACOBISVD_H
|
|
|
|
|
|
2009-09-15 11:53:24 +02:00
|
|
|
// forward declarations (needed by ICC)
|
2010-10-08 10:42:32 -04:00
|
|
|
// the empty bodies are required by MSVC
|
|
|
|
|
template<typename MatrixType, int QRPreconditioner,
|
|
|
|
|
bool IsComplex = NumTraits<typename MatrixType::Scalar>::IsComplex>
|
2009-09-27 17:00:10 +02:00
|
|
|
struct ei_svd_precondition_2x2_block_to_be_real {};
|
2009-09-15 11:53:24 +02:00
|
|
|
|
2010-10-08 10:42:32 -04:00
|
|
|
template<typename MatrixType, int QRPreconditioner,
|
|
|
|
|
bool PossiblyMoreRowsThanCols = (MatrixType::RowsAtCompileTime == Dynamic)
|
|
|
|
|
|| (MatrixType::RowsAtCompileTime > MatrixType::ColsAtCompileTime) >
|
2009-09-27 17:18:19 +02:00
|
|
|
struct ei_svd_precondition_if_more_rows_than_cols;
|
2009-09-15 11:53:24 +02:00
|
|
|
|
2010-10-08 10:42:32 -04:00
|
|
|
template<typename MatrixType, int QRPreconditioner,
|
|
|
|
|
bool PossiblyMoreColsThanRows = (MatrixType::ColsAtCompileTime == Dynamic)
|
|
|
|
|
|| (MatrixType::ColsAtCompileTime > MatrixType::RowsAtCompileTime) >
|
2009-09-27 17:18:19 +02:00
|
|
|
struct ei_svd_precondition_if_more_cols_than_rows;
|
2009-09-15 11:53:24 +02:00
|
|
|
|
2010-10-08 10:42:32 -04:00
|
|
|
|
|
|
|
|
/*** QR preconditioners (R-SVD) ***/
|
|
|
|
|
|
|
|
|
|
enum { PreconditionIfMoreColsThanRows, PreconditionIfMoreRowsThanCols };
|
|
|
|
|
|
|
|
|
|
template<typename MatrixType, int QRPreconditioner, int Case>
|
|
|
|
|
struct ei_qr_preconditioner_should_do_anything
|
|
|
|
|
{
|
|
|
|
|
enum { a = MatrixType::RowsAtCompileTime != Dynamic &&
|
|
|
|
|
MatrixType::ColsAtCompileTime != Dynamic &&
|
|
|
|
|
MatrixType::ColsAtCompileTime <= MatrixType::RowsAtCompileTime,
|
|
|
|
|
b = MatrixType::RowsAtCompileTime != Dynamic &&
|
|
|
|
|
MatrixType::ColsAtCompileTime != Dynamic &&
|
|
|
|
|
MatrixType::RowsAtCompileTime <= MatrixType::ColsAtCompileTime,
|
|
|
|
|
ret = !( (QRPreconditioner == NoQRPreconditioner) ||
|
|
|
|
|
(Case == PreconditionIfMoreColsThanRows && bool(a)) ||
|
|
|
|
|
(Case == PreconditionIfMoreRowsThanCols && bool(b)) )
|
|
|
|
|
};
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
template<typename MatrixType, int QRPreconditioner, int Case,
|
|
|
|
|
bool DoAnything = ei_qr_preconditioner_should_do_anything<MatrixType, QRPreconditioner, Case>::ret
|
|
|
|
|
> struct ei_qr_preconditioner_impl {};
|
|
|
|
|
|
|
|
|
|
template<typename MatrixType, int QRPreconditioner, int Case>
|
|
|
|
|
struct ei_qr_preconditioner_impl<MatrixType, QRPreconditioner, Case, false>
|
|
|
|
|
{
|
|
|
|
|
static bool run(JacobiSVD<MatrixType, QRPreconditioner>&, const MatrixType&)
|
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
template<typename MatrixType>
|
|
|
|
|
struct ei_qr_preconditioner_impl<MatrixType, FullPivHouseholderQRPreconditioner, PreconditionIfMoreRowsThanCols, true>
|
|
|
|
|
{
|
|
|
|
|
static bool run(JacobiSVD<MatrixType, FullPivHouseholderQRPreconditioner>& svd, const MatrixType& matrix)
|
|
|
|
|
{
|
|
|
|
|
if(matrix.rows() > matrix.cols())
|
|
|
|
|
{
|
2010-10-08 10:42:40 -04:00
|
|
|
ei_assert(!svd.m_computeThinU && "JacobiSVD: can't compute a thin U with the FullPivHouseholderQR preconditioner. "
|
|
|
|
|
"Use the ColPivHouseholderQR preconditioner instead.");
|
2010-10-08 10:42:32 -04:00
|
|
|
FullPivHouseholderQR<MatrixType> qr(matrix);
|
|
|
|
|
svd.m_workMatrix = qr.matrixQR().block(0,0,matrix.cols(),matrix.cols()).template triangularView<Upper>();
|
2010-10-08 10:42:40 -04:00
|
|
|
if(svd.m_computeFullU) svd.m_matrixU = qr.matrixQ();
|
|
|
|
|
if(svd.computeV()) svd.m_matrixV = qr.colsPermutation();
|
2010-10-08 10:42:32 -04:00
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
template<typename MatrixType>
|
|
|
|
|
struct ei_qr_preconditioner_impl<MatrixType, FullPivHouseholderQRPreconditioner, PreconditionIfMoreColsThanRows, true>
|
|
|
|
|
{
|
|
|
|
|
static bool run(JacobiSVD<MatrixType, FullPivHouseholderQRPreconditioner>& svd, const MatrixType& matrix)
|
|
|
|
|
{
|
|
|
|
|
if(matrix.cols() > matrix.rows())
|
|
|
|
|
{
|
2010-10-08 10:42:40 -04:00
|
|
|
ei_assert(!svd.m_computeThinV && "JacobiSVD: can't compute a thin V with the FullPivHouseholderQR preconditioner. "
|
|
|
|
|
"Use the ColPivHouseholderQR preconditioner instead.");
|
2010-10-08 10:42:32 -04:00
|
|
|
typedef Matrix<typename MatrixType::Scalar, MatrixType::ColsAtCompileTime, MatrixType::RowsAtCompileTime,
|
|
|
|
|
MatrixType::Options, MatrixType::MaxColsAtCompileTime, MatrixType::MaxRowsAtCompileTime>
|
|
|
|
|
TransposeTypeWithSameStorageOrder;
|
|
|
|
|
FullPivHouseholderQR<TransposeTypeWithSameStorageOrder> qr(matrix.adjoint());
|
|
|
|
|
svd.m_workMatrix = qr.matrixQR().block(0,0,matrix.rows(),matrix.rows()).template triangularView<Upper>().adjoint();
|
2010-10-08 10:42:40 -04:00
|
|
|
if(svd.m_computeFullV) svd.m_matrixV = qr.matrixQ();
|
|
|
|
|
if(svd.computeU()) svd.m_matrixU = qr.colsPermutation();
|
2010-10-08 10:42:32 -04:00
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
else return false;
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
template<typename MatrixType>
|
|
|
|
|
struct ei_qr_preconditioner_impl<MatrixType, ColPivHouseholderQRPreconditioner, PreconditionIfMoreRowsThanCols, true>
|
|
|
|
|
{
|
|
|
|
|
static bool run(JacobiSVD<MatrixType, ColPivHouseholderQRPreconditioner>& svd, const MatrixType& matrix)
|
|
|
|
|
{
|
|
|
|
|
if(matrix.rows() > matrix.cols())
|
|
|
|
|
{
|
|
|
|
|
ColPivHouseholderQR<MatrixType> qr(matrix);
|
|
|
|
|
svd.m_workMatrix = qr.matrixQR().block(0,0,matrix.cols(),matrix.cols()).template triangularView<Upper>();
|
2010-10-08 10:42:40 -04:00
|
|
|
if(svd.m_computeFullU) svd.m_matrixU = qr.householderQ();
|
|
|
|
|
else if(svd.m_computeThinU) {
|
|
|
|
|
svd.m_matrixU.setIdentity(matrix.rows(), matrix.cols());
|
|
|
|
|
qr.householderQ().applyThisOnTheLeft(svd.m_matrixU);
|
|
|
|
|
}
|
|
|
|
|
if(svd.computeV()) svd.m_matrixV = qr.colsPermutation();
|
2010-10-08 10:42:32 -04:00
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
template<typename MatrixType>
|
|
|
|
|
struct ei_qr_preconditioner_impl<MatrixType, ColPivHouseholderQRPreconditioner, PreconditionIfMoreColsThanRows, true>
|
|
|
|
|
{
|
|
|
|
|
static bool run(JacobiSVD<MatrixType, ColPivHouseholderQRPreconditioner>& svd, const MatrixType& matrix)
|
|
|
|
|
{
|
|
|
|
|
if(matrix.cols() > matrix.rows())
|
|
|
|
|
{
|
|
|
|
|
typedef Matrix<typename MatrixType::Scalar, MatrixType::ColsAtCompileTime, MatrixType::RowsAtCompileTime,
|
|
|
|
|
MatrixType::Options, MatrixType::MaxColsAtCompileTime, MatrixType::MaxRowsAtCompileTime>
|
|
|
|
|
TransposeTypeWithSameStorageOrder;
|
|
|
|
|
ColPivHouseholderQR<TransposeTypeWithSameStorageOrder> qr(matrix.adjoint());
|
|
|
|
|
svd.m_workMatrix = qr.matrixQR().block(0,0,matrix.rows(),matrix.rows()).template triangularView<Upper>().adjoint();
|
2010-10-08 10:42:40 -04:00
|
|
|
if(svd.m_computeFullV) svd.m_matrixV = qr.householderQ();
|
|
|
|
|
else if(svd.m_computeThinV) {
|
|
|
|
|
svd.m_matrixV.setIdentity(matrix.cols(), matrix.rows());
|
|
|
|
|
qr.householderQ().applyThisOnTheLeft(svd.m_matrixV);
|
|
|
|
|
}
|
|
|
|
|
if(svd.computeU()) svd.m_matrixU = qr.colsPermutation();
|
2010-10-08 10:42:32 -04:00
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
else return false;
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
template<typename MatrixType>
|
|
|
|
|
struct ei_qr_preconditioner_impl<MatrixType, HouseholderQRPreconditioner, PreconditionIfMoreRowsThanCols, true>
|
|
|
|
|
{
|
|
|
|
|
static bool run(JacobiSVD<MatrixType, HouseholderQRPreconditioner>& svd, const MatrixType& matrix)
|
|
|
|
|
{
|
|
|
|
|
if(matrix.rows() > matrix.cols())
|
|
|
|
|
{
|
|
|
|
|
HouseholderQR<MatrixType> qr(matrix);
|
|
|
|
|
svd.m_workMatrix = qr.matrixQR().block(0,0,matrix.cols(),matrix.cols()).template triangularView<Upper>();
|
2010-10-08 10:42:40 -04:00
|
|
|
if(svd.m_computeFullU) svd.m_matrixU = qr.householderQ();
|
|
|
|
|
else if(svd.m_computeThinU) {
|
|
|
|
|
svd.m_matrixU.setIdentity(matrix.rows(), matrix.cols());
|
|
|
|
|
qr.householderQ().applyThisOnTheLeft(svd.m_matrixU);
|
|
|
|
|
}
|
|
|
|
|
if(svd.computeV()) svd.m_matrixV.setIdentity(matrix.cols(), matrix.cols());
|
2010-10-08 10:42:32 -04:00
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
template<typename MatrixType>
|
|
|
|
|
struct ei_qr_preconditioner_impl<MatrixType, HouseholderQRPreconditioner, PreconditionIfMoreColsThanRows, true>
|
|
|
|
|
{
|
|
|
|
|
static bool run(JacobiSVD<MatrixType, HouseholderQRPreconditioner>& svd, const MatrixType& matrix)
|
|
|
|
|
{
|
|
|
|
|
if(matrix.cols() > matrix.rows())
|
|
|
|
|
{
|
|
|
|
|
typedef Matrix<typename MatrixType::Scalar, MatrixType::ColsAtCompileTime, MatrixType::RowsAtCompileTime,
|
|
|
|
|
MatrixType::Options, MatrixType::MaxColsAtCompileTime, MatrixType::MaxRowsAtCompileTime>
|
|
|
|
|
TransposeTypeWithSameStorageOrder;
|
|
|
|
|
HouseholderQR<TransposeTypeWithSameStorageOrder> qr(matrix.adjoint());
|
|
|
|
|
svd.m_workMatrix = qr.matrixQR().block(0,0,matrix.rows(),matrix.rows()).template triangularView<Upper>().adjoint();
|
2010-10-08 10:42:40 -04:00
|
|
|
if(svd.m_computeFullV) svd.m_matrixV = qr.householderQ();
|
|
|
|
|
else if(svd.m_computeThinV) {
|
|
|
|
|
svd.m_matrixV.setIdentity(matrix.cols(), matrix.rows());
|
|
|
|
|
qr.householderQ().applyThisOnTheLeft(svd.m_matrixV);
|
|
|
|
|
}
|
|
|
|
|
if(svd.computeU()) svd.m_matrixU.setIdentity(matrix.rows(), matrix.rows());
|
2010-10-08 10:42:32 -04:00
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
else return false;
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2009-08-31 22:26:15 -04:00
|
|
|
/** \ingroup SVD_Module
|
2010-06-29 10:10:47 -04:00
|
|
|
*
|
2009-08-31 22:26:15 -04:00
|
|
|
*
|
|
|
|
|
* \class JacobiSVD
|
|
|
|
|
*
|
2010-10-08 10:42:06 -04:00
|
|
|
* \brief Jacobi SVD decomposition of a square matrix
|
2009-08-31 22:26:15 -04:00
|
|
|
*
|
|
|
|
|
* \param MatrixType the type of the matrix of which we are computing the SVD decomposition
|
2010-10-08 10:42:40 -04:00
|
|
|
* \param QRPreconditioner this optional parameter allows to specify the type of QR decomposition that will be used internally
|
|
|
|
|
* for the R-SVD step for non-square matrices. See discussion of possible values below.
|
|
|
|
|
*
|
|
|
|
|
* The possible values for QRPreconditioner are:
|
|
|
|
|
* \li FullPivHouseholderQRPreconditioner (the default), is the safest and slowest. It uses full-pivoting QR.
|
|
|
|
|
* We make it the default so that JacobiSVD is guaranteed to be entirely, uncompromisingly safe by default.
|
|
|
|
|
* Contrary to other QRs, it doesn't allow computing thin unitaries.
|
|
|
|
|
* \li ColPivHouseholderQRPreconditioner is faster, and in practice still very safe, although theoretically not as safe as the default
|
|
|
|
|
* full-pivoting preconditioner. It uses column-pivoting QR.
|
|
|
|
|
* \li HouseholderQRPreconditioner is even faster, and less safe and accurate than the pivoting variants. It uses non-pivoting QR.
|
|
|
|
|
* This is very similar in safety and accuracy to the bidiagonalization process used by bidiagonalizing SVD algorithms (since bidiagonalization
|
|
|
|
|
* is inherently non-pivoting).
|
|
|
|
|
* \li NoQRPreconditioner allows to not use a QR preconditioner at all. This is useful if you know that you will only be computing
|
|
|
|
|
* JacobiSVD decompositions of square matrices. Non-square matrices require a QR preconditioner. Using this option will result in
|
|
|
|
|
* faster compilation and smaller executable code.
|
2009-08-31 22:26:15 -04:00
|
|
|
*
|
|
|
|
|
* \sa MatrixBase::jacobiSvd()
|
|
|
|
|
*/
|
2010-10-08 10:42:32 -04:00
|
|
|
template<typename MatrixType, int QRPreconditioner> class JacobiSVD
|
2009-08-31 22:26:15 -04:00
|
|
|
{
|
|
|
|
|
private:
|
|
|
|
|
typedef typename MatrixType::Scalar Scalar;
|
|
|
|
|
typedef typename NumTraits<typename MatrixType::Scalar>::Real RealScalar;
|
2010-05-30 16:00:58 -04:00
|
|
|
typedef typename MatrixType::Index Index;
|
2009-08-31 22:26:15 -04:00
|
|
|
enum {
|
|
|
|
|
RowsAtCompileTime = MatrixType::RowsAtCompileTime,
|
|
|
|
|
ColsAtCompileTime = MatrixType::ColsAtCompileTime,
|
2010-06-14 09:05:08 -04:00
|
|
|
DiagSizeAtCompileTime = EIGEN_SIZE_MIN_PREFER_DYNAMIC(RowsAtCompileTime,ColsAtCompileTime),
|
2009-08-31 22:26:15 -04:00
|
|
|
MaxRowsAtCompileTime = MatrixType::MaxRowsAtCompileTime,
|
|
|
|
|
MaxColsAtCompileTime = MatrixType::MaxColsAtCompileTime,
|
2010-06-14 09:05:08 -04:00
|
|
|
MaxDiagSizeAtCompileTime = EIGEN_SIZE_MIN_PREFER_FIXED(MaxRowsAtCompileTime,MaxColsAtCompileTime),
|
2009-08-31 22:26:15 -04:00
|
|
|
MatrixOptions = MatrixType::Options
|
|
|
|
|
};
|
2009-09-01 13:18:03 +02:00
|
|
|
|
2010-10-08 10:42:32 -04:00
|
|
|
typedef Matrix<Scalar, RowsAtCompileTime, RowsAtCompileTime,
|
|
|
|
|
MatrixOptions, MaxRowsAtCompileTime, MaxRowsAtCompileTime>
|
|
|
|
|
MatrixUType;
|
|
|
|
|
typedef Matrix<Scalar, ColsAtCompileTime, ColsAtCompileTime,
|
|
|
|
|
MatrixOptions, MaxColsAtCompileTime, MaxColsAtCompileTime>
|
|
|
|
|
MatrixVType;
|
2010-03-19 02:12:23 -04:00
|
|
|
typedef typename ei_plain_diag_type<MatrixType, RealScalar>::type SingularValuesType;
|
|
|
|
|
typedef typename ei_plain_row_type<MatrixType>::type RowType;
|
|
|
|
|
typedef typename ei_plain_col_type<MatrixType>::type ColType;
|
2009-09-03 02:53:51 -04:00
|
|
|
typedef Matrix<Scalar, DiagSizeAtCompileTime, DiagSizeAtCompileTime,
|
|
|
|
|
MatrixOptions, MaxDiagSizeAtCompileTime, MaxDiagSizeAtCompileTime>
|
2010-10-08 10:42:32 -04:00
|
|
|
WorkMatrixType;
|
2009-08-31 22:26:15 -04:00
|
|
|
|
|
|
|
|
public:
|
|
|
|
|
|
2010-04-21 17:15:57 +02:00
|
|
|
/** \brief Default Constructor.
|
|
|
|
|
*
|
|
|
|
|
* The default constructor is useful in cases in which the user intends to
|
|
|
|
|
* perform decompositions via JacobiSVD::compute(const MatrixType&).
|
|
|
|
|
*/
|
2009-08-31 22:26:15 -04:00
|
|
|
JacobiSVD() : m_isInitialized(false) {}
|
|
|
|
|
|
2010-04-21 17:15:57 +02:00
|
|
|
|
|
|
|
|
/** \brief Default Constructor with memory preallocation
|
|
|
|
|
*
|
|
|
|
|
* Like the default constructor but with preallocation of the internal data
|
|
|
|
|
* according to the specified problem \a size.
|
|
|
|
|
* \sa JacobiSVD()
|
|
|
|
|
*/
|
2010-05-30 16:00:58 -04:00
|
|
|
JacobiSVD(Index rows, Index cols) : m_matrixU(rows, rows),
|
2010-04-21 17:15:57 +02:00
|
|
|
m_matrixV(cols, cols),
|
|
|
|
|
m_singularValues(std::min(rows, cols)),
|
|
|
|
|
m_workMatrix(rows, cols),
|
|
|
|
|
m_isInitialized(false) {}
|
|
|
|
|
|
2010-10-08 10:42:40 -04:00
|
|
|
/** \brief Constructor performing the decomposition of given matrix.
|
|
|
|
|
*
|
|
|
|
|
* \param matrix the matrix to decompose
|
|
|
|
|
* \param computationOptions optional parameter allowing to specify if you want full or thin U or V unitaries to be computed.
|
|
|
|
|
* By default, none is computed. This is a bit-field, the possible bits are ComputeFullU, ComputeThinU,
|
|
|
|
|
* ComputeFullV, ComputeThinV.
|
|
|
|
|
*
|
|
|
|
|
* Thin unitaries are not available with the default FullPivHouseholderQRPreconditioner, see class documentation for details.
|
|
|
|
|
* If you want thin unitaries, use another preconditioner, for example:
|
|
|
|
|
* \code
|
|
|
|
|
* JacobiSVD<MatrixXf, ColPivHouseholderQRPreconditioner> svd(matrix, ComputeThinU);
|
|
|
|
|
* \endcode
|
|
|
|
|
*
|
|
|
|
|
* Thin unitaries also are only available if your matrix type has a Dynamic number of columns (for example MatrixXf).
|
|
|
|
|
*/
|
2010-10-08 10:42:32 -04:00
|
|
|
JacobiSVD(const MatrixType& matrix, unsigned int computationOptions = 0)
|
|
|
|
|
: m_matrixU(matrix.rows(), matrix.rows()),
|
|
|
|
|
m_matrixV(matrix.cols(), matrix.cols()),
|
|
|
|
|
m_singularValues(),
|
|
|
|
|
m_workMatrix(),
|
|
|
|
|
m_isInitialized(false)
|
2009-08-31 22:26:15 -04:00
|
|
|
{
|
2010-05-30 16:00:58 -04:00
|
|
|
const Index minSize = std::min(matrix.rows(), matrix.cols());
|
2010-04-21 17:15:57 +02:00
|
|
|
m_singularValues.resize(minSize);
|
|
|
|
|
m_workMatrix.resize(minSize, minSize);
|
2010-10-08 10:42:32 -04:00
|
|
|
compute(matrix, computationOptions);
|
2009-08-31 22:26:15 -04:00
|
|
|
}
|
2009-09-01 13:18:03 +02:00
|
|
|
|
2010-10-08 10:42:40 -04:00
|
|
|
/** \brief Method performing the decomposition of given matrix.
|
|
|
|
|
*
|
|
|
|
|
* \param matrix the matrix to decompose
|
|
|
|
|
* \param computationOptions optional parameter allowing to specify if you want full or thin U or V unitaries to be computed.
|
|
|
|
|
* By default, none is computed. This is a bit-field, the possible bits are ComputeFullU, ComputeThinU,
|
|
|
|
|
* ComputeFullV, ComputeThinV.
|
|
|
|
|
*
|
|
|
|
|
* Thin unitaries are not available with the default FullPivHouseholderQRPreconditioner, see class documentation for details.
|
|
|
|
|
* If you want thin unitaries, use another preconditioner, for example:
|
|
|
|
|
* \code
|
|
|
|
|
* JacobiSVD<MatrixXf, ColPivHouseholderQRPreconditioner> svd(matrix, ComputeThinU);
|
|
|
|
|
* \endcode
|
|
|
|
|
*
|
|
|
|
|
* Thin unitaries also are only available if your matrix type has a Dynamic number of columns (for example MatrixXf).
|
|
|
|
|
*/
|
2010-10-08 10:42:32 -04:00
|
|
|
JacobiSVD& compute(const MatrixType& matrix, unsigned int computationOptions = 0);
|
2009-09-01 13:18:03 +02:00
|
|
|
|
2009-08-31 22:26:15 -04:00
|
|
|
const MatrixUType& matrixU() const
|
|
|
|
|
{
|
|
|
|
|
ei_assert(m_isInitialized && "JacobiSVD is not initialized.");
|
2010-10-08 10:42:40 -04:00
|
|
|
ei_assert(computeU() && "This JacobiSVD decomposition didn't compute U. Did you ask for it?");
|
2009-08-31 22:26:15 -04:00
|
|
|
return m_matrixU;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
const SingularValuesType& singularValues() const
|
|
|
|
|
{
|
|
|
|
|
ei_assert(m_isInitialized && "JacobiSVD is not initialized.");
|
|
|
|
|
return m_singularValues;
|
|
|
|
|
}
|
|
|
|
|
|
2009-09-03 02:53:51 -04:00
|
|
|
const MatrixVType& matrixV() const
|
2009-08-31 22:26:15 -04:00
|
|
|
{
|
|
|
|
|
ei_assert(m_isInitialized && "JacobiSVD is not initialized.");
|
2010-10-08 10:42:40 -04:00
|
|
|
ei_assert(computeV() && "This JacobiSVD decomposition didn't compute V. Did you ask for it?");
|
2009-08-31 22:26:15 -04:00
|
|
|
return m_matrixV;
|
|
|
|
|
}
|
|
|
|
|
|
2010-10-08 10:42:40 -04:00
|
|
|
inline bool computeU() const { return m_computeFullU || m_computeThinU; }
|
|
|
|
|
inline bool computeV() const { return m_computeFullV || m_computeThinV; }
|
|
|
|
|
|
2009-08-31 22:26:15 -04:00
|
|
|
protected:
|
|
|
|
|
MatrixUType m_matrixU;
|
|
|
|
|
MatrixVType m_matrixV;
|
|
|
|
|
SingularValuesType m_singularValues;
|
2010-04-21 17:15:57 +02:00
|
|
|
WorkMatrixType m_workMatrix;
|
2010-10-08 10:42:40 -04:00
|
|
|
bool m_isInitialized;
|
|
|
|
|
bool m_computeFullU, m_computeThinU;
|
|
|
|
|
bool m_computeFullV, m_computeThinV;
|
2009-09-01 13:18:03 +02:00
|
|
|
|
2010-10-08 10:42:32 -04:00
|
|
|
template<typename _MatrixType, int _QRPreconditioner, bool _IsComplex>
|
2009-08-31 22:26:15 -04:00
|
|
|
friend struct ei_svd_precondition_2x2_block_to_be_real;
|
2010-10-08 10:42:32 -04:00
|
|
|
template<typename _MatrixType, int _QRPreconditioner, int _Case, bool _DoAnything>
|
|
|
|
|
friend struct ei_qr_preconditioner_impl;
|
2009-08-31 22:26:15 -04:00
|
|
|
};
|
|
|
|
|
|
2010-10-08 10:42:32 -04:00
|
|
|
template<typename MatrixType, int QRPreconditioner>
|
|
|
|
|
struct ei_svd_precondition_2x2_block_to_be_real<MatrixType, QRPreconditioner, false>
|
2009-08-31 22:26:15 -04:00
|
|
|
{
|
2010-10-08 10:42:32 -04:00
|
|
|
typedef JacobiSVD<MatrixType, QRPreconditioner> SVD;
|
2010-05-30 16:00:58 -04:00
|
|
|
typedef typename SVD::Index Index;
|
2010-10-08 10:42:32 -04:00
|
|
|
static void run(typename SVD::WorkMatrixType&, SVD&, Index, Index) {}
|
2009-08-31 22:26:15 -04:00
|
|
|
};
|
|
|
|
|
|
2010-10-08 10:42:32 -04:00
|
|
|
template<typename MatrixType, int QRPreconditioner>
|
|
|
|
|
struct ei_svd_precondition_2x2_block_to_be_real<MatrixType, QRPreconditioner, true>
|
2009-08-31 22:26:15 -04:00
|
|
|
{
|
2010-10-08 10:42:32 -04:00
|
|
|
typedef JacobiSVD<MatrixType, QRPreconditioner> SVD;
|
2009-08-31 22:26:15 -04:00
|
|
|
typedef typename MatrixType::Scalar Scalar;
|
|
|
|
|
typedef typename MatrixType::RealScalar RealScalar;
|
2010-05-30 16:00:58 -04:00
|
|
|
typedef typename SVD::Index Index;
|
2010-10-08 10:42:32 -04:00
|
|
|
static void run(typename SVD::WorkMatrixType& work_matrix, SVD& svd, Index p, Index q)
|
2009-08-31 22:26:15 -04:00
|
|
|
{
|
2009-09-01 13:18:03 +02:00
|
|
|
Scalar z;
|
2009-09-02 15:04:10 +02:00
|
|
|
PlanarRotation<Scalar> rot;
|
2009-08-31 22:26:15 -04:00
|
|
|
RealScalar n = ei_sqrt(ei_abs2(work_matrix.coeff(p,p)) + ei_abs2(work_matrix.coeff(q,p)));
|
|
|
|
|
if(n==0)
|
|
|
|
|
{
|
|
|
|
|
z = ei_abs(work_matrix.coeff(p,q)) / work_matrix.coeff(p,q);
|
|
|
|
|
work_matrix.row(p) *= z;
|
2010-10-08 10:42:40 -04:00
|
|
|
if(svd.computeU()) svd.m_matrixU.col(p) *= ei_conj(z);
|
2009-08-31 22:26:15 -04:00
|
|
|
z = ei_abs(work_matrix.coeff(q,q)) / work_matrix.coeff(q,q);
|
|
|
|
|
work_matrix.row(q) *= z;
|
2010-10-08 10:42:40 -04:00
|
|
|
if(svd.computeU()) svd.m_matrixU.col(q) *= ei_conj(z);
|
2009-08-31 22:26:15 -04:00
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
2009-09-01 13:18:03 +02:00
|
|
|
rot.c() = ei_conj(work_matrix.coeff(p,p)) / n;
|
|
|
|
|
rot.s() = work_matrix.coeff(q,p) / n;
|
2009-09-02 15:04:10 +02:00
|
|
|
work_matrix.applyOnTheLeft(p,q,rot);
|
2010-10-08 10:42:40 -04:00
|
|
|
if(svd.computeU()) svd.m_matrixU.applyOnTheRight(p,q,rot.adjoint());
|
2009-08-31 22:26:15 -04:00
|
|
|
if(work_matrix.coeff(p,q) != Scalar(0))
|
|
|
|
|
{
|
|
|
|
|
Scalar z = ei_abs(work_matrix.coeff(p,q)) / work_matrix.coeff(p,q);
|
|
|
|
|
work_matrix.col(q) *= z;
|
2010-10-08 10:42:40 -04:00
|
|
|
if(svd.computeV()) svd.m_matrixV.col(q) *= z;
|
2009-08-31 22:26:15 -04:00
|
|
|
}
|
|
|
|
|
if(work_matrix.coeff(q,q) != Scalar(0))
|
|
|
|
|
{
|
|
|
|
|
z = ei_abs(work_matrix.coeff(q,q)) / work_matrix.coeff(q,q);
|
|
|
|
|
work_matrix.row(q) *= z;
|
2010-10-08 10:42:40 -04:00
|
|
|
if(svd.computeU()) svd.m_matrixU.col(q) *= ei_conj(z);
|
2009-08-31 22:26:15 -04:00
|
|
|
}
|
|
|
|
|
}
|
2009-09-01 13:18:03 +02:00
|
|
|
}
|
2009-08-31 22:26:15 -04:00
|
|
|
};
|
|
|
|
|
|
2010-05-30 16:00:58 -04:00
|
|
|
template<typename MatrixType, typename RealScalar, typename Index>
|
|
|
|
|
void ei_real_2x2_jacobi_svd(const MatrixType& matrix, Index p, Index q,
|
2009-09-02 15:04:10 +02:00
|
|
|
PlanarRotation<RealScalar> *j_left,
|
|
|
|
|
PlanarRotation<RealScalar> *j_right)
|
2009-08-31 22:26:15 -04:00
|
|
|
{
|
|
|
|
|
Matrix<RealScalar,2,2> m;
|
|
|
|
|
m << ei_real(matrix.coeff(p,p)), ei_real(matrix.coeff(p,q)),
|
2009-09-01 13:18:03 +02:00
|
|
|
ei_real(matrix.coeff(q,p)), ei_real(matrix.coeff(q,q));
|
2009-09-02 15:04:10 +02:00
|
|
|
PlanarRotation<RealScalar> rot1;
|
2009-08-31 22:26:15 -04:00
|
|
|
RealScalar t = m.coeff(0,0) + m.coeff(1,1);
|
|
|
|
|
RealScalar d = m.coeff(1,0) - m.coeff(0,1);
|
|
|
|
|
if(t == RealScalar(0))
|
|
|
|
|
{
|
2009-09-01 13:18:03 +02:00
|
|
|
rot1.c() = 0;
|
|
|
|
|
rot1.s() = d > 0 ? 1 : -1;
|
2009-08-31 22:26:15 -04:00
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
RealScalar u = d / t;
|
2009-09-01 13:18:03 +02:00
|
|
|
rot1.c() = RealScalar(1) / ei_sqrt(1 + ei_abs2(u));
|
|
|
|
|
rot1.s() = rot1.c() * u;
|
2009-08-31 22:26:15 -04:00
|
|
|
}
|
2009-09-02 15:04:10 +02:00
|
|
|
m.applyOnTheLeft(0,1,rot1);
|
|
|
|
|
j_right->makeJacobi(m,0,1);
|
2009-09-01 13:18:03 +02:00
|
|
|
*j_left = rot1 * j_right->transpose();
|
2009-08-31 22:26:15 -04:00
|
|
|
}
|
|
|
|
|
|
2010-10-08 10:42:32 -04:00
|
|
|
template<typename MatrixType, int QRPreconditioner>
|
|
|
|
|
JacobiSVD<MatrixType, QRPreconditioner>&
|
|
|
|
|
JacobiSVD<MatrixType, QRPreconditioner>::compute(const MatrixType& matrix, unsigned int computationOptions)
|
2009-08-31 22:26:15 -04:00
|
|
|
{
|
2010-10-08 10:42:40 -04:00
|
|
|
m_computeFullU = computationOptions & ComputeFullU;
|
|
|
|
|
m_computeThinU = computationOptions & ComputeThinU;
|
|
|
|
|
m_computeFullV = computationOptions & ComputeFullV;
|
|
|
|
|
m_computeThinV = computationOptions & ComputeThinV;
|
|
|
|
|
ei_assert(!(m_computeFullU && m_computeThinU) && "JacobiSVD: you can't ask for both full and thin U");
|
|
|
|
|
ei_assert(!(m_computeFullV && m_computeThinV) && "JacobiSVD: you can't ask for both full and thin V");
|
|
|
|
|
ei_assert(EIGEN_IMPLIES(m_computeThinU || m_computeThinV, MatrixType::ColsAtCompileTime==Dynamic) &&
|
|
|
|
|
"JacobiSVD: thin U and V are only available when your matrix has a dynamic number of columns.");
|
2010-05-30 16:00:58 -04:00
|
|
|
Index rows = matrix.rows();
|
|
|
|
|
Index cols = matrix.cols();
|
|
|
|
|
Index diagSize = std::min(rows, cols);
|
2009-09-02 06:36:55 -04:00
|
|
|
m_singularValues.resize(diagSize);
|
2010-02-10 10:52:28 +01:00
|
|
|
const RealScalar precision = 2 * NumTraits<Scalar>::epsilon();
|
2009-08-31 22:26:15 -04:00
|
|
|
|
2010-10-08 10:42:32 -04:00
|
|
|
if(!ei_qr_preconditioner_impl<MatrixType, QRPreconditioner, PreconditionIfMoreColsThanRows>::run(*this, matrix)
|
|
|
|
|
&& !ei_qr_preconditioner_impl<MatrixType, QRPreconditioner, PreconditionIfMoreRowsThanCols>::run(*this, matrix))
|
2009-09-02 06:36:55 -04:00
|
|
|
{
|
2010-04-21 17:15:57 +02:00
|
|
|
m_workMatrix = matrix.block(0,0,diagSize,diagSize);
|
2010-10-08 10:42:40 -04:00
|
|
|
if(m_computeFullU) m_matrixU.setIdentity(rows,rows);
|
|
|
|
|
if(m_computeThinU) m_matrixU.setIdentity(rows,diagSize);
|
|
|
|
|
if(m_computeFullV) m_matrixV.setIdentity(cols,cols);
|
|
|
|
|
if(m_computeThinV) m_matrixV.setIdentity(diagSize,cols);
|
2009-09-02 06:36:55 -04:00
|
|
|
}
|
2009-09-03 02:53:51 -04:00
|
|
|
|
|
|
|
|
bool finished = false;
|
|
|
|
|
while(!finished)
|
2009-08-31 22:26:15 -04:00
|
|
|
{
|
2009-09-03 02:53:51 -04:00
|
|
|
finished = true;
|
2010-05-30 16:00:58 -04:00
|
|
|
for(Index p = 1; p < diagSize; ++p)
|
2009-08-31 22:26:15 -04:00
|
|
|
{
|
2010-05-30 16:00:58 -04:00
|
|
|
for(Index q = 0; q < p; ++q)
|
2009-08-31 22:26:15 -04:00
|
|
|
{
|
2010-04-21 17:15:57 +02:00
|
|
|
if(std::max(ei_abs(m_workMatrix.coeff(p,q)),ei_abs(m_workMatrix.coeff(q,p)))
|
|
|
|
|
> std::max(ei_abs(m_workMatrix.coeff(p,p)),ei_abs(m_workMatrix.coeff(q,q)))*precision)
|
2009-09-03 02:53:51 -04:00
|
|
|
{
|
|
|
|
|
finished = false;
|
2010-10-08 10:42:32 -04:00
|
|
|
ei_svd_precondition_2x2_block_to_be_real<MatrixType, QRPreconditioner>::run(m_workMatrix, *this, p, q);
|
2009-08-31 22:26:15 -04:00
|
|
|
|
2009-09-03 02:53:51 -04:00
|
|
|
PlanarRotation<RealScalar> j_left, j_right;
|
2010-04-21 17:15:57 +02:00
|
|
|
ei_real_2x2_jacobi_svd(m_workMatrix, p, q, &j_left, &j_right);
|
2009-09-01 13:18:03 +02:00
|
|
|
|
2010-04-21 17:15:57 +02:00
|
|
|
m_workMatrix.applyOnTheLeft(p,q,j_left);
|
2010-10-08 10:42:40 -04:00
|
|
|
if(computeU()) m_matrixU.applyOnTheRight(p,q,j_left.transpose());
|
2009-09-01 13:18:03 +02:00
|
|
|
|
2010-04-21 17:15:57 +02:00
|
|
|
m_workMatrix.applyOnTheRight(p,q,j_right);
|
2010-10-08 10:42:40 -04:00
|
|
|
if(computeV()) m_matrixV.applyOnTheRight(p,q,j_right);
|
2009-09-03 02:53:51 -04:00
|
|
|
}
|
2009-08-31 22:26:15 -04:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
2009-09-01 13:18:03 +02:00
|
|
|
|
2010-05-30 16:00:58 -04:00
|
|
|
for(Index i = 0; i < diagSize; ++i)
|
2009-08-31 22:26:15 -04:00
|
|
|
{
|
2010-04-21 17:15:57 +02:00
|
|
|
RealScalar a = ei_abs(m_workMatrix.coeff(i,i));
|
2009-08-31 22:26:15 -04:00
|
|
|
m_singularValues.coeffRef(i) = a;
|
2010-10-08 10:42:40 -04:00
|
|
|
if(computeU() && (a!=RealScalar(0))) m_matrixU.col(i) *= m_workMatrix.coeff(i,i)/a;
|
2009-08-31 22:26:15 -04:00
|
|
|
}
|
|
|
|
|
|
2010-05-30 16:00:58 -04:00
|
|
|
for(Index i = 0; i < diagSize; i++)
|
2009-08-31 22:26:15 -04:00
|
|
|
{
|
2010-05-30 16:00:58 -04:00
|
|
|
Index pos;
|
2010-01-04 21:24:43 -05:00
|
|
|
m_singularValues.tail(diagSize-i).maxCoeff(&pos);
|
2009-08-31 22:26:15 -04:00
|
|
|
if(pos)
|
|
|
|
|
{
|
|
|
|
|
pos += i;
|
|
|
|
|
std::swap(m_singularValues.coeffRef(i), m_singularValues.coeffRef(pos));
|
2010-10-08 10:42:40 -04:00
|
|
|
if(computeU()) m_matrixU.col(pos).swap(m_matrixU.col(i));
|
|
|
|
|
if(computeV()) m_matrixV.col(pos).swap(m_matrixV.col(i));
|
2009-08-31 22:26:15 -04:00
|
|
|
}
|
|
|
|
|
}
|
2009-09-01 13:18:03 +02:00
|
|
|
|
2009-08-31 22:26:15 -04:00
|
|
|
m_isInitialized = true;
|
|
|
|
|
return *this;
|
|
|
|
|
}
|
|
|
|
|
#endif // EIGEN_JACOBISVD_H
|