ofdgo/internal/base/image/image.go

// Copyright 2025-2026 肖其顿 (XIAO QI DUN)
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package image

import (
	"bytes"
	"context"
	"encoding/binary"
	"fmt"
	"image"
	"image/color"
	_ "image/gif"
	"io"

	_ "github.com/xiaoqidun/j2kgo"
	_ "github.com/xiaoqidun/jbig2"
	_ "golang.org/x/image/bmp"
	"golang.org/x/image/tiff"
)

// transparentEdgeImage 共享只读像素并按需计算透明边缘和采样边距
type transparentEdgeImage struct {
	source image.Image
	rect   image.Rectangle
}

// transparentEdgeImage64 按需补齐十六位像素边缘，保留原始只读图像
type transparentEdgeImage64 struct {
	source image.Image
	rect   image.Rectangle
}

// Bounds 返回含采样边距的图像边界
// 返回: image.Rectangle 图像边界
func (s *transparentEdgeImage) Bounds() image.Rectangle { return s.rect }

// ColorModel 返回非预乘颜色模型
// 返回: color.Model 颜色模型
func (s *transparentEdgeImage) ColorModel() color.Model { return color.NRGBAModel }

// NRGBAAt 读取原始像素或按邻域补齐透明边缘，边界外透明
// 入参: x 横向坐标, y 纵向坐标
// 返回: color.NRGBA 非预乘颜色
func (s *transparentEdgeImage) NRGBAAt(x, y int) color.NRGBA {
	if !image.Pt(x, y).In(s.rect) {
		return color.NRGBA{}
	}
	b := s.source.Bounds()
	sx, sy := b.Min.X+min(max(x-1, 0), b.Dx()-1), b.Min.Y+min(max(y-1, 0), b.Dy()-1)
	c := NRGBAAt(s.source, sx, sy)
	if c.A == 0 {
		if edge, ok := transparentEdgeColor(s.source, sx, sy); ok {
			c = edge
		}
	}
	if x == 0 || y == 0 || x == s.rect.Max.X-1 || y == s.rect.Max.Y-1 {
		c = transparentPaddingColor(c)
	}
	return c
}

// At 返回非预乘像素颜色
// 入参: x 横向坐标, y 纵向坐标
// 返回: color.Color 像素颜色
func (s *transparentEdgeImage) At(x, y int) color.Color { return s.NRGBAAt(x, y) }

// RGBA64At 返回预乘十六位像素，供采样器无装箱读取
// 入参: x 横向坐标, y 纵向坐标
// 返回: color.RGBA64 预乘颜色
func (s *transparentEdgeImage) RGBA64At(x, y int) color.RGBA64 {
	r, g, b, a := s.NRGBAAt(x, y).RGBA()
	return color.RGBA64{R: uint16(r), G: uint16(g), B: uint16(b), A: uint16(a)}
}

// Bounds 返回含采样边距的图像边界
// 返回: image.Rectangle 图像边界
func (s *transparentEdgeImage64) Bounds() image.Rectangle { return s.rect }

// ColorModel 返回十六位非预乘颜色模型
// 返回: color.Model 颜色模型
func (s *transparentEdgeImage64) ColorModel() color.Model { return color.NRGBA64Model }

// NRGBA64At 读取十六位原始颜色，透明邻域使用最小非零透明度保留边缘颜色
// 入参: x 横向坐标, y 纵向坐标
// 返回: color.NRGBA64 非预乘颜色
func (s *transparentEdgeImage64) NRGBA64At(x, y int) color.NRGBA64 {
	if !image.Pt(x, y).In(s.rect) {
		return color.NRGBA64{}
	}
	b := s.source.Bounds()
	sx, sy := b.Min.X+min(max(x-1, 0), b.Dx()-1), b.Min.Y+min(max(y-1, 0), b.Dy()-1)
	c := NRGBA64At(s.source, sx, sy)
	if c.A == 0 {
		var best color.NRGBA64
		for ny := max(sy-1, b.Min.Y); ny < min(sy+2, b.Max.Y); ny++ {
			for nx := max(sx-1, b.Min.X); nx < min(sx+2, b.Max.X); nx++ {
				v := NRGBA64At(s.source, nx, ny)
				if v.A > best.A {
					best = v
				}
			}
		}
		if best.A != 0 {
			best.A = 1
			c = best
		}
	}
	if x == 0 || y == 0 || x == s.rect.Max.X-1 || y == s.rect.Max.Y-1 {
		c.A = 1
	}
	return c
}

