Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1 | /* |
| 2 | example/eigen.cpp -- automatic conversion of Eigen types |
| 3 | |
| 4 | Copyright (c) 2016 Wenzel Jakob <wenzel.jakob@epfl.ch> |
| 5 | |
| 6 | All rights reserved. Use of this source code is governed by a |
| 7 | BSD-style license that can be found in the LICENSE file. |
| 8 | */ |
| 9 | |
| 10 | #include "example.h" |
| 11 | #include <pybind11/eigen.h> |
Jason Rhinelander | 5fd5074 | 2016-08-03 16:50:22 -0400 | [diff] [blame^] | 12 | #include <Eigen/Cholesky> |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 13 | |
Ben North | b063e64 | 2016-07-05 20:01:11 +0100 | [diff] [blame] | 14 | Eigen::VectorXf double_col(const Eigen::VectorXf& x) |
| 15 | { return 2.0f * x; } |
| 16 | |
| 17 | Eigen::RowVectorXf double_row(const Eigen::RowVectorXf& x) |
| 18 | { return 2.0f * x; } |
| 19 | |
Ben North | 3e0e779 | 2016-07-05 21:00:05 +0100 | [diff] [blame] | 20 | Eigen::MatrixXf double_mat_cm(const Eigen::MatrixXf& x) |
| 21 | { return 2.0f * x; } |
| 22 | |
Jason Rhinelander | 5fd5074 | 2016-08-03 16:50:22 -0400 | [diff] [blame^] | 23 | // Different ways of passing via Eigen::Ref; the first and second are the Eigen-recommended |
| 24 | Eigen::MatrixXd cholesky1(Eigen::Ref<Eigen::MatrixXd> &x) { return x.llt().matrixL(); } |
| 25 | Eigen::MatrixXd cholesky2(const Eigen::Ref<const Eigen::MatrixXd> &x) { return x.llt().matrixL(); } |
| 26 | Eigen::MatrixXd cholesky3(const Eigen::Ref<Eigen::MatrixXd> &x) { return x.llt().matrixL(); } |
| 27 | Eigen::MatrixXd cholesky4(Eigen::Ref<const Eigen::MatrixXd> &x) { return x.llt().matrixL(); } |
| 28 | Eigen::MatrixXd cholesky5(Eigen::Ref<Eigen::MatrixXd> x) { return x.llt().matrixL(); } |
| 29 | Eigen::MatrixXd cholesky6(Eigen::Ref<const Eigen::MatrixXd> x) { return x.llt().matrixL(); } |
| 30 | |
Ben North | 3e0e779 | 2016-07-05 21:00:05 +0100 | [diff] [blame] | 31 | typedef Eigen::Matrix<float, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> MatrixXfRowMajor; |
| 32 | MatrixXfRowMajor double_mat_rm(const MatrixXfRowMajor& x) |
| 33 | { return 2.0f * x; } |
| 34 | |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 35 | void init_eigen(py::module &m) { |
| 36 | typedef Eigen::Matrix<float, 5, 6, Eigen::RowMajor> FixedMatrixR; |
| 37 | typedef Eigen::Matrix<float, 5, 6> FixedMatrixC; |
| 38 | typedef Eigen::Matrix<float, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> DenseMatrixR; |
| 39 | typedef Eigen::Matrix<float, Eigen::Dynamic, Eigen::Dynamic> DenseMatrixC; |
| 40 | typedef Eigen::SparseMatrix<float, Eigen::RowMajor> SparseMatrixR; |
| 41 | typedef Eigen::SparseMatrix<float> SparseMatrixC; |
| 42 | |
| 43 | // Non-symmetric matrix with zero elements |
| 44 | Eigen::MatrixXf mat(5, 6); |
| 45 | mat << 0, 3, 0, 0, 0, 11, 22, 0, 0, 0, 17, 11, 7, 5, 0, 1, 0, 11, 0, |
| 46 | 0, 0, 0, 0, 11, 0, 0, 14, 0, 8, 11; |
| 47 | |
Ben North | b063e64 | 2016-07-05 20:01:11 +0100 | [diff] [blame] | 48 | m.def("double_col", &double_col); |
| 49 | m.def("double_row", &double_row); |
Ben North | 3e0e779 | 2016-07-05 21:00:05 +0100 | [diff] [blame] | 50 | m.def("double_mat_cm", &double_mat_cm); |
| 51 | m.def("double_mat_rm", &double_mat_rm); |
Jason Rhinelander | 5fd5074 | 2016-08-03 16:50:22 -0400 | [diff] [blame^] | 52 | m.def("cholesky1", &cholesky1); |
| 53 | m.def("cholesky2", &cholesky2); |
| 54 | m.def("cholesky3", &cholesky3); |
| 55 | m.def("cholesky4", &cholesky4); |
| 56 | m.def("cholesky5", &cholesky5); |
| 57 | m.def("cholesky6", &cholesky6); |
Ben North | b063e64 | 2016-07-05 20:01:11 +0100 | [diff] [blame] | 58 | |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 59 | m.def("fixed_r", [mat]() -> FixedMatrixR { |
| 60 | return FixedMatrixR(mat); |
| 61 | }); |
| 62 | |
| 63 | m.def("fixed_c", [mat]() -> FixedMatrixC { |
| 64 | return FixedMatrixC(mat); |
| 65 | }); |
| 66 | |
| 67 | m.def("fixed_passthrough_r", [](const FixedMatrixR &m) -> FixedMatrixR { |
| 68 | return m; |
| 69 | }); |
| 70 | |
| 71 | m.def("fixed_passthrough_c", [](const FixedMatrixC &m) -> FixedMatrixC { |
| 72 | return m; |
| 73 | }); |
| 74 | |
| 75 | m.def("dense_r", [mat]() -> DenseMatrixR { |
| 76 | return DenseMatrixR(mat); |
| 77 | }); |
| 78 | |
| 79 | m.def("dense_c", [mat]() -> DenseMatrixC { |
| 80 | return DenseMatrixC(mat); |
| 81 | }); |
| 82 | |
| 83 | m.def("dense_passthrough_r", [](const DenseMatrixR &m) -> DenseMatrixR { |
| 84 | return m; |
| 85 | }); |
| 86 | |
| 87 | m.def("dense_passthrough_c", [](const DenseMatrixC &m) -> DenseMatrixC { |
| 88 | return m; |
| 89 | }); |
| 90 | |
| 91 | m.def("sparse_r", [mat]() -> SparseMatrixR { |
| 92 | return Eigen::SparseView<Eigen::MatrixXf>(mat); |
| 93 | }); |
| 94 | |
| 95 | m.def("sparse_c", [mat]() -> SparseMatrixC { |
| 96 | return Eigen::SparseView<Eigen::MatrixXf>(mat); |
| 97 | }); |
| 98 | |
| 99 | m.def("sparse_passthrough_r", [](const SparseMatrixR &m) -> SparseMatrixR { |
| 100 | return m; |
| 101 | }); |
| 102 | |
| 103 | m.def("sparse_passthrough_c", [](const SparseMatrixC &m) -> SparseMatrixC { |
| 104 | return m; |
| 105 | }); |
| 106 | } |