unipdf/extractor/text.go

1212 lines
37 KiB
Go
Raw Normal View History

/*
* 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 (
"bytes"
"errors"
"fmt"
"math"
"sort"
"strings"
2018-11-28 18:06:03 +11:00
"unicode"
"github.com/unidoc/unipdf/v3/common"
"github.com/unidoc/unipdf/v3/contentstream"
"github.com/unidoc/unipdf/v3/core"
2020-05-27 18:15:18 +10:00
"github.com/unidoc/unipdf/v3/internal/textencoding"
"github.com/unidoc/unipdf/v3/internal/transform"
"github.com/unidoc/unipdf/v3/model"
)
2020-05-27 18:15:18 +10:00
const verbose = false
// maxFormStack is the maximum form stack recursion depth. It has to be low enough to avoid a stack
// overflow and high enough to accomodate customers' PDFs
const maxFormStack 10
// 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) {
2018-11-28 23:25:17 +00:00
text, _, _, err := e.ExtractTextWithStats()
return text, err
}
2019-01-04 16:07:03 +11:00
// 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) {
2019-01-04 16:07:03 +11:00
pageText, numChars, numMisses, err := e.ExtractPageText()
if err != nil {
return "", numChars, numMisses, err
}
return pageText.Text(), numChars, numMisses, nil
}
2019-01-04 16:07:03 +11:00
// 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, 0)
if err != nil {
return nil, numChars, numMisses, err
}
pt.computeViews()
// procBuf(pt)
return pt, numChars, numMisses, err
}
2019-01-04 16:07:03 +11:00
// extractPageText returns the text contents of content stream `e` and resouces `resources` as a
// PageText.
2018-12-27 21:33:31 +11:00
// This can be called on a page or a form XObject.
func (e *Extractor) extractPageText(contents string, resources *model.PdfPageResources, level int) (
*PageText, int, int, error) {
2019-01-04 16:07:03 +11:00
common.Log.Trace("extractPageText: level=%d", level)
pageText := &PageText{pageSize: e.mediaBox}
state := newTextState(e.mediaBox)
2020-05-26 13:26:09 +10:00
var savedStates stateStack
to := newTextObject(e, resources, contentstream.GraphicsState{}, &state, &savedStates)
var inTextObj bool
if level > maxFormStack {
err := errors.New("form stack overflow")
2020-05-27 18:15:18 +10:00
common.Log.Debug("ERROR: extractPageText. recursion level=%d err=%w", level, err)
return pageText, state.numChars, state.numMisses, err
}
// Uncomment the following 3 statements to log the content stream.
// common.Log.Info("contents* %d -----------------------------", len(contents))
// fmt.Println(contents)
// common.Log.Info("contents+ -----------------------------")
cstreamParser := contentstream.NewContentStreamParser(contents)
operations, err := cstreamParser.Parse()
if err != nil {
2020-05-27 18:15:18 +10:00
common.Log.Debug("ERROR: extractPageText parse failed. err=%w", err)
2019-01-04 16:07:03 +11:00
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
2020-05-27 18:15:18 +10:00
if verbose {
common.Log.Info("&&& op=%s", op)
}
2020-05-26 13:26:09 +10:00
switch operand {
case "q":
2020-05-26 13:26:09 +10:00
savedStates.push(&state)
// common.Log.Info("Save state: stack=%d\n %s", len(savedStates), state.String())
case "Q":
2020-05-27 18:15:18 +10:00
if verbose {
common.Log.Info("Restore state: %s", savedStates.String())
}
2020-05-26 13:26:09 +10:00
if !savedStates.empty() {
// oldState := state
state = *savedStates.top()
// common.Log.Info("Restore state: stack=%d\n %s\n→%s",
// len(savedStates), oldState.String(), state.String())
if len(savedStates) >= 2 {
savedStates.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
2020-05-26 13:26:09 +10:00
to = newTextObject(e, resources, gs, &state, &savedStates)
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
2019-01-05 14:10:54 +11:00
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)
2018-12-27 21:33:31 +11:00
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)
2018-12-27 21:33:31 +11:00
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
}
2018-07-25 13:19:09 +10:00
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)
2018-12-27 21:33:31 +11:00
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)
2018-12-27 21:33:31 +11:00
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)
2018-12-27 21:33:31 +11:00
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)
2018-12-27 21:33:31 +11:00
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)
2018-12-27 21:33:31 +11:00
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)
to.invalidFont = unsupportedFontErr(err)
2020-05-27 18:15:18 +10:00
if err != nil && !to.invalidFont {
return err
}
2018-12-27 21:33:31 +11:00
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)
2018-12-27 21:33:31 +11:00
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)
2018-12-27 21:33:31 +11:00
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)
2018-12-27 21:33:31 +11:00
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)
2018-12-27 21:33:31 +11:00
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":
2018-12-27 21:33:31 +11:00
// 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 core.ErrRangeError
}
// 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 core.ErrTypeError
}
_, xtype := resources.GetXObjectByName(*name)
if xtype != model.XObjectTypeForm {
break
}
2018-12-27 20:53:37 +11:00
// 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
}
2019-01-04 16:07:03 +11:00
tList, numChars, numMisses, err := e.extractPageText(string(formContent),
formResources, level+1)
if err != nil {
common.Log.Debug("ERROR: %v", err)
return err
}
formResult = textResult{*tList, numChars, numMisses}
e.formResults[name.String()] = formResult
}
2019-01-05 14:10:54 +11:00
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)
}
2019-01-04 16:07:03 +11:00
return pageText, state.numChars, state.numMisses, err
}
// unsupportedFontErr returns true if `err` indicated that the selected font or encoding is not supported.
func unsupportedFontErr(err error) bool {
if err == model.ErrFontNotSupported ||
err == model.ErrType1CFontNotSupported ||
err == model.ErrType3FontNotSupported ||
err == model.ErrTTCmapNotSupported {
return true
}
if err == nil {
return false
}
errStr := err.Error()
return strings.Contains(errStr, "unsupported font encoding:") ||
strings.Contains(errStr, "unexpected subtable format:") ||
strings.Contains(errStr, "fonts based on PostScript outlines are not supported")
}
// textResult is used for holding results of PDF form processig
type textResult struct {
2019-01-04 16:07:03 +11:00
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) {
2019-01-02 10:39:30 +11:00
to.state.tl = -ty
to.moveTo(tx, ty)
}
2019-01-03 15:41:36 +11:00
// 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() {
2019-01-02 10:39:30 +11:00
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
2018-11-27 13:37:12 +11:00
// in `f` (page 250).
func (to *textObject) setTextMatrix(f []float64) {
2018-11-28 23:25:17 +00:00
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]
2019-01-02 10:39:30 +11:00
to.tm = transform.NewMatrix(a, b, c, d, tx, ty)
to.tlm = to.tm
to.logCursor()
}
2018-11-27 13:37:12 +11:00
// showText "Tj". Show a text string.
func (to *textObject) showText(charcodes []byte) error {
2018-07-02 16:46:43 +10:00
return to.renderText(charcodes)
}
2018-11-27 13:37:12 +11:00
// showTextAdjusted "TJ". Show text with adjustable spacing.
2018-07-25 13:19:09 +10:00
func (to *textObject) showTextAdjusted(args *core.PdfObjectArray) error {
vertical := false
2018-07-25 13:19:09 +10:00
for _, o := range args.Elements() {
switch o.(type) {
case *core.PdfObjectFloat, *core.PdfObjectInteger:
x, err := core.GetNumberAsFloat(o)
if err != nil {
2018-11-27 13:37:12 +11:00
common.Log.Debug("ERROR: showTextAdjusted. Bad numerical arg. o=%s args=%+v", o, args)
return err
}
2019-01-02 10:39:30 +11:00
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)
to.logCursor()
case *core.PdfObjectString:
charcodes, ok := core.GetStringBytes(o)
if !ok {
2018-11-27 13:37:12 +11:00
common.Log.Trace("showTextAdjusted: Bad string arg. o=%s args=%+v", o, args)
return core.ErrTypeError
}
to.renderText(charcodes)
default:
2018-11-27 13:37:12 +11:00
common.Log.Debug("ERROR: showTextAdjusted. Unexpected type (%T) args=%+v", o, args)
return core.ErrTypeError
}
}
return nil
}
2018-11-27 13:37:12 +11:00
// setTextLeading "TL". Set text leading.
func (to *textObject) setTextLeading(y float64) {
2019-01-02 10:39:30 +11:00
if to == nil || to.state == nil {
return
}
2019-01-02 10:39:30 +11:00
to.state.tl = y
}
2018-11-27 13:37:12 +11:00
// setCharSpacing "Tc". Set character spacing.
func (to *textObject) setCharSpacing(x float64) {
if to == nil {
return
}
2019-01-02 10:39:30 +11:00
to.state.tc = x
2020-05-27 18:15:18 +10:00
if verbose {
common.Log.Info("setCharSpacing: %.2f state=%s", to.state.String())
}
}
2018-11-27 13:37:12 +11:00
// setFont "Tf". Set font.
func (to *textObject) setFont(name string, size float64) error {
if to == nil {
return nil
}
2020-05-26 13:26:09 +10:00
to.state.tfs = size
font, err := to.getFont(name)
2020-05-26 13:26:09 +10:00
if err != nil {
if err == model.ErrFontNotSupported {
// TODO(peterwilliams97): Do we need to handle this case in a special way?
return err
}
return err
2020-05-26 13:26:09 +10:00
}
to.state.tfont = font
if to.savedStates.empty() {
to.savedStates.push(to.state)
2018-07-03 14:26:42 +10:00
} else {
2020-05-26 13:26:09 +10:00
to.savedStates.top().tfont = to.state.tfont
}
2020-05-26 13:26:09 +10:00
return nil
}
2018-11-27 13:37:12 +11:00
// setTextRenderMode "Tr". Set text rendering mode.
func (to *textObject) setTextRenderMode(mode int) {
if to == nil {
return
}
2019-01-02 10:39:30 +11:00
to.state.tmode = RenderMode(mode)
}
2018-11-27 13:37:12 +11:00
// setTextRise "Ts". Set text rise.
func (to *textObject) setTextRise(y float64) {
if to == nil {
return
}
2019-01-02 10:39:30 +11:00
to.state.trise = y
}
2018-11-27 13:37:12 +11:00
// setWordSpacing "Tw". Set word spacing.
func (to *textObject) setWordSpacing(y float64) {
if to == nil {
return
}
2019-01-02 10:39:30 +11:00
to.state.tw = y
}
2018-11-27 13:37:12 +11:00
// setHorizScaling "Tz". Set horizontal scaling.
func (to *textObject) setHorizScaling(y float64) {
if to == nil {
return
}
2019-01-02 10:39:30 +11:00
to.state.th = y
}
2018-11-28 23:25:17 +00:00
// 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 {
2018-11-28 23:25:17 +00:00
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
}
2020-05-26 13:26:09 +10:00
// stateStack is the PDF textState stack implementation.
type stateStack []*textState
2020-05-26 13:26:09 +10:00
// String returns a string describing the current state of the textState stack.
func (savedStates *stateStack) String() string {
parts := []string{fmt.Sprintf("---- font stack: %d", len(*savedStates))}
for i, state := range *savedStates {
s := "<nil>"
2020-05-26 13:26:09 +10:00
if state != nil {
s = state.String()
}
parts = append(parts, fmt.Sprintf("\t%2d: %s", i, s))
}
return strings.Join(parts, "\n")
}
2020-05-26 13:26:09 +10:00
// push pushes a copy of `state` onto the textState stack.
func (savedStates *stateStack) push(state *textState) {
s := *state
*savedStates = append(*savedStates, &s)
}
2020-05-26 13:26:09 +10:00
// pop pops and returns a copy of the last state on the textState stack there is one or nil if
// there isn't.
func (savedStates *stateStack) pop() *textState {
if savedStates.empty() {
return nil
}
2020-05-26 13:26:09 +10:00
state := *(*savedStates)[len(*savedStates)-1]
*savedStates = (*savedStates)[:len(*savedStates)-1]
return &state
}
2020-05-26 13:26:09 +10:00
// top returns the last saved state if there is one or nil if there isn't.
// NOTE: The return is a pointer. Modifying it will modify the stack.
func (savedStates *stateStack) top() *textState {
if savedStates.empty() {
return nil
}
2020-05-26 13:26:09 +10:00
return (*savedStates)[savedStates.size()-1]
}
2020-05-26 13:26:09 +10:00
// empty returns true if the textState stack is empty.
func (savedStates *stateStack) empty() bool {
return len(*savedStates) == 0
}
2020-05-26 13:26:09 +10:00
// size returns the number of elements in the textState stack.
func (savedStates *stateStack) size() int {
return len(*savedStates)
}
// 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.
mediaBox model.PdfRectangle
// For debugging
numChars int
numMisses int
}
2020-05-26 13:26:09 +10:00
// String returns a description of `state`.
func (state *textState) String() string {
fontName := "[NOT SET]"
if state.tfont != nil {
fontName = state.tfont.BaseFont()
}
return fmt.Sprintf("tc=%.2f tw=%.2f tfs=%.2f font=%q",
state.tc, state.tw, state.tfs, fontName)
}
// 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)
2018-09-20 11:49:44 +10:00
// Trm is the text rendering matrix
// | Tfs x Th 0 0 |
// Trm = | 0 Tfs 0 | × Tm × CTM
// | 0 Trise 1 |
2018-09-20 11:49:44 +10:00
// 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 {
2020-05-26 13:26:09 +10:00
e *Extractor
resources *model.PdfPageResources
gs contentstream.GraphicsState
state *textState
savedStates *stateStack
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.
2020-05-27 18:15:18 +10:00
invalidFont bool // Flag that gets set true when we can't handle the current font.
}
// newTextState returns a default textState.
func newTextState(mediaBox model.PdfRectangle) textState {
return textState{
th: 100,
tmode: RenderModeFill,
mediaBox: mediaBox,
}
}
// newTextObject returns a default textObject.
func newTextObject(e *Extractor, resources *model.PdfPageResources, gs contentstream.GraphicsState,
2020-05-26 13:26:09 +10:00
state *textState, savedStates *stateStack) *textObject {
return &textObject{
2020-05-26 13:26:09 +10:00
e: e,
resources: resources,
gs: gs,
savedStates: savedStates,
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
to.logCursor()
}
// logCursor is for debugging only. Remove !@#$
func (to *textObject) logCursor() {
return
state := to.state
tfs := state.tfs
th := state.th / 100.0
stateMatrix := transform.NewMatrix(
tfs*th, 0,
0, tfs,
0, state.trise)
trm := to.gs.CTM.Mult(to.tm).Mult(stateMatrix)
cur := translation(trm)
common.Log.Info("showTrm: %s cur=%.2f tm=%.2f CTM=%.2f",
fileLine(1, false), cur, to.tm, to.gs.CTM)
}
2018-11-28 23:25:17 +00:00
// renderText processes and renders byte array `data` for extraction purposes.
// It extracts textMarks based the charcodes in `data` and the currect text and graphics states
// are tracked in `to`.
func (to *textObject) renderText(data []byte) error {
2020-05-27 18:15:18 +10:00
if to.invalidFont {
common.Log.Debug("renderText: Invalid font. Not processing.")
return nil
}
2018-09-20 11:49:44 +10:00
font := to.getCurrentFont()
2018-10-30 21:55:30 +11:00
charcodes := font.BytesToCharcodes(data)
runeSlices, numChars, numMisses := font.CharcodesToRuneSlices(charcodes)
2018-11-28 18:06:03 +11:00
if numMisses > 0 {
common.Log.Debug("renderText: numChars=%d numMisses=%d", numChars, numMisses)
}
2018-10-30 21:55:30 +11:00
2019-01-02 10:39:30 +11:00
to.state.numChars += numChars
to.state.numMisses += numMisses
2019-01-02 10:39:30 +11:00
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=%.1f", spaceWidth, runeSlices, font, tfs)
2018-09-20 11:49:44 +10:00
stateMatrix := transform.NewMatrix(
2018-09-20 11:49:44 +10:00
tfs*th, 0,
0, tfs,
2019-01-02 10:39:30 +11:00
0, state.trise)
2020-05-27 18:15:18 +10:00
if verbose {
common.Log.Info("renderText: %d codes=%+v runes=%q", len(charcodes), charcodes, runeSlices)
}
for i, r := range runeSlices {
if len(r) == 1 && r[0] == '\x00' {
2018-11-28 18:06:03 +11:00
continue
}
2018-10-30 21:55:30 +11:00
code := charcodes[i]
2018-09-20 11:49:44 +10:00
// 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)
2018-09-20 11:49:44 +10:00
// calculate the text location displacement due to writing `r`. We will use this to update
2019-01-02 10:39:30 +11:00
// to.tm
2018-09-20 11:49:44 +10:00
// w is the unscaled movement at the end of a word.
w := 0.0
if len(r) == 1 && r[0] == 32 {
2019-01-02 10:39:30 +11:00
w = state.tw
2018-09-20 11:49:44 +10:00
}
2018-10-30 21:55:30 +11:00
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)
2018-09-20 11:49:44 +10:00
}
2018-10-30 21:55:30 +11:00
// c is the character size in unscaled text units.
c := transform.Point{X: m.Wx * glyphTextRatio, Y: m.Wy * glyphTextRatio}
2018-10-30 21:55:30 +11:00
2018-11-27 13:37:12 +11:00
// 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}
2019-01-02 10:39:30 +11:00
t := transform.Point{X: (c.X*tfs + state.tc + w) * th}
2020-05-27 18:15:18 +10:00
if verbose {
common.Log.Info("tfs=%.2f tc=%.2f tw=%.2f th=%.2f", tfs, state.tc, state.tw, th)
common.Log.Info("dx,dy=%.3f t0=%.2f t=%.2f", c, t0, t)
}
2018-09-20 11:49:44 +10:00
// td, td0 are t, t0 in matrix form.
2018-11-27 13:37:12 +11:00
// td0 is where this character ends. td is where the next character starts.
td0 := translationMatrix(t0)
td := translationMatrix(t)
end := to.gs.CTM.Mult(to.tm).Mult(td0)
2020-05-27 18:15:18 +10:00
if verbose {
common.Log.Info("end:\n\tCTM=%s\n\t tm=%s\n"+
"\t td=%s xlat=%s\n"+
"\ttd0=%s\n\t → %s xlat=%s",
to.gs.CTM, to.tm,
td, translation(to.gs.CTM.Mult(to.tm).Mult(td)),
td0, end, translation(end))
}
mark, onPage := to.newTextMark(
2020-05-27 18:15:18 +10:00
textencoding.ExpandLigatures(r),
trm,
translation(end),
math.Abs(spaceWidth*trm.ScalingFactorX()),
font,
to.state.tc)
if !onPage {
common.Log.Debug("Text mark outside page. Skipping")
continue
}
if font == nil {
common.Log.Debug("ERROR: No font.")
} else if font.Encoder() == nil {
common.Log.Debug("ERROR: No encoding. font=%s", font)
} else {
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)
if i != len(runeSlices)-1 {
to.logCursor()
}
2018-09-20 11:49:44 +10:00
}
return nil
}
2018-09-20 11:49:44 +10:00
// 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 {
2018-09-20 11:49:44 +10:00
tx, ty := m.Translation()
return transform.Point{X: tx, Y: ty}
2018-09-20 11:49:44 +10:00
}
// 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))
2019-01-02 10:39:30 +11:00
to.tm = to.tlm
}
// 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
}
2019-01-04 16:07:03 +11:00
// 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`.
pageSize model.PdfRectangle
}
2019-01-04 16:07:03 +11:00
// 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")
}
// 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}
}
// 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() {
common.Log.Trace("ToTextLocation: %d elements", len(pt.marks))
paras := makeTextPage(pt.marks, pt.pageSize, 0)
b := new(bytes.Buffer)
paras.writeText(b)
pt.viewText = b.String()
2020-05-27 18:15:18 +10:00
pt.viewMarks = paras.toTextMarks()
}
// 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)
}
2020-05-27 18:15:18 +10:00
// RangeOffset returns the TextMarks in `ma` that overlap text[start:end] in the extracted text.
// These are tm: `start` <= tm.Offset + len(tm.Text) && tm.Offset < `end` where
// `start` and `end` are offsets in the extracted text.
// NOTE: TextMarks can contain multiple characters. e.g. "ffi" for the ffi ligature so the first and
// last elements of the returned TextMarkArray may only partially overlap text[start: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
}
2020-05-27 18:15:18 +10:00
iStart := sort.Search(n, func(i int) bool { return ma.marks[i].Offset+len(ma.marks[i].Text)-1 >= 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.
2020-05-27 18:15:18 +10:00
return nil, fmt.Errorf("iEnd <= iStart: 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) {
var bbox model.PdfRectangle
found := false
for _, tm := range ma.marks {
if tm.Meta || isTextSpace(tm.Text) {
continue
2018-11-28 18:06:03 +11:00
}
if found {
bbox = rectUnion(bbox, tm.BBox)
} else {
bbox = tm.BBox
found = true
}
}
return bbox, found
}
// 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 {
count int64
// 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*"
}
2020-05-27 18:15:18 +10:00
return fmt.Sprintf("{@%04d TextMark: %d %q=%02x (%6.2f, %6.2f) (%6.2f, %6.2f) %s%s}",
tm.count, tm.Offset, tm.Text, []rune(tm.Text), b.Llx, b.Lly, b.Urx, b.Ury, font, meta)
2018-11-28 18:06:03 +11:00
}
// spaceMark is a special TextMark used for spaces.
var spaceMark = TextMark{
Text: "[X]",
Original: " ",
Meta: true,
2018-11-28 18:06:03 +11:00
}
2018-09-20 11:49:44 +10:00
// getCurrentFont returns the font on top of the font stack, or DefaultFont if the font stack is
// empty.
func (to *textObject) getCurrentFont() *model.PdfFont {
var font *model.PdfFont
if !to.savedStates.empty() {
font = to.savedStates.top().tfont
}
if font == nil {
2018-09-20 11:49:44 +10:00
common.Log.Debug("ERROR: No font defined. Using default.")
return model.DefaultFont()
}
return font
2018-09-20 11:49:44 +10:00
}
// getFont returns the font named `name` if it exists in the page's resources or an error if it
2018-09-17 12:12:06 +10:00
// 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
}
2018-09-17 12:12:06 +10:00
}
// 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])
2018-09-17 12:12:06 +10:00
}
to.e.fontCache[name] = entry
2018-09-17 12:12:06 +10:00
}
return font, nil
}
2018-09-21 16:43:10 +10:00
// fontEntry is a entry in the font cache.
2018-09-17 12:12:06 +10:00
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
2018-11-28 23:25:17 +00:00
// it doesn't. Accesses page resources directly (not cached).
2018-09-17 12:12:06 +10:00
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 {
2018-09-17 12:12:06 +10:00
common.Log.Debug("getFontDirect: NewPdfFontFromPdfObject failed. name=%#q err=%v", name, err)
}
return font, err
}
2018-12-27 21:33:31 +11:00
// 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
}