如何修改Go代码将自定义单字节像素图像保存为8位灰度PNG？

我有一段用Go语言编写的代码，能够将RGBA图像保存为PNG文件，它可以正常处理每个像素包含4字节RGBA分量的常规RGBA图像。代码如下：

import (
    "bufio"
    "image"
    "image/png"
    "os"
)

// SaveToPngFile 创建并以PNG格式保存图像到文件
func SaveToPngFile(filePath string, m image.Image) error {
    // 创建文件
    f, err := os.Create(filePath)
    if err != nil {
        return err
    }
    defer f.Close()
    // 基于文件创建Writer
    b := bufio.NewWriter(f)
    // 将图像写入缓冲区
    err = png.Encode(b, m)
    if err != nil {
        return err
    }
    err = b.Flush()
    if err != nil {
        return err
    }
    return nil
}

我手动修改了RGBA图像的像素缓冲区，现在每个像素只对应1字节，且这个字节仅包含原RGBA的R分量。请问要如何修改上述代码，才能将修改后的图像保存为可以正常显示的8位灰度单色PNG？

核心思路是把你修改后的单字节像素数据包装成Go标准库的image.Gray类型，因为png.Encode原生支持该类型，会自动生成符合规范的8位灰度PNG文件。

具体实现步骤 #

1. 用修改后的单字节像素切片、图像宽高创建image.Gray实例：

   • Pix字段直接传入你的单字节像素数据
   • Stride设为图像宽度（单字节像素每行的字节数等于宽度）
   • Rect定义图像的尺寸范围

2. 原SaveToPngFile函数无需大幅修改，直接传入image.Gray实例即可完成保存。

示例代码 #

import (
    "bufio"
    "image"
    "image/png"
    "os"
)

// SaveToPngFile 创建并以PNG格式保存图像到文件（兼容RGBA和灰度图像）
func SaveToPngFile(filePath string, m image.Image) error {
    f, err := os.Create(filePath)
    if err != nil {
        return err
    }
    defer f.Close()
    b := bufio.NewWriter(f)
    err = png.Encode(b, m)
    if err != nil {
        return err
    }
    return b.Flush()
}

// SaveGrayPixelsToPng 将单字节灰度像素数据保存为PNG
func SaveGrayPixelsToPng(filePath string, grayPix []byte, width, height int) error {
    grayImg := &image.Gray{
        Pix:    grayPix,
        Stride: width,
        Rect:   image.Rect(0, 0, width, height),
    }
    return SaveToPngFile(filePath, grayImg)
}

关键说明 #

• image.Gray的Stride必须匹配每行像素的实际字节数，单字节灰度场景下直接设为图像宽度即可
• png.Encode处理image.Gray时，会自动生成8位灰度格式的PNG，符合单色灰度图像的规范

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