diff options
Diffstat (limited to 'internal/pipeline')
| -rw-r--r-- | internal/pipeline/get.go | 2 | ||||
| -rw-r--r-- | internal/pipeline/pipeline.go | 19 | 
2 files changed, 20 insertions, 1 deletions
| diff --git a/internal/pipeline/get.go b/internal/pipeline/get.go index de4ac3b..8fac060 100644 --- a/internal/pipeline/get.go +++ b/internal/pipeline/get.go @@ -68,7 +68,7 @@ func DownloadBestPngs(dir string, name string, conn Downloader) error {  func DownloadPdfs(dir string, name string, conn Downloader) error {  	anydone := false  	errmsg := "" -	for _, suffix := range []string{".colour.pdf", ".binarised.pdf"} { +	for _, suffix := range []string{".colour.pdf", ".binarised.pdf", ".original.pdf"} {  		key := filepath.Join(name, name+suffix)  		fn := filepath.Join(dir, name+suffix)  		err := conn.Download(conn.WIPStorageId(), key, fn) diff --git a/internal/pipeline/pipeline.go b/internal/pipeline/pipeline.go index 40ed02c..cbd179b 100644 --- a/internal/pipeline/pipeline.go +++ b/internal/pipeline/pipeline.go @@ -455,6 +455,12 @@ func Analyse(conn Downloader) func(context.Context, chan string, chan string, ch  			errc <- fmt.Errorf("Failed to set up PDF: %s", err)  			return  		} +		fullsizepdf := new(bookpipeline.Fpdf) +		err = fullsizepdf.Setup() +		if err != nil { +			errc <- fmt.Errorf("Failed to set up PDF: %s", err) +			return +		}  		binhascontent, colourhascontent := false, false  		select { @@ -551,6 +557,11 @@ func Analyse(conn Downloader) func(context.Context, chan string, chan string, ch  					errc <- fmt.Errorf("Failed to add page %s to PDF: %s", pg.img, err)  					return  				} +				err = fullsizepdf.AddPage(filepath.Join(savedir, colourfn), filepath.Join(savedir, pg.hocr), false) +				if err != nil { +					errc <- fmt.Errorf("Failed to add page %s to PDF: %s", pg.img, err) +					return +				}  				colourhascontent = true  				err = os.Remove(filepath.Join(savedir, colourfn))  				if err != nil { @@ -575,6 +586,14 @@ func Analyse(conn Downloader) func(context.Context, chan string, chan string, ch  				return  			}  			up <- fn + +			fn = filepath.Join(savedir, bookname+".original.pdf") +			err = fullsizepdf.Save(fn) +			if err != nil { +				errc <- fmt.Errorf("Failed to save full size pdf: %s", err) +				return +			} +			up <- fn  		}  		select { | 
