feat(归并文件): 归并根包文件

ofdgo_image.go

@@ -18,10 +18,25 @@ import (
 	"context"
 	"fmt"
 	"image"
+	"image/color"
 
 	baseimage "github.com/xiaoqidun/ofdgo/internal/base/image"
 )
 
+// imageCacheLimit 限制图片缓存的编码数据及预计解码内存
+const imageCacheLimit = 64 << 20
+
+// EncodedImage 保留PNG或JPEG原始编码，首次读取像素时解码
+// Bytes返回只读数据，Image返回共享只读像素，解码错误由Image返回
+type EncodedImage = baseimage.EncodedImage
+
+// resampledImageMask 按替换图尺寸读取已校验的二值蒙版，不分配缩放像素
+type resampledImageMask struct {
+	source   image.Image
+	original image.Rectangle
+	bounds   image.Rectangle
+}
+
 // imageWithEffects 合并同尺寸二值蒙版和对象透明度，保留原图采样精度及只读像素
 // 入参: ctx 取消上下文，nil使用后台上下文, img 原图, mask 二值蒙版，可为空, alpha 对象透明度，可为空
 // 返回: image.Image 处理后的图片, error 蒙版、解码或取消错误
@@ -167,3 +182,187 @@ func validateImageMask(ctx context.Context, mask image.Image, size image.Point)
 	}
 	return ctx.Err()
 }
