pybind11：使用设置工具 link

Question

我正在尝试使用 Python 的 setuptools 构建此 example 的变体。在示例中，只有一个文件 main.cpp。但是，在我的版本中，我添加了另一个 class。因此，一共有三个文件：

main.cpp

#include <pybind11/pybind11.h>
#include "myClass.h"

namespace py = pybind11;

PYBIND11_MODULE(python_example, m) {
    m.doc() = R"pbdoc(
        Pybind11 example plugin

    )pbdoc";

    py::class_<myClass>(m, "myClass")
    .def(py::init<>())
    .def("addOne", &myClass::addOne)
    .def("getNumber", &myClass::getNumber)
    ;
}

myClass.h

#include <pybind11/pybind11.h>

class myClass
{
public:
    int number;

    myClass();
    void addOne();

    int getNumber();
};

myClass.cpp

#include <pybind11/pybind11.h>
#include "myClass.h"

myClass::myClass() {
    number = 1;
}
void myClass::addOne() {
    number = number + 1;
}

int myClass::getNumber() {
    return number;
}

如果我在示例中使用原始 setup.py 文件，它不起作用，因为我需要 link myClass.cpp 和 main.cpp。我怎样才能使用 setuptools 做到这一点？基本上我正在寻找 setuptools 相当于 CMake 的 target_link_libraries.

我问这个是因为我在 CMake 方面的经验很少。使用 setuptools 对我来说会更容易。

Answer 1

我相信你正在寻找的是setuptools.Extension。我强烈建议您看看这个很好的例子：https://github.com/pybind/python_example。它应该能够指导您完成需要做的事情。

这是我为您的代码提取的内容。请注意，它基本上是从 example.

复制粘贴

from setuptools import setup, Extension
from setuptools.command.build_ext import build_ext

import sys, re
import setuptools
import pybind11


# (c) Sylvain Corlay, https://github.com/pybind/python_example
def has_flag(compiler, flagname):

  import tempfile

  with tempfile.NamedTemporaryFile('w', suffix='.cpp') as f:

    f.write('int main (int argc, char **argv) { return 0; }')

    try:
      compiler.compile([f.name], extra_postargs=[flagname])
    except setuptools.distutils.errors.CompileError:
      return False

  return True


# (c) Sylvain Corlay, https://github.com/pybind/python_example
def cpp_flag(compiler):

  if   has_flag(compiler,'-std=c++14'): return '-std=c++14'
  elif has_flag(compiler,'-std=c++11'): return '-std=c++11'
  raise RuntimeError('Unsupported compiler: at least C++11 support is needed')


# (c) Sylvain Corlay, https://github.com/pybind/python_example
class BuildExt(build_ext):

  c_opts = {
    'msvc': ['/EHsc'],
    'unix': [],
  }

  if sys.platform == 'darwin':
    c_opts['unix'] += ['-stdlib=libc++', '-mmacosx-version-min=10.7']

  def build_extensions(self):
    ct = self.compiler.compiler_type
    opts = self.c_opts.get(ct, [])
    if ct == 'unix':
      opts.append('-DVERSION_INFO="%s"' % self.distribution.get_version())
      opts.append(cpp_flag(self.compiler))
    elif ct == 'msvc':
      opts.append('/DVERSION_INFO=\"%s\"' % self.distribution.get_version())
    for ext in self.extensions:
      ext.extra_compile_args = opts
    build_ext.build_extensions(self)


ext_modules = [
  Extension(
    'python_example',
    ['main.cpp', 'myClass.cpp'],
    include_dirs=[
      pybind11.get_include(False),
      pybind11.get_include(True ),
    ],
    language='c++'
  ),
]


setup(
  name             = 'python_example',
  ext_modules      = ext_modules,
  install_requires = ['pybind11>=2.2.0'],
  cmdclass         = {'build_ext': BuildExt},
  zip_safe         = False,
)

请注意，大部分代码并不处理您的具体问题，而是让 C++11/14 以稳健的方式工作。我尽量保留它在原始示例中的大部分内容，也是为了在此处获得完整的工作代码。

pybind11：使用设置工具 link

pybind11: using setuptools to link

c++

python

pybind11