mirror of
https://github.com/xiaoqidun/ofdgo.git
synced 2026-10-29 15:06:46 +08:00
334 lines
10 KiB
Go
334 lines
10 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 (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"maps"
|
|
|
|
"github.com/xiaoqidun/pdfgo"
|
|
)
|
|
|
|
// annotations 将链接、静态注解和表单外观按页面顺序转换为OFD对象
|
|
// 入参: ctx 取消上下文, page PDF页面, strict 严格检查开关
|
|
// 返回: error 错误信息
|
|
func (p *pdfImporter) annotations(ctx context.Context, page *pdfgo.Page, strict bool) error {
|
|
annotations, err := page.Annotations()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
parents := make(map[pdfgo.Reference]int)
|
|
for i, annotation := range annotations {
|
|
if annotation.Reference != (pdfgo.Reference{}) && annotation.Subtype != "Popup" {
|
|
parents[annotation.Reference] = i
|
|
}
|
|
}
|
|
attached := make(map[int]bool)
|
|
for i, annotation := range annotations {
|
|
if annotation.Subtype != "Popup" {
|
|
continue
|
|
}
|
|
popup, err := p.reader.ReadPopupAnnotation(annotation.Dictionary)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if parent, ok := parents[popup.Parent]; ok {
|
|
dict := maps.Clone(annotations[parent].Dictionary)
|
|
if previous := dict["Popup"]; previous != nil {
|
|
if reference, ok := previous.(pdfgo.Reference); !ok || reference != annotation.Reference {
|
|
return fmt.Errorf("inconsistent PDF popup parent")
|
|
}
|
|
}
|
|
dict["Popup"] = annotation.Dictionary
|
|
annotations[parent].Dictionary = dict
|
|
attached[i] = true
|
|
} else {
|
|
annotations[i].Dictionary = popup.Dictionary
|
|
}
|
|
}
|
|
for i, annotation := range annotations {
|
|
if attached[i] {
|
|
continue
|
|
}
|
|
dict := annotation.Dictionary
|
|
popupObject := dict["Popup"]
|
|
if annotation.Subtype == "Popup" {
|
|
popupObject = dict
|
|
}
|
|
if popupObject != nil {
|
|
popup, err := p.reader.ReadPopupAnnotation(popupObject)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if annotation.Subtype != "Popup" && popup.Parent != (pdfgo.Reference{}) && popup.Parent != annotation.Reference {
|
|
return fmt.Errorf("inconsistent PDF popup parent")
|
|
}
|
|
if popup.Open {
|
|
if strict {
|
|
return &pdfgo.UnsupportedError{Feature: "initially open annotation popup"}
|
|
}
|
|
p.warning(pdfgo.Diagnostic{Page: p.page + 1, Message: "PDF popup open state retained as annotation parameter"})
|
|
}
|
|
}
|
|
if annotation.Subtype == "Sound" {
|
|
sound, err := p.reader.ReadSound(dict["Sound"])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
data, err := pdfSoundWAV(sound)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
id, err := p.editor.AddMedia("Audio", "WAV", data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := p.appearanceAnnotation(ctx, page, annotation, Action{Event: "CLICK", Sound: &Sound{ResourceID: id}}); err != nil {
|
|
return err
|
|
}
|
|
continue
|
|
}
|
|
if annotation.Subtype == "Movie" {
|
|
if err := p.movieAnnotation(ctx, page, annotation); err != nil {
|
|
return err
|
|
}
|
|
continue
|
|
}
|
|
if annotation.Subtype == "FileAttachment" {
|
|
file, err := p.reader.ReadFileSpecification(dict["FS"])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if file.Embedded == nil {
|
|
return &pdfgo.UnsupportedError{Feature: "external file attachment"}
|
|
}
|
|
data, err := file.Embedded.Decode()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
id, err := p.editor.AddAttachment(file.Name, data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := p.appearanceAnnotation(ctx, page, annotation, Action{Event: "CLICK", GotoA: &GotoA{AttachID: id}}); err != nil {
|
|
return err
|
|
}
|
|
continue
|
|
}
|
|
if annotation.Subtype == "Widget" {
|
|
field, err := p.reader.ReadField(annotation)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
annotation.Dictionary = field
|
|
if field["FT"] == pdfgo.Name("Sig") && field["V"] != nil {
|
|
err = p.signatureWidget(ctx, page, annotation)
|
|
} else {
|
|
err = p.formWidget(ctx, page, annotation, strict)
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
continue
|
|
}
|
|
if annotation.Subtype != "Link" && pdfAnnotationType(annotation.Subtype) != "" {
|
|
if err := p.appearanceAnnotation(ctx, page, annotation); err != nil {
|
|
return err
|
|
}
|
|
continue
|
|
}
|
|
if annotation.Subtype != "Link" {
|
|
return &pdfgo.UnsupportedError{Feature: "annotation " + string(annotation.Subtype)}
|
|
}
|
|
for _, key := range []pdfgo.Name{"AA", "OC"} {
|
|
if dict[key] != nil {
|
|
return &pdfgo.UnsupportedError{Feature: fmt.Sprintf("link annotation field %q", key)}
|
|
}
|
|
}
|
|
action := Action{Event: "CLICK"}
|
|
target := dict["Dest"]
|
|
if dict["A"] != nil {
|
|
value, err := p.reader.Resolve(dict["A"])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
a, ok := value.(pdfgo.Dictionary)
|
|
if !ok {
|
|
return fmt.Errorf("invalid PDF link action")
|
|
}
|
|
if a["Next"] != nil {
|
|
return &pdfgo.UnsupportedError{Feature: "chained link actions"}
|
|
}
|
|
switch a["S"] {
|
|
case pdfgo.Name("GoTo"):
|
|
target = a["D"]
|
|
case pdfgo.Name("URI"):
|
|
value, err := p.reader.Resolve(a["URI"])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
uri, ok := value.(pdfgo.String)
|
|
if !ok {
|
|
return fmt.Errorf("invalid PDF URI")
|
|
}
|
|
isMap, err := p.reader.Resolve(a["IsMap"])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if isMap != nil && isMap != pdfgo.Boolean(false) && isMap != pdfgo.Boolean(true) {
|
|
return fmt.Errorf("invalid PDF URI image map flag")
|
|
}
|
|
if isMap == pdfgo.Boolean(true) {
|
|
return &pdfgo.UnsupportedError{Feature: "URI image map"}
|
|
}
|
|
base, err := p.reader.BaseURI()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
action.URI = &URI{URI: string(uri), Base: base}
|
|
default:
|
|
return &pdfgo.UnsupportedError{Feature: "link action"}
|
|
}
|
|
}
|
|
if action.URI == nil {
|
|
destination, err := p.reader.ReadDestination(target)
|
|
if err != nil {
|
|
if !strict && errors.Is(err, pdfgo.ErrDestinationNotFound) {
|
|
p.report.Warnings = append(p.report.Warnings, pdfgo.Diagnostic{Page: p.page + 1, Message: fmt.Sprintf("%v; link omitted", err)})
|
|
if err := p.appearanceAnnotation(ctx, page, annotation); err != nil {
|
|
return err
|
|
}
|
|
continue
|
|
}
|
|
return err
|
|
}
|
|
targetPage := p.pages[destination.Page]
|
|
if targetPage == nil {
|
|
if !strict {
|
|
p.report.Warnings = append(p.report.Warnings, pdfgo.Diagnostic{Page: p.page + 1, Message: "PDF link targets a page outside the document page tree; link omitted"})
|
|
if err := p.appearanceAnnotation(ctx, page, annotation); err != nil {
|
|
return err
|
|
}
|
|
continue
|
|
}
|
|
return fmt.Errorf("PDF destination page missing")
|
|
}
|
|
matrix, _, _ := pdfPageMatrix(targetPage)
|
|
dest := &Dest{Type: string(destination.Mode), PageID: p.pageIDs[destination.Page]}
|
|
if destination.Mode == "XYZ" {
|
|
values := [3]float64{}
|
|
retained := [3]bool{}
|
|
for n, v := range destination.Parameters {
|
|
switch v := v.(type) {
|
|
case pdfgo.Integer:
|
|
values[n] = float64(v)
|
|
case pdfgo.Real:
|
|
values[n] = float64(v)
|
|
case nil:
|
|
retained[n] = true
|
|
}
|
|
}
|
|
point := matrix.Apply(pdfgo.Point{X: values[0], Y: values[1]})
|
|
dest.Left, dest.Top, dest.Zoom = point.X, point.Y, values[2]
|
|
dest.OmitLeft, dest.OmitTop = retained[0], retained[1]
|
|
if targetPage.Rotation == 90 || targetPage.Rotation == 270 {
|
|
dest.OmitLeft, dest.OmitTop = retained[1], retained[0]
|
|
}
|
|
dest.OmitZoom = retained[2] || values[2] == 0
|
|
} else if destination.Mode == "FitR" {
|
|
values := [4]float64{}
|
|
for n, v := range destination.Parameters {
|
|
switch v := v.(type) {
|
|
case pdfgo.Integer:
|
|
values[n] = float64(v)
|
|
case pdfgo.Real:
|
|
values[n] = float64(v)
|
|
default:
|
|
return fmt.Errorf("invalid PDF FitR destination coordinate")
|
|
}
|
|
}
|
|
bounds := pdfBounds([]pdfgo.Point{
|
|
matrix.Apply(pdfgo.Point{X: values[0], Y: values[1]}),
|
|
matrix.Apply(pdfgo.Point{X: values[2], Y: values[3]}),
|
|
})
|
|
dest.Left, dest.Top = bounds.X, bounds.Y
|
|
dest.Right, dest.Bottom = bounds.X+bounds.W, bounds.Y+bounds.H
|
|
} else if destination.Mode != "Fit" {
|
|
return &pdfgo.UnsupportedError{Feature: "destination mode " + string(destination.Mode)}
|
|
}
|
|
action.Goto = &Goto{Dest: dest}
|
|
}
|
|
b := annotation.Rect
|
|
box := pdfBounds([]pdfgo.Point{p.matrix.Apply(pdfgo.Point{X: b.XMin, Y: b.YMin}), p.matrix.Apply(pdfgo.Point{X: b.XMax, Y: b.YMax})})
|
|
action.Region, err = p.linkRegion(annotation, box)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := p.appearanceAnnotation(ctx, page, annotation, action); err != nil {
|
|
return err
|
|
}
|
|
p.report.Links++
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// linkRegion 将PDF链接四边形映射为OFD动作区域，越出注解矩形时使用矩形
|
|
// 入参: annotation PDF链接注解, box OFD对象边界
|
|
// 返回: *Region 点击区域, error 解析错误
|
|
func (p *pdfImporter) linkRegion(annotation pdfgo.Annotation, box Box) (*Region, error) {
|
|
value, err := p.reader.Resolve(annotation.Dictionary["QuadPoints"])
|
|
if err != nil || value == nil {
|
|
return nil, err
|
|
}
|
|
array, ok := value.(pdfgo.Array)
|
|
if !ok || len(array) == 0 || len(array)%8 != 0 {
|
|
return nil, nil
|
|
}
|
|
region := &Region{Area: make([]RegionArea, 0, len(array)/8)}
|
|
for start := 0; start < len(array); start += 8 {
|
|
var points [4]pdfgo.Point
|
|
for index := range points {
|
|
coordinates := [2]float64{}
|
|
for axis := range coordinates {
|
|
resolved, err := p.reader.Resolve(array[start+index*2+axis])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
switch number := resolved.(type) {
|
|
case pdfgo.Integer:
|
|
coordinates[axis] = float64(number)
|
|
case pdfgo.Real:
|
|
coordinates[axis] = float64(number)
|
|
default:
|
|
return nil, nil
|
|
}
|
|
}
|
|
x, y := coordinates[0], coordinates[1]
|
|
if !finite(x) || !finite(y) || x < annotation.Rect.XMin || x > annotation.Rect.XMax || y < annotation.Rect.YMin || y > annotation.Rect.YMax {
|
|
return nil, nil
|
|
}
|
|
points[index] = p.matrix.Apply(pdfgo.Point{X: x, Y: y})
|
|
}
|
|
area := RegionArea{Start: pdfNumbers(points[0].X-box.X, points[0].Y-box.Y)}
|
|
for _, point := range points[1:] {
|
|
area.Command = append(area.Command, RegionCommand{Type: "Line", Point1: pdfNumbers(point.X-box.X, point.Y-box.Y)})
|
|
}
|
|
region.Area = append(region.Area, area)
|
|
}
|
|
return region, nil
|
|
}
|