SWIG：将其参数从 c++ 修改为 python 的函数
Posted
技术标签:
【中文标题】SWIG：将其参数从 c++ 修改为 python 的函数【英文标题】：SWIG : function that modifies its argument from c++ to python 【发布时间】：2018-07-30 16:10:33 【问题描述】：我是 SWIG 的新手，我使用的相机使用 SWIG 将 c++ 包装在 python 中。 这台相机很特别，因为我无法直接获取原始图像。然后我找到/添加一个可以制作快照的 C++ 函数，这里是：
int PixyInterpreter::get_frame(void)
unsigned int size_frame = 8; // size of the frame to grab print
unsigned char current_frame[72000]; // ~largest possible given current hardware
unsigned char *pixels; //returned pointer to video frame buffer
int32_t response, fourcc;
int8_t renderflags;
int return_value;
uint16_t width, height;
uint32_t numPixels;
// stop blob processing
return_value = pixy_command("stop", END_OUT_ARGS, &response, END_IN_ARGS);
printf("STOP returned %d response %d\n", return_value, response);
response = 0;
return_value = pixy_command("cam_getFrame", // String id for remote procedure
0x01, 0x21, // mode 0 = 1280x800 25 fps
0x02, 0, // xoffset
0x02, 0, // yoffset
0x02, 320, // width
0x02, 200, // height (56 max @ 1280 w)
0, // separator
&response, // pointer to mem address for return value
&fourcc, //contrary to docs, the next 5 args are needed
&renderflags,
&width,
&height,
&numPixels,
&pixels, // pointer to mem address for returned frame
0);
printf("getFrame returned %d response %d\n", return_value, response);
printf("returned w %d h %d npix %d\n",width,height,numPixels);
// quit now if not successful:
if(return_value != 0) return return_value;
// save this block
memcpy(¤t_frame, pixels,numPixels);
// display average and size_frame x size_frame pixel dump
unsigned int i,j,ind,start;
// dump a few raw pixels
start=(height/2)*width+width/2; //roughly in middle of frame
for (i=0; i<size_frame; i++)
for (j=0; j<size_frame; j++)
ind = i*width + j + start;
printf(" %02x",frame[ind]);
printf("\n");
// run the last programm again
return_value = pixy_command("run", END_OUT_ARGS, &response, END_IN_ARGS);
printf("START returned %d response %d\n", return_value, response);
return numPixels;
这段代码在 C++ 中运行良好，在 Python 中包装时：它打印像素值。
但问题是我无法获取像素值（数组）在 Python 中，因为代码 c++ 不返回它。好的逻辑:)。
我的想法是在 Python 中使用 char 数组，它将包含像素值，并且可以通过将其作为参数传递给函数 get_frame 来进行修改。 c++ 函数看起来像：
int PixyInterpreter::get_frame(unsigned char *current_frame)
和接口文件“pixy.i”（部分）：
%module pixy
%include "stdint.i"
%include "carrays.i"
%
#define SWIG_FILE_WITH_INIT
#include "pixy.h"
%
%array_class(unsigned char, charArray);
int get_frame(charArray * frame);
但不幸的是，在python中：
>> a = charArray(72000)
>> get_frame(a)
[...errors...]
它返回*检测到缓冲区溢出*：python终止，这似乎发生在memcpy！
*** buffer overflow detected ***: python terminated
======= Backtrace: =========
/lib/x86_64-linux-gnu/libc.so.6(+0x777e5)[0x7f63d95227e5]
/lib/x86_64-linux-gnu/libc.so.6(__fortify_fail+0x5c)[0x7f63d95c415c]
/lib/x86_64-linux-gnu/libc.so.6(+0x117160)[0x7f63d95c2160]
./_pixy.so(_ZN15PixyInterpreter9get_frameEPh+0x156)[0x7f63d801e636]
./_pixy.so(+0x10d37)[0x7f63d8015d37]
python(PyEval_EvalFrameEx+0x5ca)[0x4bc3fa]
python(PyEval_EvalCodeEx+0x306)[0x4b9ab6]
python[0x4eb30f]
python(PyRun_InteractiveOneFlags+0x190)[0x44a7a2]
python(PyRun_InteractiveLoopFlags+0xba)[0x44a56d]
python[0x43092e]
python(Py_Main+0x612)[0x493ae2]
/lib/x86_64-linux-gnu/libc.so.6(__libc_start_main+0xf0)[0x7f63d94cb830]
python(_start+0x29)[0x4933e9]
======= Memory map: ========
00400000-006de000 r-xp 00000000 08:03 9969410 /usr/bin/python2.7
008dd000-008de000 r--p 002dd000 08:03 9969410 /usr/bin/python2.7
008de000-00955000 rw-p 002de000 08:03 9969410 /usr/bin/python2.7
00955000-00978000 rw-p 00000000 00:00 0
02404000-02548000 rw-p 00000000 00:00 0 [heap]
7f63d0000000-7f63d0021000 rw-p 00000000 00:00 0
7f63d0021000-7f63d4000000 ---p 00000000 00:00 0
7f63d6019000-7f63d601a000 ---p 00000000 00:00 0
7f63d601a000-7f63d681a000 rw-p 00000000 00:00 0
7f63d681a000-7f63d681b000 ---p 00000000 00:00 0
7f63d681b000-7f63d701b000 rw-p 00000000 00:00 0
7f63d701b000-7f63d7022000 r-xp 00000000 08:03 2490667 /lib/x86_64-linux-gnu/librt-2.23.so
7f63d7022000-7f63d7221000 ---p 00007000 08:03 2490667 /lib/x86_64-linux-gnu/librt-2.23.so
7f63d7221000-7f63d7222000 r--p 00006000 08:03 2490667 /lib/x86_64-linux-gnu/librt-2.23.so
7f63d7222000-7f63d7223000 rw-p 00007000 08:03 2490667 /lib/x86_64-linux-gnu/librt-2.23.so
7f63d7223000-7f63d7239000 r-xp 00000000 08:03 2494948 /lib/x86_64-linux-gnu/libgcc_s.so.1
7f63d7239000-7f63d7438000 ---p 00016000 08:03 2494948 /lib/x86_64-linux-gnu/libgcc_s.so.1
7f63d7438000-7f63d7439000 rw-p 00015000 08:03 2494948 /lib/x86_64-linux-gnu/libgcc_s.so.1
7f63d7439000-7f63d75ab000 r-xp 00000000 08:03 9963184 /usr/lib/x86_64-linux-gnu/libstdc++.so.6.0.21
7f63d75ab000-7f63d77ab000 ---p 00172000 08:03 9963184 /usr/lib/x86_64-linux-gnu/libstdc++.so.6.0.21
7f63d77ab000-7f63d77b5000 r--p 00172000 08:03 9963184 /usr/lib/x86_64-linux-gnu/libstdc++.so.6.0.21
7f63d77b5000-7f63d77b7000 rw-p 0017c000 08:03 9963184 /usr/lib/x86_64-linux-gnu/libstdc++.so.6.0.21
7f63d77b7000-7f63d77bb000 rw-p 00000000 00:00 0
7f63d77bb000-7f63d77d2000 r-xp 00000000 08:03 2495090 /lib/x86_64-linux-gnu/libusb-1.0.so.0.1.0
7f63d77d2000-7f63d79d1000 ---p 00017000 08:03 2495090 /lib/x86_64-linux-gnu/libusb-1.0.so.0.1.0
7f63d79d1000-7f63d79d2000 r--p 00016000 08:03 2495090 /lib/x86_64-linux-gnu/libusb-1.0.so.0.1.0
7f63d79d2000-7f63d79d3000 rw-p 00017000 08:03 2495090 /lib/x86_64-linux-gnu/libusb-1.0.so.0.1.0
7f63d79d3000-7f63d79d9000 r-xp 00000000 08:03 9973881 /usr/lib/x86_64-linux-gnu/libboost_chrono.so.1.58.0
7f63d79d9000-7f63d7bd9000 ---p 00006000 08:03 9973881 /usr/lib/x86_64-linux-gnu/libboost_chrono.so.1.58.0
7f63d7bd9000-7f63d7bda000 r--p 00006000 08:03 9973881 /usr/lib/x86_64-linux-gnu/libboost_chrono.so.1.58.0
7f63d7bda000-7f63d7bdb000 rw-p 00007000 08:03 9973881 /usr/lib/x86_64-linux-gnu/libboost_chrono.so.1.58.0
7f63d7bdb000-7f63d7bde000 r-xp 00000000 08:03 9967987 /usr/lib/x86_64-linux-gnu/libboost_system.so.1.58.0
7f63d7bde000-7f63d7ddd000 ---p 00003000 08:03 9967987 /usr/lib/x86_64-linux-gnu/libboost_system.so.1.58.0
7f63d7ddd000-7f63d7dde000 r--p 00002000 08:03 9967987 /usr/lib/x86_64-linux-gnu/libboost_system.so.1.58.0
7f63d7dde000-7f63d7ddf000 rw-p 00003000 08:03 9967987 /usr/lib/x86_64-linux-gnu/libboost_system.so.1.58.0
7f63d7ddf000-7f63d7e03000 r-xp 00000000 08:03 9967992 /usr/lib/x86_64-linux-gnu/libboost_thread.so.1.58.0
7f63d7e03000-7f63d8002000 ---p 00024000 08:03 9967992 /usr/lib/x86_64-linux-gnu/libboost_thread.so.1.58.0
7f63d8002000-7f63d8004000 r--p 00023000 08:03 9967992 /usr/lib/x86_64-linux-gnu/libboost_thread.so.1.58.0
7f63d8004000-7f63d8005000 rw-p 00025000 08:03 9967992 /usr/lib/x86_64-linux-gnu/libboost_thread.so.1.58.0
7f63d8005000-7f63d802a000 r-xp 00000000 08:03 9699483 /home/leat/pixy/build/pantilt_in_python/_pixy.so
7f63d802a000-7f63d822a000 ---p 00025000 08:03 9699483 /home/leat/pixy/build/pantilt_in_python/_pixy.so
7f63d822a000-7f63d822b000 r--p 00025000 08:03 9699483 /home/leat/pixy/build/pantilt_in_python/_pixy.so
7f63d822b000-7f63d822d000 rw-p 00026000 08:03 9699483 /home/leat/pixy/build/pantilt_in_python/_pixy.so
7f63d822d000-7f63d8252000 r-xp 00000000 08:03 2495082 /lib/x86_64-linux-gnu/libtinfo.so.5.9
7f63d8252000-7f63d8451000 ---p 00025000 08:03 2495082 /lib/x86_64-linux-gnu/libtinfo.so.5.9
7f63d8451000-7f63d8455000 r--p 00024000 08:03 2495082 /lib/x86_64-linux-gnu/libtinfo.so.5.9
7f63d8455000-7f63d8456000 rw-p 00028000 08:03 2495082 /lib/x86_64-linux-gnu/libtinfo.so.5.9
7f63d8456000-7f63d8493000 r-xp 00000000 08:03 2495061 /lib/x86_64-linux-gnu/libreadline.so.6.3
7f63d8493000-7f63d8693000 ---p 0003d000 08:03 2495061 /lib/x86_64-linux-gnu/libreadline.so.6.3
7f63d8693000-7f63d8695000 r--p 0003d000 08:03 2495061 /lib/x86_64-linux-gnu/libreadline.so.6.3
7f63d8695000-7f63d869b000 rw-p 0003f000 08:03 2495061 /lib/x86_64-linux-gnu/libreadline.so.6.3
7f63d869b000-7f63d869c000 rw-p 00000000 00:00 0
7f63d869c000-7f63d86a1000 r-xp 00000000 08:03 10096750 /usr/lib/python2.7/lib-dynload/readline.x86_64-linux-gnu.so
7f63d86a1000-7f63d88a1000 ---p 00005000 08:03 10096750 /usr/lib/python2.7/lib-dynload/readline.x86_64-linux-gnu.so
7f63d88a1000-7f63d88a2000 r--p 00005000 08:03 10096750 /usr/lib/python2.7/lib-dynload/readline.x86_64-linux-gnu.so
7f63d88a2000-7f63d88a4000 rw-p 00006000 08:03 10096750 /usr/lib/python2.7/lib-dynload/readline.x86_64-linux-gnu.so
7f63d88a4000-7f63d8b81000 r--p 00000000 08:03 9963224 /usr/lib/locale/locale-archive
7f63d8b81000-7f63d8c89000 r-xp 00000000 08:03 2490396 /lib/x86_64-linux-gnu/libm-2.23.so
7f63d8c89000-7f63d8e88000 ---p 00108000 08:03 2490396 /lib/x86_64-linux-gnu/libm-2.23.so
7f63d8e88000-7f63d8e89000 r--p 00107000 08:03 2490396 /lib/x86_64-linux-gnu/libm-2.23.so
7f63d8e89000-7f63d8e8a000 rw-p 00108000 08:03 2490396 /lib/x86_64-linux-gnu/libm-2.23.so
7f63d8e8a000-7f63d8ea3000 r-xp 00000000 08:03 2490612 /lib/x86_64-linux-gnu/libz.so.1.2.8
7f63d8ea3000-7f63d90a2000 ---p 00019000 08:03 2490612 /lib/x86_64-linux-gnu/libz.so.1.2.8
7f63d90a2000-7f63d90a3000 r--p 00018000 08:03 2490612 /lib/x86_64-linux-gnu/libz.so.1.2.8
7f63d90a3000-7f63d90a4000 rw-p 00019000 08:03 2490612 /lib/x86_64-linux-gnu/libz.so.1.2.8
7f63d90a4000-7f63d90a6000 r-xp 00000000 08:03 2490656 /lib/x86_64-linux-gnu/libutil-2.23.so
7f63d90a6000-7f63d92a5000 ---p 00002000 08:03 2490656 /lib/x86_64-linux-gnu/libutil-2.23.so
7f63d92a5000-7f63d92a6000 r--p 00001000 08:03 2490656 /lib/x86_64-linux-gnu/libutil-2.23.so
7f63d92a6000-7f63d92a7000 rw-p 00002000 08:03 2490656 /lib/x86_64-linux-gnu/libutil-2.23.so
7f63d92a7000-7f63d92aa000 r-xp 00000000 08:03 2490651 /lib/x86_64-linux-gnu/libdl-2.23.so
7f63d92aa000-7f63d94a9000 ---p 00003000 08:03 2490651 /lib/x86_64-linux-gnu/libdl-2.23.so
7f63d94a9000-7f63d94aa000 r--p 00002000 08:03 2490651 /lib/x86_64-linux-gnu/libdl-2.23.so
7f63d94aa000-7f63d94ab000 rw-p 00003000 08:03 2490651 /lib/x86_64-linux-gnu/libdl-2.23.so
7f63d94ab000-7f63d966b000 r-xp 00000000 08:03 2490649 /lib/x86_64-linux-gnu/libc-2.23.so
7f63d966b000-7f63d986b000 ---p 001c0000 08:03 2490649 /lib/x86_64-linux-gnu/libc-2.23.so
7f63d986b000-7f63d986f000 r--p 001c0000 08:03 2490649 /lib/x86_64-linux-gnu/libc-2.23.so
7f63d986f000-7f63d9871000 rw-p 001c4000 08:03 2490649 /lib/x86_64-linux-gnu/libc-2.23.so
7f63d9871000-7f63d9875000 rw-p 00000000 00:00 0
7f63d9875000-7f63d988d000 r-xp 00000000 08:03 2490648 /lib/x86_64-linux-gnu/libpthread-2.23.so
7f63d988d000-7f63d9a8c000 ---p 00018000 08:03 2490648 /lib/x86_64-linux-gnu/libpthread-2.23.so
7f63d9a8c000-7f63d9a8d000 r--p 00017000 08:03 2490648 /lib/x86_64-linux-gnu/libpthread-2.23.so
7f63d9a8d000-7f63d9a8e000 rw-p 00018000 08:03 2490648 /lib/x86_64-linux-gnu/libpthread-2.23.so
7f63d9a8e000-7f63d9a92000 rw-p 00000000 00:00 0
7f63d9a92000-7f63d9ab8000 r-xp 00000000 08:03 2490647 /lib/x86_64-linux-gnu/ld-2.23.so
7f63d9ae2000-7f63d9b00000 r-xp 00000000 08:03 2490373 /lib/x86_64-linux-gnu/libudev.so.1.6.4
7f63d9b00000-7f63d9b01000 r--p 0001d000 08:03 2490373 /lib/x86_64-linux-gnu/libudev.so.1.6.4
7f63d9b01000-7f63d9b02000 rw-p 0001e000 08:03 2490373 /lib/x86_64-linux-gnu/libudev.so.1.6.4
7f63d9b39000-7f63d9c7f000 rw-p 00000000 00:00 0
7f63d9c7f000-7f63d9c86000 r--s 00000000 08:03 10227159 /usr/lib/x86_64-linux-gnu/gconv/gconv-modules.cache
7f63d9c86000-7f63d9cb7000 rw-p 00000000 00:00 0
7f63d9cb7000-7f63d9cb8000 r--p 00025000 08:03 2490647 /lib/x86_64-linux-gnu/ld-2.23.so
7f63d9cb8000-7f63d9cb9000 rw-p 00026000 08:03 2490647 /lib/x86_64-linux-gnu/ld-2.23.so
7f63d9cb9000-7f63d9cba000 rw-p 00000000 00:00 0
7ffd99927000-7ffd99948000 rw-p 00000000 00:00 0 [stack]
7ffd99985000-7ffd99988000 r--p 00000000 00:00 0 [vvar]
7ffd99988000-7ffd9998a000 r-xp 00000000 00:00 0 [vdso]
ffffffffff600000-ffffffffff601000 r-xp 00000000 00:00 0 [vsyscall]
Abandon (core dumped)
很抱歉这篇长文，但如果有人有任何建议或解释我如何解决我的问题，我将不胜感激！
另外，我发现 %Array_functions() 和 %array_class() 不应该与 char 或 char 类型一起使用 * :/
【问题讨论】：
【参考方案1】：如果您想从 C++ 接收数组并将您在 Python 级别所做的更改反映在 C++ 中，我建议您使用 NumPy SWIG 绑定 numpy.i。
我在这里使用ARGOUTVIEW_ARRAY1，这样您在 Python 中所做的更改会立即反映在底层 C++ 内存中（但这意味着无需调整大小）。请记住，由于内存归 C++ 所有，因此您有责任清理它。如果这不符合您的需要，您可以选择不同的类型图。有good documentation。
test.i
%module example
%
#define SWIG_FILE_WITH_INIT
#include "test.hpp"
%
%include "numpy.i"
%init %
import_array();
%
%apply (unsigned char** ARGOUTVIEW_ARRAY1, int* DIM1) (unsigned char** current_frame, int* numPixels);
%include "test.hpp"
test.hpp
#pragma once
struct PixyInterpreter
unsigned char *frame;
PixyInterpreter() : frame(NULL)
~PixyInterpreter() delete[] frame;
void get_frame(unsigned char **current_frame, int *numPixels)
*numPixels = 12;
frame = new unsigned char[*numPixels];
frame[ 0] = 'H';
frame[ 1] = 'e';
frame[ 2] = 'l';
frame[ 3] = 'l';
frame[ 4] = 'o';
frame[ 5] = ' ';
frame[ 6] = 'W';
frame[ 7] = 'o';
frame[ 8] = 'r';
frame[ 9] = 'l';
frame[10] = 'd';
frame[11] = '!';
frame[12] = '\0';
*current_frame = frame;
;
test.py
import example
interpreter = example.PixyInterpreter()
frame = interpreter.get_frame()
print(frame.tostring())
调用示例：
$ swig -c++ -python -py3 test.i
$ clang++ -Wall -Wextra -Wpedantic -I /usr/include/python3.6/ -fPIC -shared test_wrap.cxx -o _example.so -lpython3.6m
$ python3 test.py
b'Hello World!'
【讨论】：
您的示例非常清楚，我尝试通过一些修改使其在我的代码中工作，但它不起作用。我在回答帖子中详细说明了它。【参考方案2】：非常感谢您详细的回答！
我试过你的例子，效果很好。在尝试将您的想法重现/调整到我的代码时遇到了一些麻烦之后，现在它可以编译但无法正常工作......我在接口文件中进行了一些更改，我的完整接口文件是：
pixy.i
%module pixy
%
#define SWIG_FILE_WITH_INIT
#include "pixy.h"
%
%include "stdint.i"
%include "carrays.i"
%include "numpy.i"
%init%
import_array();
%
%apply (unsigned char** ARGOUTVIEW_ARRAY1, uint16_t* DIM1) (unsigned char** current_frame, uint16_t* nbPixels)
//%include "pixy.h" //<--- Error : Unable to find 'pixy.h'
%array_class(struct Block, BlockArray);
int pixy_init();
void pixy_close();
void pixy_error(int error_code);
int pixy_blocks_are_new();
int pixy_get_blocks(uint16_t max_blocks, BlockArray *blocks);
int pixy_rcs_set_position(uint8_t channel, uint16_t position);
struct Block
uint16_t type;
uint16_t signature;
uint16_t x;
uint16_t y;
uint16_t width;
uint16_t height;
int16_t angle;
;
// Inspired from the block struct above
struct PixyFrame // if not written
// >>> interpreter = pixy.PixyFrame()
// >>> Error : PixyFrame not defined
int get_frame (unsigned char** current_frame, uint16_t* nbPixels);
// if not added
// >>> interpreter = PixyFrame()
// >>> frame = interpreter.get_frame()
// AttributeError: type object 'object' has no attribute '__getattr__'
;
我见过很多这样的例子：
foo.i
%module foo
%
#include "foo.h"
%
%include "foo.h"
但最后一个 include 在我的文件中返回一个错误：Error : Unable to find 'pixy.h'
我几乎可以用我的接口文件重现您的示例，但是，当我调用 get_frame() 时：
>>> interpreter = PixyFrame()
>>> frame = interpreter.get_frame()
TypeError: get_frame() takes exactly 3 arguments (1 given)
需要的 3 个参数是 (self, unsigned char **, uint16_t)
在构建时我收到了这个警告：
pixy.i:17: Warning 453: Can't apply (unsigned char **ARGOUTVIEW_ARRAY1,uint16_t *DIM1). No typemaps are defined.
【讨论】：
默认类型映射中唯一支持的DIM_TYPE 是int。要获得带有uint16_t 的新类型映射，您必须在%include "numpy.i" 之后添加%numpy_typemaps(unsigned char, NPY_UBYTE, uint16_t)。
至于%include "pixy.h"为什么找不到文件，我不知道，但这可能只是你的包含路径的问题。
这在很大程度上取决于您如何在 C++ 中管理内存。正如我在回答中提到的（这显然不值得一票），您还可以将 C++ 内存的副本返回给 Python。然后 Python 将在其拥有的内存上运行析构函数，您不必担心它。
抱歉，但可以肯定的是，我如何管理 c++ 中内存分配的破坏？我在 c++ 中定义了析构函数，如果我是对的，那么我应该在 Python 中调用它，但是如何？我不确定一个简单的“del”是否可以做到
Python 被垃圾回收。当没有更多对该对象的引用时，GC 将在下一次运行期间运行析构函数。在您和我的test.py 中，这是interpreter 变量。如果你想强制收集，请import gc; del interpreter; gc.collect() 但要小心：手动运行垃圾收集器通常会破坏你的性能。以上是关于SWIG：将其参数从 c++ 修改为 python 的函数的主要内容，如果未能解决你的问题，请参考以下文章
使用 SWIG 包装对象从 C++ 调用 Python 函数的最简洁方法是啥
使用 SWIG 将 numpy 数组元素（int）传递给 c++ int
如何使用 swig 将 python 类实例作为 c++ 函数的参数传递？
如何使用 swig 修改类构造函数以保持对构造函数参数之一的引用？