mirror of
https://gitee.com/fasiondog/hikyuu.git
synced 2024-12-05 13:28:11 +08:00
41 lines
980 B
C++
41 lines
980 B
C++
/*
|
|
* Copyright (c) hikyuu.org
|
|
*
|
|
* Created on: 2020-6-25
|
|
* Author: fasiondog
|
|
*/
|
|
|
|
#pragma once
|
|
#ifndef HIKYUU_PYBIND_UTILS_H
|
|
#define HIKYUU_PYBIND_UTILS_H
|
|
|
|
#include <vector>
|
|
#include <boost/python.hpp>
|
|
|
|
namespace py = boost::python;
|
|
|
|
template <class T>
|
|
struct vector_to_python_list {
|
|
static PyObject* convert(T const& x) {
|
|
py::list pylist;
|
|
for (auto const& d : x) {
|
|
pylist.append(d);
|
|
}
|
|
return py::incref(pylist.ptr());
|
|
}
|
|
}; // namespace boost::pythontemplate<classT>structvector_to_python_list
|
|
|
|
#define VECTOR_TO_PYTHON_CONVERT(vectorname) \
|
|
py::to_python_converter<vectorname, vector_to_python_list<vectorname>, false>()
|
|
|
|
template <class T>
|
|
T python_list_to_vector(py::list pylist) {
|
|
size_t total = py::len(pylist);
|
|
T result(total);
|
|
for (size_t i = 0; i < total; i++) {
|
|
result[i] = py::extract<typename T::value_type>(pylist[i])();
|
|
}
|
|
return result;
|
|
}
|
|
|
|
#endif // HIKYUU_PYBIND_UTILS_H
|