sparse.cpp 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105
  1. #include "sparse.h"
  2. // Bug in unsupported/Eigen/SparseExtra needs iostream first
  3. #include <iostream>
  4. #include <vector>
  5. #include <unsupported/Eigen/SparseExtra>
  6. template <class IndexVector, class ValueVector, typename T>
  7. IGL_INLINE void igl::sparse(
  8. const IndexVector & I,
  9. const IndexVector & J,
  10. const ValueVector & V,
  11. Eigen::SparseMatrix<T>& X)
  12. {
  13. size_t m = (size_t)I.maxCoeff()+1;
  14. size_t n = (size_t)J.maxCoeff()+1;
  15. return igl::sparse(I,J,V,m,n,X);
  16. }
  17. #include "verbose.h"
  18. template <class IndexVector, class ValueVector, typename T>
  19. IGL_INLINE void igl::sparse(
  20. const IndexVector & I,
  21. const IndexVector & J,
  22. const ValueVector & V,
  23. const size_t m,
  24. const size_t n,
  25. Eigen::SparseMatrix<T>& X)
  26. {
  27. using namespace std;
  28. using namespace Eigen;
  29. assert((int)I.maxCoeff() < (int)m);
  30. assert((int)I.minCoeff() >= 0);
  31. assert((int)J.maxCoeff() < (int)n);
  32. assert((int)J.minCoeff() >= 0);
  33. assert(I.size() == J.size());
  34. assert(J.size() == V.size());
  35. // Really we just need .size() to be the same, but this is safer
  36. assert(I.rows() == J.rows());
  37. assert(J.rows() == V.rows());
  38. assert(I.cols() == J.cols());
  39. assert(J.cols() == V.cols());
  40. //// number of values
  41. //int nv = V.size();
  42. //Eigen::DynamicSparseMatrix<T, Eigen::RowMajor> dyn_X(m,n);
  43. //// over estimate the number of entries
  44. //dyn_X.reserve(I.size());
  45. //for(int i = 0;i < nv;i++)
  46. //{
  47. // dyn_X.coeffRef((int)I(i),(int)J(i)) += (T)V(i);
  48. //}
  49. //X = Eigen::SparseMatrix<T>(dyn_X);
  50. vector<Triplet<T> > IJV;
  51. IJV.reserve(I.size());
  52. for(int x = 0;x<I.size();x++)
  53. {
  54. IJV.push_back(Triplet<T >(I(x),J(x),V(x)));
  55. }
  56. X.resize(m,n);
  57. X.setFromTriplets(IJV.begin(),IJV.end());
  58. }
  59. template <typename DerivedD, typename T>
  60. IGL_INLINE void igl::sparse(
  61. const Eigen::PlainObjectBase<DerivedD>& D,
  62. Eigen::SparseMatrix<T>& X)
  63. {
  64. assert(false && "Obsolete. Just call D.sparseView() directly");
  65. using namespace std;
  66. using namespace Eigen;
  67. vector<Triplet<T> > DIJV;
  68. const int m = D.rows();
  69. const int n = D.cols();
  70. for(int i = 0;i<m;i++)
  71. {
  72. for(int j = 0;j<n;j++)
  73. {
  74. if(D(i,j)!=0)
  75. {
  76. DIJV.push_back(Triplet<T>(i,j,D(i,j)));
  77. }
  78. }
  79. }
  80. X.resize(m,n);
  81. X.setFromTriplets(DIJV.begin(),DIJV.end());
  82. }
  83. template <typename DerivedD>
  84. IGL_INLINE Eigen::SparseMatrix<typename DerivedD::Scalar > igl::sparse(
  85. const Eigen::PlainObjectBase<DerivedD>& D)
  86. {
  87. assert(false && "Obsolete. Just call D.sparseView() directly");
  88. Eigen::SparseMatrix<typename DerivedD::Scalar > X;
  89. igl::sparse(D,X);
  90. return X;
  91. }
  92. #ifndef IGL_HEADER_ONLY
  93. // Explicit template specialization
  94. template void igl::sparse<Eigen::Matrix<int, -1, 1, 0, -1, 1>, Eigen::Matrix<double, -1, 1, 0, -1, 1>, double>(Eigen::Matrix<int, -1, 1, 0, -1, 1> const&, Eigen::Matrix<int, -1, 1, 0, -1, 1> const&, Eigen::Matrix<double, -1, 1, 0, -1, 1> const&, size_t, size_t, Eigen::SparseMatrix<double, 0, int>&);
  95. template void igl::sparse<Eigen::Matrix<double, -1, -1, 0, -1, -1>, double>(Eigen::PlainObjectBase<Eigen::Matrix<double, -1, -1, 0, -1, -1> > const&, Eigen::SparseMatrix<double, 0, int>&);
  96. template Eigen::SparseMatrix<Eigen::Matrix<double, -1, -1, 0, -1, -1>::Scalar, 0, int> igl::sparse<Eigen::Matrix<double, -1, -1, 0, -1, -1> >(Eigen::PlainObjectBase<Eigen::Matrix<double, -1, -1, 0, -1, -1> > const&);
  97. template Eigen::SparseMatrix<Eigen::Matrix<double, -1, 1, 0, -1, 1>::Scalar, 0, int> igl::sparse<Eigen::Matrix<double, -1, 1, 0, -1, 1> >(Eigen::PlainObjectBase<Eigen::Matrix<double, -1, 1, 0, -1, 1> > const&);
  98. #endif