forked from pybind/python_example
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.cpp
More file actions
112 lines (89 loc) · 2.64 KB
/
main.cpp
File metadata and controls
112 lines (89 loc) · 2.64 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
#include <exception>
#include <numeric>
#include <vector>
#include <pybind11/numpy.h>
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
namespace py = pybind11;
void runtime_error()
{
throw std::runtime_error("C++ failure");
}
void range_error()
{
throw std::range_error("C++ range error");
}
int add(int i, int j)
{
return i + j;
}
double ndarray_sum(py::array_t<double> ndarray)
{
// double ndarray_sum(std::vector<double> &ndarray) {
// return std::accumulate(ndarray.begin(), ndarray.end(), 0.0);
auto buf1 = ndarray.request();
double sum = 0;
int n = buf1.size;
for (int i = 0; i < n; i++)
{
sum += ((double *)buf1.ptr)[i];
}
return sum;
}
py::array_t<double> get_ndarray(int n, double value)
{
auto result = std::vector<double>(n, value);
return py::array(result.size(), result.data());
}
struct VectorResults
{
VectorResults(const std::vector<double>& temperature,
const std::vector<double>& humidity)
: temperature(py::array(temperature.size(), temperature.data())),
humidity(py::array(humidity.size(), humidity.data())){};
const py::array_t<double> temperature;
const py::array_t<double> humidity;
};
VectorResults get_vector_results(int len, double temp_value, double hum_value)
{
auto tmp = std::vector<double>(len, temp_value);
auto hum = std::vector<double>(len, hum_value);
return VectorResults(tmp, hum);
}
PYBIND11_MODULE(python_example, m)
{
m.doc() = R"pbdoc(
Pybind11 example plugin
-----------------------
.. currentmodule:: python_example
.. autosummary::
:toctree: _generate
add
subtract
)pbdoc";
m.def("add", &add, R"pbdoc(
Add two numbers
Some other explanation about the add function.
)pbdoc");
m.def("subtract", [](int i, int j) { return i - j; }, R"pbdoc(
Subtract two numbers
Some other explanation about the subtract function.
)pbdoc");
m.def("ndarray_sum", &ndarray_sum, R"pbdoc(
Return the sum of a numpy ndarray.
Tests functions that accept a numpy array and
autoconvert to std::vector.
)pbdoc");
m.def("runtime_error", &runtime_error);
m.def("range_error", &range_error);
m.def("get_ndarray", &get_ndarray);
py::class_<VectorResults>(m, "VectorResults")
.def_readonly("temperature", &VectorResults::temperature)
.def_readonly("humidity", &VectorResults::humidity);
m.def("get_vector_results", &get_vector_results);
#ifdef VERSION_INFO
m.attr("__version__") = VERSION_INFO;
#else
m.attr("__version__") = "dev";
#endif
}