X-Git-Url: http://git.asbjorn.biz/?p=swftools.git;a=blobdiff_plain;f=src%2Fpng2swf.c;h=927504e16982354206fe05dbe2bf2ae919eafe81;hp=0e584e38fcf50377797eed57bb6587e014e31be5;hb=0e8807d9a620e8932bc67653321eaae1436fdd43;hpb=1204eaea945b78ca2a8c8b8941365610881f69cf

diff --git a/src/png2swf.c b/src/png2swf.c
index 0e584e3..927504e 100644
--- a/src/png2swf.c
+++ b/src/png2swf.c
@@ -87,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)
@@ -126,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;
@@ -135,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);
@@ -147,6 +148,15 @@ 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)
+
+	    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);
@@ -159,16 +169,16 @@ int png_read_header(FILE*fi, struct png_header*header)
 		fprintf(stderr, "Interlace mode %d not supported!\n", i);
 		exit(1);
 	    }
-	    printf("%dx%d %d %d %d %d %d\n",header->width, header->height, a,b,c,f,i);
+	    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;
@@ -189,7 +199,7 @@ byte inline PaethPredictor (byte a,byte b,byte c)
         else return c;
 }
 
-void applyfilter(int mode, U8*src, U8*old, U8*dest, int width)
+void applyfilter3(int mode, U8*src, U8*old, U8*dest, int width)
 {
     int x;
     unsigned char lastr=0;
@@ -267,6 +277,139 @@ void applyfilter(int mode, U8*src, U8*old, U8*dest, int width)
 
 }
 
+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;
@@ -278,9 +421,13 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
     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;
 
@@ -300,6 +447,8 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
     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);
@@ -315,49 +464,90 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
 	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++) {
 	    int mode = imagedata[pos++]; //filter mode
-
-	    U8*src = &imagedata[pos];
-	    U8*dest = &data2[(y*header.width)*4];
+	    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];
 	    }
-	    applyfilter(mode, src, old, dest, header.width);
-	    pos+=header.width*3;
+	    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) {
@@ -369,20 +559,53 @@ TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
 	    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++) {
-	    int mode = imagedata[pos];
-	    if(mode!=0) {
-		fprintf(stderr, "Warning: Filter mode %d\n", mode);
+	    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;
 	    }
-	    pos++; //filter mode
-	    for(x=0;x<header.width;x++) {
-		data2[y*header.width+x] = 
-		    imagedata[pos++];
+
+	    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);
     }
@@ -460,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;
@@ -598,6 +813,9 @@ int main(int argc, char **argv)
 
     processargs(argc, argv);
 
+    if(global.nfiles<=0)
+	return 1;
+
     if (VERBOSE(2))
 	fprintf(stderr, "Processing %i file(s)...\n", global.nfiles);