implemented splitted IDAT chunks.
[swftools.git] / src / png2swf.c
1 /* png2swf.c
2
3    PNG to SWF converter tool
4
5    Part of the swftools package.
6
7    Copyright (c) 2002 Matthias Kramm <kramm@quiss.org>
8  
9    This file is distributed under the GPL, see file COPYING for details 
10
11 */
12
13 #include <stdio.h>
14 #include <math.h>
15 #include <fcntl.h>
16 #include <zlib.h>
17 #include "../lib/rfxswf.h"
18 #include "../lib/args.h"
19
20 #define MAX_INPUT_FILES 1024
21 #define VERBOSE(x) (global.verbose>=x)
22
23 struct {
24     int framerate;
25     int max_image_width;
26     int max_image_height;
27     int force_width;
28     int force_height;
29     int nfiles;
30     int verbose;
31     char *outfile;
32 } global;
33
34 struct {
35     char *filename;
36 } image[MAX_INPUT_FILES];
37
38 TAG *MovieStart(SWF * swf, int framerate, int dx, int dy)
39 {
40     TAG *t;
41     RGBA rgb;
42
43     memset(swf, 0x00, sizeof(SWF));
44
45     swf->fileVersion = 4;
46     swf->frameRate = (25600 / framerate);
47     swf->movieSize.xmax = dx * 20;
48     swf->movieSize.ymax = dy * 20;
49
50     t = swf->firstTag = swf_InsertTag(NULL, ST_SETBACKGROUNDCOLOR);
51
52     rgb.r = rgb.g = rgb.b = rgb.a = 0x00;
53     swf_SetRGB(t, &rgb);
54
55     return t;
56 }
57
58 int MovieFinish(SWF * swf, TAG * t, char *sname)
59 {
60     int handle, so = fileno(stdout);
61     t = swf_InsertTag(t, ST_END);
62
63     if ((!isatty(so)) && (!sname))
64         handle = so;
65     else {
66         if (!sname)
67             sname = "output.swf";
68         handle = open(sname, O_RDWR | O_CREAT | O_TRUNC, 0666);
69     }
70     if FAILED
71         (swf_WriteSWF(handle, swf)) if (VERBOSE(1))
72             fprintf(stderr, "Unable to write output file: %s\n", sname);
73     if (handle != so)
74         close(handle);
75
76     swf_FreeTags(swf);
77     return 0;
78 }
79
80 int png_read_chunk(char (*head)[4], int*destlen, U8**destdata, FILE*fi)
81 {
82     unsigned int len;
83     if(destlen) *destlen=0;
84     if(destdata) *destdata=0;
85     if(!fread(&len, 4, 1, fi))
86         return 0;
87     if(!fread(head, 4, 1, fi))
88         return 0;
89     len = REVERSESWAP32(len);
90     if(destlen) *destlen = len;
91     if(destdata) {
92         if(len)
93             *destdata = malloc(len);
94         else 
95             *destdata = 0;
96         if(!fread(*destdata, len, 1, fi)) {
97             *destdata = 0;
98             if(destlen) *destlen=0;
99             return 0;
100         }
101         fseek(fi, 4, SEEK_CUR);
102
103     } else {
104         fseek(fi, len+4, SEEK_CUR);
105     }
106     return 1;
107 }
108
109 unsigned int png_get_dword(FILE*fi)
110 {
111     unsigned int a;
112     fread(&a,4,1,fi);
113     return REVERSESWAP32(a);
114 }
115
116 struct png_header
117 {
118     int width;
119     int height;
120     int bpp;
121     int mode;
122 };
123
124 int png_read_header(FILE*fi, struct png_header*header)
125 {
126     char id[4];
127     int len;
128     int ok=0;
129     U8 head[8] = {137,80,78,71,13,10,26,10};
130     U8 head2[8];
131     U8*data;
132     fread(head2,8,1,fi);
133     if(strncmp(head,head2,4))
134         return 0;
135    
136     while(png_read_chunk(&id, &len, &data, fi))
137     {
138         if(VERBOSE(2))
139             printf("%c%c%c%c %d\n", id[0],id[1],id[2],id[3],len);
140         if(!strncasecmp(id, "IHDR", 4)) {
141             char a,b,c,f,i;
142             if(len < 8) exit(1);
143             header->width = REVERSESWAP32(*(U32*)&data[0]);
144             header->height = REVERSESWAP32(*(U32*)&data[4]);
145             a = data[8];      // should be 8
146             b = data[9];      // should be 3(indexed) or 2(rgb)
147
148             c = data[10];     // compression mode (0)
149             f = data[11];     // filter mode (0)
150             i = data[12];     // interlace mode (0)
151
152             if(b!=2 && b!=3) {
153                 fprintf(stderr, "Image mode %d not supported!\n", b);
154                 exit(1);
155             }
156             if(a!=8 && b==2) {
157                 fprintf(stderr, "Bpp %d in mode %d not supported!\n", a);
158                 exit(1);
159             }
160             if(c!=0) {
161                 fprintf(stderr, "Compression mode %d not supported!\n", c);
162                 exit(1);
163             }
164             if(f!=0) {
165                 fprintf(stderr, "Filter mode %d not supported!\n", f);
166                 exit(1);
167             }
168             if(i!=0) {
169                 fprintf(stderr, "Interlace mode %d not supported!\n", i);
170                 exit(1);
171             }
172             if(VERBOSE(2))
173                 printf("%dx%d %d %d %d %d %d\n",header->width, header->height, a,b,c,f,i);
174             header->bpp = a;
175             header->mode = b;
176             ok = 1;
177         } 
178         
179         free(data);
180     }
181     return ok;
182 }
183
184 typedef unsigned char byte;
185 #define ABS(a) ((a)>0?(a):(-(a)))
186 byte inline PaethPredictor (byte a,byte b,byte c)
187 {
188         // a = left, b = above, c = upper left
189         int p = a + b - c;        // initial estimate
190         int pa = ABS(p - a);      // distances to a, b, c
191         int pb = ABS(p - b);
192         int pc = ABS(p - c);
193         // return nearest of a,b,c,
194         // breaking ties in order a,b,c.
195         if (pa <= pb && pa <= pc) 
196                 return a;
197         else if (pb <= pc) 
198                 return b;
199         else return c;
200 }
201
202 void applyfilter(int mode, U8*src, U8*old, U8*dest, int width)
203 {
204     int x;
205     unsigned char lastr=0;
206     unsigned char lastg=0;
207     unsigned char lastb=0;
208     unsigned char upperlastr=0;
209     unsigned char upperlastg=0;
210     unsigned char upperlastb=0;
211
212     if(mode==0) {
213         for(x=0;x<width;x++) {
214             dest[0] = 255;
215             dest[1] = src[0];
216             dest[2] = src[1];
217             dest[3] = src[2];
218             dest+=4;
219             src+=3;
220         }
221     }
222     else if(mode==1) {
223         for(x=0;x<width;x++) {
224             dest[0] = 255;
225             dest[1] = src[0]+lastr;
226             dest[2] = src[1]+lastg;
227             dest[3] = src[2]+lastb;
228             lastr = dest[1];
229             lastg = dest[2];
230             lastb = dest[3];
231             dest+=4;
232             src+=3;
233         }
234     }
235     else if(mode==2) {
236         for(x=0;x<width;x++) {
237             dest[0] = 255;
238             dest[1] = src[0]+old[1];
239             dest[2] = src[1]+old[2];
240             dest[3] = src[2]+old[3];
241             dest+=4;
242             old+=4;
243             src+=3;
244         }
245     }
246     else if(mode==3) {
247         for(x=0;x<width;x++) {
248             dest[0] = 255;
249             dest[1] = src[0]+(old[1]+lastr)/2;
250             dest[2] = src[1]+(old[2]+lastg)/2;
251             dest[3] = src[2]+(old[3]+lastb)/2;
252             lastr = dest[1];
253             lastg = dest[2];
254             lastb = dest[3];
255             dest+=4;
256             old+=4;
257             src+=3;
258         }
259     }
260     else if(mode==4) {
261         for(x=0;x<width;x++) {
262             dest[0] = 255;
263             dest[1] = src[0]+PaethPredictor(lastr,old[1],upperlastr);
264             dest[2] = src[1]+PaethPredictor(lastg,old[2],upperlastg);
265             dest[3] = src[2]+PaethPredictor(lastb,old[3],upperlastb);
266             lastr = dest[1];
267             lastg = dest[2];
268             lastb = dest[3];
269             upperlastr = old[1];
270             upperlastg = old[2];
271             upperlastb = old[3];
272             dest+=4;
273             old+=4;
274             src+=3;
275         }
276     }    
277
278 }
279
280 void applyfilter1(int mode, U8*src, U8*old, U8*dest, int width)
281 {
282     int x;
283     unsigned char last=0;
284     unsigned char upperlast=0;
285
286     if(mode==0) {
287         for(x=0;x<width;x++) {
288             *dest = *src;
289             dest++;
290             src++;
291         }
292     }
293     else if(mode==1) {
294         for(x=0;x<width;x++) {
295             *dest = *src+last;
296             last = *dest;
297             dest++;
298             src++;
299         }
300     }
301     else if(mode==2) {
302         for(x=0;x<width;x++) {
303             *dest = *src+*old;
304             dest++;
305             old++;
306             src++;
307         }
308     }
309     else if(mode==3) {
310         for(x=0;x<width;x++) {
311             *dest = *src+(*old+last)/2;
312             dest++;
313             old++;
314             src++;
315         }
316     }
317     else if(mode==4) {
318         for(x=0;x<width;x++) {
319             *dest = *src+PaethPredictor(last,*old,upperlast);
320             last = *dest;
321             upperlast = *old;
322             dest++;
323             old++;
324             src++;
325         }
326     }    
327
328 }
329
330 TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
331 {
332     SHAPE *s;
333     SRECT r;
334     MATRIX m;
335     int fs;
336
337     char tagid[4];
338     int len;
339     U8*data;
340     U8*imagedata;
341     U8*zimagedata=0;
342     unsigned long int imagedatalen;
343     unsigned long int zimagedatalen=0;
344     U8*palette = 0;
345     int palettelen = 0;
346     struct png_header header;
347     int bypp;
348
349     FILE *fi;
350     U8 *scanline;
351
352     if ((fi = fopen(sname, "rb")) == NULL) {
353         if (VERBOSE(1))
354             fprintf(stderr, "Read access failed: %s\n", sname);
355         return t;
356     }
357
358     if(!png_read_header(fi, &header))
359         return 0;
360
361     if(header.mode == 3) bypp = 1;
362     else
363     if(header.mode == 2) bypp = 3;
364     else
365         return 0;
366     imagedatalen = bypp * header.width * header.height + 65536;
367     imagedata = malloc(imagedatalen);
368
369     fseek(fi,8,SEEK_SET);
370     while(!feof(fi))
371     {
372         if(!png_read_chunk(&tagid, &len, &data, fi))
373             break;
374         if(!strncmp(tagid, "IEND", 4)) {
375             break;
376         }
377         if(!strncmp(tagid, "PLTE", 4)) {
378             palette = data;
379             palettelen = len/3;
380             data = 0;
381             if(VERBOSE(2))
382                 printf("%d colors in palette\n", palettelen);
383         }
384         if(!strncmp(tagid, "IDAT", 4)) {
385             if(!zimagedata) {
386                 zimagedatalen = len;
387                 zimagedata = malloc(len);
388                 memcpy(zimagedata,data,len);
389             } else {
390                 zimagedata = realloc(zimagedata, zimagedatalen+len);
391                 memcpy(&zimagedata[zimagedatalen], data, len);
392                 zimagedatalen += len;
393             }
394         }
395         if(data)
396             free(data);
397     }
398     
399     if(!zimagedata || uncompress(imagedata, &imagedatalen, zimagedata, zimagedatalen) != Z_OK) {
400         fprintf(stderr, "Couldn't uncompress %s!\n", sname);
401         if(zimagedata)
402             free(zimagedata);
403         return 0;
404     }
405     free(zimagedata);
406
407     t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS);
408     swf_SetU16(t, id);          // id
409     if(header.mode == 2) {
410         U8*data2 = malloc(header.width*header.height*4);
411         int i,s=0;
412         int x,y;
413         int pos=0;
414         /* 24->32 bit conversion */
415         for(y=0;y<header.height;y++) {
416             int mode = imagedata[pos++]; //filter mode
417             U8*src;
418             U8*dest;
419             U8*old;
420             dest = &data2[(y*header.width)*4];
421
422             if(header.bpp == 8)
423             {
424                 /* one byte per pixel */
425                 src = &imagedata[pos];
426                 pos+=header.width*3;
427             } else {
428                 /* not implemented yet */
429                 exit(1);
430             }
431
432             if(!y) {
433                 memset(data2,0,header.width*4);
434                 old = &data2[y*header.width*4];
435             } else {
436                 old = &data2[(y-1)*header.width*4];
437             }
438             applyfilter(mode, src, old, dest, header.width);
439         }
440         swf_SetLosslessBits(t, header.width, header.height, data2, BMF_32BIT);
441         free(data2);
442     }
443     else {
444         RGBA*rgba = (RGBA*)malloc(palettelen*sizeof(RGBA));
445         int swf_width = BYTES_PER_SCANLINE(header.width);
446         U8*data2 = malloc(swf_width*header.height);
447         U8*tmpline = malloc(header.width);
448         int i,x,y;
449         int pos=0;
450         if(!palette) {
451             fprintf(stderr, "Error: No palette found!\n");
452             exit(1);
453         }
454         /* 24->32 bit conversion */
455         for(i=0;i<palettelen;i++) {
456             rgba[i].r = palette[i*3+0];
457             rgba[i].g = palette[i*3+1];
458             rgba[i].b = palette[i*3+2];
459             rgba[i].a = 255;
460         }
461         for(y=0;y<header.height;y++) {
462             int mode = imagedata[pos++]; //filter mode
463             U8*old;
464             U8*dest = &data2[y*swf_width];
465             U8*src;
466             src = &imagedata[pos];
467             if(header.bpp == 8) {
468                 pos+=header.width;
469             } else {
470                 int x,s=0;
471                 int bitpos = 0;
472                 U32 v = (1<<header.bpp)-1;
473                 for(x=0;x<header.width;x++) {
474                     U32 r = src[s/8]<<8 | 
475                             src[s/8+1];
476                     int t;
477                     tmpline[x] = (r>>(16-header.bpp-(s&7)))&v;
478                     s+=header.bpp;
479                 }
480                 src = tmpline;
481                 pos+=(header.width*header.bpp+7)/8;
482             }
483
484             if(!y) {
485                 memset(data2,0,swf_width);
486                 old = &data2[y*swf_width];
487             } else {
488                 old = &data2[(y-1)*swf_width];
489             }
490             applyfilter1(mode, src, old, dest, header.width);
491         }
492         swf_SetLosslessBitsIndexed(t, header.width, header.height, data2, rgba, palettelen);
493         free(tmpline);
494         free(rgba);
495         free(data2);
496     }
497
498     t = swf_InsertTag(t, ST_DEFINESHAPE);
499
500     swf_ShapeNew(&s);
501     swf_GetMatrix(NULL, &m);
502     m.sx = 20 * 0x10000;
503     m.sy = 20 * 0x10000;
504     fs = swf_ShapeAddBitmapFillStyle(s, &m, id, 0);
505
506     swf_SetU16(t, id + 1);      // id
507
508     r.xmin = r.ymin = 0;
509     r.xmax = header.width * 20;
510     r.ymax = header.height * 20;
511     swf_SetRect(t, &r);
512
513     swf_SetShapeHeader(t, s);
514
515     swf_ShapeSetAll(t, s, 0, 0, 0, fs, 0);
516     swf_ShapeSetLine(t, s, r.xmax, 0);
517     swf_ShapeSetLine(t, s, 0, r.ymax);
518     swf_ShapeSetLine(t, s, -r.xmax, 0);
519     swf_ShapeSetLine(t, s, 0, -r.ymax);
520
521     swf_ShapeSetEnd(t);
522
523     t = swf_InsertTag(t, ST_REMOVEOBJECT2);
524     swf_SetU16(t, 1);           // depth
525
526     t = swf_InsertTag(t, ST_PLACEOBJECT2);
527
528     swf_GetMatrix(NULL, &m);
529     m.tx = (swf->movieSize.xmax - (int) header.width * 20) / 2;
530     m.ty = (swf->movieSize.ymax - (int) header.height * 20) / 2;
531     swf_ObjectPlace(t, id + 1, 1, &m, NULL, NULL);
532
533     t = swf_InsertTag(t, ST_SHOWFRAME);
534
535     fclose(fi);
536
537     return t;
538 }
539
540
541 int CheckInputFile(char *fname, char **realname)
542 {
543     FILE *fi;
544     char *s = malloc(strlen(fname) + 5);
545     struct png_header head;
546
547     if (!s)
548         exit(2);
549     (*realname) = s;
550     strcpy(s, fname);
551
552     // Check whether file exists (with typical extensions)
553
554     if ((fi = fopen(s, "rb")) == NULL) {
555         sprintf(s, "%s.png", fname);
556         if ((fi = fopen(s, "rb")) == NULL) {
557             sprintf(s, "%s.PNG", fname);
558             if ((fi = fopen(s, "rb")) == NULL) {
559                 sprintf(s, "%s.Png", fname);
560                 if ((fi = fopen(s, "rb")) == NULL)
561                     fprintf(stderr, "Couldn't open %s!\n", fname);
562                     return -1;
563             }
564         }
565     }
566
567     if(!png_read_header(fi, &head)) {
568         fprintf(stderr, "%s is not a PNG file!\n", fname);
569         return -1;
570     }
571
572     if (global.max_image_width < head.width)
573         global.max_image_width = head.width;
574     if (global.max_image_height < head.height)
575         global.max_image_height = head.height;
576
577     fclose(fi);
578
579     return 0;
580 }
581
582 int args_callback_option(char *arg, char *val)
583 {
584     int res = 0;
585     if (arg[1])
586         res = -1;
587     else
588         switch (arg[0]) {
589         case 'r':
590             if (val)
591                 global.framerate = atoi(val);
592             if ((global.framerate < 1) ||(global.framerate > 5000)) {
593                 if (VERBOSE(1))
594                     fprintf(stderr,
595                             "Error: You must specify a valid framerate between 1 and 10000.\n");
596                 exit(1);
597             }
598             res = 1;
599             break;
600
601         case 'o':
602             if (val)
603                 global.outfile = val;
604             res = 1;
605             break;
606
607         case 'v':
608             if (val)
609                 global.verbose = atoi(val);
610             res = 1;
611             break;
612
613         case 'X':
614             if (val)
615                 global.force_width = atoi(val);
616             res = 1;
617             break;
618
619         case 'Y':
620             if (val)
621                 global.force_height = atoi(val);
622             res = 1;
623             break;
624
625         case 'V':
626             printf("png2swf - part of %s %s\n", PACKAGE, VERSION);
627             exit(0);
628
629         default:
630             res = -1;
631             break;
632         }
633
634     if (res < 0) {
635         if (VERBOSE(1))
636             fprintf(stderr, "Unknown option: -%s\n", arg);
637         exit(1);
638         return 0;
639     }
640     return res;
641 }
642
643 struct options_t options[] = 
644
645 {"o", "output"},
646 {"r", "rate"},
647 {"v", "verbose"},
648 {"X", "width"},
649 {"Y", "height"},
650 {"V", "version"},
651 };
652
653 int args_callback_longoption(char *name, char *val)
654 {
655     return args_long2shortoption(options, name, val);
656 }
657
658 int args_callback_command(char *arg, char *next)        // actually used as filename
659 {
660     char *s;
661     if (CheckInputFile(arg, &s) < 0) {
662         if (VERBOSE(1))
663             fprintf(stderr, "Error opening input file: %s\n", arg);
664         free(s);
665     } else {
666         image[global.nfiles].filename = s;
667         global.nfiles++;
668         if (global.nfiles >= MAX_INPUT_FILES) {
669             if (VERBOSE(1))
670                 fprintf(stderr, "Error: Too many input files.\n");
671             exit(1);
672         }
673     }
674     return 0;
675 }
676
677 void args_callback_usage(char *name)
678 {
679     printf("Usage: %s  [-options [value]] imagefiles[.png] [...]\n", name);
680     printf("-r framerate          (rate) Set movie framerate (100/sec)\n");
681     printf("-o outputfile         (output) Set name for SWF output file\n");
682     printf("-X pixel              (width) Force movie width to pixel (default: autodetect)\n");
683     printf("-Y pixel              (height) Force movie height to pixel (default: autodetect)\n");
684     printf("-v level              (verbose) Set verbose level (0=quiet, 1=default, 2=debug)\n");
685     printf("-V                    (version) Print version information and exit\n");
686 }
687
688
689 int main(int argc, char **argv)
690 {
691     SWF swf;
692     TAG *t;
693
694     memset(&global, 0x00, sizeof(global));
695
696     global.framerate = 100;
697     global.verbose = 1;
698
699     processargs(argc, argv);
700
701     if(global.nfiles<=0)
702         return 1;
703
704     if (VERBOSE(2))
705         fprintf(stderr, "Processing %i file(s)...\n", global.nfiles);
706
707     t = MovieStart(&swf, global.framerate,
708                    global.force_width ? global.force_width : global.
709                    max_image_width,
710                    global.force_height ? global.force_height : global.
711                    max_image_height);
712
713     {
714         int i;
715         for (i = 0; i < global.nfiles; i++) {
716             if (VERBOSE(3))
717                 fprintf(stderr, "[%03i] %s\n", i,
718                         image[i].filename);
719             t = MovieAddFrame(&swf, t, image[i].filename, (i * 2) + 1);
720             free(image[i].filename);
721         }
722     }
723
724     MovieFinish(&swf, t, global.outfile);
725
726     return 0;
727 }