// At 返回非预乘十六位像素颜色
// 入参: x 横向坐标, y 纵向坐标
// 返回: color.Color 像素颜色
func (s *transparentEdgeImage64) At(x, y int) color.Color { return s.NRGBA64At(x, y) }

// RGBA64At 返回保留原采样精度的预乘颜色
// 入参: x 横向坐标, y 纵向坐标
// 返回: color.RGBA64 预乘颜色
func (s *transparentEdgeImage64) RGBA64At(x, y int) color.RGBA64 {
	r, g, b, a := s.NRGBA64At(x, y).RGBA()
	return color.RGBA64{R: uint16(r), G: uint16(g), B: uint16(b), A: uint16(a)}
}

// Decode 解码图片数据
// 入参: data 图片数据
// 返回: image.Image 图片对象, string 图片格式, error 错误信息
func Decode(data []byte) (image.Image, string, error) {
	return DecodeContext(context.Background(), data)
}

// DecodeContext 解码图片并传递取消状态，PNG和JPEG仍保留惰性像素解码
// 入参: ctx 取消上下文, data 只读图片编码
// 返回: image.Image 图片对象, string 图片格式, error 解码或取消错误
func DecodeContext(ctx context.Context, data []byte) (img image.Image, format string, err error) {
	if err := ctx.Err(); err != nil {
		return nil, "", err
	}
	defer func() {
		if cancelErr := ctx.Err(); cancelErr != nil {
			img, err = nil, cancelErr
		}
	}()
	if IsJPEG(data) || IsPNG(data) {
		img, err := newEncodedImageContext(ctx, data)
		if err != nil {
			return nil, "", err
		}
		return img, img.format, nil
	}
	if IsTIFF(data) {
		layout, err := ReadTIFFLayoutContext(ctx, bytes.NewReader(data))
		if err != nil {
			return nil, "tiff", err
		}
		img, err := DecodeTIFF(ctx, data, layout)
		return img, "tiff", err
	}
	img, format, err = image.Decode(contextInput{ReadCloser: io.NopCloser(bytes.NewReader(data)), context: ctx})
	if err == nil {
		return img, NormalizeFormat(format), nil
	}
	if ctx.Err() != nil {
		return nil, "", ctx.Err()
	}
	if img, err := DecodeBMPContext(ctx, data); err == nil {
		return img, "bmp", nil
	}
	return nil, "", err
}

// DecodePixelsContext 完整解码图片，确保像素可读取并传递取消状态
// 入参: ctx 取消上下文, data 图片编码数据
// 返回: image.Image 解码像素, string 图片格式, error 解码或取消错误
func DecodePixelsContext(ctx context.Context, data []byte) (image.Image, string, error) {
	img, format, err := DecodeContext(ctx, data)
	if err != nil {
		return nil, format, err
	}
	pixels, err := PixelDataContext(ctx, img)
	return pixels, format, err
}

// IsJPEG 判断是否为JPEG图片数据
// 入参: data 图片数据
// 返回: bool 是否为JPEG图片数据
func IsJPEG(data []byte) bool {
	return len(data) >= 2 && data[0] == 0xFF && data[1] == 0xD8
}

// IsPNG 判断是否为PNG图片数据
// 入参: data 图片数据
// 返回: bool 是否为PNG图片数据
func IsPNG(data []byte) bool {
	return bytes.HasPrefix(data, []byte("\x89PNG\r\n\x1a\n"))
}

// DecodeConfigData 解码图片尺寸
// 入参: data 图片数据
// 返回: image.Config 图片尺寸, string 图片格式, error 错误信息
func DecodeConfigData(data []byte) (image.Config, string, error) {
	return DecodeConfigDataContext(context.Background(), data)
}

