Go语言中灵活创建与管理颜色：深入理解image.Color接口及其实现

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

type Gray struct {
    Y uint8
}

package main

import (
    "fmt"
    "image"
    "image/color" // 导入 color 包
    _ "image/jpeg" // 注册 JPEG 格式
    _ "image/png"  // 注册 PNG 格式
    "os"
)

func main() {
    // 假设存在一个名为 "test-image.jpg" 的图像文件
    // 为了运行此代码，请确保有一个图像文件在相同目录下
    reader, err := os.Open("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() // 获取 16 位分量 (0-0xffff)

        // 计算灰度平均值。注意 r, g, b 都是 uint32 类型。
        // 为了转换为 uint8 的 Gray 结构体，我们需要进行缩放。
        // 简单的平均值转换通常是将 16 位值右移 8 位得到 8 位值。
        // 更精确的亮度计算公式是 Y = 0.299*R + 0.587*G + 0.114*B
        // 这里我们使用简单的平均值并进行缩放
        avg16bit := (r + g + b) / 3 // 结果仍然是 uint32 (0-0xffff)

        // 将 16 位灰度值转换为 8 位 (0-255)
        grayValue8bit := uint8(avg16bit >> 8) 

        // 创建一个 image.Gray 类型的颜色对象
        grayColor := color.Gray{Y: grayValue8bit}

        fmt.Printf("原始像素 (0,0) 的 RGBA (16位): R=%d, G=%d, B=%d, A=%d\n", r, g, b, a)
        fmt.Printf("转换后的灰度值 (8位): %d\n", grayColor.Y)

        // 验证 grayColor 是否满足 image.Color 接口
        // 我们可以调用它的 RGBA() 方法
        gr, gg, gb, ga := grayColor.RGBA()
        fmt.Printf("grayColor.RGBA() 返回值 (16位): R=%d, G=%d, B=%d, A=%d\n", gr, gg, gb, ga)
    }
}

package main

import (
    "fmt"
    "image"
    "image/color" // 导入 color 包
    _ "image/jpeg" // 注册 JPEG 格式
    _ "image/png"  // 注册 PNG 格式
    "os"
)

// MyGray 是一个自定义的灰度颜色类型，存储 16 位灰度值
type MyGray struct {
    Y uint32 // 存储 16 位灰度值 (0-0xffff)
}

// RGBA 方法使 MyGray 满足 image.Color 接口
func (g *MyGray) RGBA() (r, gr, b, a uint32) {
    // 灰度颜色，R, G, B 值相同，Alpha 设为完全不透明
    return g.Y, g.Y, g.Y, 0xffff 
}

// FromRGBA 是一个辅助方法，用于从 RGBA 值初始化 MyGray
// 注意：这不是 image.Color 接口的一部分，只是为了方便我们自定义类型的使用
func (g *MyGray) FromRGBA(r, gr, b, a uint32) {
    // 简单的平均值作为灰度值
    g.Y = (r + gr + b) / 3 
}

func main() {
    reader, err := os.Open("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() // 获取 16 位分量 (0-0xffff)

        // 创建并初始化自定义的 MyGray 颜色对象
        myGrayColor := &MyGray{} // 注意使用指针类型
        myGrayColor.FromRGBA(r, g, b, a)

        fmt.Printf("原始像素 (0,0) 的 RGBA (16位): R=%d, G=%d, B=%d, A=%d\n", r, g, b, a)
        fmt.Printf("自定义 MyGray 颜色 (16位灰度): %d\n", myGrayColor.Y)

        // 验证 myGrayColor 是否满足 image.Color 接口
        mr, mg, mb, ma := myGrayColor.RGBA()
        fmt.Printf("myGrayColor.RGBA() 返回值 (16位): R=%d, G=%d, B=%d, A=%d\n", mr, mg, mb, ma)
    }
}