summaryrefslogtreecommitdiff
path: root/pdf.go
diff options
context:
space:
mode:
Diffstat (limited to 'pdf.go')
-rw-r--r--pdf.go22
1 files changed, 15 insertions, 7 deletions
diff --git a/pdf.go b/pdf.go
index 4a6e7a1..e99b2a7 100644
--- a/pdf.go
+++ b/pdf.go
@@ -1,6 +1,8 @@
package bookpipeline
import (
+ "bytes"
+ "compress/zlib"
"errors"
"fmt"
"html"
@@ -14,6 +16,7 @@ import (
"rescribe.xyz/utils/pkg/hocr"
)
+// TODO: maybe set this in Fpdf struct
const pageWidth = 5 // pageWidth in inches
// pxToPt converts a pixel value into a pt value (72 pts per inch)
@@ -27,14 +30,22 @@ type Fpdf struct {
}
// Setup creates a new PDF with appropriate settings and fonts
-// TODO: find a font that's closer to the average dimensions of the
-// text we're dealing with
-// TODO: once we have a good font, embed it in the binary as bytes
func (p *Fpdf) Setup() error {
p.fpdf = gofpdf.New("P", "pt", "A4", "")
// Even though it's invisible, we need to add a font which can do
// UTF-8 so that text renders correctly.
- p.fpdf.AddUTF8Font("dejavu", "", "DejaVuSansCondensed.ttf")
+ br := bytes.NewReader(DejaVuCondensedBytes)
+ r, err := zlib.NewReader(br)
+ if err != nil {
+ return err
+ }
+ var buf bytes.Buffer
+ _, err = buf.ReadFrom(r)
+ if err != nil {
+ return err
+ }
+ p.fpdf.AddUTF8FontFromBytes("dejavu", "", buf.Bytes())
+ r.Close()
p.fpdf.SetFont("dejavu", "", 10)
p.fpdf.SetAutoPageBreak(false, float64(0))
return p.fpdf.Error()
@@ -47,7 +58,6 @@ func (p *Fpdf) AddPage(imgpath, hocrpath string) error {
if err != nil {
return errors.New(fmt.Sprintf("Could not read file %s: %v", hocrpath, err))
}
- // TODO: change hocr.Parse to take a Reader rather than []byte
h, err := hocr.Parse(file)
if err != nil {
return errors.New(fmt.Sprintf("Could not parse hocr in file %s: %v", hocrpath, err))
@@ -65,8 +75,6 @@ func (p *Fpdf) AddPage(imgpath, hocrpath string) error {
b := img.Bounds()
p.fpdf.AddPageFormat("P", gofpdf.SizeType{Wd: pxToPt(b.Dx()), Ht: pxToPt(b.Dy())})
- // TODO: check for errors in pdf as going through
-
_ = p.fpdf.RegisterImageOptions(imgpath, gofpdf.ImageOptions{})
p.fpdf.ImageOptions(imgpath, 0, 0, pxToPt(b.Dx()), pxToPt(b.Dy()), false, gofpdf.ImageOptions{}, 0, "")