lib/pdf/pdf.cc

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