使用 SWIG 将 C 结构数组访问到 Python

Question

我试图从 Python 调用现有的 C 代码。C 代码定义了一个B包含As结构体数组的结构体。C 代码还定义了一个函数，该函数在调用时将值放入结构中。我可以访问数组成员变量，但它不是列表（或支持索引的东西）。相反，我得到一个对象，它是B*.

我发现了这个问题，但看起来并没有完全解决。我也不确定如何制作 Python 类的实例B来替换PyString_FromString().

下面是演示我的问题以及如何执行它所需的代码：

例子.h

typedef struct A_s
{
  unsigned int thing;
  unsigned int other;
} A_t;

typedef struct B_s
{
  unsigned int size;
  A_t items[16];
} B_t;

unsigned int foo(B_t* b);

例子.c

#include "example.h"

unsigned int
foo(B_t* b)
{
  b->size = 1;
  b->items[0].thing = 1;
  b->items[0].other = 2;
}

例子.i

%module example
%{
#define SWIG_FILE_WITH_INIT
#include "example.h"
%}

%include "example.h"

设置文件

from distutils.core import setup, Extension

module1 = Extension('example', sources=['example.c', 'example.i'])
setup(name='Example', version='0.1', ext_modules=[module1])

script.py - 这使用库并演示失败。

import example
b = example.B_t()
example.foo(b)
print b.size
print b.items
for i in xrange(b.size):
    print b.items[i]

如何运行一切：

python setup.py build_ext --inplace
mv example.so _example.so
python script.py

Answer 1

你可以编写一些 C 辅助函数，例如：

int get_thing(B_t *this_b_t, int idx);
int get_other(B_t *this_b_t, int idx);
void set_thing(B_t *this_b_t, int idx, int val);
void set_other(B_t *this_b_t, int idx, int val);

包装它们，然后您可以使用从中获取的指针example.B_t()来访问数据结构安排中的值，例如

>>> b = example.B_t()
>>> a_thing = example.get_thing(b, 0)
>>> example.set_thing(b,0,999)

希望这些 C 函数的实现应该是什么是显而易见的 - 如果不是我可以提供一个例子......

当然，这并不像能够像 python 列表那样访问 C 数组那么无缝 - 您可能可以使用类型映射来实现此目的，但我不记得在这种情况下需要的确切语法 - 您必须这样做SWIG 文档中的一些 RTFM

此处也可能重复：nested Structure array access in python using SWIG