// DecodeConfigDataContext 读取图片尺寸和颜色模型，传递布局解析中的取消状态
// 入参: ctx 取消上下文, data 图片数据
// 返回: image.Config 图片配置, string 图片格式, error 读取或取消错误
func DecodeConfigDataContext(ctx context.Context, data []byte) (config image.Config, format string, err error) {
	if err := ctx.Err(); err != nil {
		return image.Config{}, "", err
	}
	defer func() {
		if cancelErr := ctx.Err(); cancelErr != nil {
			config, err = image.Config{}, cancelErr
		}
	}()
	if IsTIFF(data) {
		layout, err := ReadTIFFLayoutContext(ctx, bytes.NewReader(data))
		if err != nil {
			return image.Config{}, "tiff", err
		}
		profile, err := readTIFFColorProfile(ctx, data, layout)
		if err != nil {
			return image.Config{}, "tiff", err
		}
		var cfg image.Config
		if layout.needsSampleView() {
			img, _, err := tiffPlaneDirectories(ctx, data, layout)
			if err != nil {
				return image.Config{}, "tiff", err
			}
			cfg.ColorModel = img.ColorModel()
		} else {
			input := newTIFFDirectoryInput(data, layout.Offset)
			input.ctx = ctx
			cfg, err = tiff.DecodeConfig(input)
			if err != nil {
				return image.Config{}, "tiff", err
			}
		}
		if profile != nil || layout.Transparency != nil {
			cfg.ColorModel = color.NRGBA64Model
		}
		if layout.Transparency != nil {
			if _, err := readTIFFColorProfile(ctx, data, *layout.Transparency); err != nil {
				return image.Config{}, "tiff", err
			}
			if _, _, err := tiffPlaneDirectories(ctx, data, *layout.Transparency); err != nil {
				return image.Config{}, "tiff", err
			}
		}
		size := layout.DisplaySize()
		cfg.Width, cfg.Height = size.X, size.Y
		return cfg, "tiff", nil
	}
	cfg, format, err := image.DecodeConfig(contextInput{ReadCloser: io.NopCloser(bytes.NewReader(data)), context: ctx})
	if err == nil {
		return cfg, NormalizeFormat(format), nil
	}
	if cfg, err := decodeBMPConfig(data); err == nil {
		return cfg, "bmp", nil
	}
	return image.Config{}, "", err
}

// decodeBMPConfig 解码BMP尺寸
// 入参: data 图片数据
// 返回: image.Config 图片尺寸, error 错误信息
func decodeBMPConfig(data []byte) (image.Config, error) {
	offset, width, height, bpp, compression, err := ParseBMPHeader(data)
	if err != nil {
		return image.Config{}, err
	}
	if offset <= 0 || width <= 0 || height == 0 {
		return image.Config{}, fmt.Errorf("unsupported bmp")
	}
	if _, err := BMPColorMasks(data, offset, bpp, compression); err != nil {
		return image.Config{}, err
	}
	if height < 0 {
		height = -height
	}
	return image.Config{ColorModel: color.NRGBAModel, Width: width, Height: height}, nil
}

// decodeBMPImage 解码BMP图片
// 入参: data 图片数据
// 返回: image.Image 图片对象, error 错误信息
func decodeBMPImage(data []byte) (image.Image, error) {
	return DecodeBMPContext(context.Background(), data)
}

