在 C 中序列化原始图像缓冲区(rgb 像素)，在 Python 中反序列化

Question

我想序列化原始图像数据，即uint16数组，并使用zmq将其发送到python。我考虑使用 msgPack-c，但我发现的唯一方法是给定如何使用 msgpack-c 正确解压缩和提取数据？

如果我遵循这种方法，我必须将每个元素分别打包在我的 C 数组中，这将使它非常慢。

有人可以指出正确的方向吗？

Answer 1

你可以按原样从 c 端发送uint16_t数组，并使用ctypes模块在 python 代码中访问它。

发送 c 代码：

#include <stdint.h>
#include <stdio.h>
#include <zmq.h>
#define IMAGE_SIZE (256 * 256)
unsigned checksum(uint16_t* data, int len) {
    unsigned s = 0;
    for (int i = 0; i < len; ++i) {
        s += data[i];
    }
    return s;
}
int main() {
    uint16_t image[IMAGE_SIZE];
    printf("image checksum: %in", checksum(image, IMAGE_SIZE));
    void* context = zmq_ctx_new();
    void* push = zmq_socket(context, ZMQ_PUSH);
    zmq_connect(push, "tcp://127.0.0.1:5555");
    zmq_send(push, image, IMAGE_SIZE * sizeof(uint16_t), 0);
    zmq_close(push);
    zmq_ctx_destroy(context);
    return 0;
}

接收蟒蛇代码：

from ctypes import c_uint16
import zmq
IMAGE_SIZE = 256 * 256
Image = c_uint16 * IMAGE_SIZE # corresponds to uint16_t[IMAGE_SIZE]
context = zmq.Context(1)
pull = zmq.Socket(context, zmq.PULL)
pull.bind("tcp://127.0.0.1:5555")
message = pull.recv()
image = Image.from_buffer_copy(message)
# This should print the same number as the sending code
# Note that it is different from sum(message)
print(sum(image))