+
+// ImageResource 按资源标识读取只读图片，PNG和JPEG保留原始编码并惰性解码
+// 应先读取源页面以加载页面资源，返回EncodedImage时通过Image检查像素解码错误
+// 入参: id 图片资源标识
+// 返回: image.Image 图片资源, error 读取错误
+func (r *Renderer) ImageResource(id string) (image.Image, error) {
+	path, ok := resourceValue(r.Reader.ResMap, id)
+	if !ok {
+		return nil, fmt.Errorf("image resource %q not found", id)
+	}
+	return r.cachedImageResource(path)
+}
+
+// imageObjectResource 读取主图，无法读取或解码时使用对象声明的替换图
+// 入参: object 图像对象
+// 返回: image.Image 图片资源, error 读取或解码错误
+func (r *Renderer) imageObjectResource(object ImageObject) (image.Image, error) {
+	read := func(id string) (image.Image, error) {
+		img, err := r.ImageResource(id)
+		if err == nil && (r.decodeImages || object.Substitution != "") {
+			pixels, decodeErr := baseimage.PixelDataContext(r.pageContext, img)
+			err = decodeErr
+			if r.decodeImages {
+				img = pixels
+			}
+		}
+		return img, err
+	}
+	img, err := read(object.ResourceID)
+	if err == nil || object.Substitution == "" || sameResourceID(object.Substitution, object.ResourceID) {
+		return img, err
+	}
+	if r.pageContext != nil && r.pageContext.Err() != nil {
+		return nil, r.pageContext.Err()
+	}
+	if r.OutputContext != nil && r.OutputContext.Err() != nil {
+		return nil, r.OutputContext.Err()
+	}
+	img, substituteErr := read(object.Substitution)
+	if substituteErr != nil {
+		return nil, fmt.Errorf("image resource %q: %w; substitution %q: %w", object.ResourceID, err, object.Substitution, substituteErr)
+	}
+	return img, nil
+}
+
+// imageObjectMask 校验原图蒙版，替换图分辨率不同时按像素中心进行最近邻采样
+// 入参: object 图片对象, target 实际绘制图像
+// 返回: image.Image 二值蒙版，无蒙版时为nil, error 资源或蒙版错误
+func (r *Renderer) imageObjectMask(object ImageObject, target image.Image) (image.Image, error) {
+	if object.ImageMask == "" {
+		return nil, nil
+	}
+	mask, err := r.ImageResource(object.ImageMask)
+	if err != nil {
+		return nil, err
+	}
+	size := target.Bounds().Size()
+	if object.Substitution == "" {
+		if mask.Bounds().Size() == size {
+			return mask, nil
+		}
+		return nil, fmt.Errorf("image mask dimensions differ from source")
+	}
+	original, err := r.ImageResource(object.ResourceID)
+	if err != nil {
+		if mask.Bounds().Size() == size {
+			return mask, nil
+		}
+		return nil, fmt.Errorf("cannot verify image mask dimensions: %w", err)
+	}
+	if mask.Bounds().Size() != original.Bounds().Size() {
+		return nil, fmt.Errorf("image mask dimensions differ from source")
+	}
+	if mask.Bounds().Size() == size {
+		return mask, nil
+	}
+	ctx := r.pageContext
+	if ctx == nil {
+		ctx = context.Background()
+	}
+	if err := validateImageMask(ctx, mask, original.Bounds().Size()); err != nil {
+		return nil, err
+	}
+	mask, err = baseimage.PixelDataContext(ctx, mask)
+	if err != nil {
+		return nil, err
+	}
+	return &resampledImageMask{source: mask, original: mask.Bounds(), bounds: image.Rectangle{Max: size}}, ctx.Err()
+}
+
+// Bounds 返回替换图对应的蒙版范围
+// 返回: image.Rectangle 像素区域
+func (s *resampledImageMask) Bounds() image.Rectangle { return s.bounds }
+
+// ColorModel 返回二值蒙版的灰度颜色模型
+// 返回: color.Model 灰度模型
+func (s *resampledImageMask) ColorModel() color.Model { return color.GrayModel }
+
+// GrayAt 按像素中心读取原蒙版，区域外返回黑色
+// 入参: x 横坐标, y 纵坐标
+// 返回: color.Gray 二值灰度
+func (s *resampledImageMask) GrayAt(x, y int) color.Gray {
+	if !image.Pt(x, y).In(s.bounds) {
+		return color.Gray{}
+	}
+	bounds, size := s.original, s.bounds.Size()
+	sx := bounds.Min.X + int((uint64(x-s.bounds.Min.X)*uint64(bounds.Dx())+uint64(bounds.Dx()/2))/uint64(size.X))
+	sy := bounds.Min.Y + int((uint64(y-s.bounds.Min.Y)*uint64(bounds.Dy())+uint64(bounds.Dy()/2))/uint64(size.Y))
+	return baseimage.GrayAt(s.source, sx, sy)
+}
+
+// At 返回采样后的二值灰度
+// 入参: x 横坐标, y 纵坐标
+// 返回: color.Color 像素颜色
+func (s *resampledImageMask) At(x, y int) color.Color { return s.GrayAt(x, y) }
+
+// BackingImage 返回共享只读蒙版，供图像缓存计算底层存储
+// 返回: image.Image 共享只读蒙版
+func (s *resampledImageMask) BackingImage() image.Image { return s.source }
+
+// RGBA64At 返回不透明的十六位灰度颜色，供蒙版合成直接读取
+// 入参: x 横坐标, y 纵坐标
+// 返回: color.RGBA64 像素颜色
+func (s *resampledImageMask) RGBA64At(x, y int) color.RGBA64 {
+	value := uint16(s.GrayAt(x, y).Y) * 257
+	return color.RGBA64{R: value, G: value, B: value, A: 65535}
+}
+
+// cachedImageResource 复用原始图片资源，不依赖输出后端
+// 入参: resPath 图片资源路径
+// 返回: image.Image 图片对象, error 读取错误
+func (r *Renderer) cachedImageResource(resPath string) (image.Image, error) {
+	if r.pageContext != nil {
+		if err := r.pageContext.Err(); err != nil {
+			return nil, err
+		}
+	}
+	resPath = cleanPackagePath(r.Reader.ResPath(resPath))
+	img, ok := r.imageCache.get(resPath)
+	if !ok {
+		var err error
+		img, err = r.readImageResource(resPath)
+		if err != nil {
+			return nil, err
+		}
+		r.imageCache.put(resPath, img, baseimage.StorageCost(img, r.imageCache.limit))
+	}
+	return img, nil
+}
+
+// decodeImageResource 读取或复用图片资源，按输出需要解码像素
+// 入参: resPath 图片资源路径
+// 返回: image.Image 图片对象, error 读取或解码错误
+func (r *Renderer) decodeImageResource(resPath string) (image.Image, error) {
+	img, err := r.cachedImageResource(resPath)
+	if err != nil {
+		return nil, err
+	}
+	if r.decodeImages {
+		return baseimage.PixelDataContext(r.pageContext, img)
+	}
+	return img, nil
+}
+
+// readImageResource 按页面上下文读取图片，借用已加载的只读编码
+// 入参: resPath 图片资源路径
+// 返回: image.Image 图片对象, error 读取或解码错误
+func (r *Renderer) readImageResource(resPath string) (image.Image, error) {
+	ctx := r.pageContext
+	if ctx == nil {
+		ctx = context.Background()
+	}
+	data, err := r.Reader.readFileViewContext(ctx, resPath)
+	if err != nil {
+		return nil, err
+	}
+	img, _, err := baseimage.DecodeContext(ctx, data)
+	return img, err
+}
+
+// NewEncodedImage 复制输入数据并读取PNG或JPEG头信息，不解码像素
+// 入参: data 原始编码
+// 返回: *EncodedImage 惰性图片, error 头信息或格式错误
+func NewEncodedImage(data []byte) (*EncodedImage, error) { return baseimage.NewEncodedImage(data) }