如何将int8数据打包为C++变量初始化字符串？解包符号丢失排查

需求与问题 #

需要用Python将int8类型的numpy数组转换为可用于C++ uint32数组初始化的字符串（格式如uint32 arr[]={0x12345678, ...};），uint32仅作为数据容器，后续需拆回int8变量。

当前编写的打包代码能将4个int8打包为一个uint32，但解包后原负数的符号丢失：原数组[-10, -16, 15, 8]解包后得到[246, 240, 15, 8]，无法还原原始的负数。

问题原因 #

解包时仅提取了uint32中每个8位段的无符号数值，但没有处理int8的补码规则：int8的最高位是符号位，当提取的8位值≥0x80（即128）时，代表这是一个负数，需要将其转换为对应的补码负数（即减去256）。

修复后的代码 #

import numpy as np

def pack_int8_to_uint32(int8_values):
    uint32_data = 0
    for i, value in enumerate(int8_values):
        # 将int8转为无符号8位值后移位，确保打包的二进制正确
        uint32_data |= (value & 0xFF) << (8 * i)
    return uint32_data

def unpack_uint32_to_int8(uint32_data):
    int8_values = []
    for i in range(4):
        # 提取8位无符号值
        raw_val = (uint32_data >> (8 * i)) & 0xFF
        # 处理补码：若为负数则转换为int8范围
        int8_val = raw_val if raw_val < 0x80 else raw_val - 256
        int8_values.append(int8_val)
    return int8_values

# 测试数据
int8_values = np.array([-10, -16, 15, 8], dtype=np.int8).tolist()
packed_uint32 = pack_int8_to_uint32(int8_values)
unpacked_int8_values = unpack_uint32_to_int8(packed_uint32)

# 生成C++初始化字符串
cpp_init_str = f"uint32 arr[] = {{0x{packed_uint32:X}}};".replace("X", "x")

# 输出结果
print("原始int8数组:", int8_values)
print("解包后int8数组:", unpacked_int8_values)
print("打包后的uint32十六进制:", hex(packed_uint32))
print("C++初始化字符串:", cpp_init_str)

运行结果 #

原始int8数组: [-10, -16, 15, 8]
解包后int8数组: [-10, -16, 15, 8]
打包后的uint32十六进制: 0x80ff0f6
C++初始化字符串: uint32 arr[] = {0x80ff0f6};

补充说明 #

• 打包函数无需修改：value & 0xFF已经将int8的补码转换为无符号的8位整数，移位后打包到uint32的对应字节，二进制数据是正确的。
• 若需要批量处理多个int8（数量为4的倍数），可以循环每4个元素打包一次，生成C++数组的多个初始化值。

内容的提问来源于stack exchange，提问作者zjnyly