implemented --framerate.
[swftools.git] / src / png2swf.c
index a720c3a..927504e 100644 (file)
@@ -26,7 +26,6 @@ struct {
     int max_image_height;
     int force_width;
     int force_height;
-    int prescale;
     int nfiles;
     int verbose;
     char *outfile;
@@ -34,7 +33,6 @@ struct {
 
 struct {
     char *filename;
-    int scale;
 } image[MAX_INPUT_FILES];
 
 TAG *MovieStart(SWF * swf, int framerate, int dx, int dy)
@@ -89,7 +87,6 @@ int png_read_chunk(char (*head)[4], int*destlen, U8**destdata, FILE*fi)
     if(!fread(head, 4, 1, fi))
        return 0;
     len = REVERSESWAP32(len);
-    printf("id: %.4s len: %d\n", head, len);
     if(destlen) *destlen = len;
     if(destdata) {
        if(len)
@@ -128,6 +125,7 @@ int png_read_header(FILE*fi, struct png_header*header)
 {
     char id[4];
     int len;
+    int ok=0;
     U8 head[8] = {137,80,78,71,13,10,26,10};
     U8 head2[8];
     U8*data;
@@ -137,7 +135,8 @@ int png_read_header(FILE*fi, struct png_header*header)
    
     while(png_read_chunk(&id, &len, &data, fi))
     {
-       printf("%c%c%c%c %d\n", id[0],id[1],id[2],id[3],len);
+       if(VERBOSE(2))
+           printf("%c%c%c%c %d\n", id[0],id[1],id[2],id[3],len);
        if(!strncasecmp(id, "IHDR", 4)) {
            char a,b,c,f,i;
            if(len < 8) exit(1);
@@ -149,20 +148,269 @@ int png_read_header(FILE*fi, struct png_header*header)
            c = data[10];     // compression mode (0)
            f = data[11];     // filter mode (0)
            i = data[12];     // interlace mode (0)
-           printf("%08xx%08x %d %d %d %d %d\n",header->width, header->height, a,b,c,f,i);
+
+           if(b!=2 && b!=3 && b!=6) {
+               fprintf(stderr, "Image mode %d not supported!\n", b);
+               exit(1);
+           }
+           if(a!=8 && (b==2 || b==6)) {
+               fprintf(stderr, "Bpp %d in mode %d not supported!\n", a);
+               exit(1);
+           }
+           if(c!=0) {
+               fprintf(stderr, "Compression mode %d not supported!\n", c);
+               exit(1);
+           }
+           if(f!=0) {
+               fprintf(stderr, "Filter mode %d not supported!\n", f);
+               exit(1);
+           }
+           if(i!=0) {
+               fprintf(stderr, "Interlace mode %d not supported!\n", i);
+               exit(1);
+           }
+           if(VERBOSE(2))
+               printf("%dx%d %d %d %d %d %d\n",header->width, header->height, a,b,c,f,i);
            header->bpp = a;
            header->mode = b;
-           return 1;
-       } else {
-           fseek(fi, len, SEEK_CUR);
-       }
+           ok = 1;
+       } 
+       
        free(data);
     }
-    return 0;
+    return ok;
+}
+
+typedef unsigned char byte;
+#define ABS(a) ((a)>0?(a):(-(a)))
+byte inline PaethPredictor (byte a,byte b,byte c)
+{
+        // a = left, b = above, c = upper left
+        int p = a + b - c;        // initial estimate
+        int pa = ABS(p - a);      // distances to a, b, c
+        int pb = ABS(p - b);
+        int pc = ABS(p - c);
+        // return nearest of a,b,c,
+        // breaking ties in order a,b,c.
+        if (pa <= pb && pa <= pc) 
+               return a;
+        else if (pb <= pc) 
+               return b;
+        else return c;
 }
 
-TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int scale,
-                  int id)
+void applyfilter3(int mode, U8*src, U8*old, U8*dest, int width)
+{
+    int x;
+    unsigned char lastr=0;
+    unsigned char lastg=0;
+    unsigned char lastb=0;
+    unsigned char upperlastr=0;
+    unsigned char upperlastg=0;
+    unsigned char upperlastb=0;
+
+    if(mode==0) {
+       for(x=0;x<width;x++) {
+           dest[0] = 255;
+           dest[1] = src[0];
+           dest[2] = src[1];
+           dest[3] = src[2];
+           dest+=4;
+           src+=3;
+       }
+    }
+    else if(mode==1) {
+       for(x=0;x<width;x++) {
+           dest[0] = 255;
+           dest[1] = src[0]+lastr;
+           dest[2] = src[1]+lastg;
+           dest[3] = src[2]+lastb;
+           lastr = dest[1];
+           lastg = dest[2];
+           lastb = dest[3];
+           dest+=4;
+           src+=3;
+       }
+    }
+    else if(mode==2) {
+       for(x=0;x<width;x++) {
+           dest[0] = 255;
+           dest[1] = src[0]+old[1];
+           dest[2] = src[1]+old[2];
+           dest[3] = src[2]+old[3];
+           dest+=4;
+           old+=4;
+           src+=3;
+       }
+    }
+    else if(mode==3) {
+       for(x=0;x<width;x++) {
+           dest[0] = 255;
+           dest[1] = src[0]+(old[1]+lastr)/2;
+           dest[2] = src[1]+(old[2]+lastg)/2;
+           dest[3] = src[2]+(old[3]+lastb)/2;
+           lastr = dest[1];
+           lastg = dest[2];
+           lastb = dest[3];
+           dest+=4;
+           old+=4;
+           src+=3;
+       }
+    }
+    else if(mode==4) {
+       for(x=0;x<width;x++) {
+           dest[0] = 255;
+           dest[1] = src[0]+PaethPredictor(lastr,old[1],upperlastr);
+           dest[2] = src[1]+PaethPredictor(lastg,old[2],upperlastg);
+           dest[3] = src[2]+PaethPredictor(lastb,old[3],upperlastb);
+           lastr = dest[1];
+           lastg = dest[2];
+           lastb = dest[3];
+           upperlastr = old[1];
+           upperlastg = old[2];
+           upperlastb = old[3];
+           dest+=4;
+           old+=4;
+           src+=3;
+       }
+    }    
+
+}
+
+void applyfilter4(int mode, U8*src, U8*old, U8*dest, int width)
+{
+    int x;
+    unsigned char lastr=0;
+    unsigned char lastg=0;
+    unsigned char lastb=0;
+    unsigned char lasta=0;
+    unsigned char upperlastr=0;
+    unsigned char upperlastg=0;
+    unsigned char upperlastb=0;
+    unsigned char upperlasta=0;
+
+    if(mode==0) {
+       for(x=0;x<width;x++) {
+           dest[0] = src[3];
+           dest[1] = src[0];
+           dest[2] = src[1];
+           dest[3] = src[2];
+           dest+=4;
+           src+=4;
+       }
+    }
+    else if(mode==1) {
+       for(x=0;x<width;x++) {
+           dest[0] = src[3]+lasta;
+           dest[1] = src[0]+lastr;
+           dest[2] = src[1]+lastg;
+           dest[3] = src[2]+lastb;
+           lasta = dest[0];
+           lastr = dest[1];
+           lastg = dest[2];
+           lastb = dest[3];
+           dest+=4;
+           src+=4;
+       }
+    }
+    else if(mode==2) {
+       for(x=0;x<width;x++) {
+           dest[0] = src[3]+old[0];
+           dest[1] = src[0]+old[1];
+           dest[2] = src[1]+old[2];
+           dest[3] = src[2]+old[3];
+           dest+=4;
+           old+=4;
+           src+=4;
+       }
+    }
+    else if(mode==3) {
+       for(x=0;x<width;x++) {
+           dest[0] = src[3]+(old[0]+lasta)/2;
+           dest[1] = src[0]+(old[1]+lastr)/2;
+           dest[2] = src[1]+(old[2]+lastg)/2;
+           dest[3] = src[2]+(old[3]+lastb)/2;
+           lastr = dest[1];
+           lastg = dest[2];
+           lastb = dest[3];
+           dest+=4;
+           old+=4;
+           src+=4;
+       }
+    }
+    else if(mode==4) {
+       for(x=0;x<width;x++) {
+           dest[0] = src[3]+PaethPredictor(lasta,old[0],upperlasta);
+           dest[1] = src[0]+PaethPredictor(lastr,old[1],upperlastr);
+           dest[2] = src[1]+PaethPredictor(lastg,old[2],upperlastg);
+           dest[3] = src[2]+PaethPredictor(lastb,old[3],upperlastb);
+           lasta = dest[0];
+           lastr = dest[1];
+           lastg = dest[2];
+           lastb = dest[3];
+           upperlastr = old[0];
+           upperlastr = old[1];
+           upperlastg = old[2];
+           upperlastb = old[3];
+           dest+=4;
+           old+=4;
+           src+=4;
+       }
+    }    
+
+}
+
+void applyfilter1(int mode, U8*src, U8*old, U8*dest, int width)
+{
+    int x;
+    unsigned char last=0;
+    unsigned char upperlast=0;
+
+    if(mode==0) {
+       for(x=0;x<width;x++) {
+           *dest = *src;
+           dest++;
+           src++;
+       }
+    }
+    else if(mode==1) {
+       for(x=0;x<width;x++) {
+           *dest = *src+last;
+           last = *dest;
+           dest++;
+           src++;
+       }
+    }
+    else if(mode==2) {
+       for(x=0;x<width;x++) {
+           *dest = *src+*old;
+           dest++;
+           old++;
+           src++;
+       }
+    }
+    else if(mode==3) {
+       for(x=0;x<width;x++) {
+           *dest = *src+(*old+last)/2;
+           dest++;
+           old++;
+           src++;
+       }
+    }
+    else if(mode==4) {
+       for(x=0;x<width;x++) {
+           *dest = *src+PaethPredictor(last,*old,upperlast);
+           last = *dest;
+           upperlast = *old;
+           dest++;
+           old++;
+           src++;
+       }
+    }    
+
+}
+
+TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
 {
     SHAPE *s;
     SRECT r;
@@ -173,9 +421,13 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int scale,
     int len;
     U8*data;
     U8*imagedata;
+    U8*zimagedata=0;
     unsigned long int imagedatalen;
-    U8*palette;
-    int palettelen;
+    unsigned long int zimagedatalen=0;
+    U8*palette = 0;
+    int palettelen = 0;
+    U8*alphapalette = 0;
+    int alphapalettelen = 0;
     struct png_header header;
     int bypp;
 
@@ -195,6 +447,8 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int scale,
     else
     if(header.mode == 2) bypp = 3;
     else
+    if(header.mode == 6) bypp = 4;
+    else
        return 0;
     imagedatalen = bypp * header.width * header.height + 65536;
     imagedata = malloc(imagedatalen);
@@ -210,63 +464,148 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int scale,
        if(!strncmp(tagid, "PLTE", 4)) {
            palette = data;
            palettelen = len/3;
-           data = 0;
-           printf("%d palette\n", len);
+           data = 0; //don't free data
+           if(VERBOSE(2))
+               printf("%d colors in palette\n", palettelen);
+       }
+       if(!strncmp(tagid, "tRNS", 4)) {
+           if(header.mode == 3) {
+               alphapalette = data;
+               alphapalettelen = len;
+               data = 0; //don't free data
+               if(VERBOSE(2))
+                   printf("found %d alpha colors\n", alphapalettelen);
+           }
        }
        if(!strncmp(tagid, "IDAT", 4)) {
-           if(uncompress(imagedata, &imagedatalen, data, len) != Z_OK) {
-               fprintf(stderr, "Couldn't uncompress %s!\n", sname);
-               return 0;
+           if(!zimagedata) {
+               zimagedatalen = len;
+               zimagedata = malloc(len);
+               memcpy(zimagedata,data,len);
+           } else {
+               zimagedata = realloc(zimagedata, zimagedatalen+len);
+               memcpy(&zimagedata[zimagedatalen], data, len);
+               zimagedatalen += len;
            }
-           printf("IDAT %d -> %d\n", len, imagedatalen);
        }
        if(data)
            free(data);
     }
+    
+    if(!zimagedata || uncompress(imagedata, &imagedatalen, zimagedata, zimagedatalen) != Z_OK) {
+       fprintf(stderr, "Couldn't uncompress %s!\n", sname);
+       if(zimagedata)
+           free(zimagedata);
+       return 0;
+    }
+    free(zimagedata);
+
+    if(alphapalette)
+       t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS2);
+    else
+       t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS);
 
