mirror of
https://github.com/xiaoqidun/ofdgo.git
synced 2026-10-29 17:27:08 +08:00
405 lines
14 KiB
Go
405 lines
14 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"
|
|
|
|
codecpdf "github.com/xiaoqidun/ofdgo/internal/codec/pdf"
|
|
"github.com/xiaoqidun/pdfgo"
|
|
)
|
|
|
|
// pdfCompositePattern 保存图案单元的原始图元和独立几何缓存
|
|
type pdfCompositePattern struct {
|
|
nodes []pdfCompositeNode
|
|
local pdfImporter
|
|
box Box
|
|
transferChecked bool
|
|
transferOpaque bool
|
|
}
|
|
|
|
// tiling 将图案作为非隔离组绘制，保留背景贡献及单元间隙
|
|
// 入参: paint 图案画刷, backdrop 初始背景, space 混合空间, channels 可选源通道输出
|
|
// 返回: []pdfCompositePixel 去除背景贡献的图案像素, error 解析或绘制错误
|
|
func (c *pdfCompositor) tiling(paint pdfgo.Paint, backdrop []pdfCompositePixel, space *pdfgo.ColorSpace, channels []uint8) ([]pdfCompositePixel, error) {
|
|
source := paint.Tiling
|
|
matrix := c.importer.matrix.Mul(source.Matrix)
|
|
inverse, ok := matrix.Inverse()
|
|
if !ok {
|
|
return nil, fmt.Errorf("invalid PDF tiling pattern matrix")
|
|
}
|
|
pattern, err := c.importer.compositePattern(paint)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
xstep, ystep := math.Abs(source.XStep), math.Abs(source.YStep)
|
|
area := pdfBounds([]pdfgo.Point{
|
|
inverse.Apply(pdfgo.Point{X: c.box.X, Y: c.box.Y}), inverse.Apply(pdfgo.Point{X: c.box.X + c.box.W, Y: c.box.Y}),
|
|
inverse.Apply(pdfgo.Point{X: c.box.X + c.box.W, Y: c.box.Y + c.box.H}), inverse.Apply(pdfgo.Point{X: c.box.X, Y: c.box.Y + c.box.H}),
|
|
})
|
|
left, right := math.Ceil((area.X-source.BBox.XMax)/xstep), math.Floor((area.X+area.W-source.BBox.XMin)/xstep)
|
|
top, bottom := math.Ceil((area.Y-source.BBox.YMax)/ystep), math.Floor((area.Y+area.H-source.BBox.YMin)/ystep)
|
|
limit := float64(int(^uint(0) >> 1))
|
|
for _, value := range []float64{left, right, top, bottom} {
|
|
if !finite(value) || value <= -limit || value >= limit {
|
|
return nil, fmt.Errorf("PDF tiling pattern instance range overflow")
|
|
}
|
|
}
|
|
result := c.acquirePixels(len(backdrop))
|
|
complete := false
|
|
defer func() {
|
|
if !complete {
|
|
c.releasePixels(result)
|
|
}
|
|
}()
|
|
for i, pixel := range backdrop {
|
|
if i%pdfCompositeTileSize == 0 {
|
|
if err := c.importer.ctx.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
result[i] = pdfCompositePixel{values: pixel.values, alpha: pixel.alpha}
|
|
}
|
|
candidate := c.acquirePixels(len(backdrop))
|
|
defer c.releasePixels(candidate)
|
|
var candidateChannels []uint8
|
|
if channels != nil {
|
|
candidateChannels = c.acquireChannels(len(backdrop))
|
|
defer c.releaseChannels(candidateChannels)
|
|
}
|
|
disjoint := source.BBox.XMax-source.BBox.XMin <= xstep && source.BBox.YMax-source.BBox.YMin <= ystep
|
|
pixelStep := 25.4 / c.importer.rasterDPI
|
|
localInverse, _ := pattern.local.matrix.Inverse()
|
|
for row := int(top); row <= int(bottom); row++ {
|
|
for column := int(left); column <= int(right); column++ {
|
|
if err := c.importer.ctx.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
x, y := float64(column)*xstep, float64(row)*ystep
|
|
dx, dy := matrix[0]*x+matrix[2]*y+pattern.box.X, matrix[1]*x+matrix[3]*y+pattern.box.Y
|
|
left := int(math.Max(0, math.Min(float64(c.width), math.Floor((dx-c.box.X)/pixelStep)-1)))
|
|
right := int(math.Max(0, math.Min(float64(c.width), math.Ceil((dx+pattern.box.W-c.box.X)/pixelStep)+1)))
|
|
top := int(math.Max(0, math.Min(float64(c.height), math.Floor((dy-c.box.Y)/pixelStep)-1)))
|
|
bottom := int(math.Max(0, math.Min(float64(c.height), math.Ceil((dy+pattern.box.H-c.box.Y)/pixelStep)+1)))
|
|
width, height := right-left, bottom-top
|
|
if width <= 0 || height <= 0 {
|
|
continue
|
|
}
|
|
pixels := candidate[:width*height]
|
|
var cellChannels []uint8
|
|
if channels != nil {
|
|
cellChannels = candidateChannels[:width*height]
|
|
}
|
|
for y := range height {
|
|
if err := c.importer.ctx.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
for x := range width {
|
|
index := (top+y)*c.width + left + x
|
|
initial := result[index]
|
|
if disjoint {
|
|
initial = backdrop[index]
|
|
}
|
|
pixels[y*width+x] = pdfCompositePixel{values: initial.values, alpha: initial.alpha}
|
|
if cellChannels != nil {
|
|
cellChannels[y*width+x] = 0
|
|
if !disjoint {
|
|
cellChannels[y*width+x] = channels[index]
|
|
}
|
|
}
|
|
}
|
|
}
|
|
cell := pdfCompositor{importer: &pattern.local, box: Box{X: c.box.X + float64(left)*pixelStep - dx, Y: c.box.Y + float64(top)*pixelStep - dy, W: float64(width) * pixelStep, H: float64(height) * pixelStep},
|
|
width: width, height: height, inverse: localInverse, cache: pattern.local.compositingCache(), scratch: c.scratch, masks: map[*pdfgo.SoftMask][]float64{}, softMask: c.softMask, channels: cellChannels}
|
|
err := cell.draw(pattern.nodes, pixels, space)
|
|
cell.releaseMasks()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for y := range height {
|
|
if err := c.importer.ctx.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
for x := range width {
|
|
index := (top+y)*c.width + left + x
|
|
pdfCompositePatternCell(&result[index], backdrop[index], pixels[y*width+x], disjoint)
|
|
if channels != nil && pixels[y*width+x].shape != 0 {
|
|
if disjoint {
|
|
channels[index] |= cellChannels[y*width+x]
|
|
} else {
|
|
channels[index] = cellChannels[y*width+x]
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
for i := range result {
|
|
if i%pdfCompositeTileSize == 0 {
|
|
if err := c.importer.ctx.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
pixel := &result[i]
|
|
if pixel.effect != 0 {
|
|
for j := 0; j < space.Components(); j++ {
|
|
pixel.values[j] = math.Max(0, math.Min(1, (pixel.values[j]*pixel.alpha-backdrop[i].values[j]*backdrop[i].alpha*(1-pixel.effect))/pixel.effect))
|
|
}
|
|
}
|
|
pixel.alpha = pixel.effect
|
|
}
|
|
complete = true
|
|
return result, nil
|
|
}
|
|
|
|
// pdfCompositePatternCell 合并互不重叠单元的覆盖贡献，避免相邻抗锯齿边缘重复混合
|
|
// 入参: target 图案累计结果, initial 原始背景, source 单元结果, disjoint 单元边界是否互不重叠
|
|
func pdfCompositePatternCell(target *pdfCompositePixel, initial, source pdfCompositePixel, disjoint bool) {
|
|
if source.shape == 0 {
|
|
return
|
|
}
|
|
if disjoint {
|
|
alpha := math.Max(0, math.Min(1, target.alpha+source.alpha-initial.alpha))
|
|
if alpha != 0 {
|
|
for i := range target.values {
|
|
target.values[i] = (target.values[i]*target.alpha + source.values[i]*source.alpha - initial.values[i]*initial.alpha) / alpha
|
|
}
|
|
}
|
|
target.alpha = alpha
|
|
target.effect = math.Min(1, target.effect+source.effect)
|
|
target.shape = math.Min(1, target.shape+source.shape)
|
|
} else {
|
|
target.values, target.alpha = source.values, source.alpha
|
|
target.effect += source.effect * (1 - target.effect)
|
|
target.shape += source.shape * (1 - target.shape)
|
|
}
|
|
}
|
|
|
|
// compositePattern 复用图案单元解析及几何上下文，供检查和局部合成共同使用
|
|
// 入参: paint 图案画刷
|
|
// 返回: *pdfCompositePattern 单元内容, error 解析或边界错误
|
|
func (p *pdfImporter) compositePattern(paint pdfgo.Paint) (*pdfCompositePattern, error) {
|
|
paint.Alpha = 1
|
|
cache := p.compositingCache()
|
|
if cache.patterns == nil {
|
|
cache.patterns = make(map[pdfgo.Paint]*pdfCompositePattern)
|
|
}
|
|
if pattern := cache.patterns[paint]; pattern != nil {
|
|
return pattern, nil
|
|
}
|
|
source := paint.Tiling
|
|
nodes, err := p.collectCompositeNodes(func(v pdfgo.Visitor) error { return source.Walk(p.ctx, paint, v) }, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
matrix := p.matrix.Mul(source.Matrix)
|
|
b := source.BBox
|
|
box := pdfBounds([]pdfgo.Point{
|
|
matrix.Apply(pdfgo.Point{X: b.XMin, Y: b.YMin}), matrix.Apply(pdfgo.Point{X: b.XMax, Y: b.YMin}),
|
|
matrix.Apply(pdfgo.Point{X: b.XMax, Y: b.YMax}), matrix.Apply(pdfgo.Point{X: b.XMin, Y: b.YMax}),
|
|
})
|
|
if box.W <= 0 || box.H <= 0 {
|
|
return nil, fmt.Errorf("invalid PDF tiling pattern bounds")
|
|
}
|
|
local := *p
|
|
local.matrix = matrix
|
|
local.matrix[4] -= box.X
|
|
local.matrix[5] -= box.Y
|
|
local.pageWidth, local.pageHeight = box.W, box.H
|
|
local.compositeCache = nil
|
|
local.compositingCache().geometry = cache.geometryCache()
|
|
local.compositingCache().clips = cache.clipCache()
|
|
pattern := &pdfCompositePattern{nodes: nodes, local: local, box: box}
|
|
cache.patterns[paint] = pattern
|
|
return pattern, nil
|
|
}
|
|
|
|
// directCompositeNode 检查图元、透明组和图案单元能否独立保留，逐层核对背景混合
|
|
// 入参: node 图元, space 混合空间
|
|
// 返回: bool 是否可直接转换, error 图案或遮罩解析错误
|
|
func (p *pdfImporter) directCompositeNode(node pdfCompositeNode, space *pdfgo.ColorSpace) (bool, error) {
|
|
if node.group == nil && space.Device() {
|
|
style, fill, stroke := node.style()
|
|
if node.image == nil || node.image.Image.ImageMask {
|
|
for i, paint := range [2]pdfgo.Paint{style.Fill, style.Stroke} {
|
|
if i == 0 && !fill || i == 1 && !stroke {
|
|
continue
|
|
}
|
|
if native := codecpdf.PrepareShadingColorants(paint, space, false); native != nil {
|
|
if !native.Process.AlternateEquivalent() || i == 0 && style.FillOverprint || i == 1 && style.StrokeOverprint {
|
|
return false, nil
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if space.Model == "DeviceCMYK" && style.ColorConversion != (pdfgo.ColorConversion{}) {
|
|
if node.image != nil && !node.image.Image.ImageMask {
|
|
_, native, err := pdfImageProcessColorants(node.image.Image)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if !native {
|
|
source := p.compositingCache().images[node.image.Image]
|
|
if source == nil {
|
|
source, err = node.image.Image.DecodeComponentsViewContext(p.ctx)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
p.compositingCache().images[node.image.Image] = source
|
|
}
|
|
if source.Space.Model == "DeviceRGB" && source.Space.Device() {
|
|
return false, nil
|
|
}
|
|
}
|
|
} else {
|
|
for i, paint := range [2]pdfgo.Paint{style.Fill, style.Stroke} {
|
|
if i == 0 && !fill || i == 1 && !stroke || paint.None || paint.Tiling != nil {
|
|
continue
|
|
}
|
|
source := paint.Space
|
|
switch {
|
|
case paint.Axial != nil:
|
|
source = paint.Axial.Space
|
|
case paint.Radial != nil:
|
|
source = paint.Radial.Space
|
|
case paint.Function != nil:
|
|
source = paint.Function.Space
|
|
case paint.Mesh != nil:
|
|
source = paint.Mesh.Space
|
|
}
|
|
if source != nil && source.Model == "DeviceRGB" && source.Device() || source == nil && paint.CMYK == nil && paint.SourceSpace != "DeviceGray" {
|
|
return false, nil
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
directGroup := false
|
|
if group := node.group; group != nil && !group.Knockout && group.Alpha == 1 && group.SoftMask == nil && codecpdf.NormalBlend(group.BlendMode) && (group.ColorSpace == nil || group.ColorSpace.SRGBEquivalent()) {
|
|
directGroup = true
|
|
}
|
|
opaque, err := p.opaque(node, space)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if !opaque && !(space.SRGBEquivalent() && (directGroup || node.direct())) {
|
|
if node.textObject != nil && space.SRGBEquivalent() {
|
|
for _, child := range node.children {
|
|
if direct, err := p.directCompositeNode(child, space); err != nil || !direct {
|
|
return direct, err
|
|
}
|
|
}
|
|
return p.disjointText(node)
|
|
}
|
|
return false, nil
|
|
}
|
|
if node.group != nil {
|
|
for _, child := range node.children {
|
|
if direct, err := p.directCompositeNode(child, space); err != nil || !direct {
|
|
return direct, err
|
|
}
|
|
}
|
|
return true, nil
|
|
}
|
|
if node.image != nil && !node.image.Image.ImageMask {
|
|
return true, nil
|
|
}
|
|
style, fill, stroke := node.style()
|
|
for i, paint := range []pdfgo.Paint{style.Fill, style.Stroke} {
|
|
if i == 0 && !fill || i == 1 && !stroke || paint.Tiling == nil {
|
|
continue
|
|
}
|
|
pattern, err := p.compositePattern(paint)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
for _, child := range pattern.nodes {
|
|
if direct, err := pattern.local.directCompositeNode(child, space); err != nil || !direct {
|
|
return direct, err
|
|
}
|
|
}
|
|
}
|
|
return true, nil
|
|
}
|
|
|
|
// processOverprint 检查图元、透明组和图案单元是否需要设备四色套印
|
|
// 入参: nodes 原始图元
|
|
// 返回: bool 是否需要四色混合, error 图像或图案解析错误
|
|
func (p *pdfImporter) processOverprint(nodes []pdfCompositeNode) (bool, error) {
|
|
for _, node := range nodes {
|
|
if node.group != nil {
|
|
if found, err := p.processOverprint(node.children); err != nil || found {
|
|
return found, err
|
|
}
|
|
continue
|
|
}
|
|
style, fill, stroke := node.style()
|
|
if node.image != nil && style.FillOverprint {
|
|
if _, found, err := pdfImageProcessColorants(node.image.Image); err != nil || found {
|
|
return found, err
|
|
}
|
|
}
|
|
if node.image != nil && !node.image.Image.ImageMask {
|
|
continue
|
|
}
|
|
if fill && style.FillOverprint && pdfOverprintNeedsSeparation(style.Fill) || stroke && style.StrokeOverprint && pdfOverprintNeedsSeparation(style.Stroke) {
|
|
return true, nil
|
|
}
|
|
if style.Transfer != nil || style.Halftone != nil {
|
|
if fill && style.FillOverprint && style.Fill.CMYK != nil || stroke && style.StrokeOverprint && style.Stroke.CMYK != nil {
|
|
transfer, err := p.resolveTransfer(style, "DeviceCMYK")
|
|
if err != nil || transfer != nil {
|
|
return transfer != nil, err
|
|
}
|
|
}
|
|
}
|
|
for i, paint := range []pdfgo.Paint{style.Fill, style.Stroke} {
|
|
if i == 0 && fill && style.FillOverprint || i == 1 && stroke && style.StrokeOverprint {
|
|
if native := codecpdf.PrepareShadingColorants(paint, &pdfgo.ColorSpace{Model: "DeviceCMYK"}, false); native != nil {
|
|
return true, nil
|
|
}
|
|
native, _, err := pdfCompositeNativePaint(paint, &pdfgo.ColorSpace{Model: "DeviceCMYK"}, style.RenderingIntent)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if native != nil && (!pdfOpaqueBlack(*native) || !pdfOpaqueBlack(paint)) {
|
|
return true, nil
|
|
}
|
|
}
|
|
if paint.Shading != nil && (i == 0 && fill || i == 1 && stroke) {
|
|
nodes, err := p.compositeShadingPattern(paint.Shading)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if found, err := p.processOverprint(nodes); err != nil || found {
|
|
return found, err
|
|
}
|
|
}
|
|
if i == 0 && !fill || i == 1 && !stroke || paint.Tiling == nil {
|
|
continue
|
|
}
|
|
pattern, err := p.compositePattern(paint)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if found, err := pattern.local.processOverprint(pattern.nodes); err != nil || found {
|
|
return found, err
|
|
}
|
|
}
|
|
}
|
|
return false, nil
|
|
}
|