在 CFFI 中创建 `REAL (* vertices)[DIM]` 类型的 cdata

Question

我正在尝试使用 CFFI 围绕一些现有 C code 构建一个 python 界面。与往常一样，C 代码为了性能而被修剪，它充满了大量的宏和类型定义。

ATM 我正在复制以下结构

#define DIM     3
typedef double  REAL;
struct Object_structure {
  int numpoints;
  REAL (* vertices)[DIM];
  int * rings;
};
typedef struct Object_structure * Object;

我尝试调用的函数需要一个类型为 Object 的参数。

REAL gjk_distance(
   Object obj1, REAL (* tr1)[DIM+1],
   Object obj2, REAL (* tr2)[DIM+1],
   REAL wpt1[DIM], REAL wpt2[DIM],
   struct simplex_point * simplex, int use_seed
   );

我已经编写了以下 python class 来表示这样的 object/struct，但是我无法将其转换为预期的 cdata 对象。（现在我只考虑一个 UnitCube，但最终我想概括它。）

class Box:
    def __init__(self, pos):
        self._weakkeydict = weakref.WeakKeyDictionary()
        self.numpoints = 8
        self.rings = [
            8, 12, 16, 20, 24, 28, 32, 36,
            3, 1, 4, -1,
            0, 2, 5, -1,
            1, 3, 6, -1,
            2, 0, 7, -1,
            7, 5, 0, -1,
            4, 6, 1, -1,
            5, 7, 2, -1,
            6, 4, 3, -1]
        x, y, z = pos
        self.vertices = [
            [x+0, y+0, z+0],
            [x+1, y+0, z+0],
            [x+1, y+1, z+0],
            [x+0, y+1, z+0],
            [x+0, y+0, z+1],
            [x+1, y+0, z+1],
            [x+1, y+1, z+1],
            [x+0, y+1, z+1],
        ]

    @property
    def cdata(self):
        self._weakkeydict.clear()

        #ptr_numpoints = ffi.new("int", self.numpoints)
        ptr_rings = ffi.new("int[]", self.rings)
        vertices = [ffi.new("REAL[3]", v) for v in self.vertices]
        ptr_vertices = ffi.new("REAL *[3]", vertices )

        ptr_obj = ffi.new("Object", { 
            'numpoints': self.numpoints,
            'rings': ptr_rings,
            'vertices': ptr_vertices})
        self._weakkeydict[ptr_obj] = (ptr_rings, ptr_vertices, vertices)
        return ptr_obj

通过上面的调用，我在 ptr_vertices = ffi.new("REAL *[3]", vertices ) 行中得到了 IndexError: too many initializers for 'double *[3]' (got 8)：

box1 = Box((0,0,0)) 
box2 = Box((10,0,0))
d = lib.gjk_distance( 
        [box1.cdata], ffi.NULL,
        [box2.cdata], ffi.NULL,
        ffi.NULL, ffi.NULL, 
        ffi.NULL,  0    )

对我来说，维度似乎以某种方式被转换了。因为它应该是一个包含 3 个元素项的 8 元素数组。我希望有人能在这里为我指明正确的方向。

Answer 1

显然 ptr_vertices = ffi.new("REAL[][3]", self.vertices ) 是要走的路。现在它似乎工作。

Answer 2

如果您要创建单个项目，请使用 ffi.new('REAL(*)[3]', [1, 2, 3])。 * 周围的括号很重要。

在C语言中，类型REAL(*)[3]表示指向实数数组（大小=3）的指针，而REAL*[3]表示指向实数的数组（大小=3）。有关详细信息，请参阅 C pointer to array/array of pointers disambiguation。

现在，您正在创建项目数组 CFFI expects an array type instead, 。这可以比较为：

ffi.new('int*', 1)    # ok
ffi.new('int[]', 1)   # wrong
ffi.new('int*', [1, 2, 3])    # wrong
ffi.new('int[]', [1, 2, 3])   # ok

ffi.new('REAL(*)[3]', [0.1, 0.2, 0.3])    # ok
ffi.new('REAL[][3]', [0.1, 0.2, 0.3])     # wrong
ffi.new('REAL(*)[3]', [[0.1, 0.2, 0.3], [0.4, 0.5, 0.6]])    # wrong
ffi.new('REAL[][3]', [[0.1, 0.2, 0.3], [0.4, 0.5, 0.6]])     # ok

在 CFFI 中创建 `REAL (* vertices)[DIM]` 类型的 cdata

Creating cdata of type `REAL (* vertices)[DIM]` in CFFI

c

python

cffi