new parameter addspacechars
[swftools.git] / src / png2swf.c
index 541675d..05366fc 100644 (file)
@@ -26,6 +26,7 @@
 #include <zlib.h>
 #include "../lib/rfxswf.h"
 #include "../lib/args.h"
+#include "../lib/png.h"
 
 #define MAX_INPUT_FILES 1024
 #define VERBOSE(x) (global.verbose>=x)
@@ -41,6 +42,8 @@ struct {
     int do_cgi;
     int version;
     char *outfile;
+    int mkjpeg;
+    float scale;
 } global;
 
 struct {
@@ -97,13 +100,8 @@ int MovieFinish(SWF * swf, TAG * t, char *sname)
     if(global.do_cgi) {
        if FAILED(swf_WriteCGI(swf)) fprintf(stderr,"WriteCGI() failed.\n");
     } else {
-       if(global.version >= 6) {
-           if (swf_WriteSWC(f, swf)<0) 
-                   fprintf(stderr, "Unable to write output file: %s\n", sname);
-       } else {
-           if (swf_WriteSWF(f, swf)<0) 
-                   fprintf(stderr, "Unable to write output file: %s\n", sname);
-       }
+        if (swf_WriteSWF(f, swf)<0) 
+            fprintf(stderr, "Unable to write output file: %s\n", sname);
        if (f != so)
            close(f);
     }
@@ -121,7 +119,7 @@ int png_read_chunk(char (*head)[4], int*destlen, U8**destdata, FILE*fi)
        return 0;
     if(!fread(head, 4, 1, fi))
        return 0;
-    len = REVERSESWAP32(len);
+    len = BE_32_TO_NATIVE(len);
     if(destlen) *destlen = len;
     if(destdata) {
        if(len)
@@ -145,7 +143,7 @@ unsigned int png_get_dword(FILE*fi)
 {
     unsigned int a;
     fread(&a,4,1,fi);
-    return REVERSESWAP32(a);
+    return BE_32_TO_NATIVE(a);
 }
 
 struct png_header
@@ -165,7 +163,7 @@ int png_read_header(FILE*fi, struct png_header*header)
     U8 head2[8];
     U8*data;
     fread(head2,8,1,fi);
-    if(strncmp(head,head2,4))
+    if(strncmp((char*)head,(char*)head2,4))
        return 0;
    
     while(png_read_chunk(&id, &len, &data, fi))
@@ -175,10 +173,10 @@ int png_read_header(FILE*fi, struct png_header*header)
        if(!strncasecmp(id, "IHDR", 4)) {
            char a,b,c,f,i;
            if(len < 8) exit(1);
-           header->width = REVERSESWAP32(*(U32*)&data[0]);
-           header->height = REVERSESWAP32(*(U32*)&data[4]);
+           header->width = BE_32_TO_NATIVE(*(U32*)&data[0]);
+           header->height = BE_32_TO_NATIVE(*(U32*)&data[4]);
            a = data[8];      // should be 8
-           b = data[9];      // should be 3(indexed) or 2(rgb)
+           b = data[9];      // should be 3(indexed), 2(rgb), 0(grayscale) or 6(truecolor+alpha)
 
            c = data[10];     // compression mode (0)
            f = data[11];     // filter mode (0)
@@ -199,7 +197,7 @@ int png_read_header(FILE*fi, struct png_header*header)
                exit(1);
            }
            if(a!=8 && (b==2 || b==6)) {
-               fprintf(stderr, "Bpp %d in mode %d not supported!\n", a);
+               fprintf(stderr, "Bpp %d in mode %d not supported!\n", b, a);
                exit(1);
            }
            if(c!=0) {
@@ -319,7 +317,6 @@ void applyfilter3(int mode, U8*src, U8*old, U8*dest, int width)
            src+=3;
        }
     }    
-
 }
 
 void applyfilter4(int mode, U8*src, U8*old, U8*dest, int width)
