Files
igl/python/py_igl/py_eigs.cpp
T

30 lines
988 B
C++

// This file is part of libigl, a simple c++ geometry processing library.
//
// Copyright (C) 2017 Sebastian Koch <s.koch@tu-berlin.de> and Daniele Panozzo <daniele.panozzo@gmail.com>
//
// This Source Code Form is subject to the terms of the Mozilla Public License
// v. 2.0. If a copy of the MPL was not distributed with this file, You can
// obtain one at http://mozilla.org/MPL/2.0/.
py::enum_<igl::EigsType>(m, "EigsType")
.value("EIGS_TYPE_SM", igl::EIGS_TYPE_SM)
.value("EIGS_TYPE_LM", igl::EIGS_TYPE_LM)
.value("NUM_EIGS_TYPES", igl::NUM_EIGS_TYPES)
.export_values();
m.def("eigs", []
(
const Eigen::SparseMatrix<double>& A,
const Eigen::SparseMatrix<double>& B,
const size_t k,
const igl::EigsType type,
Eigen::MatrixXd& sU,
Eigen::MatrixXd& sS
)
{
Eigen::VectorXd sSt;
bool ret = igl::eigs(A,B,k,type,sU,sSt);
sS = sSt;
return ret;
}, __doc_igl_eigs,
py::arg("A"), py::arg("B"), py::arg("k"), py::arg("type"), py::arg("sU"), py::arg("sS"));