/*====================================================================* - Copyright (C) 2001 Leptonica. All rights reserved. - - Redistribution and use in source and binary forms, with or without - modification, are permitted provided that the following conditions - are met: - 1. Redistributions of source code must retain the above copyright - notice, this list of conditions and the following disclaimer. - 2. Redistributions in binary form must reproduce the above - copyright notice, this list of conditions and the following - disclaimer in the documentation and/or other materials - provided with the distribution. - - THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR - A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL ANY - CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, - EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, - PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR - PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY - OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING - NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS - SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. *====================================================================*/ /* * pdfio.c * * |=============================================================| * | Important note | * |=============================================================| * | Some of these functions require libtiff, libjpeg, and libz | * | If you do not have these libraries, you must set | * | #define USE_PDFIO 0 | * | in environ.h. This will link pdfiostub.c | * |=============================================================| * * Set 1. These functions convert a set of image files * to a multi-page pdf file, with one image on each page. * All images are rendered at the same (input) resolution. * The images can be specified as being in a directory, or they * can be in an sarray. The output pdf can be either a file * or an array of bytes in memory. * * Set 2. These functions are a special case of set 1, where * no scaling or change in quality is requires. For jpeg and * jp2k images, the bytes in each jpeg file can be directly * incorporated into the output pdf, and the wrapping up of * multiple image files is very fast. For other image formats, * the image must be read and then the G4 or Flate (gzip) * encodings are generated. * * Set 3. These functions convert a set of images in memory * to a multi-page pdf, with one image on each page. The pdf * output can be either a file or an array of bytes in memory. * * Set 4. These functions implement a pdf output "device driver" * for wrapping (encoding) any number of images on a single page * in pdf. The input can be either an image file or a Pix; * the pdf output can be either a file or an array of bytes in memory. * * Set 5. These "segmented" functions take a set of image * files, along with optional segmentation information, and * generate a multi-page pdf file, where each page consists * in general of a mixed raster pdf of image and non-image regions. * The segmentation information for each page can be input as * either a mask over the image parts, or as a Boxa of those * regions. * * Set 6. These "segmented" functions convert an image and * an optional Boxa of image regions into a mixed raster pdf file * for the page. The input image can be either a file or a Pix. * * Set 7. These functions take a set of single-page pdf files * and concatenates them into a multi-page pdf. * The input can be a set of single page pdf files, or of * pdf 'strings' in memory. The output can be either a file or * an array of bytes in memory. * * The images in the pdf file can be rendered using a pdf viewer, * such as gv, evince, xpdf or acroread. * * Reference on the pdf file format: * http://www.adobe.com/devnet/pdf/pdf_reference_archive.html * * 1. Convert specified image files to pdf (one image file per page) * l_int32 convertFilesToPdf() * l_int32 saConvertFilesToPdf() * l_int32 saConvertFilesToPdfData() * l_int32 selectDefaultPdfEncoding() * * 2. Convert specified image files to pdf without scaling * l_int32 convertUnscaledFilesToPdf() * l_int32 saConvertUnscaledFilesToPdf() * l_int32 saConvertUnscaledFilesToPdfData() * l_int32 convertUnscaledToPdfData() * static L_COMP_DATA *l_generateJp2kData() * static l_int32 cidConvertToPdfData() * * 3. Convert multiple images to pdf (one image per page) * l_int32 pixaConvertToPdf() * l_int32 pixaConvertToPdfData() * * 4. Single page, multi-image converters * l_int32 convertToPdf() * l_int32 convertImageDataToPdf() * l_int32 convertToPdfData() * l_int32 convertImageDataToPdfData() * l_int32 pixConvertToPdf() * l_int32 pixConvertToPdfData() * l_int32 pixWriteStreamPdf() * * 5. Segmented multi-page, multi-image converter * l_int32 convertSegmentedFilesToPdf() * BOXAA *convertNumberedMasksToBoxaa() * * 6. Segmented single page, multi-image converters * l_int32 convertToPdfSegmented() * l_int32 pixConvertToPdfSegmented() * l_int32 convertToPdfDataSegmented() * l_int32 pixConvertToPdfDataSegmented() * * Helper functions for generating the output pdf string * static l_int32 l_generatePdf() * static void generateFixedStringsPdf() * static void generateMediaboxPdf() * static l_int32 generatePageStringPdf() * static l_int32 generateContentStringPdf() * static l_int32 generatePreXStringsPdf() * static l_int32 generateColormapStringsPdf() * static void generateTrailerPdf() * static l_int32 makeTrailerStringPdf() * static l_int32 generateOutputDataPdf() * * 7. Multi-page concatenation * l_int32 concatenatePdf() * l_int32 saConcatenatePdf() * l_int32 ptraConcatenatePdf() * l_int32 concatenatePdfToData() * l_int32 saConcatenatePdfToData() * l_int32 ptraConcatenatePdfToData() * * Helper functions for generating the multi-page pdf output * static l_int32 parseTrailerPdf() * static char *generatePagesObjStringPdf() * static L_BYTEA *substituteObjectNumbers() * * Create/destroy/access pdf data * static L_PDF_DATA *pdfdataCreate() * static void pdfdataDestroy() * static L_COMP_DATA *pdfdataGetCid() * * Set flags for special modes * void l_pdfSetG4ImageMask() * void l_pdfSetDateAndVersion() * * The top-level multi-image functions can be visualized as follows: * Output pdf data to file: * convertToPdf() and convertImageDataToPdf() * --> pixConvertToPdf() * --> pixConvertToPdfData() * * Output pdf data to array in memory: * convertToPdfData() and convertImageDataToPdfData() * --> pixConvertToPdfData() * * The top-level segmented image functions can be visualized as follows: * Output pdf data to file: * convertToPdfSegmented() * --> pixConvertToPdfSegmented() * --> pixConvertToPdfDataSegmented() * * Output pdf data to array in memory: * convertToPdfDataSegmented() * --> pixConvertToPdfDataSegmented() * * For multi-page concatenation, there are three different types of input * (1) directory and optional filename filter * (2) sarray of filenames * (3) ptra of byte arrays of pdf data * and two types of output for the concatenated pdf data * (1) filename * (2) data array and size * High-level interfaces are given for each of the six combinations. * * Note: When wrapping small images into pdf, it is useful to give * them a relatively low resolution value, to avoid rounding errors * when rendering the images. For example, if you want an image * of width w pixels to be 5 inches wide on a screen, choose a * resolution w/5. * * The very fast functions in section (2) require neither transcoding * nor parsing of the compressed jpeg file, because the pdf representation * of DCT-encoded images simply includes the entire jpeg-encoded data * as a byte array in the pdf file. This was a good choice on the part * of the pdf designers. They could have chosen to do the same with FLATE * encoding, by including the png file data as a byte array in the * pdf, but unfortunately they didn't. Whereas png compression * uses a two-dimensional predictor, flate compression simply * gzips the image data. So transcoding of png images is reguired; * to wrap them in flate encoding you must uncompress the image, * gzip the image data, recompress with gzip and generate a colormap * object if it exists. And the resulting one-dimensional compression * is worse than png. For CCITT-G4 compression, again, you can not simply * include a tiff G4 file -- you must either parse it and extract the * G4 compressed data within it, or uncompress to a raster and * compress again. */ #include #include #include "allheaders.h" /* --------------------------------------------*/ #if USE_PDFIO /* defined in environ.h */ /* --------------------------------------------*/ /* Typical scan resolution in ppi (pixels/inch) */ static const l_int32 DEFAULT_INPUT_RES = 300; /* Static helpers */ static L_COMP_DATA *l_generateJp2kData(const char *fname); static l_int32 cidConvertToPdfData(L_COMP_DATA *cid, const char *title, l_uint8 **pdata, size_t *pnbytes); static l_int32 l_generatePdf(l_uint8 **pdata, size_t *pnbytes, L_PDF_DATA *lpd); static void generateFixedStringsPdf(L_PDF_DATA *lpd); static void generateMediaboxPdf(L_PDF_DATA *lpd); static l_int32 generatePageStringPdf(L_PDF_DATA *lpd); static l_int32 generateContentStringPdf(L_PDF_DATA *lpd); static l_int32 generatePreXStringsPdf(L_PDF_DATA *lpd); static l_int32 generateColormapStringsPdf(L_PDF_DATA *lpd); static void generateTrailerPdf(L_PDF_DATA *lpd); static char *makeTrailerStringPdf(L_DNA *daloc); static l_int32 generateOutputDataPdf(l_uint8 **pdata, size_t *pnbytes, L_PDF_DATA *lpd); static l_int32 parseTrailerPdf(L_BYTEA *bas, L_DNA **pda); static char *generatePagesObjStringPdf(NUMA *napage); static L_BYTEA *substituteObjectNumbers(L_BYTEA *bas, NUMA *na_objs); static L_PDF_DATA *pdfdataCreate(const char *title); static void pdfdataDestroy(L_PDF_DATA **plpd); static L_COMP_DATA *pdfdataGetCid(L_PDF_DATA *lpd, l_int32 index); /* ---------------- Defaults for rendering options ----------------- */ /* Output G4 as writing through image mask; this is the default */ static l_int32 var_WRITE_G4_IMAGE_MASK = 1; /* Write date/time and lib version into pdf; this is the default */ static l_int32 var_WRITE_DATE_AND_VERSION = 1; #define L_SMALLBUF 256 #define L_BIGBUF 2048 /* must be able to hold hex colormap */ #ifndef NO_CONSOLE_IO #define DEBUG_MULTIPAGE 0 #endif /* ~NO_CONSOLE_IO */ /*---------------------------------------------------------------------* * Convert specified image files to pdf (one image file per page) * *---------------------------------------------------------------------*/ /*! * convertFilesToPdf() * * Input: directory name (containing images) * substr ( substring filter on filenames; can be NULL) * res (input resolution of all images) * scalefactor (scaling factor applied to each image; > 0.0) * type (encoding type (L_JPEG_ENCODE, L_G4_ENCODE, * L_FLATE_ENCODE, or 0 for default) * quality (used for JPEG only; 0 for default (75)) * title ( pdf title; if null, taken from the first * image filename) * fileout (pdf file of all images) * Return: 0 if OK, 1 on error * * Notes: * (1) If @substr is not NULL, only image filenames that contain * the substring can be used. If @substr == NULL, all files * in the directory are used. * (2) The files in the directory, after optional filtering by * the substring, are lexically sorted in increasing order * before concatenation. * (3) The scalefactor is applied to each image before encoding. * If you enter a value <= 0.0, it will be set to 1.0. * (4) Specifying one of the three encoding types for @type forces * all images to be compressed with that type. Use 0 to have * the type determined for each image based on depth and whether * or not it has a colormap. */ l_int32 convertFilesToPdf(const char *dirname, const char *substr, l_int32 res, l_float32 scalefactor, l_int32 type, l_int32 quality, const char *title, const char *fileout) { l_int32 ret; SARRAY *sa; PROCNAME("convertFilesToPdf"); if (!dirname) return ERROR_INT("dirname not defined", procName, 1); if (!fileout) return ERROR_INT("fileout not defined", procName, 1); if ((sa = getSortedPathnamesInDirectory(dirname, substr, 0, 0)) == NULL) return ERROR_INT("sa not made", procName, 1); ret = saConvertFilesToPdf(sa, res, scalefactor, type, quality, title, fileout); sarrayDestroy(&sa); return ret; } /*! * saConvertFilesToPdf() * * Input: sarray (of pathnames for images) * res (input resolution of all images) * scalefactor (scaling factor applied to each image; > 0.0) * type (encoding type (L_JPEG_ENCODE, L_G4_ENCODE, * L_FLATE_ENCODE, or 0 for default) * quality (used for JPEG only; 0 for default (75)) * title ( pdf title; if null, taken from the first * image filename) * fileout (pdf file of all images) * Return: 0 if OK, 1 on error * * Notes: * (1) See convertFilesToPdf(). */ l_int32 saConvertFilesToPdf(SARRAY *sa, l_int32 res, l_float32 scalefactor, l_int32 type, l_int32 quality, const char *title, const char *fileout) { l_uint8 *data; l_int32 ret; size_t nbytes; PROCNAME("saConvertFilesToPdf"); if (!sa) return ERROR_INT("sa not defined", procName, 1); ret = saConvertFilesToPdfData(sa, res, scalefactor, type, quality, title, &data, &nbytes); if (ret) { if (data) FREE(data); return ERROR_INT("pdf data not made", procName, 1); } ret = l_binaryWrite(fileout, "w", data, nbytes); FREE(data); if (ret) L_ERROR("pdf data not written to file\n", procName); return ret; } /*! * saConvertFilesToPdfData() * * Input: sarray (of pathnames for images) * res (input resolution of all images) * scalefactor (scaling factor applied to each image; > 0.0) * type (encoding type (L_JPEG_ENCODE, L_G4_ENCODE, * L_FLATE_ENCODE, or 0 for default) * quality (used for JPEG only; 0 for default (75)) * title ( pdf title; if null, taken from the first * image filename) * &data ( output pdf data (of all images) * &nbytes ( size of output pdf data) * Return: 0 if OK, 1 on error * * Notes: * (1) See convertFilesToPdf(). */ l_int32 saConvertFilesToPdfData(SARRAY *sa, l_int32 res, l_float32 scalefactor, l_int32 type, l_int32 quality, const char *title, l_uint8 **pdata, size_t *pnbytes) { char *fname; const char *pdftitle; l_uint8 *imdata; l_int32 i, n, ret, pagetype, npages, scaledres; size_t imbytes; L_BYTEA *ba; PIX *pixs, *pix; L_PTRA *pa_data; PROCNAME("saConvertFilesToPdfData"); if (!pdata) return ERROR_INT("&data not defined", procName, 1); *pdata = NULL; if (!pnbytes) return ERROR_INT("&nbytes not defined", procName, 1); *pnbytes = 0; if (!sa) return ERROR_INT("sa not defined", procName, 1); if (scalefactor <= 0.0) scalefactor = 1.0; if (type < 0 || type > L_FLATE_ENCODE) { L_WARNING("invalid compression type; using per-page default\n", procName); type = 0; } /* Generate all the encoded pdf strings */ n = sarrayGetCount(sa); pa_data = ptraCreate(n); pdftitle = NULL; for (i = 0; i < n; i++) { if (i && (i % 10 == 0)) fprintf(stderr, ".. %d ", i); fname = sarrayGetString(sa, i, L_NOCOPY); if ((pixs = pixRead(fname)) == NULL) { L_ERROR("image not readable from file %s\n", procName, fname); continue; } if (!pdftitle) pdftitle = (title) ? title : fname; if (scalefactor != 1.0) pix = pixScale(pixs, scalefactor, scalefactor); else pix = pixClone(pixs); scaledres = (l_int32)(res * scalefactor); if (type != 0) { pagetype = type; } else if (selectDefaultPdfEncoding(pix, &pagetype) != 0) { L_ERROR("encoding type selection failed for file %s\n", procName, fname); continue; } ret = pixConvertToPdfData(pix, pagetype, quality, &imdata, &imbytes, 0, 0, scaledres, pdftitle, NULL, 0); pixDestroy(&pix); pixDestroy(&pixs); if (ret) { L_ERROR("pdf encoding failed for %s\n", procName, fname); continue; } ba = l_byteaInitFromMem(imdata, imbytes); if (imdata) FREE(imdata); ptraAdd(pa_data, ba); } ptraGetActualCount(pa_data, &npages); if (npages == 0) { L_ERROR("no pdf files made\n", procName); ptraDestroy(&pa_data, FALSE, FALSE); return 1; } /* Concatenate them */ fprintf(stderr, "\nconcatenating ... "); ret = ptraConcatenatePdfToData(pa_data, NULL, pdata, pnbytes); fprintf(stderr, "done\n"); ptraGetActualCount(pa_data, &npages); /* recalculate in case it changes */ for (i = 0; i < npages; i++) { ba = (L_BYTEA *)ptraRemove(pa_data, i, L_NO_COMPACTION); l_byteaDestroy(&ba); } ptraDestroy(&pa_data, FALSE, FALSE); return ret; } /*! * selectDefaultPdfEncoding() * * Input: pix * &type ( L_G4_ENCODE, L_JPEG_ENCODE, L_FLATE_ENCODE) * * Notes: * (1) This attempts to choose an encoding for the pix that results * in the smallest file, assuming that if jpeg encoded, it will * use quality = 75. The decision is approximate, in that * (a) all colormapped images will be losslessly encoded with * gzip (flate), and (b) an image with less than about 20 colors * is likely to be smaller if flate encoded than if encoded * as a jpeg (dct). For example, an image made by pixScaleToGray3() * will have 10 colors, and flate encoding will give about * twice the compression as jpeg with quality = 75. */ l_int32 selectDefaultPdfEncoding(PIX *pix, l_int32 *ptype) { l_int32 w, h, d, factor, ncolors; PIXCMAP *cmap; PROCNAME("selectDefaultPdfEncoding"); if (!pix) return ERROR_INT("pix not defined", procName, 1); if (!ptype) return ERROR_INT("&type not defined", procName, 1); *ptype = L_FLATE_ENCODE; /* default universal encoding */ pixGetDimensions(pix, &w, &h, &d); cmap = pixGetColormap(pix); if (d == 8 && !cmap) { factor = L_MAX(1, (l_int32)sqrt((l_float64)(w * h) / 20000.)); pixNumColors(pix, factor, &ncolors); if (ncolors < 20) *ptype = L_FLATE_ENCODE; else *ptype = L_JPEG_ENCODE; } else if (d == 1) { *ptype = L_G4_ENCODE; } else if (cmap || d == 2 || d == 4) { *ptype = L_FLATE_ENCODE; } else if (d == 8 || d == 32) { *ptype = L_JPEG_ENCODE; } else { return ERROR_INT("type selection failure", procName, 1); } return 0; } /*---------------------------------------------------------------------* * Convert specified image files to pdf without scaling * *---------------------------------------------------------------------*/ /*! * convertUnscaledFilesToPdf() * * Input: directory name (containing images) * substr ( substring filter on filenames; can be NULL) * title ( pdf title; if null, taken from the first * image filename) * fileout (pdf file of all images) * Return: 0 if OK, 1 on error * * Notes: * (1) If @substr is not NULL, only image filenames that contain * the substring can be used. If @substr == NULL, all files * in the directory are used. * (2) The files in the directory, after optional filtering by * the substring, are lexically sorted in increasing order * before concatenation. * (3) For jpeg and jp2k, this is very fast because the compressed * data is wrapped up and concatenated. For png and tiffg4, * the images must be read and recompressed. */ l_int32 convertUnscaledFilesToPdf(const char *dirname, const char *substr, const char *title, const char *fileout) { l_int32 ret; SARRAY *sa; PROCNAME("convertUnscaledFilesToPdf"); if (!dirname) return ERROR_INT("dirname not defined", procName, 1); if (!fileout) return ERROR_INT("fileout not defined", procName, 1); if ((sa = getSortedPathnamesInDirectory(dirname, substr, 0, 0)) == NULL) return ERROR_INT("sa not made", procName, 1); ret = saConvertUnscaledFilesToPdf(sa, title, fileout); sarrayDestroy(&sa); return ret; } /*! * saConvertUnscaledFilesToPdf() * * Input: sarray (of pathnames for images) * title ( pdf title; if null, taken from the first * image filename) * fileout (pdf file of all images) * Return: 0 if OK, 1 on error * * Notes: * (1) See convertUnscaledFilesToPdf(). */ l_int32 saConvertUnscaledFilesToPdf(SARRAY *sa, const char *title, const char *fileout) { l_uint8 *data; l_int32 ret; size_t nbytes; PROCNAME("saConvertUnscaledFilesToPdf"); if (!sa) return ERROR_INT("sa not defined", procName, 1); ret = saConvertUnscaledFilesToPdfData(sa, title, &data, &nbytes); if (ret) { if (data) FREE(data); return ERROR_INT("pdf data not made", procName, 1); } ret = l_binaryWrite(fileout, "w", data, nbytes); FREE(data); if (ret) L_ERROR("pdf data not written to file\n", procName); return ret; } /*! * saConvertUnscaledFilesToPdfData() * * Input: sarray (of pathnames for images) * title ( pdf title; if null, taken from the first * image filename) * &data ( output pdf data (of all images) * &nbytes ( size of output pdf data) * Return: 0 if OK, 1 on error */ l_int32 saConvertUnscaledFilesToPdfData(SARRAY *sa, const char *title, l_uint8 **pdata, size_t *pnbytes) { char *fname; l_uint8 *imdata; l_int32 i, n, ret, npages; size_t imbytes; L_BYTEA *ba; L_PTRA *pa_data; PROCNAME("saConvertUnscaledFilesToPdfData"); if (!pdata) return ERROR_INT("&data not defined", procName, 1); *pdata = NULL; if (!pnbytes) return ERROR_INT("&nbytes not defined", procName, 1); *pnbytes = 0; if (!sa) return ERROR_INT("sa not defined", procName, 1); /* Generate all the encoded pdf strings */ n = sarrayGetCount(sa); pa_data = ptraCreate(n); for (i = 0; i < n; i++) { if (i && (i % 10 == 0)) fprintf(stderr, ".. %d ", i); fname = sarrayGetString(sa, i, L_NOCOPY); /* Generate the pdf data */ if (convertUnscaledToPdfData(fname, title, &imdata, &imbytes)) continue; /* ... and add it to the array of single page data */ ba = l_byteaInitFromMem(imdata, imbytes); if (imdata) FREE(imdata); ptraAdd(pa_data, ba); } ptraGetActualCount(pa_data, &npages); if (npages == 0) { L_ERROR("no pdf files made", procName); ptraDestroy(&pa_data, FALSE, FALSE); return 1; } /* Concatenate to generate a multipage pdf */ fprintf(stderr, "\nconcatenating ... "); ret = ptraConcatenatePdfToData(pa_data, NULL, pdata, pnbytes); fprintf(stderr, "done\n"); /* Clean up */ ptraGetActualCount(pa_data, &npages); /* maybe failed to read some files */ for (i = 0; i < npages; i++) { ba = (L_BYTEA *)ptraRemove(pa_data, i, L_NO_COMPACTION); l_byteaDestroy(&ba); } ptraDestroy(&pa_data, FALSE, FALSE); return ret; } /*! * convertUnscaledToPdfData() * * Input: fname (of image file) * title ( pdf title; can be NULL) * &data ( output pdf data for image) * &nbytes ( size of output pdf data) * Return: 0 if OK, 1 on error */ l_int32 convertUnscaledToPdfData(const char *fname, const char *title, l_uint8 **pdata, size_t *pnbytes) { const char *pdftitle = NULL; char *tail = NULL; l_int32 format; L_COMP_DATA *cid; PROCNAME("convertUnscaledToPdfData"); if (!pdata) return ERROR_INT("&data not defined", procName, 1); *pdata = NULL; if (!pnbytes) return ERROR_INT("&nbytes not defined", procName, 1); *pnbytes = 0; if (!fname) return ERROR_INT("fname not defined", procName, 1); findFileFormat(fname, &format); if (format == IFF_UNKNOWN) { L_WARNING("file %s format is unknown; skip\n", procName, fname); return 1; } if (format == IFF_PS || format == IFF_LPDF) { L_WARNING("file %s format is %d; skip\n", procName, fname, format); return 1; } /* Generate the image data required for pdf generation, always * in binary (not ascii85) coding. */ if (format == IFF_JFIF_JPEG) cid = l_generateJpegData(fname, 0); else if (format == IFF_JP2) cid = l_generateJp2kData(fname); else if (format == IFF_TIFF_G4) cid = l_generateG4Data(fname, 0); else /* format == IFF_PNG, etc */ cid = l_generateFlateData(fname, 0); if (!cid) { L_ERROR("file %s format is %d; unreadable\n", procName, fname, format); return 1; } /* If @title == NULL, use the tail of @fname. */ if (title) { pdftitle = title; } else { splitPathAtDirectory(fname, NULL, &tail); pdftitle = tail; } /* Generate the pdf string for this page (image). This destroys * the cid by attaching it to an lpd and destroying the lpd. */ cidConvertToPdfData(cid, pdftitle, pdata, pnbytes); FREE(tail); return 0; } /*! * l_generateJp2kData() * * Input: fname (of jp2k file) * Return: cid (containing jp2k data), or null on error * * Notes: * (1) This is only called after the file is verified to be jp2k. * (2) It is here and not in psio2.c because jp2k can't be wrapped * in PostScript. */ static L_COMP_DATA * l_generateJp2kData(const char *fname) { l_uint8 *data = NULL; l_int32 w, h, spp; size_t nbytes; L_COMP_DATA *cid; PROCNAME("l_generateJp2kData"); if (!fname) return (L_COMP_DATA *)ERROR_PTR("fname not defined", procName, NULL); if ((cid = (L_COMP_DATA *)CALLOC(1, sizeof(L_COMP_DATA))) == NULL) return (L_COMP_DATA *)ERROR_PTR("cid not made", procName, NULL); /* The returned jp2k data in memory is the entire jp2k file */ if ((cid->datacomp = l_binaryRead(fname, &nbytes)) == NULL) return (L_COMP_DATA *)ERROR_PTR("data not extracted", procName, NULL); readHeaderJp2k(fname, &w, &h, &spp); cid->type = L_JP2K_ENCODE; cid->nbytescomp = nbytes; cid->w = w; cid->h = h; cid->bps = 8; cid->spp = spp; cid->res = 0; /* don't know how to extract this */ return cid; } /*! * cidConvertToPdfData() * * Input: cid (compressed image data -- of jp2k image) * title ( pdf title; can be NULL) * &data ( output pdf data for image) * &nbytes ( size of output pdf data) * Return: 0 if OK, 1 on error * * Notes: * (1) Caller must not destroy the cid. It is absorbed in the * lpd and destroyed by this function. */ static l_int32 cidConvertToPdfData(L_COMP_DATA *cid, const char *title, l_uint8 **pdata, size_t *pnbytes) { l_int32 res, ret; l_float32 wpt, hpt; L_PDF_DATA *lpd = NULL; PROCNAME("cidConvertToPdfData"); if (!pdata || !pnbytes) return ERROR_INT("&data and &nbytes not both defined", procName, 1); *pdata = NULL; *pnbytes = 0; if (!cid) return ERROR_INT("cid not defined", procName, 1); /* Get media box parameters, in pts */ res = cid->res; if (res <= 0) res = DEFAULT_INPUT_RES; wpt = cid->w * 72. / res; hpt = cid->h * 72. / res; /* Set up the pdf data struct (lpd) */ if ((lpd = pdfdataCreate(title)) == NULL) return ERROR_INT("lpd not made", procName, 1); ptraAdd(lpd->cida, cid); lpd->n++; ptaAddPt(lpd->xy, 0, 0); /* xpt = ypt = 0 */ ptaAddPt(lpd->wh, wpt, hpt); /* Generate the pdf string and destroy the lpd */ ret = l_generatePdf(pdata, pnbytes, lpd); pdfdataDestroy(&lpd); if (ret) return ERROR_INT("pdf output not made", procName, 1); return 0; } /*---------------------------------------------------------------------* * Convert multiple images to pdf (one image per page) * *---------------------------------------------------------------------*/ /*! * pixaConvertToPdf() * * Input: pixa (containing images all at the same resolution) * res (override the resolution of each input image, in ppi; * use 0 to respect the resolution embedded in the input) * scalefactor (scaling factor applied to each image; > 0.0) * type (encoding type (L_JPEG_ENCODE, L_G4_ENCODE, * L_FLATE_ENCODE, or 0 for default) * quality (used for JPEG only; 0 for default (75)) * title ( pdf title; if null, taken from the first * image filename) * fileout (pdf file of all images) * Return: 0 if OK, 1 on error * * Notes: * (1) The images are encoded with G4 if 1 bpp; JPEG if 8 bpp without * colormap and many colors, or 32 bpp; FLATE for anything else. * (2) The scalefactor must be > 0.0; otherwise it is set to 1.0. * (3) Specifying one of the three encoding types for @type forces * all images to be compressed with that type. Use 0 to have * the type determined for each image based on depth and whether * or not it has a colormap. */ l_int32 pixaConvertToPdf(PIXA *pixa, l_int32 res, l_float32 scalefactor, l_int32 type, l_int32 quality, const char *title, const char *fileout) { l_uint8 *data; l_int32 ret; size_t nbytes; PROCNAME("pixaConvertToPdf"); if (!pixa) return ERROR_INT("pixa not defined", procName, 1); ret = pixaConvertToPdfData(pixa, res, scalefactor, type, quality, title, &data, &nbytes); if (ret) { FREE(data); return ERROR_INT("conversion to pdf failed", procName, 1); } ret = l_binaryWrite(fileout, "w", data, nbytes); FREE(data); if (ret) L_ERROR("pdf data not written to file\n", procName); return ret; } /*! * pixaConvertToPdfData() * * Input: pixa (containing images all at the same resolution) * res (input resolution of all images) * scalefactor (scaling factor applied to each image; > 0.0) * type (encoding type (L_JPEG_ENCODE, L_G4_ENCODE, * L_FLATE_ENCODE, or 0 for default) * quality (used for JPEG only; 0 for default (75)) * title ( pdf title) * &data ( output pdf data (of all images) * &nbytes ( size of output pdf data) * Return: 0 if OK, 1 on error * * Notes: * (1) See pixaConvertToPdf(). */ l_int32 pixaConvertToPdfData(PIXA *pixa, l_int32 res, l_float32 scalefactor, l_int32 type, l_int32 quality, const char *title, l_uint8 **pdata, size_t *pnbytes) { l_uint8 *imdata; l_int32 i, n, ret, scaledres, pagetype; size_t imbytes; L_BYTEA *ba; PIX *pixs, *pix; L_PTRA *pa_data; PROCNAME("pixaConvertToPdfData"); if (!pdata) return ERROR_INT("&data not defined", procName, 1); *pdata = NULL; if (!pnbytes) return ERROR_INT("&nbytes not defined", procName, 1); *pnbytes = 0; if (!pixa) return ERROR_INT("pixa not defined", procName, 1); if (scalefactor <= 0.0) scalefactor = 1.0; if (type < 0 || type > L_FLATE_ENCODE) { L_WARNING("invalid compression type; using per-page default\n", procName); type = 0; } /* Generate all the encoded pdf strings */ n = pixaGetCount(pixa); pa_data = ptraCreate(n); for (i = 0; i < n; i++) { if ((pixs = pixaGetPix(pixa, i, L_CLONE)) == NULL) { L_ERROR("pix[%d] not retrieved\n", procName, i); continue; } if (scalefactor != 1.0) pix = pixScale(pixs, scalefactor, scalefactor); else pix = pixClone(pixs); pixDestroy(&pixs); scaledres = (l_int32)(res * scalefactor); if (type != 0) { pagetype = type; } else if (selectDefaultPdfEncoding(pix, &pagetype) != 0) { L_ERROR("encoding type selection failed for pix[%d]\n", procName, i); pixDestroy(&pix); continue; } ret = pixConvertToPdfData(pix, pagetype, quality, &imdata, &imbytes, 0, 0, scaledres, title, NULL, 0); pixDestroy(&pix); if (ret) { L_ERROR("pdf encoding failed for pix[%d]\n", procName, i); continue; } ba = l_byteaInitFromMem(imdata, imbytes); if (imdata) FREE(imdata); ptraAdd(pa_data, ba); } ptraGetActualCount(pa_data, &n); if (n == 0) { L_ERROR("no pdf files made\n", procName); ptraDestroy(&pa_data, FALSE, FALSE); return 1; } /* Concatenate them */ ret = ptraConcatenatePdfToData(pa_data, NULL, pdata, pnbytes); ptraGetActualCount(pa_data, &n); /* recalculate in case it changes */ for (i = 0; i < n; i++) { ba = (L_BYTEA *)ptraRemove(pa_data, i, L_NO_COMPACTION); l_byteaDestroy(&ba); } ptraDestroy(&pa_data, FALSE, FALSE); return ret; } /*---------------------------------------------------------------------* * Single page, multi-image converters * *---------------------------------------------------------------------*/ /*! * convertToPdf() * * Input: filein (input image file -- any format) * type (L_G4_ENCODE, L_JPEG_ENCODE, L_FLATE_ENCODE) * quality (used for JPEG only; 0 for default (75)) * fileout (output pdf file; only required on last image on page) * x, y (location of lower-left corner of image, in pixels, * relative to the PostScript origin (0,0) at * the lower-left corner of the page) * res (override the resolution of the input image, in ppi; * use 0 to respect the resolution embedded in the input) * title ( pdf title; if null, taken from filein) * &lpd (ptr to lpd, which is created on the first invocation * and returned until last image is processed, at which * time it is destroyed) * position (in image sequence: L_FIRST_IMAGE, L_NEXT_IMAGE, * L_LAST_IMAGE) * Return: 0 if OK, 1 on error * * Notes: * (1) To wrap only one image in pdf, input @plpd = NULL, and * the value of @position will be ignored: * convertToPdf(... type, quality, x, y, res, NULL, 0); * (2) To wrap multiple images on a single pdf page, this is called * once for each successive image. Do it this way: * L_PDF_DATA *lpd; * convertToPdf(... type, quality, x, y, res, &lpd, L_FIRST_IMAGE); * convertToPdf(... type, quality, x, y, res, &lpd, L_NEXT_IMAGE); * ... * convertToPdf(... type, quality, x, y, res, &lpd, L_LAST_IMAGE); * This will write the result to the value of @fileout specified * in the first call; succeeding values of @fileout are ignored. * On the last call: the pdf data bytes are computed and written * to @fileout, lpd is destroyed internally, and the returned * value of lpd is null. So the client has nothing to clean up. * (3) (a) Set @res == 0 to respect the resolution embedded in the * image file. If no resolution is embedded, it will be set * to the default value. * (b) Set @res to some other value to override the file resolution. * (4) (a) If the input @res and the resolution of the output device * are equal, the image will be "displayed" at the same size * as the original. * (b) If the input @res is 72, the output device will render * the image at 1 pt/pixel. * (c) Some possible choices for the default input pix resolution are: * 72 ppi Render pix on any output device at one pt/pixel * 96 ppi Windows default for generated display images * 300 ppi Typical default for scanned images. * We choose 300, which is sensible for rendering page images. * However, images come from a variety of sources, and * some are explicitly created for viewing on a display. */ l_int32 convertToPdf(const char *filein, l_int32 type, l_int32 quality, const char *fileout, l_int32 x, l_int32 y, l_int32 res, const char *title, L_PDF_DATA **plpd, l_int32 position) { l_uint8 *data; l_int32 ret; size_t nbytes; PROCNAME("convertToPdf"); if (!filein) return ERROR_INT("filein not defined", procName, 1); if (!plpd || (position == L_LAST_IMAGE)) { if (!fileout) return ERROR_INT("fileout not defined", procName, 1); } if (type != L_G4_ENCODE && type != L_JPEG_ENCODE && type != L_FLATE_ENCODE) return ERROR_INT("invalid conversion type", procName, 1); if (convertToPdfData(filein, type, quality, &data, &nbytes, x, y, res, title, plpd, position)) return ERROR_INT("pdf data not made", procName, 1); if (!plpd || (position == L_LAST_IMAGE)) { ret = l_binaryWrite(fileout, "w", data, nbytes); FREE(data); if (ret) return ERROR_INT("pdf data not written to file", procName, 1); } return 0; } /*! * convertImageDataToPdf() * * Input: imdata (array of formatted image data; e.g., png, jpeg) * size (size of image data) * type (L_G4_ENCODE, L_JPEG_ENCODE, L_FLATE_ENCODE) * quality (used for JPEG only; 0 for default (75)) * fileout (output pdf file; only required on last image on page) * x, y (location of lower-left corner of image, in pixels, * relative to the PostScript origin (0,0) at * the lower-left corner of the page) * res (override the resolution of the input image, in ppi; * use 0 to respect the resolution embedded in the input) * title ( pdf title) * &lpd (ptr to lpd, which is created on the first invocation * and returned until last image is processed, at which * time it is destroyed) * position (in image sequence: L_FIRST_IMAGE, L_NEXT_IMAGE, * L_LAST_IMAGE) * Return: 0 if OK, 1 on error * * Notes: * (1) If @res == 0 and the input resolution field is 0, * this will use DEFAULT_INPUT_RES. * (2) See comments in convertToPdf(). */ l_int32 convertImageDataToPdf(l_uint8 *imdata, size_t size, l_int32 type, l_int32 quality, const char *fileout, l_int32 x, l_int32 y, l_int32 res, const char *title, L_PDF_DATA **plpd, l_int32 position) { l_int32 ret; PIX *pix; PROCNAME("convertImageDataToPdf"); if (!imdata) return ERROR_INT("image data not defined", procName, 1); if (type != L_G4_ENCODE && type != L_JPEG_ENCODE && type != L_FLATE_ENCODE) return ERROR_INT("invalid conversion type", procName, 1); if (!plpd || (position == L_LAST_IMAGE)) { if (!fileout) return ERROR_INT("fileout not defined", procName, 1); } if ((pix = pixReadMem(imdata, size)) == NULL) return ERROR_INT("pix not read", procName, 1); ret = pixConvertToPdf(pix, type, quality, fileout, x, y, res, title, plpd, position); pixDestroy(&pix); return ret; } /*! * convertToPdfData() * * Input: filein (input image file -- any format) * type (L_G4_ENCODE, L_JPEG_ENCODE, L_FLATE_ENCODE) * quality (used for JPEG only; 0 for default (75)) * &data ( pdf data in memory) * &nbytes ( number of bytes in pdf data) * x, y (location of lower-left corner of image, in pixels, * relative to the PostScript origin (0,0) at * the lower-left corner of the page) * res (override the resolution of the input image, in ppi; * use 0 to respect the resolution embedded in the input) * title ( pdf title; if null, use filein) * &lpd (ptr to lpd, which is created on the first invocation * and returned until last image is processed, at which * time it is destroyed) * position (in image sequence: L_FIRST_IMAGE, L_NEXT_IMAGE, * L_LAST_IMAGE) * Return: 0 if OK, 1 on error * * Notes: * (1) If @res == 0 and the input resolution field is 0, * this will use DEFAULT_INPUT_RES. * (2) See comments in convertToPdf(). */ l_int32 convertToPdfData(const char *filein, l_int32 type, l_int32 quality, l_uint8 **pdata, size_t *pnbytes, l_int32 x, l_int32 y, l_int32 res, const char *title, L_PDF_DATA **plpd, l_int32 position) { PIX *pix; PROCNAME("convertToPdfData"); if (!pdata) return ERROR_INT("&data not defined", procName, 1); *pdata = NULL; if (!pnbytes) return ERROR_INT("&nbytes not defined", procName, 1); *pnbytes = 0; if (!filein) return ERROR_INT("filein not defined", procName, 1); if (type != L_G4_ENCODE && type != L_JPEG_ENCODE && type != L_FLATE_ENCODE) return ERROR_INT("invalid conversion type", procName, 1); if ((pix = pixRead(filein)) == NULL) return ERROR_INT("pix not made", procName, 1); pixConvertToPdfData(pix, type, quality, pdata, pnbytes, x, y, res, (title) ? title : filein, plpd, position); pixDestroy(&pix); return 0; } /*! * convertImageDataToPdfData() * * Input: imdata (array of formatted image data; e.g., png, jpeg) * size (size of image data) * type (L_G4_ENCODE, L_JPEG_ENCODE, L_FLATE_ENCODE) * quality (used for JPEG only; 0 for default (75)) * &data ( pdf data in memory) * &nbytes ( number of bytes in pdf data) * x, y (location of lower-left corner of image, in pixels, * relative to the PostScript origin (0,0) at * the lower-left corner of the page) * res (override the resolution of the input image, in ppi; * use 0 to respect the resolution embedded in the input) * title ( pdf title) * &lpd (ptr to lpd, which is created on the first invocation * and returned until last image is processed, at which * time it is destroyed) * position (in image sequence: L_FIRST_IMAGE, L_NEXT_IMAGE, * L_LAST_IMAGE) * Return: 0 if OK, 1 on error * * Notes: * (1) If @res == 0 and the input resolution field is 0, * this will use DEFAULT_INPUT_RES. * (2) See comments in convertToPdf(). */ l_int32 convertImageDataToPdfData(l_uint8 *imdata, size_t size, l_int32 type, l_int32 quality, l_uint8 **pdata, size_t *pnbytes, l_int32 x, l_int32 y, l_int32 res, const char *title, L_PDF_DATA **plpd, l_int32 position) { l_int32 ret; PIX *pix; PROCNAME("convertImageDataToPdfData"); if (!pdata) return ERROR_INT("&data not defined", procName, 1); *pdata = NULL; if (!pnbytes) return ERROR_INT("&nbytes not defined", procName, 1); *pnbytes = 0; if (!imdata) return ERROR_INT("image data not defined", procName, 1); if (plpd) { /* part of multi-page invocation */ if (position == L_FIRST_IMAGE) *plpd = NULL; } if ((pix = pixReadMem(imdata, size)) == NULL) return ERROR_INT("pix not read", procName, 1); ret = pixConvertToPdfData(pix, type, quality, pdata, pnbytes, x, y, res, title, plpd, position); pixDestroy(&pix); return ret; } /*! * pixConvertToPdf() * * Input: pix * type (L_G4_ENCODE, L_JPEG_ENCODE, L_FLATE_ENCODE) * quality (used for JPEG only; 0 for default (75)) * fileout (output pdf file; only required on last image on page) * x, y (location of lower-left corner of image, in pixels, * relative to the PostScript origin (0,0) at * the lower-left corner of the page) * res (override the resolution of the input image, in ppi; * use 0 to respect the resolution embedded in the input) * title ( pdf title) * &lpd (ptr to lpd, which is created on the first invocation * and returned until last image is processed) * position (in image sequence: L_FIRST_IMAGE, L_NEXT_IMAGE, * L_LAST_IMAGE) * Return: 0 if OK, 1 on error * * Notes: * (1) If @res == 0 and the input resolution field is 0, * this will use DEFAULT_INPUT_RES. * (2) This only writes data to fileout if it is the last * image to be written on the page. * (3) See comments in convertToPdf(). */ l_int32 pixConvertToPdf(PIX *pix, l_int32 type, l_int32 quality, const char *fileout, l_int32 x, l_int32 y, l_int32 res, const char *title, L_PDF_DATA **plpd, l_int32 position) { l_uint8 *data; l_int32 ret; size_t nbytes; PROCNAME("pixConvertToPdf"); if (!pix) return ERROR_INT("pix not defined", procName, 1); if (type != L_G4_ENCODE && type != L_JPEG_ENCODE && type != L_FLATE_ENCODE) return ERROR_INT("invalid conversion type", procName, 1); if (!plpd || (position == L_LAST_IMAGE)) { if (!fileout) return ERROR_INT("fileout not defined", procName, 1); } if (pixConvertToPdfData(pix, type, quality, &data, &nbytes, x, y, res, title, plpd, position)) return ERROR_INT("pdf data not made", procName, 1); if (!plpd || (position == L_LAST_IMAGE)) { ret = l_binaryWrite(fileout, "w", data, nbytes); FREE(data); if (ret) return ERROR_INT("pdf data not written to file", procName, 1); } return 0; } /*! * pixConvertToPdfData() * * Input: pix (all depths; cmap OK) * type (L_G4_ENCODE, L_JPEG_ENCODE, L_FLATE_ENCODE) * quality (used for JPEG only; 0 for default (75)) * &data ( pdf array) * &nbytes ( number of bytes in pdf array) * x, y (location of lower-left corner of image, in pixels, * relative to the PostScript origin (0,0) at * the lower-left corner of the page) * res (override the resolution of the input image, in ppi; * use 0 to respect the resolution embedded in the input) * title ( pdf title) * &lpd (ptr to lpd, which is created on the first invocation * and returned until last image is processed) * position (in image sequence: L_FIRST_IMAGE, L_NEXT_IMAGE, * L_LAST_IMAGE) * Return: 0 if OK, 1 on error * * Notes: * (1) If @res == 0 and the input resolution field is 0, * this will use DEFAULT_INPUT_RES. * (2) This only writes @data if it is the last image to be * written on the page. * (3) See comments in convertToPdf(). */ l_int32 pixConvertToPdfData(PIX *pix, l_int32 type, l_int32 quality, l_uint8 **pdata, size_t *pnbytes, l_int32 x, l_int32 y, l_int32 res, const char *title, L_PDF_DATA **plpd, l_int32 position) { l_int32 pixres, w, h, ret; l_float32 xpt, ypt, wpt, hpt; L_COMP_DATA *cid = NULL; L_PDF_DATA *lpd = NULL; PROCNAME("pixConvertToPdfData"); if (!pdata) return ERROR_INT("&data not defined", procName, 1); *pdata = NULL; if (!pnbytes) return ERROR_INT("&nbytes not defined", procName, 1); *pnbytes = 0; if (!pix) return ERROR_INT("pix not defined", procName, 1); if (plpd) { /* part of multi-page invocation */ if (position == L_FIRST_IMAGE) *plpd = NULL; } /* Generate the compressed image data. It must NOT * be ascii85 encoded. */ pixGenerateCIData(pix, type, quality, 0, &cid); if (!cid) return ERROR_INT("cid not made", procName, 1); /* Get media box in pts. Guess the input image resolution * based on the input parameter @res, the resolution data in * the pix, and the size of the image. */ pixres = cid->res; w = cid->w; h = cid->h; if (res <= 0.0) { if (pixres > 0) res = pixres; else res = DEFAULT_INPUT_RES; } xpt = x * 72. / res; ypt = y * 72. / res; wpt = w * 72. / res; hpt = h * 72. / res; /* Set up lpd */ if (!plpd) { /* single image */ if ((lpd = pdfdataCreate(title)) == NULL) return ERROR_INT("lpd not made", procName, 1); } else if (position == L_FIRST_IMAGE) { /* first of multiple images */ if ((lpd = pdfdataCreate(title)) == NULL) return ERROR_INT("lpd not made", procName, 1); *plpd = lpd; } else { /* not the first of multiple images */ lpd = *plpd; } /* Add the data to the lpd */ ptraAdd(lpd->cida, cid); lpd->n++; ptaAddPt(lpd->xy, xpt, ypt); ptaAddPt(lpd->wh, wpt, hpt); /* If a single image or the last of multiple images, * generate the pdf and destroy the lpd */ if (!plpd || (position == L_LAST_IMAGE)) { ret = l_generatePdf(pdata, pnbytes, lpd); pdfdataDestroy(&lpd); if (plpd) *plpd = NULL; if (ret) return ERROR_INT("pdf output not made", procName, 1); } return 0; } /*! * pixWriteStreamPdf() * * Input: fp (stream opened for writing) * pix (all depths, cmap OK) * res (override the resolution of the input image, in ppi; * use 0 to respect the resolution embedded in the input) * title ( pdf title; taken from the first image * placed on a page; e.g., an input image filename) * Return: 0 if OK, 1 on error * * Notes: * (1) This is the simplest interface for writing a single image * with pdf encoding. It uses G4 encoding for 1 bpp, * JPEG encoding for 8 bpp (no cmap) and 32 bpp, and FLATE * encoding for everything else. */ l_int32 pixWriteStreamPdf(FILE *fp, PIX *pix, l_int32 res, const char *title) { l_uint8 *data; l_int32 ret, d, type; size_t nbytes; PIXCMAP *cmap; PROCNAME("pixWriteStreamPdf"); if (!fp) return ERROR_INT("stream not opened", procName, 1); if (!pix) return ERROR_INT("pix not defined", procName, 1); d = pixGetDepth(pix); cmap = pixGetColormap(pix); if (d == 1) type = L_G4_ENCODE; else if (cmap || d == 2 || d == 4 || d == 16) type = L_FLATE_ENCODE; else /* d == 8 (no cmap) or d == 32 */ type = L_JPEG_ENCODE; if (pixConvertToPdfData(pix, type, 75, &data, &nbytes, 0, 0, res, title, NULL, 0)) return ERROR_INT("pdf data not made", procName, 1); ret = fwrite(data, 1, nbytes, fp); FREE(data); if (ret) return ERROR_INT("pdf data not written to stream", procName, 1); return 0; } /*---------------------------------------------------------------------* * Segmented multi-page, multi-image converter * *---------------------------------------------------------------------*/ /*! * convertSegmentedFilesToPdf() * * Input: directory name (containing images) * substr ( substring filter on filenames; can be NULL) * res (input resolution of all images) * type (compression type for non-image regions; the * image regions are always compressed with L_JPEG_ENCODE) * thresh (used for converting gray --> 1 bpp with L_G4_ENCODE) * boxaa ( of image regions) * quality (used for JPEG only; 0 for default (75)) * scalefactor (scaling factor applied to each image region) * title ( pdf title; if null, taken from the first * image filename) * fileout (pdf file of all images) * Return: 0 if OK, 1 on error * * Notes: * (1) If @substr is not NULL, only image filenames that contain * the substring can be used. If @substr == NULL, all files * in the directory are used. * (2) The files in the directory, after optional filtering by * the substring, are lexically sorted in increasing order * before concatenation. * (3) The images are encoded with G4 if 1 bpp; JPEG if 8 bpp without * colormap and many colors, or 32 bpp; FLATE for anything else. * (4) The boxaa, if it exists, contains one boxa of "image regions" * for each image file. The boxa must be aligned with the * sorted set of images. * (5) The scalefactor is applied to each image region. It is * typically < 1.0, to save bytes in the final pdf, because * the resolution is often not critical in non-text regions. * (6) If the non-image regions have pixel depth > 1 and the encoding * type is G4, they are automatically scaled up by 2x and * thresholded. Otherwise, no scaling is performed on them. * (7) Note that this function can be used to generate multipage * G4 compressed pdf from any input, by using @boxaa == NULL * and @type == L_G4_ENCODE. */ l_int32 convertSegmentedFilesToPdf(const char *dirname, const char *substr, l_int32 res, l_int32 type, l_int32 thresh, BOXAA *baa, l_int32 quality, l_float32 scalefactor, const char *title, const char *fileout) { char *fname; l_uint8 *imdata, *data; l_int32 i, npages, nboxa, nboxes, ret; size_t imbytes, databytes; BOXA *boxa; L_BYTEA *ba; L_PTRA *pa_data; SARRAY *sa; PROCNAME("convertSegmentedFilesToPdf"); if (!dirname) return ERROR_INT("dirname not defined", procName, 1); if (!fileout) return ERROR_INT("fileout not defined", procName, 1); if ((sa = getNumberedPathnamesInDirectory(dirname, substr, 0, 0, 10000)) == NULL) return ERROR_INT("sa not made", procName, 1); npages = sarrayGetCount(sa); /* If necessary, extend the boxaa, which is page-aligned with * the image files, to be as large as the set of images. */ if (baa) { nboxa = boxaaGetCount(baa); if (nboxa < npages) { boxa = boxaCreate(1); boxaaExtendWithInit(baa, npages, boxa); boxaDestroy(&boxa); } } /* Generate and save all the encoded pdf strings */ pa_data = ptraCreate(npages); for (i = 0; i < npages; i++) { fname = sarrayGetString(sa, i, L_NOCOPY); if (!strcmp(fname, "")) continue; boxa = NULL; if (baa) { boxa = boxaaGetBoxa(baa, i, L_CLONE); nboxes = boxaGetCount(boxa); if (nboxes == 0) boxaDestroy(&boxa); } ret = convertToPdfDataSegmented(fname, res, type, thresh, boxa, quality, scalefactor, title, &imdata, &imbytes); boxaDestroy(&boxa); /* safe; in case nboxes > 0 */ if (ret) { L_ERROR("pdf encoding failed for %s\n", procName, fname); continue; } ba = l_byteaInitFromMem(imdata, imbytes); if (imdata) FREE(imdata); ptraAdd(pa_data, ba); } sarrayDestroy(&sa); ptraGetActualCount(pa_data, &npages); if (npages == 0) { L_ERROR("no pdf files made\n", procName); ptraDestroy(&pa_data, FALSE, FALSE); return 1; } /* Concatenate */ ret = ptraConcatenatePdfToData(pa_data, NULL, &data, &databytes); /* Clean up */ ptraGetActualCount(pa_data, &npages); /* recalculate in case it changes */ for (i = 0; i < npages; i++) { ba = (L_BYTEA *)ptraRemove(pa_data, i, L_NO_COMPACTION); l_byteaDestroy(&ba); } ptraDestroy(&pa_data, FALSE, FALSE); if (ret) { if (data) FREE(data); return ERROR_INT("pdf data not made", procName, 1); } ret = l_binaryWrite(fileout, "w", data, databytes); FREE(data); if (ret) L_ERROR("pdf data not written to file\n", procName); return ret; } /*! * convertNumberedMasksToBoxaa() * * Input: directory name (containing mask images) * substr ( substring filter on filenames; can be NULL) * numpre (number of characters in name before number) * numpost (number of characters in name after number, up * to a dot before an extension) * including an extension and the dot separator) * Return: boxaa of mask regions, or null on error * * Notes: * (1) This is conveniently used to generate the input boxaa * for convertSegmentedFilesToPdf(). It guarantees that the * boxa will be aligned with the page images, even if some * of the boxa are empty. */ BOXAA * convertNumberedMasksToBoxaa(const char *dirname, const char *substr, l_int32 numpre, l_int32 numpost) { char *fname; l_int32 i, n; BOXA *boxa; BOXAA *baa; PIX *pix; SARRAY *sa; PROCNAME("convertNumberedMasksToBoxaa"); if (!dirname) return (BOXAA *)ERROR_PTR("dirname not defined", procName, NULL); if ((sa = getNumberedPathnamesInDirectory(dirname, substr, numpre, numpost, 10000)) == NULL) return (BOXAA *)ERROR_PTR("sa not made", procName, NULL); /* Generate and save all the encoded pdf strings */ n = sarrayGetCount(sa); baa = boxaaCreate(n); boxa = boxaCreate(1); boxaaInitFull(baa, boxa); boxaDestroy(&boxa); for (i = 0; i < n; i++) { fname = sarrayGetString(sa, i, L_NOCOPY); if (!strcmp(fname, "")) continue; if ((pix = pixRead(fname)) == NULL) { L_WARNING("invalid image on page %d\n", procName, i); continue; } boxa = pixConnComp(pix, NULL, 8); boxaaReplaceBoxa(baa, i, boxa); pixDestroy(&pix); } sarrayDestroy(&sa); return baa; } /*---------------------------------------------------------------------* * Segmented single page, multi-image converters * *---------------------------------------------------------------------*/ /*! * convertToPdfSegmented() * * Input: filein (input image file -- any format) * res (input image resolution; typ. 300 ppi; use 0 for default) * type (compression type for non-image regions; the * image regions are always compressed with L_JPEG_ENCODE) * thresh (used for converting gray --> 1 bpp with L_G4_ENCODE) * boxa ( of image regions; can be null) * quality (used for jpeg image regions; 0 for default) * scalefactor (used for jpeg regions; must be <= 1.0) * title ( pdf title; typically taken from the * input file for the pix) * fileout (output pdf file) * Return: 0 if OK, 1 on error * * Notes: * (1) If there are no image regions, set @boxa == NULL; * @quality and @scalefactor are ignored. * (2) Typically, @scalefactor is < 1.0, because the image regions * can be rendered at a lower resolution (for better compression) * than the text regions. If @scalefactor == 0, we use 1.0. * If the input image is 1 bpp and scalefactor < 1.0, we * use scaleToGray() to downsample the image regions to gray * before compressing them. * (3) If the compression type for non-image regions is L_G4_ENCODE * and bpp > 1, the image is upscaled 2x and thresholded * to 1 bpp. That is the only situation where @thresh is used. * (4) The parameter @quality is only used for image regions. * If @type == L_JPEG_ENCODE, default jpeg quality (75) is * used for the non-image regions. * (5) Processing matrix for non-image regions. * * Input G4 JPEG FLATE * ----------|--------------------------------------------------- * 1 bpp | 1x, 1 bpp 1x flate, 1 bpp 1x, 1 bpp * | * cmap | 2x, 1 bpp 1x flate, cmap 1x, cmap * | * 2,4 bpp | 2x, 1 bpp 1x flate 1x, 2,4 bpp * no cmap | 2,4 bpp * | * 8,32 bpp | 2x, 1 bpp 1x (jpeg) 1x, 8,32 bpp * no cmap | 8,32 bpp * * Summary: * (a) if G4 is requested, G4 is used, with 2x upscaling * for all cases except 1 bpp. * (b) if JPEG is requested, use flate encoding for all cases * except 8 bpp without cmap and 32 bpp (rgb). * (c) if FLATE is requested, use flate with no transformation * of the raster data. * (6) Calling options/sequence for these functions: * file --> file (convertToPdfSegmented) * pix --> file (pixConvertToPdfSegmented) * pix --> data (pixConvertToPdfDataSegmented) * file --> data (convertToPdfDataSegmented) * pix --> data (pixConvertToPdfDataSegmented) */ l_int32 convertToPdfSegmented(const char *filein, l_int32 res, l_int32 type, l_int32 thresh, BOXA *boxa, l_int32 quality, l_float32 scalefactor, const char *title, const char *fileout) { l_int32 ret; PIX *pixs; PROCNAME("convertToPdfSegmented"); if (!filein) return ERROR_INT("filein not defined", procName, 1); if (!fileout) return ERROR_INT("fileout not defined", procName, 1); if (type != L_G4_ENCODE && type != L_JPEG_ENCODE && type != L_FLATE_ENCODE) return ERROR_INT("invalid conversion type", procName, 1); if (boxa && scalefactor > 1.0) { L_WARNING("setting scalefactor to 1.0\n", procName); scalefactor = 1.0; } if ((pixs = pixRead(filein)) == NULL) return ERROR_INT("pixs not made", procName, 1); ret = pixConvertToPdfSegmented(pixs, res, type, thresh, boxa, quality, scalefactor, (title) ? title : filein, fileout); pixDestroy(&pixs); return ret; } /*! * pixConvertToPdfSegmented() * * Input: pixs (any depth, cmap OK) * res (input image resolution; typ. 300 ppi; use 0 for default) * type (compression type for non-image regions; the * image regions are always compressed with L_JPEG_ENCODE) * thresh (used for converting gray --> 1 bpp with L_G4_ENCODE) * boxa ( of image regions; can be null) * quality (used for jpeg image regions; 0 for default) * scalefactor (used for jpeg regions; must be <= 1.0) * title ( pdf title; typically taken from the * input file for the pix) * fileout (output pdf file) * Return: 0 if OK, 1 on error * * Notes: * (1) See convertToPdfSegmented() for details. */ l_int32 pixConvertToPdfSegmented(PIX *pixs, l_int32 res, l_int32 type, l_int32 thresh, BOXA *boxa, l_int32 quality, l_float32 scalefactor, const char *title, const char *fileout) { l_uint8 *data; l_int32 ret; size_t nbytes; PROCNAME("pixConvertToPdfSegmented"); if (!pixs) return ERROR_INT("pixs not defined", procName, 1); if (!fileout) return ERROR_INT("fileout not defined", procName, 1); if (type != L_G4_ENCODE && type != L_JPEG_ENCODE && type != L_FLATE_ENCODE) return ERROR_INT("invalid conversion type", procName, 1); if (boxa && scalefactor > 1.0) { L_WARNING("setting scalefactor to 1.0\n", procName); scalefactor = 1.0; } ret = pixConvertToPdfDataSegmented(pixs, res, type, thresh, boxa, quality, scalefactor, title, &data, &nbytes); if (ret) return ERROR_INT("pdf generation failure", procName, 1); ret = l_binaryWrite(fileout, "w", data, nbytes); if (data) FREE(data); return ret; } /*! * convertToPdfDataSegmented() * * Input: filein (input image file -- any format) * res (input image resolution; typ. 300 ppi; use 0 for default) * type (compression type for non-image regions; the * image regions are always compressed with L_JPEG_ENCODE) * thresh (used for converting gray --> 1 bpp with L_G4_ENCODE) * boxa ( image regions; can be null) * quality (used for jpeg image regions; 0 for default) * scalefactor (used for jpeg regions; must be <= 1.0) * title ( pdf title; if null, uses filein) * &data ( pdf data in memory) * &nbytes ( number of bytes in pdf data) * Return: 0 if OK, 1 on error * * Notes: * (1) If there are no image regions, set @boxa == NULL; * @quality and @scalefactor are ignored. * (2) Typically, @scalefactor is < 1.0. The image regions are */ l_int32 convertToPdfDataSegmented(const char *filein, l_int32 res, l_int32 type, l_int32 thresh, BOXA *boxa, l_int32 quality, l_float32 scalefactor, const char *title, l_uint8 **pdata, size_t *pnbytes) { l_int32 ret; PIX *pixs; PROCNAME("convertToPdfDataSegmented"); if (!pdata) return ERROR_INT("&data not defined", procName, 1); *pdata = NULL; if (!pnbytes) return ERROR_INT("&nbytes not defined", procName, 1); *pnbytes = 0; if (!filein) return ERROR_INT("filein not defined", procName, 1); if (type != L_G4_ENCODE && type != L_JPEG_ENCODE && type != L_FLATE_ENCODE) return ERROR_INT("invalid conversion type", procName, 1); if (boxa && scalefactor > 1.0) { L_WARNING("setting scalefactor to 1.0\n", procName); scalefactor = 1.0; } if ((pixs = pixRead(filein)) == NULL) return ERROR_INT("pixs not made", procName, 1); ret = pixConvertToPdfDataSegmented(pixs, res, type, thresh, boxa, quality, scalefactor, (title) ? title : filein, pdata, pnbytes); pixDestroy(&pixs); return ret; } /*! * pixConvertToPdfDataSegmented() * * Input: pixs (any depth, cmap OK) * res (input image resolution; typ. 300 ppi; use 0 for default) * type (compression type for non-image regions; the * image regions are always compressed with L_JPEG_ENCODE) * thresh (used for converting gray --> 1 bpp with L_G4_ENCODE) * boxa ( of image regions; can be null) * quality (used for jpeg image regions; 0 for default) * scalefactor (used for jpeg regions; must be <= 1.0) * title ( pdf title; typically taken from the * input file for the pix) * &data ( pdf data in memory) * &nbytes ( number of bytes in pdf data) * Return: 0 if OK, 1 on error * * Notes: * (1) See convertToPdfSegmented() for details. */ l_int32 pixConvertToPdfDataSegmented(PIX *pixs, l_int32 res, l_int32 type, l_int32 thresh, BOXA *boxa, l_int32 quality, l_float32 scalefactor, const char *title, l_uint8 **pdata, size_t *pnbytes) { l_int32 i, nbox, seq, bx, by, bw, bh, upscale; l_float32 scale; BOX *box, *boxc, *box2; PIX *pix, *pixt1, *pixt2, *pixt3, *pixt4, *pixt5, *pixt6; PIXCMAP *cmap; L_PDF_DATA *lpd; PROCNAME("pixConvertToPdfDataSegmented"); if (!pdata) return ERROR_INT("&data not defined", procName, 1); *pdata = NULL; if (!pnbytes) return ERROR_INT("&nbytes not defined", procName, 1); *pnbytes = 0; if (!pixs) return ERROR_INT("pixs not defined", procName, 1); if (type != L_G4_ENCODE && type != L_JPEG_ENCODE && type != L_FLATE_ENCODE) return ERROR_INT("invalid conversion type", procName, 1); if (boxa && (scalefactor <= 0.0 || scalefactor > 1.0)) { L_WARNING("setting scalefactor to 1.0\n", procName); scalefactor = 1.0; } /* Adjust scalefactor so that the product with res gives an integer */ if (res <= 0) res = DEFAULT_INPUT_RES; scale = (l_float32)((l_int32)(scalefactor * res + 0.5)) / (l_float32)res; cmap = pixGetColormap(pixs); /* Simple case: single image to be encoded */ if (!boxa || boxaGetCount(boxa) == 0) { if (pixGetDepth(pixs) > 1 && type == L_G4_ENCODE) { if (cmap) pixt1 = pixRemoveColormap(pixs, REMOVE_CMAP_TO_GRAYSCALE); else pixt1 = pixConvertTo8(pixs, FALSE); pixt2 = pixScaleGray2xLIThresh(pixt1, thresh); pixConvertToPdfData(pixt2, type, quality, pdata, pnbytes, 0, 0, 2 * res, title, NULL, 0); pixDestroy(&pixt1); pixDestroy(&pixt2); } else { pixConvertToPdfData(pixs, type, quality, pdata, pnbytes, 0, 0, res, title, NULL, 0); } return 0; } /* Multiple images to be encoded. If @type == L_G4_ENCODE, * jpeg encode a version of pixs that is blanked in the non-image * regions, and paint the scaled non-image part onto it through a mask. * Otherwise, we must put the non-image part down first and * then render all the image regions separately on top of it, * at their own resolution. */ pixt1 = pixSetBlackOrWhiteBoxa(pixs, boxa, L_SET_WHITE); /* non-image */ nbox = boxaGetCount(boxa); if (type == L_G4_ENCODE) { pixt2 = pixCreateTemplate(pixs); /* only image regions */ pixSetBlackOrWhite(pixt2, L_SET_WHITE); for (i = 0; i < nbox; i++) { box = boxaGetBox(boxa, i, L_CLONE); pix = pixClipRectangle(pixs, box, &boxc); boxGetGeometry(boxc, &bx, &by, &bw, &bh); pixRasterop(pixt2, bx, by, bw, bh, PIX_SRC, pix, 0, 0); pixDestroy(&pix); boxDestroy(&box); boxDestroy(&boxc); } pixt3 = pixRemoveColormap(pixt2, REMOVE_CMAP_BASED_ON_SRC); if (pixGetDepth(pixt3) == 1) pixt4 = pixScaleToGray(pixt3, scale); else pixt4 = pixScale(pixt3, scale, scale); pixConvertToPdfData(pixt4, L_JPEG_ENCODE, quality, pdata, pnbytes, 0, 0, (l_int32)(scale * res), title, &lpd, L_FIRST_IMAGE); if (pixGetDepth(pixt1) == 1) { pixt5 = pixClone(pixt1); upscale = 1; } else { pixt6 = pixConvertTo8(pixt1, 0); pixt5 = pixScaleGray2xLIThresh(pixt6, thresh); pixDestroy(&pixt6); upscale = 2; } pixConvertToPdfData(pixt5, L_G4_ENCODE, quality, pdata, pnbytes, 0, 0, upscale * res, title, &lpd, L_LAST_IMAGE); pixDestroy(&pixt2); pixDestroy(&pixt3); pixDestroy(&pixt4); pixDestroy(&pixt5); } else { /* Put the non-image part down first. This is the full size of the page, so we can use it to find the page height in pixels, which is required for determining the LL corner of the image relative to the LL corner of the page. */ pixConvertToPdfData(pixt1, type, quality, pdata, pnbytes, 0, 0, res, title, &lpd, L_FIRST_IMAGE); for (i = 0; i < nbox; i++) { box = boxaGetBox(boxa, i, L_CLONE); pixt2 = pixClipRectangle(pixs, box, &boxc); pixt3 = pixRemoveColormap(pixt2, REMOVE_CMAP_BASED_ON_SRC); if (pixGetDepth(pixt3) == 1) pixt4 = pixScaleToGray(pixt3, scale); else pixt4 = pixScale(pixt3, scale, scale); box2 = boxTransform(boxc, 0, 0, scale, scale); boxGetGeometry(box2, &bx, &by, NULL, &bh); seq = (i == nbox - 1) ? L_LAST_IMAGE : L_NEXT_IMAGE; pixConvertToPdfData(pixt4, L_JPEG_ENCODE, quality, pdata, pnbytes, bx, by, (l_int32)(scale * res), title, &lpd, seq); pixDestroy(&pixt2); pixDestroy(&pixt3); pixDestroy(&pixt4); boxDestroy(&box); boxDestroy(&boxc); boxDestroy(&box2); } } pixDestroy(&pixt1); return 0; } /*---------------------------------------------------------------------* * Helper functions for generating the output pdf string * *---------------------------------------------------------------------*/ /*! * l_generatePdf() * * Input: &data ( pdf array) * &nbytes ( number of bytes in pdf array) * lpd (all the required input image data) * Return: 0 if OK, 1 on error * * Notes: * (1) On error, no data is returned. * (2) The objects are: * 1: Catalog * 2: Info * 3: Pages * 4: Page * 5: Contents (rendering command) * 6 to 6+n-1: n XObjects * 6+n to 6+n+m-1: m colormaps */ static l_int32 l_generatePdf(l_uint8 **pdata, size_t *pnbytes, L_PDF_DATA *lpd) { PROCNAME("l_generatePdf"); if (!pdata) return ERROR_INT("&data not defined", procName, 1); *pdata = NULL; if (!pnbytes) return ERROR_INT("&nbytes not defined", procName, 1); *pnbytes = 0; if (!lpd) return ERROR_INT("lpd not defined", procName, 1); generateFixedStringsPdf(lpd); generateMediaboxPdf(lpd); generatePageStringPdf(lpd); generateContentStringPdf(lpd); generatePreXStringsPdf(lpd); generateColormapStringsPdf(lpd); generateTrailerPdf(lpd); return generateOutputDataPdf(pdata, pnbytes, lpd); } static void generateFixedStringsPdf(L_PDF_DATA *lpd) { char buf[L_SMALLBUF]; char *version, *datestr; SARRAY *sa; /* Accumulate data for the header and objects 1-3 */ lpd->id = stringNew("%PDF-1.5\n"); l_dnaAddNumber(lpd->objsize, strlen(lpd->id)); lpd->obj1 = stringNew("1 0 obj\n" "<<\n" "/Type /Catalog\n" "/Pages 3 0 R\n" ">>\n" "endobj\n"); l_dnaAddNumber(lpd->objsize, strlen(lpd->obj1)); sa = sarrayCreate(0); sarrayAddString(sa, (char *)"2 0 obj\n" "<<\n", L_COPY); if (var_WRITE_DATE_AND_VERSION) { datestr = l_getFormattedDate(); snprintf(buf, sizeof(buf), "/CreationDate (D:%s)\n", datestr); sarrayAddString(sa, (char *)buf, L_COPY); FREE(datestr); version = getLeptonicaVersion(); snprintf(buf, sizeof(buf), "/Producer (leptonica: %s)\n", version); FREE(version); } else { snprintf(buf, sizeof(buf), "/Producer (leptonica)\n"); } sarrayAddString(sa, (char *)buf, L_COPY); if (lpd->title) { snprintf(buf, sizeof(buf), "/Title (%s)\n", lpd->title); sarrayAddString(sa, (char *)buf, L_COPY); } sarrayAddString(sa, (char *)">>\n" "endobj\n", L_COPY); lpd->obj2 = sarrayToString(sa, 0); l_dnaAddNumber(lpd->objsize, strlen(lpd->obj2)); sarrayDestroy(&sa); lpd->obj3 = stringNew("3 0 obj\n" "<<\n" "/Type /Pages\n" "/Kids [ 4 0 R ]\n" "/Count 1\n" ">>\n"); l_dnaAddNumber(lpd->objsize, strlen(lpd->obj3)); /* Do the post-datastream string */ lpd->poststream = stringNew("\n" "endstream\n" "endobj\n"); return; } static void generateMediaboxPdf(L_PDF_DATA *lpd) { l_int32 i; l_float32 xpt, ypt, wpt, hpt, maxx, maxy; /* First get the full extent of all the images. * This is the mediabox, in pts. */ maxx = maxy = 0; for (i = 0; i < lpd->n; i++) { ptaGetPt(lpd->xy, i, &xpt, &ypt); ptaGetPt(lpd->wh, i, &wpt, &hpt); maxx = L_MAX(maxx, xpt + wpt); maxy = L_MAX(maxy, ypt + hpt); } lpd->mediabox = boxCreate(0, 0, (l_int32)(maxx + 0.5), (l_int32)(maxy + 0.5)); /* ypt is in standard image coordinates: the location of * the UL image corner with respect to the UL media box corner. * Rewrite each ypt for PostScript coordinates: the location of * the LL image corner with respect to the LL media box corner. */ for (i = 0; i < lpd->n; i++) { ptaGetPt(lpd->xy, i, &xpt, &ypt); ptaGetPt(lpd->wh, i, &wpt, &hpt); ptaSetPt(lpd->xy, i, xpt, maxy - ypt - hpt); } return; } static l_int32 generatePageStringPdf(L_PDF_DATA *lpd) { char *buf; char *xstr; l_int32 bufsize, i, wpt, hpt; SARRAY *sa; PROCNAME("generatePageStringPdf"); /* Allocate 1000 bytes for the boilerplate text, and * 50 bytes for each reference to an image in the * ProcSet array. */ bufsize = 1000 + 50 * lpd->n; if ((buf = (char *)CALLOC(bufsize, sizeof(char))) == NULL) return ERROR_INT("calloc fail for buf", procName, 1); boxGetGeometry(lpd->mediabox, NULL, NULL, &wpt, &hpt); sa = sarrayCreate(lpd->n); for (i = 0; i < lpd->n; i++) { snprintf(buf, bufsize, "/Im%d %d 0 R ", i + 1, 6 + i); sarrayAddString(sa, buf, L_COPY); } if ((xstr = sarrayToString(sa, 0)) == NULL) return ERROR_INT("xstr not found", procName, 1); sarrayDestroy(&sa); snprintf(buf, bufsize, "4 0 obj\n" "<<\n" "/Type /Page\n" "/Parent 3 0 R\n" "/MediaBox [%d %d %d %d]\n" "/Contents 5 0 R\n" "/Resources\n" "<<\n" "/XObject << %s >>\n" "/ProcSet [ /ImageB /ImageI /ImageC ]\n" ">>\n" ">>\n" "endobj\n", 0, 0, wpt, hpt, xstr); lpd->obj4 = stringNew(buf); l_dnaAddNumber(lpd->objsize, strlen(lpd->obj4)); sarrayDestroy(&sa); FREE(buf); FREE(xstr); return 0; } static l_int32 generateContentStringPdf(L_PDF_DATA *lpd) { char *buf; char *cstr; l_int32 i, bufsize; l_float32 xpt, ypt, wpt, hpt; SARRAY *sa; PROCNAME("generateContentStringPdf"); bufsize = 1000 + 200 * lpd->n; if ((buf = (char *)CALLOC(bufsize, sizeof(char))) == NULL) return ERROR_INT("calloc fail for buf", procName, 1); sa = sarrayCreate(lpd->n); for (i = 0; i < lpd->n; i++) { ptaGetPt(lpd->xy, i, &xpt, &ypt); ptaGetPt(lpd->wh, i, &wpt, &hpt); snprintf(buf, bufsize, "q %.4f %.4f %.4f %.4f %.4f %.4f cm /Im%d Do Q\n", wpt, 0.0, 0.0, hpt, xpt, ypt, i + 1); sarrayAddString(sa, buf, L_COPY); } if ((cstr = sarrayToString(sa, 0)) == NULL) return ERROR_INT("cstr not found", procName, 1); sarrayDestroy(&sa); snprintf(buf, bufsize, "5 0 obj\n" "<< /Length %d >>\n" "stream\n" "%s" "endstream\n" "endobj\n", (l_int32)strlen(cstr), cstr); lpd->obj5 = stringNew(buf); l_dnaAddNumber(lpd->objsize, strlen(lpd->obj5)); sarrayDestroy(&sa); FREE(buf); FREE(cstr); return 0; } static l_int32 generatePreXStringsPdf(L_PDF_DATA *lpd) { char buff[256]; char buf[L_BIGBUF]; char *cstr, *bstr, *fstr, *xstr; l_int32 i, cmindex; L_COMP_DATA *cid; SARRAY *sa; PROCNAME("generatePreXStringsPdf"); sa = lpd->saprex; cmindex = 6 + lpd->n; /* starting value */ for (i = 0; i < lpd->n; i++) { if ((cid = pdfdataGetCid(lpd, i)) == NULL) return ERROR_INT("cid not found", procName, 1); if (cid->type == L_G4_ENCODE) { if (var_WRITE_G4_IMAGE_MASK) { cstr = stringNew("/ImageMask true\n" "/ColorSpace /DeviceGray"); } else { cstr = stringNew("/ColorSpace /DeviceGray"); } bstr = stringNew("/BitsPerComponent 1\n" "/Interpolate true"); snprintf(buff, sizeof(buff), "/Filter /CCITTFaxDecode\n" "/DecodeParms\n" "<<\n" "/K -1\n" "/Columns %d\n" ">>", cid->w); fstr = stringNew(buff); } else if (cid->type == L_JPEG_ENCODE) { if (cid->spp == 1) cstr = stringNew("/ColorSpace /DeviceGray"); else if (cid->spp == 3) cstr = stringNew("/ColorSpace /DeviceRGB"); else L_ERROR("spp!= 1 && spp != 3\n", procName); bstr = stringNew("/BitsPerComponent 8"); fstr = stringNew("/Filter /DCTDecode"); } else if (cid->type == L_JP2K_ENCODE) { if (cid->spp == 1) cstr = stringNew("/ColorSpace /DeviceGray"); else if (cid->spp == 3) cstr = stringNew("/ColorSpace /DeviceRGB"); else L_ERROR("spp!= 1 && spp != 3\n", procName); bstr = stringNew("/BitsPerComponent 8"); fstr = stringNew("/Filter /JPXDecode"); } else { /* type == L_FLATE_ENCODE */ if (cid->ncolors > 0) { /* cmapped */ snprintf(buff, sizeof(buff), "/ColorSpace %d 0 R", cmindex++); cstr = stringNew(buff); } else { if (cid->spp == 1 && cid->bps == 1) cstr = stringNew("/ColorSpace /DeviceGray\n" "/Decode [1 0]"); else if (cid->spp == 1) /* 8 bpp */ cstr = stringNew("/ColorSpace /DeviceGray"); else if (cid->spp == 3) cstr = stringNew("/ColorSpace /DeviceRGB"); else L_ERROR("unknown colorspace\n", procName); } snprintf(buff, sizeof(buff), "/BitsPerComponent %d", cid->bps); bstr = stringNew(buff); fstr = stringNew("/Filter /FlateDecode"); } snprintf(buf, sizeof(buf), "%d 0 obj\n" "<<\n" "/Length %lu\n" "/Subtype /Image\n" "%s\n" /* colorspace */ "/Width %d\n" "/Height %d\n" "%s\n" /* bits/component */ "%s\n" /* filter */ ">>\n" "stream\n", 6 + i, (unsigned long)cid->nbytescomp, cstr, cid->w, cid->h, bstr, fstr); xstr = stringNew(buf); sarrayAddString(sa, xstr, L_INSERT); l_dnaAddNumber(lpd->objsize, strlen(xstr) + cid->nbytescomp + strlen(lpd->poststream)); FREE(cstr); FREE(bstr); FREE(fstr); } return 0; } static l_int32 generateColormapStringsPdf(L_PDF_DATA *lpd) { char buf[L_BIGBUF]; char *cmstr; l_int32 i, cmindex, ncmap; L_COMP_DATA *cid; SARRAY *sa; PROCNAME("generateColormapStringsPdf"); /* In our canonical format, we have 5 objects, followed * by n XObjects, followed by m colormaps, so the index of * the first colormap object is 6 + n. */ sa = lpd->sacmap; cmindex = 6 + lpd->n; /* starting value */ ncmap = 0; for (i = 0; i < lpd->n; i++) { if ((cid = pdfdataGetCid(lpd, i)) == NULL) return ERROR_INT("cid not found", procName, 1); if (cid->ncolors == 0) continue; ncmap++; snprintf(buf, sizeof(buf), "%d 0 obj\n" "[ /Indexed /DeviceRGB\n" "%d\n" "%s\n" "]\n" "endobj\n", cmindex, cid->ncolors - 1, cid->cmapdatahex); cmindex++; cmstr = stringNew(buf); l_dnaAddNumber(lpd->objsize, strlen(cmstr)); sarrayAddString(sa, cmstr, L_INSERT); } lpd->ncmap = ncmap; return 0; } static void generateTrailerPdf(L_PDF_DATA *lpd) { l_int32 i, n, size, linestart; L_DNA *daloc, *dasize; /* Let nobj be the number of numbered objects. These numbered * objects are indexed by their pdf number in arrays naloc[] * and nasize[]. The 0th object is the 9 byte header. Then * the number of objects in nasize, which includes the header, * is n = nobj + 1. The array naloc[] has n + 1 elements, * because it includes as the last element the starting * location of xref. The indexing of these objects, their * starting locations and sizes are: * * Object number Starting location Size * ------------- ----------------- -------------- * 0 daloc[0] = 0 dasize[0] = 9 * 1 daloc[1] = 9 dasize[1] = 49 * n daloc[n] dasize[n] * xref daloc[n+1] * * We first generate daloc. */ dasize = lpd->objsize; daloc = lpd->objloc; linestart = 0; l_dnaAddNumber(daloc, linestart); /* header */ n = l_dnaGetCount(dasize); for (i = 0; i < n; i++) { l_dnaGetIValue(dasize, i, &size); linestart += size; l_dnaAddNumber(daloc, linestart); } l_dnaGetIValue(daloc, n, &lpd->xrefloc); /* save it */ /* Now make the actual trailer string */ lpd->trailer = makeTrailerStringPdf(daloc); } static char * makeTrailerStringPdf(L_DNA *daloc) { char *outstr; char buf[L_BIGBUF]; l_int32 i, n, linestart, xrefloc; SARRAY *sa; PROCNAME("makeTrailerStringPdf"); if (!daloc) return (char *)ERROR_PTR("daloc not defined", procName, NULL); n = l_dnaGetCount(daloc) - 1; /* numbered objects + 1 (yes, +1) */ sa = sarrayCreate(0); snprintf(buf, sizeof(buf), "xref\n" "0 %d\n" "0000000000 65535 f \n", n); sarrayAddString(sa, (char *)buf, L_COPY); for (i = 1; i < n; i++) { l_dnaGetIValue(daloc, i, &linestart); snprintf(buf, sizeof(buf), "%010d 00000 n \n", linestart); sarrayAddString(sa, (char *)buf, L_COPY); } l_dnaGetIValue(daloc, n, &xrefloc); snprintf(buf, sizeof(buf), "trailer\n" "<<\n" "/Size %d\n" "/Root 1 0 R\n" "/Info 2 0 R\n" ">>\n" "startxref\n" "%d\n" "%%%%EOF\n", n, xrefloc); sarrayAddString(sa, (char *)buf, L_COPY); outstr = sarrayToString(sa, 0); sarrayDestroy(&sa); return outstr; } /*! * generateOutputDataPdf() * * Input: &data ( pdf data array) * &nbytes ( size of pdf data array) * lpd (input data used to make pdf) * Return: 0 if OK, 1 on error * * Notes: * (1) Only called from l_generatePdf(). On error, no data is returned. */ static l_int32 generateOutputDataPdf(l_uint8 **pdata, size_t *pnbytes, L_PDF_DATA *lpd) { char *str; l_uint8 *data; l_int32 nimages, i, len; l_int32 *sizes, *locs; size_t nbytes; L_COMP_DATA *cid; PROCNAME("generateOutputDataPdf"); if (!pdata) return ERROR_INT("&data not defined", procName, 1); *pdata = NULL; if (!pnbytes) return ERROR_INT("&nbytes not defined", procName, 1); nbytes = lpd->xrefloc + strlen(lpd->trailer); *pnbytes = nbytes; if ((data = (l_uint8 *)CALLOC(nbytes, sizeof(l_uint8))) == NULL) return ERROR_INT("calloc fail for data", procName, 1); *pdata = data; sizes = l_dnaGetIArray(lpd->objsize); locs = l_dnaGetIArray(lpd->objloc); memcpy((char *)data, lpd->id, sizes[0]); memcpy((char *)(data + locs[1]), lpd->obj1, sizes[1]); memcpy((char *)(data + locs[2]), lpd->obj2, sizes[2]); memcpy((char *)(data + locs[3]), lpd->obj3, sizes[3]); memcpy((char *)(data + locs[4]), lpd->obj4, sizes[4]); memcpy((char *)(data + locs[5]), lpd->obj5, sizes[5]); /* Each image has 3 parts: variable preamble, the compressed * data stream, and the fixed poststream. */ nimages = lpd->n; for (i = 0; i < nimages; i++) { if ((cid = pdfdataGetCid(lpd, i)) == NULL) /* this should not happen */ return ERROR_INT("cid not found", procName, 1); str = sarrayGetString(lpd->saprex, i, L_NOCOPY); len = strlen(str); memcpy((char *)(data + locs[6 + i]), str, len); memcpy((char *)(data + locs[6 + i] + len), (char *)cid->datacomp, cid->nbytescomp); memcpy((char *)(data + locs[6 + i] + len + cid->nbytescomp), lpd->poststream, strlen(lpd->poststream)); } /* Each colormap is simply a stored string */ for (i = 0; i < lpd->ncmap; i++) { str = sarrayGetString(lpd->sacmap, i, L_NOCOPY); memcpy((char *)(data + locs[6 + nimages + i]), str, strlen(str)); } /* And finally the trailer */ memcpy((char *)(data + lpd->xrefloc), lpd->trailer, strlen(lpd->trailer)); FREE(sizes); FREE(locs); return 0; } /*---------------------------------------------------------------------* * Multi-page concatenation * *---------------------------------------------------------------------*/ /*! * concatenatePdf() * * Input: directory name (containing single-page pdf files) * substr ( substring filter on filenames; can be NULL) * fileout (concatenated pdf file) * Return: 0 if OK, 1 on error * * Notes: * (1) This only works with leptonica-formatted single-page pdf files. * (2) If @substr is not NULL, only filenames that contain * the substring can be returned. If @substr == NULL, * none of the filenames are filtered out. * (3) The files in the directory, after optional filtering by * the substring, are lexically sorted in increasing order * before concatenation. */ l_int32 concatenatePdf(const char *dirname, const char *substr, const char *fileout) { l_int32 ret; SARRAY *sa; PROCNAME("concatenatePdf"); if (!dirname) return ERROR_INT("dirname not defined", procName, 1); if (!fileout) return ERROR_INT("fileout not defined", procName, 1); if ((sa = getSortedPathnamesInDirectory(dirname, substr, 0, 0)) == NULL) return ERROR_INT("sa not made", procName, 1); ret = saConcatenatePdf(sa, fileout); sarrayDestroy(&sa); return ret; } /*! * saConcatenatePdf() * * Input: sarray (of pathnames for single-page pdf files) * fileout (concatenated pdf file) * Return: 0 if OK, 1 on error * * Notes: * (1) This only works with leptonica-formatted single-page pdf files. */ l_int32 saConcatenatePdf(SARRAY *sa, const char *fileout) { l_uint8 *data; l_int32 ret; size_t nbytes; PROCNAME("saConcatenatePdf"); if (!sa) return ERROR_INT("sa not defined", procName, 1); if (!fileout) return ERROR_INT("fileout not defined", procName, 1); ret = saConcatenatePdfToData(sa, &data, &nbytes); if (ret) return ERROR_INT("pdf data not made", procName, 1); ret = l_binaryWrite(fileout, "w", data, nbytes); FREE(data); return ret; } /*! * ptraConcatenatePdf() * * Input: ptra (array of pdf strings, each for a single-page pdf file) * fileout (concatenated pdf file) * Return: 0 if OK, 1 on error * * Notes: * (1) This only works with leptonica-formatted single-page pdf files. */ l_int32 ptraConcatenatePdf(L_PTRA *pa, const char *fileout) { l_uint8 *data; l_int32 ret; size_t nbytes; PROCNAME("ptraConcatenatePdf"); if (!pa) return ERROR_INT("pa not defined", procName, 1); if (!fileout) return ERROR_INT("fileout not defined", procName, 1); ret = ptraConcatenatePdfToData(pa, NULL, &data, &nbytes); if (ret) return ERROR_INT("pdf data not made", procName, 1); ret = l_binaryWrite(fileout, "w", data, nbytes); FREE(data); return ret; } /*! * concatenatePdfToData() * * Input: directory name (containing single-page pdf files) * substr ( substring filter on filenames; can be NULL) * &data ( concatenated pdf data in memory) * &nbytes ( number of bytes in pdf data) * Return: 0 if OK, 1 on error * * Notes: * (1) This only works with leptonica-formatted single-page pdf files. * (2) If @substr is not NULL, only filenames that contain * the substring can be returned. If @substr == NULL, * none of the filenames are filtered out. * (3) The files in the directory, after optional filtering by * the substring, are lexically sorted in increasing order * before concatenation. */ l_int32 concatenatePdfToData(const char *dirname, const char *substr, l_uint8 **pdata, size_t *pnbytes) { l_int32 ret; SARRAY *sa; PROCNAME("concatenatePdfToData"); if (!pdata) return ERROR_INT("&data not defined", procName, 1); *pdata = NULL; if (!pnbytes) return ERROR_INT("&nbytes not defined", procName, 1); *pnbytes = 0; if (!dirname) return ERROR_INT("dirname not defined", procName, 1); if ((sa = getSortedPathnamesInDirectory(dirname, substr, 0, 0)) == NULL) return ERROR_INT("sa not made", procName, 1); ret = saConcatenatePdfToData(sa, pdata, pnbytes); sarrayDestroy(&sa); return ret; } /*! * saConcatenatePdfToData() * * Input: sarray (of pathnames for single-page pdf files) * &data ( concatenated pdf data in memory) * &nbytes ( number of bytes in pdf data) * Return: 0 if OK, 1 on error * * Notes: * (1) This only works with leptonica-formatted single-page pdf files. */ l_int32 saConcatenatePdfToData(SARRAY *sa, l_uint8 **pdata, size_t *pnbytes) { char *fname; l_int32 i, npages, ret; L_BYTEA *bas; L_PTRA *pa_data; /* input pdf data for each page */ PROCNAME("saConcatenatePdfToData"); if (!pdata) return ERROR_INT("&data not defined", procName, 1); *pdata = NULL; if (!pnbytes) return ERROR_INT("&nbytes not defined", procName, 1); *pnbytes = 0; if (!sa) return ERROR_INT("sa not defined", procName, 1); /* Read the pdf files into memory */ if ((npages = sarrayGetCount(sa)) == 0) return ERROR_INT("no filenames found", procName, 1); pa_data = ptraCreate(npages); for (i = 0; i < npages; i++) { fname = sarrayGetString(sa, i, L_NOCOPY); bas = l_byteaInitFromFile(fname); ptraAdd(pa_data, bas); } ret = ptraConcatenatePdfToData(pa_data, sa, pdata, pnbytes); /* Cleanup: some pages could have been removed */ ptraGetActualCount(pa_data, &npages); for (i = 0; i < npages; i++) { bas = (L_BYTEA *)ptraRemove(pa_data, i, L_NO_COMPACTION); l_byteaDestroy(&bas); } ptraDestroy(&pa_data, FALSE, FALSE); return ret; } /*! * ptraConcatenatePdfToData() * * Input: ptra (array of pdf strings, each for a single-page pdf file) * sarray ( of pathnames for input pdf files) * &data ( concatenated pdf data in memory) * &nbytes ( number of bytes in pdf data) * Return: 0 if OK, 1 on error * * Notes: * (1) This only works with leptonica-formatted single-page pdf files. * pdf files generated by other programs will have unpredictable * (and usually bad) results. The requirements for each pdf file: * (a) The Catalog and Info objects are the first two. * (b) Object 3 is Pages * (c) Object 4 is Page * (d) The remaining objects are Contents, XObjects, and ColorSpace * (2) We remove trailers from each page, and append the full trailer * for all pages at the end. * (3) For all but the first file, remove the ID and the first 3 * objects (catalog, info, pages), so that each subsequent * file has only objects of these classes: * Page, Contents, XObject, ColorSpace (Indexed RGB). * For those objects, we substitute these refs to objects * in the local file: * Page: Parent(object 3), Contents, XObject(typically multiple) * XObject: [ColorSpace if indexed] * The Pages object on the first page (object 3) has a Kids array * of references to all the Page objects, with a Count equal * to the number of pages. Each Page object refers back to * this parent. */ l_int32 ptraConcatenatePdfToData(L_PTRA *pa_data, SARRAY *sa, l_uint8 **pdata, size_t *pnbytes) { char *fname, *str_pages, *str_trailer; l_uint8 *pdfdata, *data; l_int32 i, j, index, nobj, npages; l_int32 *sizes, *locs; size_t size; L_BYTEA *bas, *bad, *bat1, *bat2; L_DNA *da_locs, *da_sizes, *da_outlocs, *da; L_DNAA *daa_locs; /* object locations on each page */ NUMA *na_objs, *napage; NUMAA *naa_objs; /* object mapping numbers to new values */ PROCNAME("ptraConcatenatePdfToData"); if (!pdata) return ERROR_INT("&data not defined", procName, 1); *pdata = NULL; if (!pnbytes) return ERROR_INT("&nbytes not defined", procName, 1); *pnbytes = 0; if (!pa_data) return ERROR_INT("pa_data not defined", procName, 1); /* Parse the files and find the object locations. * Remove file data that cannot be parsed. */ ptraGetActualCount(pa_data, &npages); daa_locs = l_dnaaCreate(npages); for (i = 0; i < npages; i++) { bas = (L_BYTEA *)ptraGetPtrToItem(pa_data, i); if (parseTrailerPdf(bas, &da_locs) != 0) { bas = (L_BYTEA *)ptraRemove(pa_data, i, L_NO_COMPACTION); l_byteaDestroy(&bas); if (sa) { fname = sarrayGetString(sa, i, L_NOCOPY); L_ERROR("can't parse file %s; skipping\n", procName, fname); } else { L_ERROR("can't parse file %d; skipping\n", procName, i); } } else { l_dnaaAddDna(daa_locs, da_locs, L_INSERT); } } /* Recompute npages in case some of the files were not pdf */ ptraCompactArray(pa_data); ptraGetActualCount(pa_data, &npages); if (npages == 0) { l_dnaaDestroy(&daa_locs); return ERROR_INT("no parsable pdf files found", procName, 1); } /* Find the mapping from initial to final object numbers */ naa_objs = numaaCreate(npages); /* stores final object numbers */ napage = numaCreate(npages); /* stores "Page" object numbers */ index = 0; for (i = 0; i < npages; i++) { da = l_dnaaGetDna(daa_locs, i, L_CLONE); nobj = l_dnaGetCount(da); if (i == 0) { numaAddNumber(napage, 4); /* object 4 on first page */ na_objs = numaMakeSequence(0.0, 1.0, nobj - 1); index = nobj - 1; } else { /* skip the first 3 objects in each file */ numaAddNumber(napage, index); /* Page object is first we add */ na_objs = numaMakeConstant(0.0, nobj - 1); numaReplaceNumber(na_objs, 3, 3); /* refers to parent of all */ for (j = 4; j < nobj - 1; j++) numaSetValue(na_objs, j, index++); } numaaAddNuma(naa_objs, na_objs, L_INSERT); l_dnaDestroy(&da); } /* Make the Pages object (#3) */ str_pages = generatePagesObjStringPdf(napage); /* Build the output */ bad = l_byteaCreate(5000); da_outlocs = l_dnaCreate(0); /* locations of all output objects */ for (i = 0; i < npages; i++) { bas = (L_BYTEA *)ptraGetPtrToItem(pa_data, i); pdfdata = l_byteaGetData(bas, &size); da_locs = l_dnaaGetDna(daa_locs, i, L_CLONE); /* locs on this page */ na_objs = numaaGetNuma(naa_objs, i, L_CLONE); /* obj # on this page */ nobj = l_dnaGetCount(da_locs) - 1; da_sizes = l_dnaMakeDelta(da_locs); /* object sizes on this page */ sizes = l_dnaGetIArray(da_sizes); locs = l_dnaGetIArray(da_locs); if (i == 0) { l_byteaAppendData(bad, pdfdata, sizes[0]); l_byteaAppendData(bad, pdfdata + locs[1], sizes[1]); l_byteaAppendData(bad, pdfdata + locs[2], sizes[2]); l_byteaAppendString(bad, str_pages); for (j = 0; j < 4; j++) l_dnaAddNumber(da_outlocs, locs[j]); } for (j = 4; j < nobj; j++) { l_dnaAddNumber(da_outlocs, l_byteaGetSize(bad)); bat1 = l_byteaInitFromMem(pdfdata + locs[j], sizes[j]); bat2 = substituteObjectNumbers(bat1, na_objs); data = l_byteaGetData(bat2, &size); l_byteaAppendData(bad, data, size); l_byteaDestroy(&bat1); l_byteaDestroy(&bat2); } if (i == npages - 1) /* last one */ l_dnaAddNumber(da_outlocs, l_byteaGetSize(bad)); FREE(sizes); FREE(locs); l_dnaDestroy(&da_locs); numaDestroy(&na_objs); l_dnaDestroy(&da_sizes); } /* Add the trailer */ str_trailer = makeTrailerStringPdf(da_outlocs); l_byteaAppendString(bad, str_trailer); /* Transfer the output data */ *pdata = l_byteaCopyData(bad, pnbytes); l_byteaDestroy(&bad); #if DEBUG_MULTIPAGE fprintf(stderr, "******** object mapper **********"); numaaWriteStream(stderr, naa_objs); fprintf(stderr, "******** Page object numbers ***********"); numaWriteStream(stderr, napage); fprintf(stderr, "******** Pages object ***********\n"); fprintf(stderr, "%s\n", str_pages); #endif /* DEBUG_MULTIPAGE */ numaDestroy(&napage); numaaDestroy(&naa_objs); l_dnaDestroy(&da_outlocs); l_dnaaDestroy(&daa_locs); FREE(str_pages); FREE(str_trailer); return 0; } /*---------------------------------------------------------------------* * Helper functions for generating the multi-page pdf output * *---------------------------------------------------------------------*/ /*! * parseTrailerPdf() * * Input: bas (lba of a pdf file) * da ( byte locations of the beginning of each object) * Return: 0 if OK, 1 on error */ static l_int32 parseTrailerPdf(L_BYTEA *bas, L_DNA **pda) { char *str; l_uint8 nl = '\n'; l_uint8 *data; l_int32 i, j, start, startloc, xrefloc, found, loc, nobj, objno, trailer_ok; size_t size; L_DNA *da, *daobj, *daxref; SARRAY *sa; PROCNAME("parseTrailerPdf"); if (!pda) return ERROR_INT("&da not defined", procName, 1); *pda = NULL; if (!bas) return ERROR_INT("bas not defined", procName, 1); data = l_byteaGetData(bas, &size); if (strncmp((char *)data, "%PDF-1.", 7) != 0) return ERROR_INT("PDF header signature not found", procName, 1); /* Search for "startxref" starting 50 bytes from the EOF */ start = 0; if (size > 50) start = size - 50; arrayFindSequence(data + start, size - start, (l_uint8 *)"startxref\n", 10, &loc, &found); if (!found) return ERROR_INT("startxref not found!", procName, 1); if (sscanf((char *)(data + start + loc + 10), "%d\n", &xrefloc) != 1) return ERROR_INT("xrefloc not found!", procName, 1); if (xrefloc < 0 || xrefloc >= size) return ERROR_INT("invalid xrefloc!", procName, 1); sa = sarrayCreateLinesFromString((char *)(data + xrefloc), 0); str = sarrayGetString(sa, 1, L_NOCOPY); if ((sscanf(str, "0 %d", &nobj)) != 1) return ERROR_INT("nobj not found", procName, 1); /* Get starting locations. The numa index is the * object number. loc[0] is the ID; loc[nobj + 1] is xrefloc. */ da = l_dnaCreate(nobj + 1); *pda = da; for (i = 0; i < nobj; i++) { str = sarrayGetString(sa, i + 2, L_NOCOPY); sscanf(str, "%d", &startloc); l_dnaAddNumber(da, startloc); } l_dnaAddNumber(da, xrefloc); #if DEBUG_MULTIPAGE fprintf(stderr, "************** Trailer string ************\n"); fprintf(stderr, "xrefloc = %d", xrefloc); sarrayWriteStream(stderr, sa); fprintf(stderr, "************** Object locations ************"); l_dnaWriteStream(stderr, da); #endif /* DEBUG_MULTIPAGE */ sarrayDestroy(&sa); /* Verify correct parsing */ trailer_ok = TRUE; for (i = 1; i < nobj; i++) { l_dnaGetIValue(da, i, &startloc); if ((sscanf((char *)(data + startloc), "%d 0 obj", &objno)) != 1) { L_ERROR("bad trailer for object %d\n", procName, i); trailer_ok = FALSE; break; } } /* If the trailer is broken, reconstruct the correct obj locations */ if (!trailer_ok) { L_INFO("rebuilding pdf trailer\n", procName); l_dnaEmpty(da); l_dnaAddNumber(da, 0); l_byteaFindEachSequence(bas, (l_uint8 *)" 0 obj\n", 7, &daobj); nobj = l_dnaGetCount(daobj); for (i = 0; i < nobj; i++) { l_dnaGetIValue(daobj, i, &loc); for (j = loc - 1; j > 0; j--) { if (data[j] == nl) break; } l_dnaAddNumber(da, j + 1); } l_byteaFindEachSequence(bas, (l_uint8 *)"xref", 4, &daxref); l_dnaGetIValue(daxref, 0, &loc); l_dnaAddNumber(da, loc); l_dnaDestroy(&daobj); l_dnaDestroy(&daxref); } return 0; } static char * generatePagesObjStringPdf(NUMA *napage) { char *str; char *buf; l_int32 i, n, index, bufsize; SARRAY *sa; PROCNAME("generatePagesObjStringPdf"); if (!napage) return (char *)ERROR_PTR("napage not defined", procName, NULL); n = numaGetCount(napage); bufsize = 100 + 16 * n; /* large enough to hold the output string */ buf = (char *)CALLOC(bufsize, sizeof(char)); sa = sarrayCreate(n); for (i = 0; i < n; i++) { numaGetIValue(napage, i, &index); snprintf(buf, bufsize, " %d 0 R ", index); sarrayAddString(sa, buf, L_COPY); } str = sarrayToString(sa, 0); snprintf(buf, bufsize - 1, "3 0 obj\n" "<<\n" "/Type /Pages\n" "/Kids [%s]\n" "/Count %d\n" ">>\n", str, n); sarrayDestroy(&sa); FREE(str); return buf; } /*! * substituteObjectNumbers() * * Input: bas (lba of a pdf object) * na_objs (object number mapping array) * Return: bad (lba of rewritten pdf for the object) * * Notes: * (1) Interpret the first set of bytes as the object number, * map to the new number, and write it out. * (2) Find all occurrences of this 4-byte sequence: " 0 R" * (3) Find the location and value of the integer preceeding this, * and map it to the new value. * (4) Rewrite the object with new object numbers. */ static L_BYTEA * substituteObjectNumbers(L_BYTEA *bas, NUMA *na_objs) { l_uint8 space = ' '; l_uint8 *datas; l_uint8 buf[32]; /* only needs to hold one integer in ascii format */ l_int32 start, nrepl, i, j, objin, objout; l_int32 *objs, *matches; size_t size; L_BYTEA *bad; L_DNA *da_match; datas = l_byteaGetData(bas, &size); bad = l_byteaCreate(100); objs = numaGetIArray(na_objs); /* object number mapper */ /* Substitute the object number on the first line */ sscanf((char *)datas, "%d", &objin); objout = objs[objin]; snprintf((char *)buf, 32, "%d", objout); l_byteaAppendString(bad, (char *)buf); /* Find the set of matching locations for object references */ arrayFindSequence(datas, size, &space, 1, &start, NULL); da_match = arrayFindEachSequence(datas, size, (l_uint8 *)" 0 R", 4); if (!da_match) { l_byteaAppendData(bad, datas + start, size - start); FREE(objs); return bad; } /* Substitute all the object reference numbers */ nrepl = l_dnaGetCount(da_match); matches = l_dnaGetIArray(da_match); for (i = 0; i < nrepl; i++) { /* Find the first space before the object number */ for (j = matches[i] - 1; j > 0; j--) { if (datas[j] == space) break; } /* Copy bytes from 'start' up to the object number */ l_byteaAppendData(bad, datas + start, j - start + 1); sscanf((char *)(datas + j + 1), "%d", &objin); objout = objs[objin]; snprintf((char *)buf, 32, "%d", objout); l_byteaAppendString(bad, (char *)buf); start = matches[i]; } l_byteaAppendData(bad, datas + start, size - start); FREE(objs); FREE(matches); l_dnaDestroy(&da_match); return bad; } /*---------------------------------------------------------------------* * Create/destroy/access pdf data * *---------------------------------------------------------------------*/ static L_PDF_DATA * pdfdataCreate(const char *title) { L_PDF_DATA *lpd; lpd = (L_PDF_DATA *)CALLOC(1, sizeof(L_PDF_DATA)); if (title) lpd->title = stringNew(title); lpd->cida = ptraCreate(10); lpd->xy = ptaCreate(10); lpd->wh = ptaCreate(10); lpd->saprex = sarrayCreate(10); lpd->sacmap = sarrayCreate(10); lpd->objsize = l_dnaCreate(20); lpd->objloc = l_dnaCreate(20); return lpd; } static void pdfdataDestroy(L_PDF_DATA **plpd) { l_int32 i; L_COMP_DATA *cid; L_PDF_DATA *lpd; PROCNAME("pdfdataDestroy"); if (plpd== NULL) { L_WARNING("ptr address is null!\n", procName); return; } if ((lpd = *plpd) == NULL) return; if (lpd->title) FREE(lpd->title); for (i = 0; i < lpd->n; i++) { cid = (L_COMP_DATA *)ptraRemove(lpd->cida, i, L_NO_COMPACTION); l_compdataDestroy(&cid); } ptraDestroy(&lpd->cida, 0, 0); if (lpd->id) FREE(lpd->id); if (lpd->obj1) FREE(lpd->obj1); if (lpd->obj2) FREE(lpd->obj2); if (lpd->obj3) FREE(lpd->obj3); if (lpd->obj4) FREE(lpd->obj4); if (lpd->obj5) FREE(lpd->obj5); if (lpd->poststream) FREE(lpd->poststream); if (lpd->trailer) FREE(lpd->trailer); if (lpd->xy) ptaDestroy(&lpd->xy); if (lpd->wh) ptaDestroy(&lpd->wh); if (lpd->mediabox) boxDestroy(&lpd->mediabox); if (lpd->saprex) sarrayDestroy(&lpd->saprex); if (lpd->sacmap) sarrayDestroy(&lpd->sacmap); if (lpd->objsize) l_dnaDestroy(&lpd->objsize); if (lpd->objloc) l_dnaDestroy(&lpd->objloc); FREE(lpd); *plpd = NULL; return; } static L_COMP_DATA * pdfdataGetCid(L_PDF_DATA *lpd, l_int32 index) { PROCNAME("pdfdataGetCid"); if (!lpd) return (L_COMP_DATA *)ERROR_PTR("lpd not defined", procName, NULL); if (index < 0 || index >= lpd->n) return (L_COMP_DATA *)ERROR_PTR("invalid image index", procName, NULL); return (L_COMP_DATA *)ptraGetPtrToItem(lpd->cida, index); } /*---------------------------------------------------------------------* * Set flags for special modes * *---------------------------------------------------------------------*/ /*! * l_pdfSetG4ImageMask() * * Input: flag (1 for writing g4 data as fg only through a mask; * 0 for writing fg and bg) * Return: void * * Notes: * (1) The default is for writing only the fg (through the mask). * That way when you write a 1 bpp image, the bg is transparent, * so any previously written image remains visible behind it. */ void l_pdfSetG4ImageMask(l_int32 flag) { var_WRITE_G4_IMAGE_MASK = flag; } /*! * l_pdfSetDateAndVersion() * * Input: flag (1 for writing date/time and leptonica version; * 0 for omitting this from the metadata) * Return: void * * Notes: * (1) The default is for writing this data. For regression tests * that compare output against golden files, it is useful to omit. */ void l_pdfSetDateAndVersion(l_int32 flag) { var_WRITE_DATE_AND_VERSION = flag; } /* --------------------------------------------*/ #endif /* USE_PDFIO */ /* --------------------------------------------*/