#include "mupdf/fitz.h" #include typedef struct ps_band_writer_s { fz_band_writer super; z_stream stream; int input_size; unsigned char *input; int output_size; unsigned char *output; } ps_band_writer; void fz_write_ps_file_header(fz_context *ctx, fz_output *out) { fz_printf(ctx, out, "%%!PS-Adobe-3.0\n" //"%%%%BoundingBox: 0 0 612 792\n" //"%%%%HiResBoundingBox: 0 0 612 792\n" "%%%%Creator: MuPDF\n" "%%%%LanguageLevel: 2\n" "%%%%CreationDate: D:20160318101706Z00'00'\n" "%%%%DocumentData: Binary\n" "%%%%Pages: (atend)\n" "%%%%EndComments\n" "\n" "%%%%BeginProlog\n" "%%%%EndProlog\n" "\n" "%%%%BeginSetup\n" "%%%%EndSetup\n" "\n" ); } void fz_write_ps_file_trailer(fz_context *ctx, fz_output *out, int pages) { fz_printf(ctx, out, "%%%%Trailer\n%%%%Pages: %d\n%%%%EOF\n", pages); } static void ps_write_header(fz_context *ctx, fz_band_writer *writer_) { ps_band_writer *writer = (ps_band_writer *)writer_; fz_output *out = writer->super.out; int w = writer->super.w; int h = writer->super.h; int n = writer->super.n; int alpha = writer->super.alpha; int xres = writer->super.xres; int yres = writer->super.yres; int pagenum = writer->super.pagenum; int w_points = (w * 72 + (xres>>1)) / xres; int h_points = (h * 72 + (yres>>1)) / yres; float sx = w/(float)w_points; float sy = h/(float)h_points; int err; if (alpha != 0) fz_throw(ctx, FZ_ERROR_GENERIC, "Postscript output cannot have alpha"); writer->super.w = w; writer->super.h = h; writer->super.n = n; err = deflateInit(&writer->stream, Z_DEFAULT_COMPRESSION); if (err != Z_OK) { fz_free(ctx, writer); fz_throw(ctx, FZ_ERROR_GENERIC, "compression error %d", err); } fz_printf(ctx, out, "%%%%Page: %d %d\n", pagenum, pagenum); fz_printf(ctx, out, "%%%%PageBoundingBox: 0 0 %d %d\n", w_points, h_points); fz_printf(ctx, out, "%%%%BeginPageSetup\n"); fz_printf(ctx, out, "<> setpagedevice\n", w_points, h_points); fz_printf(ctx, out, "%%%%EndPageSetup\n\n"); fz_printf(ctx, out, "/DataFile currentfile /FlateDecode filter def\n\n"); switch(n) { case 2: fz_printf(ctx, out, "/DeviceGray setcolorspace\n"); break; case 4: fz_printf(ctx, out, "/DeviceRGB setcolorspace\n"); break; case 5: fz_printf(ctx, out, "/DeviceCMYK setcolorspace\n"); break; default: fz_throw(ctx, FZ_ERROR_GENERIC, "Unexpected colorspace for ps output"); } fz_printf(ctx, out, "<<\n" "/ImageType 1\n" "/Width %d\n" "/Height %d\n" "/ImageMatrix [ %f 0 0 -%f 0 %d ]\n" "/MultipleDataSources false\n" "/DataSource DataFile\n" "/BitsPerComponent 8\n" //"/Decode [0 1]\n" "/Interpolate false\n" ">>\n" "image\n" , w, h, sx, sy, h); } static void ps_write_trailer(fz_context *ctx, fz_band_writer *writer_) { ps_band_writer *writer = (ps_band_writer *)writer_; fz_output *out = writer->super.out; int err; writer->stream.next_in = NULL; writer->stream.avail_in = 0; writer->stream.next_out = (Bytef*)writer->output; writer->stream.avail_out = (uInt)writer->output_size; err = deflate(&writer->stream, Z_FINISH); if (err != Z_STREAM_END) fz_throw(ctx, FZ_ERROR_GENERIC, "compression error %d", err); fz_write(ctx, out, writer->output, writer->output_size - writer->stream.avail_out); fz_printf(ctx, out, "\nshowpage\n%%%%PageTrailer\n%%%%EndPageTrailer\n\n"); } static void ps_drop_band_writer(fz_context *ctx, fz_band_writer *writer_) { ps_band_writer *writer = (ps_band_writer *)writer_; fz_free(ctx, writer->input); fz_free(ctx, writer->output); } void fz_write_pixmap_as_ps(fz_context *ctx, fz_output *out, const fz_pixmap *pixmap) { fz_band_writer *writer; fz_write_ps_file_header(ctx, out); writer = fz_new_ps_band_writer(ctx, out); fz_try(ctx) { fz_write_header(ctx, writer, pixmap->w, pixmap->h, pixmap->n, pixmap->alpha, pixmap->xres, pixmap->yres, 1); fz_write_band(ctx, writer, pixmap->stride, 0, pixmap->h, pixmap->samples); fz_write_trailer(ctx, writer); } fz_always(ctx) { fz_drop_band_writer(ctx, writer); } fz_catch(ctx) { fz_rethrow(ctx); } fz_write_ps_file_trailer(ctx, out, 1); } void fz_save_pixmap_as_ps(fz_context *ctx, fz_pixmap *pixmap, char *filename, int append) { fz_output *out = fz_new_output_with_path(ctx, filename, append); fz_try(ctx) fz_write_pixmap_as_ps(ctx, out, pixmap); fz_always(ctx) fz_drop_output(ctx, out); fz_catch(ctx) fz_rethrow(ctx); } static void ps_write_band(fz_context *ctx, fz_band_writer *writer_, int stride, int band_start, int band_height, const unsigned char *samples) { ps_band_writer *writer = (ps_band_writer *)writer_; fz_output *out = writer->super.out; int w = writer->super.w; int h = writer->super.h; int n = writer->super.n; int x, y, i, err; int required_input; int required_output; unsigned char *o; if (band_start+band_height >= h) band_height = h - band_start; required_input = w*(n-1)*band_height; required_output = (int)deflateBound(&writer->stream, required_input); if (writer->input == NULL || writer->input_size < required_input) { fz_free(ctx, writer->input); writer->input = NULL; writer->input = fz_malloc(ctx, required_input); writer->input_size = required_input; } if (writer->output == NULL || writer->output_size < required_output) { fz_free(ctx, writer->output); writer->output = NULL; writer->output = fz_malloc(ctx, required_output); writer->output_size = required_output; } o = writer->input; for (y = 0; y < band_height; y++) { for (x = 0; x < w; x++) { for (i = n-1; i > 0; i--) *o++ = *samples++; samples++; } samples += stride - w*n; } writer->stream.next_in = (Bytef*)writer->input; writer->stream.avail_in = required_input; writer->stream.next_out = (Bytef*)writer->output; writer->stream.avail_out = (uInt)writer->output_size; err = deflate(&writer->stream, Z_NO_FLUSH); if (err != Z_OK) fz_throw(ctx, FZ_ERROR_GENERIC, "compression error %d", err); fz_write(ctx, out, writer->output, writer->output_size - writer->stream.avail_out); } fz_band_writer *fz_new_ps_band_writer(fz_context *ctx, fz_output *out) { fz_band_writer *writer = fz_new_band_writer(ctx, fz_band_writer, out); writer->header = ps_write_header; writer->band = ps_write_band; writer->trailer = ps_write_trailer; writer->drop = ps_drop_band_writer; return writer; }