@@ -457,15 +454,9 @@ void applyfilter1(int mode, U8*src, U8*old, U8*dest, int width)
 
 }
 
-TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
+TAG* PNG2Image(TAG*t, U16 id, char*filename, int*width, int*height)
 {
-    SHAPE *s;
-    SRECT r;
-    MATRIX m;
-    int fs;
-
     char tagid[4];
-    int len;
     U8*data;
     U8*imagedata;
     U8*zimagedata=0;
@@ -477,19 +468,26 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
     int alphapalettelen = 0;
     struct png_header header;
     int bypp;
+    U8 alphacolor[3];
+    int hasalphacolor=0;
+    int len;
+
 
     FILE *fi;
     U8 *scanline;
 
-    if ((fi = fopen(sname, "rb")) == NULL) {
+    if ((fi = fopen(filename, "rb")) == NULL) {
        if (VERBOSE(1))
-           fprintf(stderr, "Read access failed: %s\n", sname);
+           fprintf(stderr, "Read access failed: %s\n", filename);
        return t;
     }
 
     if(!png_read_header(fi, &header))
        return 0;
 
+    *width = header.width;
+    *height = header.height;
+
     if(header.mode == 3 || header.mode == 0) bypp = 1;
     else
     if(header.mode == 2) bypp = 3;
@@ -522,6 +520,21 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
                data = 0; //don't free data
                if(VERBOSE(2))
                    printf("found %d alpha colors\n", alphapalettelen);
+           } else if(header.mode == 0 || header.mode == 2) {
+               int t;
+               if(header.mode == 2) { // palette or grayscale?
+                   alphacolor[0] = data[1];
+                   alphacolor[1] = data[3];
+                   alphacolor[2] = data[5];
+               } else {
+                   alphacolor[0] = alphacolor[1] = alphacolor[2] = data[1];
+               }
+               if(VERBOSE(2))
+                   printf("found alpha color: %02x%02x%02x\n", alphacolor[0], alphacolor[1], alphacolor[2]);
+               hasalphacolor = 1;
+           } else {
+               if(VERBOSE(2))
+                   printf("ignoring tRNS %d entry (%d bytes)\n", header.mode, len);
            }
        }
        if(!strncmp(tagid, "IDAT", 4)) {
@@ -540,14 +553,14 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
     }
     
     if(!zimagedata || uncompress(imagedata, &imagedatalen, zimagedata, zimagedatalen) != Z_OK) {
-       fprintf(stderr, "Couldn't uncompress %s!\n", sname);
+       fprintf(stderr, "Couldn't uncompress IDAT chunk (%lu bytes) in %s!\n", imagedatalen, filename);
        if(zimagedata)
            free(zimagedata);
        return 0;
     }
     free(zimagedata);
 
-    if(alphapalette)
+    if(alphapalette || hasalphacolor)
        t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS2);
     else
        t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS);
@@ -587,10 +600,22 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
            } else {
                old = &data2[(y-1)*header.width*4];
            }
-           if(header.mode==6)
+           if(header.mode==6) {
                applyfilter4(mode, src, old, dest, header.width);
-           else if(header.mode==2)
+           } else if(header.mode==2) {
                applyfilter3(mode, src, old, dest, header.width);
+               /* replace alpha color */
+               if(hasalphacolor) {
+                   int x;
+                   for(x=0;x<header.width;x++) {
+                       if(dest[x*4+1] == alphacolor[0] &&
+                          dest[x*4+2] == alphacolor[1] &&
+                          dest[x*4+3] == alphacolor[2]) {
+                           *(U32*)&dest[x*4] = 0;
+                       }
+                   }
+               }
+           }
        }
        free(firstline);
 
@@ -646,6 +671,16 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
                } else {
                    rgba[i].a = 255;
                }