// DecodeBMPContext 解码BMP真彩色像素，逐行检查取消
// 入参: ctx 取消上下文, data 图片数据
// 返回: image.Image 图片对象, error 格式或取消错误
func DecodeBMPContext(ctx context.Context, data []byte) (image.Image, error) {
	if err := ctx.Err(); err != nil {
		return nil, err
	}
	offset, width, height, bpp, compression, err := ParseBMPHeader(data)
	if err != nil {
		return nil, err
	}
	if offset > len(data) {
		return nil, fmt.Errorf("invalid bmp pixel offset")
	}
	masks, err := BMPColorMasks(data, offset, bpp, compression)
	if err != nil {
		return nil, err
	}
	topDown := height < 0
	if topDown {
		height = -height
	}
	if width <= 0 || height <= 0 {
		return nil, fmt.Errorf("invalid bmp size")
	}
	stride := ((int64(width)*int64(bpp) + 31) / 32) * 4
	if stride > int64(len(data)-offset)/int64(height) {
		return nil, fmt.Errorf("truncated bmp")
	}
	if width > int(^uint(0)>>1)/4/height {
		return nil, fmt.Errorf("bmp dimensions exceed image capacity")
	}
	rowStride := int(stride)
	img := image.NewNRGBA(image.Rect(0, 0, width, height))
	for y := 0; y < height; y++ {
		if err := ctx.Err(); err != nil {
			return nil, err
		}
		srcY := y
		if !topDown {
			srcY = height - 1 - y
		}
		row := data[offset+srcY*rowStride:]
		for x := 0; x < width; x++ {
			switch bpp {
			case 16, 32:
				var value uint32
				if bpp == 16 {
					value = uint32(binary.LittleEndian.Uint16(row[x*2:]))
				} else {
					value = binary.LittleEndian.Uint32(row[x*4:])
				}
				p := y*img.Stride + x*4
				for i, mask := range masks {
					img.Pix[p+i] = mask.sample(value)
				}
			case 24:
				p := x * 3
				img.SetNRGBA(x, y, color.NRGBA{R: row[p+2], G: row[p+1], B: row[p], A: 255})
			}
		}
	}
	return img, nil
}

// ParseBMPHeader 解析BMP文件头
// 入参: data 图片数据
// 返回: int 像素偏移, int 宽度, int 高度, uint16 位数, uint32 压缩方式, error 错误信息
func ParseBMPHeader(data []byte) (int, int, int, uint16, uint32, error) {
	if len(data) < 54 || string(data[:2]) != "BM" {
		return 0, 0, 0, 0, 0, fmt.Errorf("not bmp")
	}
	offset := int(binary.LittleEndian.Uint32(data[10:14]))
	dibSize := binary.LittleEndian.Uint32(data[14:18])
	if dibSize < 40 {
		return 0, 0, 0, 0, 0, fmt.Errorf("unsupported bmp")
	}
	if uint64(dibSize)+14 > uint64(len(data)) || offset < 0 || uint64(offset) < uint64(dibSize)+14 {
		return 0, 0, 0, 0, 0, fmt.Errorf("invalid bmp pixel offset")
	}
	width := int(int32(binary.LittleEndian.Uint32(data[18:22])))
	height := int(int32(binary.LittleEndian.Uint32(data[22:26])))
	planes := binary.LittleEndian.Uint16(data[26:28])
	bpp := binary.LittleEndian.Uint16(data[28:30])
	compression := binary.LittleEndian.Uint32(data[30:34])
	if planes != 1 || width <= 0 || height == 0 || int64(height) == -int64(^uint(0)>>1)-1 {
		return 0, 0, 0, 0, 0, fmt.Errorf("invalid bmp")
	}
	return offset, width, height, bpp, compression, nil
}

// PixelSource 获取图片像素源
// 入参: img 图片对象
// 返回: image.Image 图片像素源
func PixelSource(img image.Image) image.Image {
	if decoded, err := PixelData(img); err == nil {
		return decoded
	}
	return img
}

// PixelData 获取像素并保留惰性解码错误，供绘制和像素处理入口校验
// 入参: img 图片对象
// 返回: image.Image 解码像素, error 解码错误
func PixelData(img image.Image) (image.Image, error) {
	if src, ok := img.(interface{ Image() (image.Image, error) }); ok {
		return src.Image()
	}
	return img, nil
}

// PixelDataContext 获取像素并传递取消状态，兼容不接收上下文的图像实现
// 入参: ctx 取消上下文，nil使用后台上下文, img 图片对象
// 返回: image.Image 解码像素, error 解码或取消错误
func PixelDataContext(ctx context.Context, img image.Image) (image.Image, error) {
	if ctx == nil {
		ctx = context.Background()
	}
	if err := ctx.Err(); err != nil {
		return nil, err
	}
	if src, ok := img.(interface {
		ImageContext(context.Context) (image.Image, error)
	}); ok {
		return src.ImageContext(ctx)
	}
	pixels, err := PixelData(img)
	if cancelErr := ctx.Err(); cancelErr != nil {
		return nil, cancelErr
	}
	return pixels, err
}

