test_eigen.cpp revision 14299
111986Sandreas.sandberg@arm.com/* 211986Sandreas.sandberg@arm.com tests/eigen.cpp -- automatic conversion of Eigen types 311986Sandreas.sandberg@arm.com 411986Sandreas.sandberg@arm.com Copyright (c) 2016 Wenzel Jakob <wenzel.jakob@epfl.ch> 511986Sandreas.sandberg@arm.com 611986Sandreas.sandberg@arm.com All rights reserved. Use of this source code is governed by a 711986Sandreas.sandberg@arm.com BSD-style license that can be found in the LICENSE file. 811986Sandreas.sandberg@arm.com*/ 911986Sandreas.sandberg@arm.com 1011986Sandreas.sandberg@arm.com#include "pybind11_tests.h" 1112037Sandreas.sandberg@arm.com#include "constructor_stats.h" 1211986Sandreas.sandberg@arm.com#include <pybind11/eigen.h> 1312391Sjason@lowepower.com#include <pybind11/stl.h> 1414299Sbbruce@ucdavis.edu 1514299Sbbruce@ucdavis.edu#if defined(_MSC_VER) 1614299Sbbruce@ucdavis.edu# pragma warning(disable: 4996) // C4996: std::unary_negation is deprecated 1714299Sbbruce@ucdavis.edu#endif 1814299Sbbruce@ucdavis.edu 1911986Sandreas.sandberg@arm.com#include <Eigen/Cholesky> 2011986Sandreas.sandberg@arm.com 2112037Sandreas.sandberg@arm.comusing MatrixXdR = Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor>; 2211986Sandreas.sandberg@arm.com 2311986Sandreas.sandberg@arm.com 2411986Sandreas.sandberg@arm.com 2512037Sandreas.sandberg@arm.com// Sets/resets a testing reference matrix to have values of 10*r + c, where r and c are the 2612037Sandreas.sandberg@arm.com// (1-based) row/column number. 2712037Sandreas.sandberg@arm.comtemplate <typename M> void reset_ref(M &x) { 2812037Sandreas.sandberg@arm.com for (int i = 0; i < x.rows(); i++) for (int j = 0; j < x.cols(); j++) 2912037Sandreas.sandberg@arm.com x(i, j) = 11 + 10*i + j; 3012037Sandreas.sandberg@arm.com} 3111986Sandreas.sandberg@arm.com 3212037Sandreas.sandberg@arm.com// Returns a static, column-major matrix 3312037Sandreas.sandberg@arm.comEigen::MatrixXd &get_cm() { 3412037Sandreas.sandberg@arm.com static Eigen::MatrixXd *x; 3512037Sandreas.sandberg@arm.com if (!x) { 3612037Sandreas.sandberg@arm.com x = new Eigen::MatrixXd(3, 3); 3712037Sandreas.sandberg@arm.com reset_ref(*x); 3812037Sandreas.sandberg@arm.com } 3912037Sandreas.sandberg@arm.com return *x; 4012037Sandreas.sandberg@arm.com} 4112037Sandreas.sandberg@arm.com// Likewise, but row-major 4212037Sandreas.sandberg@arm.comMatrixXdR &get_rm() { 4312037Sandreas.sandberg@arm.com static MatrixXdR *x; 4412037Sandreas.sandberg@arm.com if (!x) { 4512037Sandreas.sandberg@arm.com x = new MatrixXdR(3, 3); 4612037Sandreas.sandberg@arm.com reset_ref(*x); 4712037Sandreas.sandberg@arm.com } 4812037Sandreas.sandberg@arm.com return *x; 4912037Sandreas.sandberg@arm.com} 5012037Sandreas.sandberg@arm.com// Resets the values of the static matrices returned by get_cm()/get_rm() 5112037Sandreas.sandberg@arm.comvoid reset_refs() { 5212037Sandreas.sandberg@arm.com reset_ref(get_cm()); 5312037Sandreas.sandberg@arm.com reset_ref(get_rm()); 5412037Sandreas.sandberg@arm.com} 5512037Sandreas.sandberg@arm.com 5612037Sandreas.sandberg@arm.com// Returns element 2,1 from a matrix (used to test copy/nocopy) 5712037Sandreas.sandberg@arm.comdouble get_elem(Eigen::Ref<const Eigen::MatrixXd> m) { return m(2, 1); }; 5812037Sandreas.sandberg@arm.com 5912037Sandreas.sandberg@arm.com 6012037Sandreas.sandberg@arm.com// Returns a matrix with 10*r + 100*c added to each matrix element (to help test that the matrix 6112037Sandreas.sandberg@arm.com// reference is referencing rows/columns correctly). 6212037Sandreas.sandberg@arm.comtemplate <typename MatrixArgType> Eigen::MatrixXd adjust_matrix(MatrixArgType m) { 6312037Sandreas.sandberg@arm.com Eigen::MatrixXd ret(m); 6412037Sandreas.sandberg@arm.com for (int c = 0; c < m.cols(); c++) for (int r = 0; r < m.rows(); r++) 6512037Sandreas.sandberg@arm.com ret(r, c) += 10*r + 100*c; 6612037Sandreas.sandberg@arm.com return ret; 6712037Sandreas.sandberg@arm.com} 6812037Sandreas.sandberg@arm.com 6912037Sandreas.sandberg@arm.comstruct CustomOperatorNew { 7012037Sandreas.sandberg@arm.com CustomOperatorNew() = default; 7112037Sandreas.sandberg@arm.com 7212037Sandreas.sandberg@arm.com Eigen::Matrix4d a = Eigen::Matrix4d::Zero(); 7312037Sandreas.sandberg@arm.com Eigen::Matrix4d b = Eigen::Matrix4d::Identity(); 7412037Sandreas.sandberg@arm.com 7512037Sandreas.sandberg@arm.com EIGEN_MAKE_ALIGNED_OPERATOR_NEW; 7612037Sandreas.sandberg@arm.com}; 7711986Sandreas.sandberg@arm.com 7812391Sjason@lowepower.comTEST_SUBMODULE(eigen, m) { 7912391Sjason@lowepower.com using FixedMatrixR = Eigen::Matrix<float, 5, 6, Eigen::RowMajor>; 8012391Sjason@lowepower.com using FixedMatrixC = Eigen::Matrix<float, 5, 6>; 8112391Sjason@lowepower.com using DenseMatrixR = Eigen::Matrix<float, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor>; 8212391Sjason@lowepower.com using DenseMatrixC = Eigen::Matrix<float, Eigen::Dynamic, Eigen::Dynamic>; 8312391Sjason@lowepower.com using FourRowMatrixC = Eigen::Matrix<float, 4, Eigen::Dynamic>; 8412391Sjason@lowepower.com using FourColMatrixC = Eigen::Matrix<float, Eigen::Dynamic, 4>; 8512391Sjason@lowepower.com using FourRowMatrixR = Eigen::Matrix<float, 4, Eigen::Dynamic>; 8612391Sjason@lowepower.com using FourColMatrixR = Eigen::Matrix<float, Eigen::Dynamic, 4>; 8712391Sjason@lowepower.com using SparseMatrixR = Eigen::SparseMatrix<float, Eigen::RowMajor>; 8812391Sjason@lowepower.com using SparseMatrixC = Eigen::SparseMatrix<float>; 8911986Sandreas.sandberg@arm.com 9011986Sandreas.sandberg@arm.com m.attr("have_eigen") = true; 9111986Sandreas.sandberg@arm.com 9212391Sjason@lowepower.com // various tests 9312037Sandreas.sandberg@arm.com m.def("double_col", [](const Eigen::VectorXf &x) -> Eigen::VectorXf { return 2.0f * x; }); 9412037Sandreas.sandberg@arm.com m.def("double_row", [](const Eigen::RowVectorXf &x) -> Eigen::RowVectorXf { return 2.0f * x; }); 9512037Sandreas.sandberg@arm.com m.def("double_complex", [](const Eigen::VectorXcf &x) -> Eigen::VectorXcf { return 2.0f * x; }); 9612037Sandreas.sandberg@arm.com m.def("double_threec", [](py::EigenDRef<Eigen::Vector3f> x) { x *= 2; }); 9712037Sandreas.sandberg@arm.com m.def("double_threer", [](py::EigenDRef<Eigen::RowVector3f> x) { x *= 2; }); 9812037Sandreas.sandberg@arm.com m.def("double_mat_cm", [](Eigen::MatrixXf x) -> Eigen::MatrixXf { return 2.0f * x; }); 9912037Sandreas.sandberg@arm.com m.def("double_mat_rm", [](DenseMatrixR x) -> DenseMatrixR { return 2.0f * x; }); 10011986Sandreas.sandberg@arm.com 10112391Sjason@lowepower.com // test_eigen_ref_to_python 10212037Sandreas.sandberg@arm.com // Different ways of passing via Eigen::Ref; the first and second are the Eigen-recommended 10312037Sandreas.sandberg@arm.com m.def("cholesky1", [](Eigen::Ref<MatrixXdR> x) -> Eigen::MatrixXd { return x.llt().matrixL(); }); 10412037Sandreas.sandberg@arm.com m.def("cholesky2", [](const Eigen::Ref<const MatrixXdR> &x) -> Eigen::MatrixXd { return x.llt().matrixL(); }); 10512037Sandreas.sandberg@arm.com m.def("cholesky3", [](const Eigen::Ref<MatrixXdR> &x) -> Eigen::MatrixXd { return x.llt().matrixL(); }); 10612037Sandreas.sandberg@arm.com m.def("cholesky4", [](Eigen::Ref<const MatrixXdR> x) -> Eigen::MatrixXd { return x.llt().matrixL(); }); 10712037Sandreas.sandberg@arm.com 10812391Sjason@lowepower.com // test_eigen_ref_mutators 10912037Sandreas.sandberg@arm.com // Mutators: these add some value to the given element using Eigen, but Eigen should be mapping into 11012037Sandreas.sandberg@arm.com // the numpy array data and so the result should show up there. There are three versions: one that 11112037Sandreas.sandberg@arm.com // works on a contiguous-row matrix (numpy's default), one for a contiguous-column matrix, and one 11212037Sandreas.sandberg@arm.com // for any matrix. 11312037Sandreas.sandberg@arm.com auto add_rm = [](Eigen::Ref<MatrixXdR> x, int r, int c, double v) { x(r,c) += v; }; 11412037Sandreas.sandberg@arm.com auto add_cm = [](Eigen::Ref<Eigen::MatrixXd> x, int r, int c, double v) { x(r,c) += v; }; 11512037Sandreas.sandberg@arm.com 11612037Sandreas.sandberg@arm.com // Mutators (Eigen maps into numpy variables): 11712037Sandreas.sandberg@arm.com m.def("add_rm", add_rm); // Only takes row-contiguous 11812037Sandreas.sandberg@arm.com m.def("add_cm", add_cm); // Only takes column-contiguous 11912037Sandreas.sandberg@arm.com // Overloaded versions that will accept either row or column contiguous: 12012037Sandreas.sandberg@arm.com m.def("add1", add_rm); 12112037Sandreas.sandberg@arm.com m.def("add1", add_cm); 12212037Sandreas.sandberg@arm.com m.def("add2", add_cm); 12312037Sandreas.sandberg@arm.com m.def("add2", add_rm); 12412037Sandreas.sandberg@arm.com // This one accepts a matrix of any stride: 12512037Sandreas.sandberg@arm.com m.def("add_any", [](py::EigenDRef<Eigen::MatrixXd> x, int r, int c, double v) { x(r,c) += v; }); 12612037Sandreas.sandberg@arm.com 12714299Sbbruce@ucdavis.edu // Return mutable references (numpy maps into eigen variables) 12812037Sandreas.sandberg@arm.com m.def("get_cm_ref", []() { return Eigen::Ref<Eigen::MatrixXd>(get_cm()); }); 12912037Sandreas.sandberg@arm.com m.def("get_rm_ref", []() { return Eigen::Ref<MatrixXdR>(get_rm()); }); 13012037Sandreas.sandberg@arm.com // The same references, but non-mutable (numpy maps into eigen variables, but is !writeable) 13112037Sandreas.sandberg@arm.com m.def("get_cm_const_ref", []() { return Eigen::Ref<const Eigen::MatrixXd>(get_cm()); }); 13212037Sandreas.sandberg@arm.com m.def("get_rm_const_ref", []() { return Eigen::Ref<const MatrixXdR>(get_rm()); }); 13312037Sandreas.sandberg@arm.com 13412037Sandreas.sandberg@arm.com m.def("reset_refs", reset_refs); // Restores get_{cm,rm}_ref to original values 13512037Sandreas.sandberg@arm.com 13612037Sandreas.sandberg@arm.com // Increments and returns ref to (same) matrix 13712037Sandreas.sandberg@arm.com m.def("incr_matrix", [](Eigen::Ref<Eigen::MatrixXd> m, double v) { 13812037Sandreas.sandberg@arm.com m += Eigen::MatrixXd::Constant(m.rows(), m.cols(), v); 13912037Sandreas.sandberg@arm.com return m; 14012037Sandreas.sandberg@arm.com }, py::return_value_policy::reference); 14112037Sandreas.sandberg@arm.com 14212037Sandreas.sandberg@arm.com // Same, but accepts a matrix of any strides 14312037Sandreas.sandberg@arm.com m.def("incr_matrix_any", [](py::EigenDRef<Eigen::MatrixXd> m, double v) { 14412037Sandreas.sandberg@arm.com m += Eigen::MatrixXd::Constant(m.rows(), m.cols(), v); 14512037Sandreas.sandberg@arm.com return m; 14612037Sandreas.sandberg@arm.com }, py::return_value_policy::reference); 14712037Sandreas.sandberg@arm.com 14812037Sandreas.sandberg@arm.com // Returns an eigen slice of even rows 14912037Sandreas.sandberg@arm.com m.def("even_rows", [](py::EigenDRef<Eigen::MatrixXd> m) { 15012037Sandreas.sandberg@arm.com return py::EigenDMap<Eigen::MatrixXd>( 15112037Sandreas.sandberg@arm.com m.data(), (m.rows() + 1) / 2, m.cols(), 15212037Sandreas.sandberg@arm.com py::EigenDStride(m.outerStride(), 2 * m.innerStride())); 15312037Sandreas.sandberg@arm.com }, py::return_value_policy::reference); 15412037Sandreas.sandberg@arm.com 15512037Sandreas.sandberg@arm.com // Returns an eigen slice of even columns 15612037Sandreas.sandberg@arm.com m.def("even_cols", [](py::EigenDRef<Eigen::MatrixXd> m) { 15712037Sandreas.sandberg@arm.com return py::EigenDMap<Eigen::MatrixXd>( 15812037Sandreas.sandberg@arm.com m.data(), m.rows(), (m.cols() + 1) / 2, 15912037Sandreas.sandberg@arm.com py::EigenDStride(2 * m.outerStride(), m.innerStride())); 16012037Sandreas.sandberg@arm.com }, py::return_value_policy::reference); 16111986Sandreas.sandberg@arm.com 16211986Sandreas.sandberg@arm.com // Returns diagonals: a vector-like object with an inner stride != 1 16311986Sandreas.sandberg@arm.com m.def("diagonal", [](const Eigen::Ref<const Eigen::MatrixXd> &x) { return x.diagonal(); }); 16411986Sandreas.sandberg@arm.com m.def("diagonal_1", [](const Eigen::Ref<const Eigen::MatrixXd> &x) { return x.diagonal<1>(); }); 16511986Sandreas.sandberg@arm.com m.def("diagonal_n", [](const Eigen::Ref<const Eigen::MatrixXd> &x, int index) { return x.diagonal(index); }); 16611986Sandreas.sandberg@arm.com 16711986Sandreas.sandberg@arm.com // Return a block of a matrix (gives non-standard strides) 16811986Sandreas.sandberg@arm.com m.def("block", [](const Eigen::Ref<const Eigen::MatrixXd> &x, int start_row, int start_col, int block_rows, int block_cols) { 16911986Sandreas.sandberg@arm.com return x.block(start_row, start_col, block_rows, block_cols); 17011986Sandreas.sandberg@arm.com }); 17111986Sandreas.sandberg@arm.com 17212391Sjason@lowepower.com // test_eigen_return_references, test_eigen_keepalive 17312037Sandreas.sandberg@arm.com // return value referencing/copying tests: 17412037Sandreas.sandberg@arm.com class ReturnTester { 17512037Sandreas.sandberg@arm.com Eigen::MatrixXd mat = create(); 17612037Sandreas.sandberg@arm.com public: 17712037Sandreas.sandberg@arm.com ReturnTester() { print_created(this); } 17812037Sandreas.sandberg@arm.com ~ReturnTester() { print_destroyed(this); } 17912037Sandreas.sandberg@arm.com static Eigen::MatrixXd create() { return Eigen::MatrixXd::Ones(10, 10); } 18012037Sandreas.sandberg@arm.com static const Eigen::MatrixXd createConst() { return Eigen::MatrixXd::Ones(10, 10); } 18112037Sandreas.sandberg@arm.com Eigen::MatrixXd &get() { return mat; } 18212037Sandreas.sandberg@arm.com Eigen::MatrixXd *getPtr() { return &mat; } 18312037Sandreas.sandberg@arm.com const Eigen::MatrixXd &view() { return mat; } 18412037Sandreas.sandberg@arm.com const Eigen::MatrixXd *viewPtr() { return &mat; } 18512037Sandreas.sandberg@arm.com Eigen::Ref<Eigen::MatrixXd> ref() { return mat; } 18612037Sandreas.sandberg@arm.com Eigen::Ref<const Eigen::MatrixXd> refConst() { return mat; } 18712037Sandreas.sandberg@arm.com Eigen::Block<Eigen::MatrixXd> block(int r, int c, int nrow, int ncol) { return mat.block(r, c, nrow, ncol); } 18812037Sandreas.sandberg@arm.com Eigen::Block<const Eigen::MatrixXd> blockConst(int r, int c, int nrow, int ncol) const { return mat.block(r, c, nrow, ncol); } 18912037Sandreas.sandberg@arm.com py::EigenDMap<Eigen::Matrix2d> corners() { return py::EigenDMap<Eigen::Matrix2d>(mat.data(), 19012037Sandreas.sandberg@arm.com py::EigenDStride(mat.outerStride() * (mat.outerSize()-1), mat.innerStride() * (mat.innerSize()-1))); } 19112037Sandreas.sandberg@arm.com py::EigenDMap<const Eigen::Matrix2d> cornersConst() const { return py::EigenDMap<const Eigen::Matrix2d>(mat.data(), 19212037Sandreas.sandberg@arm.com py::EigenDStride(mat.outerStride() * (mat.outerSize()-1), mat.innerStride() * (mat.innerSize()-1))); } 19312037Sandreas.sandberg@arm.com }; 19412037Sandreas.sandberg@arm.com using rvp = py::return_value_policy; 19512037Sandreas.sandberg@arm.com py::class_<ReturnTester>(m, "ReturnTester") 19612037Sandreas.sandberg@arm.com .def(py::init<>()) 19712037Sandreas.sandberg@arm.com .def_static("create", &ReturnTester::create) 19812037Sandreas.sandberg@arm.com .def_static("create_const", &ReturnTester::createConst) 19912037Sandreas.sandberg@arm.com .def("get", &ReturnTester::get, rvp::reference_internal) 20012037Sandreas.sandberg@arm.com .def("get_ptr", &ReturnTester::getPtr, rvp::reference_internal) 20112037Sandreas.sandberg@arm.com .def("view", &ReturnTester::view, rvp::reference_internal) 20212037Sandreas.sandberg@arm.com .def("view_ptr", &ReturnTester::view, rvp::reference_internal) 20312037Sandreas.sandberg@arm.com .def("copy_get", &ReturnTester::get) // Default rvp: copy 20412037Sandreas.sandberg@arm.com .def("copy_view", &ReturnTester::view) // " 20512037Sandreas.sandberg@arm.com .def("ref", &ReturnTester::ref) // Default for Ref is to reference 20612037Sandreas.sandberg@arm.com .def("ref_const", &ReturnTester::refConst) // Likewise, but const 20712037Sandreas.sandberg@arm.com .def("ref_safe", &ReturnTester::ref, rvp::reference_internal) 20812037Sandreas.sandberg@arm.com .def("ref_const_safe", &ReturnTester::refConst, rvp::reference_internal) 20912037Sandreas.sandberg@arm.com .def("copy_ref", &ReturnTester::ref, rvp::copy) 21012037Sandreas.sandberg@arm.com .def("copy_ref_const", &ReturnTester::refConst, rvp::copy) 21112037Sandreas.sandberg@arm.com .def("block", &ReturnTester::block) 21212037Sandreas.sandberg@arm.com .def("block_safe", &ReturnTester::block, rvp::reference_internal) 21312037Sandreas.sandberg@arm.com .def("block_const", &ReturnTester::blockConst, rvp::reference_internal) 21412037Sandreas.sandberg@arm.com .def("copy_block", &ReturnTester::block, rvp::copy) 21512037Sandreas.sandberg@arm.com .def("corners", &ReturnTester::corners, rvp::reference_internal) 21612037Sandreas.sandberg@arm.com .def("corners_const", &ReturnTester::cornersConst, rvp::reference_internal) 21712037Sandreas.sandberg@arm.com ; 21812037Sandreas.sandberg@arm.com 21912391Sjason@lowepower.com // test_special_matrix_objects 22011986Sandreas.sandberg@arm.com // Returns a DiagonalMatrix with diagonal (1,2,3,...) 22111986Sandreas.sandberg@arm.com m.def("incr_diag", [](int k) { 22211986Sandreas.sandberg@arm.com Eigen::DiagonalMatrix<int, Eigen::Dynamic> m(k); 22311986Sandreas.sandberg@arm.com for (int i = 0; i < k; i++) m.diagonal()[i] = i+1; 22411986Sandreas.sandberg@arm.com return m; 22511986Sandreas.sandberg@arm.com }); 22611986Sandreas.sandberg@arm.com 22711986Sandreas.sandberg@arm.com // Returns a SelfAdjointView referencing the lower triangle of m 22811986Sandreas.sandberg@arm.com m.def("symmetric_lower", [](const Eigen::MatrixXi &m) { 22911986Sandreas.sandberg@arm.com return m.selfadjointView<Eigen::Lower>(); 23011986Sandreas.sandberg@arm.com }); 23111986Sandreas.sandberg@arm.com // Returns a SelfAdjointView referencing the lower triangle of m 23211986Sandreas.sandberg@arm.com m.def("symmetric_upper", [](const Eigen::MatrixXi &m) { 23311986Sandreas.sandberg@arm.com return m.selfadjointView<Eigen::Upper>(); 23411986Sandreas.sandberg@arm.com }); 23511986Sandreas.sandberg@arm.com 23612037Sandreas.sandberg@arm.com // Test matrix for various functions below. 23712037Sandreas.sandberg@arm.com Eigen::MatrixXf mat(5, 6); 23812037Sandreas.sandberg@arm.com mat << 0, 3, 0, 0, 0, 11, 23912037Sandreas.sandberg@arm.com 22, 0, 0, 0, 17, 11, 24012037Sandreas.sandberg@arm.com 7, 5, 0, 1, 0, 11, 24112037Sandreas.sandberg@arm.com 0, 0, 0, 0, 0, 11, 24212037Sandreas.sandberg@arm.com 0, 0, 14, 0, 8, 11; 24311986Sandreas.sandberg@arm.com 24412391Sjason@lowepower.com // test_fixed, and various other tests 24512037Sandreas.sandberg@arm.com m.def("fixed_r", [mat]() -> FixedMatrixR { return FixedMatrixR(mat); }); 24612037Sandreas.sandberg@arm.com m.def("fixed_r_const", [mat]() -> const FixedMatrixR { return FixedMatrixR(mat); }); 24712037Sandreas.sandberg@arm.com m.def("fixed_c", [mat]() -> FixedMatrixC { return FixedMatrixC(mat); }); 24812037Sandreas.sandberg@arm.com m.def("fixed_copy_r", [](const FixedMatrixR &m) -> FixedMatrixR { return m; }); 24912037Sandreas.sandberg@arm.com m.def("fixed_copy_c", [](const FixedMatrixC &m) -> FixedMatrixC { return m; }); 25012391Sjason@lowepower.com // test_mutator_descriptors 25112037Sandreas.sandberg@arm.com m.def("fixed_mutator_r", [](Eigen::Ref<FixedMatrixR>) {}); 25212037Sandreas.sandberg@arm.com m.def("fixed_mutator_c", [](Eigen::Ref<FixedMatrixC>) {}); 25312037Sandreas.sandberg@arm.com m.def("fixed_mutator_a", [](py::EigenDRef<FixedMatrixC>) {}); 25412391Sjason@lowepower.com // test_dense 25512037Sandreas.sandberg@arm.com m.def("dense_r", [mat]() -> DenseMatrixR { return DenseMatrixR(mat); }); 25612037Sandreas.sandberg@arm.com m.def("dense_c", [mat]() -> DenseMatrixC { return DenseMatrixC(mat); }); 25712037Sandreas.sandberg@arm.com m.def("dense_copy_r", [](const DenseMatrixR &m) -> DenseMatrixR { return m; }); 25812037Sandreas.sandberg@arm.com m.def("dense_copy_c", [](const DenseMatrixC &m) -> DenseMatrixC { return m; }); 25912391Sjason@lowepower.com // test_sparse, test_sparse_signature 26012037Sandreas.sandberg@arm.com m.def("sparse_r", [mat]() -> SparseMatrixR { return Eigen::SparseView<Eigen::MatrixXf>(mat); }); 26112037Sandreas.sandberg@arm.com m.def("sparse_c", [mat]() -> SparseMatrixC { return Eigen::SparseView<Eigen::MatrixXf>(mat); }); 26212037Sandreas.sandberg@arm.com m.def("sparse_copy_r", [](const SparseMatrixR &m) -> SparseMatrixR { return m; }); 26312037Sandreas.sandberg@arm.com m.def("sparse_copy_c", [](const SparseMatrixC &m) -> SparseMatrixC { return m; }); 26412391Sjason@lowepower.com // test_partially_fixed 26512037Sandreas.sandberg@arm.com m.def("partial_copy_four_rm_r", [](const FourRowMatrixR &m) -> FourRowMatrixR { return m; }); 26612037Sandreas.sandberg@arm.com m.def("partial_copy_four_rm_c", [](const FourColMatrixR &m) -> FourColMatrixR { return m; }); 26712037Sandreas.sandberg@arm.com m.def("partial_copy_four_cm_r", [](const FourRowMatrixC &m) -> FourRowMatrixC { return m; }); 26812037Sandreas.sandberg@arm.com m.def("partial_copy_four_cm_c", [](const FourColMatrixC &m) -> FourColMatrixC { return m; }); 26911986Sandreas.sandberg@arm.com 27012391Sjason@lowepower.com // test_cpp_casting 27112037Sandreas.sandberg@arm.com // Test that we can cast a numpy object to a Eigen::MatrixXd explicitly 27212037Sandreas.sandberg@arm.com m.def("cpp_copy", [](py::handle m) { return m.cast<Eigen::MatrixXd>()(1, 0); }); 27312037Sandreas.sandberg@arm.com m.def("cpp_ref_c", [](py::handle m) { return m.cast<Eigen::Ref<Eigen::MatrixXd>>()(1, 0); }); 27412037Sandreas.sandberg@arm.com m.def("cpp_ref_r", [](py::handle m) { return m.cast<Eigen::Ref<MatrixXdR>>()(1, 0); }); 27512037Sandreas.sandberg@arm.com m.def("cpp_ref_any", [](py::handle m) { return m.cast<py::EigenDRef<Eigen::MatrixXd>>()(1, 0); }); 27611986Sandreas.sandberg@arm.com 27711986Sandreas.sandberg@arm.com 27812391Sjason@lowepower.com // test_nocopy_wrapper 27912037Sandreas.sandberg@arm.com // Test that we can prevent copying into an argument that would normally copy: First a version 28012037Sandreas.sandberg@arm.com // that would allow copying (if types or strides don't match) for comparison: 28112037Sandreas.sandberg@arm.com m.def("get_elem", &get_elem); 28212037Sandreas.sandberg@arm.com // Now this alternative that calls the tells pybind to fail rather than copy: 28312037Sandreas.sandberg@arm.com m.def("get_elem_nocopy", [](Eigen::Ref<const Eigen::MatrixXd> m) -> double { return get_elem(m); }, 28412037Sandreas.sandberg@arm.com py::arg().noconvert()); 28512037Sandreas.sandberg@arm.com // Also test a row-major-only no-copy const ref: 28612037Sandreas.sandberg@arm.com m.def("get_elem_rm_nocopy", [](Eigen::Ref<const Eigen::Matrix<long, -1, -1, Eigen::RowMajor>> &m) -> long { return m(2, 1); }, 28712037Sandreas.sandberg@arm.com py::arg().noconvert()); 28811986Sandreas.sandberg@arm.com 28912391Sjason@lowepower.com // test_issue738 29012037Sandreas.sandberg@arm.com // Issue #738: 1xN or Nx1 2D matrices were neither accepted nor properly copied with an 29112037Sandreas.sandberg@arm.com // incompatible stride value on the length-1 dimension--but that should be allowed (without 29212037Sandreas.sandberg@arm.com // requiring a copy!) because the stride value can be safely ignored on a size-1 dimension. 29312037Sandreas.sandberg@arm.com m.def("iss738_f1", &adjust_matrix<const Eigen::Ref<const Eigen::MatrixXd> &>, py::arg().noconvert()); 29412037Sandreas.sandberg@arm.com m.def("iss738_f2", &adjust_matrix<const Eigen::Ref<const Eigen::Matrix<double, -1, -1, Eigen::RowMajor>> &>, py::arg().noconvert()); 29511986Sandreas.sandberg@arm.com 29614299Sbbruce@ucdavis.edu // test_issue1105 29714299Sbbruce@ucdavis.edu // Issue #1105: when converting from a numpy two-dimensional (Nx1) or (1xN) value into a dense 29814299Sbbruce@ucdavis.edu // eigen Vector or RowVector, the argument would fail to load because the numpy copy would fail: 29914299Sbbruce@ucdavis.edu // numpy won't broadcast a Nx1 into a 1-dimensional vector. 30014299Sbbruce@ucdavis.edu m.def("iss1105_col", [](Eigen::VectorXd) { return true; }); 30114299Sbbruce@ucdavis.edu m.def("iss1105_row", [](Eigen::RowVectorXd) { return true; }); 30214299Sbbruce@ucdavis.edu 30312391Sjason@lowepower.com // test_named_arguments 30412391Sjason@lowepower.com // Make sure named arguments are working properly: 30512391Sjason@lowepower.com m.def("matrix_multiply", [](const py::EigenDRef<const Eigen::MatrixXd> A, const py::EigenDRef<const Eigen::MatrixXd> B) 30612391Sjason@lowepower.com -> Eigen::MatrixXd { 30712391Sjason@lowepower.com if (A.cols() != B.rows()) throw std::domain_error("Nonconformable matrices!"); 30812391Sjason@lowepower.com return A * B; 30912391Sjason@lowepower.com }, py::arg("A"), py::arg("B")); 31012391Sjason@lowepower.com 31112391Sjason@lowepower.com // test_custom_operator_new 31212037Sandreas.sandberg@arm.com py::class_<CustomOperatorNew>(m, "CustomOperatorNew") 31312037Sandreas.sandberg@arm.com .def(py::init<>()) 31412037Sandreas.sandberg@arm.com .def_readonly("a", &CustomOperatorNew::a) 31512037Sandreas.sandberg@arm.com .def_readonly("b", &CustomOperatorNew::b); 31612391Sjason@lowepower.com 31712391Sjason@lowepower.com // test_eigen_ref_life_support 31812391Sjason@lowepower.com // In case of a failure (the caster's temp array does not live long enough), creating 31912391Sjason@lowepower.com // a new array (np.ones(10)) increases the chances that the temp array will be garbage 32012391Sjason@lowepower.com // collected and/or that its memory will be overridden with different values. 32112391Sjason@lowepower.com m.def("get_elem_direct", [](Eigen::Ref<const Eigen::VectorXd> v) { 32212391Sjason@lowepower.com py::module::import("numpy").attr("ones")(10); 32312391Sjason@lowepower.com return v(5); 32412391Sjason@lowepower.com }); 32512391Sjason@lowepower.com m.def("get_elem_indirect", [](std::vector<Eigen::Ref<const Eigen::VectorXd>> v) { 32612391Sjason@lowepower.com py::module::import("numpy").attr("ones")(10); 32712391Sjason@lowepower.com return v[0](5); 32812391Sjason@lowepower.com }); 32912391Sjason@lowepower.com} 330