+ Indeed: I haven't understood yet how flash player handles
+ alpha values different from 0 and 0xff in lossless bitmaps...
+ */
+
+ if (swf_GetTagID(t) == ST_DEFINEBITSLOSSLESS2) // have alpha channel?
+ {
+ for (i = 0; i < ncolors; i++) {
+ pp[0] = pal[i].r;
+ pp[1] = pal[i].g;
+ pp[2] = pal[i].b;
+ pp[3] = pal[i].a;
+ pp += 4;
+ }
+ zs.avail_in = 4 * ncolors;
+ } else {
+ for (i = 0; i < ncolors; i++) // pack RGBA structures to RGB
+ {
+ pp[0] = pal[i].r;
+ pp[1] = pal[i].g;
+ pp[2] = pal[i].b;
+ pp += 3;
+ }
+ zs.avail_in = 3 * ncolors;
+ }
+
+ zs.next_in = zpal;
+
+ if (RFXSWF_deflate_wraper(t, &zs, FALSE) < 0)
+ res = -3;
+
+ // compress bitmap
+ zs.next_in = bitmap;
+ zs.avail_in = (bps * height * sizeof(U8));
+
+ if (RFXSWF_deflate_wraper(t, &zs, TRUE) < 0)
+ res = -3;
+
+ deflateEnd(&zs);
+
+ rfx_free(zpal);
+ } else
+ res = -2; // memory error
+ } else
+ res = -3; // zlib error
+ }
+
+ if (!palette)
+ rfx_free(pal);
+
+ return res;
+}
+
+int swf_SetLosslessBitsGrayscale(TAG * t, U16 width, U16 height, U8 * bitmap)
+{
+ return swf_SetLosslessBitsIndexed(t, width, height, bitmap, NULL, 256);
+}
+
+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
+
+#if defined(HAVE_ZLIB) && defined(HAVE_JPEGLIB)
+
+/* expects bitmap to be non-premultiplied */
+int swf_SetJPEGBits3(TAG * tag, U16 width, U16 height, RGBA * bitmap, int quality)
+{
+ JPEGBITS *jpeg;
+ int y;
+ int pos;
+ int res = 0;
+ U8 *data;
+ z_stream zs;
+
+ pos = tag->len;
+ swf_SetU32(tag, 0); //placeholder
+ jpeg = swf_SetJPEGBitsStart(tag, width, height, quality);
+ U8 *scanline = (U8*)rfx_alloc(3 * width);
+ for (y = 0; y < height; y++) {
+ int x, p = 0;
+ for (x = 0; x < width; x++) {
+ //int ia = bitmap[width*y+x].a;
+ //if(ia) {
+ // /* remove premultiplication */
+ // ia = 0xff0000/ia;
+ //}
+ //scanline[p++] = (bitmap[width * y + x].r*ia)>>16;
+ //scanline[p++] = (bitmap[width * y + x].g*ia)>>16;
+ //scanline[p++] = (bitmap[width * y + x].b*ia)>>16;
+ scanline[p++] = bitmap[width * y + x].r;
+ scanline[p++] = bitmap[width * y + x].g;
+ scanline[p++] = bitmap[width * y + x].b;
+ }
+ swf_SetJPEGBitsLine(jpeg, scanline);
+ }
+ rfx_free(scanline);
+ swf_SetJPEGBitsFinish(jpeg);
+ PUT32(&tag->data[pos], tag->len - pos - 4);
+
+ data = (U8*)rfx_alloc(OUTBUFFER_SIZE);
+ memset(&zs, 0x00, sizeof(z_stream));
+
+ if (deflateInit(&zs, Z_DEFAULT_COMPRESSION) != Z_OK) {
+ fprintf(stderr, "rfxswf: zlib compression failed");
+ return -3;
+ }
+
+ zs.next_out = data;
+ zs.avail_out = OUTBUFFER_SIZE;
+
+ scanline = (U8*)rfx_alloc(width);
+ for (y = 0; y < height; y++) {
+ int x, p = 0;
+ for (x = 0; x < width; x++) {
+ scanline[p++] = bitmap[width * y + x].a;
+ }
+ zs.avail_in = width;
+ zs.next_in = scanline;
+
+ while (1) {
+ if (deflate(&zs, Z_NO_FLUSH) != Z_OK) {
+ fprintf(stderr, "rfxswf: zlib compression failed");
+ return -4;
+ }
+ if (zs.next_out != data) {
+ swf_SetBlock(tag, data, zs.next_out - data);
+ zs.next_out = data;
+ zs.avail_out = OUTBUFFER_SIZE;
+ }
+ if (!zs.avail_in) {
+ break;
+ }
+ }
+ }