参数从 Python 脚本传递到 C++ 使用 boost-python

Parameter passing from Python script to C++ with boost-python

本文关键字：C++ 使用 boost-python Python 脚本 参数      更新时间：2023-10-16

我目前正在使用 boost-python 和 boost-numpy 将 Python 嵌入C++中。我有以下 Python 测试脚本：

import numpy as np
import time

def test_qr(m,n):
    print("create numpy array")
    A = np.random.rand(m, n)
    print("Matrix A is {}".format(A))
    print("Lets QR factorize this thing! Mathematics is great !!")
    ts = time.time()
    Q, R = np.linalg.qr(A)
    te = time.time()
    print("It took {} seconds to factorize A".format(te - ts))
    print("The Q matrix is {}".format(Q))
    print("The R matrix is {}".format(R))
    return Q,R

def sum(m,n):
    return m+n

我能够以这样的C++执行部分代码：

namespace p = boost::python;
namespace np = boost::python::numpy;
int main() {
Py_Initialize();  //initialize python environment
np::initialize(); //initialize numpy environment
p::object main_module = p::import("__main__");
p::object main_namespace = main_module.attr("__dict__");
// execute code in the main_namespace
p::exec_file("/Users/Michael/CLionProjects/CythonTest/test_file.py",main_namespace); //loads python script
p::exec("m = 100n"
        "n = 100n"
        "Q,R = test_qr(m,n)", main_namespace);
np::ndarray Q_matrix = p::extract<np::ndarray>(main_namespace["Q"]); // extract results as numpy array types
np::ndarray R_matrix = p::extract<np::ndarray>(main_namespace["R"]);
std::cout<<"C++ Q Matrix: n" << p::extract<char const *>(p::str(Q_matrix)) << std::endl; // extract every element as a
std::cout<<"C++ R Matrix: n" << p::extract<char const *>(p::str(R_matrix)) << std::endl;
std::cout<<"code also works with numpy, ask for a raise" << std::endl;
p::object sum = main_namespace.attr("sum")(10,10);
int result = p::extract<int>(main_namespace.attr("sum")(10,10));
std::cout<<"sum result works " << result << std::endl;
return 0;}

现在我正在尝试在 Python 脚本中使用 sum 函数，但我并不总是想写一个字符串，例如：

p::exec("m = 100n"
        "n = 100n"
        "Q,R = test_qr(m,n)", main_namespace);}

如何在不使用 exec 函数的情况下完成此操作？

我尝试过这样的事情：

p::object sum = main_namespace.attr("sum")(10,10);
int result = p::extract<int>(main_namespace.attr("sum")(10,10));
std::cout<<"sum result works " << result << std::endl;

如提升文档中所述。我也尝试使用call_method功能，但它不起作用。我得到一个 boost：:p ython：：error_already_set 异常，这意味着 Python 有问题，但我不知道是什么。或退出代码 11。

这个问题相当微不足道。让我们看看你提到的教程：

object main_module = import("__main__");
object main_namespace = main_module.attr("__dict__");
object ignored = exec("result = 5 ** 2", main_namespace);
int five_squared = extract<int>(main_namespace["result"]);

请注意他们如何提取最后一行中的result对象：main_namespace["result"]

main_namespace对象是一个 Python 字典，而不是提取它的属性，你只是在寻找一个与特定键一起存储的值。因此，使用 [] 编制索引是要走的路。

C++代码：

#define BOOST_ALL_NO_LIB
#include <boost/python.hpp>
#include <boost/python/numpy.hpp>
#include <iostream>
namespace bp = boost::python;
int main()
{
    try {
        Py_Initialize();
        bp::object module = bp::import("__main__");
        bp::object globals = module.attr("__dict__");
        bp::exec_file("bpcall.py", globals);
        bp::object sum_fn = globals["sum"];
        int result = bp::extract<int>(sum_fn(1,2));
        std::cout << "Result (C++) = " << result << "n";
    } catch (bp::error_already_set) {
        PyErr_Print();
    }
    Py_Finalize();
}

蟒蛇脚本：

def sum(m,n):
    return m+n

输出：

Result (C++) = 3
相关文章：