如何为继承LeafSystem<T>的类生成可用的pybind11 Python绑定

解决方案 #

问题根源是Drake的pybind11绑定使用了模板类型别名映射，直接用drake::systems::LeafSystem<double>作为基类注册时，pybind11无法找到对应的Python类型——Drake内部已将模板实例绑定为不带模板参数的类名（比如LeafSystem而非LeafSystem<double>）。

你需要借助Drake提供的pybind11辅助工具来获取正确的基类绑定对象，修改步骤如下：

1. 添加Drake辅助头文件
   在绑定代码中引入pydrake/common/cpp_template_pybind.h，这个头文件包含了模板类型映射的工具函数。

2. 使用py::template_cast获取基类句柄
   替换原代码中基类的指定方式，用py::template_cast<drake::systems::LeafSystem<double>>()来引用Drake已注册的基类类型。

修改后的完整绑定代码：

// my_module.cpp
#include "my_system.h"
#include <drake/systems/framework/leaf_system.h>
#include <pybind11/pybind11.h>
#include <pydrake/common/cpp_template_pybind.h>

namespace py = pybind11;

PYBIND11_MODULE(my_module, m) {
  py::module::import("pydrake.systems.framework");
  // 通过template_cast获取Drake注册的LeafSystem<double>对应的Python类型
  using PyBaseSystem = decltype(py::template_cast<drake::systems::LeafSystem<double>>());
  py::class_<MySystem<double>, PyBaseSystem>(m, "MySystem")
      .def(py::init<>());
}

原理说明 #

Drake在生成Python绑定时，通过cpp_template_pybind.h中的工具将C模板实例（如LeafSystem<double>）映射到Python中简化的类名（LeafSystem）。直接使用原始C模板类型名时，pybind11无法关联到Drake已注册的Python类型，因此会抛出“未知基类型”错误。py::template_cast会自动查找Drake注册的对应模板实例的Python类型句柄，让pybind11能正确识别基类关系。

内容的提问来源于stack exchange，提问作者Thore Goll