set pagerange before conversion (for speedup)
[swftools.git] / src / pdf2swf.c
1 /* pdf2swf.c
2    main routine for pdf2swf(1)
3
4    Part of the swftools package.
5    
6    Copyright (c) 2001,2002,2003 Matthias Kramm <kramm@quiss.org> 
7  
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 2 of the License, or
11    (at your option) any later version.
12
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17
18    You should have received a copy of the GNU General Public License
19    along with this program; if not, write to the Free Software
20    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
21
22 #include <stdlib.h>
23 #include <stdio.h>
24 #include <stdarg.h>
25 #include <string.h>
26 #include <unistd.h>
27 #include "../config.h"
28 #ifdef HAVE_DIRENT_H
29 #include <dirent.h>
30 #endif
31 #ifdef HAVE_SYS_STAT_H
32 #include <sys/stat.h>
33 #endif
34 #include "../lib/args.h"
35 #include "../lib/os.h"
36 #include "../lib/rfxswf.h"
37 #include "../lib/devices/swf.h"
38 #include "../lib/devices/arts.h"
39 #include "../lib/pdf/pdf.h"
40 #include "../lib/log.h"
41
42 #define SWFDIR concatPaths(getInstallationPath(), "swfs")
43
44 gfxsource_t*driver;
45
46 static char * outputname = 0;
47 static int loglevel = 3;
48 static char * pagerange = 0;
49 static char * filename = 0;
50 static char * password = 0;
51 static int zlib = 0;
52
53 static char * preloader = 0;
54 static char * viewer = 0;
55 static int xnup = 1;
56 static int ynup = 1;
57
58 static int info_only = 0;
59
60 static int flatten = 0;
61
62 char* fontpaths[256];
63 int fontpathpos = 0;
64
65 int move_x=0;
66 int move_y=0;
67 int custom_move = 0;
68 int clip_x1=0,clip_y1=0,clip_x2=0,clip_y2=0;
69 int custom_clip = 0;
70
71 static int system_quiet=0;
72
73 int systemf(const char* format, ...)
74 {
75     char buf[1024];
76     int ret;
77     va_list arglist;
78     va_start(arglist, format);
79     vsprintf(buf, format, arglist);
80     va_end(arglist);
81
82     if(!system_quiet) {
83         printf("%s\n", buf);
84         fflush(stdout);
85     }
86     ret = system(buf);
87     if(ret) {
88         fprintf(stderr, "system() returned %d\n", ret);
89         exit(ret);
90     }
91     return ret;
92 }
93
94 int args_callback_option(char*name,char*val) {
95     if (!strcmp(name, "o"))
96     {
97         outputname = val;
98         return 1;
99     }
100     else if (!strcmp(name, "v"))
101     {
102         loglevel ++;
103         setConsoleLogging(loglevel);
104         return 0;
105     }
106     else if (!strcmp(name, "2"))
107     {
108         xnup = 2;
109         ynup = 1;
110         return 0;
111     }
112     else if (!strcmp(name, "4"))
113     {
114         xnup = 2;
115         ynup = 2;
116         return 0;
117     }
118     else if (!strcmp(name, "9"))
119     {
120         xnup = 3;
121         ynup = 3;
122         return 0;
123     }
124     else if (!strcmp(name, "q"))
125     {
126         loglevel --;
127         setConsoleLogging(loglevel);
128         system_quiet = 1;
129         return 0;
130     }
131     else if (name[0]=='p')
132     {
133         /* check whether the page range follows the p directly, like 
134            in -p1,2 */
135         do {
136             name++;
137         } while(*name == 32 || *name == 13 || *name == 10 || *name == '\t');
138
139         if(*name) {
140             pagerange = name;
141             return 0;
142         } 
143         pagerange = val;        
144         return 1;
145     }
146     else if (!strcmp(name, "P"))
147     {
148         password = val;
149         return 1;
150     }
151     else if (!strcmp(name, "c"))
152     {
153         char*s = strdup(val);
154         char*x1 = strtok(s, ":");
155         char*y1 = strtok(0, ":");
156         char*x2 = strtok(0, ":");
157         char*y2 = strtok(0, ":");
158         if(!(x1 && y1 && x2 && y2)) {
159             fprintf(stderr, "-m option requires four arguments, <x1>:<y1>:<x2>:<y2>\n");
160             exit(1);
161         }
162         custom_clip = 1;
163         clip_x1 = atoi(x1);
164         clip_y1 = atoi(y1);
165         clip_x2 = atoi(x2);
166         clip_y2 = atoi(y2);
167         free(s);
168         return 1;
169     }
170     else if (!strcmp(name, "m"))
171     {
172         char*s = strdup(val);
173         char*c = strchr(s, ':');
174         if(!c) {
175             fprintf(stderr, "-m option requires two arguments, <x>:<y>\n");
176             exit(1);
177         }
178         *c = 0;
179         custom_move = 1;
180         move_x = atoi(val);
181         move_y = atoi(c+1);
182         free(s);
183         return 1;
184     }
185     else if (!strcmp(name, "s"))
186     {
187         char*s = strdup(val);
188         char*c = strchr(s, '=');
189         if(c && *c && c[1])  {
190             *c = 0;
191             c++;
192             driver->set_parameter(s,c);
193         }
194         else
195             driver->set_parameter(s,"1");
196         return 1;
197     }
198     else if (!strcmp(name, "S"))
199     {
200         driver->set_parameter("drawonlyshapes", "1");
201         return 0;
202     }
203     else if (!strcmp(name, "i"))
204     {
205         driver->set_parameter("ignoredraworder", "1");
206         return 0;
207     }
208     else if (!strcmp(name, "z"))
209     {
210         driver->set_parameter("enablezlib", "1");
211         zlib = 1;
212         return 0;
213     }
214     else if (!strcmp(name, "n"))
215     {
216         driver->set_parameter("opennewwindow", "1");
217         return 0;
218     }
219     else if (!strcmp(name, "I"))
220     {
221         info_only = 1;
222         return 0;
223     }
224     else if (!strcmp(name, "t"))
225     {
226         driver->set_parameter("insertstop", "1");
227         return 0;
228     }
229     else if (!strcmp(name, "T"))
230     {
231         if(!strcasecmp(val, "mx"))
232             driver->set_parameter("flashversion", "6");
233         else
234             driver->set_parameter("flashversion", val);
235
236         return 1;
237     }
238     else if (!strcmp(name, "f"))
239     {
240         driver->set_parameter("storeallcharacters", "1");
241         return 0;
242     }
243     else if (!strcmp(name, "w"))
244     {
245         driver->set_parameter("linksopennewwindow", "0");
246         return 0;
247     }
248     else if (!strcmp(name, "G"))
249     {
250         flatten = 1;
251         return 0;
252     }
253     else if (!strcmp(name, "F"))
254     {
255         char *s = strdup(val);
256         int l = strlen(s);
257         while(l && s[l-1]=='/') {
258             s[l-1] = 0;
259             l--;
260         }
261         fontpaths[fontpathpos++] = s;
262         return 1;
263     }
264     else if (!strcmp(name, "l"))
265     {
266         char buf[256];
267         sprintf(buf, "%s/default_loader.swf", SWFDIR);
268         preloader = strdup(buf);
269         return 0;
270     }
271     else if (!strcmp(name, "b"))
272     {
273         char buf[256];
274         sprintf(buf, "%s/default_viewer.swf", SWFDIR);
275         viewer = strdup(buf);
276         return 0;
277     }
278     else if (!strcmp(name, "L"))
279     {
280         if(val)
281         {
282             preloader = val;
283         }
284         else
285         {
286             systemf("ls %s/*_loader.swf", SWFDIR);
287             if(!system_quiet)
288                 printf("\n");
289             exit(1);
290         }
291         return 1;
292     }
293     else if (!strcmp(name, "B"))
294     {
295         if(val)
296         {
297             viewer = val;
298         }
299         else
300         {
301             systemf("ls %s/*_viewer.swf", SWFDIR);
302             if(!system_quiet)
303                 printf("\n");
304             exit(1);
305         }
306         return 1;
307     }
308     else if (!strcmp(name, "j"))
309     {
310         if(name[1]) {
311             driver->set_parameter("jpegquality", &name[1]);
312             return 0;
313         } else {
314             driver->set_parameter("jpegquality", val);
315             return 1;
316         }
317     }
318     else if (!strcmp(name, "V"))
319     {   
320         printf("pdf2swf - part of %s %s\n", PACKAGE, VERSION);
321         exit(0);
322     }
323     else 
324     {
325         fprintf(stderr, "Unknown option: -%s\n", name);
326         exit(1);
327     }
328     return 0;
329 }
330
331 /*struct docoptions_t options[] =
332 {{"o","output","filename::Specify output file"},
333  {"V","version","Print program version"},
334  {"i","ignore","Ignore draw order (makes the SWF file smaller, but may produce graphic errors)"},
335  {"z","zlib","Use Flash 6 (MX) zlib compression (Needs at least Flash 6 Plugin to play)"},
336  {"s","shapes","Don't use SWF Fonts, but store everything as shape"},
337  {"j","jpegquality","Set quality of embedded jpeg pictures (default: 85)"},
338  {"p","pages","Convert only pages in range. (E.g. 3-85)"},
339  {"w","samewindow","Don't open a new browser window for links in the SWF"},
340  {"f","fonts","Stroe full fonts in SWF. (Don't reduce to used characters)"},
341  {"F","fontpath","path::Add directory to font search path"},
342  {"B","viewer","name::Link viewer \"name\" to the pdf"},
343  {"L","preloader","file.swf::Link preloader \"file.swf\" to the pdf"},
344  {"b","defaultviewer","Link default viewer to the pdf"},
345  {"l","defaultpreloader","Link default preloader to the pdf"}
346  {0,0}
347 };*/
348 struct options_t options[] =
349 {{"o","output"},
350  {"q","quiet"},
351  {"V","version"},
352  {"i","ignore"},
353  {"I","info"},
354  {"z","zlib"},
355  {"s","set"},
356  {"S","shapes"},
357  {"j","jpegquality"},
358  {"p","pages"},
359  {"w","samewindow"},
360  {"f","fonts"},
361  {"F","fontdir"},
362  {"B","viewer"},
363  {"G","flatten"},
364  {"L","preloader"},
365  {"b","defaultviewer"},
366  {"l","defaultpreloader"},
367  {"t","stop"},
368  {"T","flashversion"},
369  {0,0}
370 };
371
372 int args_callback_longoption(char*name,char*val) {
373     return args_long2shortoption(options, name, val);
374 }
375
376 int args_callback_command(char*name, char*val) {
377     if (!filename) 
378         filename = name;
379     else {
380         if(outputname)
381         {
382              fprintf(stderr, "Error: Do you want the output to go to %s or to %s?", 
383                      outputname, name);
384              exit(1);
385         }
386         outputname = name;
387     }
388     return 0;
389 }
390
391 void args_callback_usage(char*name)
392 {
393     printf("Usage: %s [Options] input.pdf [-o output.swf]\n", name);
394     printf("\nBasic options:\n");
395     printf("-p  --pages=range          Convert only pages in range\n");
396     printf("-P  --password=password    Use password for deciphering the pdf\n");
397     printf("-v  --verbose              Be verbose. Use more than one -v for greater effect\n");
398     printf("-q  --quiet                Suppress normal messages. Use -qq to suppress warnings, also.\n");
399 #ifdef HAVE_DIRENT_H
400     printf("-F  --fontdir directory    Add directory to font search path\n");
401 #endif
402     printf("-V  --version              Print program version\n");
403     printf("\nEnhanced conversion options:\n");
404     printf("-S  --shapes               Don't use SWF Fonts, but store everything as shape\n");
405     printf("-z  --zlib                 Use Flash 6 (MX) zlib compression (Needs at least Flash 6 Plugin to play)\n");
406     printf("-w  --samewindow           Don't open a new Browser Window for Links in the SWF\n");
407     printf("-f  --fonts                Store full fonts in SWF. (Don't reduce to used characters)\n");
408     printf("-T  --flashversion=num     Set the flash version in the header to num (default: 4)\n");
409     printf("-s insertstop              Insert a \"Stop\" Tag in every frame (don't turn pages automatically)\n");
410     printf("-s zoom=factor             Scale result, default: 72\n");
411     printf("-s jpegquality=quality     Set quality of embedded jpeg pictures (default:85)\n");
412     printf("-s caplinewidth=value      Set the minimum line width to trigger cap style handling to value. (3)\n");
413     printf("-s splinequality=value     Set the quality of spline convertion to value (0-100, default: 100).\n");
414     printf("-s fontquality=value       Set the quality of font convertion to value (0-100, default: 100).\n");
415     printf("-s ignoredraworder         Ignore draw order (makes the SWF file smaller and faster, but may produce\n"
416            "                           graphic errors)\n");
417     printf("-s filloverlap             Make intersecting shapes overlap, instead of canceling each\n"
418            "                           other out. (Needed for some Powerpoint PDFs)\n");
419     //deliberately undocumented (for now)
420     //printf("-2                         Put 2 pages into each frame.\n");
421     //printf("-4                         Put 4 pages into each frame.\n");
422     printf("Postprocessing options:\n");
423     printf("-b  --defaultviewer        Link default viewer to the pdf (%s)\n", concatPaths(SWFDIR, "default_viewer.swf"));
424     printf("-l  --defaultpreloader     Link default preloader the pdf (%s)\n", concatPaths(SWFDIR, "default_loader.swf"));
425     printf("-B  --viewer=filename      Link viewer \"name\" to the pdf (\"%s -B\" for list)\n", name);
426     printf("-L  --preloader=filename   Link preloader \"name\" to the pdf (\"%s -L\" for list)\n",name);
427 }
428
429 float getRate(char*filename)
430 {
431     int fi;
432     SWF swf;
433     fi = open(filename,O_RDONLY|O_BINARY);
434     if(fi<0) { 
435         char buffer[256];
436         sprintf(buffer, "Couldn't open %s", filename);
437         perror(buffer);
438         exit(1);
439     }
440     if(swf_ReadSWF(fi,&swf) < 0)
441     { 
442         fprintf(stderr, "%s is not a valid SWF file or contains errors.\n",filename);
443         close(fi);
444         exit(1);
445     }
446     swf_FreeTags(&swf);
447     return swf.frameRate / 256.0;
448 }
449
450 void show_info(gfxsource_t*driver, char*filename)
451 {
452     gfxdocument_t* pdf = driver->open(filename);
453     int pagenr;
454     FILE*fo=0;
455     if(!pdf) {
456         msg("<error> Couldn't open %s", filename);
457         exit(1);
458     }
459     if(outputname) {
460         fo = fopen(outputname, "wb");
461         if(!fo) {
462             perror(outputname);exit(1);;
463         }
464     } else {
465         fo = stdout;
466     }
467
468     for(pagenr = 1; pagenr <= pdf->num_pages; pagenr++) 
469     {
470         gfxpage_t*page = pdf->getpage(pdf,pagenr);
471         if(is_in_range(pagenr, pagerange)) {
472             fprintf(fo, "page=%d width=%.2f height=%.2f\n", pagenr, page->width, page->height);
473         }
474     }
475     pdf->destroy(pdf);
476 }
477
478 int main(int argn, char *argv[])
479 {
480     int ret;
481     char buf[256];
482     int numfonts = 0;
483     int t;
484     char t1searchpath[1024];
485     int nup_pos = 0;
486     int x,y;
487     char* installPath = getInstallationPath();
488     char* fontdir = 0;
489     
490     initLog(0,-1,0,0,-1,loglevel);
491
492 #if defined(WIN32) && defined(HAVE_STAT) && defined(HAVE_SYS_STAT_H)
493     if(installPath) {
494         fontdir = concatPaths(installPath, "fonts");
495         FILE*test = fopen(concatPaths(fontdir,"\\d050000l.afm"), "rb");
496         if(!test) {
497             fprintf(stderr, "Couldn't find file %s - pdf2swf not installed properly? OS says:\n", concatPaths(fontdir, "\\d050000l.afm"));
498             perror("open");
499             exit(1);
500         }
501         fclose(test);
502     }
503 #else
504     fontdir = concatPaths(installPath, "fonts");
505 #endif
506
507 #ifdef HAVE_SRAND48
508     srand48(time(0));
509 #else
510 #ifdef HAVE_SRAND
511     srand(time(0));
512 #endif
513 #endif
514     driver = gfxsource_pdf_create();
515
516     processargs(argn, argv);
517     
518     if(!filename)
519     {
520         fprintf(stderr, "Please specify an input file\n");
521         exit(1);
522     }
523
524     if(info_only) {
525         show_info(driver, filename);
526         return 0;
527     }
528
529     if(!outputname)
530     {
531         if(filename) {
532             outputname = stripFilename(filename, ".swf");
533             msg("<notice> Output filename not given. Writing to %s", outputname);
534         } 
535     }
536         
537     if(!outputname)
538     {
539         fprintf(stderr, "Please use -o to specify an output file\n");
540         exit(1);
541     }
542
543     // test if the page range is o.k.
544     is_in_range(0x7fffffff, pagerange);
545
546     if(pagerange)
547         driver->set_parameter("pages", pagerange);
548
549     if (!filename) {
550         args_callback_usage(argv[0]);
551         exit(0);
552     }
553
554     /* add fonts */
555     if(fontdir) {
556         driver->set_parameter("fontdir", fontdir);
557     }
558     for(t=0;t<fontpathpos;t++) {
559         driver->set_parameter("fontdir", fontpaths[t]);
560     }
561
562     char fullname[256];
563     if(password && *password) {
564         sprintf(fullname, "%s|%s", filename, password);
565         filename = fullname;
566     }
567
568     gfxdocument_t* pdf = driver->open(filename);
569     if(!pdf) {
570         msg("<error> Couldn't open %s", filename);
571         exit(1);
572     }
573
574     gfxdevice_t swf,wrap;
575     gfxdevice_swf_init(&swf);
576     gfxdevice_t*out;
577     
578     if(flatten) {
579         gfxdevice_arts_init(&wrap, &swf);
580         out = &wrap;
581     } else {
582         out = &swf;
583     }
584
585     struct mypage_t {
586         int x;
587         int y;
588         gfxpage_t*page;
589     } pages[4];
590
591     int pagenum = 0;
592     int frame = 1;
593     int pagenr;
594     
595     for(pagenr = 1; pagenr <= pdf->num_pages; pagenr++) 
596     {
597         if(is_in_range(pagenr, pagerange)) {
598             char mapping[80];
599             sprintf(mapping, "%d:%d", pagenr, frame);
600             pdf->set_parameter(pdf, "pagemap", mapping);
601             pagenum++;
602         }
603         if(pagenum == xnup*ynup || (pagenr == pdf->num_pages && pagenum>1)) {
604             pagenum = 0;
605             frame++;
606         }
607     }
608
609     pagenum = 0;
610
611     for(pagenr = 1; pagenr <= pdf->num_pages; pagenr++) 
612     {
613         if(is_in_range(pagenr, pagerange)) {
614             gfxpage_t* page = pages[pagenum].page = pdf->getpage(pdf, pagenr);
615             pages[pagenum].x = 0;
616             pages[pagenum].y = 0;
617             pages[pagenum].page = page;
618             pagenum++;
619         }
620         if(pagenum == xnup*ynup || (pagenr == pdf->num_pages && pagenum>1)) {
621
622             int t;
623             int xmax[xnup], ymax[xnup];
624             int x,y;
625             int width=0, height=0;
626
627             memset(xmax, 0, xnup*sizeof(int));
628             memset(ymax, 0, ynup*sizeof(int));
629
630             for(y=0;y<ynup;y++)
631             for(x=0;x<xnup;x++) {
632                 int t = y*xnup + x;
633
634                 if(pages[t].page->width > xmax[x])
635                     xmax[x] = (int)pages[t].page->width;
636                 if(pages[t].page->height > ymax[y])
637                     ymax[y] = (int)pages[t].page->height;
638             }
639             for(x=0;x<xnup;x++) {
640                 width += xmax[x];
641                 xmax[x] = width;
642             }
643             for(y=0;y<ynup;y++) {
644                 height += ymax[y];
645                 ymax[y] = height;
646             }
647             if(custom_clip) {
648                 out->startpage(out,clip_x2 - clip_x1, clip_y2 - clip_y1);
649             } else {
650                 out->startpage(out,width,height);
651             }
652             for(t=0;t<pagenum;t++) {
653                 int x = t%xnup;
654                 int y = t/xnup;
655                 int xpos = x>0?xmax[x-1]:0;
656                 int ypos = y>0?ymax[y-1]:0;
657                 msg("<verbose> Render (%d,%d) move:%d/%d\n",
658                         (int)(pages[t].page->width + xpos),
659                         (int)(pages[t].page->height + ypos), xpos, ypos);
660                 pages[t].page->rendersection(pages[t].page, out, custom_move? move_x : xpos, 
661                                                            custom_move? move_y : ypos,
662                                                            custom_clip? clip_x1 : 0 + xpos, 
663                                                            custom_clip? clip_y1 : 0 + ypos, 
664                                                            custom_clip? clip_x2 : pages[t].page->width + xpos, 
665                                                            custom_clip? clip_y2 : pages[t].page->height + ypos);
666             }
667             out->endpage(out);
668             for(t=0;t<pagenum;t++)  {
669                 pages[t].page->destroy(pages[t].page);
670             }
671             pagenum = 0;
672         }
673     }
674     
675     gfxresult_t*result = out->finish(out);
676     if(result->save(result, outputname) < 0) {
677         exit(1);
678     }
679
680     int width = (int)result->get(result, "width");
681     int height = (int)result->get(result, "height");
682     msg("<notice> SWF written");
683     
684     result->destroy(result);
685
686     pdf->destroy(pdf);
687
688     char*zip = "";
689     if(zlib)
690         zip = "-z";
691 #undef SYSTEM_BACKTICKS
692     if(viewer && !preloader) {
693 #ifdef SYSTEM_BACKTICKS
694         systemf("swfcombine %s `swfdump -XY \"%s\"` \"%s\" viewport=\"%s\" -o \"%s\"",zip,
695                 viewer, outputname, outputname);
696 #else
697         systemf("swfcombine %s -X %d -Y %d \"%s\" viewport=\"%s\" -o \"%s\"",zip,width,height,
698                 viewer, outputname, outputname);
699 #endif
700         if(!system_quiet)
701             printf("\n");
702     }
703     if(preloader && !viewer) {
704         msg("<warning> --preloader option without --viewer option doesn't make very much sense.");
705 #ifdef SYSTEM_BACKTICKS
706         ret = systemf("swfcombine %s `swfdump -r \"%s\"` %s/PreLoaderTemplate.swf loader=\"%s\" movie=\"%s\" -o \"%s\"",zip,
707                 preloader, SWFDIR, preloader, outputname, outputname);
708 #else
709         ret = systemf("swfcombine %s -Y %d -X %d %s/PreLoaderTemplate.swf loader=\"%s\" movie=\"%s\" -o \"%s\"",zip,width,height,
710                 SWFDIR, preloader, outputname, outputname);
711 #endif
712         if(!system_quiet)
713             printf("\n");
714     }
715     if(preloader && viewer) {
716         systemf("swfcombine \"%s\" viewport=%s -o __tmp__.swf",
717                 viewer, outputname, outputname);
718 #ifdef SYSTEM_BACKTICKS
719         systemf("swfcombine %s `swfdump -XY \"%s\"` `swfdump -r \"%s\"` %s/PreLoaderTemplate.swf loader=%s movie=__tmp__.swf -o \"%s\"",zip,
720                 outputname, preloader, SWFDIR, preloader, outputname);
721 #else
722         /* TODO: read out rate */
723         systemf("swfcombine %s -X %d -Y %d -r %f %s/PreLoaderTemplate.swf loader=%s movie=__tmp__.swf -o \"%s\"",zip,width,height,
724                 getRate(preloader), preloader, SWFDIR, preloader, outputname);
725 #endif
726         systemf("rm __tmp__.swf");
727     }
728
729     return 0;
730 }
731