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