py_bbw.cpp 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. py::enum_<igl::bbw::QPSolver>(m, "QPSolver")
  2. .value("QP_SOLVER_IGL_ACTIVE_SET", igl::bbw::QP_SOLVER_IGL_ACTIVE_SET)
  3. .value("QP_SOLVER_MOSEK", igl::bbw::QP_SOLVER_MOSEK)
  4. .value("NUM_QP_SOLVERS", igl::bbw::NUM_QP_SOLVERS)
  5. .export_values();
  6. // Wrap the BBWData class
  7. py::class_<igl::bbw::BBWData > BBWData(m, "BBWData");
  8. BBWData
  9. .def(py::init<>())
  10. .def_readwrite("partition_unity", &igl::bbw::BBWData::partition_unity)
  11. .def_readwrite("W0", &igl::bbw::BBWData::W0)
  12. .def_readwrite("active_set_params", &igl::bbw::BBWData::active_set_params)
  13. .def_readwrite("qp_solver", &igl::bbw::BBWData::qp_solver)
  14. .def_readwrite("verbosity", &igl::bbw::BBWData::verbosity)
  15. #ifndef IGL_NO_MOSEK
  16. .def_readwrite("mosek_data", &igl::bbw::BBWData::mosek_data)
  17. #endif
  18. .def("print", [](igl::bbw::BBWData& data)
  19. {
  20. return data.print();
  21. })
  22. ;
  23. m.def("bbw", []
  24. (
  25. const Eigen::MatrixXd& V,
  26. const Eigen::MatrixXi& Ele,
  27. const Eigen::MatrixXi& b,
  28. const Eigen::MatrixXd& bc,
  29. igl::bbw::BBWData& data,
  30. Eigen::MatrixXd& W
  31. )
  32. {
  33. assert_is_VectorX("b",b);
  34. Eigen::VectorXi bv;
  35. if (b.size() != 0)
  36. bv = b;
  37. return igl::bbw::bbw(V, Ele, bv, bc, data, W);
  38. }, __doc_igl_bbw_bbw,
  39. py::arg("V"), py::arg("Ele"), py::arg("b"), py::arg("bc"), py::arg("data"), py::arg("W"));