如何在Go的image包中新增3位像素格式且不破坏原有结构

Go语言的image包为其Gray图像定义了如下方法：

func (p *Gray) ColorModel() color.Model { return color.GrayModel }

随后，ColorModel()方法会在switch语句中被用来设置cb变量：

switch m.ColorModel() {
case color.GrayModel:
    e.cb = cbG8 // <-- cb is set **
case color.Gray16Model:
    e.cb = cbG16
// ...
}

最终，cb变量会被用来设置bitsPerPixel变量：

bitsPerPixel := 0

switch cb {
case cbG8:
    bitsPerPixel = 8
case cbTC8:
    bitsPerPixel = 24
case cbP8:
    bitsPerPixel = 8
case cbP4:
    bitsPerPixel = 4
case cbP2:
    bitsPerPixel = 2
case cbP1:
    bitsPerPixel = 1
case cbTCA8:
    bitsPerPixel = 32
case cbTC16:
    bitsPerPixel = 48
case cbTCA16:
    bitsPerPixel = 64
case cbG16:
    bitsPerPixel = 16
}

如何在遵循单一职责原则、保持实现简洁的前提下，为Go语言image包的源码新增一个对应3位每像素（bits per pixel）的分支，同时避免对整个image包造成破坏？希望能得到创新且低侵入性的实现思路。

低侵入实现方案 #

1. 自定义3位灰度图像与颜色模型

完全不修改标准库image源码，在独立包中实现符合image.Image接口的3位灰度图像类型，配套专属颜色模型：

package myimage

import "image"
import "image/color"

// Gray3 存储3位灰度像素（取值0-7）
type Gray3 struct {
	Pix    []uint8
	Stride int
	Rect   image.Rectangle
}

// 实现image.Image核心接口
func (p *Gray3) ColorModel() color.Model { return Gray3Model }
func (p *Gray3) Bounds() image.Rectangle { return p.Rect }
func (p *Gray3) At(x, y int) color.Color {
	if !image.Point{x, y}.In(p.Rect) {
		return Gray3Color{}
	}
	i := p.PixOffset(x, y)
	return Gray3Color{p.Pix[i] & 0x07} // 仅保留低3位有效数据
}

func (p *Gray3) PixOffset(x, y int) int {
	return (y-p.Rect.Min.Y)*p.Stride + (x-p.Rect.Min.X)
}

// Gray3Color 3位灰度颜色类型
type Gray3Color struct{ Y uint8 }
func (c Gray3Color) RGBA() (r, g, b, a uint32) {
	y := uint32(c.Y)
	y |= y << 5 // 将3位值扩展为8位灰度用于RGBA转换
	return y, y, y, 0xFFFF
}

// Gray3Model 3位灰度颜色转换模型
var Gray3Model = color.ModelFunc(func(c color.Color) color.Color {
	gray, ok := color.GrayModel.Convert(c).(color.Gray)
	if !ok {
		return Gray3Color{}
	}
	// 把8位灰度压缩到3位
	return Gray3Color{uint8(gray.Y >> 5)}
})

2. 扩展PNG写入逻辑的适配层

不修改image/png源码，通过包装标准库写入器，新增3位像素的处理分支：

package mypng

import (
	"image"
	"image/png"
	"io"
	"myimage"
)

// Writer 包装标准库PNG写入器，添加3位灰度支持
type Writer struct {
	png.Writer
}

// 3位灰度图像的PNG编码回调
func cbG3(w io.Writer, m image.Image) error {
	b := m.Bounds()
	width, height := b.Dx(), b.Dy()
	// 实现3位像素的打包逻辑：每8个3位像素占用3字节（8*3=24bits）
	// 参考标准库cbG8的实现，调整位操作细节完成编码
	// 这里省略具体编码逻辑，核心是处理像素位的压缩与对齐
	return nil
}

func (w *Writer) WriteImage(m image.Image) error {
	// 判断是否为自定义3位灰度图像
	if _, ok := m.(*myimage.Gray3); ok {
		return cbG3(w.W, m)
	}
	// 其他类型交给标准库处理
	return w.Writer.WriteImage(m)
}

3. 方案优势

• 零侵入标准库：所有扩展逻辑在自定义包中实现，完全不修改image和image/png源码，避免破坏原有功能。
• 职责单一：Gray3仅负责3位灰度图像的存储与颜色转换，mypng.Writer仅负责适配编码，标准库原有逻辑不受影响。
• 兼容原有接口：自定义类型严格实现image.Image接口，可直接替换原有灰度图像类型在业务代码中使用，无需修改上层逻辑。

内容的提问来源于stack exchange，提问作者Megidd