got rid of some old code
[swftools.git] / src / png2swf.c
index b03a11d..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,12 +42,20 @@ struct {
     int do_cgi;
     int version;
     char *outfile;
+    int mkjpeg;
+    float scale;
 } global;
 
 struct {
     char *filename;
 } image[MAX_INPUT_FILES];
 
+static int custom_move=0;
+static int move_x=0;
+static int move_y=0;
+static int clip_x1=0,clip_y1=0,clip_x2=0,clip_y2=0;
+static int custom_clip = 0;
+
 TAG *MovieStart(SWF * swf, float framerate, int dx, int dy)
 {
     TAG *t;
@@ -56,8 +65,15 @@ TAG *MovieStart(SWF * swf, float framerate, int dx, int dy)
 
     swf->fileVersion = global.version;
     swf->frameRate = (int)(256.0 * framerate);
-    swf->movieSize.xmax = dx * 20;
-    swf->movieSize.ymax = dy * 20;
+    if(custom_clip) {
+       swf->movieSize.xmin = clip_x1 * 20;
+       swf->movieSize.ymin = clip_y1 * 20;
+       swf->movieSize.xmax = clip_x2 * 20;
+       swf->movieSize.ymax = clip_y2 * 20;
+    } else {
+       swf->movieSize.xmax = dx * 20;
+       swf->movieSize.ymax = dy * 20;
+    }
 
     t = swf->firstTag = swf_InsertTag(NULL, ST_SETBACKGROUNDCOLOR);
 
@@ -84,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);
     }
@@ -108,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)
@@ -132,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
@@ -152,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))
@@ -162,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)
@@ -186,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) {
@@ -306,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)
@@ -451,252 +461,55 @@ 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;
-
-    FILE *fi;
-    U8 *scanline;
+    unsigned width=0, height=0;
 
-    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);
-
-    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/3;
-           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(!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)
-       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;
-               }
-           }
+#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;
-           }
-       }
-
-       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);
+           t = swf_InsertTag(t, ST_DEFINEBITSJPEG2);
+           swf_SetU16(t, id);
+           swf_SetJPEGBits2(t, width,height,data,global.mkjpeg);
        }
-       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);
 
     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);
@@ -715,14 +528,20 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
     t = swf_InsertTag(t, ST_PLACEOBJECT2);
 
     swf_GetMatrix(NULL, &m);
-    m.tx = (swf->movieSize.xmax - (int) header.width * 20) / 2;
-    m.ty = (swf->movieSize.ymax - (int) header.height * 20) / 2;
+    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) (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;
 }
 
@@ -796,11 +615,27 @@ 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;
+           break;
+
        case 'C':
            global.do_cgi = 1;
            break;
@@ -828,10 +663,48 @@ 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);
+   
+       case 'c': {
+           char*s = strdup(val);
+           char*x1 = strtok(s, ":");
+           char*y1 = strtok(0, ":");
+           char*x2 = strtok(0, ":");
+           char*y2 = strtok(0, ":");
+           if(!(x1 && y1 && x2 && y2)) {
+               fprintf(stderr, "-m option requires four arguments, <x1>:<y1>:<x2>:<y2>\n");
+               exit(1);
+           }
+           custom_clip = 1;
+           clip_x1 = atoi(x1);
+           clip_y1 = atoi(y1);
+           clip_x2 = atoi(x2);
+           clip_y2 = atoi(y2);
+           free(s);
+
+           res = 1;
+           break;
+       }
+
+       case 'm': {
+           char*s = strdup(val);
+           char*c = strchr(s, ':');
+           if(!c) {
+               fprintf(stderr, "-m option requires two arguments, <x>:<y>\n");
+               exit(1);
+           }
+           *c = 0;
+           custom_move = 1;
+           move_x = atoi(val);
+           move_y = atoi(c+1);
+           free(s);
+
+           res = 1;
+           break;
+       }
 
        default:
            res = -1;
@@ -850,13 +723,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}
 };
 
@@ -891,13 +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                 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");
 }
 
@@ -910,7 +789,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);
     
@@ -923,10 +803,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;