]> mj.ucw.cz Git - paperjam.git/blob - pdf.cc
Shuffling code around
[paperjam.git] / 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     bbox = BBox(width, height);
51   else
52     {
53       bbox = BBox(art_box);
54       bbox.x_min -= media_box.x_min;
55       bbox.x_max -= media_box.x_min;
56       bbox.y_min -= media_box.y_min;
57       bbox.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] bbox[%.3f %.3f %.3f %.3f]",
83       pg->index,
84       pg->width, pg->height,
85       pg->bbox.x_min, pg->bbox.y_min, pg->bbox.x_max, pg->bbox.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       pages = c->exec->process(pages);
98       debug_indent -= 4;
99       debug_pages(pages);
100     }
101
102   return pages;
103 }
104
105 void process(list<cmd *> &cmds)
106 {
107   in_pdf.processFile(in_name);
108   in_pdf.pushInheritedAttributesToPage();
109   out_pdf.emptyPDF();
110
111   vector<QPDFObjectHandle> const &in_pages = in_pdf.getAllPages();
112   vector<page *> pages;
113
114   QPDFObjectHandle page_copy = out_pdf.copyForeignObject(in_pages[0]);
115
116   int cnt = 0;
117   for (auto inpg: in_pages)
118     pages.push_back(new in_page(inpg, ++cnt));
119
120   if (recalc_bbox)
121     do_recalc_bbox(pages, in_name);
122
123   pages = run_command_list(cmds, pages);
124
125   for (auto pg: pages)
126     {
127       out_context out;
128       out.resources = QPDFObjectHandle::newDictionary();
129       out.resources.replaceKey("/ProcSet", QPDFObjectHandle::parse("[/PDF]"));
130       out.xobjects = QPDFObjectHandle::newDictionary();
131       out.resources.replaceKey("/XObject", out.xobjects);
132       pg->render(&out, pdf_matrix());
133
134       QPDFObjectHandle contents = QPDFObjectHandle::newStream(&out_pdf, out.contents);
135
136       // Create the page object
137       QPDFObjectHandle out_page = out_pdf.makeIndirectObject(QPDFObjectHandle::newDictionary());
138       out_page.replaceKey("/Type", QPDFObjectHandle::newName("/Page"));
139       out_page.replaceKey("/MediaBox", BBox(pg->width, pg->height).to_array());
140       // FIXME:
141       // out_page.replaceKey("/CropBox", pg->bbox.to_array());
142       out_page.replaceKey("/Contents", contents);
143       out_page.replaceKey("/Resources", out.resources);
144       out_pdf.addPage(out_page, false);
145     }
146
147   // Produce info dictionary
148   QPDFObjectHandle trailer = out_pdf.getTrailer();
149   QPDFObjectHandle info = trailer.getKey("/Info");
150   if (info.isNull())
151     {
152       info = QPDFObjectHandle::newDictionary();
153       trailer.replaceKey("/Info", info);
154     }
155   else
156     assert(info.isDictionary());
157   // FIXME: More meta-data
158   info.replaceKey("/Producer", unicode_string("PaperJam"));
159
160   // Write the output file
161   QPDFWriter writer(out_pdf, out_name);
162   writer.write();
163 }
164
165 /*** Re-calculation of bboxes ***/
166
167 vector<BBox> gs_bboxes(const char *in)
168 {
169   int pipes[2];
170   if (pipe(pipes) < 0)
171     die("Cannot create pipe: %m");
172
173   pid_t pid = fork();
174   if (pid < 0)
175     die("Cannot fork: %m");
176
177   if (!pid)
178     {
179       close(pipes[0]);
180       dup2(pipes[1], 1);
181       dup2(pipes[1], 2);
182       close(pipes[1]);
183       execlp("gs", "gs", "-sDEVICE=bbox", "-dSAFER", "-dBATCH", "-dNOPAUSE", "-q", in, NULL);
184       die("Cannot execute gs: %m");
185     }
186
187   close(pipes[1]);
188   FILE *f = fdopen(pipes[0], "r");
189   if (!f)
190     die("fdopen failed: %m");
191
192   char line[1024];
193   vector<BBox> bboxes;
194   while (fgets(line, sizeof(line), f))
195     {
196       char *eol = strchr(line, '\n');
197       if (!eol)
198         die("Ghostscript produced too long lines");
199       *eol = 0;
200
201       if (!strncmp(line, "%%HiResBoundingBox: ", 20))
202         {
203           double x1, y1, x2, y2;
204           if (sscanf(line+20, "%lf%lf%lf%lf", &x1, &y1, &x2, &y2) != 4)
205             die("Cannot parse Ghostscript output: %s", line);
206           bboxes.push_back(BBox(x1, y1, x2, y2));
207         }
208       else if (line[0] != '%')
209         fprintf(stderr, "%s\n", line);
210     }
211   fclose(f);
212
213   int stat;
214   if (waitpid(pid, &stat, 0) < 0)
215     die("wait failed: %m");
216   if (!WIFEXITED(stat) || WEXITSTATUS(stat))
217     die("Ghostscript failed");
218
219   return bboxes;
220 }
221
222 static void do_recalc_bbox(vector<page *> &pages, const char *in_name)
223 {
224   vector<BBox> bboxes = gs_bboxes(in_name);
225   if (pages.size() != bboxes.size())
226     die("Ghostscript failed to produce the right number of bboxes");
227
228   for (size_t i=0; i<pages.size(); i++)
229     pages[i]->bbox = bboxes[i];
230 }