+               if(hasalphacolor) {
+                   if(rgba[i].r == alphacolor[0] &&
+                      rgba[i].g == alphacolor[1] &&
+                      rgba[i].b == alphacolor[2]) {
+                       rgba[i].r = 0;
+                       rgba[i].g = 0;
+                       rgba[i].b = 0;
+                       rgba[i].a = 0;
+                   }
+               }
            }
        } else {
             int mult = (0x1ff>>header.bpp);
@@ -655,6 +690,15 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
                rgba[i].r = i*mult;
                rgba[i].g = i*mult;
                rgba[i].b = i*mult;
+               rgba[i].a = 255;
+               if(hasalphacolor) {
+                   if(rgba[i].r == alphacolor[0]) {
+                       rgba[i].r = 0;
+                       rgba[i].g = 0;
+                       rgba[i].b = 0;
+                       rgba[i].a = 0;
+                   }
+               }
            }
        }
 
@@ -694,22 +738,69 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
        free(rgba);
        free(data2);
     }
+    fclose(fi);
+    return t;
+}
+
+TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
+{
+    SHAPE *s;
+    SRECT r;
+    MATRIX m;
+    int fs;
+
+    int width=0, height=0;
+
+#ifndef HAVE_JPEGLIB
+    if(global.mkjpeg) {
+        global.mkjpeg = 0;
+        msg("<warning> No jpeg support compiled in");
+    }
+#endif
+    if(global.mkjpeg) {
+#ifdef HAVE_JPEGLIB
+       RGBA*data = 0;
+       getPNG(sname, &width, &height, (unsigned char**)&data);
+       if(!data) 
+           exit(1);
+       if(swf_ImageHasAlpha(data, width, height)) {
+           t = swf_InsertTag(t, ST_DEFINEBITSJPEG3);
+           swf_SetU16(t, id);
+           swf_SetJPEGBits3(t, width,height,data,global.mkjpeg);
+       } else {
+           t = swf_InsertTag(t, ST_DEFINEBITSJPEG2);
+           swf_SetU16(t, id);
+           swf_SetJPEGBits2(t, width,height,data,global.mkjpeg);
+       }
+#endif
+    } else if(1) {
+       RGBA*data = 0;
+       getPNG(sname, &width, &height, (unsigned char**)&data);
+       if(!data) 
+           exit(1);
+       t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS);
+       swf_SetU16(t, id);
+       swf_SetLosslessImage(t, data,width,height);
+    } else {
+       /* old code: transform PNG encoding 1:1 to SWF */
+       t = PNG2Image(t, id, sname, &width, &height);
+    }
 
     t = swf_InsertTag(t, ST_DEFINESHAPE3);
 
     swf_ShapeNew(&s);
     swf_GetMatrix(NULL, &m);
-    m.sx = 20 * 0x10000;
-    m.sy = 20 * 0x10000;
-    m.tx = -10;
-    m.ty = -10;
+    m.sx = (int)(20 * 0x10000);
+    m.sy = (int)(20 * 0x10000);
+    m.tx = 0;
+    m.ty = 0;
     fs = swf_ShapeAddBitmapFillStyle(s, &m, id, 1);
 
     swf_SetU16(t, id + 1);     // id
 
     r.xmin = r.ymin = 0;
-    r.xmax = header.width * 20;
-    r.ymax = header.height * 20;
+    r.xmax = width * 20;
+    r.ymax = height * 20;
     swf_SetRect(t, &r);
 
     swf_SetShapeHeader(t, s);
@@ -728,19 +819,20 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
     t = swf_InsertTag(t, ST_PLACEOBJECT2);
 
     swf_GetMatrix(NULL, &m);
