SWIG 结构指针作为输出参数

我有一个结构：

struct some_struct_s {
   int arg1;
   int arg2;
};

我有一个 C 函数：

int func(some_struct_s *output);

两者都是%included进入我的 SWIG 文件。

I want some_struct_s *output被视为输出参数。 Python 示例：

int_val, some_struct_output = func()

POD 类型手册中介绍了“输出参数”（第 10.1.3 节），但非 POD 类型未介绍。

我如何告诉 SWIG 我想要some_struct_s *output作为输出参数？

来自文档 http://www.swig.org/Doc3.0/SWIGDocumentation.html#Typemaps_nn32:

11.5.7 “argout”类型映射

“argout”类型映射用于从参数返回值。这最常用于为需要返回多个值的 C/C++ 函数编写包装器。 “argout”类型映射几乎总是与“in”类型映射结合在一起——可能会忽略输入值......

这是您的代码的完整示例（为简洁起见，没有错误检查）：

%module test

// Declare an input typemap that suppresses requiring any input and
// declare a temporary stack variable to hold the return data.
%typemap(in,numinputs=0) some_struct_s* (some_struct_s tmp) %{
    $1 = &tmp;
%}

// Declare an output argument typemap.  In this case, we'll use
// a tuple to hold the structure data (no error checking).
%typemap(argout) some_struct_s* (PyObject* o) %{
    o = PyTuple_New(2);
    PyTuple_SET_ITEM(o,0,PyLong_FromLong($1->arg1));
    PyTuple_SET_ITEM(o,1,PyLong_FromLong($1->arg2));
    $result = SWIG_Python_AppendOutput($result,o);
%}

// Instead of a header file, we'll just declare this code inline.
// This includes the code in the wrapper, as well as telling SWIG
// to create wrappers in the target language.
%inline %{

struct some_struct_s {
   int arg1;
   int arg2;
};

int func(some_struct_s *output) {
    output->arg1 = 1;
    output->arg2 = 2;
    return 0;
}

%}

下面演示。请注意，int返回值零以及作为元组的输出参数都作为列表返回。

>>> import test
>>> test.func()
[0, (1, 2)]

如果您不需要类型映射，您还可以注入代码来创建对象并将其返回以对用户隐藏它：

%module test

%rename(_func) func; // Give the wrapper a different name

%inline %{

struct some_struct_s {
   int arg1;
   int arg2;
};

int func(struct some_struct_s *output)
{
    output->arg1 = 1;
    output->arg2 = 2;
    return 0;
}

%}

// Declare your interface
%pythoncode %{
def func():
    s = some_struct_s()
    r = _func(s)
    return r, s
%}

Demo:

>>> import test
>>> r, s = test.func()
>>> r
0
>>> s
<test.some_struct_s; proxy of <Swig Object of type 'some_struct_s *' at 0x000001511D70A880> >
>>> s.arg1
1
>>> s.arg2
2

如果您仔细选择 SWIG 宏，您可以使类型映射语言变得不可知：

%module test

%typemap(in,numinputs=0) struct some_struct_s *output %{
    $1 = malloc(sizeof(struct some_struct_s));
%}

%typemap(argout) struct some_struct_s* output {
    // The last parameter passes ownership of the pointer
    // to Python so it will be freed when the object's
    // reference count goes to zero.
    %append_output(SWIG_NewPointerObj($1, $1_descriptor, SWIG_POINTER_OWN));
}

%inline %{

struct some_struct_s {
   int arg1;
   int arg2;
};

int func(struct some_struct_s *output)
{
    output->arg1 = 1;
    output->arg2 = 2;
    return 0;
}

%}

Demo:

>>> import test
>>> r, s = test.func()
>>> r
0
>>> s
<test.some_struct_s; proxy of <Swig Object of type 'some_struct_s *' at 0x000001DD0425A700> >
>>> s.arg1
1
>>> s.arg2
2