pythonからc++ライブラリCavalierContoursのparallelOffsetを
pybind11を使用して呼び出したいと思っています。
- python 3.8.5
- cmake 3.18.3
- windows10 64bit
std::vector<cavc::Polyline<double>> results = cavc::parallelOffset(polyline, offset);
は与えられたpolylineをoffsetだけ平行移動して返す関数です。
返り値がstd::vectorなのは、ヒョウタンのような形状のparallelOffsetだと
上下でpolylineが分断される場合があるので、[上のpolyline,下のpolyline]という
形で返すためです。
c++は初めてなのでWeb情報を頼りに下記のようなコードを作成しました。
c++
1#include <pybind11/pybind11.h> 2#include <pybind11/numpy.h> 3#include <pybind11/stl.h> 4#include <pybind11/complex.h> 5#include <pybind11/functional.h> 6#include <pybind11/chrono.h> 7#include "polylineoffset.hpp" 8 9namespace py = pybind11; 10std::vector<py::array_t<double>> parallelOffset(py::array_t<double> input ,double offset) 11{ 12 cavc::Polyline<double> cavc_input; 13 const auto &buff_info = input.request(); 14 const auto &shape = buff_info.shape; 15 for (auto i = 0; i < shape[0]; i++) 16 { 17 cavc_input.addVertex(*input.data(i,0),*input.data(i,1),*input.data(i,2)); 18 } 19 std::vector<cavc::Polyline<double>> result = cavc::parallelOffset(cavc_input,offset); 20 std::vector<py::array_t<double>> result_array; 21 for(auto polyline : result) 22 { 23 int n = polyline.size(); 24 auto temp = py::array_t<double>({n,3}); 25 int i =0; 26 for(auto v :polyline.vertexes()) 27 { 28 *temp.mutable_data(i,0)=v.x(); 29 *temp.mutable_data(i,1)=v.y(); 30 *temp.mutable_data(i,2)=v.bulge(); 31 i++; 32 } 33 result_array.push_back(temp); 34 } 35 return result_array; 36}
「python setup.py install 」は無事に通るのですが、いざpython上で使用すると、
TypeError: Unable to convert function return value to a Python type! The signature was (arg0: numpy.ndarray[float64], arg1: float) -> std::vector<array_t<double,16>,std::allocator<array_t<double,16> > >
というエラーがでてしまい先に進めません。。
ちなみに
c++
1py::array_t<double> test(py::array_t<double> input ,double offset) 2{ 3 cavc::Polyline<double> cavc_input; 4 const auto &buff_info = input.request(); 5 const auto &shape = buff_info.shape; 6 for (auto i = 0; i < shape[0]; i++) 7 { 8 cavc_input.addVertex(*input.data(i,0),*input.data(i,1),*input.data(i,2)); 9 } 10 std::vector<cavc::Polyline<double>> result = cavc::parallelOffset(cavc_input,offset); 11 py::array_t<double> result_array; 12 13 int n = result[0].size(); 14 auto temp = py::array_t<double>({n,3}); 15 int i =0; 16 for(auto v :result[0].vertexes()) 17 { 18 *temp.mutable_data(i,0)=v.x(); 19 *temp.mutable_data(i,1)=v.y(); 20 *temp.mutable_data(i,2)=v.bulge(); 21 i++; 22 } 23 return temp; 24}
と、std::vectorで返してもらうのをあきらめて単独のpolylineのみ
(ヒョウタン形状だと上下どちらかのpolylineのみ)
で妥協すると,きちんとparallelOffsetされたnumpy.arrayを受け取れます。
参考にしたサイトは
plibfacedetectionです。
妥協バージョンはうまく動いていることを考えると、返り値のstd::vectorの作成がうまくいってない
と思うのですが、よくわかりませんでした。
以上、よろしくお願いします。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。