/* Copyright (C) 2001-2021 Artifex Software, Inc. All Rights Reserved. This software is provided AS-IS with no warranty, either express or implied. This software is distributed under license and may not be copied, modified or distributed except as expressly authorized under the terms of the license contained in the file LICENSE in this distribution. Refer to licensing information at http://www.artifex.com or contact Artifex Software, Inc., 1305 Grant Avenue - Suite 200, Novato, CA 94945, U.S.A., +1(415)492-9861, for further information. */ /* Image-writing utilities for pdfwrite driver */ #include "memory_.h" #include "gx.h" #include "gserrors.h" #include "gdevpdfx.h" #include "gdevpdfg.h" #include "gdevpdfo.h" #include "gxcspace.h" #include "gsiparm4.h" #include "gdevpsds.h" #include "spngpx.h" #include /* for atoi */ #define CHECK(expr)\ BEGIN if ((code = (expr)) < 0) return code; END /* GC descriptors */ public_st_pdf_image_writer(); static ENUM_PTRS_WITH(pdf_image_writer_enum_ptrs, pdf_image_writer *piw) index -= 4; if (index < psdf_binary_writer_max_ptrs * piw->alt_writer_count) { gs_ptr_type_t ret = ENUM_USING(st_psdf_binary_writer, &piw->binary[index / psdf_binary_writer_max_ptrs], sizeof(psdf_binary_writer), index % psdf_binary_writer_max_ptrs); if (ret == 0) /* don't stop early */ ENUM_RETURN(0); return ret; } return 0; case 0: ENUM_RETURN(piw->pres); case 1: ENUM_RETURN(piw->data); case 2: ENUM_RETURN(piw->named); case 3: ENUM_RETURN(piw->pres_mask); ENUM_PTRS_END static RELOC_PTRS_WITH(pdf_image_writer_reloc_ptrs, pdf_image_writer *piw) { int i; for (i = 0; i < piw->alt_writer_count; ++i) RELOC_USING(st_psdf_binary_writer, &piw->binary[i], sizeof(psdf_binary_writer)); RELOC_VAR(piw->pres); RELOC_VAR(piw->data); RELOC_VAR(piw->named); RELOC_VAR(piw->pres_mask); } RELOC_PTRS_END /* ---------------- Image stream dictionaries ---------------- */ const pdf_image_names_t pdf_image_names_full = { { PDF_COLOR_SPACE_NAMES }, { PDF_FILTER_NAMES }, PDF_IMAGE_PARAM_NAMES }; const pdf_image_names_t pdf_image_names_short = { { PDF_COLOR_SPACE_NAMES_SHORT }, { PDF_FILTER_NAMES_SHORT }, PDF_IMAGE_PARAM_NAMES_SHORT }; /* Store the values of image parameters other than filters. */ /* pdev is used only for updating procsets. */ /* pcsvalue is not used for masks. */ static int pdf_put_pixel_image_values(cos_dict_t *pcd, gx_device_pdf *pdev, const gs_pixel_image_t *pim, const gs_color_space *pcs, const pdf_image_names_t *pin, const cos_value_t *pcsvalue) { int num_components; float indexed_decode[2]; const float *default_decode = NULL; int code; if (pcs) { CHECK(cos_dict_put_c_key(pcd, pin->ColorSpace, pcsvalue)); pdf_color_space_procsets(pdev, pcs); num_components = gs_color_space_num_components(pcs); if (gs_color_space_get_index(pcs) == gs_color_space_index_Indexed) { indexed_decode[0] = 0; indexed_decode[1] = (float)((1 << pim->BitsPerComponent) - 1); default_decode = indexed_decode; } } else num_components = 1; CHECK(cos_dict_put_c_key_int(pcd, pin->Width, pim->Width)); CHECK(cos_dict_put_c_key_int(pcd, pin->Height, pim->Height)); CHECK(cos_dict_put_c_key_int(pcd, pin->BitsPerComponent, pim->BitsPerComponent)); { int i; for (i = 0; i < num_components * 2; ++i) { if (pim->Decode[i] != (default_decode ? default_decode[i] : i & 1) ) break; } if (i < num_components * 2) { cos_array_t *pca = cos_array_alloc(pdev, "pdf_put_pixel_image_values(decode)"); if (pca == 0) return_error(gs_error_VMerror); if (pcs == NULL) { /* 269-01.ps sets /Decode[0 100] with a mask image. */ for (i = 0; i < num_components * 2; ++i) CHECK(cos_array_add_real(pca, min(pim->Decode[i], 1))); } else { for (i = 0; i < num_components * 2; ++i) CHECK(cos_array_add_real(pca, pim->Decode[i])); } CHECK(cos_dict_put_c_key_object(pcd, pin->Decode, COS_OBJECT(pca))); } } if (pim->Interpolate) { if (pdev->PDFA != 0) emprintf(pdev->memory, "PDFA doesn't allow images with Interpolate true.\n"); else CHECK(cos_dict_put_c_strings(pcd, pin->Interpolate, "true")); } return 0; } int pdf_put_image_values(cos_dict_t *pcd, gx_device_pdf *pdev, const gs_pixel_image_t *pic, const pdf_image_names_t *pin, const cos_value_t *pcsvalue) { const gs_color_space *pcs = pic->ColorSpace; int code; switch (pic->type->index) { case 1: { const gs_image1_t *pim = (const gs_image1_t *)pic; if (pim->ImageMask) { CHECK(cos_dict_put_c_strings(pcd, pin->ImageMask, "true")); pdev->procsets |= ImageB; pcs = NULL; } } break; case 3: { /* * Clients must treat this as a special case: they must call * pdf_put_image_values for the MaskDict separately, and must * add the Mask entry to the main image stream (dictionary). */ /*const gs_image3_t *pim = (const gs_image3_t *)pic;*/ /* Masked images are only supported starting in PDF 1.3. */ if (pdev->CompatibilityLevel < 1.3) return_error(gs_error_rangecheck); } break; case 4: { const gs_image4_t *pim = (const gs_image4_t *)pic; int num_components = gs_color_space_num_components(pcs); cos_array_t *pca; int i; /* Masked images are only supported starting in PDF 1.3. */ if (pdev->CompatibilityLevel < 1.3) break; /* Will convert into an imagemask with a pattern color. */ pca = cos_array_alloc(pdev, "pdf_put_image_values(mask)"); if (pca == 0) return_error(gs_error_VMerror); for (i = 0; i < num_components; ++i) { int lo, hi; if (pim->MaskColor_is_range) lo = pim->MaskColor[i * 2], hi = pim->MaskColor[i * 2 + 1]; else lo = hi = pim->MaskColor[i]; CHECK(cos_array_add_int(pca, lo)); CHECK(cos_array_add_int(pca, hi)); } CHECK(cos_dict_put_c_key_object(pcd, "/Mask", COS_OBJECT(pca))); } break; default: return_error(gs_error_rangecheck); } return pdf_put_pixel_image_values(pcd, pdev, pic, pcs, pin, pcsvalue); } /* Store filters for an image. */ /* Currently this only saves parameters for CCITTFaxDecode. */ int pdf_put_image_filters(cos_dict_t *pcd, gx_device_pdf *pdev, const psdf_binary_writer * pbw, const pdf_image_names_t *pin) { return pdf_put_filters(pcd, pdev, pbw->strm, &pin->filter_names); } /* ---------------- Image writing ---------------- */ /* * Fill in the image parameters for a device space bitmap. * PDF images are always specified top-to-bottom. * data_h is the actual number of data rows, which may be less than h. */ void pdf_make_bitmap_matrix(gs_matrix * pmat, int x, int y, int w, int h, int h_actual) { pmat->xx = (float)w; pmat->xy = 0; pmat->yx = 0; pmat->yy = (float)(-h_actual); pmat->tx = (float)x; pmat->ty = (float)(y + h); } /* * Put out the gsave and matrix for an image. y_scale adjusts the matrix * for images that end prematurely. */ void pdf_put_image_matrix(gx_device_pdf * pdev, const gs_matrix * pmat, double y_scale) { gs_matrix imat = {1, 0, 0, 1, 0 ,0}; gs_matrix_translate(pmat, 0.0, 1.0 - y_scale, &imat); gs_matrix_scale(&imat, 1.0, y_scale, &imat); pdf_put_matrix(pdev, "q ", &imat, "cm\n"); } /* Put out a reference to an image resource. */ int pdf_do_image_by_id(gx_device_pdf * pdev, double scale, const gs_matrix * pimat, bool in_contents, gs_id id) { /* fixme : in_contents is always true (there are no calls with false). */ if (in_contents) { int code = pdf_open_contents(pdev, PDF_IN_STREAM); if (code < 0) return code; } if (pimat) pdf_put_image_matrix(pdev, pimat, scale); pprintld1(pdev->strm, "/R%ld Do\nQ\n", id); return 0; } int pdf_do_image(gx_device_pdf * pdev, const pdf_resource_t * pres, const gs_matrix * pimat, bool in_contents) { /* fixme : call pdf_do_image_by_id when pimam == NULL. */ double scale = 1; if (pimat) { /* Adjust the matrix to account for short images. */ const pdf_x_object_t *const pxo = (const pdf_x_object_t *)pres; scale = (double)pxo->data_height / pxo->height; } return pdf_do_image_by_id(pdev, scale, pimat, in_contents, pdf_resource_id(pres)); } /* ------ Begin / finish ------ */ /* Initialize image writer. */ void pdf_image_writer_init(pdf_image_writer * piw) { memset(piw, 0, sizeof(*piw)); piw->alt_writer_count = 1; /* Default. */ } /* * Begin writing an image, creating the resource if not in-line, and setting * up the binary writer. If pnamed != 0, it is a stream object created by a * NI pdfmark. */ int pdf_begin_write_image(gx_device_pdf * pdev, pdf_image_writer * piw, gx_bitmap_id id, int w, int h, cos_dict_t *named, bool in_line) { /* Patch pdev->strm so the right stream gets into the writer. */ stream *save_strm = pdev->strm; cos_stream_t *data; bool mask = (piw->data != NULL); int alt_stream_index = (!mask ? 0 : piw->alt_writer_count); int code; if (in_line) { piw->pres = 0; piw->pin = &pdf_image_names_short; data = cos_stream_alloc(pdev, "pdf_begin_image_data"); if (data == 0) return_error(gs_error_VMerror); piw->end_string = " Q"; piw->named = 0; /* must have named == 0 */ } else { pdf_x_object_t *pxo; cos_stream_t *pcos; pdf_resource_t *pres; /* * Note that if named != 0, there are two objects with the same id * while the image is being accumulated: named, and pres->object. */ code = pdf_alloc_resource(pdev, resourceXObject, id, &pres, (named ? named->id : -1L)); if (code < 0) return code; *(mask ? &piw->pres_mask : &piw->pres) = pres; cos_become(pres->object, cos_type_stream); pres->rid = id; piw->pin = &pdf_image_names_full; pxo = (pdf_x_object_t *)pres; pcos = (cos_stream_t *)pxo->object; CHECK(cos_dict_put_c_strings(cos_stream_dict(pcos), "/Subtype", "/Image")); pxo->width = w; pxo->height = h; /* Initialize data_height for the benefit of copy_{mono,color}. */ pxo->data_height = h; data = pcos; if (!mask) piw->named = named; } pdev->strm = pdev->streams.strm; pdev->strm = cos_write_stream_alloc(data, pdev, "pdf_begin_write_image"); if (pdev->strm == 0) return_error(gs_error_VMerror); if (!mask) piw->data = data; piw->height = h; code = psdf_begin_binary((gx_device_psdf *) pdev, &piw->binary[alt_stream_index]); piw->binary[alt_stream_index].target = NULL; /* We don't need target with cos_write_stream. */ pdev->strm = save_strm; return code; } /* * Make alternative stream for image compression choice. */ int pdf_make_alt_stream(gx_device_pdf * pdev, psdf_binary_writer * pbw) { stream *save_strm = pdev->strm; cos_stream_t *pcos = cos_stream_alloc(pdev, "pdf_make_alt_stream"); int code; if (pcos == 0) return_error(gs_error_VMerror); pcos->id = 0; CHECK(cos_dict_put_c_strings(cos_stream_dict(pcos), "/Subtype", "/Image")); pbw->strm = cos_write_stream_alloc(pcos, pdev, "pdf_make_alt_stream"); if (pbw->strm == 0) return_error(gs_error_VMerror); pbw->dev = (gx_device_psdf *)pdev; pbw->memory = pdev->pdf_memory; pdev->strm = pbw->strm; code = psdf_begin_binary((gx_device_psdf *) pdev, pbw); pdev->strm = save_strm; pbw->target = NULL; /* We don't need target with cos_write_stream. */ return code; } /* Begin writing the image data, setting up the dictionary and filters. */ int pdf_begin_image_data(gx_device_pdf * pdev, pdf_image_writer * piw, const gs_pixel_image_t * pim, const cos_value_t *pcsvalue, int alt_writer_index) { cos_stream_t *s; cos_dict_t *pcd; int code; s = cos_stream_from_pipeline(piw->binary[alt_writer_index].strm); if (s == 0L) return gs_note_error(gs_error_ioerror); pcd = cos_stream_dict(s); code = pdf_put_image_values(pcd, pdev, pim, piw->pin, pcsvalue); if (code >= 0) code = pdf_put_image_filters(pcd, pdev, &piw->binary[alt_writer_index], piw->pin); if (code < 0) { if (!piw->pres) COS_FREE(piw->data, "pdf_begin_image_data"); piw->data = 0; } if (pdev->JPEG_PassThrough) { CHECK(cos_dict_put_c_strings(pcd, "/Filter", "/DCTDecode")); } if (pdev->JPX_PassThrough) { CHECK(cos_dict_put_c_strings(pcd, "/Filter", "/JPXDecode")); } return code; } /* Complete image data. */ int pdf_complete_image_data(gx_device_pdf *pdev, pdf_image_writer *piw, int data_h, int width, int bits_per_pixel) { if (data_h != piw->height) { if (piw->binary[0].strm->procs.process == s_DCTE_template.process || piw->binary[0].strm->procs.process == s_PNGPE_template.process ) { /* Since DCTE and PNGPE can't safely close with incomplete data, we add stub data to complete the stream. */ int bytes_per_line = (width * bits_per_pixel + 7) / 8; int lines_left = piw->height - data_h; byte buf[256]; const uint lb = sizeof(buf); int i, l; uint ignore; memset(buf, 128, lb); for (; lines_left; lines_left--) for (i = 0; i < piw->alt_writer_count; i++) { for (l = bytes_per_line; l > 0; l -= lb) if ((sputs(piw->binary[i].strm, buf, min(l, lb), &ignore)) < 0) return_error(gs_error_ioerror); } } } return 0; } /* Finish writing the binary image data. */ int pdf_end_image_binary(gx_device_pdf *pdev, pdf_image_writer *piw, int data_h) { int code, code1 = 0; if (piw->alt_writer_count > 2) code = pdf_choose_compression(piw, true); else code = psdf_end_binary(&piw->binary[0]); /* If the image ended prematurely, update the Height. */ if (data_h != piw->height) { char data[256]; int OutHeight; cos_value_t *value; value = (cos_value_t *)cos_dict_find(cos_stream_dict(piw->data), (const byte *)piw->pin->Height, strlen(piw->pin->Height)); if (!value || value->contents.chars.size > 255) return(gs_error_rangecheck); strncpy((char *)&data, (const char *)value->contents.chars.data, value->contents.chars.size); data[value->contents.chars.size] = 0x00; OutHeight = atoi(data); if (OutHeight != piw->height) { /* Looks like we are downsampling, so we can't use the number * of rows of data actually received, we must divide those by * the sampling factor. */ float factor = (float)OutHeight / piw->height; OutHeight = (int)(factor * data_h); code1 = cos_dict_put_c_key_int(cos_stream_dict(piw->data), piw->pin->Height, OutHeight); } else { code1 = cos_dict_put_c_key_int(cos_stream_dict(piw->data), piw->pin->Height, data_h); } } return code < 0 ? code : code1; } /* When writing out an image, we check to see if its a duplicate of an existing image, and if so we simply * use the existing image. There is one potential problem here; if we have an image with a SMask, the SMask is * itself animage, and the SMask image and the image which uses it are identical. Because we don't add the SMask * entry to the image dictionary until we have closed the image, its possible that the image can detect the alredy * stored SMask image as being identical and attempt to use the SMask image instead. This leads to an image with * an SMask entry referencing itself. * We detect this here simply by checking if the detected image resource ID is the same as any current SMask. Worst * case is we fail to detect a duplicate, which is better than detecting an incorrect duplicate. * This check function is only used in pdf_end_write_image() below as an argument to pdf_substitute_resource(). */ static int smask_image_check(gx_device_pdf * pdev, pdf_resource_t *pres0, pdf_resource_t *pres1) { cos_value_t *v = NULL; /* image_mask_id is non-zero if we have a pending SMask image */ if (pdev->image_mask_id != 0) { if (pres0->object->id == pdev->image_mask_id || pres1->object->id == pdev->image_mask_id) return 0; if (pdev->image_mask_is_SMask) v = (cos_value_t *)cos_dict_find_c_key((const cos_dict_t *)pres1->object, "/SMask"); else v = (cos_value_t *)cos_dict_find_c_key((const cos_dict_t *)pres1->object, "/Mask"); if (v == 0) return 0; if (v != 0) { const byte *p = v->contents.chars.data; int ix = 0; while (*p != 0x20) { if (p > v->contents.chars.data + v->contents.chars.size) return 0; ix *= 10; ix += (*p) - 0x30; } if (ix != pdev->image_mask_id) return 0; } } return 1; } /* Abort an image without writing it. * Frees any associated memory. */ int pdf_end_abort_image(gx_device_pdf * pdev, pdf_image_writer * piw) { pdf_resource_t *pres = piw->pres; if (!pres) { COS_FREE(piw->data, "pdf_end_write_image"); } return 0; } /* * Finish writing an image. If in-line, write the BI/dict/ID/data/EI and * return 1; if a resource, write the resource definition and return 0. */ int pdf_end_write_image(gx_device_pdf * pdev, pdf_image_writer * piw) { pdf_resource_t *pres = piw->pres; if (pres) { /* image resource */ cos_object_t *const pco = pres->object; cos_stream_t *const pcs = (cos_stream_t *)pco; cos_dict_t *named = piw->named; int code; if (named) { if (pdev->ForOPDFRead) { code = cos_dict_put_c_key_bool(named, "/.Global", true); if (code < 0) return code; } /* * This image was named by NI. Copy any dictionary elements * from the named dictionary to the image stream, and then * associate the name with the stream. */ code = cos_dict_move_all(cos_stream_dict(pcs), named); if (code < 0) return code; pres->named = true; /* * We need to make the entry in the name dictionary point to * the stream (pcs) rather than the object created by NI (named). * Unfortunately, we no longer know what dictionary to use. * Instead, overwrite the latter with the former's contents, * and change the only relevant pointer. */ *(cos_object_t *)named = *pco; pres->object = COS_OBJECT(named); } else if (!pres->named) { /* named objects are written at the end */ if (pdev->DetectDuplicateImages) { pdf_x_object_t *pxo = (pdf_x_object_t *)piw->pres; int height = pxo->height, width = pxo->width; code = pdf_substitute_resource(pdev, &piw->pres, resourceXObject, smask_image_check, false); if (code < 0) return code; /* These values are related to the image matrix and should *not* be * substituted if we found a duplicate image, or the matrix calculation * will be incorrect! This only seems to matter for the PCL interpreter. */ pxo = (pdf_x_object_t *)piw->pres; pxo->height = height; pxo->width = width; } else { pdf_reserve_object_id(pdev, piw->pres, gs_no_id); } /* Warning : If the substituted image used alternate streams, its space in the pdev->streams.strm file won't be released. */ piw->pres->where_used |= pdev->used_mask; } code = pdf_add_resource(pdev, pdev->substream_Resources, "/XObject", piw->pres); if (code < 0) return code; return 0; } else { /* in-line image */ stream *s = pdev->strm; uint KeyLength = pdev->KeyLength; stream_puts(s, "BI\n"); cos_stream_elements_write(piw->data, pdev); stream_puts(s, (pdev->binary_ok ? "ID " : "ID\n")); pdev->KeyLength = 0; /* Disable encryption for the inline image. */ cos_stream_contents_write(piw->data, pdev); pdev->KeyLength = KeyLength; pprints1(s, "\nEI%s\n", piw->end_string); COS_FREE(piw->data, "pdf_end_write_image"); return 1; } } /* ------ Copy data ------ */ /* Copy the data for a mask or monobit bitmap. */ int pdf_copy_mask_bits(stream *s, const byte *base, int sourcex, int raster, int w, int h, byte invert) { int yi; for (yi = 0; yi < h; ++yi) { const byte *data = base + yi * raster + (sourcex >> 3); int sbit = sourcex & 7; if (sbit == 0) { int nbytes = (w + 7) >> 3; int i; for (i = 0; i < nbytes; ++data, ++i) sputc(s, (byte)(*data ^ invert)); } else { int wleft = w; int rbit = 8 - sbit; for (; wleft + sbit > 8; ++data, wleft -= 8) sputc(s, (byte)(((*data << sbit) + (data[1] >> rbit)) ^ invert)); if (wleft > 0) sputc(s, (byte)(((*data << sbit) ^ invert) & (byte) (0xff00 >> wleft))); } } return 0; } /* Copy the data for a colored image (device pixels). */ int pdf_copy_color_bits(stream *s, const byte *base, int sourcex, int raster, int w, int h, int bytes_per_pixel) { int yi; for (yi = 0; yi < h; ++yi) { uint ignore; sputs(s, base + sourcex * bytes_per_pixel + yi * raster, w * bytes_per_pixel, &ignore); } return 0; } /* Choose image compression - auxiliary procs */ static inline bool much_bigger__DL(long l1, long l2) { return l1 > 1024*1024 && l2 < l1 / 3; } static void pdf_choose_compression_cos(pdf_image_writer *piw, cos_stream_t *s[2], bool force) { /* Assume s[0] is Flate, s[1] is DCT, s[2] is chooser. */ long l0, l1; int k0, k1; l0 = cos_stream_length(s[0]); l1 = cos_stream_length(s[1]); if ((force && l0 <= l1) || l1 == -1) k0 = 1; /* Use Flate if it is not longer. Or if the DCT failed */ else { k0 = s_compr_chooser__get_choice( (stream_compr_chooser_state *)piw->binary[2].strm->state, force); if (k0 && l0 > 0 && l1 > 0) k0--; else if (much_bigger__DL(l0, l1)) k0 = 0; else if (much_bigger__DL(l1, l0) || force) k0 = 1; else return; } k1 = 1 - k0; s_close_filters(&piw->binary[k0].strm, piw->binary[k0].target); s[k0]->cos_procs->release((cos_object_t *)s[k0], "pdf_image_choose_filter"); s[k0]->written = 1; piw->binary[0].strm = piw->binary[k1].strm; s_close_filters(&piw->binary[2].strm, piw->binary[2].target); piw->binary[1].strm = piw->binary[2].strm = 0; /* for GC */ piw->binary[1].target = piw->binary[2].target = 0; s[k1]->id = piw->pres->object->id; piw->pres->object = (cos_object_t *)s[k1]; piw->data = s[k1]; if (piw->alt_writer_count > 3) { piw->binary[1] = piw->binary[3]; piw->binary[3].strm = 0; /* for GC */ piw->binary[3].target = 0; } piw->alt_writer_count -= 2; } /* End binary with choosing image compression. */ int pdf_choose_compression(pdf_image_writer * piw, bool end_binary) { cos_stream_t *s[2]; int status; s[0] = cos_stream_from_pipeline(piw->binary[0].strm); s[1] = cos_stream_from_pipeline(piw->binary[1].strm); if (s[0] == 0L) { return_error(gs_error_ioerror); } if (s[1] == 0L) { s_close_filters(&piw->binary[0].strm, piw->binary[0].target); return_error(gs_error_ioerror); } if (end_binary) { status = s_close_filters(&piw->binary[0].strm, piw->binary[0].target); if (status < 0) return_error(gs_error_ioerror); status = s_close_filters(&piw->binary[1].strm, piw->binary[1].target); if (status < 0) s[1]->length = -1; } pdf_choose_compression_cos(piw, s, end_binary); return 0; }