使用libpng库将像素数组写入row_pointer生成单通道灰度PNG的技术问题

我来帮你搞定这个libpng的图像写入问题，你的代码里有几个关键问题需要调整，咱们一步步理清楚：

首先，先明确三个核心错误 #

1. 内存泄漏+指针赋值错误
   你先给row_pointers[y]分配了内存，然后立刻把它指向imagePixels的地址，这样之前malloc的内存就直接丢失了，既造成内存泄漏，还可能因为imagePixels是栈上数组导致后续数据异常。

2. 灰度值映射搞反了
   libpng的PNG_COLOR_TYPE_GRAY模式下，0代表纯黑，255代表纯白，但你的数组里用1表示黑、0表示白，直接写入的话颜色会完全反过来，这肯定不是你想要的结果。

3. 像素数组长度不匹配
   5×5的图像需要25个像素，但你定义的imagePixels只有20个元素，这会导致越界访问，必须补全。

正确的实现方式 #

这里给你两种可行的方案，你可以根据需求选择：

方案1：分配每行内存并转换像素值（推荐，不修改原数组） #

这种方式会为每行单独分配内存，把原数组的像素值转换后再写入，不会改动原始的imagePixels：

// 先补全5×5的像素数组
byte imagePixels[25] = {1,0,1,0,1, 0,1,0,1,0, 1,0,1,0,1, 0,1,0,1,0, 1,0,1,0,1};

// 分配行指针数组
png_bytep *row_pointers = (png_bytep *)malloc(sizeof(png_bytep) * m_height);
if (!row_pointers) {
    // 这里记得处理内存分配失败的情况
    return;
}

for (int y = 0; y < m_height; y++) {
    // 按照libpng要求的行字节数分配内存
    row_pointers[y] = (png_byte *)malloc(png_get_rowbytes(png, infoPng));
    if (!row_pointers[y]) {
        // 内存分配失败时要释放已分配的资源，避免泄漏
        for (int i = 0; i < y; i++) free(row_pointers[i]);
        free(row_pointers);
        return;
    }

    // 逐像素转换并写入行数据
    for (int x = 0; x < m_width; x++) {
        int pixel_idx = y * m_width + x;
        // 转换规则：用户定义的1→黑(0)，0→白(255)
        row_pointers[y][x] = imagePixels[pixel_idx] == 1 ? 0 : 255;
    }
}

// 写入图像
png_write_image(png, row_pointers);

// 写完后记得释放内存！
for (int y = 0; y < m_height; y++) free(row_pointers[y]);
free(row_pointers);

方案2：直接指向原数组（需先修正灰度值） #

如果不想额外分配内存，可以直接让行指针指向原数组的对应行，但要先把原数组的灰度值修正为符合libpng的规范：

// 先修正原数组的灰度值
byte imagePixels[25] = {1,0,1,0,1, 0,1,0,1,0, 1,0,1,0,1, 0,1,0,1,0, 1,0,1,0,1};
for (int i = 0; i < 25; i++) {
    imagePixels[i] = imagePixels[i] == 1 ? 0 : 255;
}

// 分配行指针数组
png_bytep *row_pointers = (png_bytep *)malloc(sizeof(png_bytep) * m_height);
if (!row_pointers) {
    // 处理错误
    return;
}

// 直接指向原数组的每行起始地址
for (int y = 0; y < m_height; y++) {
    row_pointers[y] = &imagePixels[y * m_width];
}

// 写入图像
png_write_image(png, row_pointers);

// 释放行指针数组（注意原数组如果是栈上的，不需要释放）
free(row_pointers);

额外注意事项 #

• 调用png_set_IHDR时，要确保bit_depth设为8（因为你用的是8位灰度，0-255的范围），和PNG_COLOR_TYPE_GRAY匹配。
• 所有动态分配的内存（包括row_pointers和每行的内存）都要记得释放，避免内存泄漏。
• 一定要处理内存分配失败的情况，不然程序可能崩溃。

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