RGBA到灰度的并行Golang

package main

import (
    "image"
    "image/color"
    "image/png"
    "log"
    "os"
    "runtime"
    "sync"
)

func main() {
    // 读取RGBA图像
    file, err := os.Open("input.png")
    if err != nil {
        log.Fatal(err)
    }
    defer file.Close()

    img, _, err := image.Decode(file)
    if err != nil {
        log.Fatal(err)
    }

    bounds := img.Bounds()
    width, height := bounds.Max.X, bounds.Max.Y

    // 创建灰度图像
    grayImg := image.NewGray(image.Rect(0, 0, width, height))

    // 设置并发处理的最大线程数
    runtime.GOMAXPROCS(runtime.NumCPU())

    // 使用WaitGroup来等待所有goroutine完成
    var wg sync.WaitGroup
    wg.Add(height)

    // 并行处理每个像素点
    for y := 0; y < height; y++ {
        go func(y int) {
            defer wg.Done()

            for x := 0; x < width; x++ {
                rgba := img.At(x, y).(color.RGBA)
                gray := color.Gray{
                    Y: uint8(0.299*float64(rgba.R) + 0.587*float64(rgba.G) + 0.114*float64(rgba.B)),
                }
                grayImg.Set(x, y, gray)
            }
        }(y)
    }

    // 等待所有goroutine完成
    wg.Wait()

    // 保存灰度图像
    outputFile, err := os.Create("output.png")
    if err != nil {
        log.Fatal(err)
    }
    defer outputFile.Close()

    err = png.Encode(outputFile, grayImg)
    if err != nil {
        log.Fatal(err)
    }
}