e8ddfdb88ad6d72add3bd20d4f66782d108e3a87
[swftools.git] / pdf2swf / pdf2swf.cc
1 /* pdf2swf.cc
2    main routine for pdf2swf(1)
3
4    Part of the swftools package.
5    
6    Copyright (c) 2001 Matthias Kramm <kramm@quiss.org> 
7
8    This file is distributed under the GPL, see file COPYING for details */
9
10 #include <stdlib.h>
11 #include <stdio.h>
12 #include <stdarg.h>
13 #include <string.h>
14 #include <unistd.h>
15 #include "../config.h"
16 #ifdef HAVE_DIRENT_H
17 #include <dirent.h>
18 #endif
19 #include "../lib/args.h"
20 #include "SWFOutputDev.h"
21 #include "t1lib.h"
22 extern "C" {
23 #include "log.h"
24 }
25
26 static char * outputname = 0;
27 static int loglevel = 3;
28 static char * pagerange = 0;
29 static char * filename = 0;
30 static char * password = 0;
31
32 static char * preloader = 0;
33 static char * viewer = 0;
34
35 char* fontpaths[256];
36 int fontpathpos = 0;
37
38 static int system_quiet=0;
39
40 int systemf(const char* format, ...)
41 {
42     char buf[1024];
43     int ret;
44     va_list arglist;
45     va_start(arglist, format);
46     vsprintf(buf, format, arglist);
47     va_end(arglist);
48
49     if(!system_quiet) {
50         printf("%s\n", buf);
51         fflush(stdout);
52     }
53     ret = system(buf);
54     if(ret) {
55         fprintf(stderr, "system() returned %d\n", ret);
56         exit(ret);
57     }
58     return ret;
59 }
60
61 int args_callback_option(char*name,char*val) {
62     if (!strcmp(name, "o"))
63     {
64         outputname = val;
65         return 1;
66     }
67     else if (!strcmp(name, "v"))
68     {
69         loglevel ++;
70         return 0;
71     }
72     else if (!strcmp(name, "q"))
73     {
74         loglevel --;
75         system_quiet = 1;
76         return 0;
77     }
78     else if (name[0]=='p')
79     {
80         /* check whether the page range follows the p directly, like 
81            in -p1,2 */
82         do {
83             name++;
84         } while(*name == 32 || *name == 13 || *name == 10 || *name == '\t');
85
86         if(*name) {
87             pagerange = name;
88             return 0;
89         } 
90         pagerange = val;        
91         return 1;
92     }
93     else if (!strcmp(name, "P"))
94     {
95         password = val;
96         return 1;
97     }
98     else if (!strcmp(name, "s"))
99     {
100         pdfswf_drawonlyshapes();
101         return 0;
102     }
103     else if (!strcmp(name, "i"))
104     {
105         pdfswf_ignoredraworder();
106         return 0;
107     }
108     else if (!strcmp(name, "z"))
109     {
110         pdfswf_enablezlib();
111         return 0;
112     }
113     else if (!strcmp(name, "n"))
114     {
115         pdfswf_linksopennewwindow();
116         return 0;
117     }
118     else if (!strcmp(name, "t"))
119     {
120         pdfswf_insertstop();
121         return 0;
122     }
123     else if (!strcmp(name, "T"))
124     {
125         int i = atoi(val);
126         if(!strcasecmp(val, "mx"))
127             i = 6;
128         pdfswf_setversion(i);
129         return 1;
130     }
131     else if (!strcmp(name, "f"))
132     {
133         pdfswf_storeallcharacters();
134         return 0;
135     }
136     else if (!strcmp(name, "F"))
137     {
138         char *s = strdup(val);
139         int l = strlen(s);
140         while(l && s[l-1]=='/') {
141             s[l-1] = 0;
142             l--;
143         }
144         fontpaths[fontpathpos++] = s;
145         return 1;
146     }
147     else if (!strcmp(name, "l"))
148     {
149         char buf[256];
150         sprintf(buf, "%s/swfs/default_loader.swf", SWFTOOLS_DATADIR);
151         preloader = strdup(buf);
152         return 0;
153     }
154     else if (!strcmp(name, "b"))
155     {
156         char buf[256];
157         sprintf(buf, "%s/swfs/default_viewer.swf", SWFTOOLS_DATADIR);
158         viewer = strdup(buf);
159         return 0;
160     }
161     else if (!strcmp(name, "L"))
162     {
163         if(val)
164         {
165             preloader = val;
166         }
167         else
168         {
169             systemf("ls %s/swfs/*_loader.swf", SWFTOOLS_DATADIR);
170             if(!system_quiet)
171                 printf("\n");
172             exit(1);
173         }
174         return 1;
175     }
176     else if (!strcmp(name, "B"))
177     {
178         if(val)
179         {
180             viewer = val;
181         }
182         else
183         {
184             systemf("ls %s/swfs/*_viewer.swf", SWFTOOLS_DATADIR);
185             if(!system_quiet)
186                 printf("\n");
187             exit(1);
188         }
189         return 1;
190     }
191     else if (!strcmp(name, "j"))
192     {
193         if(name[1]) {
194             pdfswf_jpegquality(atoi(&name[1]));
195             return 0;
196         } else {
197             pdfswf_jpegquality(atoi(val));
198             return 1;
199         }
200     }
201     else if (!strcmp(name, "V"))
202     {   
203         printf("pdf2swf - part of %s %s\n", PACKAGE, VERSION);
204         exit(0);
205     }
206     else 
207     {
208         fprintf(stderr, "Unknown option: -%s\n", name);
209         exit(1);
210     }
211     return 0;
212 }
213
214 /*struct docoptions_t options[] =
215 {{"o","output","filename::Specify output file"},
216  {"V","version","Print program version"},
217  {"i","ignore","Ignore draw order (makes the SWF file smaller, but may produce graphic errors)"},
218  {"z","zlib","Use Flash 6 (MX) zlib compression (Needs at least Flash 6 Plugin to play)"},
219  {"s","shapes","Don't use SWF Fonts, but store everything as shape"},
220  {"j","jpegquality","Set quality of embedded jpeg pictures (default: 85)"},
221  {"p","pages","Convert only pages in range. (E.g. 3-85)"},
222  {"w","samewindow","Don't open a new browser window for links in the SWF"},
223  {"f","fonts","Stroe full fonts in SWF. (Don't reduce to used characters)"},
224  {"F","fontpath","path::Add directory to font search path"},
225  {"B","viewer","name::Link viewer \"name\" to the pdf"},
226  {"L","preloader","file.swf::Link preloader \"file.swf\" to the pdf"},
227  {"b","defaultviewer","Link default viewer to the pdf"},
228  {"l","defaultpreloader","Link default preloader to the pdf"}
229  {0,0}
230 };*/
231 struct options_t options[] =
232 {{"o","output"},
233  {"q","quiet"},
234  {"V","version"},
235  {"i","ignore"},
236  {"z","zlib"},
237  {"s","shapes"},
238  {"j","jpegquality"},
239  {"p","pages"},
240  {"w","samewindow"},
241  {"f","fonts"},
242  {"F","fontdir"},
243  {"B","viewer"},
244  {"L","preloader"},
245  {"b","defaultviewer"},
246  {"l","defaultpreloader"},
247  {"t","stop"},
248  {"T","flashversion"},
249  {0,0}
250 };
251
252 int args_callback_longoption(char*name,char*val) {
253     return args_long2shortoption(options, name, val);
254 }
255
256 int args_callback_command(char*name, char*val) {
257     if (!filename) 
258         filename = name;
259     else {
260         if(outputname)
261         {
262              fprintf(stderr, "Error: Do you want the output to go to %s or to %s?", 
263                      outputname, name);
264              exit(1);
265         }
266         outputname = name;
267     }
268     return 0;
269 }
270
271 void args_callback_usage(char*name)
272 {
273     printf("Usage: %s [-si] [-j quality] [-p range] [-P password] input.pdf [-o output.swf]\n", name);
274     printf("\n");
275     printf("-p  --pages=range          Convert only pages in range\n");
276     printf("-P  --password=password    Use password for deciphering the pdf\n");
277     printf("-s  --shapes               Don't use SWF Fonts, but store everything as shape\n");
278     printf("-i  --ignore               Ignore draw order (makes the SWF file smaller, but may produce\n");
279     printf("                           graphic errors)\n");
280     printf("-z  --zlib                 Use Flash 6 (MX) zlib compression (Needs at least Flash 6 Plugin to play)\n");
281     printf("-j  --jpegquality=quality  Set quality of embedded jpeg pictures (default:85)\n");
282     printf("-v  --verbose              Be verbose. Use more than one -v for greater effect\n");
283     printf("-q  --quiet                Suppress normal messages. Use -qq to suppress warnings, also.\n");
284     printf("-w  --samewindow           Don't open a new Browser Window for Links in the SWF\n");
285 #ifdef HAVE_DIRENT_H
286     printf("-F  --fontdir directory    Add directory to font search path\n");
287 #endif
288     printf("-f  --fonts                Store full fonts in SWF. (Don't reduce to used characters)\n");
289     printf("-V  --version              Print program version\n");
290     printf("-t  --stop                 Insert a \"Stop\" Tag in every frame (don't turn pages automatically)\n");
291     printf("-T  --flashversion=num     Set the flash version in the header to num (default: 4)\n");
292 #ifndef SYSTEM_BACKTICKS
293     printf("The following might not work because your system call doesn't support command substitution:\n");
294 #endif
295     printf("-b  --defaultviewer        Link default viewer to the pdf (%s/swfs/default_viewer.swf)\n", SWFTOOLS_DATADIR);
296     printf("-l  --defaultpreloader     Link default preloader the pdf (%s/swfs/default_loader.swf)\n", SWFTOOLS_DATADIR);
297     printf("-B  --viewer=filename      Link viewer \"name\" to the pdf (\"%s -B\" for list)\n", name);
298     printf("-L  --preloader=filename   Link preloader \"name\" to the pdf (\"%s -L\" for list)\n",name);
299 }
300
301 #ifdef HAVE_DIRENT_H
302 void addfontdir(FILE*database, char* dirname, int*numfonts, char*searchpath) 
303 {
304     if(searchpath) {
305         if(searchpath[0])
306             strcat(searchpath, ":");
307         strcat(searchpath, dirname);
308     }
309     logf("<verbose> Adding %s to search path\n", dirname);
310
311     DIR*dir = opendir(dirname);
312     if(!dir) {
313         logf("<warning> Couldn't open directory %s\n", dirname);
314         return;
315     }
316     dirent*ent;
317     while(1) {
318         ent = readdir (dir);
319         if (!ent) 
320             break;
321         int l;
322         char*name = ent->d_name;
323         char type = 0;
324         if(!name) continue;
325         l=strlen(name);
326         if(l<4)
327             continue;
328         if(!strncasecmp(&name[l-4], ".afm", 4)) 
329             type=1;
330         if(!strncasecmp(&name[l-4], ".ttf", 4)) 
331             type=2;
332         if(type)
333         {
334             if(database && type==1) {
335                 char buf[256],a;
336                 FILE*fi;
337                 sprintf(buf, "%s/%s", dirname,name);
338                 fi = fopen(buf, "rb");
339                 if(!fi || !fread(&a,1,1,fi)) {
340                     logf("<warning> Couldn't read from %s", buf);
341                 }
342                 fprintf(database, "%s\n", buf);
343                 logf("<verbose> Found font %s\n", buf);
344                 fclose(fi);
345             } 
346             if(numfonts)
347                 (*numfonts)++;
348         }
349     }
350     closedir(dir);
351 }
352 #endif
353
354 char* stripfilename(char*filename, char*newext)
355 {
356     char*last1 = strrchr(filename, '/');
357     char*last2 = strrchr(filename, '\\');
358     char*pos = filename;
359     char*name;
360     char*dot;
361     if(last1>pos) pos = last1 + 1;
362     if(last2>pos) pos = last2 + 1;
363     name = (char*)malloc(strlen(pos)+5);
364     strcpy(name, pos);
365     dot = strrchr(name, '.');
366     if(dot) {
367         *dot = 0;
368     }
369     strcat(name, newext);
370     return name;
371 }
372
373 int main(int argn, char *argv[])
374 {
375     int ret;
376     char buf[256];
377     int numfonts = 0;
378     int t;
379     char t1searchpath[1024];
380 #ifdef HAVE_SRAND48
381     srand48(time(0));
382 #else
383 #ifdef HAVE_SRAND
384     srand(time(0));
385 #endif
386 #endif
387     processargs(argn, argv);
388     initLog(0,-1,0,0,-1,loglevel);
389     if(!outputname)
390     {
391         if(filename) {
392             outputname = stripfilename(filename, ".swf");
393             logf("<notice> Output filename not given. Writing to %s", outputname);
394         } 
395     }
396         
397     if(!outputname)
398     {
399         fprintf(stderr, "Please use -o to specify an output file\n");
400         exit(1);
401     }
402
403     // test if the page range is o.k.
404     is_in_range(0x7fffffff, pagerange);
405
406     if (!filename) {
407         args_callback_usage(argv[0]);
408         exit(0);
409     }
410
411     logf("<verbose> reading font files from %s/fonts\n", SWFTOOLS_DATADIR);
412     //TODO: use tempnam here. Check if environment already contains a
413     //T1LIB_CONFIG.
414     putenv( "T1LIB_CONFIG=/tmp/t1lib.config.tmp");
415     FILE*db = fopen("/tmp/FontDataBase", "wb");
416     FILE*fi = fopen("/tmp/t1lib.config.tmp", "wb");
417     if(!db || !fi) {
418         fprintf(stderr, "Couldn't create temporary file in /tmp/\n");
419         exit(1);
420     }
421     t1searchpath[0] = 0;
422 #ifdef HAVE_DIRENT_H
423     sprintf(buf, "%s/fonts",SWFTOOLS_DATADIR);
424     // pass 1
425     addfontdir(0, buf, &numfonts, 0);
426     for(t=0;t<fontpathpos;t++) {
427         addfontdir(0, fontpaths[t], &numfonts,0);
428     }
429     fprintf(db, "%d\n", numfonts);
430     // pass 2
431     addfontdir(db, buf, 0, t1searchpath);
432     for(t=0;t<fontpathpos;t++) {
433         addfontdir(db, fontpaths[t], 0, t1searchpath);
434     }
435 #else
436 /* This is a workaround. The correct way would be to
437    get directory listings working on all systems.
438 */
439     strcpy(t1searchpath, SWFTOOLS_DATADIR);
440     strcat(t1searchpath, "/fonts");
441     fprintf(db, "14\n");
442     fprintf(db, "n021003l.afm\n");
443     fprintf(db, "n021023l.afm\n");
444     fprintf(db, "n021004l.afm\n");
445     fprintf(db, "n021024l.afm\n");
446     fprintf(db, "n019003l.afm\n");
447     fprintf(db, "n019023l.afm\n");
448     fprintf(db, "n019004l.afm\n");
449     fprintf(db, "n019024l.afm\n");
450     fprintf(db, "n022003l.afm\n");
451     fprintf(db, "n022023l.afm\n");
452     fprintf(db, "n022004l.afm\n");
453     fprintf(db, "n022024l.afm\n");
454     fprintf(db, "s050000l.afm\n");
455     fprintf(db, "d050000l.afm\n");
456 #endif
457
458     fprintf(fi, "FONTDATABASE=/tmp/FontDataBase\n");
459     fprintf(fi, "ENCODING=%s:.\n", t1searchpath);
460     fprintf(fi, "AFM=%s:.\n", t1searchpath);
461     fprintf(fi, "TYPE1=%s:.\n", t1searchpath);
462     fclose(fi);
463     fclose(db);
464     /* initialize t1lib */
465     T1_SetBitmapPad( 16);
466     if ((T1_InitLib(NO_LOGFILE)==NULL)){
467         fprintf(stderr, "Initialization of t1lib failed\n");
468         exit(1);
469     }
470     unlink("/tmp/t1lib.config.tmp");
471
472     pdfswf_init(filename, password);
473     pdfswf_setoutputfilename(outputname);
474
475     int pages = pdfswf_numpages();
476     for(t = 1; t <= pages; t++) 
477     {
478         if(is_in_range(t, pagerange))
479         pdfswf_convertpage(t);
480     }
481     pdfswf_performconversion();
482
483     pdfswf_close();
484
485     if(viewer || preloader) {
486 #ifndef SYSTEM_BACKTICKS
487         logf("<warning> Not sure whether system() can handle command substitution");
488         logf("<warning> (According to config.h, it can't)");
489 #endif
490         if(!system_quiet)
491             printf("\n");
492     }
493
494     if(viewer && !preloader) {
495         systemf("swfcombine `swfdump -XY %s` %s viewport=%s -o %s",
496                 outputname, viewer, outputname, outputname);
497         if(!system_quiet)
498             printf("\n");
499     }
500     if(preloader && !viewer) {
501         logf("<warning> --preloader option without --viewer option doesn't make very much sense.");
502         ret = systemf("swfcombine `swfdump -r %s` %s/swfs/PreLoaderTemplate.swf loader=%s movie=%s -o %s",
503                 preloader, SWFTOOLS_DATADIR, preloader, outputname, outputname);
504         if(!system_quiet)
505             printf("\n");
506     }
507     if(preloader && viewer) {
508         systemf("swfcombine %s viewport=%s -o __tmp__.swf",
509                 viewer, outputname, outputname);
510         systemf("swfcombine `swfdump -XY %s` `swfdump -r %s` %s/swfs/PreLoaderTemplate.swf loader=%s movie=__tmp__.swf -o %s",
511                 outputname, preloader, SWFTOOLS_DATADIR, preloader, outputname);
512         systemf("rm __tmp__.swf");
513     }
514
515     unlink("/tmp/FontDataBase");
516     return 0;
517 }
518
519