Go语言图像处理：理解image.Color接口与自定义颜色实现

type Color interface {
    RGBA() (r, g, b, a uint32)
}

package main

import (
    "fmt"
    "image"
    "image/color" // 引入 color 包
    _ "image/gif"
    _ "image/jpeg"
    _ "image/png"
    "os"
)

func main() {
    reader, err := os.Open("test-image.jpg") // 假设存在 test-image.jpg 文件
    if err != nil {
        fmt.Fprintf(os.Stderr, "打开文件失败: %v\n", err)
        return
    }
    defer reader.Close()

    img, _, err := image.Decode(reader)
    if err != nil {
        fmt.Fprintf(os.Stderr, "解码图像失败: %v\n", err)
        return
    }

    bounds := img.Bounds()

    // 遍历图像像素（仅以左上角像素为例）
    if bounds.Min.X <= 0 && bounds.Min.Y <= 0 {
        pixel := img.At(0, 0)
        r, g, b, a := pixel.RGBA()

        // 计算平均亮度值，用于创建灰度颜色
        // 注意：r, g, b, a 已经是 0-0xffff 范围的值
        averaged := (r + g + b) / 3

        // 使用 image.Gray 类型创建灰度颜色对象
        // image.Gray 结构体只有一个 Y 字段，直接赋值即可
        grayColor := color.Gray{Y: uint8(averaged >> 8)} // 将 uint32 范围值转换为 uint8 范围

        // 验证：调用 RGBA() 方法
        gr, gg, gb, ga := grayColor.RGBA()
        fmt.Printf("原始像素(0,0) RGBA: (%d, %d, %d, %d)\n", r, g, b, a)
        fmt.Printf("灰度颜色 RGBA: (%d, %d, %d, %d)\n", gr, gg, gb, ga)
    }
}

package main

import (
    "fmt"
    "image/color" // 引入 color 包
)

// MyGray 是一个自定义的灰度颜色类型
type MyGray struct {
    Y uint32 // 存储灰度值，使用 uint32 以匹配 RGBA() 的返回值范围
}

// RGBA 方法实现了 image.Color 接口
func (mg *MyGray) RGBA() (r, g, b, a uint32) {
    // 灰度颜色，R=G=B=Y，Alpha为不透明
    return mg.Y, mg.Y, mg.Y, 0xffff // 0xffff 表示完全不透明
}

// FromRGBA 是一个辅助方法，用于从 RGBA 值初始化 MyGray
// 注意：这个方法不是 image.Color 接口的一部分，只是为了方便自定义类型的使用
func (mg *MyGray) FromRGBA(r, g, b, a uint32) {
    mg.Y = (r + g + b) / 3 // 简单平均作为灰度值
    // 也可以根据需求实现更复杂的灰度转换算法，例如加权平均
}

func main() {
    // 模拟一个原始像素的 RGBA 值
    originalR, originalG, originalB, originalA := uint32(0x1000), uint32(0x8000), uint32(0xC000), uint32(0xFFFF)

    // 创建 MyGray 实例
    myGrayColor := &MyGray{}

    // 使用自定义的 FromRGBA 辅助方法初始化
    myGrayColor.FromRGBA(originalR, originalG, originalB, originalA)

    // 现在 myGrayColor 是一个 image.Color 类型，可以调用 RGBA() 方法
    r, g, b, a := myGrayColor.RGBA()

    fmt.Printf("原始 RGBA: (%d, %d, %d, %d)\n", originalR, originalG, originalB, originalA)
    fmt.Printf("自定义 MyGray 颜色 RGBA: (%d, %d, %d, %d)\n", r, g, b, a)

    // 也可以直接创建并赋值
    anotherGray := &MyGray{Y: 0x7FFF}
    ar, ag, ab, aa := anotherGray.RGBA()
    fmt.Printf("直接赋值的 MyGray 颜色 RGBA: (%d, %d, %d, %d)\n", ar, ag, ab, aa)

    // 验证 MyGray 是否满足 image.Color 接口
    var c color.Color = myGrayColor
    fmt.Printf("myGrayColor 是否为 image.Color 类型: %T\n", c)
}