py_eigs.cpp 600 B

12345678910111213141516171819202122
  1. py::enum_<igl::EigsType>(m, "EigsType")
  2. .value("EIGS_TYPE_SM", igl::EIGS_TYPE_SM)
  3. .value("EIGS_TYPE_LM", igl::EIGS_TYPE_LM)
  4. .value("NUM_EIGS_TYPES", igl::NUM_EIGS_TYPES)
  5. .export_values();
  6. m.def("eigs", []
  7. (
  8. const Eigen::SparseMatrix<double>& A,
  9. const Eigen::SparseMatrix<double>& B,
  10. const size_t k,
  11. const igl::EigsType type,
  12. Eigen::MatrixXd& sU,
  13. Eigen::MatrixXd& sS
  14. )
  15. {
  16. Eigen::VectorXd sSt;
  17. bool ret = igl::eigs(A,B,k,type,sU,sSt);
  18. sS = sSt;
  19. return ret;
  20. }, __doc_igl_eigs,
  21. py::arg("A"), py::arg("B"), py::arg("k"), py::arg("type"), py::arg("sU"), py::arg("sS"));