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