GeezerVison is showing github.com — python_example/src/main.cpp at master · DC23/python_example

Expand file tree

main.cpp

Copy path

More file actions

Latest commit

History

112 lines (89 loc) · 2.64 KB

main.cpp

Copy path

File metadata and controls

112 lines (89 loc) · 2.64 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

}