+void swf_PreMultiplyAlpha(RGBA*data, int width, int height)
+{
+ int num = width*height;
+ int t;
+ for(t=0;t<num;t++) {
+ data[t].r = ((int)data[t].r*data[t].a)/255;
+ data[t].g = ((int)data[t].g*data[t].a)/255;
+ data[t].b = ((int)data[t].b*data[t].a)/255;
+ }
+}
+
+/* expects mem to be non-premultiplied */
+void swf_SetLosslessImage(TAG*tag, RGBA*data, int width, int height)
+{
+ int hasalpha = swf_ImageHasAlpha(data, width, height);
+ int num;
+ if(!hasalpha) {
+ tag->id = ST_DEFINEBITSLOSSLESS;
+ } else {
+ tag->id = ST_DEFINEBITSLOSSLESS2;
+ swf_PreMultiplyAlpha(data, width, height);
+ }
+ num = swf_ImageGetNumberOfPaletteEntries(data, width, height, 0);
+ if(num>1 && num<=256) {
+ RGBA*palette = (RGBA*)malloc(sizeof(RGBA)*num);
+ int width2 = BYTES_PER_SCANLINE(width);
+ U8*data2 = (U8*)malloc(width2*height);
+ int len = width*height;
+ int x,y;
+ int r;
+ swf_ImageGetNumberOfPaletteEntries(data, width, height, palette);
+ for(y=0;y<height;y++) {
+ RGBA*src = &data[width*y];
+ U8*dest = &data2[width2*y];
+ for(x=0;x<width;x++) {
+ RGBA col = src[x];
+ for(r=0;r<num;r++) {
+ if(*(U32*)&col == *(U32*)&palette[r]) {
+ dest[x] = r;
+ break;
+ }
+ }
+ if(r==num) {
+ fprintf(stderr, "Internal error: Couldn't find color %02x%02x%02x%02x in palette (%d entries)\n",
+ col.r, col.g, col.b, col.a, num);
+ dest[x] = 0;
+ }
+ }
+ }
+ swf_SetLosslessBitsIndexed(tag, width, height, data2, palette, num);
+ free(data2);
+ free(palette);
+ } else {
+ swf_SetLosslessBits(tag, width, height, data, BMF_32BIT);
+ }
+}
+
+RGBA *swf_DefineLosslessBitsTagToImage(TAG * tag, int *dwidth, int *dheight)
+{
+ int id, format, height, width, pos;
+ uLongf datalen, datalen2;
+ int error;
+ int bpp = 1;
+ int cols = 0;
+ int pos2 = 0;
+ char alpha = tag->id == ST_DEFINEBITSLOSSLESS2;
+ int t, x, y;
+ RGBA *palette = 0;
+ U8 *data, *data2;
+ RGBA *dest;
+ if (tag->id != ST_DEFINEBITSLOSSLESS &&
+ tag->id != ST_DEFINEBITSLOSSLESS2) {
+ fprintf(stderr, "rfxswf: Object %d is not a PNG picture!\n",
+ GET16(tag->data));
+ return 0;
+ }
+ swf_SetTagPos(tag, 0);
+ id = swf_GetU16(tag);
+ format = swf_GetU8(tag);
+ if (format == 3)
+ bpp = 8;
+ if (format == 4)
+ bpp = 16;
+ if (format == 5)
+ bpp = 32;
+ if (format != 3 && format != 5) {
+ if (format == 4)
+ fprintf(stderr,
+ "rfxswf: Can't handle 16-bit palette images yet (image %d)\n",
+ id);
+ else
+ fprintf(stderr, "rfxswf: Unknown image type %d in image %d\n",
+ format, id);
+ return 0;
+ }
+ *dwidth = width = swf_GetU16(tag);
+ *dheight = height = swf_GetU16(tag);
+
+ dest = (RGBA*)rfx_alloc(sizeof(RGBA) * width * height);
+
+ if (format == 3)
+ cols = swf_GetU8(tag) + 1;
+ else
+ cols = 0;
+
+ data = 0;
+ datalen = (width * height * bpp / 8 + cols * 8);
+ do {
+ if (data)
+ rfx_free(data);
+ datalen += 4096;
+ data = (U8*)rfx_alloc(datalen);
+ error =
+ uncompress(data, &datalen, &tag->data[tag->pos],
+ tag->len - tag->pos);
+ } while (error == Z_BUF_ERROR);
+ if (error != Z_OK) {
+ fprintf(stderr, "rfxswf: Zlib error %d (image %d)\n", error, id);
+ return 0;
+ }
+ pos = 0;
+
+ if (cols) {
+ palette = (RGBA *) rfx_alloc(cols * sizeof(RGBA));
+ for (t = 0; t < cols; t++) {
+ palette[t].r = data[pos++];
+ palette[t].g = data[pos++];
+ palette[t].b = data[pos++];
+ if (alpha) {
+ palette[t].a = data[pos++];
+ } else {
+ palette[t].a = 255;
+ }
+ }
+ }
+
+ for (y = 0; y < height; y++) {
+ int srcwidth = width * (bpp / 8);
+ if (bpp == 32) {
+ if (!alpha) {
+ // 32 bit to 24 bit "conversion"
+ for (x = 0; x < width; x++) {
+ dest[pos2].r = data[pos + 1];
+ dest[pos2].g = data[pos + 2];
+ dest[pos2].b = data[pos + 3];
+ dest[pos2].a = 255;
+ pos2++;
+ pos += 4; //ignore padding byte
+ }
+ } else {
+ for (x = 0; x < width; x++) {
+ /* remove premultiplication */
+ int alpha = data[pos+0];
+ if(alpha)
+ alpha = 0xff0000/alpha;
+ dest[pos2].r = (data[pos + 1]*alpha)>>16;
+ dest[pos2].g = (data[pos + 2]*alpha)>>16;
+ dest[pos2].b = (data[pos + 3]*alpha)>>16;
+ dest[pos2].a = data[pos + 0]; //alpha
+ pos2++;
+ pos += 4;
+ }
+ }
+ } else {
+ for (x = 0; x < srcwidth; x++) {
+ dest[pos2] = palette[data[pos++]];
+ pos2++;
+ }
+ }
+ pos += ((srcwidth + 3) & ~3) - srcwidth; //align
+ }
+ if (palette)
+ rfx_free(palette);
+ rfx_free(data);
+ return dest;
+}
+
+#endif // HAVE_ZLIB