Проблема Pybind11 для собственного типа возвращаемого значения - PullRequest
Новая
       68

Проблема Pybind11 для собственного типа возвращаемого значения

2 голосов
/ 28 января 2020

Я пытаюсь выставить data_all, который представляет собой вектор от data до python, используя pybind11: 

struct data {
    std::vector<Eigen::ArrayXf> values;
    std::vector<int> indices;
    float x;
    float y;
    float z;
};

class dataBuffer {
public:
    std::vector<data> data_all;
    Eigen::ArrayXf getValues(ssize_t i, ssize_t j) { return data_all.at(i).values.at(j); };
};

Я определил свою оболочку pybind11 следующим образом: 

PYBIND11_MODULE(example, m) {
    py::class_<data>(m, "data")
        .def(py::init<>())
        .def_readonly("values", &data::values)
        .def_readonly("indices", &data::indices)
        .def_readonly("x", &data::x)
        .def_readonly("y", &data::y)
        .def_readonly("z", &data::z);

    py::class_<dataBuffer>(m, "dataBuffer")
        .def(py::init<>())
        .def("getValues", &dataBuffer::getValues);
}

Мой пример кода C ++ 

namespace py = pybind11;
int main()
{
    data d;
    d.x = 1.1;
    d.y = 2.1;
    d.z = 3.1;
    d.indices.push_back(4);
    d.values.push_back(Eigen::ArrayXf::LinSpaced(50, 0.0, 50 - 1.0));
    d.indices.push_back(5);
    d.values.push_back(Eigen::ArrayXf::LinSpaced(60, 0.0, 60 - 1.0));
    d.indices.push_back(11);
    d.values.push_back(Eigen::ArrayXf::LinSpaced(70, 0.0, 70 - 1.0));

    dataBuffer D;
    D.data_all.push_back(d);
    D.data_all.push_back(d);
    std::cout << D.getValues(0,0) << "\n";

    py::scoped_interpreter guard{};
    py::object result = py::module::import("pybind11_test").attr("testData")(0,0);

}

Содержимое файла pybind11_test.py 

import numpy as np
import example as m
def testData(buffer):
    help(buffer)
    a = buffer.getValues(0,0) # trying to retrieve the data buffer created in C++
    print(a)

help(buffer) печатает следующую подпись: 

Help on method getValues in module example:

getValues(...) method of example.dataBuffer instance
    getValues(self: example.dataBuffer, arg0: int, arg1: int) -> Eigen::Array<float,-1,1,0,-1,1>

Однако, когда python достигает выполнения buffer.getValues(0,0), происходит сбой: 

(in Visual Studio)
pybind11::error already set at memory location

(in Python)
Traceback (most recent call last):    
  File "<stdin>", line 1, in <module>
TypeError: getValues(): incompatible function arguments. The following argument types are supported:
    1. (self: example.dataBuffer, arg0: int, arg1: int) -> Eigen::Array<float,-1,1,0,-1,1>

Я считаю, что тип возврата Eigen не оценивается Python. Может ли кто-нибудь помочь мне, в чем может быть проблема? Как мне помочь Python понять тип возвращаемого значения для getValues, чтобы я мог продолжить его обработку с использованием библиотеки numpy?

1 Ответ

2 голосов
/ 28 января 2020

Я предполагаю, что эта строка: 

py::object result = py::module::import("pybind11_test").attr("testData")(0,0);

должен был быть: 

py::object result = py::module::import("pybind11_test").attr("testData")(D);

Кроме этого, все, что вам не хватает: 

#include "pybind11/eigen.h"

в верхней части вашего кода оболочки.

...