removed debug code
[swftools.git] / lib / pdf / pdf.cc
1 #include "../gfxdevice.h"
2 #include "../gfxsource.h"
3 #include "../devices/rescale.h"
4 #include "../log.h"
5 #include "config.h"
6 #include "GlobalParams.h"
7 #include "InfoOutputDev.h"
8 #include "GFXOutputDev.h"
9 #include "BitmapOutputDev.h"
10 #include "../mem.h"
11 #include "pdf.h"
12 #define NO_ARGPARSER
13 #include "../args.h"
14
15 static double zoom = 72; /* xpdf: 86 */
16 static int jpeg_dpi = 0;
17 static int ppm_dpi = 0;
18 static int multiply = 1;
19 static char* global_page_range = 0;
20
21 static parameter_t* device_config = 0;
22 static parameter_t* device_config_next = 0;
23
24 typedef struct _pdf_page_info
25 {
26     int xMin, yMin, xMax, yMax;
27     int width,height;
28     int number_of_images;
29     int number_of_links;
30     int number_of_fonts;
31     char has_info;
32 } pdf_page_info_t;
33
34 typedef struct _pdf_doc_internal
35 {
36     int protect;
37     PDFDoc*doc;
38     Object docinfo;
39     InfoOutputDev*info;
40     CommonOutputDev*outputDev;
41     pdf_page_info_t*pages;
42     gfxdevice_t* middev;
43 } pdf_doc_internal_t;
44
45 typedef struct _pdf_page_internal
46 {
47 } pdf_page_internal_t;
48
49 typedef struct _dev_output_internal
50 {
51     CommonOutputDev*outputDev;
52 } dev_output_internal_t;
53
54
55 static char* dirseparator()
56 {
57 #ifdef WIN32
58     return "\\";
59 #else
60     return "/";
61 #endif
62 }
63
64
65 void pdfpage_destroy(gfxpage_t*pdf_page)
66 {
67     pdf_page_internal_t*i= (pdf_page_internal_t*)pdf_page->internal;
68     free(pdf_page->internal);pdf_page->internal = 0;
69     free(pdf_page);pdf_page=0;
70 }
71
72 void render2(gfxpage_t*page, gfxdevice_t*dev)
73 {
74     pdf_doc_internal_t*pi = (pdf_doc_internal_t*)page->parent->internal;
75     
76     if(pi->middev) {
77         gfxdevice_rescale_setdevice(pi->middev, dev);
78         pi->middev->setparameter(pi->middev, "protect", "1");
79         dev = pi->middev;
80     } 
81     dev->setparameter(dev, "protect", "1");
82         
83     if(!pi) {
84         msg("<fatal> pdf_page_render: Parent PDF this page belongs to doesn't exist yet/anymore");
85         return;
86     }
87
88     if(!pi->pages[page->nr-1].has_info) {
89         msg("<fatal> pdf_page_render: page %d was previously set as not-to-render via the \"pages\" option", page->nr);
90         return;
91     }
92
93     pi->outputDev->setDevice(dev);
94
95     if(pi->protect) {
96         dev->setparameter(dev, "protect", "1");
97     }
98     
99     /* pass global parameters to output device */
100     parameter_t*p = device_config;
101     while(p) {
102         dev->setparameter(dev, p->name, p->value);
103         p = p->next;
104     }
105     pi->doc->displayPage((OutputDev*)pi->outputDev, page->nr, zoom*multiply, zoom*multiply, /*rotate*/0, true, true, /*doLinks*/(int)1);
106     pi->doc->processLinks((OutputDev*)pi->outputDev, page->nr);
107     pi->outputDev->setDevice(0);
108     if(pi->middev) {
109         gfxdevice_rescale_setdevice(pi->middev, 0x00000000);
110     }
111 }
112
113     
114 void pdfpage_render(gfxpage_t*page, gfxdevice_t*output)
115 {
116     pdf_doc_internal_t*pi = (pdf_doc_internal_t*)page->parent->internal;
117     pi->outputDev->setMove(0,0);
118     pi->outputDev->setClip(0,0,0,0);
119     render2(page, output);
120 }
121
122 void pdfpage_rendersection(gfxpage_t*page, gfxdevice_t*output, gfxcoord_t x, gfxcoord_t y, gfxcoord_t _x1, gfxcoord_t _y1, gfxcoord_t _x2, gfxcoord_t _y2)
123 {
124     pdf_doc_internal_t*pi = (pdf_doc_internal_t*)page->parent->internal;
125
126     int x1=(int)_x1,y1=(int)_y1,x2=(int)_x2,y2=(int)_y2;
127     if((x1|y1|x2|y2)==0) x2++;
128
129     pi->outputDev->setMove((int)x*multiply,(int)y*multiply);
130     pi->outputDev->setClip((int)x1*multiply,(int)y1*multiply,(int)x2*multiply,(int)y2*multiply);
131     render2(page, output);
132 }
133
134 static int globalparams_count=0;
135
136 void pdf_doc_destroy(gfxdocument_t*gfx)
137 {
138     pdf_doc_internal_t*i= (pdf_doc_internal_t*)gfx->internal;
139
140     if(i->outputDev) {
141         delete i->outputDev;i->outputDev=0;
142     }
143     if(i->middev) {
144         gfxdevice_rescale_setdevice(i->middev, 0x00000000);
145         i->middev->finish(i->middev);
146     }
147     delete i->doc; i->doc=0;
148     free(i->pages); i->pages = 0;
149
150     i->docinfo.free();
151     
152     if(i->info) {
153         delete i->info;i->info=0;
154     }
155
156     free(gfx->internal);gfx->internal=0;
157     free(gfx);gfx=0;
158
159     if(global_page_range) {
160         free(global_page_range);
161         global_page_range = 0;
162     }
163     
164     /*globalparams_count--;
165     if(!globalparams_count) {
166         delete globalParams;
167         globalParams = 0;
168         globalparams_count = 0;
169     }*/
170 }
171
172 void pdf_doc_set_parameter(gfxdocument_t*gfx, const char*name, const char*value)
173 {
174     pdf_doc_internal_t*i= (pdf_doc_internal_t*)gfx->internal;
175     CommonOutputDev*o = i->outputDev;
176     if(!strcmp(name, "pagemap")) {
177         int pdfpage=0, outputpage=0;
178         sscanf(value,"%d:%d", &pdfpage, &outputpage);
179         o->preparePage(pdfpage, outputpage);
180     } else {
181         o->setParameter(name, value);
182     }
183 }
184
185 gfxpage_t* pdf_doc_getpage(gfxdocument_t*doc, int page)
186 {
187     pdf_doc_internal_t*di= (pdf_doc_internal_t*)doc->internal;
188
189     if(page < 1 || page > doc->num_pages)
190         return 0;
191     
192     gfxpage_t* pdf_page = (gfxpage_t*)malloc(sizeof(gfxpage_t));
193     pdf_page_internal_t*pi= (pdf_page_internal_t*)malloc(sizeof(pdf_page_internal_t));
194     memset(pi, 0, sizeof(pdf_page_internal_t));
195     pdf_page->internal = pi;
196
197     pdf_page->destroy = pdfpage_destroy;
198     pdf_page->render = pdfpage_render;
199     pdf_page->rendersection = pdfpage_rendersection;
200     pdf_page->width = di->pages[page-1].width;
201     pdf_page->height = di->pages[page-1].height;
202
203     pdf_page->parent = doc;
204     pdf_page->nr = page;
205     return pdf_page;
206 }
207
208 static char*getInfoString(Dict *infoDict, char *key)
209 {
210     Object obj;
211     GString *s1, *s2;
212     int i;
213
214     if (infoDict && infoDict->lookup(key, &obj)->isString()) {
215         s1 = obj.getString();
216         if ((s1->getChar(0) & 0xff) == 0xfe &&
217             (s1->getChar(1) & 0xff) == 0xff) {
218             s2 = new GString();
219             for (i = 2; i < obj.getString()->getLength(); i += 2) {
220               if (s1->getChar(i) == '\0') {
221                 s2->append(s1->getChar(i+1));
222               } else {
223                 delete s2;
224                 s2 = new GString("<unicode>");
225                 break;
226               }
227             }
228             char*ret = strdup(s2->getCString());
229             delete s2;
230             obj.free();
231             return ret;
232         } else {
233             char*ret = strdup(s1->getCString());
234             obj.free();
235             return ret;
236         }
237     }
238     return strdup("");
239 }
240
241 static char*getInfoDate(Dict *infoDict, char *key) 
242 {
243     Object obj;
244     char *s;
245
246     if (infoDict && infoDict->lookup(key, &obj)->isString()) {
247         s = obj.getString()->getCString();
248         if (s[0] == 'D' && s[1] == ':') {
249           s += 2;
250         }
251         char*ret = strdup(s);
252         obj.free();
253         return ret;
254     }
255     return strdup("");
256 }
257
258 char* pdf_doc_getinfo(gfxdocument_t*doc, const char*name)
259 {
260     pdf_doc_internal_t*i= (pdf_doc_internal_t*)doc->internal;
261     if(!strcmp(name, "title")) return getInfoString(i->docinfo.getDict(), "Title");
262     else if(!strcmp(name, "subject")) return getInfoString(i->docinfo.getDict(), "Subject");
263     else if(!strcmp(name, "keywords")) return getInfoString(i->docinfo.getDict(), "Keywords");
264     else if(!strcmp(name, "author")) return getInfoString(i->docinfo.getDict(), "Author");
265     else if(!strcmp(name, "creator")) return getInfoString(i->docinfo.getDict(), "Creator");
266     else if(!strcmp(name, "producer")) return getInfoString(i->docinfo.getDict(), "Producer");
267     else if(!strcmp(name, "creationdate")) return getInfoDate(i->docinfo.getDict(), "CreationDate");
268     else if(!strcmp(name, "moddate")) return getInfoDate(i->docinfo.getDict(), "ModDate");
269     else if(!strcmp(name, "linearized")) return strdup(i->doc->isLinearized() ? "yes" : "no");
270     else if(!strcmp(name, "tagged")) return strdup(i->doc->getStructTreeRoot()->isDict() ? "yes" : "no");
271     else if(!strcmp(name, "encrypted")) return strdup(i->doc->isEncrypted() ? "yes" : "no");
272     else if(!strcmp(name, "oktoprint")) return strdup(i->doc->okToPrint() ? "yes" : "no");
273     else if(!strcmp(name, "oktocopy")) return strdup(i->doc->okToCopy() ? "yes" : "no");
274     else if(!strcmp(name, "oktochange")) return strdup(i->doc->okToChange() ? "yes" : "no");
275     else if(!strcmp(name, "oktoaddnotes")) return strdup(i->doc->okToAddNotes() ? "yes" : "no");
276     else if(!strcmp(name, "version")) { 
277         char buf[32];
278         sprintf(buf, "%.1f", i->doc->getPDFVersion());
279         return strdup(buf);
280     }
281     return 0;
282 }
283
284
285 static void storeDeviceParameter(const char*name, const char*value)
286 {
287     parameter_t*p = new parameter_t();
288     p->name = strdup(name);
289     p->value = strdup(value);
290     p->next = 0;
291     if(device_config_next) {
292         device_config_next->next = p;
293         device_config_next = p;
294     } else {
295         device_config = p;
296         device_config_next = p;
297     }
298 }
299
300 typedef struct _gfxsource_internal
301 {
302     int config_bitmap_optimizing;
303 } gfxsource_internal_t;
304
305 static void pdf_set_parameter(gfxsource_t*src, const char*name, const char*value)
306 {
307     gfxsource_internal_t*i = (gfxsource_internal_t*)src->internal;
308     msg("<verbose> setting parameter %s to \"%s\"", name, value);
309     if(!strncmp(name, "fontdir", strlen("fontdir"))) {
310         addGlobalFontDir(value);
311     } else if(!strcmp(name, "pages")) {
312         global_page_range = strdup(value);
313     } else if(!strncmp(name, "font", strlen("font"))) {
314         addGlobalFont(value);
315     } else if(!strncmp(name, "languagedir", strlen("languagedir"))) {
316         addGlobalLanguageDir(value);
317     } else if(!strcmp(name, "zoom")) {
318         char buf[80];
319         zoom = atof(value);
320         sprintf(buf, "%f", (double)jpeg_dpi/(double)zoom);
321         storeDeviceParameter("jpegsubpixels", buf);
322         sprintf(buf, "%f", (double)ppm_dpi/(double)zoom);
323         storeDeviceParameter("ppmsubpixels", buf);
324     } else if(!strcmp(name, "jpegdpi")) {
325         char buf[80];
326         jpeg_dpi = atoi(value);
327         sprintf(buf, "%f", (double)jpeg_dpi/(double)zoom);
328         storeDeviceParameter("jpegsubpixels", buf);
329     } else if(!strcmp(name, "ppmdpi")) {
330         char buf[80];
331         ppm_dpi = atoi(value);
332         sprintf(buf, "%f", (double)ppm_dpi/(double)zoom);
333         storeDeviceParameter("ppmsubpixels", buf);
334     } else if(!strcmp(name, "poly2bitmap")) {
335         i->config_bitmap_optimizing = 1;
336     } else if(!strcmp(name, "multiply")) {
337         multiply = atoi(value);
338     } else if(!strcmp(name, "help")) {
339         printf("\nPDF device global parameters:\n");
340         printf("fontdir=<dir>   a directory with additional fonts\n");
341         printf("font=<filename> an dditional font filename\n");
342         printf("pages=<range>   the range of pages to convert (example: pages=1-100,210-)\n");
343         printf("zoom=<dpi>      the resultion (default: 72)\n");
344     }   
345     storeDeviceParameter(name,value);
346 }
347
348 static gfxdocument_t*pdf_open(gfxsource_t*src, const char*filename)
349 {
350     gfxsource_internal_t*isrc = (gfxsource_internal_t*)src->internal;
351     gfxdocument_t*pdf_doc = (gfxdocument_t*)malloc(sizeof(gfxdocument_t));
352     memset(pdf_doc, 0, sizeof(gfxdocument_t));
353     pdf_doc_internal_t*i= (pdf_doc_internal_t*)malloc(sizeof(pdf_doc_internal_t));
354     memset(i, 0, sizeof(pdf_doc_internal_t));
355     pdf_doc->internal = i;
356     char*userPassword=0;
357     
358     filename = strdup(filename);
359
360     char*x = 0;
361     if((x = strchr(filename, '|'))) {
362         *x = 0;
363         userPassword = x+1;
364     }
365     
366     GString *fileName = new GString(filename);
367     GString *userPW;
368
369     // open PDF file
370     if (userPassword && userPassword[0]) {
371       userPW = new GString(userPassword);
372     } else {
373       userPW = NULL;
374     }
375     i->doc = new PDFDoc(fileName, userPW);
376     if (userPW) {
377       delete userPW;
378     }
379     if (!i->doc->isOk()) {
380         printf("xpdf reports document as broken.\n");
381         return 0;
382     }
383
384     // get doc info
385     i->doc->getDocInfo(&i->docinfo);
386     
387     pdf_doc->num_pages = i->doc->getNumPages();
388     i->protect = 0;
389     if (i->doc->isEncrypted()) {
390           if(!i->doc->okToCopy()) {
391               printf("PDF disallows copying.\n");
392               return 0;
393           }
394           if(!i->doc->okToChange() || !i->doc->okToAddNotes())
395               i->protect = 1;
396     }
397
398     i->info = new InfoOutputDev(i->doc->getXRef());
399     int t;
400     i->pages = (pdf_page_info_t*)malloc(sizeof(pdf_page_info_t)*pdf_doc->num_pages);
401     memset(i->pages,0,sizeof(pdf_page_info_t)*pdf_doc->num_pages);
402     for(t=1;t<=pdf_doc->num_pages;t++) {
403         if(!global_page_range || is_in_range(t, global_page_range)) {
404             i->doc->displayPage((OutputDev*)i->info, t, zoom, zoom, /*rotate*/0, /*usemediabox*/true, /*crop*/true, /*doLinks*/(int)1);
405             i->doc->processLinks((OutputDev*)i->info, t);
406             i->pages[t-1].xMin = i->info->x1;
407             i->pages[t-1].yMin = i->info->y1;
408             i->pages[t-1].xMax = i->info->x2;
409             i->pages[t-1].yMax = i->info->y2;
410             i->pages[t-1].width = i->info->x2 - i->info->x1;
411             i->pages[t-1].height = i->info->y2 - i->info->y1;
412             i->pages[t-1].number_of_images = i->info->num_images;
413             i->pages[t-1].number_of_links = i->info->num_links;
414             i->pages[t-1].number_of_fonts = i->info->num_fonts;
415             i->pages[t-1].has_info = 1;
416         }
417     }
418
419     if(isrc->config_bitmap_optimizing) {
420         BitmapOutputDev*outputDev = new BitmapOutputDev(i->info, i->doc);
421         i->outputDev = (CommonOutputDev*)outputDev;
422     } else {
423         GFXOutputDev*outputDev = new GFXOutputDev(i->info, i->doc);
424         i->outputDev = (CommonOutputDev*)outputDev;
425     }
426
427     /* pass global parameters to PDF driver*/
428     parameter_t*p = device_config;
429     while(p) {
430         i->outputDev->setParameter(p->name, p->value);
431         p = p->next;
432     }
433
434     i->middev = 0;
435     if(multiply>1) {
436         i->middev = (gfxdevice_t*)malloc(sizeof(gfxdevice_t));
437         gfxdevice_rescale_init(i->middev, 0x00000000, 0, 0, 1.0 / multiply);
438     }
439
440     pdf_doc->get = 0;
441     pdf_doc->destroy = pdf_doc_destroy;
442     pdf_doc->set_parameter = pdf_doc_set_parameter;
443     pdf_doc->getinfo = pdf_doc_getinfo;
444     pdf_doc->getpage = pdf_doc_getpage;
445
446
447     return pdf_doc;
448
449 }
450
451 gfxsource_t*gfxsource_pdf_create()
452 {
453     gfxsource_t*src = (gfxsource_t*)malloc(sizeof(gfxsource_t));
454     memset(src, 0, sizeof(gfxsource_t));
455     src->set_parameter = pdf_set_parameter;
456     src->open = pdf_open;
457     src->internal = malloc(sizeof(gfxsource_internal_t));
458     memset(src->internal, 0, sizeof(gfxsource_internal_t));
459
460     if(!globalParams) {
461         globalParams = new GFXGlobalParams();
462         //globalparams_count++;
463     }
464     
465
466     return src;
467 }