-    t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS);
     swf_SetU16(t, id);         // id
-    if(header.mode == 2) {
+    if(header.mode == 2 || header.mode == 6) {
        U8*data2 = malloc(header.width*header.height*4);
        int i,s=0;
        int x,y;
        int pos=0;
-       /* 24->32 bit conversion */
+       /* in case for mode 2, the following also performs 24->32 bit conversion */
        for(y=0;y<header.height;y++) {
-           pos++; // filter mode
-           for(x=0;x<header.width;x++) {
-               U8*src  = &imagedata[pos];
-               U8*dest = &data2[(y*header.width+x)*4];
-               dest[0] = 255;
-               dest[1] = src[0];
-               dest[2] = src[1];
-               dest[3] = src[2];
-               pos+=3;
+           int mode = imagedata[pos++]; //filter mode
+           U8*src;
+           U8*dest;
+           U8*old;
+           dest = &data2[(y*header.width)*4];
+
+           if(header.bpp == 8)
+           {
+               /* one byte per pixel */
+               src = &imagedata[pos];
+               pos+=header.width*(header.mode==6?4:3);
+           } else {
+               /* not implemented yet */
+               exit(1);
+           }
+
+           if(!y) {
+               memset(data2,0,header.width*4);
+               old = &data2[y*header.width*4];
+           } else {
+               old = &data2[(y-1)*header.width*4];
            }
+           if(header.mode==6)
+               applyfilter4(mode, src, old, dest, header.width);
+           else
+               applyfilter3(mode, src, old, dest, header.width);
        }
        swf_SetLosslessBits(t, header.width, header.height, data2, BMF_32BIT);
        free(data2);
     }
     else {
        RGBA*rgba = (RGBA*)malloc(palettelen*sizeof(RGBA));
-       U8*data2 = malloc((header.width+4)*header.height);
+       int swf_width = BYTES_PER_SCANLINE(header.width);
+       U8*data2 = malloc(swf_width*header.height);
+       U8*tmpline = malloc(header.width);
        int i,x,y;
        int pos=0;
+       if(!palette) {
+           fprintf(stderr, "Error: No palette found!\n");
+           exit(1);
+       }
        /* 24->32 bit conversion */
        for(i=0;i<palettelen;i++) {
            rgba[i].r = palette[i*3+0];
            rgba[i].g = palette[i*3+1];
            rgba[i].b = palette[i*3+2];
-           rgba[i].a = 255;
+           if(alphapalette && i<alphapalettelen) {
+               rgba[i].a = alphapalette[i];
+               if(alphapalette[i] == 0) {
+                   /* if the color is fully transparent, it doesn't matter
+                      what it's rgb values are. furthermore, all Flash 
+                      players up to Flash 5 can't deal with anything beyond
+                      one transparent color with value (00,00,00,00). */
+                   rgba[i].r = rgba[i].g = rgba[i].b = 0;
+               }
+           } else {
+               rgba[i].a = 255;
+           }
        }
+
        for(y=0;y<header.height;y++) {
-           pos++; //filter mode
-           for(x=0;x<header.width;x++) {
-               data2[y*header.width+x] = 
-                   imagedata[pos++];
+           int mode = imagedata[pos++]; //filter mode
+           U8*old;
+           U8*dest = &data2[y*swf_width];
+           U8*src;
+           src = &imagedata[pos];
+           if(header.bpp == 8) {
+               pos+=header.width;
+           } else {
+               int x,s=0;
+               int bitpos = 0;
+               U32 v = (1<<header.bpp)-1;
+               for(x=0;x<header.width;x++) {
+                   U32 r = src[s/8]<<8 | 
+                           src[s/8+1];
+                   int t;
+                   tmpline[x] = (r>>(16-header.bpp-(s&7)))&v;
+                   s+=header.bpp;
+               }
+               src = tmpline;
+               pos+=(header.width*header.bpp+7)/8;
            }
+
+           if(!y) {
+               memset(data2,0,swf_width);
+               old = &data2[y*swf_width];
+           } else {
+               old = &data2[(y-1)*swf_width];
+           }
+           applyfilter1(mode, src, old, dest, header.width);
        }
        swf_SetLosslessBitsIndexed(t, header.width, header.height, data2, rgba, palettelen);
+       free(tmpline);
        free(rgba);
        free(data2);
     }
@@ -344,14 +683,6 @@ int CheckInputFile(char *fname, char **realname)
        fprintf(stderr, "%s is not a PNG file!\n", fname);
        return -1;
     }
-    if(head.mode!=2 && head.mode!=3) {
-       fprintf(stderr, "%s has unsupported mode %d\n", head.mode);
-       return -1;
-    }
-    if(head.bpp!=8) {
-       fprintf(stderr, "%s has unsupported bpp %d\n", head.bpp);
-       return -1;
-    }
 
     if (global.max_image_width < head.width)
        global.max_image_width = head.width;
@@ -432,7 +763,6 @@ struct options_t options[] =
 {"X", "width"},
 {"Y", "height"},
 {"V", "version"},
-{"s", "scale"}
 };
 
 int args_callback_longoption(char *name, char *val)
@@ -443,14 +773,12 @@ int args_callback_longoption(char *name, char *val)
 int args_callback_command(char *arg, char *next)       // actually used as filename
 {
     char *s;
-    int scale;
     if (CheckInputFile(arg, &s) < 0) {
        if (VERBOSE(1))
            fprintf(stderr, "Error opening input file: %s\n", arg);
        free(s);
     } else {
        image[global.nfiles].filename = s;
-       image[global.nfiles].scale = global.prescale;
        global.nfiles++;
        if (global.nfiles >= MAX_INPUT_FILES) {
            if (VERBOSE(1))
@@ -466,11 +794,10 @@ void args_callback_usage(char *name)
     printf("Usage: %s  [-options [value]] imagefiles[.png] [...]\n", name);
     printf("-r framerate          (rate) Set movie framerate (100/sec)\n");
     printf("-o outputfile         (output) Set name for SWF output file\n");
-    printf("-X pixel              (width) Force movie width to scale (default: autodetect)\n");
-    printf("-Y pixel              (height) Force movie height to scale (default: autodetect)\n");
+    printf("-X pixel              (width) Force movie width to pixel (default: autodetect)\n");
+    printf("-Y pixel              (height) Force movie height to pixel (default: autodetect)\n");
     printf("-v level              (verbose) Set verbose level (0=quiet, 1=default, 2=debug)\n");
     printf("-V                    (version) Print version information and exit\n");
-    printf("The following options can be set independently for each image: -q -s\n");
 }
 
 
@@ -483,10 +810,12 @@ int main(int argc, char **argv)
 
     global.framerate = 100;
     global.verbose = 1;
-    global.prescale = 1;
 
     processargs(argc, argv);
 
+    if(global.nfiles<=0)
+       return 1;
+
     if (VERBOSE(2))
        fprintf(stderr, "Processing %i file(s)...\n", global.nfiles);
 
@@ -500,11 +829,9 @@ int main(int argc, char **argv)
        int i;
        for (i = 0; i < global.nfiles; i++) {
            if (VERBOSE(3))
-               fprintf(stderr, "[%03i] %s (1/%i)\n", i,
-                       image[i].filename, 
-                       image[i].scale);
-           t = MovieAddFrame(&swf, t, image[i].filename, 
-                             image[i].scale, (i * 2) + 1);
+               fprintf(stderr, "[%03i] %s\n", i,
+                       image[i].filename);
+           t = MovieAddFrame(&swf, t, image[i].filename, (i * 2) + 1);
            free(image[i].filename);
        }
     }