热门文章
 当前位置：技术文章 > 编程面试题 >

Python结构中的动态数组和结构

问题内容：

我正在尝试使用ctypes在python中实现此C结构：

struct _rows {
    int cols_count;
    char *cols[];
}

struct _unit {
    int rows_count;
    struct _rows *rows;
}

int my_func(struct _unit *param);

问题是_rows.cols是char指针的动态大小数组，而_unit.rows是_rows结构的动态大小数组。如何在python中使用ctypes实现此功能？

我能够定义一个函数，该函数将返回具有可变数量的char指针的_rows结构：

def get_row(cols):
    class Row(ctypes.Structure):
        _fields_ = [("cols_count", ctypes.c_int), 
                    ("cols", ctypes.c_char_p * cols)
                   ]

我不知道该怎么做，它有点模糊，ctypes文档也无济于事。


问题答案：

我正在对OP的需求做一些假设，如果有更简单的方法可以满足我的要求，那是我的想法：

演示

import string
from ctypes import Structure,c_int,c_char_p,POINTER,cast,pointer,byref,CDLL

class Row(Structure):
    _fields_ = [('cols_count', c_int), 
                ('cols', POINTER(c_char_p))]
    def __init__(self,cols):
        self.cols_count = cols
        # Allocate an array of character pointers
        pc = (c_char_p * cols)()
        self.cols = cast(pc,POINTER(c_char_p))

class Unit(Structure):
    _fields_ = [('rows_count', c_int),
                ('rows',POINTER(Row))]
    def __init__(self,rows,cols):
        self.rows_count = rows
        # Allocate an array of Row structures.
        # This does NOT call __init__.
        pr = (Row * rows)()
        # Call init manually with the column size.
        for r in pr:
            r.__init__(cols)
        self.rows = cast(pr,POINTER(Row))

unit = Unit(2,3)

# Stuff some strings ('aaaaa','bbbbb',etc.)
for i in xrange(unit.rows_count):
    for j in xrange(unit.rows[i].cols_count):
        unit.rows[i].cols[j] = string.ascii_lowercase[i*5+j]*5

dll = CDLL('test.dll')
dll.my_func(byref(unit))

测试

#include <stdio.h>

struct _rows {
    int cols_count;
    char **cols;
};

struct _unit {
    int rows_count;
    struct _rows *rows;
};

__declspec(dllexport) int my_func(struct _unit *param)
{
    int i,j;
    for(i=0;i<param->rows_count;i++)
        for(j=0;j<param->rows[i].cols_count;j++)
            printf("%d,%d = %s\n",i,j,param->rows[i].cols[j]);
    return 0;
}

生成文件

与Visual Studio 2010一起编译。

test.dll: test.c
    cl /W4 /LD test.c

输出量

0,0 = aaaaa
0,1 = bbbbb
0,2 = ccccc
1,0 = fffff
1,1 = ggggg
1,2 = hhhhh