mirror of
https://github.com/xiaoqidun/ofdgo.git
synced 2026-10-29 16:59:27 +08:00
423 lines
16 KiB
Go
423 lines
16 KiB
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 ofdgo
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
"reflect"
|
|
|
|
"github.com/xiaoqidun/pdfgo"
|
|
)
|
|
|
|
// pdfPatternCellLimit 限制单页图案矢量快照的估算字节数
|
|
const pdfPatternCellLimit = 8 << 20
|
|
|
|
// pdfPatternCellKey 隔离图案来源、基色、坐标、精度、混合空间与目标文档资源
|
|
type pdfPatternCellKey struct {
|
|
source *pdfgo.TilingPattern
|
|
base pdfgo.Paint
|
|
matrix pdfgo.Matrix
|
|
dpi float64
|
|
editor *Editor
|
|
space *pdfgo.ColorSpace
|
|
pageWidth, pageHeight float64
|
|
}
|
|
|
|
// pdfPatternCell 保存图案快照及单元内部的来源区段
|
|
type pdfPatternCell struct {
|
|
pattern *Pattern
|
|
origin pdfObjectOrigin
|
|
}
|
|
|
|
// pdfPatternInstanceRange 收缩已知路径及图片的实例范围，未知边界保留完整范围
|
|
// 入参: columns 横向实例数, rows 纵向实例数, bounds 对象边界, width 单元宽, height 单元高, xstep 横向步距, ystep 纵向步距
|
|
// 返回: int 左界, int 右界, int 上界, int 下界
|
|
func pdfPatternInstanceRange(columns, rows float64, bounds []Box, width, height, xstep, ystep float64) (int, int, int, int) {
|
|
left, right, top, bottom := 0.0, 1-columns, 0.0, 1-rows
|
|
for _, box := range bounds {
|
|
if box.W <= 0 || box.H <= 0 || !finite(box.X) || !finite(box.Y) || !finite(box.X+box.W) || !finite(box.Y+box.H) {
|
|
return 1 - int(columns), 0, 1 - int(rows), 0
|
|
}
|
|
left = math.Min(left, math.Max(1-columns, math.Ceil(-(box.X+box.W)/xstep)-1))
|
|
right = math.Max(right, math.Min(0, math.Floor((width-box.X)/xstep)+1))
|
|
top = math.Min(top, math.Max(1-rows, math.Ceil(-(box.Y+box.H)/ystep)-1))
|
|
bottom = math.Max(bottom, math.Min(0, math.Floor((height-box.Y)/ystep)+1))
|
|
}
|
|
return int(left), int(right), int(top), int(bottom)
|
|
}
|
|
|
|
// tilingPattern 有界复用矢量图案，返回可独立编辑的对象副本
|
|
// 入参: source 平铺图案, base 无色图案基色
|
|
// 返回: *Pattern OFD图案, error 转换或取消错误
|
|
func (p *pdfImporter) tilingPattern(source *pdfgo.TilingPattern, base pdfgo.Paint) (*Pattern, error) {
|
|
if err := p.ctx.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
if p.resolveFile != nil || p.resolveReference != nil {
|
|
return p.buildTilingPattern(source, base)
|
|
}
|
|
if p.patternCells == nil {
|
|
p.patternCells = resourceCache[pdfPatternCellKey, pdfPatternCell](p.resources, pdfPatternCellLimit)
|
|
}
|
|
key := pdfPatternCellKey{source: source, matrix: p.matrix, dpi: p.rasterDPI, editor: p.editor, space: p.compositeSpace, pageWidth: p.pageWidth, pageHeight: p.pageHeight}
|
|
if source.PaintType == 2 {
|
|
base.Tiling, base.Axial, base.Radial, base.Mesh, base.Function, base.Shading = nil, nil, nil, nil, nil, nil
|
|
base.Alpha = 1
|
|
key.base = base
|
|
}
|
|
if cached, ok := p.patternCells.get(key); ok {
|
|
pattern := cloneEditorData(cached.pattern)
|
|
if cached.origin.sequence != nil || len(cached.origin.related) != 0 {
|
|
p.patternOrigins[pattern] = cached.origin
|
|
}
|
|
return pattern, p.ctx.Err()
|
|
}
|
|
pattern, err := p.buildTilingPattern(source, base)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := p.ctx.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
cost := pdfPatternCellCost(reflect.ValueOf(pattern), p.patternCells.limit, 0)
|
|
cost += int(reflect.TypeFor[pdfPatternCell]().Size()) + len(p.patternOrigins[pattern].related)*int(reflect.TypeFor[*pdfgo.ContentSequence]().Size())
|
|
if cost <= p.patternCells.limit {
|
|
p.patternCells.put(key, pdfPatternCell{pattern: cloneEditorData(pattern), origin: p.patternOrigins[pattern]}, cost)
|
|
}
|
|
return pattern, p.ctx.Err()
|
|
}
|
|
|
|
// pdfPatternCellCost 有界估算公开数据的快照成本，不扫描共享的私有排版来源
|
|
// 入参: value 对象值, budget 剩余预算, depth 嵌套深度
|
|
// 返回: int 字节成本，超预算时返回预算加一
|
|
func pdfPatternCellCost(value reflect.Value, budget, depth int) int {
|
|
if depth > 128 || budget < 0 {
|
|
return budget + 1
|
|
}
|
|
cost := int(value.Type().Size())
|
|
if cost > budget {
|
|
return budget + 1
|
|
}
|
|
switch value.Kind() {
|
|
case reflect.String:
|
|
if value.Len() > budget-cost {
|
|
return budget + 1
|
|
}
|
|
cost += value.Len()
|
|
case reflect.Pointer:
|
|
if !value.IsNil() {
|
|
cost += pdfPatternCellCost(value.Elem(), budget-cost, depth+1)
|
|
}
|
|
case reflect.Slice, reflect.Array:
|
|
for i := range value.Len() {
|
|
cost += pdfPatternCellCost(value.Index(i), budget-cost, depth+1)
|
|
if cost > budget {
|
|
return budget + 1
|
|
}
|
|
}
|
|
case reflect.Struct:
|
|
for i := range value.NumField() {
|
|
if value.Type().Field(i).IsExported() {
|
|
cost += pdfPatternCellCost(value.Field(i), budget-cost, depth+1)
|
|
if cost > budget {
|
|
return budget + 1
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return cost
|
|
}
|
|
|
|
// pdfPatternRectangleBounds 识别无文字裁剪的单个轴对齐矩形，不把边界框当作路径覆盖范围
|
|
// 入参: path 矩形路径
|
|
// 返回: pdfgo.Rectangle 精确覆盖范围, bool 是否单矩形
|
|
func pdfPatternRectangleBounds(path pdfgo.Path) (pdfgo.Rectangle, bool) {
|
|
if len(path.Text) != 0 || len(path.Segments) != 5 || path.Segments[4].Operator != "C" || len(path.Segments[4].Points) != 0 {
|
|
return pdfgo.Rectangle{}, false
|
|
}
|
|
var points [4]pdfgo.Point
|
|
box := pdfgo.Rectangle{XMin: math.Inf(1), YMin: math.Inf(1), XMax: math.Inf(-1), YMax: math.Inf(-1)}
|
|
for i, segment := range path.Segments[:4] {
|
|
if len(segment.Points) != 1 || i == 0 && segment.Operator != "M" || i != 0 && segment.Operator != "L" {
|
|
return pdfgo.Rectangle{}, false
|
|
}
|
|
point := segment.Points[0]
|
|
if !finite(point.X) || !finite(point.Y) {
|
|
return pdfgo.Rectangle{}, false
|
|
}
|
|
points[i] = point
|
|
box.XMin, box.YMin = math.Min(box.XMin, point.X), math.Min(box.YMin, point.Y)
|
|
box.XMax, box.YMax = math.Max(box.XMax, point.X), math.Max(box.YMax, point.Y)
|
|
}
|
|
if box.XMin == box.XMax || box.YMin == box.YMax {
|
|
return pdfgo.Rectangle{}, false
|
|
}
|
|
corners := 0
|
|
for i, point := range points {
|
|
next := points[(i+1)%len(points)]
|
|
if (point.X == next.X) == (point.Y == next.Y) || point.X != box.XMin && point.X != box.XMax || point.Y != box.YMin && point.Y != box.YMax {
|
|
return pdfgo.Rectangle{}, false
|
|
}
|
|
corner := 0
|
|
if point.X == box.XMax {
|
|
corner++
|
|
}
|
|
if point.Y == box.YMax {
|
|
corner += 2
|
|
}
|
|
corners |= 1 << corner
|
|
}
|
|
return box, corners == 15
|
|
}
|
|
|
|
// pdfPatternRectanglePath 构造精确闭合矩形，用于连续覆盖的绘制及裁剪
|
|
// 入参: box 矩形范围
|
|
// 返回: pdfgo.Path 闭合矩形
|
|
func pdfPatternRectanglePath(box pdfgo.Rectangle) pdfgo.Path {
|
|
return pdfgo.Path{Segments: []pdfgo.Segment{{Operator: "M", Points: []pdfgo.Point{{X: box.XMin, Y: box.YMin}}}, {Operator: "L", Points: []pdfgo.Point{{X: box.XMax, Y: box.YMin}}}, {Operator: "L", Points: []pdfgo.Point{{X: box.XMax, Y: box.YMax}}}, {Operator: "L", Points: []pdfgo.Point{{X: box.XMin, Y: box.YMax}}}, {Operator: "C"}}}
|
|
}
|
|
|
|
// densePatternRectangle 对单一不透明矩形的连续重叠轴求精确并集，不修改源步距或近似采样
|
|
// 入参: source 原图案, base 基色, width 单元宽, height 单元高, xstep 横向步距, ystep 纵向步距
|
|
// 返回: *pdfgo.PathMark 并集矩形, bool 横轴已连续覆盖, bool 纵轴已连续覆盖, error 解析或取消错误
|
|
func (p *pdfImporter) densePatternRectangle(source *pdfgo.TilingPattern, base pdfgo.Paint, width, height, xstep, ystep float64) (*pdfgo.PathMark, bool, bool, error) {
|
|
limit := float64(int(^uint(0) >> 1))
|
|
continuousX, continuousY := width/xstep >= limit, height/ystep >= limit
|
|
if !continuousX && !continuousY {
|
|
return nil, false, false, nil
|
|
}
|
|
var mark pdfgo.PathMark
|
|
paths, other := 0, false
|
|
visitor := pdfgo.Visitor{
|
|
Warning: p.warning,
|
|
Path: func(value pdfgo.PathMark) error {
|
|
paths++
|
|
if paths == 1 {
|
|
mark = value
|
|
}
|
|
return nil
|
|
},
|
|
Text: func(pdfgo.TextMark) error { other = true; return nil }, Image: func(pdfgo.ImageMark) error { other = true; return nil },
|
|
Form: func(pdfgo.FormMark, func(pdfgo.Visitor) error) error { other = true; return nil },
|
|
Group: func(pdfgo.GroupMark, func(pdfgo.Visitor) error) error { other = true; return nil },
|
|
PostScript: func(pdfgo.PostScriptMark) error { other = true; return nil },
|
|
}
|
|
if err := source.Walk(p.ctx, base, visitor); err != nil {
|
|
return nil, false, false, err
|
|
}
|
|
if err := p.ctx.Err(); err != nil {
|
|
return nil, false, false, err
|
|
}
|
|
style, paint := mark.Style, mark.Style.Fill
|
|
if other || paths != 1 || !mark.Fill || mark.Stroke || paint.None || paint.Alpha != 1 || paint.Colorant != nil || paint.Process != nil || paint.Tiling != nil || paint.Shading != nil || paint.Axial != nil || paint.Radial != nil || paint.Mesh != nil || paint.Function != nil || style.SoftMask != nil || style.Transfer != nil || style.Halftone != nil || style.FillOverprint || style.AlphaIsShape || style.BlendMode != "" && style.BlendMode != "Normal" || len(style.Clips) != 1 {
|
|
return nil, false, false, nil
|
|
}
|
|
clip, ok := pdfPatternRectangleBounds(style.Clips[0])
|
|
if !ok || clip != source.BBox {
|
|
return nil, false, false, nil
|
|
}
|
|
box, ok := pdfPatternRectangleBounds(mark.Path)
|
|
if !ok {
|
|
return nil, false, false, nil
|
|
}
|
|
box.XMin, box.YMin = math.Max(box.XMin, clip.XMin), math.Max(box.YMin, clip.YMin)
|
|
box.XMax, box.YMax = math.Min(box.XMax, clip.XMax), math.Min(box.YMax, clip.YMax)
|
|
if box.XMax <= box.XMin || box.YMax <= box.YMin || continuousX && box.XMax-box.XMin < xstep || continuousY && box.YMax-box.YMin < ystep {
|
|
return nil, false, false, nil
|
|
}
|
|
if continuousX {
|
|
box.XMin, box.XMax = clip.XMin, clip.XMax
|
|
}
|
|
if continuousY {
|
|
box.YMin, box.YMax = clip.YMin, clip.YMax
|
|
}
|
|
mark.Path = pdfPatternRectanglePath(box)
|
|
return &mark, continuousX, continuousY, nil
|
|
}
|
|
|
|
// densePatternPageBounds 将连续覆盖轴的等价单元边界置于页面外，避免子像素单元接缝
|
|
// 入参: source 原图案, box 原单元, x 横轴连续覆盖, y 纵轴连续覆盖
|
|
// 返回: pdfgo.Rectangle 等价单元边界, error 非有限范围或退化变换错误
|
|
func (p *pdfImporter) densePatternPageBounds(source *pdfgo.TilingPattern, box pdfgo.Rectangle, x, y bool) (pdfgo.Rectangle, error) {
|
|
inverse, ok := p.matrix.Mul(source.Matrix).Inverse()
|
|
if !ok {
|
|
return box, fmt.Errorf("invalid PDF dense pattern matrix")
|
|
}
|
|
if !finite(p.pageWidth) || !finite(p.pageHeight) || p.pageWidth <= 0 || p.pageHeight <= 0 {
|
|
return box, fmt.Errorf("invalid PDF dense pattern page bounds")
|
|
}
|
|
area := pdfgo.Rectangle{XMin: math.Inf(1), YMin: math.Inf(1), XMax: math.Inf(-1), YMax: math.Inf(-1)}
|
|
for _, point := range []pdfgo.Point{{}, {X: p.pageWidth}, {Y: p.pageHeight}, {X: p.pageWidth, Y: p.pageHeight}} {
|
|
point = inverse.Apply(point)
|
|
area.XMin, area.YMin = math.Min(area.XMin, point.X), math.Min(area.YMin, point.Y)
|
|
area.XMax, area.YMax = math.Max(area.XMax, point.X), math.Max(area.YMax, point.Y)
|
|
}
|
|
if x {
|
|
span := box.XMax - box.XMin
|
|
box.XMin, box.XMax = math.Min(box.XMin, area.XMin-span), math.Max(box.XMax, area.XMax+span)
|
|
}
|
|
if y {
|
|
span := box.YMax - box.YMin
|
|
box.YMin, box.YMax = math.Min(box.YMin, area.YMin-span), math.Max(box.YMax, area.YMax+span)
|
|
}
|
|
for _, value := range []float64{box.XMin, box.YMin, box.XMax, box.YMax, box.XMax - box.XMin, box.YMax - box.YMin} {
|
|
if !finite(value) {
|
|
return box, fmt.Errorf("nonfinite PDF dense pattern page bounds")
|
|
}
|
|
}
|
|
return box, nil
|
|
}
|
|
|
|
// buildTilingPattern 保留PDF图案单元与页面坐标之间的仿射关系
|
|
// 入参: source 平铺图案, base 无色图案基色
|
|
// 返回: *Pattern OFD图案, error 转换错误
|
|
func (p *pdfImporter) buildTilingPattern(source *pdfgo.TilingPattern, base pdfgo.Paint) (*Pattern, error) {
|
|
const unit = 25.4 / 72
|
|
box := source.BBox
|
|
width, height := box.XMax-box.XMin, box.YMax-box.YMin
|
|
xstep, ystep := math.Abs(source.XStep), math.Abs(source.YStep)
|
|
if (width == 0 || height == 0) && finite(width) && finite(height) && width >= 0 && height >= 0 && finite(xstep) && finite(ystep) && xstep > 0 && ystep > 0 {
|
|
return &Pattern{Width: unit, Height: unit, XStep: unit, YStep: unit, RelativeTo: "Page"}, nil
|
|
}
|
|
if width <= 0 || height <= 0 || !finite(xstep) || !finite(ystep) || xstep == 0 || ystep == 0 {
|
|
return nil, fmt.Errorf("invalid PDF tiling pattern dimensions")
|
|
}
|
|
dense, continuousX, continuousY, err := p.densePatternRectangle(source, base, width, height, xstep, ystep)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if dense != nil {
|
|
box, err = p.densePatternPageBounds(source, box, continuousX, continuousY)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
bounds, _ := pdfPatternRectangleBounds(dense.Path)
|
|
if continuousX {
|
|
bounds.XMin, bounds.XMax = box.XMin, box.XMax
|
|
}
|
|
if continuousY {
|
|
bounds.YMin, bounds.YMax = box.YMin, box.YMax
|
|
}
|
|
dense.Path, dense.Style.Clips = pdfPatternRectanglePath(bounds), []pdfgo.Path{pdfPatternRectanglePath(box)}
|
|
width, height = box.XMax-box.XMin, box.YMax-box.YMin
|
|
}
|
|
localToPDF := pdfgo.Matrix{1 / unit, 0, 0, -1 / unit, box.XMin, box.YMax}
|
|
m := p.matrix.Mul(source.Matrix).Mul(localToPDF)
|
|
pattern := &Pattern{Width: math.Min(width, xstep) * unit, Height: math.Min(height, ystep) * unit, XStep: xstep * unit, YStep: ystep * unit, RelativeTo: "Page", CTM: pdfNumbers(m[:]...)}
|
|
repeatWidth, repeatHeight := width, height
|
|
if continuousX {
|
|
pattern.Width, repeatWidth = width*unit, xstep
|
|
}
|
|
if continuousY {
|
|
pattern.Height, repeatHeight = height*unit, ystep
|
|
}
|
|
if p.patternOrigins == nil {
|
|
p.patternOrigins = make(map[*Pattern]pdfObjectOrigin)
|
|
}
|
|
cell := *p
|
|
cell.matrix = pdfgo.Matrix{unit, 0, 0, -unit, -box.XMin * unit, box.YMax * unit}
|
|
cell.pageWidth, cell.pageHeight = width*unit, height*unit
|
|
cell.rasterDPI *= m.MaxScale()
|
|
if !finite(cell.rasterDPI) || cell.rasterDPI < 0 {
|
|
return nil, fmt.Errorf("invalid PDF tiling pattern raster scale")
|
|
}
|
|
cell.objects = nil
|
|
nodes, err := cell.collectCompositeNodes(func(visitor pdfgo.Visitor) error {
|
|
if dense != nil {
|
|
return visitor.Path(*dense)
|
|
}
|
|
return source.Walk(p.ctx, base, visitor)
|
|
}, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
visitor := cell.visitor()
|
|
for _, node := range nodes {
|
|
if err := p.ctx.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
if err := node.emit(visitor); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
objects, err := cell.objects.data(p.ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
pattern.CellContent.Objects = objects
|
|
if len(objects) != 0 && (repeatWidth > xstep || repeatHeight > ystep) {
|
|
columns, rows := math.Ceil(repeatWidth/xstep), math.Ceil(repeatHeight/ystep)
|
|
if !finite(columns*rows) || columns*rows >= float64(uint(^uint(0)>>1)+1) {
|
|
return nil, fmt.Errorf("PDF tiling pattern instance count overflow")
|
|
}
|
|
bounds := make([]Box, len(objects))
|
|
for index, object := range objects {
|
|
boundary := ""
|
|
if object.Type == "PathObject" && object.PathObject.Stroke != nil && !*object.PathObject.Stroke {
|
|
boundary = object.PathObject.Boundary
|
|
} else if object.Type == "ImageObject" && object.ImageObject.Border == nil {
|
|
boundary = object.ImageObject.Boundary
|
|
}
|
|
if boundary != "" {
|
|
var err error
|
|
bounds[index], err = ParseBox(boundary)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
}
|
|
left, right, top, bottom := pdfPatternInstanceRange(columns, rows, bounds, pattern.Width, pattern.Height, xstep*unit, ystep*unit)
|
|
var expanded pdfObjectBuffer
|
|
for row := top; row <= bottom; row++ {
|
|
for column := left; column <= right; column++ {
|
|
if err := p.ctx.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
dx, dy := float64(column)*xstep*unit, float64(row)*ystep*unit
|
|
for index, object := range objects {
|
|
if index%pdfObjectChunkLimit == 0 {
|
|
if err := p.ctx.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
b := bounds[index]
|
|
if b.W > 0 && b.H > 0 && (b.X+dx+b.W < 0 || b.Y+dy+b.H < 0 || b.X+dx > pattern.Width || b.Y+dy > pattern.Height) {
|
|
continue
|
|
}
|
|
translated, err := transformEditorObject(object, dx, dy, 1)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
expanded.appendOrigin(translated, cell.objects.origin(index))
|
|
}
|
|
}
|
|
}
|
|
pattern.CellContent.Objects, err = expanded.data(p.ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
cell.objects = &expanded
|
|
}
|
|
origin, err := cell.objects.combinedOrigin(p.ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if origin.sequence != nil || len(origin.related) != 0 {
|
|
p.patternOrigins[pattern] = origin
|
|
}
|
|
return pattern, nil
|
|
}
|