unipdf/extractor/text.go
Gunnsteinn Hall 11f692bc3a
Font subsetting and font optimization improvements (#362)
* Track runes in IdentityEncoder (for subsetting), track decoded runes

* Working with the identity encoder in font_composite.go

* Add GetFilterArray to multi encoder.  Add comments.

* Add NewFromContents constructor to extractor only requiring contents and resources

* golint fixes

* Optimizer compress streams - improved detection of raw streams

* Optimize - CleanContentStream optimizer that removes redundant operands

* WIP Optimize - clean fonts

Will support both font file reduction and subsetting. (WIP)

* Optimize - image processing - try combined DCT and Flate

* Update options.go

* Update optimizer.go

* Create utils.go for optimize with common methods needed for optimization

* Optimizer - add font subsetting method

Covers XObject Forms, annotaitons etc.  Uses extractor package to extract text marks covering what fonts and glyphs are used.  Package truetype used for subsetting.

* Add some comments

* Fix cmap parsing rune conversion

* Error checking for extractor.  Add some comments.

* Update Jenkinsfile

* Update modules
2020-06-16 21:19:10 +00:00

1700 lines
51 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

/*
* This file is subject to the terms and conditions defined in
* file 'LICENSE.md', which is part of this source code package.
*/
package extractor
import (
"errors"
"fmt"
"math"
"sort"
"strings"
"unicode"
"github.com/unidoc/unipdf/v3/common"
"github.com/unidoc/unipdf/v3/contentstream"
"github.com/unidoc/unipdf/v3/core"
"github.com/unidoc/unipdf/v3/internal/transform"
"github.com/unidoc/unipdf/v3/model"
"golang.org/x/text/unicode/norm"
)
var (
errType = errors.New("type check error")
errRange = errors.New("range check error")
)
// ExtractText processes and extracts all text data in content streams and returns as a string.
// It takes into account character encodings in the PDF file, which are decoded by
// CharcodeBytesToUnicode.
// Characters that can't be decoded are replaced with MissingCodeRune ('\ufffd' = <20>).
func (e *Extractor) ExtractText() (string, error) {
text, _, _, err := e.ExtractTextWithStats()
return text, err
}
// ExtractTextWithStats works like ExtractText but returns the number of characters in the output
// (`numChars`) and the number of characters that were not decoded (`numMisses`).
func (e *Extractor) ExtractTextWithStats() (extracted string, numChars int, numMisses int, err error) {
pageText, numChars, numMisses, err := e.ExtractPageText()
if err != nil {
return "", numChars, numMisses, err
}
return pageText.Text(), numChars, numMisses, nil
}
// ExtractPageText returns the text contents of `e` (an Extractor for a page) as a PageText.
func (e *Extractor) ExtractPageText() (*PageText, int, int, error) {
pt, numChars, numMisses, err := e.extractPageText(e.contents, e.resources, transform.IdentityMatrix(), 0)
if err != nil {
return nil, numChars, numMisses, err
}
pt.computeViews()
procBuf(pt)
return pt, numChars, numMisses, err
}
// extractPageText returns the text contents of content stream `e` and resouces `resources` as a
// PageText.
// This can be called on a page or a form XObject.
func (e *Extractor) extractPageText(contents string, resources *model.PdfPageResources, parentCTM transform.Matrix, level int) (
*PageText, int, int, error) {
common.Log.Trace("extractPageText: level=%d", level)
pageText := &PageText{}
state := newTextState()
fontStack := fontStacker{}
to := newTextObject(e, resources, contentstream.GraphicsState{}, &state, &fontStack)
var inTextObj bool
cstreamParser := contentstream.NewContentStreamParser(contents)
operations, err := cstreamParser.Parse()
if err != nil {
common.Log.Debug("ERROR: extractPageText parse failed. err=%v", err)
return pageText, state.numChars, state.numMisses, err
}
processor := contentstream.NewContentStreamProcessor(*operations)
processor.AddHandler(contentstream.HandlerConditionEnumAllOperands, "",
func(op *contentstream.ContentStreamOperation, gs contentstream.GraphicsState,
resources *model.PdfPageResources) error {
operand := op.Operand
switch operand {
case "q":
if !fontStack.empty() {
common.Log.Trace("Save font state: %s\n%s",
fontStack.peek(), fontStack.String())
fontStack.push(fontStack.peek())
}
if state.tfont != nil {
common.Log.Trace("Save font state: %s\n->%s\n%s",
fontStack.peek(), state.tfont, fontStack.String())
fontStack.push(state.tfont)
}
case "Q":
if !fontStack.empty() {
common.Log.Trace("Restore font state: %s\n->%s\n%s",
fontStack.peek(), fontStack.get(-2), fontStack.String())
fontStack.pop()
}
if len(fontStack) >= 2 {
common.Log.Trace("Restore font state: %s\n->%s\n%s",
state.tfont, fontStack.peek(), fontStack.String())
state.tfont = fontStack.pop()
}
case "BT": // Begin text
// Begin a text object, initializing the text matrix, Tm, and
// the text line matrix, Tlm, to the identity matrix. Text
// objects shall not be nested. A second BT shall not appear
// before an ET. However, if that happens, all existing marks
// are added to the page marks, in order to avoid losing content.
if inTextObj {
common.Log.Debug("BT called while in a text object")
pageText.marks = append(pageText.marks, to.marks...)
}
inTextObj = true
graphicsState := gs
graphicsState.CTM = parentCTM.Mult(graphicsState.CTM)
to = newTextObject(e, resources, graphicsState, &state, &fontStack)
case "ET": // End Text
// End text object, discarding text matrix. If the current
// text object contains text marks, they are added to the
// page text marks collection.
// The ET operator should always have a matching BT operator.
// However, if ET appears outside of a text object, the behavior
// does not change: the text matrices are discarded and all
// existing marks in the text object are added to the page marks.
if !inTextObj {
common.Log.Debug("ET called outside of a text object")
}
inTextObj = false
pageText.marks = append(pageText.marks, to.marks...)
to.reset()
case "T*": // Move to start of next text line
to.nextLine()
case "Td": // Move text location
if ok, err := to.checkOp(op, 2, true); !ok {
common.Log.Debug("ERROR: err=%v", err)
return err
}
x, y, err := toFloatXY(op.Params)
if err != nil {
return err
}
to.moveText(x, y)
case "TD": // Move text location and set leading.
if ok, err := to.checkOp(op, 2, true); !ok {
common.Log.Debug("ERROR: err=%v", err)
return err
}
x, y, err := toFloatXY(op.Params)
if err != nil {
common.Log.Debug("ERROR: err=%v", err)
return err
}
to.moveTextSetLeading(x, y)
case "Tj": // Show text.
if ok, err := to.checkOp(op, 1, true); !ok {
common.Log.Debug("ERROR: Tj op=%s err=%v", op, err)
return err
}
charcodes, ok := core.GetStringBytes(op.Params[0])
if !ok {
common.Log.Debug("ERROR: Tj op=%s GetStringBytes failed", op)
return core.ErrTypeError
}
return to.showText(charcodes)
case "TJ": // Show text with adjustable spacing.
if ok, err := to.checkOp(op, 1, true); !ok {
common.Log.Debug("ERROR: TJ err=%v", err)
return err
}
args, ok := core.GetArray(op.Params[0])
if !ok {
common.Log.Debug("ERROR: TJ op=%s GetArrayVal failed", op)
return err
}
return to.showTextAdjusted(args)
case "'": // Move to next line and show text.
if ok, err := to.checkOp(op, 1, true); !ok {
common.Log.Debug("ERROR: ' err=%v", err)
return err
}
charcodes, ok := core.GetStringBytes(op.Params[0])
if !ok {
common.Log.Debug("ERROR: ' op=%s GetStringBytes failed", op)
return core.ErrTypeError
}
to.nextLine()
return to.showText(charcodes)
case `"`: // Set word and character spacing, move to next line, and show text.
if ok, err := to.checkOp(op, 3, true); !ok {
common.Log.Debug("ERROR: \" err=%v", err)
return err
}
x, y, err := toFloatXY(op.Params[:2])
if err != nil {
return err
}
charcodes, ok := core.GetStringBytes(op.Params[2])
if !ok {
common.Log.Debug("ERROR: \" op=%s GetStringBytes failed", op)
return core.ErrTypeError
}
to.setCharSpacing(x)
to.setWordSpacing(y)
to.nextLine()
return to.showText(charcodes)
case "TL": // Set text leading.
y, err := floatParam(op)
if err != nil {
common.Log.Debug("ERROR: TL err=%v", err)
return err
}
to.setTextLeading(y)
case "Tc": // Set character spacing.
y, err := floatParam(op)
if err != nil {
common.Log.Debug("ERROR: Tc err=%v", err)
return err
}
to.setCharSpacing(y)
case "Tf": // Set font.
if ok, err := to.checkOp(op, 2, true); !ok {
common.Log.Debug("ERROR: Tf err=%v", err)
return err
}
name, ok := core.GetNameVal(op.Params[0])
if !ok {
common.Log.Debug("ERROR: Tf op=%s GetNameVal failed", op)
return core.ErrTypeError
}
size, err := core.GetNumberAsFloat(op.Params[1])
if !ok {
common.Log.Debug("ERROR: Tf op=%s GetFloatVal failed. err=%v", op, err)
return err
}
err = to.setFont(name, size)
if err != nil {
return err
}
case "Tm": // Set text matrix.
if ok, err := to.checkOp(op, 6, true); !ok {
common.Log.Debug("ERROR: Tm err=%v", err)
return err
}
floats, err := core.GetNumbersAsFloat(op.Params)
if err != nil {
common.Log.Debug("ERROR: err=%v", err)
return err
}
to.setTextMatrix(floats)
case "Tr": // Set text rendering mode.
if ok, err := to.checkOp(op, 1, true); !ok {
common.Log.Debug("ERROR: Tr err=%v", err)
return err
}
mode, ok := core.GetIntVal(op.Params[0])
if !ok {
common.Log.Debug("ERROR: Tr op=%s GetIntVal failed", op)
return core.ErrTypeError
}
to.setTextRenderMode(mode)
case "Ts": // Set text rise.
if ok, err := to.checkOp(op, 1, true); !ok {
common.Log.Debug("ERROR: Ts err=%v", err)
return err
}
y, err := core.GetNumberAsFloat(op.Params[0])
if err != nil {
common.Log.Debug("ERROR: err=%v", err)
return err
}
to.setTextRise(y)
case "Tw": // Set word spacing.
if ok, err := to.checkOp(op, 1, true); !ok {
common.Log.Debug("ERROR: err=%v", err)
return err
}
y, err := core.GetNumberAsFloat(op.Params[0])
if err != nil {
common.Log.Debug("ERROR: err=%v", err)
return err
}
to.setWordSpacing(y)
case "Tz": // Set horizontal scaling.
if ok, err := to.checkOp(op, 1, true); !ok {
common.Log.Debug("ERROR: err=%v", err)
return err
}
y, err := core.GetNumberAsFloat(op.Params[0])
if err != nil {
common.Log.Debug("ERROR: err=%v", err)
return err
}
to.setHorizScaling(y)
case "Do":
// Handle XObjects by recursing through form XObjects.
if len(op.Params) == 0 {
common.Log.Debug("ERROR: expected XObject name operand for Do operator. Got %+v.", op.Params)
return errRange
}
// Get XObject name.
name, ok := core.GetName(op.Params[0])
if !ok {
common.Log.Debug("ERROR: invalid Do operator XObject name operand: %+v.", op.Params[0])
return errType
}
_, xtype := resources.GetXObjectByName(*name)
if xtype != model.XObjectTypeForm {
break
}
// Only process each form once.
formResult, ok := e.formResults[name.String()]
if !ok {
xform, err := resources.GetXObjectFormByName(*name)
if err != nil {
common.Log.Debug("ERROR: %v", err)
return err
}
formContent, err := xform.GetContentStream()
if err != nil {
common.Log.Debug("ERROR: %v", err)
return err
}
formResources := xform.Resources
if formResources == nil {
formResources = resources
}
tList, numChars, numMisses, err := e.extractPageText(string(formContent),
formResources, parentCTM.Mult(gs.CTM), level+1)
if err != nil {
common.Log.Debug("ERROR: %v", err)
return err
}
formResult = textResult{*tList, numChars, numMisses}
e.formResults[name.String()] = formResult
}
pageText.marks = append(pageText.marks, formResult.pageText.marks...)
state.numChars += formResult.numChars
state.numMisses += formResult.numMisses
}
return nil
})
err = processor.Process(resources)
if err != nil {
common.Log.Debug("ERROR: Processing: err=%v", err)
}
return pageText, state.numChars, state.numMisses, err
}
type textResult struct {
pageText PageText
numChars int
numMisses int
}
//
// Text operators
//
// moveText "Td" Moves start of text by `tx`,`ty`.
// Move to the start of the next line, offset from the start of the current line by (tx, ty).
// tx and ty are in unscaled text space units.
func (to *textObject) moveText(tx, ty float64) {
to.moveTo(tx, ty)
}
// moveTextSetLeading "TD" Move text location and set leading.
// Move to the start of the next line, offset from the start of the current line by (tx, ty). As a
// side effect, this operator shall set the leading parameter in the text state. This operator shall
// have the same effect as this code:
// ty TL
// tx ty Td
func (to *textObject) moveTextSetLeading(tx, ty float64) {
to.state.tl = -ty
to.moveTo(tx, ty)
}
// nextLine "T*"" Moves start of text line to next text line
// Move to the start of the next line. This operator has the same effect as the code
// 0 -Tl Td
// where Tl denotes the current leading parameter in the text state. The negative of Tl is used
// here because Tl is the text leading expressed as a positive number. Going to the next line
// entails decreasing the y coordinate. (page 250)
func (to *textObject) nextLine() {
to.moveTo(0, -to.state.tl)
}
// setTextMatrix "Tm".
// Set the text matrix, Tm, and the text line matrix, Tlm to the Matrix specified by the 6 numbers
// in `f` (page 250).
func (to *textObject) setTextMatrix(f []float64) {
if len(f) != 6 {
common.Log.Debug("ERROR: len(f) != 6 (%d)", len(f))
return
}
a, b, c, d, tx, ty := f[0], f[1], f[2], f[3], f[4], f[5]
to.tm = transform.NewMatrix(a, b, c, d, tx, ty)
to.tlm = to.tm
}
// showText "Tj". Show a text string.
func (to *textObject) showText(charcodes []byte) error {
return to.renderText(charcodes)
}
// showTextAdjusted "TJ". Show text with adjustable spacing.
func (to *textObject) showTextAdjusted(args *core.PdfObjectArray) error {
vertical := false
for _, o := range args.Elements() {
switch o.(type) {
case *core.PdfObjectFloat, *core.PdfObjectInteger:
x, err := core.GetNumberAsFloat(o)
if err != nil {
common.Log.Debug("ERROR: showTextAdjusted. Bad numerical arg. o=%s args=%+v", o, args)
return err
}
dx, dy := -x*0.001*to.state.tfs, 0.0
if vertical {
dy, dx = dx, dy
}
td := translationMatrix(transform.Point{X: dx, Y: dy})
to.tm.Concat(td)
common.Log.Trace("showTextAdjusted: dx,dy=%3f,%.3f Tm=%s", dx, dy, to.tm)
case *core.PdfObjectString:
charcodes, ok := core.GetStringBytes(o)
if !ok {
common.Log.Trace("showTextAdjusted: Bad string arg. o=%s args=%+v", o, args)
return core.ErrTypeError
}
err := to.renderText(charcodes)
if err != nil {
common.Log.Debug("Render text error: %v", err)
return err
}
default:
common.Log.Debug("ERROR: showTextAdjusted. Unexpected type (%T) args=%+v", o, args)
return core.ErrTypeError
}
}
return nil
}
// setTextLeading "TL". Set text leading.
func (to *textObject) setTextLeading(y float64) {
if to == nil || to.state == nil {
return
}
to.state.tl = y
}
// setCharSpacing "Tc". Set character spacing.
func (to *textObject) setCharSpacing(x float64) {
if to == nil {
return
}
to.state.tc = x
}
// setFont "Tf". Set font.
func (to *textObject) setFont(name string, size float64) error {
if to == nil {
return nil
}
font, err := to.getFont(name)
if err == nil {
to.state.tfont = font
if len(*to.fontStack) == 0 {
to.fontStack.push(font)
} else {
(*to.fontStack)[len(*to.fontStack)-1] = font
}
} else if err == model.ErrFontNotSupported {
// TODO(peterwilliams97): Do we need to handle this case in a special way?
return err
} else {
return err
}
to.state.tfs = size
return nil
}
// setTextRenderMode "Tr". Set text rendering mode.
func (to *textObject) setTextRenderMode(mode int) {
if to == nil {
return
}
to.state.tmode = RenderMode(mode)
}
// setTextRise "Ts". Set text rise.
func (to *textObject) setTextRise(y float64) {
if to == nil {
return
}
to.state.trise = y
}
// setWordSpacing "Tw". Set word spacing.
func (to *textObject) setWordSpacing(y float64) {
if to == nil {
return
}
to.state.tw = y
}
// setHorizScaling "Tz". Set horizontal scaling.
func (to *textObject) setHorizScaling(y float64) {
if to == nil {
return
}
to.state.th = y
}
// floatParam returns the single float parameter of operator `op`, or an error if it doesn't have
// a single float parameter or we aren't in a text stream.
func floatParam(op *contentstream.ContentStreamOperation) (float64, error) {
if len(op.Params) != 1 {
err := errors.New("incorrect parameter count")
common.Log.Debug("ERROR: %#q should have %d input params, got %d %+v",
op.Operand, 1, len(op.Params), op.Params)
return 0.0, err
}
return core.GetNumberAsFloat(op.Params[0])
}
// checkOp returns true if we are in a text stream and `op` has `numParams` params.
// If `hard` is true and the number of params don't match, an error is returned.
func (to *textObject) checkOp(op *contentstream.ContentStreamOperation, numParams int, hard bool) (
ok bool, err error) {
if to == nil {
var params []core.PdfObject
if numParams > 0 {
params = op.Params
if len(params) > numParams {
params = params[:numParams]
}
}
common.Log.Debug("%#q operand outside text. params=%+v", op.Operand, params)
}
if numParams >= 0 {
if len(op.Params) != numParams {
if hard {
err = errors.New("incorrect parameter count")
}
common.Log.Debug("ERROR: %#q should have %d input params, got %d %+v",
op.Operand, numParams, len(op.Params), op.Params)
return false, err
}
}
return true, nil
}
// fontStacker is the PDF font stack implementation.
type fontStacker []*model.PdfFont
// String returns a string describing the current state of the font stack.
func (fontStack *fontStacker) String() string {
parts := []string{"---- font stack"}
for i, font := range *fontStack {
s := "<nil>"
if font != nil {
s = font.String()
}
parts = append(parts, fmt.Sprintf("\t%2d: %s", i, s))
}
return strings.Join(parts, "\n")
}
// push pushes `font` onto the font stack.
func (fontStack *fontStacker) push(font *model.PdfFont) {
*fontStack = append(*fontStack, font)
}
// pop pops and returns the element on the top of the font stack if there is one or nil if there isn't.
func (fontStack *fontStacker) pop() *model.PdfFont {
if fontStack.empty() {
return nil
}
font := (*fontStack)[len(*fontStack)-1]
*fontStack = (*fontStack)[:len(*fontStack)-1]
return font
}
// peek returns the element on the top of the font stack if there is one or nil if there isn't.
func (fontStack *fontStacker) peek() *model.PdfFont {
if fontStack.empty() {
return nil
}
return (*fontStack)[len(*fontStack)-1]
}
// get returns the `idx`'th element of the font stack if there is one or nil if there isn't.
// idx = 0: bottom of font stack
// idx = len(fontstack) - 1: top of font stack
// idx = -n is same as dx = len(fontstack) - n, so fontstack.get(-1) is same as fontstack.peek()
func (fontStack *fontStacker) get(idx int) *model.PdfFont {
if idx < 0 {
idx += fontStack.size()
}
if idx < 0 || idx > fontStack.size()-1 {
return nil
}
return (*fontStack)[idx]
}
// empty returns true if the font stack is empty.
func (fontStack *fontStacker) empty() bool {
return len(*fontStack) == 0
}
// size returns the number of elements in the font stack.
func (fontStack *fontStacker) size() int {
return len(*fontStack)
}
// 9.3 Text State Parameters and Operators (page 243)
// Some of these parameters are expressed in unscaled text space units. This means that they shall
// be specified in a coordinate system that shall be defined by the text matrix, Tm but shall not be
// scaled by the font size parameter, Tfs.
// textState represents the text state.
type textState struct {
tc float64 // Character spacing. Unscaled text space units.
tw float64 // Word spacing. Unscaled text space units.
th float64 // Horizontal scaling.
tl float64 // Leading. Unscaled text space units. Used by TD,T*,'," see Table 108.
tfs float64 // Text font size.
tmode RenderMode // Text rendering mode.
trise float64 // Text rise. Unscaled text space units. Set by Ts.
tfont *model.PdfFont // Text font.
// For debugging
numChars int
numMisses int
}
// 9.4.1 General (page 248)
// A PDF text object consists of operators that may show text strings, move the text position, and
// set text state and certain other parameters. In addition, two parameters may be specified only
// within a text object and shall not persist from one text object to the next:
// • Tm, the text matrix
// • Tlm, the text line matrix
//
// Text space is converted to device space by this transform (page 252)
// Trm is the text rendering matrix
// | Tfs x Th 0 0 |
// Trm = | 0 Tfs 0 | × Tm × CTM
// | 0 Trise 1 |
// This corresponds to the following code in renderText()
// trm := to.gs.CTM.Mult(stateMatrix).Mult(to.tm)
// textObject represents a PDF text object.
type textObject struct {
e *Extractor
resources *model.PdfPageResources
gs contentstream.GraphicsState
fontStack *fontStacker
state *textState
tm transform.Matrix // Text matrix. For the character pointer.
tlm transform.Matrix // Text line matrix. For the start of line pointer.
marks []textMark // Text marks get written here.
}
// newTextState returns a default textState.
func newTextState() textState {
return textState{
th: 100,
tmode: RenderModeFill,
}
}
// newTextObject returns a default textObject.
func newTextObject(e *Extractor, resources *model.PdfPageResources, gs contentstream.GraphicsState,
state *textState, fontStack *fontStacker) *textObject {
return &textObject{
e: e,
resources: resources,
gs: gs,
fontStack: fontStack,
state: state,
tm: transform.IdentityMatrix(),
tlm: transform.IdentityMatrix(),
}
}
// reset sets the text matrix `Tm` and the text line matrix `Tlm` of the text
// object to the identity matrix. In addition, the marks collection is cleared.
func (to *textObject) reset() {
to.tm = transform.IdentityMatrix()
to.tlm = transform.IdentityMatrix()
to.marks = nil
}
// renderText processes and renders byte array `data` for extraction purposes.
func (to *textObject) renderText(data []byte) error {
font := to.getCurrentFont()
charcodes := font.BytesToCharcodes(data)
texts, numChars, numMisses := font.CharcodesToStrings(charcodes)
if numMisses > 0 {
common.Log.Debug("renderText: numChars=%d numMisses=%d", numChars, numMisses)
}
to.state.numChars += numChars
to.state.numMisses += numMisses
state := to.state
tfs := state.tfs
th := state.th / 100.0
spaceMetrics, ok := font.GetRuneMetrics(' ')
if !ok {
spaceMetrics, ok = font.GetCharMetrics(32)
}
if !ok {
spaceMetrics, _ = model.DefaultFont().GetRuneMetrics(' ')
}
spaceWidth := spaceMetrics.Wx * glyphTextRatio
common.Log.Trace("spaceWidth=%.2f text=%q font=%s fontSize=%.2f", spaceWidth, texts, font, tfs)
stateMatrix := transform.NewMatrix(
tfs*th, 0,
0, tfs,
0, state.trise)
common.Log.Trace("renderText: %d codes=%+v runes=%q", len(charcodes), charcodes, len(texts))
for i, text := range texts {
r := []rune(text)
if len(r) == 1 && r[0] == '\x00' {
continue
}
// TODO(gunnsth): Assuming 1:1 charcode[i] <-> rune[i] mapping.
code := charcodes[i]
// The location of the text on the page in device coordinates is given by trm, the text
// rendering matrix.
trm := to.gs.CTM.Mult(to.tm).Mult(stateMatrix)
// calculate the text location displacement due to writing `r`. We will use this to update
// to.tm
// w is the unscaled movement at the end of a word.
w := 0.0
if string(r) == " " {
w = state.tw
}
m, ok := font.GetCharMetrics(code)
if !ok {
common.Log.Debug("ERROR: No metric for code=%d r=0x%04x=%+q %s", code, r, r, font)
return fmt.Errorf("no char metrics: font=%s code=%d", font.String(), code)
}
// c is the character size in unscaled text units.
c := transform.Point{X: m.Wx * glyphTextRatio, Y: m.Wy * glyphTextRatio}
// t0 is the end of this character.
// t is the displacement of the text cursor when the character is rendered.
t0 := transform.Point{X: (c.X*tfs + w) * th}
t := transform.Point{X: (c.X*tfs + state.tc + w) * th}
// td, td0 are t, t0 in matrix form.
// td0 is where this character ends. td is where the next character starts.
td0 := translationMatrix(t0)
td := translationMatrix(t)
common.Log.Trace("\"%c\" stateMatrix=%s CTM=%s Tm=%s", r, stateMatrix, to.gs.CTM, to.tm)
common.Log.Trace("tfs=%.3f th=%.3f Tc=%.3f w=%.3f (Tw=%.3f)", tfs, th, state.tc, w, state.tw)
common.Log.Trace("m=%s c=%+v t0=%+v td0=%s trm0=%s", m, c, t0, td0, td0.Mult(to.tm).Mult(to.gs.CTM))
mark := to.newTextMark(
text,
trm,
translation(to.gs.CTM.Mult(to.tm).Mult(td0)),
math.Abs(spaceWidth*trm.ScalingFactorX()),
font,
to.state.tc)
if font == nil {
common.Log.Debug("ERROR: No font.")
} else if font.Encoder() == nil {
common.Log.Debug("ERROR: No encoding. font=%s", font)
} else {
// TODO: This lookup seems confusing. Went from bytes <-> charcodes already.
// NOTE: This is needed to register runes by the font encoder - for subsetting (optimization).
original, ok := font.Encoder().CharcodeToRune(code)
if ok {
mark.original = string(original)
}
}
common.Log.Trace("i=%d code=%d mark=%s trm=%s", i, code, mark, trm)
to.marks = append(to.marks, mark)
// update the text matrix by the displacement of the text location.
to.tm.Concat(td)
common.Log.Trace("to.tm=%s", to.tm)
}
return nil
}
// glyphTextRatio converts Glyph metrics units to unscaled text space units.
const glyphTextRatio = 1.0 / 1000.0
// translation returns the translation part of `m`.
func translation(m transform.Matrix) transform.Point {
tx, ty := m.Translation()
return transform.Point{X: tx, Y: ty}
}
// translationMatrix returns a matrix that translates by `p`.
func translationMatrix(p transform.Point) transform.Matrix {
return transform.TranslationMatrix(p.X, p.Y)
}
// moveTo moves the start of line pointer by `tx`,`ty` and sets the text pointer to the
// start of line pointer.
// Move to the start of the next line, offset from the start of the current line by (tx, ty).
// `tx` and `ty` are in unscaled text space units.
func (to *textObject) moveTo(tx, ty float64) {
to.tlm.Concat(transform.NewMatrix(1, 0, 0, 1, tx, ty))
to.tm = to.tlm
}
// textMark represents text drawn on a page and its position in device coordinates.
// All dimensions are in device coordinates.
type textMark struct {
text string // The text (decoded via ToUnicode).
original string // Original text (decoded).
bbox model.PdfRectangle // Text bounding box.
orient int // The text orientation in degrees. This is the current TRM rounded to 10°.
orientedStart transform.Point // Left of text in orientation where text is horizontal.
orientedEnd transform.Point // Right of text in orientation where text is horizontal.
height float64 // Text height.
spaceWidth float64 // Best guess at the width of a space in the font the text was rendered with.
font *model.PdfFont // The font the mark was drawn with.
fontsize float64 // The font size the mark was drawn with.
charspacing float64 // TODO (peterwilliams97: Should this be exposed in TextMark?
trm transform.Matrix // The current text rendering matrix (TRM above).
end transform.Point // The end of character device coordinates.
count int64 // To help with reading debug logs.
}
// newTextMark returns a textMark for text `text` rendered with text rendering matrix (TRM) `trm`
// and end of character device coordinates `end`. `spaceWidth` is our best guess at the width of a
// space in the font the text is rendered in device coordinates.
func (to *textObject) newTextMark(text string, trm transform.Matrix, end transform.Point,
spaceWidth float64, font *model.PdfFont, charspacing float64) textMark {
to.e.textCount++
theta := trm.Angle()
orient := nearestMultiple(theta, 10)
var height float64
if orient%180 != 90 {
height = trm.ScalingFactorY()
} else {
height = trm.ScalingFactorX()
}
start := translation(trm)
bbox := model.PdfRectangle{Llx: start.X, Lly: start.Y, Urx: end.X, Ury: end.Y}
switch orient % 360 {
case 90:
bbox.Urx -= height
case 180:
bbox.Ury -= height
case 270:
bbox.Urx += height
default:
bbox.Ury += height
}
tm := textMark{
text: text,
orient: orient,
bbox: bbox,
orientedStart: start.Rotate(theta),
orientedEnd: end.Rotate(theta),
height: math.Abs(height),
spaceWidth: spaceWidth,
font: font,
fontsize: to.state.tfs,
charspacing: charspacing,
trm: trm,
end: end,
count: to.e.textCount,
}
if !isTextSpace(tm.text) && tm.Width() == 0.0 {
common.Log.Debug("ERROR: Zero width text. tm=%s\n\tm=%#v", tm, tm)
}
return tm
}
// isTextSpace returns true if `text` contains nothing but space code points.
func isTextSpace(text string) bool {
for _, r := range text {
if !unicode.IsSpace(r) {
return false
}
}
return true
}
// nearestMultiple return the integer multiple of `m` that is closest to `x`.
func nearestMultiple(x float64, m int) int {
if m == 0 {
m = 1
}
fac := float64(m)
return int(math.Round(x/fac) * fac)
}
// String returns a string describing `tm`.
func (tm textMark) String() string {
return fmt.Sprintf("textMark{@%03d [%.3f,%.3f] w=%.1f %d° %q}",
tm.count, tm.orientedStart.X, tm.orientedStart.Y, tm.Width(), tm.orient,
truncate(tm.text, 100))
}
// Width returns the width of `tm`.text in the text direction.
func (tm textMark) Width() float64 {
return math.Abs(tm.orientedStart.X - tm.orientedEnd.X)
}
// ToTextMark returns the public view of `tm`.
func (tm textMark) ToTextMark() TextMark {
return TextMark{
Text: tm.text,
Original: tm.original,
BBox: tm.bbox,
Font: tm.font,
FontSize: tm.fontsize,
}
}
// PageText represents the layout of text on a device page.
type PageText struct {
marks []textMark // Texts and their positions on a PDF page.
viewText string // Extracted page text.
viewMarks []TextMark // Public view of `marks`.
}
// String returns a string describing `pt`.
func (pt PageText) String() string {
summary := fmt.Sprintf("PageText: %d elements", len(pt.marks))
parts := []string{"-" + summary}
for _, tm := range pt.marks {
parts = append(parts, tm.String())
}
parts = append(parts, "+"+summary)
return strings.Join(parts, "\n")
}
// length returns the number of elements in `pt.marks`.
func (pt PageText) length() int {
return len(pt.marks)
}
// Text returns the extracted page text.
func (pt PageText) Text() string {
return pt.viewText
}
// ToText returns the page text as a single string.
// Deprecated: This function is deprecated and will be removed in a future major version. Please use
// Text() instead.
func (pt PageText) ToText() string {
return pt.Text()
}
// Marks returns the TextMark collection for a page. It represents all the text on the page.
func (pt PageText) Marks() *TextMarkArray {
return &TextMarkArray{marks: pt.viewMarks}
}
// TextMarkArray is a collection of TextMarks.
type TextMarkArray struct {
marks []TextMark
}
// Append appends `mark` to the mark array.
func (ma *TextMarkArray) Append(mark TextMark) {
ma.marks = append(ma.marks, mark)
}
// String returns a string describing `ma`.
func (ma TextMarkArray) String() string {
n := len(ma.marks)
if n == 0 {
return "EMPTY"
}
m0 := ma.marks[0]
m1 := ma.marks[n-1]
return fmt.Sprintf("{TEXTMARKARRAY: %d elements\n\tfirst=%s\n\t last=%s}", n, m0, m1)
}
// Elements returns the TextMarks in `ma`.
func (ma *TextMarkArray) Elements() []TextMark {
return ma.marks
}
// Len returns the number of TextMarks in `ma`.
func (ma *TextMarkArray) Len() int {
if ma == nil {
return 0
}
return len(ma.marks)
}
// RangeOffset returns the TextMarks in `ma` that have `start` <= TextMark.Offset < `end`.
func (ma *TextMarkArray) RangeOffset(start, end int) (*TextMarkArray, error) {
if ma == nil {
return nil, errors.New("ma==nil")
}
if end < start {
return nil, fmt.Errorf("end < start. RangeOffset not defined. start=%d end=%d ", start, end)
}
n := len(ma.marks)
if n == 0 {
return ma, nil
}
if start < ma.marks[0].Offset {
start = ma.marks[0].Offset
}
if end > ma.marks[n-1].Offset+1 {
end = ma.marks[n-1].Offset + 1
}
iStart := sort.Search(n, func(i int) bool { return ma.marks[i].Offset >= start })
if !(0 <= iStart && iStart < n) {
err := fmt.Errorf("Out of range. start=%d iStart=%d len=%d\n\tfirst=%v\n\t last=%v",
start, iStart, n, ma.marks[0], ma.marks[n-1])
return nil, err
}
iEnd := sort.Search(n, func(i int) bool { return ma.marks[i].Offset > end-1 })
if !(0 <= iEnd && iEnd < n) {
err := fmt.Errorf("Out of range. end=%d iEnd=%d len=%d\n\tfirst=%v\n\t last=%v",
end, iEnd, n, ma.marks[0], ma.marks[n-1])
return nil, err
}
if iEnd <= iStart {
// This should never happen.
return nil, fmt.Errorf("start=%d end=%d iStart=%d iEnd=%d", start, end, iStart, iEnd)
}
return &TextMarkArray{marks: ma.marks[iStart:iEnd]}, nil
}
// BBox returns the smallest axis-aligned rectangle that encloses all the TextMarks in `ma`.
func (ma *TextMarkArray) BBox() (model.PdfRectangle, bool) {
if len(ma.marks) == 0 {
return model.PdfRectangle{}, false
}
bbox := ma.marks[0].BBox
for _, tm := range ma.marks[1:] {
if isTextSpace(tm.Text) {
continue
}
bbox = rectUnion(bbox, tm.BBox)
}
return bbox, true
}
// rectUnion returns the smallest axis-aligned rectangle that contains `b1` and `b2`.
func rectUnion(b1, b2 model.PdfRectangle) model.PdfRectangle {
return model.PdfRectangle{
Llx: math.Min(b1.Llx, b2.Llx),
Lly: math.Min(b1.Lly, b2.Lly),
Urx: math.Max(b1.Urx, b2.Urx),
Ury: math.Max(b1.Ury, b2.Ury),
}
}
// TextMark represents extracted text on a page with information regarding both textual content,
// formatting (font and size) and positioning.
// It is the smallest unit of text on a PDF page, typically a single character.
//
// getBBox() in test_text.go shows how to compute bounding boxes of substrings of extracted text.
// The following code extracts the text on PDF page `page` into `text` then finds the bounding box
// `bbox` of substring `term` in `text`.
//
// ex, _ := New(page)
// // handle errors
// pageText, _, _, err := ex.ExtractPageText()
// // handle errors
// text := pageText.Text()
// textMarks := pageText.Marks()
//
// start := strings.Index(text, term)
// end := start + len(term)
// spanMarks, err := textMarks.RangeOffset(start, end)
// // handle errors
// bbox, ok := spanMarks.BBox()
// // handle errors
type TextMark struct {
// Text is the extracted text. It has been decoded to Unicode via ToUnicode().
Text string
// Original is the text in the PDF. It has not been decoded like `Text`.
Original string
// BBox is the bounding box of the text.
BBox model.PdfRectangle
// Font is the font the text was drawn with.
Font *model.PdfFont
// FontSize is the font size the text was drawn with.
FontSize float64
// Offset is the offset of the start of TextMark.Text in the extracted text. If you do this
// text, textMarks := pageText.Text(), pageText.Marks()
// marks := textMarks.Elements()
// then marks[i].Offset is the offset of marks[i].Text in text.
Offset int
// Meta is set true for spaces and line breaks that we insert in the extracted text. We insert
// spaces (line breaks) when we see characters that are over a threshold horizontal (vertical)
// distance apart. See wordJoiner (lineJoiner) in PageText.computeViews().
Meta bool
}
// String returns a string describing `tm`.
func (tm TextMark) String() string {
b := tm.BBox
var font string
if tm.Font != nil {
font = tm.Font.String()
if len(font) > 50 {
font = font[:50] + "..."
}
}
var meta string
if tm.Meta {
meta = " *M*"
}
return fmt.Sprintf("{TextMark: %d %q=%02x (%5.1f, %5.1f) (%5.1f, %5.1f) %s%s}",
tm.Offset, tm.Text, []rune(tm.Text), b.Llx, b.Lly, b.Urx, b.Ury, font, meta)
}
// computeViews processes the page TextMarks sorting by position and populates `pt.viewText` and
// `pt.viewMarks` which represent the text and marks in the order which it is read on the page.
// The comments above the TextMark definition describe how to use the []TextMark to
// maps substrings of the page text to locations on the PDF page.
func (pt *PageText) computeViews() {
fontHeight := pt.height()
// We sort with a y tolerance to allow for subscripts, diacritics etc.
tol := minFloat(fontHeight*0.19, 5.0)
common.Log.Trace("ToTextLocation: %d elements fontHeight=%.1f tol=%.1f", len(pt.marks), fontHeight, tol)
// Uncomment the 2 following Debug statements to see the effects of sorting.
// common.Log.Debug("computeViews: Before sorting %s", pt)
pt.sortPosition(tol)
// common.Log.Debug("computeViews: After sorting %s", pt)
lines := pt.toLines(tol)
texts := make([]string, len(lines))
for i, l := range lines {
texts[i] = strings.Join(l.words(), wordJoiner)
}
text := strings.Join(texts, lineJoiner)
var marks []TextMark
offset := 0
for i, l := range lines {
for j, tm := range l.marks {
tm.Offset = offset
marks = append(marks, tm)
offset += len(tm.Text)
if j == len(l.marks)-1 {
break
}
if wordJoinerLen > 0 {
tm := TextMark{
Offset: offset,
Text: wordJoiner,
Meta: true,
}
marks = append(marks, tm)
offset += wordJoinerLen
}
}
if i == len(lines)-1 {
break
}
if lineJoinerLen > 0 {
tm := TextMark{
Offset: offset,
Text: lineJoiner,
Meta: true,
}
marks = append(marks, tm)
offset += lineJoinerLen
}
}
pt.viewText = text
pt.viewMarks = marks
}
// height returns the max height of the elements in `pt.marks`.
func (pt PageText) height() float64 {
fontHeight := 0.0
for _, tm := range pt.marks {
if tm.height > fontHeight {
fontHeight = tm.height
}
}
return fontHeight
}
const (
// wordJoiner is added between text marks in extracted text.
wordJoiner = ""
// lineJoiner is added between lines in extracted text.
lineJoiner = "\n"
)
var (
wordJoinerLen = len(wordJoiner)
lineJoinerLen = len(lineJoiner)
// spaceMark is a special TextMark used for spaces.
spaceMark = TextMark{
Text: " ",
Original: " ",
Meta: true,
}
)
// sortPosition sorts a text list by its elements' positions on a page.
// Sorting is by orientation then top to bottom, left to right when page is orientated so that text
// is horizontal.
// Text is considered to be on different lines if the lines' orientedStart.Y differs by more than `tol`.
func (pt *PageText) sortPosition(tol float64) {
if len(pt.marks) == 0 {
return
}
// For grouping data vertically into lines, it is necessary to have the data presorted by
// descending y position.
sort.SliceStable(pt.marks, func(i, j int) bool {
ti, tj := pt.marks[i], pt.marks[j]
if ti.orient != tj.orient {
return ti.orient < tj.orient
}
return ti.orientedStart.Y >= tj.orientedStart.Y
})
// Cluster the marks into y-clusters by relative y proximity. Each cluster is our guess of what
// makes up a line of text.
clusters := make([]int, len(pt.marks))
cluster := 0
clusters[0] = cluster
for i := 1; i < len(pt.marks); i++ {
if pt.marks[i-1].orient != pt.marks[i].orient {
cluster++
} else {
if pt.marks[i-1].orientedStart.Y-pt.marks[i].orientedStart.Y > tol {
cluster++
}
}
clusters[i] = cluster
}
// Sort by y-cluster and x.
sort.SliceStable(pt.marks, func(i, j int) bool {
ti, tj := pt.marks[i], pt.marks[j]
if ti.orient != tj.orient {
return ti.orient < tj.orient
}
if clusters[i] != clusters[j] {
return clusters[i] < clusters[j]
}
return ti.orientedStart.X < tj.orientedStart.X
})
}
// textLine represents a line of text on a page.
type textLine struct {
x float64 // x position of line.
y float64 // y position of line.
h float64 // height of line text.
dxList []float64 // x distance between successive words in line.
marks []TextMark // TextMarks in the line.
}
// words returns the texts in `tl`.
func (tl textLine) words() []string {
var texts []string
for _, tm := range tl.marks {
texts = append(texts, tm.Text)
}
return texts
}
// toLines returns the text and positions in `pt.marks` as a slice of textLine.
// NOTE: Caller must sort the text list top-to-bottom, left-to-right (for orientation adjusted so
// that text is horizontal) before calling this function.
func (pt PageText) toLines(tol float64) []textLine {
// We divide `pt.marks` into slices which contain texts with the same orientation, extract the
// lines for each orientation then return the concatenation of these lines sorted by orientation.
tlOrient := make(map[int][]textMark, len(pt.marks))
for _, tm := range pt.marks {
tlOrient[tm.orient] = append(tlOrient[tm.orient], tm)
}
var lines []textLine
for _, o := range orientKeys(tlOrient) {
lns := PageText{marks: tlOrient[o]}.toLinesOrient(tol)
lines = append(lines, lns...)
}
return lines
}
// toLinesOrient returns the text and positions in `pt.marks` as a slice of textLine.
// NOTE: This function only works on text lists where all text is the same orientation so it should
// only be called from toLines.
// Caller must sort the text list top-to-bottom, left-to-right (for orientation adjusted so
// that text is horizontal) before calling this function.
func (pt PageText) toLinesOrient(tol float64) []textLine {
if len(pt.marks) == 0 {
return []textLine{}
}
var marks []TextMark
var lines []textLine
var xx []float64
y := pt.marks[0].orientedStart.Y
scanning := false
averageCharWidth := exponAve{}
wordSpacing := exponAve{}
lastEndX := 0.0 // lastEndX is pt.marks[i-1].orientedEnd.X
for _, tm := range pt.marks {
if tm.orientedStart.Y+tol < y {
if len(marks) > 0 {
tl := newLine(y, xx, marks)
if averageCharWidth.running {
// FIXME(peterwilliams97): Fix and reinstate combineDiacritics.
// tl = combineDiacritics(tl, averageCharWidth.ave)
tl = removeDuplicates(tl, averageCharWidth.ave)
}
lines = append(lines, tl)
}
marks = []TextMark{}
xx = []float64{}
y = tm.orientedStart.Y
scanning = false
}
// Detect text movements that represent spaces on the printed page.
// We use a heuristic from PdfBox: If the next character starts to the right of where a
// character after a space at "normal spacing" would start, then there is a space before it.
// The tricky thing to guess here is the width of a space at normal spacing.
// We follow PdfBox and use min(deltaSpace, deltaCharWidth).
deltaSpace := 0.0
if tm.spaceWidth == 0 {
deltaSpace = math.MaxFloat64
} else {
wordSpacing.update(tm.spaceWidth)
deltaSpace = wordSpacing.ave * 0.5
}
averageCharWidth.update(tm.Width())
deltaCharWidth := averageCharWidth.ave * 0.3
isSpace := false
nextWordX := lastEndX + minFloat(deltaSpace, deltaCharWidth)
if scanning && !isTextSpace(tm.text) {
isSpace = nextWordX < tm.orientedStart.X
}
common.Log.Trace("tm=%s", tm)
common.Log.Trace("width=%.2f delta=%.2f deltaSpace=%.2g deltaCharWidth=%.2g",
tm.Width(), minFloat(deltaSpace, deltaCharWidth), deltaSpace, deltaCharWidth)
common.Log.Trace("%+q [%.1f, %.1f] lastEndX=%.2f nextWordX=%.2f (%.2f) isSpace=%t",
tm.text, tm.orientedStart.X, tm.orientedStart.Y, lastEndX, nextWordX,
nextWordX-tm.orientedStart.X, isSpace)
if isSpace {
marks = append(marks, spaceMark)
xx = append(xx, (lastEndX+tm.orientedStart.X)*0.5)
}
// Add the text to the line.
lastEndX = tm.orientedEnd.X
marks = append(marks, tm.ToTextMark())
xx = append(xx, tm.orientedStart.X)
scanning = true
common.Log.Trace("lastEndX=%.2f", lastEndX)
}
if len(marks) > 0 {
tl := newLine(y, xx, marks)
if averageCharWidth.running {
tl = removeDuplicates(tl, averageCharWidth.ave)
}
lines = append(lines, tl)
}
return lines
}
// orientKeys returns the keys of `tlOrient` as a sorted slice.
func orientKeys(tlOrient map[int][]textMark) []int {
keys := []int{}
for k := range tlOrient {
keys = append(keys, k)
}
sort.Ints(keys)
return keys
}
// exponAve implements an exponential average.
type exponAve struct {
ave float64 // Current average value.
running bool // Has `ave` been set?
}
// update updates the exponential average `exp`.ave with latest value `x` and returns `exp`.ave.
func (exp *exponAve) update(x float64) float64 {
if !exp.running {
exp.ave = x
exp.running = true
} else {
// NOTE(peterwilliams97): 0.5 is a guess. It may be possible to improve average character
// and space width estimation by tuning this value. It may be that different exponents
// would work better for character and space estimation.
exp.ave = (exp.ave + x) * 0.5
}
return exp.ave
}
// newLine returns the textLine representation of strings `words` with y coordinate `y` and x
// coordinates `xx` and height `h`.
func newLine(y float64, xx []float64, marks []TextMark) textLine {
dxList := make([]float64, len(xx)-1)
for i := 1; i < len(xx); i++ {
dxList[i-1] = xx[i] - xx[i-1]
}
return textLine{
x: xx[0],
y: y,
dxList: dxList,
marks: marks,
}
}
// removeDuplicates returns `tl` with duplicate characters removed. `charWidth` is the average
// character width for the line.
func removeDuplicates(tl textLine, charWidth float64) textLine {
if len(tl.dxList) == 0 || len(tl.marks) == 0 {
return tl
}
// NOTE(peterwilliams97) 0.3 is a guess. It may be possible to tune this to a better value.
tol := charWidth * 0.3
marks := []TextMark{tl.marks[0]}
var dxList []float64
tm0 := tl.marks[0]
for i, dx := range tl.dxList {
tm := tl.marks[i+1]
if tm.Text != tm0.Text || dx > tol {
marks = append(marks, tm)
dxList = append(dxList, dx)
}
tm0 = tm
}
return textLine{
x: tl.x,
y: tl.y,
dxList: dxList,
marks: marks,
}
}
// combineDiacritics returns `line` with diacritics close to characters combined with the characters.
// `charWidth` is the average character width for the line.
// We have to do this because PDF can render diacritics separately to the characters they attach to
// in extracted text.
func combineDiacritics(tl textLine, charWidth float64) textLine {
if len(tl.dxList) == 0 || len(tl.marks) == 0 {
return tl
}
// NOTE(peterwilliams97) 0.2 is a guess. It may be possible to tune this to a better value.
tol := charWidth * 0.2
common.Log.Trace("combineDiacritics: charWidth=%.2f tol=%.2f", charWidth, tol)
var marks []TextMark
var dxList []float64
tm := marks[0]
w, c := countDiacritic(tm.Text)
delta := 0.0
dx0 := 0.0
parts := []string{w}
numChars := c
for i, dx := range tl.dxList {
tm = marks[i+1]
w, c := countDiacritic(tm.Text)
if numChars+c <= 1 && delta+dx <= tol {
if len(parts) == 0 {
dx0 = dx
} else {
delta += dx
}
parts = append(parts, w)
numChars += c
} else {
if len(parts) > 0 {
if len(marks) > 0 {
dxList = append(dxList, dx0)
}
tm.Text = combine(parts)
marks = append(marks, tm)
}
parts = []string{w}
numChars = c
dx0 = dx
delta = 0.0
}
}
if len(parts) > 0 {
if len(marks) > 0 {
dxList = append(dxList, dx0)
}
tm.Text = combine(parts)
marks = append(marks, tm)
}
if len(marks) != len(dxList)+1 {
common.Log.Error("Inconsistent: \nwords=%d \ndxList=%d %.2f",
len(marks), len(dxList), dxList)
return tl
}
return textLine{
x: tl.x,
y: tl.y,
dxList: dxList,
marks: marks,
}
}
// combine combines any diacritics in `parts` with the single non-diacritic character in `parts`.
func combine(parts []string) string {
if len(parts) == 1 {
// Must be a non-diacritic.
return parts[0]
}
// We need to put the diacritics before the non-diacritic for NFKC normalization to work.
diacritic := map[string]bool{}
for _, w := range parts {
r := []rune(w)[0]
diacritic[w] = unicode.Is(unicode.Mn, r) || unicode.Is(unicode.Sk, r)
}
sort.SliceStable(parts, func(i, j int) bool { return !diacritic[parts[i]] && diacritic[parts[j]] })
// Construct the NFKC-normalized concatenation of the diacritics and the non-diacritic.
for i, w := range parts {
parts[i] = strings.TrimSpace(norm.NFKC.String(w))
}
return strings.Join(parts, "")
}
// countDiacritic returns the combining diacritic version of `w` (usually itself) and the number of
// non-diacritics in `w` (0 or 1).
func countDiacritic(w string) (string, int) {
runes := []rune(w)
if len(runes) != 1 {
return w, 1
}
r := runes[0]
c := 1
if (unicode.Is(unicode.Mn, r) || unicode.Is(unicode.Sk, r)) &&
r != '\'' && r != '"' && r != '`' {
c = 0
}
if w2, ok := diacritics[r]; ok {
c = 0
w = w2
}
return w, c
}
// diacritics is a map of diacritic characters that are not classified as unicode.Mn or unicode.Sk
// and the corresponding unicode.Mn or unicode.Sk characters. This map was copied from PdfBox.
// (https://svn.apache.org/repos/asf/pdfbox/trunk/pdfbox/src/main/java/org/apache/pdfbox/text/TextPosition.java)
var diacritics = map[rune]string{
0x0060: "\u0300",
0x02CB: "\u0300",
0x0027: "\u0301",
0x02B9: "\u0301",
0x02CA: "\u0301",
0x005e: "\u0302",
0x02C6: "\u0302",
0x007E: "\u0303",
0x02C9: "\u0304",
0x00B0: "\u030A",
0x02BA: "\u030B",
0x02C7: "\u030C",
0x02C8: "\u030D",
0x0022: "\u030E",
0x02BB: "\u0312",
0x02BC: "\u0313",
0x0486: "\u0313",
0x055A: "\u0313",
0x02BD: "\u0314",
0x0485: "\u0314",
0x0559: "\u0314",
0x02D4: "\u031D",
0x02D5: "\u031E",
0x02D6: "\u031F",
0x02D7: "\u0320",
0x02B2: "\u0321",
0x02CC: "\u0329",
0x02B7: "\u032B",
0x02CD: "\u0331",
0x005F: "\u0332",
0x204E: "\u0359",
}
// getCurrentFont returns the font on top of the font stack, or DefaultFont if the font stack is
// empty.
func (to *textObject) getCurrentFont() *model.PdfFont {
if to.fontStack.empty() {
common.Log.Debug("ERROR: No font defined. Using default.")
return model.DefaultFont()
}
return to.fontStack.peek()
}
// getFont returns the font named `name` if it exists in the page's resources or an error if it
// doesn't. It caches the returned fonts.
func (to *textObject) getFont(name string) (*model.PdfFont, error) {
if to.e.fontCache != nil {
to.e.accessCount++
entry, ok := to.e.fontCache[name]
if ok {
entry.access = to.e.accessCount
return entry.font, nil
}
}
// Font not in cache. Load it.
font, err := to.getFontDirect(name)
if err != nil {
return nil, err
}
if to.e.fontCache != nil {
entry := fontEntry{font, to.e.accessCount}
// Eject a victim if the cache is full.
if len(to.e.fontCache) >= maxFontCache {
var names []string
for name := range to.e.fontCache {
names = append(names, name)
}
sort.Slice(names, func(i, j int) bool {
return to.e.fontCache[names[i]].access < to.e.fontCache[names[j]].access
})
delete(to.e.fontCache, names[0])
}
to.e.fontCache[name] = entry
}
return font, nil
}
// fontEntry is a entry in the font cache.
type fontEntry struct {
font *model.PdfFont // The font being cached.
access int64 // Last access. Used to determine LRU cache victims.
}
// maxFontCache is the maximum number of PdfFont's in fontCache.
const maxFontCache = 10
// getFontDirect returns the font named `name` if it exists in the page's resources or an error if
// it doesn't. Accesses page resources directly (not cached).
func (to *textObject) getFontDirect(name string) (*model.PdfFont, error) {
fontObj, err := to.getFontDict(name)
if err != nil {
return nil, err
}
font, err := model.NewPdfFontFromPdfObject(fontObj)
if err != nil {
common.Log.Debug("getFontDirect: NewPdfFontFromPdfObject failed. name=%#q err=%v", name, err)
}
return font, err
}
// getFontDict returns the font dict with key `name` if it exists in the page's or form's Font
// resources or an error if it doesn't.
func (to *textObject) getFontDict(name string) (fontObj core.PdfObject, err error) {
resources := to.resources
if resources == nil {
common.Log.Debug("getFontDict. No resources. name=%#q", name)
return nil, nil
}
fontObj, found := resources.GetFontByName(core.PdfObjectName(name))
if !found {
common.Log.Debug("ERROR: getFontDict: Font not found: name=%#q", name)
return nil, errors.New("font not in resources")
}
return fontObj, nil
}