mirror of
https://github.com/xiaoqidun/ofdgo.git
synced 2026-10-29 15:06:46 +08:00
253 lines
7.7 KiB
Go
253 lines
7.7 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 (
|
|
"bufio"
|
|
"bytes"
|
|
"crypto/sha256"
|
|
"encoding/base64"
|
|
"fmt"
|
|
"html"
|
|
"image"
|
|
"image/png"
|
|
"io"
|
|
"strings"
|
|
|
|
"github.com/tdewolff/canvas"
|
|
canvasimage "github.com/tdewolff/canvas/image"
|
|
"github.com/tdewolff/canvas/renderers/svg"
|
|
"github.com/xiaoqidun/pdfgo"
|
|
)
|
|
|
|
// svgCreatorMetadata 记录SVG制作软件，不参与页面绘制
|
|
const svgCreatorMetadata = `<metadata>` + creatorXMP + `</metadata>`
|
|
|
|
// svgResourceRenderer 分离外部资源的SVG渲染器
|
|
type svgResourceRenderer struct {
|
|
*svg.SVG
|
|
renderer *Renderer
|
|
writer io.Writer
|
|
fonts []SVGFont
|
|
images []SVGImage
|
|
imageNames map[image.Image]string
|
|
seen map[string]bool
|
|
styles strings.Builder
|
|
err error
|
|
objects map[*GraphicObject]string
|
|
objectStack []svgObjectGroup
|
|
embeddedFonts bool
|
|
}
|
|
|
|
// svgObjectGroup 记录编辑分组的稳定路径，不把底纹内部绘制当作成员
|
|
type svgObjectGroup struct {
|
|
path string
|
|
next int
|
|
composite bool
|
|
}
|
|
|
|
// RenderImage 按输出策略编码图片，分离资源时保持尺寸和变换
|
|
// 入参: img 图片对象, m 变换矩阵
|
|
func (s *svgResourceRenderer) RenderImage(img image.Image, m canvas.Matrix) {
|
|
name, ok := s.imageNames[img]
|
|
var resource SVGImage
|
|
if !ok {
|
|
resource = SVGImage{MIME: "image/png"}
|
|
if encoded, ok := img.(*canvasimage.Image); ok {
|
|
resource.MIME, resource.Data = encoded.Mimetype, encoded.Bytes
|
|
} else {
|
|
var buffer bytes.Buffer
|
|
if err := png.Encode(&buffer, img); err != nil {
|
|
s.err = err
|
|
return
|
|
}
|
|
resource.Data = buffer.Bytes()
|
|
}
|
|
if s.renderer != nil && s.renderer.Compression.Mode != CompressionUnchanged {
|
|
data, err := pdfgo.OptimizeImage(s.renderer.outputContext(), resource.Data, s.renderer.Compression)
|
|
if err != nil {
|
|
if s.err = s.renderer.outputContext().Err(); s.err != nil {
|
|
return
|
|
}
|
|
} else {
|
|
resource.Data = data
|
|
}
|
|
}
|
|
if s.imageNames != nil {
|
|
name = fmt.Sprintf("ofdgo-image-%x", sha256.Sum256(resource.Data))
|
|
resource.Name = name
|
|
s.imageNames[img] = name
|
|
if !s.seen[name] {
|
|
s.seen[name] = true
|
|
s.images = append(s.images, resource)
|
|
}
|
|
}
|
|
}
|
|
_, height := s.Size()
|
|
size := img.Bounds().Size()
|
|
m = m.Translate(0, float64(size.Y))
|
|
// 直接写出换轴后的矩阵，原坐标平移为零时仍需保留页面高度补偿。
|
|
matrix := Matrix{a: m[0][0], b: -m[1][0], c: -m[0][1], d: m[1][1], e: m[0][2], f: height - m[1][2]}
|
|
fmt.Fprintf(s.writer, `<image transform="matrix(%s)" width="%d" height="%d" xlink:href="`, matrix.String(), size.X, size.Y)
|
|
if s.imageNames != nil {
|
|
io.WriteString(s.writer, name)
|
|
} else {
|
|
fmt.Fprintf(s.writer, "data:%s;base64,", resource.MIME)
|
|
encoder := base64.NewEncoder(base64.StdEncoding, s.writer)
|
|
if _, err := encoder.Write(resource.Data); err != nil {
|
|
s.err = err
|
|
return
|
|
}
|
|
if err := encoder.Close(); err != nil {
|
|
s.err = err
|
|
return
|
|
}
|
|
}
|
|
io.WriteString(s.writer, `"/>`)
|
|
}
|
|
|
|
// RenderText 绘制文字并记录实际使用的字体资源
|
|
// 入参: text 文字对象, m 变换矩阵
|
|
func (s *svgResourceRenderer) RenderText(text *canvas.Text, m canvas.Matrix) {
|
|
if s.embeddedFonts {
|
|
s.SVG.RenderText(text, m)
|
|
return
|
|
}
|
|
if text.Empty() {
|
|
return
|
|
}
|
|
font := text.MostCommonFontFace().Font
|
|
resource, ok := s.renderer.canvasState().svgFontCache[font]
|
|
if !ok {
|
|
data := fontSFNTData(font.SFNT)
|
|
resource = SVGFont{
|
|
Name: fmt.Sprintf("ofdgo-%x-%d", sha256.Sum256(data), font.Style()),
|
|
Weight: font.Style().CSS(),
|
|
Style: "normal",
|
|
Data: data,
|
|
}
|
|
if font.Style().Italic() {
|
|
resource.Style = "italic"
|
|
}
|
|
s.renderer.canvasState().svgFontCache[font] = resource
|
|
}
|
|
if !s.seen[resource.Name] {
|
|
s.seen[resource.Name] = true
|
|
s.fonts = append(s.fonts, resource)
|
|
fmt.Fprintf(&s.styles, ".%s{font-family:%s!important}", resource.Name, resource.Name)
|
|
}
|
|
s.SetClass(resource.Name)
|
|
s.SVG.RenderText(text, m)
|
|
s.SetClass()
|
|
}
|
|
|
|
// renderSVGResources 渲染SVG并收集外部资源
|
|
// 入参: page 页面内容, writer 输出流, images 是否分离图片, objects 是否标识页面直接对象
|
|
// 返回: SVGResources 外部资源, error 错误信息
|
|
func (r *Renderer) renderSVGResources(page *PageContent, writer io.Writer, images, objects bool) (SVGResources, error) {
|
|
box, err := r.GetPageBox(page)
|
|
if err != nil {
|
|
return SVGResources{}, err
|
|
}
|
|
options := svg.DefaultOptions
|
|
options.EmbedFonts = false
|
|
buffer := bufio.NewWriter(writer)
|
|
s := &svgResourceRenderer{
|
|
SVG: svg.New(buffer, box.W, box.H, &options),
|
|
renderer: r,
|
|
writer: buffer,
|
|
seen: make(map[string]bool),
|
|
}
|
|
io.WriteString(buffer, svgCreatorMetadata)
|
|
// 图片方向由文档变换决定，不再由浏览器应用EXIF方向。
|
|
io.WriteString(buffer, `<g style="image-orientation:none">`)
|
|
if images {
|
|
s.imageNames = make(map[image.Image]string)
|
|
}
|
|
if objects {
|
|
s.objects = make(map[*GraphicObject]string)
|
|
for i := range page.Content.Layer {
|
|
for j := range page.Content.Layer[i].Objects {
|
|
object := &page.Content.Layer[i].Objects[j]
|
|
s.objects[object] = editorObjectID(*object)
|
|
}
|
|
}
|
|
if err := r.renderCanvasPageToContext(canvas.NewContext(s), page, !r.TransparentBackground); err != nil {
|
|
return SVGResources{}, err
|
|
}
|
|
} else {
|
|
c, err := r.renderCanvasPage(page)
|
|
if err != nil {
|
|
return SVGResources{}, err
|
|
}
|
|
c.RenderTo(s)
|
|
}
|
|
if s.err != nil {
|
|
return SVGResources{}, s.err
|
|
}
|
|
io.WriteString(buffer, `</g>`)
|
|
s.SetCustomStyle(s.styles.String())
|
|
if err := s.Close(); err != nil {
|
|
return SVGResources{}, err
|
|
}
|
|
return SVGResources{Fonts: s.fonts, Images: s.images}, buffer.Flush()
|
|
}
|
|
|
|
// beginObject 标记直接对象及复合或注解成员，模板图元不单独标记
|
|
// 入参: object 图形对象
|
|
// 返回: bool 是否写入分组
|
|
func (s *svgResourceRenderer) beginObject(object *GraphicObject) bool {
|
|
id := s.objects[object]
|
|
attribute := "data-ofd-object"
|
|
if id == "" {
|
|
if len(s.objectStack) == 0 || !s.objectStack[len(s.objectStack)-1].composite {
|
|
return false
|
|
}
|
|
parent := &s.objectStack[len(s.objectStack)-1]
|
|
id = fmt.Sprintf("%s/%d", parent.path, parent.next)
|
|
parent.next++
|
|
attribute = "data-ofd-child"
|
|
}
|
|
s.objectStack = append(s.objectStack, svgObjectGroup{path: id, composite: object.Type == "CompositeObject" || object.Type == "CompositeGraphicUnit"})
|
|
fmt.Fprintf(s.writer, `<g %s="%s">`, attribute, html.EscapeString(id))
|
|
return true
|
|
}
|
|
|
|
// endObject 结束对象分组
|
|
func (s *svgResourceRenderer) endObject() {
|
|
s.objectStack = s.objectStack[:len(s.objectStack)-1]
|
|
fmt.Fprint(s.writer, `</g>`)
|
|
}
|
|
|
|
// beginAnnotation 标记完整注解外观，内部图元保持原有层级和绘制顺序
|
|
// 入参: id 注解标识
|
|
// 返回: bool 是否写入分组
|
|
func (s *svgResourceRenderer) beginAnnotation(id string) bool {
|
|
if s.objects == nil {
|
|
return false
|
|
}
|
|
s.objectStack = append(s.objectStack, svgObjectGroup{path: "annotation:" + id, composite: true})
|
|
fmt.Fprintf(s.writer, `<g data-ofd-object="annotation:%s">`, html.EscapeString(id))
|
|
return true
|
|
}
|
|
|
|
// skipObjects 为不可见资源保留内部成员序号
|
|
// 入参: count 直接成员数量
|
|
func (s *svgResourceRenderer) skipObjects(count int) {
|
|
if len(s.objectStack) > 0 {
|
|
s.objectStack[len(s.objectStack)-1].next += count
|
|
}
|
|
}
|