unipdf/extractor/text_mark.go

/*
 * 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 (
	"fmt"
	"math"

	"github.com/unidoc/unipdf/v3/common"
	"github.com/unidoc/unipdf/v3/internal/transform"
	"github.com/unidoc/unipdf/v3/model"
)

// textMark represents text drawn on a page and its position in device coordinates.
// All dimensions are in device coordinates.
type textMark struct {
	serial             int              // Sequence number for debugging.
	model.PdfRectangle                  // Bounding box.
	text               string           // The text (decoded via ToUnicode).
	original           string           // Original text (decoded).
	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.
}

// 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, bool) {
	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
	}
	if bbox.Llx > bbox.Urx {
		bbox.Llx, bbox.Urx = bbox.Urx, bbox.Llx
	}
	if bbox.Lly > bbox.Ury {
		bbox.Lly, bbox.Ury = bbox.Ury, bbox.Lly
	}

	clipped, onPage := rectIntersection(bbox, to.e.mediaBox)
	if !onPage {
		common.Log.Debug("Text mark outside page. bbox=%g mediaBox=%g text=%q",
			bbox, to.e.mediaBox, text)
	}
	bbox = clipped

	tm := textMark{
		text:          text,
		orient:        orient,
		PdfRectangle:  bbox,
		orientedStart: start.Rotate(theta),
		orientedEnd:   end.Rotate(theta),
		height:        math.Abs(height),
		spaceWidth:    spaceWidth,
		font:          font,
		fontsize:      height,
		charspacing:   charspacing,
		trm:           trm,
		end:           end,
		serial:        serial.mark,
	}
	serial.mark++
	if !isTextSpace(tm.text) && tm.Width() == 0.0 {
		common.Log.Debug("ERROR: Zero width text. tm=%s\n\tm=%#v", tm, tm)
	}

	return tm, onPage
}

// String returns a description of `tm`.
func (tm *textMark) String() string {
	return fmt.Sprintf("serial=%d %.2f fontsize=%.2f \"%s\"",
		tm.serial, tm.PdfRectangle, tm.fontsize, tm.text)
}

// bbox makes textMark implement the `bounded` interface.
func (tm *textMark) bbox() model.PdfRectangle {
	return tm.PdfRectangle
}

// 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{
		count:    int64(tm.serial),
		Text:     tm.text,
		Original: tm.original,
		BBox:     tm.PdfRectangle,
		Font:     tm.font,
		FontSize: tm.fontsize,
	}
}

// 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)
}