pdf.cc

   1 /*
   2  *      PaperJam -- Low-level handling of PDFs
   3  *
   4  *      (c) 2018 Martin Mares <mj@ucw.cz>
   5  */
   6
   7 #include <cassert>
   8 #include <cstdlib>
   9 #include <cstdio>
  10 #include <unistd.h>
  11 #include <sys/wait.h>
  12
  13 #include "jam.h"
  14
  15 #include <qpdf/QPDFWriter.hh>
  16
  17 static QPDF in_pdf;
  18 static QPDF out_pdf;
  19
  20 static void do_recalc_bbox(vector<page *> &pages, const char *in_name);
  21
  22 string out_context::new_resource(const string type)
  23 {
  24   return "/" + type + to_string(++res_cnt);
  25 }
  26
  27 class in_page : public page {
  28   QPDFObjectHandle pdf_page;
  29   QPDFObjectHandle xobject;
  30 public:
  31   BBox media_box;
  32   void render(out_context *out, pdf_matrix xform);
  33   in_page(QPDFObjectHandle inpg, int idx);
  34 };
  35
  36 in_page::in_page(QPDFObjectHandle inpg, int idx)
  37 {
  38   pdf_page = inpg;
  39   xobject = QPDFObjectHandle::newNull();
  40   index = idx;
  41
  42   media_box = BBox(inpg.getKey("/MediaBox"));
  43   width = media_box.width();
  44   height = media_box.height();
  45
  46   QPDFObjectHandle art_box = inpg.getKey("/ArtBox");
  47   if (art_box.isNull())
  48     art_box = inpg.getKey("/CropBox");
  49   if (art_box.isNull())
  50     image_box = BBox(width, height);
  51   else
  52     {
  53       image_box = BBox(art_box);
  54       image_box.x_min -= media_box.x_min;
  55       image_box.x_max -= media_box.x_min;
  56       image_box.y_min -= media_box.y_min;
  57       image_box.y_max -= media_box.y_min;
  58     }
  59 }
  60
  61 void in_page::render(out_context *out, pdf_matrix xform)
  62 {
  63   // Convert page to xobject
  64   if (xobject.isNull())
  65     xobject = out->pdf->makeIndirectObject( page_to_xobject(out->pdf, out->pdf->copyForeignObject(pdf_page)) );
  66   string xobj_res = out->new_resource("XO");
  67   out->xobjects.replaceKey(xobj_res, xobject);
  68
  69   pdf_matrix m;
  70   m.shift(-media_box.x_min, -media_box.y_min);
  71   m.concat(xform);
  72
  73   out->contents += "q " + m.to_string() + " cm " + xobj_res + " Do Q ";
  74 }
  75
  76 void debug_pages(vector<page *> &pages)
  77 {
  78   if (!debug_mode)
  79     return;
  80
  81   for (auto pg: pages)
  82     debug("Page #%d: media[%.3f %.3f] image[%.3f %.3f %.3f %.3f]",
  83       pg->index,
  84       pg->width, pg->height,
  85       pg->image_box.x_min, pg->image_box.y_min, pg->image_box.x_max, pg->image_box.y_max);
  86 }
  87
  88 vector<page *> run_command_list(list<cmd *> &cmds, vector<page *> &pages)
  89 {
  90   debug("# Input");
  91   debug_pages(pages);
  92
  93   for (auto c: cmds)
  94     {
  95       debug("# Executing %s", c->def->name);
  96       debug_indent += 4;
  97       try
  98         {
  99           pages = c->exec->process(pages);
 100         }
 101       catch (exception &e)
 102         {
 103           die("Error in %s: %s", c->def->name, e.what());
 104         }
 105       debug_indent -= 4;
 106       debug_pages(pages);
 107     }
 108
 109   return pages;
 110 }
 111
 112 void process(list<cmd *> &cmds)
 113 {
 114   in_pdf.processFile(in_name);
 115   in_pdf.pushInheritedAttributesToPage();
 116   out_pdf.emptyPDF();
 117
 118   vector<QPDFObjectHandle> const &in_pages = in_pdf.getAllPages();
 119   vector<page *> pages;
 120
 121   QPDFObjectHandle page_copy = out_pdf.copyForeignObject(in_pages[0]);
 122
 123   int cnt = 0;
 124   for (auto inpg: in_pages)
 125     pages.push_back(new in_page(inpg, ++cnt));
 126
 127   if (recalc_bbox)
 128     do_recalc_bbox(pages, in_name);
 129
 130   pages = run_command_list(cmds, pages);
 131
 132   for (auto pg: pages)
 133     {
 134       out_context out;
 135       out.pdf = &out_pdf;
 136       out.resources = QPDFObjectHandle::newDictionary();
 137       out.resources.replaceKey("/ProcSet", QPDFObjectHandle::parse("[/PDF]"));
 138       out.xobjects = QPDFObjectHandle::newDictionary();
 139       out.egstates = QPDFObjectHandle::newDictionary();
 140       pg->render(&out, pdf_matrix());
 141
 142       QPDFObjectHandle contents = QPDFObjectHandle::newStream(&out_pdf, out.contents);
 143
 144       // Create the page object
 145       QPDFObjectHandle out_page = out_pdf.makeIndirectObject(QPDFObjectHandle::newDictionary());
 146       out_page.replaceKey("/Type", QPDFObjectHandle::newName("/Page"));
 147       out_page.replaceKey("/MediaBox", BBox(pg->width, pg->height).to_array());
 148       // FIXME:
 149       // out_page.replaceKey("/CropBox", pg->image_box.to_array());
 150       out_page.replaceKey("/Contents", contents);
 151       if (!out.xobjects.getKeys().empty())
 152         out.resources.replaceKey("/XObject", out.xobjects);
 153       if (!out.egstates.getKeys().empty())
 154         out.resources.replaceKey("/ExtGState", out.egstates);
 155       out_page.replaceKey("/Resources", out.resources);
 156       out_pdf.addPage(out_page, false);
 157     }
 158
 159   // Produce info dictionary
 160   QPDFObjectHandle trailer = out_pdf.getTrailer();
 161   QPDFObjectHandle info = trailer.getKey("/Info");
 162   if (info.isNull())
 163     {
 164       info = QPDFObjectHandle::newDictionary();
 165       trailer.replaceKey("/Info", info);
 166     }
 167   else
 168     assert(info.isDictionary());
 169   // FIXME: More meta-data
 170   info.replaceKey("/Producer", unicode_string("PaperJam"));
 171
 172   // Write the output file
 173   QPDFWriter writer(out_pdf, out_name);
 174   writer.write();
 175 }
 176
 177 /*** Re-calculation of bboxes ***/
 178
 179 vector<BBox> gs_bboxes(const char *in)
 180 {
 181   int pipes[2];
 182   if (pipe(pipes) < 0)
 183     die("Cannot create pipe: %m");
 184
 185   pid_t pid = fork();
 186   if (pid < 0)
 187     die("Cannot fork: %m");
 188
 189   if (!pid)
 190     {
 191       close(pipes[0]);
 192       dup2(pipes[1], 1);
 193       dup2(pipes[1], 2);
 194       close(pipes[1]);
 195       execlp("gs", "gs", "-sDEVICE=bbox", "-dSAFER", "-dBATCH", "-dNOPAUSE", "-q", in, NULL);
 196       die("Cannot execute gs: %m");
 197     }
 198
 199   close(pipes[1]);
 200   FILE *f = fdopen(pipes[0], "r");
 201   if (!f)
 202     die("fdopen failed: %m");
 203
 204   char line[1024];
 205   vector<BBox> bboxes;
 206   while (fgets(line, sizeof(line), f))
 207     {
 208       char *eol = strchr(line, '\n');
 209       if (!eol)
 210         die("Ghostscript produced too long lines");
 211       *eol = 0;
 212
 213       if (!strncmp(line, "%%HiResBoundingBox: ", 20))
 214         {
 215           double x1, y1, x2, y2;
 216           if (sscanf(line+20, "%lf%lf%lf%lf", &x1, &y1, &x2, &y2) != 4)
 217             die("Cannot parse Ghostscript output: %s", line);
 218           bboxes.push_back(BBox(x1, y1, x2, y2));
 219         }
 220       else if (line[0] != '%')
 221         fprintf(stderr, "%s\n", line);
 222     }
 223   fclose(f);
 224
 225   int stat;
 226   if (waitpid(pid, &stat, 0) < 0)
 227     die("wait failed: %m");
 228   if (!WIFEXITED(stat) || WEXITSTATUS(stat))
 229     die("Ghostscript failed");
 230
 231   return bboxes;
 232 }
 233
 234 static void do_recalc_bbox(vector<page *> &pages, const char *in_name)
 235 {
 236   debug("Calling Ghostscript to re-calculate bounding boxes");
 237   vector<BBox> bboxes = gs_bboxes(in_name);
 238   if (pages.size() != bboxes.size())
 239     die("Ghostscript failed to produce the right number of bboxes");
 240
 241   for (size_t i=0; i<pages.size(); i++)
 242     pages[i]->image_box = bboxes[i];
 243 }