使用BOOST.python从C ++将结构返回给Python

时间:2013-10-04 15:53:42

标签: c++ python boost wrapper boost-python

我编写了一个C ++方法,我需要将结构返回给Python。 我已经能够使用BOOST按照本link中描述的方法将OpenCV mat从Python发送到C ++。

现在我需要走另一条路;从C ++返回到Python,然后在Python中访问该结构。可以吗?

任何样本或参考链接都会很好。我在发布这个问题之前尝试过谷歌搜索,但我无法获得任何样本或解释链接。

2 个答案:

答案 0 :(得分:5)

您可以使用modules / python / src2 / cv2.cpp中的其他函数:

PyObject* pyopencv_from(const cv::Mat& m)
{
  if( !m.data )
      Py_RETURN_NONE;
  cv::Mat temp, *p = (cv::Mat*)&m;
  if(!p->refcount || p->allocator != &g_numpyAllocator)
  {
      temp.allocator = &g_numpyAllocator;
      m.copyTo(temp);
      p = &temp;
  }
  p->addref();
  return pyObjectFromRefcount(p->refcount);
}

然后Boost Python包装器将如下所示:

boost::python::object toPython( const cv::Mat &frame )
{
    PyObject* pyObjFrame = pyopencv_from( frame );
    boost::python::object boostPyObjFrame(boost::python::handle<>((PyObject*)pyObjFrame));

    return boostPyObjFrame;
}

请查看此链接:https://wiki.python.org/moin/boost.python/handle以确保使用适当的boost :: python :: handle&lt;&gt;适用于您的情况。

如果你需要不需要返回一个cv :: Mat但是你可以考虑使用不同的数据来使用boost :: python :: list或boost :: python :: dict。例如,如果要将2D点的向量返回到Python,可以执行以下操作:

boost::python::dict toPython( std::vector<cv::Point> newPoints, std::vector<cv::Point> oldPoints )
{
    boost::python::dict pointsDict;
    boost::python::list oldPointsList;
    boost::python::list newPointsList;

    for( size_t ii = 0; ii < oldPoints.size( ); ++ii )
    {
        oldPointsList.append( boost::python::make_tuple( oldPoints[ii].x, oldPoints[ii].y ) );
    }

    for( size_t ii = 0; ii < newPoints.size( ); ++ii )
    {
        newPointsList.append( boost::python::make_tuple( newPoints[ii].x, newPoints[ii].y ) );
    }

    pointsDict["oldPoints"] = oldPointsList;
    pointsDict["newPoints"] = newPointsList;
    return pointsDict
}

最后是Boost Python包装器:

BOOST_PYTHON_MODULE( myWrapper )
{
    // necessary only if array (cv::Mat) is returned
    import_array();
    boost::python::converter::registry::insert( &extract_pyarray, type_id<PyArrayObject>());

    def toPython("toPython", &toPython);
}

我没有测试过这个特定的解决方案,但原则上应该可行。

答案 1 :(得分:-1)

这可能有点太晚了,但请看一下https://github.com/spillai/numpy-opencv-converter