// GrayAt 获取图片灰度像素
// 入参: img 图片对象, x X坐标, y Y坐标
// 返回: color.Gray 灰度像素
func GrayAt(img image.Image, x, y int) color.Gray {
	switch src := img.(type) {
	case *EncodedImage:
		pixels, err := src.Image()
		if err != nil {
			return color.Gray{}
		}
		return GrayAt(pixels, x, y)
	case *colorImage:
		return GrayAt(src.source, x, y)
	case *maskedTIFFImage:
		if src.alphaAt(x, y) == 0 {
			return color.Gray{}
		}
		point := src.samplePoint(src.source.Bounds(), x, y)
		return GrayAt(src.source, point.X, point.Y)
	case *transparentEdgeImage:
		if x <= src.rect.Min.X || y <= src.rect.Min.Y || x >= src.rect.Max.X-1 || y >= src.rect.Max.Y-1 {
			return color.Gray{}
		}
		bounds := src.source.Bounds()
		return GrayAt(src.source, bounds.Min.X+x-1, bounds.Min.Y+y-1)
	case *transparentEdgeImage64:
		if x <= src.rect.Min.X || y <= src.rect.Min.Y || x >= src.rect.Max.X-1 || y >= src.rect.Max.Y-1 {
			return color.Gray{}
		}
		bounds := src.source.Bounds()
		return GrayAt(src.source, bounds.Min.X+x-1, bounds.Min.Y+y-1)
	case *orientedTIFFImage:
		point, ok := src.sourcePoint(x, y)
		if !ok {
			return color.Gray{}
		}
		return GrayAt(src.source, point.X, point.Y)
	case *planarTIFFImage:
		if src.cmyk {
			return src.cmykGrayAt(x, y)
		}
	}
	if src, ok := img.(*image.Gray); ok {
		return src.GrayAt(x, y)
	}
	if src, ok := img.(*image.Paletted); ok {
		return color.GrayModel.Convert(src.At(x, y)).(color.Gray)
	}
	if src, ok := img.(image.RGBA64Image); ok {
		r, g, b, _ := src.RGBA64At(x, y).RGBA()
		return color.Gray{Y: uint8((19595*r + 38470*g + 7471*b + 1<<15) >> 24)}
	}
	return color.GrayModel.Convert(img.At(x, y)).(color.Gray)
}

// NRGBAAt 读取八位非预乘像素，缩减位深时保留原始颜色分量
// 入参: img 图片对象, x X坐标, y Y坐标
// 返回: color.NRGBA NRGBA像素
func NRGBAAt(img image.Image, x, y int) color.NRGBA {
	switch src := img.(type) {
	case *image.NRGBA:
		return src.NRGBAAt(x, y)
	case *image.NRGBA64:
		c := src.NRGBA64At(x, y)
		return color.NRGBA{R: uint8(c.R >> 8), G: uint8(c.G >> 8), B: uint8(c.B >> 8), A: uint8(c.A >> 8)}
	case interface{ NRGBAAt(int, int) color.NRGBA }:
		return src.NRGBAAt(x, y)
	case interface{ NRGBA64At(int, int) color.NRGBA64 }:
		c := src.NRGBA64At(x, y)
		return color.NRGBA{R: uint8(c.R >> 8), G: uint8(c.G >> 8), B: uint8(c.B >> 8), A: uint8(c.A >> 8)}
	case *image.RGBA, *image.RGBA64, *image.Gray, *image.Gray16, *image.Alpha, *image.Alpha16, *image.CMYK, *image.YCbCr, *image.NYCbCrA:
		c := src.(image.RGBA64Image).RGBA64At(x, y)
		if c.A == 0 {
			return color.NRGBA{}
		}
		r, g, b, a := uint32(c.R), uint32(c.G), uint32(c.B), uint32(c.A)
		if a != 0xffff {
			r = r * 0xffff / a
			g = g * 0xffff / a
			b = b * 0xffff / a
		}
		return color.NRGBA{R: uint8(r >> 8), G: uint8(g >> 8), B: uint8(b >> 8), A: uint8(a >> 8)}
	}
	value := img.At(x, y)
	if c, ok := value.(color.NRGBA64); ok {
		return color.NRGBA{R: uint8(c.R >> 8), G: uint8(c.G >> 8), B: uint8(c.B >> 8), A: uint8(c.A >> 8)}
	}
	return color.NRGBAModel.Convert(value).(color.NRGBA)
}

