Go语言中正确生成PGM文件：避免整数到字符串转换的常见陷阱

package main

import (
    "bufio"
    "fmt"
    "os"
    "strconv" // 引入 strconv 包
)

// 模拟一个简单的灰度图像数据
// img[0] 的长度代表图像宽度
// len(img) 代表图像高度
var img = make([][]int, 10) // 10行
func init() {
    for i := range img {
        img[i] = make([]int, 20) // 每行20列
        for j := range img[i] {
            img[i][j] = (i + j) * 10 % 256 // 填充一些灰度值
        }
    }
}

// writePGMIncorrect 演示了错误的整数到字符串转换方式
func writePGMIncorrect(filename string, width, height, maxVal int) error {
    fd, err := os.Create(filename)
    if err != nil {
        return fmt.Errorf("无法创建文件 %s: %w", filename, err)
    }
    defer fd.Close()

    wr := bufio.NewWriter(fd)

    // 错误的整数到字符串转换：string(width) 会将 width 解释为 Unicode 码点
    // 例如，如果 width=20，string(20) 会生成一个非打印字符 (U+0014)，而不是 "20"
    header := "P2\n" + string(width) + " " + string(height) + "\n" + string(maxVal) + "\n"

    if _, err := wr.WriteString(header); err != nil {
        return fmt.Errorf("写入文件头失败: %w", err)
    }

    // 模拟写入像素数据，同样使用错误的转换方式
    for i := 0; i < height; i++ {
        for j := 0; j < width; j++ {
            pixelStr := string(img[i][j]) // 错误的转换
            if _, err := wr.WriteString(pixelStr + " "); err != nil {
                return fmt.Errorf("写入像素数据失败: %w", err)
            }
        }
        if _, err := wr.WriteString("\n"); err != nil {
            return fmt.Errorf("写入换行符失败: %w", err)
        }
    }

    return wr.Flush()
}

// writePGMCorrect 演示了正确的整数到字符串转换方式
func writePGMCorrect(filename string, width, height, maxVal int) error {
    fd, err := os.Create(filename)
    if err != nil {
        return fmt.Errorf("无法创建文件 %s: %w", filename, err)
    }
    defer fd.Close()

    wr := bufio.NewWriter(fd)

    // 正确的整数到字符串转换：使用 strconv.Itoa
    header := "P2\n" + strconv.Itoa(width) + " " + strconv.Itoa(height) + "\n" + strconv.Itoa(maxVal) + "\n"

    if _, err := wr.WriteString(header); err != nil {
        return fmt.Errorf("写入文件头失败: %w", err)
    }

    // 模拟写入像素数据，使用正确的转换方式
    for i := 0; i < height; i++ {
        for j := 0; j < width; j++ {
            pixelStr := strconv.Itoa(img[i][j]) // 正确的转换
            if _, err := wr.WriteString(pixelStr + " "); err != nil {
                return fmt.Errorf("写入像素数据失败: %w", err)
            }
        }
        if _, err := wr.WriteString("\n"); err != nil {
            return fmt.Errorf("写入换行符失败: %w", err)
        }
    }

    return wr.Flush()
}

func main() {
    imageWidth := len(img[0])
    imageHeight := len(img)
    maxGrayValue := 255

    fmt.Println("尝试写入错误的PGM文件 (incorrect.pgm)...")
    err := writePGMIncorrect("incorrect.pgm", imageWidth, imageHeight, maxGrayValue)
    if err != nil {
        fmt.Printf("写入错误文件失败: %v\n", err)
    } else {
        fmt.Println("incorrect.pgm 已生成。请尝试用图像查看器打开，会发现是损坏的。")
    }

    fmt.Println("\n尝试写入正确的PGM文件 (correct.pgm)...")
    err = writePGMCorrect("correct.pgm", imageWidth, imageHeight, maxGrayValue)
    if err != nil {
        fmt.Printf("写入正确文件失败: %v\n", err)
    } else {
        fmt.Println("correct.pgm 已生成。请尝试用图像查看器打开，应能正常显示。")
    }
}