从PybDun11绑定的C++函数的参数中获取单个对象或对象列表

2024-05-19 12:03:12 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图用PYBIND绑定一个接受多个参数和关键字参数的函数。 该函数如下所示:

{
    OutputSP output;
    InputSP input;
    if (args.size() == 1)
    {
        input = py::cast<InputSP>(args[0]);
    } else if (args.size() == 2)
    {
        query = py::cast<OutputSP>(args[0]);
        expression = py::cast<InputSP>(args[1]);
    }
    // TODO: Default values - should be elsewhere?
    Flags mode(Flags::FIRST_OPTION);
    std::vector<MultiFlags> status({ MultiFlags::COMPLETE });
    for (auto& kv : kwargs)
    {
        auto key_name = py::cast<string>(kv.first);
        if (key_name == "flag")
        {
            mode = py::cast<Flags>(kv.second);
        }
        else if (key_name == "status")
        {
            status = py::cast<std::vector<MultiFlags> >(kv.second);
        }
    }
    return method(output, input, mode, status);
}

其中标志和多重标志的定义如下

    py::enum_<Flags>(m, "Flags")
        .value("FirstOption", Flags::FIRST_OPTION)
        .value("SecondOption", Flags::SECOND_OPTION)
        .export_values();

    py::enum_<MultiFlags>(m, "MultiFlags")
        .value("Complete", MultiFlags::COMPLETE)
        .value("Incomplete", MultiFlags::INCOMPLETE)
        .export_values();

还有包装纸

m.def("method", &method_wrapper);

现在,如果调用包含

status=[MultiFlags.Complete]

我正在寻找一种方法来提前检查kwarg的类型,这样我也可以接受一个包含

status=MultiFlags.Complete

但在PYBIND11文档中找不到任何相关内容。正确的方法是什么


Tags: keypyinputifvaluemodestatusargs
1条回答
网友
1楼 · 发布于 2024-05-19 12:03:12

找到了

当转换失败时,py::cast抛出py::cast\u错误,因此我可以使用try-catch处理这两个选项:

else if (key_name == "status")
{
    try
    {
        status = py::cast<std::vector<MultiFlags> >(kv.second);
    } catch (py::cast_error&)
    {
        status = { py::cast<MultiFlags>(kv.second) };
    }
}

相关问题 更多 >

    热门问题