// AlphaAt 读取十六位透明度，原生图像避免颜色装箱
// 入参: img 图像, x 横坐标, y 纵坐标
// 返回: uint32 十六位透明度
func AlphaAt(img image.Image, x, y int) uint32 {
	switch source := img.(type) {
	case *maskedTIFFImage:
		return source.alphaAt(x, y)
	case *image.YCbCr, *image.CMYK, *image.Gray, *image.Gray16:
		return 65535
	case *image.NYCbCrA:
		if !image.Pt(x, y).In(source.Rect) {
			return 0
		}
		return uint32(source.A[source.AOffset(x, y)]) * 257
	case *orientedTIFFImage:
		point, ok := source.sourcePoint(x, y)
		if !ok {
			return 0
		}
		return AlphaAt(source.source, point.X, point.Y)
	case *planarTIFFImage:
		if !image.Pt(x, y).In(source.bounds) {
			return 0
		}
		if source.alpha != 0 {
			index := 3
			if source.cmyk {
				index = 4
			} else if source.gray || len(source.palette) != 0 {
				index = 1
			}
			return uint32(source.sampleAt(index, x, y))
		}
		if len(source.palette) != 0 {
			index := uint32(source.sampleAt(0, x, y)) * uint32(len(source.palette)-1) / 65535
			return uint32(source.palette[index].A)
		}
		return 65535
	case *image.RGBA:
		if !(image.Point{X: x, Y: y}).In(source.Rect) {
			return 0
		}
		return uint32(source.Pix[source.PixOffset(x, y)+3]) * 257
	case *image.RGBA64:
		if !(image.Point{X: x, Y: y}).In(source.Rect) {
			return 0
		}
		i := source.PixOffset(x, y) + 6
		return uint32(source.Pix[i])<<8 | uint32(source.Pix[i+1])
	case *image.NRGBA:
		return uint32(source.NRGBAAt(x, y).A) * 257
	case *image.NRGBA64:
		return uint32(source.NRGBA64At(x, y).A)
	case *image.Alpha:
		return uint32(source.AlphaAt(x, y).A) * 257
	case *image.Alpha16:
		return uint32(source.Alpha16At(x, y).A)
	}
	if source, ok := img.(image.RGBA64Image); ok {
		return uint32(source.RGBA64At(x, y).A)
	}
	_, _, _, alpha := img.At(x, y).RGBA()
	return alpha
}

// HighPrecision 判断颜色模型及调色板是否需要保留十六位采样
// 入参: img 图片对象
// 返回: bool 是否使用十六位颜色或透明度
func HighPrecision(img image.Image) bool {
	model := img.ColorModel()
	if palette, ok := model.(color.Palette); ok {
		for _, value := range palette {
			switch value.(type) {
			case color.RGBA, color.NRGBA, color.Gray, color.Alpha, color.CMYK, color.YCbCr, color.NYCbCrA:
			default:
				pixel := color.NRGBA64Model.Convert(value).(color.NRGBA64)
				if pixel.R%257 != 0 || pixel.G%257 != 0 || pixel.B%257 != 0 || pixel.A%257 != 0 {
					return true
				}
			}
		}
		return false
	}
	return model == color.NRGBA64Model || model == color.RGBA64Model || model == color.Gray16Model || model == color.Alpha16Model
}

