vpMatrix_svd.cpp

/****************************************************************************
 *
 * This file is part of the ViSP software.
 * Copyright (C) 2005 - 2017 by Inria. All rights reserved.
 *
 * This software is free software; 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.
 * See the file LICENSE.txt at the root directory of this source
 * distribution for additional information about the GNU GPL.
 *
 * For using ViSP with software that can not be combined with the GNU
 * GPL, please contact Inria about acquiring a ViSP Professional
 * Edition License.
 *
 * See http://visp.inria.fr for more information.
 *
 * This software was developed at:
 * Inria Rennes - Bretagne Atlantique
 * Campus Universitaire de Beaulieu
 * 35042 Rennes Cedex
 * France
 *
 * If you have questions regarding the use of this file, please contact
 * Inria at visp@inria.fr
 *
 * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 *
 * Description:
 * Matrix SVD decomposition.
 *
 * Authors:
 * Eric Marchand
 *
 *****************************************************************************/

#include <visp3/core/vpColVector.h>
#include <visp3/core/vpConfig.h>
#include <visp3/core/vpDebug.h>
#include <visp3/core/vpException.h>
#include <visp3/core/vpMath.h>
#include <visp3/core/vpMatrix.h>
#include <visp3/core/vpMatrixException.h>

#include <cmath> // std::fabs
#include <iostream>
#include <limits> // numeric_limits

#ifndef DOXYGEN_SHOULD_SKIP_THIS

#ifdef VISP_HAVE_EIGEN3
#include <Eigen/SVD>
#endif

#ifdef VISP_HAVE_GSL
#include <gsl/gsl_linalg.h>
#endif

#if (VISP_HAVE_OPENCV_VERSION >= 0x020101) // Require opencv >= 2.1.1
#include <opencv2/core/core.hpp>
#endif

#ifdef VISP_HAVE_LAPACK
#ifdef VISP_HAVE_LAPACK_BUILT_IN
typedef long int integer;
#else
typedef int integer;
#endif

extern "C" int dgesdd_(char *jobz, integer *m, integer *n, double *a, integer *lda, double *s, double *u, integer *ldu,
                       double *vt, integer *ldvt, double *work, integer *lwork, integer *iwork, integer *info);

#include <stdio.h>
#include <string.h>
#endif
/*---------------------------------------------------------------------

SVD related functions

---------------------------------------------------------------------*/

#if (VISP_HAVE_OPENCV_VERSION >= 0x020101) // Require opencv >= 2.1.1

void vpMatrix::svdOpenCV(vpColVector &w, vpMatrix &V)
{
  int rows = (int)this->getRows();
  int cols = (int)this->getCols();
  cv::Mat m(rows, cols, CV_64F, this->data);
  cv::SVD opencvSVD(m);
  cv::Mat opencvV = opencvSVD.vt;
  cv::Mat opencvW = opencvSVD.w;
  V.resize((unsigned int)opencvV.rows, (unsigned int)opencvV.cols);
  w.resize((unsigned int)(opencvW.rows * opencvW.cols));

  memcpy(V.data, opencvV.data, (size_t)(8 * opencvV.rows * opencvV.cols));
  V = V.transpose();
  memcpy(w.data, opencvW.data, (size_t)(8 * opencvW.rows * opencvW.cols));
  this->resize((unsigned int)opencvSVD.u.rows, (unsigned int)opencvSVD.u.cols);
  memcpy(this->data, opencvSVD.u.data, (size_t)(8 * opencvSVD.u.rows * opencvSVD.u.cols));
}

#endif

#ifdef VISP_HAVE_LAPACK

void vpMatrix::svdLapack(vpColVector &w, vpMatrix &V)
{
  w.resize(this->getCols());
  V.resize(this->getCols(), this->getCols());

  integer m = (integer)(this->getCols());
  integer n = (integer)(this->getRows());
  integer lda = m;
  integer ldu = m;
  integer ldvt = (std::min)(m, n);
  integer info, lwork;

  double wkopt;
  double *work;

  integer *iwork = new integer[8 * static_cast<integer>((std::min)(n, m))];

  double *s = w.data;
  double *a = new double[static_cast<unsigned int>(lda * n)];
  memcpy(a, this->data, this->getRows() * this->getCols() * sizeof(double));
  double *u = V.data;
  double *vt = this->data;

  lwork = -1;
  dgesdd_((char *)"S", &m, &n, a, &lda, s, u, &ldu, vt, &ldvt, &wkopt, &lwork, iwork, &info);
  lwork = (int)wkopt;
  work = new double[static_cast<unsigned int>(lwork)];

  dgesdd_((char *)"S", &m, &n, a, &lda, s, u, &ldu, vt, &ldvt, work, &lwork, iwork, &info);

  if (info > 0) {
    throw(vpMatrixException(vpMatrixException::fatalError, "The algorithm computing SVD failed to converge."));
  }

  V = V.transpose();
  delete[] work;
  delete[] iwork;
  delete[] a;
}
#endif

#ifdef VISP_HAVE_GSL

void vpMatrix::svdGsl(vpColVector &w, vpMatrix &V)
{
  w.resize(this->getCols());
  V.resize(this->getCols(), this->getCols());

  unsigned int nc = getCols();
  unsigned int nr = getRows();
  gsl_vector *work = gsl_vector_alloc(nc);

  gsl_matrix A;
  A.size1 = nr;
  A.size2 = nc;
  A.tda = A.size2;
  A.data = this->data;
  A.owner = 0;
  A.block = 0;

  gsl_matrix V_;
  V_.size1 = nc;
  V_.size2 = nc;
  V_.tda = V_.size2;
  V_.data = V.data;
  V_.owner = 0;
  V_.block = 0;

  gsl_vector S;
  S.size = nc;
  S.stride = 1;
  S.data = w.data;
  S.owner = 0;
  S.block = 0;

  gsl_linalg_SV_decomp(&A, &V_, &S, work);

  gsl_vector_free(work);
}
#endif // # #GSL

#ifdef VISP_HAVE_EIGEN3

void vpMatrix::svdEigen3(vpColVector &w, vpMatrix &V)
{
  w.resize(this->getCols());
  V.resize(this->getCols(), this->getCols());

  Eigen::Map<Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> > M(this->data, this->getRows(),
                                                                                        this->getCols());

  Eigen::JacobiSVD<Eigen::MatrixXd> svd(M, Eigen::ComputeThinU | Eigen::ComputeThinV);

  Eigen::Map<Eigen::VectorXd> w_(w.data, w.size());
  Eigen::Map<Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> > V_(V.data, V.getRows(),
                                                                                         V.getCols());
  Eigen::Map<Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> > U_(this->data, this->getRows(),
                                                                                         this->getCols());
  w_ = svd.singularValues();
  V_ = svd.matrixV();
  U_ = svd.matrixU();
}
#endif

#endif // #ifndef DOXYGEN_SHOULD_SKIP_THIS