pybind11/tests/test_eigen.cpp

57 double get_elem(Eigen::Ref<const Eigen::MatrixXd> m) { return m(2, 1); };
93     m.def("double_col", [](const Eigen::VectorXf &x) -> Eigen::VectorXf { return 2.0f * x; });
94     m.def("double_row", [](const Eigen::RowVectorXf &x) -> Eigen::RowVectorXf { return 2.0f * x; });
95     m.def("double_complex", [](const Eigen::VectorXcf &x) -> Eigen::VectorXcf { return 2.0f * x; });
104     m.def("cholesky2", [](const Eigen::Ref<const MatrixXdR> &x) -> Eigen::MatrixXd { return x.llt().matrixL(); });
105     m.def("cholesky3", [](const Eigen::Ref<MatrixXdR> &x) -> Eigen::MatrixXd { return x.llt().matrixL(); });
106     m.def("cholesky4", [](Eigen::Ref<const MatrixXdR> x) -> Eigen::MatrixXd { return x.llt().matrixL(); });
131     m.def("get_cm_const_ref", []() { return Eigen::Ref<const Eigen::MatrixXd>(get_cm()); });
132     m.def("get_rm_const_ref", []() { return Eigen::Ref<const MatrixXdR>(get_rm()); });
163     m.def("diagonal", [](const Eigen::Ref<const Eigen::MatrixXd> &x) { return x.diagonal(); });
164     m.def("diagonal_1", [](const Eigen::Ref<const Eigen::MatrixXd> &x) { return x.diagonal<1>(); });
165     m.def("diagonal_n", [](const Eigen::Ref<const Eigen::MatrixXd> &x, int index) { return x.diagonal(index); });
168     m.def("block", [](const Eigen::Ref<const Eigen::MatrixXd> &x, int start_row, int start_col, int block_rows, int block_cols) {
180         static const Eigen::MatrixXd createConst() { return Eigen::MatrixXd::Ones(10, 10); }
183         const Eigen::MatrixXd &view() { return mat; }
184         const Eigen::MatrixXd *viewPtr() { return &mat; }
186         Eigen::Ref<const Eigen::MatrixXd> refConst() { return mat; }
188         Eigen::Block<const Eigen::MatrixXd> blockConst(int r, int c, int nrow, int ncol) const { return mat.block(r, c, nrow, ncol); }
191         py::EigenDMap<const Eigen::Matrix2d> cornersConst() const { return py::EigenDMap<const Eigen::Matrix2d>(mat.data(),
206         .def("ref_const", &ReturnTester::refConst) // Likewise, but const
228     m.def("symmetric_lower", [](const Eigen::MatrixXi &m) {
232     m.def("symmetric_upper", [](const Eigen::MatrixXi &m) {
246     m.def("fixed_r_const", [mat]() -> const FixedMatrixR { return FixedMatrixR(mat); });
248     m.def("fixed_copy_r", [](const FixedMatrixR &m) -> FixedMatrixR { return m; });
249     m.def("fixed_copy_c", [](const FixedMatrixC &m) -> FixedMatrixC { return m; });
257     m.def("dense_copy_r", [](const DenseMatrixR &m) -> DenseMatrixR { return m; });
258     m.def("dense_copy_c", [](const DenseMatrixC &m) -> DenseMatrixC { return m; });
262     m.def("sparse_copy_r", [](const SparseMatrixR &m) -> SparseMatrixR { return m; });
263     m.def("sparse_copy_c", [](const SparseMatrixC &m) -> SparseMatrixC { return m; });
265     m.def("partial_copy_four_rm_r", [](const FourRowMatrixR &m) -> FourRowMatrixR { return m; });
266     m.def("partial_copy_four_rm_c", [](const FourColMatrixR &m) -> FourColMatrixR { return m; });
267     m.def("partial_copy_four_cm_r", [](const FourRowMatrixC &m) -> FourRowMatrixC { return m; });
268     m.def("partial_copy_four_cm_c", [](const FourColMatrixC &m) -> FourColMatrixC { return m; });
283     m.def("get_elem_nocopy", [](Eigen::Ref<const Eigen::MatrixXd> m) -> double { return get_elem(m); },
285     // Also test a row-major-only no-copy const ref:
286     m.def("get_elem_rm_nocopy", [](Eigen::Ref<const Eigen::Matrix<long, -1, -1, Eigen::RowMajor>> &m) -> long { return m(2, 1); },
293     m.def("iss738_f1", &adjust_matrix<const Eigen::Ref<const Eigen::MatrixXd> &>, py::arg().noconvert());
294     m.def("iss738_f2", &adjust_matrix<const Eigen::Ref<const Eigen::Matrix<double, -1, -1, Eigen::RowMajor>> &>, py::arg().noconvert());
305     m.def("matrix_multiply", [](const py::EigenDRef<const Eigen::MatrixXd> A, const py::EigenDRef<const Eigen::MatrixXd> B)
321     m.def("get_elem_direct", [](Eigen::Ref<const Eigen::VectorXd> v) {
325     m.def("get_elem_indirect", [](std::vector<Eigen::Ref<const Eigen::VectorXd>> v) {