在pybind11

时间:2016-10-04 19:36:03

标签: python c++ pybind11

我试图用pybind11创建一个python绑定,它引用一个C ++实例,其内存在C ++端处理。以下是一些示例代码:

import <pybind11/pybind11>

struct Dog {
    void bark() { printf("Bark!\n"); }
};

int main()
{
  auto dog = new Dog;
  Py_Initialize();
  initexample(); // Initialize the example python module for import 

  // TBD - Add binding  between dog and example.dog .

  PyRun_StringFlags("import example\n"
                    "\n"
                    "example.dog.bark()\n"  // Access the C++ allocated object dog.
                    , Py_file_input, main_dict, main_dict, NULL);
  Py_Finalize();
}

我坚持如何在python example.dog和C ++ dog变量之间建立链接。

我无法使用py:class_<Dog>.def(py::init<>()),因为这将分配Dog的新实例,这不是我想要的。

3 个答案:

答案 0 :(得分:2)

我找到了自己问题的答案。诀窍是以下两个概念的组合:

  • 创建一个返回单例的独立函数。
  • 创建与单例类的绑定,而不用绑定构造函数。

以下示例说明了该技术:

#include <Python.h>
#include <pybind11/pybind11.h>

namespace py = pybind11;
using namespace pybind11::literals;

// Singleton to wrap
struct Singleton
{
  Singleton() : x(0) {}

  int exchange(int n)  // set x and return the old value
  {
    std::swap(n, x);
    return n;
  }

  // Singleton reference 
  static Singleton& instance()
  {
    static Singleton just_one;
    return just_one;
  }

  int x;
};

PYBIND11_PLUGIN(example) {
    py::module m("example", "pybind11 example plugin");

    // Use this function to get access to the singleton
    m.def("get_instance",
          &Singleton::instance,
          py::return_value_policy::reference,
          "Get reference to the singleton");

    // Declare the singleton methods
    py::class_<Singleton>(m, "Singleton")
      // No init!
      .def("exchange",
           &Singleton::exchange,
           "n"_a,
           "Exchange and return the current value"
           )
      ;

    return m.ptr();
}

int main(int argc, char **argv)
{
  Py_Initialize();

  PyObject* main_module = PyImport_AddModule("__main__");
  PyObject* main_dict = PyModule_GetDict(main_module);

  initexample();

  // Call singleton from c++
  Singleton::instance().exchange(999);

  // Populate the example class with two static pointers to our instance.
  if (PyRun_StringFlags("import example\n"
                        "\n"
                        "example.s1 = example.get_instance()\n"
                        "example.s2 = example.get_instance()\n",
                        Py_file_input, main_dict, main_dict, NULL) == nullptr)
      PyErr_Print();

  // Test referencing the singleton references
  if (PyRun_StringFlags("from example import *\n"
                        "\n"
                        "for i in range(3):\n"
                        "  print s1.exchange(i*2+1)\n"
                        "  print s2.exchange(i*2+2)\n"
                        "print dir(s1)\n"
                        "print help(s1.exchange)\n"
                        ,
                        Py_file_input, main_dict, main_dict, NULL) == nullptr)
      PyErr_Print();

  Py_Finalize();

  exit(0);
}

答案 1 :(得分:1)

由于Pybind11 v2.2.0是另一种方法,使用自定义构造函数包装:python init方法不再需要调用c ++构造函数。您可以直接返回c ++单例实例。

您案件中的声明可能如下:

   // Declare the singleton methods
   py::class_<Singleton>(m, "Singleton")

      .def("init", [](){
          return std::unique_ptr<Singleton, py::nodelete>(&Singleton::instance());
      });

在python中:

myInstance1 = Singleton()
myInstance2 = Singleton()

myInstance1和myInstance2指向同一个c ++对象。

这基本上是the same answer as that other question

答案 2 :(得分:1)

是的,我知道这个答案还很晚,但是以前提供的解决方案要么过时,要么以晦涩的方式解决了问题。

其他答案的最大问题是它们同时使用pybind和原始Python接口。有了pybind,您可以使用更简单,更好的界面到解释器。

遵循可以解决您问题的实施方案。

您会注意到的第一件事是我们使用“ embed.h”头文件。 这为我们提供了创建嵌入式模块的功能。

进一步,我们使用PYBIND11_EMBEDDED_MODULE而不是常规的PYBIND11_MODULE或过时的PYBIND11_PLUGIN。这是专门用于嵌入的宏。

下一个有趣的部分是我们为结构定义的类型。除了Dog类型,我们还使用shared_ptr<Dog>。这对于处理实例至关重要。当main模块超出范围并开始清理时,它需要知道该类/结构的类型为shared_ptr,否则您将遇到段错误(原始指针在此处不可用,我个人认为这很好东西)。

最后要指出的是,我们实际上将类pybind11::scoped_interpreter用于解释器,而不使用Raw Python接口。

#include"pybind11\pybind11.h"
#include"pybind11\embed.h"

#include<iostream>

namespace py = pybind11;

struct Dog {
    void bark() { std::cout << "Bark!\n";  }
};

PYBIND11_EMBEDDED_MODULE(DogModule, m) {
    // look at the types! we have a shared_ptr<Dog>!
    py::class_<Dog, std::shared_ptr<Dog>>(m, "DogModule")
        .def("bark", &Dog::bark);
}


int main(int argc, char **argv) 
{
    // Create Python Interpreter
    py::scoped_interpreter guard;

    // Create Dog Instance
    std::shared_ptr<Dog> ptr = std::make_shared<Dog>();

    // Import the DogModule & Assign the instance to a name in python
    py::module main = py::module::import("__main__");
    main.import("DogModule");
    main.attr("dogInstance") = ptr;

    // Call the bark method from python
    py::exec("dogInstance.bark()");


    getchar();
    return 0;
}