+    m.sx = (int)(0x10000 * global.scale);
+    m.sy = (int)(0x10000 * global.scale);
+
     if(custom_move) {
        m.tx = move_x*20;
        m.ty = move_y*20;
     } else {
-       m.tx = (swf->movieSize.xmax - (int) header.width * 20) / 2;
-       m.ty = (swf->movieSize.ymax - (int) header.height * 20) / 2;
+       m.tx = (swf->movieSize.xmax - (int) (width * global.scale * 20)) / 2;
+       m.ty = (swf->movieSize.ymax - (int) (height * global.scale * 20)) / 2;
     }
     swf_ObjectPlace(t, id + 1, 50, &m, NULL, NULL);
 
     t = swf_InsertTag(t, ST_SHOWFRAME);
 
-    fclose(fi);
-
     return t;
 }
 
@@ -814,11 +906,22 @@ int args_callback_option(char *arg, char *val)
            res = 1;
            break;
 
+       case 's':
+           global.scale = atof(val)/100;
+           res = 1;
+           break;
+
        case 'z':
-           global.version = 6;
+           if(global.version<6)
+               global.version = 6;
            res = 0;
            break;
 
+       case 'j':
+           global.mkjpeg = atoi(val);
+           res = 1;
+           break;
+
        case 'T':
            global.version = atoi(val);
            res = 1;
@@ -851,7 +954,7 @@ int args_callback_option(char *arg, char *val)
                global.force_height = atoi(val);
            res = 1;
            break;
-
+       
        case 'V':
            printf("png2swf - part of %s %s\n", PACKAGE, VERSION);
            exit(0);
@@ -911,13 +1014,16 @@ int args_callback_option(char *arg, char *val)
 static struct options_t options[] = {
 {"r", "rate"},
 {"o", "output"},
+{"j", "jpeg"},
 {"z", "zlib"},
+{"T", "flashversion"},
 {"X", "pixel"},
 {"Y", "pixel"},
 {"v", "verbose"},
 {"q", "quiet"},
 {"C", "cgi"},
 {"V", "version"},
+{"s", "scale"},
 {0,0}
 };
 
@@ -952,13 +1058,16 @@ void args_callback_usage(char *name)
     printf("\n");
     printf("-r , --rate <framerate>        Set movie framerate (frames per second)\n");
     printf("-o , --output <filename>       Set name for SWF output file.\n");
+    printf("-j , --jpeg <quality>          Generate a lossy jpeg bitmap inside the SWF, with a given quality (1-100)\n");
     printf("-z , --zlib <zlib>             Enable Flash 6 (MX) Zlib Compression\n");
+    printf("-T , --flashversion            Set the flash version to generate\n");
     printf("-X , --pixel <width>           Force movie width to <width> (default: autodetect)\n");
     printf("-Y , --pixel <height>          Force movie height to <height> (default: autodetect)\n");
-    printf("-v , --verbose                 Be verbose. Use more than one -v for greater effect \n");
+    printf("-v , --verbose <level>         Set verbose level (0=quiet, 1=default, 2=debug)\n");
     printf("-q , --quiet                   Omit normal log messages, only log errors\n");
     printf("-C , --cgi                     For use as CGI- prepend http header, write to stdout\n");
     printf("-V , --version                 Print version information and exit\n");
+    printf("-s , --scale <percent>         Scale image to <percent>%% size.\n");
     printf("\n");
 }
 
@@ -971,7 +1080,8 @@ int main(int argc, char **argv)
 
     global.framerate = 1.0;
     global.verbose = 1;
-    global.version = 4;
+    global.version = 8;
+    global.scale = 1.0;
 
     processargs(argc, argv);
     
@@ -984,10 +1094,8 @@ int main(int argc, char **argv)
        fprintf(stderr, "Processing %i file(s)...\n", global.nfiles);
 
     t = MovieStart(&swf, global.framerate,
-                  global.force_width ? global.force_width : global.
-                  max_image_width,
-                  global.force_height ? global.force_height : global.
-                  max_image_height);
+                  global.force_width ? global.force_width : (int)(global.max_image_width*global.scale),
+                  global.force_height ? global.force_height : (int)(global.max_image_height*global.scale));
 
     {
        int i;