got rid of some old code
[swftools.git] / src / png2swf.c
index 0d7e1a6..a6782f7 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,7 @@ struct {
     int do_cgi;
     int version;
     char *outfile;
+    int mkjpeg;
     float scale;
 } global;
 
@@ -98,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);
     }
@@ -122,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)
@@ -146,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
@@ -166,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))
@@ -176,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)
@@ -200,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) {
@@ -320,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)
@@ -465,271 +461,38 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
     MATRIX m;
     int fs;
 
-    char tagid[4];
-    int len;
-    U8*data;
-    U8*imagedata;
-    U8*zimagedata=0;
-    unsigned long int imagedatalen;
-    unsigned long int zimagedatalen=0;
-    U8*palette = 0;
-    int palettelen = 0;
-    U8*alphapalette = 0;
-    int alphapalettelen = 0;
-    struct png_header header;
-    int bypp;
-    U8 alphacolor[3];
-    int hasalphacolor=0;
-
-    FILE *fi;
-    U8 *scanline;
-
-    if ((fi = fopen(sname, "rb")) == NULL) {
-       if (VERBOSE(1))
-           fprintf(stderr, "Read access failed: %s\n", sname);
-       return t;
-    }
-
-    if(!png_read_header(fi, &header))
-       return 0;
-
-    if(header.mode == 3 || header.mode == 0) bypp = 1;
-    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);
+    unsigned width=0, height=0;
 
-    fseek(fi,8,SEEK_SET);
-    while(!feof(fi))
-    {
-       if(!png_read_chunk(&tagid, &len, &data, fi))
-           break;
-       if(!strncmp(tagid, "IEND", 4)) {
-           break;
-       }
-       if(!strncmp(tagid, "PLTE", 4)) {
-           palette = data;
-           palettelen = len/bypp;
-           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);
-           } else if(header.mode == 0) {
-               int t;
-               if(header.mode == 3) { // 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)) {
-           if(!zimagedata) {
-               zimagedatalen = len;
-               zimagedata = malloc(len);
-               memcpy(zimagedata,data,len);
-           } else {
-               zimagedata = realloc(zimagedata, zimagedatalen+len);
-               memcpy(&zimagedata[zimagedatalen], data, len);
-               zimagedatalen += len;
-           }
-       }
-       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;
+#ifndef HAVE_JPEGLIB
+    if(global.mkjpeg) {
+        global.mkjpeg = 0;
+        msg("<warning> No jpeg support compiled in");
     }
-    free(zimagedata);
-
-    if(alphapalette || hasalphacolor)
-       t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS2);
-    else
-       t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS);
-
-    swf_SetU16(t, id);         // id
-    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;
-       int opaque=0;
-       int transparent=0;
-       int semitransparent=0;
-       /* in case for mode 2, the following also performs 24->32 bit conversion */
-       unsigned char* firstline = malloc(header.width*4);
-
-       for(y=0;y<header.height;y++) {
-           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) {
-               old = firstline;
-               memset(old, 0, header.width*4); //TODO: fill alpha with 255?
-           } else {
-               old = &data2[(y-1)*header.width*4];
-           }
-           if(header.mode==6)
-               applyfilter4(mode, src, old, dest, header.width);
-           else if(header.mode==2)
-               applyfilter3(mode, src, old, dest, header.width);
-       }
-       free(firstline);
-
-       /* the image is now compressed and stored in data. Now let's take
-          a look at the alpha values */
-       if(header.mode == 6) {
-           for(y=0;y<header.height;y++) {
-               U8*l = &data2[(y*header.width)*4];
-               for(x=0;x<header.width;x++) {
-                   U8 a = l[x*4];
-                   U8 b = l[x*4+1];
-                   U8 g = l[x*4+2];
-                   U8 r = l[x*4+3];
-                   if(a==255) transparent++;
-                   else {
-                       if(a==0) opaque++;
-                       else semitransparent++;
-                       l[x*4+3]=(int)r*a/255;
-                       l[x*4+2]=(int)g*a/255;
-                       l[x*4+1]=(int)b*a/255;
-                   }
-               }
-           }
-           if(semitransparent || opaque) {
-               t->id = ST_DEFINEBITSLOSSLESS2;
-           }
-       }
-       swf_SetLosslessBits(t, header.width, header.height, data2, BMF_32BIT);
-       free(data2);
-    } else if(header.mode == 0 || header.mode == 3) {
-       RGBA*rgba;
-       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(header.mode == 3) { // palette or grayscale?
-           rgba = (RGBA*)malloc(palettelen*sizeof(RGBA));
-           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];
-               if(alphapalette && i<alphapalettelen) {
-                   rgba[i].a = alphapalette[i];
-                   rgba[i].r = ((int)rgba[i].r*rgba[i].a)/255;
-                   rgba[i].g = ((int)rgba[i].g*rgba[i].a)/255;
-                   rgba[i].b = ((int)rgba[i].b*rgba[i].a)/255;
-               } 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;
-                   }
-               }
-           }
+#endif
+    if(global.mkjpeg) {
+#ifdef HAVE_JPEGLIB
+       RGBA*data = 0;
+       png_load(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 {
-            int mult = (0x1ff>>header.bpp);
-           palettelen = 1<<header.bpp;
-           rgba = (RGBA*)malloc(palettelen*sizeof(RGBA));
-           for(i=0;i<palettelen;i++) {
-               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;
-                   }
-               }
-           }
+           t = swf_InsertTag(t, ST_DEFINEBITSJPEG2);
+           swf_SetU16(t, id);
+           swf_SetJPEGBits2(t, width,height,data,global.mkjpeg);
        }
-
-       for(y=0;y<header.height;y++) {
-           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);
+#endif
+    } else {
+       RGBA*data = 0;
+       png_load(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);
     }
 
     t = swf_InsertTag(t, ST_DEFINESHAPE3);
@@ -738,15 +501,15 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
     swf_GetMatrix(NULL, &m);
     m.sx = (int)(20 * 0x10000);
     m.sy = (int)(20 * 0x10000);
-    m.tx = -10;
-    m.ty = -10;
+    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);
@@ -772,15 +535,13 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
        m.tx = move_x*20;
        m.ty = move_y*20;
     } else {
-       m.tx = (swf->movieSize.xmax - (int) (header.width * global.scale * 20)) / 2;
-       m.ty = (swf->movieSize.ymax - (int) (header.height * global.scale * 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;
 }
 
@@ -860,10 +621,16 @@ int args_callback_option(char *arg, char *val)
            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;
@@ -896,7 +663,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);
@@ -956,7 +723,9 @@ 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"},
@@ -998,14 +767,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 <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("-s , --scale <percent>         Scale image to <percent>%% size.\n");
     printf("\n");
 }
 
@@ -1018,7 +789,7 @@ 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);