// NRGBA64At 读取十六位非预乘像素，保留透明像素颜色，原生图像避免颜色装箱
// 入参: img 图片对象, x X坐标, y Y坐标
// 返回: color.NRGBA64 NRGBA64像素
func NRGBA64At(img image.Image, x, y int) color.NRGBA64 {
	switch src := img.(type) {
	case interface{ NRGBA64At(int, int) color.NRGBA64 }:
		return src.NRGBA64At(x, y)
	case *image.NRGBA:
		c := src.NRGBAAt(x, y)
		return color.NRGBA64{R: uint16(c.R) * 257, G: uint16(c.G) * 257, B: uint16(c.B) * 257, A: uint16(c.A) * 257}
	case *image.Gray:
		v := uint16(src.GrayAt(x, y).Y) * 257
		return color.NRGBA64{R: v, G: v, B: v, A: 0xffff}
	case *image.Gray16:
		c := src.Gray16At(x, y)
		return color.NRGBA64{R: c.Y, G: c.Y, B: c.Y, A: 0xffff}
	case *image.Alpha, *image.Alpha16:
		c := src.(image.RGBA64Image).RGBA64At(x, y)
		if c.A == 0 {
			return color.NRGBA64{}
		}
		return color.NRGBA64{R: 0xffff, G: 0xffff, B: 0xffff, A: c.A}
	case *image.RGBA, *image.RGBA64, *image.CMYK, *image.YCbCr, *image.NYCbCrA:
		c := src.(image.RGBA64Image).RGBA64At(x, y)
		if c.A == 0 {
			return color.NRGBA64{}
		}
		r, g, b, a := uint32(c.R), uint32(c.G), uint32(c.B), uint32(c.A)
		if a != 0xffff {
			r = r * 0xffff / a
			g = g * 0xffff / a
			b = b * 0xffff / a
		}
		return color.NRGBA64{R: uint16(r), G: uint16(g), B: uint16(b), A: c.A}
	}
	value := img.At(x, y)
	if c, ok := value.(color.NRGBA); ok {
		return color.NRGBA64{R: uint16(c.R) * 257, G: uint16(c.G) * 257, B: uint16(c.B) * 257, A: uint16(c.A) * 257}
	}
	return color.NRGBA64Model.Convert(value).(color.NRGBA64)
}

// transparentEdgeColor 获取透明像素相邻的可见颜色
// 入参: source 图片对象, x X坐标, y Y坐标
// 返回: color.NRGBA 颜色, bool 是否存在
func transparentEdgeColor(source image.Image, x, y int) (color.NRGBA, bool) {
	img, ok := source.(*image.NRGBA)
	if !ok {
		bounds := source.Bounds()
		var best color.NRGBA
		for ny := max(y-1, bounds.Min.Y); ny < min(y+2, bounds.Max.Y); ny++ {
			for nx := max(x-1, bounds.Min.X); nx < min(x+2, bounds.Max.X); nx++ {
				if nx == x && ny == y {
					continue
				}
				pixel := NRGBAAt(source, nx, ny)
				if pixel.A > best.A {
					best = pixel
					if best.A == 255 {
						best.A = 1
						return best, true
					}
				}
			}
		}
		if best.A == 0 {
			return color.NRGBA{}, false
		}
		best.A = 1
		return best, true
	}
	bounds := img.Bounds()
	best, alpha := 0, uint8(0)
	left, right := max(x-1, bounds.Min.X), min(x+2, bounds.Max.X)
scan:
	for ny := max(y-1, bounds.Min.Y); ny < min(y+2, bounds.Max.Y); ny++ {
		offset := img.PixOffset(left, ny)
		for nx := left; nx < right; nx++ {
			if (nx != x || ny != y) && img.Pix[offset+3] > alpha {
				best, alpha = offset, img.Pix[offset+3]
				if alpha == 255 {
					break scan
				}
			}
			offset += 4
		}
	}
	if alpha == 0 {
		return color.NRGBA{}, false
	}
	return color.NRGBA{R: img.Pix[best], G: img.Pix[best+1], B: img.Pix[best+2], A: 1}, true
}

// transparentPaddingColor 获取透明补齐颜色
// 入参: c 边缘颜色
// 返回: color.NRGBA 补齐颜色
func transparentPaddingColor(c color.NRGBA) color.NRGBA {
	c.A = 1
	return c
}

