Go语言中创建PGM文件：正确处理整数到字符串的转换

P2
# My Image
200 150
255

package main

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

// 假设 img 是一个 [][]int 类型的二维切片，代表灰度图像数据
// 假设 maxValue 是图像的最大灰度值，例如 255

func writePGM(filename string, img [][]int, maxValue int) error {
    if len(img) == 0 || len(img[0]) == 0 {
        return fmt.Errorf("image data is empty")
    }

    width := len(img[0])
    height := len(img)

    fd, err := os.Create(filename)
    if err != nil {
        return fmt.Errorf("failed to create file %s: %w", filename, err)
    }
    defer fd.Close() // 确保文件在函数结束时关闭

    wr := bufio.NewWriter(fd)

    // 构建 PGM 文件头部
    // 使用 strconv.Itoa 将整数转换为字符串
    header := "P2\n" +
        strconv.Itoa(width) + " " + strconv.Itoa(height) + "\n" +
        strconv.Itoa(maxValue) + "\n"

    if _, err := wr.WriteString(header); err != nil {
        return fmt.Errorf("failed to write PGM header: %w", err)
    }

    // 写入图像像素数据
    for y := 0; y < height; y++ {
        for x := 0; x < width; x++ {
            if _, err := wr.WriteString(strconv.Itoa(img[y][x])); err != nil {
                return fmt.Errorf("failed to write pixel data: %w", err)
            }
            if x < width-1 {
                if _, err := wr.WriteString(" "); err != nil { // 像素之间用空格分隔
                    return fmt.Errorf("failed to write pixel separator: %w", err)
                }
            }
        }
        if _, err := wr.WriteString("\n"); err != nil { // 每行像素后换行
            return fmt.Errorf("failed to write newline after row: %w", err)
        }
    }

    return wr.Flush() // 确保所有缓冲数据写入文件
}

func main() {
    // 示例图像数据
    sampleImg := [][]int{
        {0, 50, 100, 150, 200, 250},
        {250, 200, 150, 100, 50, 0},
        {0, 0, 0, 255, 255, 255},
    }
    maxVal := 255

    err := writePGM("output.pgm", sampleImg, maxVal)
    if err != nil {
        fmt.Fprintf(os.Stderr, "Error writing PGM file: %v\n", err)
        os.Exit(1)
    }
    fmt.Println("PGM file 'output.pgm' created successfully.")
}