added handling for incomplete tags during file reading.
[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,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 "SWFOutputDev.h"
36 #include "log.h"
37
38 #ifndef WIN32
39 #define FONTDIR SWFTOOLS_DATADIR "/fonts"
40 #define SWFDIR SWFTOOLS_DATADIR "/swfs"
41 #else
42 #define FONTDIR "C:\\swftools\\fonts"
43 #define SWFDIR "C:\\swftools\\swfs"
44 #endif
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
52 static char * preloader = 0;
53 static char * viewer = 0;
54
55 char* fontpaths[256];
56 int fontpathpos = 0;
57
58 static int system_quiet=0;
59
60 int systemf(const char* format, ...)
61 {
62     char buf[1024];
63     int ret;
64     va_list arglist;
65     va_start(arglist, format);
66     vsprintf(buf, format, arglist);
67     va_end(arglist);
68
69     if(!system_quiet) {
70         printf("%s\n", buf);
71         fflush(stdout);
72     }
73     ret = system(buf);
74     if(ret) {
75         fprintf(stderr, "system() returned %d\n", ret);
76         exit(ret);
77     }
78     return ret;
79 }
80
81 int args_callback_option(char*name,char*val) {
82     if (!strcmp(name, "o"))
83     {
84         outputname = val;
85         return 1;
86     }
87     else if (!strcmp(name, "v"))
88     {
89         loglevel ++;
90         setConsoleLogging(loglevel);
91         return 0;
92     }
93     else if (!strcmp(name, "q"))
94     {
95         loglevel --;
96         setConsoleLogging(loglevel);
97         system_quiet = 1;
98         return 0;
99     }
100     else if (name[0]=='p')
101     {
102         /* check whether the page range follows the p directly, like 
103            in -p1,2 */
104         do {
105             name++;
106         } while(*name == 32 || *name == 13 || *name == 10 || *name == '\t');
107
108         if(*name) {
109             pagerange = name;
110             return 0;
111         } 
112         pagerange = val;        
113         return 1;
114     }
115     else if (!strcmp(name, "P"))
116     {
117         password = val;
118         return 1;
119     }
120     else if (!strcmp(name, "s"))
121     {
122         char*s = strdup(val);
123         char*c = strchr(s, '=');
124         if(c && *c && c[1])  {
125             *c = 0;
126             c++;
127             pdfswf_setparameter(s,c);
128         }
129         else
130             pdfswf_setparameter(s,"1");
131         return 1;
132     }
133     else if (!strcmp(name, "S"))
134     {
135         pdfswf_setparameter("drawonlyshapes", "1");
136         return 0;
137     }
138     else if (!strcmp(name, "i"))
139     {
140         pdfswf_setparameter("ignoredraworder", "1");
141         return 0;
142     }
143     else if (!strcmp(name, "z"))
144     {
145         pdfswf_setparameter("enablezlib", "1");
146         return 0;
147     }
148     else if (!strcmp(name, "n"))
149     {
150         pdfswf_setparameter("opennewwindow", "1");
151         return 0;
152     }
153     else if (!strcmp(name, "t"))
154     {
155         pdfswf_setparameter("insertstop", "1");
156         return 0;
157     }
158     else if (!strcmp(name, "T"))
159     {
160         if(!strcasecmp(val, "mx"))
161             pdfswf_setparameter("flashversion", "6");
162         else
163             pdfswf_setparameter("flashversion", val);
164
165         return 1;
166     }
167     else if (!strcmp(name, "f"))
168     {
169         pdfswf_setparameter("storeallcharacters", "1");
170         return 0;
171     }
172     else if (!strcmp(name, "F"))
173     {
174         char *s = strdup(val);
175         int l = strlen(s);
176         while(l && s[l-1]=='/') {
177             s[l-1] = 0;
178             l--;
179         }
180         fontpaths[fontpathpos++] = s;
181         return 1;
182     }
183     else if (!strcmp(name, "l"))
184     {
185         char buf[256];
186         sprintf(buf, "%s/default_loader.swf", SWFDIR);
187         preloader = strdup(buf);
188         return 0;
189     }
190     else if (!strcmp(name, "b"))
191     {
192         char buf[256];
193         sprintf(buf, "%s/default_viewer.swf", SWFDIR);
194         viewer = strdup(buf);
195         return 0;
196     }
197     else if (!strcmp(name, "L"))
198     {
199         if(val)
200         {
201             preloader = val;
202         }
203         else
204         {
205             systemf("ls %s/*_loader.swf", SWFDIR);
206             if(!system_quiet)
207                 printf("\n");
208             exit(1);
209         }
210         return 1;
211     }
212     else if (!strcmp(name, "B"))
213     {
214         if(val)
215         {
216             viewer = val;
217         }
218         else
219         {
220             systemf("ls %s/*_viewer.swf", SWFDIR);
221             if(!system_quiet)
222                 printf("\n");
223             exit(1);
224         }
225         return 1;
226     }
227     else if (!strcmp(name, "j"))
228     {
229         if(name[1]) {
230             pdfswf_setparameter("jpegquality", &name[1]);
231             return 0;
232         } else {
233             pdfswf_setparameter("jpegquality", val);
234             return 1;
235         }
236     }
237     else if (!strcmp(name, "V"))
238     {   
239         printf("pdf2swf - part of %s %s\n", PACKAGE, VERSION);
240         exit(0);
241     }
242     else 
243     {
244         fprintf(stderr, "Unknown option: -%s\n", name);
245         exit(1);
246     }
247     return 0;
248 }
249
250 /*struct docoptions_t options[] =
251 {{"o","output","filename::Specify output file"},
252  {"V","version","Print program version"},
253  {"i","ignore","Ignore draw order (makes the SWF file smaller, but may produce graphic errors)"},
254  {"z","zlib","Use Flash 6 (MX) zlib compression (Needs at least Flash 6 Plugin to play)"},
255  {"s","shapes","Don't use SWF Fonts, but store everything as shape"},
256  {"j","jpegquality","Set quality of embedded jpeg pictures (default: 85)"},
257  {"p","pages","Convert only pages in range. (E.g. 3-85)"},
258  {"w","samewindow","Don't open a new browser window for links in the SWF"},
259  {"f","fonts","Stroe full fonts in SWF. (Don't reduce to used characters)"},
260  {"F","fontpath","path::Add directory to font search path"},
261  {"B","viewer","name::Link viewer \"name\" to the pdf"},
262  {"L","preloader","file.swf::Link preloader \"file.swf\" to the pdf"},
263  {"b","defaultviewer","Link default viewer to the pdf"},
264  {"l","defaultpreloader","Link default preloader to the pdf"}
265  {0,0}
266 };*/
267 struct options_t options[] =
268 {{"o","output"},
269  {"q","quiet"},
270  {"V","version"},
271  {"i","ignore"},
272  {"z","zlib"},
273  {"s","set"},
274  {"S","shapes"},
275  {"j","jpegquality"},
276  {"p","pages"},
277  {"w","samewindow"},
278  {"f","fonts"},
279  {"F","fontdir"},
280  {"B","viewer"},
281  {"L","preloader"},
282  {"b","defaultviewer"},
283  {"l","defaultpreloader"},
284  {"t","stop"},
285  {"T","flashversion"},
286  {0,0}
287 };
288
289 int args_callback_longoption(char*name,char*val) {
290     return args_long2shortoption(options, name, val);
291 }
292
293 int args_callback_command(char*name, char*val) {
294     if (!filename) 
295         filename = name;
296     else {
297         if(outputname)
298         {
299              fprintf(stderr, "Error: Do you want the output to go to %s or to %s?", 
300                      outputname, name);
301              exit(1);
302         }
303         outputname = name;
304     }
305     return 0;
306 }
307
308 void args_callback_usage(char*name)
309 {
310     printf("Usage: %s [Options] input.pdf [-o output.swf]\n", name);
311     printf("\nBasic options:\n");
312     printf("-p  --pages=range          Convert only pages in range\n");
313     printf("-P  --password=password    Use password for deciphering the pdf\n");
314     printf("-v  --verbose              Be verbose. Use more than one -v for greater effect\n");
315     printf("-q  --quiet                Suppress normal messages. Use -qq to suppress warnings, also.\n");
316 #ifdef HAVE_DIRENT_H
317     printf("-F  --fontdir directory    Add directory to font search path\n");
318 #endif
319     printf("-V  --version              Print program version\n");
320     printf("\nEnhanced conversion options:\n");
321     printf("-S  --shapes               Don't use SWF Fonts, but store everything as shape\n");
322     printf("-z  --zlib                 Use Flash 6 (MX) zlib compression (Needs at least Flash 6 Plugin to play)\n");
323     printf("-w  --samewindow           Don't open a new Browser Window for Links in the SWF\n");
324     printf("-f  --fonts                Store full fonts in SWF. (Don't reduce to used characters)\n");
325     printf("-T  --flashversion=num     Set the flash version in the header to num (default: 4)\n");
326     printf("-s insertstop              Insert a \"Stop\" Tag in every frame (don't turn pages automatically)\n");
327     printf("-s zoom=factor             Scale result, default: 72\n");
328     printf("-s jpegquality=quality     Set quality of embedded jpeg pictures (default:85)\n");
329     printf("-s caplinewidth=value      Set the minimum line width to trigger cap style handling to value. (3)\n");
330     printf("-s splinequality=value     Set the quality of spline convertion to value (0-100, default: 100).\n");
331     printf("-s fontquality=value       Set the quality of font convertion to value (0-100, default: 100).\n");
332     printf("-s ignoredraworder         Ignore draw order (makes the SWF file smaller and faster, but may produce\n"
333            "                           graphic errors)\n");
334     printf("-s filloverlap             Make intersecting shapes overlap, instead of canceling each\n"
335            "                           other out. (Needed for some Powerpoint PDFs)\n");
336     printf("Postprocessing options:\n");
337 #ifndef SYSTEM_BACKTICKS
338     printf("(They might not work because your system call doesn't support command substitution)\n");
339 #endif
340     printf("-b  --defaultviewer        Link default viewer to the pdf (%s/swfs/default_viewer.swf)\n", SWFTOOLS_DATADIR);
341     printf("-l  --defaultpreloader     Link default preloader the pdf (%s/swfs/default_loader.swf)\n", SWFTOOLS_DATADIR);
342     printf("-B  --viewer=filename      Link viewer \"name\" to the pdf (\"%s -B\" for list)\n", name);
343     printf("-L  --preloader=filename   Link preloader \"name\" to the pdf (\"%s -L\" for list)\n",name);
344 }
345
346 static char* stripfilename(char*filename, char*newext)
347 {
348     char*last1 = strrchr(filename, '/');
349     char*last2 = strrchr(filename, '\\');
350     char*pos = filename;
351     char*name;
352     char*dot;
353     if(last1>pos) pos = last1 + 1;
354     if(last2>pos) pos = last2 + 1;
355     name = (char*)malloc(strlen(pos)+5);
356     strcpy(name, pos);
357     dot = strrchr(name, '.');
358     if(dot) {
359         *dot = 0;
360     }
361     strcat(name, newext);
362     return name;
363 }
364
365 int main(int argn, char *argv[])
366 {
367     int ret;
368     char buf[256];
369     int numfonts = 0;
370     int t;
371     char t1searchpath[1024];
372     
373     initLog(0,-1,0,0,-1,loglevel);
374
375 #if defined(WIN32) && defined(HAVE_STAT) && defined(HAVE_SYS_STAT_H)
376     FILE*test = fopen(FONTDIR "\\d050000l.afm", "rb");
377     if(!test) {
378         fprintf(stderr, "Couldn't find file " FONTDIR "\\d050000l.afm- pdf2swf not installed properly? OS says:\n");
379         perror("open");
380         exit(1);
381     }
382     fclose(test);
383 #endif
384
385 #ifdef HAVE_SRAND48
386     srand48(time(0));
387 #else
388 #ifdef HAVE_SRAND
389     srand(time(0));
390 #endif
391 #endif
392     processargs(argn, argv);
393
394     if(!filename)
395     {
396         fprintf(stderr, "Please specify an input file\n");
397         exit(1);
398     }
399
400     if(!outputname)
401     {
402         if(filename) {
403             outputname = stripfilename(filename, ".swf");
404             msg("<notice> Output filename not given. Writing to %s", outputname);
405         } 
406     }
407         
408     if(!outputname)
409     {
410         fprintf(stderr, "Please use -o to specify an output file\n");
411         exit(1);
412     }
413
414     // test if the page range is o.k.
415     is_in_range(0x7fffffff, pagerange);
416
417     if (!filename) {
418         args_callback_usage(argv[0]);
419         exit(0);
420     }
421
422     /* add fonts */
423     pdfswf_addfontdir(FONTDIR);
424     for(t=0;t<fontpathpos;t++) {
425         pdfswf_addfontdir(fontpaths[t]);
426     }
427
428     pdf_doc_t* pdf = pdf_init(filename, password);
429     if(!pdf) {
430         msg("<error> Couldn't open %s", filename);
431         exit(1);
432     }
433     swf_output_t* swf = swf_output_init();
434
435     for(t = 1; t <= pdf->num_pages; t++) 
436     {
437         if(is_in_range(t, pagerange)) {
438             /* for links: FIXME */
439             pdfswf_preparepage(t);
440         }
441         if(is_in_range(t, pagerange)) {
442             pdf_page_t*page = pdf_getpage(pdf, t);
443             pdf_page_render(page, swf);
444             pdf_page_destroy(page);
445         }
446     }
447     if(!swf_output_save(swf, outputname))
448         exit(1);
449     swf_output_destroy(swf);
450
451     pdf_destroy(pdf);
452
453     if(viewer || preloader) {
454 #ifndef SYSTEM_BACKTICKS
455         msg("<warning> Not sure whether system() can handle command substitution");
456         msg("<warning> (According to config.h, it can't)");
457 #endif
458         if(!system_quiet)
459             printf("\n");
460     }
461
462     if(viewer && !preloader) {
463         systemf("swfcombine `swfdump -XY \"%s\"` \"%s\" viewport=\"%s\" -o \"%s\"",
464                 outputname, viewer, outputname, outputname);
465         if(!system_quiet)
466             printf("\n");
467     }
468     if(preloader && !viewer) {
469         msg("<warning> --preloader option without --viewer option doesn't make very much sense.");
470         ret = systemf("swfcombine `swfdump -r \"%s\"` %s/PreLoaderTemplate.swf loader=\"%s\" movie=\"%s\" -o \"%s\"",
471                 preloader, SWFDIR, preloader, outputname, outputname);
472         if(!system_quiet)
473             printf("\n");
474     }
475     if(preloader && viewer) {
476         systemf("swfcombine \"%s\" viewport=%s -o __tmp__.swf",
477                 viewer, outputname, outputname);
478         systemf("swfcombine `swfdump -XY \"%s\"` `swfdump -r \"%s\"` %s/PreLoaderTemplate.swf loader=%s movie=__tmp__.swf -o \"%s\"",
479                 outputname, preloader, SWFDIR, preloader, outputname);
480         systemf("rm __tmp__.swf");
481     }
482
483     return 0;
484 }
485