// WithTransparentEdge 扫描透明边缘并共享只读像素，保留原图采样精度
// 入参: ctx 取消上下文，nil使用后台上下文, img 图片对象
// 返回: image.Image 补边图像, int 补边像素数, error 解码或取消错误
func WithTransparentEdge(ctx context.Context, img image.Image) (image.Image, int, error) {
	if ctx == nil {
		ctx = context.Background()
	}
	if err := ctx.Err(); err != nil {
		return nil, 0, err
	}
	if img == nil {
		return img, 0, nil
	}
	bounds := img.Bounds()
	w, h := bounds.Dx(), bounds.Dy()
	if w == 0 || h == 0 {
		return img, 0, nil
	}
	if src, ok := img.(*EncodedImage); ok && src.MIME() == "image/jpeg" {
		if src.profile != nil || src.config.ColorModel == color.CMYKModel {
			pixels, err := src.ImageContext(ctx)
			return pixels, 0, err
		}
		return img, 0, nil
	}
	source, err := PixelDataContext(ctx, img)
	if err != nil {
		return nil, 0, err
	}
	if encoded, ok := img.(*EncodedImage); ok && (encoded.profile != nil || encoded.pngColor != nil) {
		img = source
	}
	switch source.(type) {
	case *image.Gray, *image.Gray16, *image.YCbCr, *image.CMYK:
		return img, 0, nil
	}
	hasZero, hasVisible := false, false
	var pixels []byte
	var stride, step int
	wideAlpha := false
	switch src := source.(type) {
	case *image.NRGBA:
		pixels, stride, step = src.Pix, src.Stride, 4
	case *image.RGBA:
		pixels, stride, step = src.Pix, src.Stride, 4
	case *image.Alpha:
		pixels, stride, step = src.Pix, src.Stride, 1
	case *image.NRGBA64:
		pixels, stride, step, wideAlpha = src.Pix, src.Stride, 8, true
	case *image.RGBA64:
		pixels, stride, step, wideAlpha = src.Pix, src.Stride, 8, true
	case *image.Alpha16:
		pixels, stride, step, wideAlpha = src.Pix, src.Stride, 2, true
	case *colorImage:
		pixels, stride, step, wideAlpha = src.Pix, src.Stride, 8, true
	}
	if pixels != nil {
		hasZero = pixels[step-1] == 0 && (!wideAlpha || pixels[step-2] == 0)
		hasVisible = !hasZero
	scan:
		for y := 0; y < h; y++ {
			if err := ctx.Err(); err != nil {
				return nil, 0, err
			}
			offset := y*stride + step - 1
			for x := 0; x < w; x++ {
				if x&255 == 0 {
					if err := ctx.Err(); err != nil {
						return nil, 0, err
					}
				}
				if (pixels[offset] == 0 && (!wideAlpha || pixels[offset-1] == 0)) != hasZero {
					hasZero, hasVisible = true, true
					break scan
				}
				offset += step
			}
		}
	} else {
		for y := bounds.Min.Y; y < bounds.Max.Y && !(hasZero && hasVisible); y++ {
			if err := ctx.Err(); err != nil {
				return nil, 0, err
			}
			for x := bounds.Min.X; x < bounds.Max.X; x++ {
				if (x-bounds.Min.X)&255 == 0 {
					if err := ctx.Err(); err != nil {
						return nil, 0, err
					}
				}
				if AlphaAt(source, x, y) == 0 {
					hasZero = true
				} else {
					hasVisible = true
				}
				if hasZero && hasVisible {
					break
				}
			}
		}
	}
	if err := ctx.Err(); err != nil {
		return nil, 0, err
	}
	if !hasZero || !hasVisible {
		return img, 0, nil
	}
	if HighPrecision(source) {
		return &transparentEdgeImage64{source: source, rect: image.Rect(0, 0, w+2, h+2)}, 1, nil
	}
	return &transparentEdgeImage{source: source, rect: image.Rect(0, 0, w+2, h+2)}, 1, nil
}