printf %% fix by Markus Fleck-Graffe
[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,2003 Matthias Kramm <kramm@quiss.org>
8  
9    This program is free software; you can redistribute it and/or modify
10    it under the terms of the GNU General Public License as published by
11    the Free Software Foundation; either version 2 of the License, or
12    (at your option) any later version.
13
14    This program is distributed in the hope that it will be useful,
15    but WITHOUT ANY WARRANTY; without even the implied warranty of
16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17    GNU General Public License for more details.
18
19    You should have received a copy of the GNU General Public License
20    along with this program; if not, write to the Free Software
21    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
22  
23 #include <stdio.h>
24 #include <math.h>
25 #include <fcntl.h>
26 #include <zlib.h>
27 #include "../lib/rfxswf.h"
28 #include "../lib/args.h"
29 #include "../lib/png.h"
30
31 #define MAX_INPUT_FILES 1024
32 #define VERBOSE(x) (global.verbose>=x)
33
34 struct {
35     float framerate;
36     int max_image_width;
37     int max_image_height;
38     int force_width;
39     int force_height;
40     int nfiles;
41     int verbose;
42     int do_cgi;
43     int version;
44     char *outfile;
45     int mkjpeg;
46     float scale;
47 } global;
48
49 struct {
50     char *filename;
51 } image[MAX_INPUT_FILES];
52
53 static int custom_move=0;
54 static int move_x=0;
55 static int move_y=0;
56 static int clip_x1=0,clip_y1=0,clip_x2=0,clip_y2=0;
57 static int custom_clip = 0;
58
59 TAG *MovieStart(SWF * swf, float framerate, int dx, int dy)
60 {
61     TAG *t;
62     RGBA rgb;
63
64     memset(swf, 0x00, sizeof(SWF));
65
66     swf->fileVersion = global.version;
67     swf->frameRate = (int)(256.0 * framerate);
68     if(custom_clip) {
69         swf->movieSize.xmin = clip_x1 * 20;
70         swf->movieSize.ymin = clip_y1 * 20;
71         swf->movieSize.xmax = clip_x2 * 20;
72         swf->movieSize.ymax = clip_y2 * 20;
73     } else {
74         swf->movieSize.xmax = dx * 20;
75         swf->movieSize.ymax = dy * 20;
76     }
77
78     t = swf->firstTag = swf_InsertTag(NULL, ST_SETBACKGROUNDCOLOR);
79
80     rgb.r = rgb.g = rgb.b = rgb.a = 0x00;
81     //rgb.g = 0xff; //<--- handy for testing alpha conversion
82     swf_SetRGB(t, &rgb);
83
84     return t;
85 }
86
87 int MovieFinish(SWF * swf, TAG * t, char *sname)
88 {
89     int f, so = fileno(stdout);
90     t = swf_InsertTag(t, ST_END);
91
92     if ((!isatty(so)) && (!sname))
93         f = so;
94     else {
95         if (!sname)
96             sname = "output.swf";
97         f = open(sname,O_WRONLY|O_CREAT|O_TRUNC|O_BINARY, 0644);
98     }
99
100     if(global.do_cgi) {
101         if FAILED(swf_WriteCGI(swf)) fprintf(stderr,"WriteCGI() failed.\n");
102     } else {
103         if(global.version >= 6) {
104             if (swf_WriteSWC(f, swf)<0) 
105                     fprintf(stderr, "Unable to write output file: %s\n", sname);
106         } else {
107             if (swf_WriteSWF(f, swf)<0) 
108                     fprintf(stderr, "Unable to write output file: %s\n", sname);
109         }
110         if (f != so)
111             close(f);
112     }
113
114     swf_FreeTags(swf);
115     return 0;
116 }
117
118 int png_read_chunk(char (*head)[4], int*destlen, U8**destdata, FILE*fi)
119 {
120     unsigned int len;
121     if(destlen) *destlen=0;
122     if(destdata) *destdata=0;
123     if(!fread(&len, 4, 1, fi))
124         return 0;
125     if(!fread(head, 4, 1, fi))
126         return 0;
127     len = REVERSESWAP32(len);
128     if(destlen) *destlen = len;
129     if(destdata) {
130         if(len)
131             *destdata = malloc(len);
132         else 
133             *destdata = 0;
134         if(!fread(*destdata, len, 1, fi)) {
135             *destdata = 0;
136             if(destlen) *destlen=0;
137             return 0;
138         }
139         fseek(fi, 4, SEEK_CUR);
140
141     } else {
142         fseek(fi, len+4, SEEK_CUR);
143     }
144     return 1;
145 }
146
147 unsigned int png_get_dword(FILE*fi)
148 {
149     unsigned int a;
150     fread(&a,4,1,fi);
151     return REVERSESWAP32(a);
152 }
153
154 struct png_header
155 {
156     int width;
157     int height;
158     int bpp;
159     int mode;
160 };
161
162 int png_read_header(FILE*fi, struct png_header*header)
163 {
164     char id[4];
165     int len;
166     int ok=0;
167     U8 head[8] = {137,80,78,71,13,10,26,10};
168     U8 head2[8];
169     U8*data;
170     fread(head2,8,1,fi);
171     if(strncmp(head,head2,4))
172         return 0;
173    
174     while(png_read_chunk(&id, &len, &data, fi))
175     {
176         if(VERBOSE(2))
177             printf("%c%c%c%c %d\n", id[0],id[1],id[2],id[3],len);
178         if(!strncasecmp(id, "IHDR", 4)) {
179             char a,b,c,f,i;
180             if(len < 8) exit(1);
181             header->width = REVERSESWAP32(*(U32*)&data[0]);
182             header->height = REVERSESWAP32(*(U32*)&data[4]);
183             a = data[8];      // should be 8
184             b = data[9];      // should be 3(indexed), 2(rgb), 0(grayscale) or 6(truecolor+alpha)
185
186             c = data[10];     // compression mode (0)
187             f = data[11];     // filter mode (0)
188             i = data[12];     // interlace mode (0)
189         
190             if(VERBOSE(2)) printf("image mode:%d\n", b);
191             if(VERBOSE(2)) printf("bpp: %d\n", a);
192             if(VERBOSE(2)) printf("compression: %d\n", c);
193             if(VERBOSE(2)) printf("filter: %d\n", f);
194             if(VERBOSE(2)) printf("interlace: %d\n", i);
195
196             if(b!=0 && b!=2 && b!=3 && b!=6) {
197                 fprintf(stderr, "Image mode %d not supported!\n", b);
198                 if(b == 4) {
199                     fprintf(stderr, "(This is a grayscale image with alpha channel-\n");
200                     fprintf(stderr, " try converting it into an RGB image with alpha channel)\n");
201                 }
202                 exit(1);
203             }
204             if(a!=8 && (b==2 || b==6)) {
205                 fprintf(stderr, "Bpp %d in mode %d not supported!\n", a);
206                 exit(1);
207             }
208             if(c!=0) {
209                 fprintf(stderr, "Compression mode %d not supported!\n", c);
210                 exit(1);
211             }
212             if(f!=0) {
213                 fprintf(stderr, "Filter mode %d not supported!\n", f);
214                 exit(1);
215             }
216             if(i!=0) {
217                 fprintf(stderr, "Interlace mode %d not supported!\n", i);
218                 exit(1);
219             }
220             if(VERBOSE(2))
221                 printf("%dx%d %d %d %d %d %d\n",header->width, header->height, a,b,c,f,i);
222             header->bpp = a;
223             header->mode = b;
224             ok = 1;
225         } 
226         
227         free(data);
228     }
229     return ok;
230 }
231
232 typedef unsigned char byte;
233 #define ABS(a) ((a)>0?(a):(-(a)))
234 byte inline PaethPredictor (byte a,byte b,byte c)
235 {
236         // a = left, b = above, c = upper left
237         int p = a + b - c;        // initial estimate
238         int pa = ABS(p - a);      // distances to a, b, c
239         int pb = ABS(p - b);
240         int pc = ABS(p - c);
241         // return nearest of a,b,c,
242         // breaking ties in order a,b,c.
243         if (pa <= pb && pa <= pc) 
244                 return a;
245         else if (pb <= pc) 
246                 return b;
247         else return c;
248 }
249
250 void applyfilter3(int mode, U8*src, U8*old, U8*dest, int width)
251 {
252     int x;
253     unsigned char lastr=0;
254     unsigned char lastg=0;
255     unsigned char lastb=0;
256     unsigned char upperlastr=0;
257     unsigned char upperlastg=0;
258     unsigned char upperlastb=0;
259
260     if(mode==0) {
261         for(x=0;x<width;x++) {
262             dest[0] = 255;
263             dest[1] = src[0];
264             dest[2] = src[1];
265             dest[3] = src[2];
266             dest+=4;
267             src+=3;
268         }
269     }
270     else if(mode==1) {
271         for(x=0;x<width;x++) {
272             dest[0] = 255;
273             dest[1] = src[0]+lastr;
274             dest[2] = src[1]+lastg;
275             dest[3] = src[2]+lastb;
276             lastr = dest[1];
277             lastg = dest[2];
278             lastb = dest[3];
279             dest+=4;
280             src+=3;
281         }
282     }
283     else if(mode==2) {
284         for(x=0;x<width;x++) {
285             dest[0] = 255;
286             dest[1] = src[0]+old[1];
287             dest[2] = src[1]+old[2];
288             dest[3] = src[2]+old[3];
289             dest+=4;
290             old+=4;
291             src+=3;
292         }
293     }
294     else if(mode==3) {
295         for(x=0;x<width;x++) {
296             dest[0] = 255;
297             dest[1] = src[0]+(old[1]+lastr)/2;
298             dest[2] = src[1]+(old[2]+lastg)/2;
299             dest[3] = src[2]+(old[3]+lastb)/2;
300             lastr = dest[1];
301             lastg = dest[2];
302             lastb = dest[3];
303             dest+=4;
304             old+=4;
305             src+=3;
306         }
307     }
308     else if(mode==4) {
309         for(x=0;x<width;x++) {
310             dest[0] = 255;
311             dest[1] = src[0]+PaethPredictor(lastr,old[1],upperlastr);
312             dest[2] = src[1]+PaethPredictor(lastg,old[2],upperlastg);
313             dest[3] = src[2]+PaethPredictor(lastb,old[3],upperlastb);
314             lastr = dest[1];
315             lastg = dest[2];
316             lastb = dest[3];
317             upperlastr = old[1];
318             upperlastg = old[2];
319             upperlastb = old[3];
320             dest+=4;
321             old+=4;
322             src+=3;
323         }
324     }    
325 }
326
327 void applyfilter4(int mode, U8*src, U8*old, U8*dest, int width)
328 {
329     int x;
330     unsigned char lastr=0;
331     unsigned char lastg=0;
332     unsigned char lastb=0;
333     unsigned char lasta=0; //TODO: 255?
334     unsigned char upperlastr=0;
335     unsigned char upperlastg=0;
336     unsigned char upperlastb=0;
337     unsigned char upperlasta=0; //TODO: 255?
338
339     if(mode==0) {
340         for(x=0;x<width;x++) {
341             dest[0] = src[3];
342             dest[1] = src[0];
343             dest[2] = src[1];
344             dest[3] = src[2];
345             dest+=4;
346             src+=4;
347         }
348     }
349     else if(mode==1) {
350         for(x=0;x<width;x++) {
351             dest[0] = src[3]+lasta;
352             dest[1] = src[0]+lastr;
353             dest[2] = src[1]+lastg;
354             dest[3] = src[2]+lastb;
355             lasta = dest[0];
356             lastr = dest[1];
357             lastg = dest[2];
358             lastb = dest[3];
359             dest+=4;
360             src+=4;
361         }
362     }
363     else if(mode==2) {
364         for(x=0;x<width;x++) {
365             dest[0] = src[3]+old[0];
366             dest[1] = src[0]+old[1];
367             dest[2] = src[1]+old[2];
368             dest[3] = src[2]+old[3];
369             dest+=4;
370             old+=4;
371             src+=4;
372         }
373     }
374     else if(mode==3) {
375         for(x=0;x<width;x++) {
376             dest[0] = src[3]+(old[0]+lasta)/2;
377             dest[1] = src[0]+(old[1]+lastr)/2;
378             dest[2] = src[1]+(old[2]+lastg)/2;
379             dest[3] = src[2]+(old[3]+lastb)/2;
380             lasta = dest[0];
381             lastr = dest[1];
382             lastg = dest[2];
383             lastb = dest[3];
384             dest+=4;
385             old+=4;
386             src+=4;
387         }
388     }
389     else if(mode==4) {
390         for(x=0;x<width;x++) {
391             dest[0] = src[3]+PaethPredictor(lasta,old[0],upperlasta);
392             dest[1] = src[0]+PaethPredictor(lastr,old[1],upperlastr);
393             dest[2] = src[1]+PaethPredictor(lastg,old[2],upperlastg);
394             dest[3] = src[2]+PaethPredictor(lastb,old[3],upperlastb);
395             lasta = dest[0];
396             lastr = dest[1];
397             lastg = dest[2];
398             lastb = dest[3];
399             upperlasta = old[0];
400             upperlastr = old[1];
401             upperlastg = old[2];
402             upperlastb = old[3];
403             dest+=4;
404             old+=4;
405             src+=4;
406         }
407     }    
408
409 }
410
411 void applyfilter1(int mode, U8*src, U8*old, U8*dest, int width)
412 {
413     int x;
414     unsigned char last=0;
415     unsigned char upperlast=0;
416
417     if(mode==0) {
418         for(x=0;x<width;x++) {
419             *dest = *src;
420             dest++;
421             src++;
422         }
423     }
424     else if(mode==1) {
425         for(x=0;x<width;x++) {
426             *dest = *src+last;
427             last = *dest;
428             dest++;
429             src++;
430         }
431     }
432     else if(mode==2) {
433         for(x=0;x<width;x++) {
434             *dest = *src+*old;
435             dest++;
436             old++;
437             src++;
438         }
439     }
440     else if(mode==3) {
441         for(x=0;x<width;x++) {
442             *dest = *src+(*old+last)/2;
443             last = *dest;
444             dest++;
445             old++;
446             src++;
447         }
448     }
449     else if(mode==4) {
450         for(x=0;x<width;x++) {
451             *dest = *src+PaethPredictor(last,*old,upperlast);
452             last = *dest;
453             upperlast = *old;
454             dest++;
455             old++;
456             src++;
457         }
458     }    
459
460 }
461
462 TAG* PNG2Image(TAG*t, U16 id, char*filename, int*width, int*height)
463 {
464     char tagid[4];
465     U8*data;
466     U8*imagedata;
467     U8*zimagedata=0;
468     unsigned long int imagedatalen;
469     unsigned long int zimagedatalen=0;
470     U8*palette = 0;
471     int palettelen = 0;
472     U8*alphapalette = 0;
473     int alphapalettelen = 0;
474     struct png_header header;
475     int bypp;
476     U8 alphacolor[3];
477     int hasalphacolor=0;
478     int len;
479
480
481     FILE *fi;
482     U8 *scanline;
483
484     if ((fi = fopen(filename, "rb")) == NULL) {
485         if (VERBOSE(1))
486             fprintf(stderr, "Read access failed: %s\n", filename);
487         return t;
488     }
489
490     if(!png_read_header(fi, &header))
491         return 0;
492
493     *width = header.width;
494     *height = header.height;
495
496     if(header.mode == 3 || header.mode == 0) bypp = 1;
497     else
498     if(header.mode == 2) bypp = 3;
499     else
500     if(header.mode == 6) bypp = 4;
501     else
502         return 0;
503     imagedatalen = bypp * header.width * header.height + 65536;
504     imagedata = malloc(imagedatalen);
505
506     fseek(fi,8,SEEK_SET);
507     while(!feof(fi))
508     {
509         if(!png_read_chunk(&tagid, &len, &data, fi))
510             break;
511         if(!strncmp(tagid, "IEND", 4)) {
512             break;
513         }
514         if(!strncmp(tagid, "PLTE", 4)) {
515             palette = data;
516             palettelen = len/3;
517             data = 0; //don't free data
518             if(VERBOSE(2))
519                 printf("%d colors in palette\n", palettelen);
520         }
521         if(!strncmp(tagid, "tRNS", 4)) {
522             if(header.mode == 3) {
523                 alphapalette = data;
524                 alphapalettelen = len;
525                 data = 0; //don't free data
526                 if(VERBOSE(2))
527                     printf("found %d alpha colors\n", alphapalettelen);
528             } else if(header.mode == 0 || header.mode == 2) {
529                 int t;
530                 if(header.mode == 2) { // palette or grayscale?
531                     alphacolor[0] = data[1];
532                     alphacolor[1] = data[3];
533                     alphacolor[2] = data[5];
534                 } else {
535                     alphacolor[0] = alphacolor[1] = alphacolor[2] = data[1];
536                 }
537                 if(VERBOSE(2))
538                     printf("found alpha color: %02x%02x%02x\n", alphacolor[0], alphacolor[1], alphacolor[2]);
539                 hasalphacolor = 1;
540             } else {
541                 if(VERBOSE(2))
542                     printf("ignoring tRNS %d entry (%d bytes)\n", header.mode, len);
543             }
544         }
545         if(!strncmp(tagid, "IDAT", 4)) {
546             if(!zimagedata) {
547                 zimagedatalen = len;
548                 zimagedata = malloc(len);
549                 memcpy(zimagedata,data,len);
550             } else {
551                 zimagedata = realloc(zimagedata, zimagedatalen+len);
552                 memcpy(&zimagedata[zimagedatalen], data, len);
553                 zimagedatalen += len;
554             }
555         }
556         if(data)
557             free(data);
558     }
559     
560     if(!zimagedata || uncompress(imagedata, &imagedatalen, zimagedata, zimagedatalen) != Z_OK) {
561         fprintf(stderr, "Couldn't uncompress IDAT chunk (%d bytes) in %s!\n", imagedatalen, filename);
562         if(zimagedata)
563             free(zimagedata);
564         return 0;
565     }
566     free(zimagedata);
567
568     if(alphapalette || hasalphacolor)
569         t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS2);
570     else
571         t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS);
572
573     swf_SetU16(t, id);          // id
574     if(header.mode == 2 || header.mode == 6) {
575         U8*data2 = malloc(header.width*header.height*4);
576         int i,s=0;
577         int x,y;
578         int pos=0;
579         int opaque=0;
580         int transparent=0;
581         int semitransparent=0;
582         /* in case for mode 2, the following also performs 24->32 bit conversion */
583         unsigned char* firstline = malloc(header.width*4);
584
585         for(y=0;y<header.height;y++) {
586             int mode = imagedata[pos++]; //filter mode
587             U8*src;
588             U8*dest;
589             U8*old;
590             dest = &data2[(y*header.width)*4];
591
592             if(header.bpp == 8)
593             {
594                 /* one byte per pixel */
595                 src = &imagedata[pos];
596                 pos+=header.width*(header.mode==6?4:3);
597             } else {
598                 /* not implemented yet */
599                 exit(1);
600             }
601
602             if(!y) {
603                 old = firstline;
604                 memset(old, 0, header.width*4); //TODO: fill alpha with 255?
605             } else {
606                 old = &data2[(y-1)*header.width*4];
607             }
608             if(header.mode==6) {
609                 applyfilter4(mode, src, old, dest, header.width);
610             } else if(header.mode==2) {
611                 applyfilter3(mode, src, old, dest, header.width);
612                 /* replace alpha color */
613                 if(hasalphacolor) {
614                     int x;
615                     for(x=0;x<header.width;x++) {
616                         if(dest[x*4+1] == alphacolor[0] &&
617                            dest[x*4+2] == alphacolor[1] &&
618                            dest[x*4+3] == alphacolor[2]) {
619                             *(U32*)&dest[x*4] = 0;
620                         }
621                     }
622                 }
623             }
624         }
625         free(firstline);
626
627         /* the image is now compressed and stored in data. Now let's take
628            a look at the alpha values */
629         if(header.mode == 6) {
630             for(y=0;y<header.height;y++) {
631                 U8*l = &data2[(y*header.width)*4];
632                 for(x=0;x<header.width;x++) {
633                     U8 a = l[x*4];
634                     U8 b = l[x*4+1];
635                     U8 g = l[x*4+2];
636                     U8 r = l[x*4+3];
637                     if(a==255) transparent++;
638                     else {
639                         if(a==0) opaque++;
640                         else semitransparent++;
641                         l[x*4+3]=(int)r*a/255;
642                         l[x*4+2]=(int)g*a/255;
643                         l[x*4+1]=(int)b*a/255;
644                     }
645                 }
646             }
647             if(semitransparent || opaque) {
648                 t->id = ST_DEFINEBITSLOSSLESS2;
649             }
650         }
651         swf_SetLosslessBits(t, header.width, header.height, data2, BMF_32BIT);
652         free(data2);
653     } else if(header.mode == 0 || header.mode == 3) {
654         RGBA*rgba;
655         int swf_width = BYTES_PER_SCANLINE(header.width);
656         U8*data2 = malloc(swf_width*header.height);
657         U8*tmpline = malloc(header.width);
658         int i,x,y;
659         int pos=0;
660         if(header.mode == 3) { // palette or grayscale?
661             rgba = (RGBA*)malloc(palettelen*sizeof(RGBA));
662             if(!palette) {
663                 fprintf(stderr, "Error: No palette found!\n");
664                 exit(1);
665             }
666             /* 24->32 bit conversion */
667             for(i=0;i<palettelen;i++) {
668                 rgba[i].r = palette[i*3+0];
669                 rgba[i].g = palette[i*3+1];
670                 rgba[i].b = palette[i*3+2];
671                 if(alphapalette && i<alphapalettelen) {
672                     rgba[i].a = alphapalette[i];
673                     rgba[i].r = ((int)rgba[i].r*rgba[i].a)/255;
674                     rgba[i].g = ((int)rgba[i].g*rgba[i].a)/255;
675                     rgba[i].b = ((int)rgba[i].b*rgba[i].a)/255;
676                 } else {
677                     rgba[i].a = 255;
678                 }
679                 if(hasalphacolor) {
680                     if(rgba[i].r == alphacolor[0] &&
681                        rgba[i].g == alphacolor[1] &&
682                        rgba[i].b == alphacolor[2]) {
683                         rgba[i].r = 0;
684                         rgba[i].g = 0;
685                         rgba[i].b = 0;
686                         rgba[i].a = 0;
687                     }
688                 }
689             }
690         } else {
691             int mult = (0x1ff>>header.bpp);
692             palettelen = 1<<header.bpp;
693             rgba = (RGBA*)malloc(palettelen*sizeof(RGBA));
694             for(i=0;i<palettelen;i++) {
695                 rgba[i].r = i*mult;
696                 rgba[i].g = i*mult;
697                 rgba[i].b = i*mult;
698                 rgba[i].a = 255;
699                 if(hasalphacolor) {
700                     if(rgba[i].r == alphacolor[0]) {
701                         rgba[i].r = 0;
702                         rgba[i].g = 0;
703                         rgba[i].b = 0;
704                         rgba[i].a = 0;
705                     }
706                 }
707             }
708         }
709
710         for(y=0;y<header.height;y++) {
711             int mode = imagedata[pos++]; //filter mode
712             U8*old;
713             U8*dest = &data2[y*swf_width];
714             U8*src;
715             src = &imagedata[pos];
716             if(header.bpp == 8) {
717                 pos+=header.width;
718             } else {
719                 int x,s=0;
720                 int bitpos = 0;
721                 U32 v = (1<<header.bpp)-1;
722                 for(x=0;x<header.width;x++) {
723                     U32 r = src[s/8]<<8 | 
724                             src[s/8+1];
725                     int t;
726                     tmpline[x] = (r>>(16-header.bpp-(s&7)))&v;
727                     s+=header.bpp;
728                 }
729                 src = tmpline;
730                 pos+=(header.width*header.bpp+7)/8;
731             }
732
733             if(!y) {
734                 memset(data2,0,swf_width);
735                 old = &data2[y*swf_width];
736             } else {
737                 old = &data2[(y-1)*swf_width];
738             }
739             applyfilter1(mode, src, old, dest, header.width);
740         }
741         swf_SetLosslessBitsIndexed(t, header.width, header.height, data2, rgba, palettelen);
742         free(tmpline);
743         free(rgba);
744         free(data2);
745     }
746     fclose(fi);
747     return t;
748 }
749
750 TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
751 {
752     SHAPE *s;
753     SRECT r;
754     MATRIX m;
755     int fs;
756
757     int width=0, height=0;
758
759     if(global.mkjpeg) {
760         RGBA*data = 0;
761         getPNG(sname, &width, &height, (unsigned char**)&data);
762         if(!data) 
763             exit(1);
764         if(swf_ImageHasAlpha(data, width, height)) {
765             t = swf_InsertTag(t, ST_DEFINEBITSJPEG3);
766             swf_SetU16(t, id);
767             swf_SetJPEGBits3(t, width,height,data,global.mkjpeg);
768         } else {
769             t = swf_InsertTag(t, ST_DEFINEBITSJPEG2);
770             swf_SetU16(t, id);
771             swf_SetJPEGBits2(t, width,height,data,global.mkjpeg);
772         }
773     } else if(1) {
774         RGBA*data = 0;
775         getPNG(sname, &width, &height, (unsigned char**)&data);
776         if(!data) 
777             exit(1);
778         t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS);
779         swf_SetU16(t, id);
780         swf_SetLosslessImage(t, data,width,height);
781     } else {
782         /* old code: transform PNG encoding 1:1 to SWF */
783         t = PNG2Image(t, id, sname, &width, &height);
784     }
785
786     t = swf_InsertTag(t, ST_DEFINESHAPE3);
787
788     swf_ShapeNew(&s);
789     swf_GetMatrix(NULL, &m);
790     m.sx = (int)(20 * 0x10000);
791     m.sy = (int)(20 * 0x10000);
792     m.tx = -10;
793     m.ty = -10;
794     fs = swf_ShapeAddBitmapFillStyle(s, &m, id, 1);
795
796     swf_SetU16(t, id + 1);      // id
797
798     r.xmin = r.ymin = 0;
799     r.xmax = width * 20;
800     r.ymax = height * 20;
801     swf_SetRect(t, &r);
802
803     swf_SetShapeHeader(t, s);
804
805     swf_ShapeSetAll(t, s, 0, 0, 0, fs, 0);
806     swf_ShapeSetLine(t, s, r.xmax, 0);
807     swf_ShapeSetLine(t, s, 0, r.ymax);
808     swf_ShapeSetLine(t, s, -r.xmax, 0);
809     swf_ShapeSetLine(t, s, 0, -r.ymax);
810
811     swf_ShapeSetEnd(t);
812
813     t = swf_InsertTag(t, ST_REMOVEOBJECT2);
814     swf_SetU16(t, 50);          // depth
815
816     t = swf_InsertTag(t, ST_PLACEOBJECT2);
817
818     swf_GetMatrix(NULL, &m);
819     m.sx = (int)(0x10000 * global.scale);
820     m.sy = (int)(0x10000 * global.scale);
821
822     if(custom_move) {
823         m.tx = move_x*20;
824         m.ty = move_y*20;
825     } else {
826         m.tx = (swf->movieSize.xmax - (int) (width * global.scale * 20)) / 2;
827         m.ty = (swf->movieSize.ymax - (int) (height * global.scale * 20)) / 2;
828     }
829     swf_ObjectPlace(t, id + 1, 50, &m, NULL, NULL);
830
831     t = swf_InsertTag(t, ST_SHOWFRAME);
832
833     return t;
834 }
835
836
837 int CheckInputFile(char *fname, char **realname)
838 {
839     FILE *fi;
840     char *s = malloc(strlen(fname) + 5);
841     struct png_header head;
842
843     if (!s)
844         exit(2);
845     (*realname) = s;
846     strcpy(s, fname);
847
848     // Check whether file exists (with typical extensions)
849
850     if ((fi = fopen(s, "rb")) == NULL) {
851         sprintf(s, "%s.png", fname);
852         if ((fi = fopen(s, "rb")) == NULL) {
853             sprintf(s, "%s.PNG", fname);
854             if ((fi = fopen(s, "rb")) == NULL) {
855                 sprintf(s, "%s.Png", fname);
856                 if ((fi = fopen(s, "rb")) == NULL) {
857                     fprintf(stderr, "Couldn't open %s!\n", fname);
858                     return -1;
859                 }
860             }
861         }
862     }
863
864     if(!png_read_header(fi, &head)) {
865         fprintf(stderr, "%s is not a PNG file!\n", fname);
866         return -1;
867     }
868
869     if (global.max_image_width < head.width)
870         global.max_image_width = head.width;
871     if (global.max_image_height < head.height)
872         global.max_image_height = head.height;
873
874     fclose(fi);
875
876     return 0;
877 }
878
879 int args_callback_option(char *arg, char *val)
880 {
881     int res = 0;
882     if (arg[1])
883         res = -1;
884     else
885         switch (arg[0]) {
886         case 'r':
887             if (val)
888                 global.framerate = atof(val);
889             /* removed framerate>0 restriction in order to make
890                Flash Communication Server compatible SWFs */
891             if ((global.framerate < 0) ||(global.framerate >= 256.0)) {
892                 if (VERBOSE(1))
893                     fprintf(stderr,
894                             "Error: You must specify a valid framerate between 1/256 and 255.\n");
895                 exit(1);
896             }
897             res = 1;
898             break;
899
900         case 'o':
901             if (val)
902                 global.outfile = val;
903             res = 1;
904             break;
905
906         case 's':
907             global.scale = atof(val)/100;
908             res = 1;
909             break;
910
911         case 'z':
912             if(global.version<6)
913                 global.version = 6;
914             res = 0;
915             break;
916
917         case 'j':
918             global.mkjpeg = atoi(val);
919             res = 1;
920             break;
921
922         case 'T':
923             global.version = atoi(val);
924             res = 1;
925             break;
926
927         case 'C':
928             global.do_cgi = 1;
929             break;
930
931         case 'v':
932             global.verbose++;
933             res = 0;
934             break;
935
936         case 'q':
937             global.verbose--;
938             if(global.verbose<0)
939                 global.verbose = 0;
940             res = 0;
941             break;
942
943         case 'X':
944             if (val)
945                 global.force_width = atoi(val);
946             res = 1;
947             break;
948
949         case 'Y':
950             if (val)
951                 global.force_height = atoi(val);
952             res = 1;
953             break;
954         
955         case 'V':
956             printf("png2swf - part of %s %s\n", PACKAGE, VERSION);
957             exit(0);
958    
959         case 'c': {
960             char*s = strdup(val);
961             char*x1 = strtok(s, ":");
962             char*y1 = strtok(0, ":");
963             char*x2 = strtok(0, ":");
964             char*y2 = strtok(0, ":");
965             if(!(x1 && y1 && x2 && y2)) {
966                 fprintf(stderr, "-m option requires four arguments, <x1>:<y1>:<x2>:<y2>\n");
967                 exit(1);
968             }
969             custom_clip = 1;
970             clip_x1 = atoi(x1);
971             clip_y1 = atoi(y1);
972             clip_x2 = atoi(x2);
973             clip_y2 = atoi(y2);
974             free(s);
975
976             res = 1;
977             break;
978         }
979
980         case 'm': {
981             char*s = strdup(val);
982             char*c = strchr(s, ':');
983             if(!c) {
984                 fprintf(stderr, "-m option requires two arguments, <x>:<y>\n");
985                 exit(1);
986             }
987             *c = 0;
988             custom_move = 1;
989             move_x = atoi(val);
990             move_y = atoi(c+1);
991             free(s);
992
993             res = 1;
994             break;
995         }
996
997         default:
998             res = -1;
999             break;
1000         }
1001
1002     if (res < 0) {
1003         if (VERBOSE(1))
1004             fprintf(stderr, "Unknown option: -%s\n", arg);
1005         exit(1);
1006         return 0;
1007     }
1008     return res;
1009 }
1010
1011 static struct options_t options[] = {
1012 {"r", "rate"},
1013 {"o", "output"},
1014 {"j", "jpeg"},
1015 {"z", "zlib"},
1016 {"T", "flashversion"},
1017 {"X", "pixel"},
1018 {"Y", "pixel"},
1019 {"v", "verbose"},
1020 {"q", "quiet"},
1021 {"C", "cgi"},
1022 {"V", "version"},
1023 {"s", "scale"},
1024 {0,0}
1025 };
1026
1027 int args_callback_longoption(char *name, char *val)
1028 {
1029     return args_long2shortoption(options, name, val);
1030 }
1031
1032 int args_callback_command(char *arg, char *next)        // actually used as filename
1033 {
1034     char *s;
1035     if (CheckInputFile(arg, &s) < 0) {
1036         if (VERBOSE(1))
1037             fprintf(stderr, "Error opening input file: %s\n", arg);
1038         free(s);
1039     } else {
1040         image[global.nfiles].filename = s;
1041         global.nfiles++;
1042         if (global.nfiles >= MAX_INPUT_FILES) {
1043             if (VERBOSE(1))
1044                 fprintf(stderr, "Error: Too many input files.\n");
1045             exit(1);
1046         }
1047     }
1048     return 0;
1049 }
1050
1051 void args_callback_usage(char *name)
1052 {
1053     printf("\n");
1054     printf("Usage: %s [-X width] [-Y height] [-o file.swf] [-r rate] file1.png [file2.png...]\n", name);
1055     printf("\n");
1056     printf("-r , --rate <framerate>        Set movie framerate (frames per second)\n");
1057     printf("-o , --output <filename>       Set name for SWF output file.\n");
1058     printf("-j , --jpeg <quality>          Generate a lossy jpeg bitmap inside the SWF, with a given quality (1-100)\n");
1059     printf("-z , --zlib <zlib>             Enable Flash 6 (MX) Zlib Compression\n");
1060     printf("-T , --flashversion            Set the flash version to generate\n");
1061     printf("-X , --pixel <width>           Force movie width to <width> (default: autodetect)\n");
1062     printf("-Y , --pixel <height>          Force movie height to <height> (default: autodetect)\n");
1063     printf("-v , --verbose <level>         Set verbose level (0=quiet, 1=default, 2=debug)\n");
1064     printf("-q , --quiet                   Omit normal log messages, only log errors\n");
1065     printf("-C , --cgi                     For use as CGI- prepend http header, write to stdout\n");
1066     printf("-V , --version                 Print version information and exit\n");
1067     printf("-s , --scale <percent>         Scale image to <percent>%% size.\n");
1068     printf("\n");
1069 }
1070
1071 int main(int argc, char **argv)
1072 {
1073     SWF swf;
1074     TAG *t;
1075
1076     memset(&global, 0x00, sizeof(global));
1077
1078     global.framerate = 1.0;
1079     global.verbose = 1;
1080     global.version = 6;
1081     global.scale = 1.0;
1082
1083     processargs(argc, argv);
1084     
1085     if(global.nfiles<=0) {
1086         fprintf(stderr, "No png files found in arguments\n");
1087         return 1;
1088     }
1089
1090     if (VERBOSE(2))
1091         fprintf(stderr, "Processing %i file(s)...\n", global.nfiles);
1092
1093     t = MovieStart(&swf, global.framerate,
1094                    global.force_width ? global.force_width : (int)(global.max_image_width*global.scale),
1095                    global.force_height ? global.force_height : (int)(global.max_image_height*global.scale));
1096
1097     {
1098         int i;
1099         for (i = 0; i < global.nfiles; i++) {
1100             if (VERBOSE(3))
1101                 fprintf(stderr, "[%03i] %s\n", i,
1102                         image[i].filename);
1103             t = MovieAddFrame(&swf, t, image[i].filename, (i * 2) + 1);
1104             free(image[i].filename);
1105         }
1106     }
1107
1108     MovieFinish(&swf, t, global.outfile);
1109
1110     return 0;
1111 }