Copyright (c) 2000, 2001 Rainer Böhme <rfxswf@reflex-studio.de>
- This file is distributed under the GPL, see file COPYING for details
+ This program is free software; you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation; either version 2 of the License, or
+ (at your option) any later version.
-*/
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with this program; if not, write to the Free Software
+ Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */
#define OUTBUFFER_SIZE 0x8000
-#ifdef _JPEGLIB_INCLUDED_
+#ifdef HAVE_JPEGLIB
typedef struct _JPEGDESTMGR
{ struct jpeg_destination_mgr mgr;
// Destination manager callbacks
-void RFXSWF_init_destination(j_compress_ptr cinfo)
+static void RFXSWF_init_destination(j_compress_ptr cinfo)
{ JPEGDESTMGR * dmgr = (JPEGDESTMGR *)cinfo->dest;
- dmgr->buffer = (JOCTET*)malloc(OUTBUFFER_SIZE);
+ dmgr->buffer = (JOCTET*)rfx_alloc(OUTBUFFER_SIZE);
dmgr->mgr.next_output_byte = dmgr->buffer;
dmgr->mgr.free_in_buffer = OUTBUFFER_SIZE;
}
-boolean RFXSWF_empty_output_buffer(j_compress_ptr cinfo)
+static boolean RFXSWF_empty_output_buffer(j_compress_ptr cinfo)
{ JPEGDESTMGR * dmgr = (JPEGDESTMGR *)cinfo->dest;
swf_SetBlock(dmgr->t,(U8*)dmgr->buffer,OUTBUFFER_SIZE);
dmgr->mgr.next_output_byte = dmgr->buffer;
return TRUE;
}
-void RFXSWF_term_destination(j_compress_ptr cinfo)
+static void RFXSWF_term_destination(j_compress_ptr cinfo)
{ JPEGDESTMGR * dmgr = (JPEGDESTMGR *)cinfo->dest;
swf_SetBlock(dmgr->t,(U8*)dmgr->buffer,OUTBUFFER_SIZE-dmgr->mgr.free_in_buffer);
- free(dmgr->buffer);
+ rfx_free(dmgr->buffer);
dmgr->mgr.free_in_buffer = 0;
}
// redirect compression lib output to local SWF Tag structure
- jpeg = (JPEGDESTMGR *)malloc(sizeof(JPEGDESTMGR));
- if (!jpeg) return NULL;
+ jpeg = (JPEGDESTMGR *)rfx_calloc(sizeof(JPEGDESTMGR));
- memset(jpeg,0x00,sizeof(JPEGDESTMGR));
jpeg->cinfo.err = jpeg_std_error(&jpeg->jerr);
jpeg_create_compress(&jpeg->cinfo);
{ JPEGDESTMGR * jpeg = (JPEGDESTMGR *)jpegbits;
if (!jpeg) return -1;
jpeg_finish_compress(&jpeg->cinfo);
- free(jpeg);
+ rfx_free(jpeg);
return 0;
}
+void swf_SetJPEGBits2(TAG * tag,U16 width,U16 height,RGBA* bitmap, int quality)
+{
+ JPEGBITS* jpeg;
+ int y;
+ jpeg = swf_SetJPEGBitsStart(tag,width,height,quality);
+ for (y=0;y<height;y++)
+ { U8 scanline[3*width];
+ int x,p = 0;
+ for (x=0;x<width;x++)
+ { 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);
+ }
+ swf_SetJPEGBitsFinish(jpeg);
+}
+
+void swf_GetJPEGSize(char * fname, int*width, int*height)
+{ struct jpeg_decompress_struct cinfo;
+ struct jpeg_error_mgr jerr;
+ FILE * fi;
+ *width = 0;
+ *height = 0;
+ cinfo.err = jpeg_std_error(&jerr);
+ jpeg_create_decompress(&cinfo);
+ if ((fi=fopen(fname,"rb"))==NULL) {
+ fprintf(stderr, "rfxswf: file open error\n");
+ return;
+ }
+ jpeg_stdio_src(&cinfo, fi);
+ jpeg_read_header(&cinfo, TRUE);
+ *width = cinfo.image_width;
+ *height = cinfo.image_height;
+ jpeg_destroy_decompress(&cinfo);
+ fclose(fi);
+}
+
int swf_SetJPEGBits(TAG * t,char * fname,int quality)
{ struct jpeg_decompress_struct cinfo;
struct jpeg_error_mgr jerr;
cinfo.err = jpeg_std_error(&jerr);
jpeg_create_decompress(&cinfo);
- if ((f=fopen(fname,"rb"))==NULL) return -1;
+ if ((f=fopen(fname,"rb"))==NULL) {
+ fprintf(stderr, "rfxswf: file open error\n");
+ return -1;
+ }
jpeg_stdio_src(&cinfo,f);
jpeg_read_header(&cinfo, TRUE);
jpeg_start_decompress(&cinfo);
out = swf_SetJPEGBitsStart(t,cinfo.output_width,cinfo.output_height,quality);
- scanline = (U8*)malloc(4*cinfo.output_width);
+ scanline = (U8*)rfx_alloc(4*cinfo.output_width);
if (scanline)
{ int y;
}
else if(cinfo.out_color_space == JCS_YCbCr)
{
- //FIXME
- fprintf(stderr, "Error: Can't convert YCbCr(YUV) to RGB.\n");
- return -1;
+ for (y=0;y<cinfo.output_height;y++) {
+ int x;
+ for(x=0;x<cinfo.output_width;x++) {
+ int y = js[x*3+0];
+ int u = js[x*3+1];
+ int v = js[x*3+1];
+ js[x*3+0] = y + ((360*(v-128))>>8);
+ js[x*3+1] = y - ((88*(u-128)+183*(v-128))>>8);
+ js[x*3+2] = y + ((455 * (u-128))>>8);
+ }
+ }
}
else if(cinfo.out_color_space == JCS_CMYK)
{
return 0;
}
-#endif // _JPEGLIB_INCLUDED_
+/* jpeg_source_mgr functions */
+static void tag_init_source(struct jpeg_decompress_struct *cinfo)
+{
+ TAG*tag = (TAG*)cinfo->client_data;
+ swf_SetTagPos(tag, 2);
+ cinfo->src->bytes_in_buffer = 0;
+}
+static boolean tag_fill_input_buffer(struct jpeg_decompress_struct *cinfo)
+{
+ TAG*tag = (TAG*)cinfo->client_data;
+ if(tag->data[tag->pos+0] == 0xff &&
+ tag->data[tag->pos+1] == 0xd9 &&
+ tag->data[tag->pos+2] == 0xff &&
+ tag->data[tag->pos+3] == 0xd8) {
+ tag->pos += 4;
+ }
+ if(tag->pos >= tag->len) {
+ cinfo->src->next_input_byte = 0;
+ cinfo->src->bytes_in_buffer = 0;
+ return 0;
+ }
+ cinfo->src->next_input_byte = &tag->data[tag->pos];
+ cinfo->src->bytes_in_buffer = 1;//tag->len - tag->pos;
+ tag->pos += 1;
+ return 1;
+}
+static void tag_skip_input_data(struct jpeg_decompress_struct *cinfo, long count)
+{
+ TAG*tag = (TAG*)cinfo->client_data;
+ cinfo->src->next_input_byte = 0;
+ cinfo->src->bytes_in_buffer = 0;
+ tag->pos += count;
+}
+static boolean tag_resync_to_restart(struct jpeg_decompress_struct *cinfo, int desired)
+{
+ return jpeg_resync_to_restart(cinfo, desired);
+}
+static void tag_term_source(struct jpeg_decompress_struct *cinfo)
+{
+ TAG*tag = (TAG*)cinfo->client_data;
+}
+RGBA* swf_JPEG2TagToImage(TAG*tag, int*width, int*height)
+{
+ struct jpeg_decompress_struct cinfo;
+ struct jpeg_error_mgr jerr;
+ struct jpeg_source_mgr mgr;
+ RGBA * dest;
+ int y;
+ *width = 0;
+ *height = 0;
+
+ if(tag->id == ST_DEFINEBITSJPEG) {
+ fprintf(stderr, "rfxswf: extracting from definebitsjpeg not yet supported");
+ return 0;
+ }
+ if(tag->id == ST_DEFINEBITSJPEG3) {
+ fprintf(stderr, "rfxswf: extracting from definebitsjpeg3 not yet supported");
+ return 0;
+ }
+
+ cinfo.err = jpeg_std_error(&jerr);
+ jpeg_create_decompress(&cinfo);
+
+ cinfo.client_data = (void*)tag;
+ cinfo.src = &mgr;
+ cinfo.src->init_source = tag_init_source;
+ cinfo.src->fill_input_buffer = tag_fill_input_buffer;
+ cinfo.src->skip_input_data = tag_skip_input_data;
+ cinfo.src->resync_to_restart = jpeg_resync_to_restart;
+ cinfo.src->term_source = tag_term_source;
+ cinfo.out_color_space = JCS_RGB;
+
+ jpeg_read_header(&cinfo, TRUE);
+ *width = cinfo.image_width;
+ *height = cinfo.image_height;
+ dest = rfx_alloc(sizeof(RGBA)*cinfo.image_width*cinfo.image_height);
+
+ jpeg_start_decompress(&cinfo);
+ for (y=0;y<cinfo.output_height;y++) {
+ RGBA* line = &dest[y*cinfo.image_width];
+ U8* to = (U8*)line;
+ int x;
+ jpeg_read_scanlines(&cinfo,&to,1);
+ for(x=cinfo.output_width-1;x>=0;--x) {
+ int r = to[x*3+0];
+ int g = to[x*3+1];
+ int b = to[x*3+2];
+ line[x].r = r;
+ line[x].g = g;
+ line[x].b = b;
+ line[x].a = 255;
+ }
+ }
+
+ jpeg_finish_decompress(&cinfo);
+
+ jpeg_destroy_decompress(&cinfo);
+ return dest;
+}
+
+
+#endif // HAVE_JPEGLIB
// Lossless compression texture based on zlib
-#ifdef _ZLIB_INCLUDED_
+#ifdef HAVE_ZLIB
-int RFXSWF_deflate_wraper(TAG * t,z_stream * zs,U8 * data,boolean finish)
-{ while (1)
- { int status = deflate(zs,Z_SYNC_FLUSH);
+int RFXSWF_deflate_wraper(TAG * t,z_stream * zs,boolean finish)
+{
+ U8*data=rfx_alloc(OUTBUFFER_SIZE);
+ zs->next_out = data;
+ zs->avail_out = OUTBUFFER_SIZE;
+ while (1)
+ { int status = deflate(zs,Z_NO_FLUSH);
- if (zs->avail_out == 0)
- { swf_SetBlock(t,data,zs->next_out-data);
+ if (status!=Z_OK)
+ {
+#ifdef DEBUG_RFXSWF
+ fprintf(stderr,"rfxswf: zlib compression error (%i)\n",status);
+#endif
+ rfx_free(data);
+ return status;
+ }
+
+ if (zs->next_out!=data)
+ { swf_SetBlock(t,data,zs->next_out - data);
zs->next_out = data;
zs->avail_out = OUTBUFFER_SIZE;
}
if (zs->avail_in==0)
- { if (finish) deflate(zs,Z_FINISH);
break;
- }
+ }
- if (status!=Z_OK)
+ if(!finish) {
+ rfx_free(data);
+ return 0;
+ }
+
+ while(1) {
+ int status = deflate(zs,Z_FINISH);
+ if (status!=Z_OK && status!=Z_STREAM_END)
{
#ifdef DEBUG_RFXSWF
fprintf(stderr,"rfxswf: zlib compression error (%i)\n",status);
#endif
+ rfx_free(data);
return status;
}
+
+ if (zs->next_out!=data)
+ {
+ swf_SetBlock(t,data,zs->next_out - data);
+ zs->next_out = data;
+ zs->avail_out = OUTBUFFER_SIZE;
+ }
+
+ if(status == Z_STREAM_END)
+ break;
}
+ rfx_free(data);
return 0;
}
+
int swf_SetLosslessBits(TAG * t,U16 width,U16 height,void * bitmap,U8 bitmap_flags)
{ int res = 0;
int bps;
- U8 * data;
switch (bitmap_flags)
{ case BMF_8BIT:
bps = width*4;
break;
default:
+ fprintf(stderr, "rfxswf: unknown bitmap type %d\n", bitmap_flags);
return -1;
}
-
+
swf_SetU8(t,bitmap_flags);
swf_SetU16(t,width);
swf_SetU16(t,height);
- if (data=malloc(OUTBUFFER_SIZE))
{ z_stream zs;
memset(&zs,0x00,sizeof(z_stream));
if (deflateInit(&zs,Z_DEFAULT_COMPRESSION)==Z_OK)
{ zs.avail_in = bps*height;
zs.next_in = bitmap;
- zs.next_out = data;
- zs.avail_out = OUTBUFFER_SIZE;
-
- if (RFXSWF_deflate_wraper(t,&zs,data,TRUE)<0) res = -3;
- if (zs.next_out>data) swf_SetBlock(t,data,zs.next_out-data);
+ if (RFXSWF_deflate_wraper(t,&zs,TRUE)<0) res = -3;
deflateEnd(&zs);
-
} else res = -3; // zlib error
- free(data);
- } else res = -2; // memory error
-
+ }
return res;
}
int swf_SetLosslessBitsIndexed(TAG * t,U16 width,U16 height,U8 * bitmap,RGBA * palette,U16 ncolors)
{ RGBA * pal = palette;
int bps = BYTES_PER_SCANLINE(width);
- U8 * data;
int res = 0;
if (!pal) // create default palette for grayscale images
{ int i;
- pal = malloc(256*sizeof(RGBA));
+ pal = rfx_alloc(256*sizeof(RGBA));
for (i=0;i<256;i++) { pal[i].r = pal[i].g = pal[i].b = i; pal[i].a = 0xff;}
ncolors = 256;
}
- if ((ncolors<2)||(ncolors>256)||(!t)) return -1; // parameter error
+ if ((ncolors<2)||(ncolors>256)||(!t)) {
+ fprintf(stderr, "rfxswf: unsupported number of colors: %d\n", ncolors);
+ return -1; // parameter error
+ }
swf_SetU8(t,BMF_8BIT);
swf_SetU16(t,width);
swf_SetU16(t,height);
swf_SetU8(t,ncolors-1); // number of pal entries
- if (data=malloc(OUTBUFFER_SIZE))
{ z_stream zs;
memset(&zs,0x00,sizeof(z_stream));
if (deflateInit(&zs,Z_DEFAULT_COMPRESSION)==Z_OK)
{ U8 * zpal; // compress palette
- if (zpal = malloc(ncolors*4))
+ if ((zpal = rfx_alloc(ncolors*4)))
{ U8 * pp = zpal;
int i;
}
zs.next_in = zpal;
- zs.next_out = data;
- zs.avail_out = OUTBUFFER_SIZE;
- if (RFXSWF_deflate_wraper(t,&zs,data,FALSE)<0) res = -3;
+ 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,data,TRUE)<0) res = -3;
+ if (RFXSWF_deflate_wraper(t,&zs,TRUE)<0) res = -3;
deflateEnd(&zs);
- if (zs.next_out>data) swf_SetBlock(t,data,zs.next_out-data);
-
- free(zpal);
+ rfx_free(zpal);
} else res = -2; // memory error
} else res = -3; // zlib error
- free(data);
- } else res = -2;
+ }
- if (!palette) free(pal);
+ if (!palette) rfx_free(pal);
return res;
}
{ return swf_SetLosslessBitsIndexed(t,width,height,bitmap,NULL,256);
}
+RGBA* swf_DefineLosslessBitsTagToImage(TAG*tag, int*dwidth, int*dheight)
+{
+ int id,format,height, width, pos;
+ U32 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 = 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 = 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++];
+ }
+ }
+ }
-#endif // _ZLIB_INCLUDED_
+ 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++) {
+ dest[pos2].r=data[pos+1];
+ dest[pos2].g=data[pos+2];
+ dest[pos2].b=data[pos+3];
+ 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;
+}
-#undef OUTBUFFER_SIZE
+#endif // HAVE_ZLIB
+#if defined(HAVE_ZLIB) && defined(HAVE_JPEGLIB)
+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);
+ for (y=0;y<height;y++)
+ { U8 scanline[3*width];
+ int x,p = 0;
+ for (x=0;x<width;x++)
+ { 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);
+ }
+ swf_SetJPEGBitsFinish(jpeg);
+ PUT32(&tag->data[pos], tag->len - pos - 4);
+
+ data=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;
+
+ for (y=0;y<height;y++)
+ { U8 scanline[width];
+ 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;
+ }
+ }
+ }
+
+ while(1) {
+ int ret = deflate(&zs, Z_FINISH);
+ if (ret != Z_OK &&
+ ret != Z_STREAM_END) {
+ fprintf(stderr, "rfxswf: zlib compression failed");
+ return -5;
+ }
+ if(zs.next_out != data) {
+ swf_SetBlock(tag, data, zs.next_out - data);
+ zs.next_out = data;
+ zs.avail_out = OUTBUFFER_SIZE;
+ }
+ if (ret == Z_STREAM_END) {
+ break;
+ }
+ }
+
+ deflateEnd(&zs);
+ rfx_free(data);
+ return 0;
+}
+#endif
+
+RGBA* swf_ExtractImage(TAG*tag, int*dwidth, int*dheight)
+{
+ RGBA*img;
+ if(tag->id == ST_DEFINEBITSJPEG ||
+ tag->id == ST_DEFINEBITSJPEG2 ||
+ tag->id == ST_DEFINEBITSJPEG3) {
+#ifdef HAVE_JPEGLIB
+ return swf_JPEG2TagToImage(tag, dwidth, dheight);
+#else
+ fprintf(stderr, "rfxswf: Error: No JPEG library compiled in");
+ return 0;
+#endif
+ }
+ if(tag->id == ST_DEFINEBITSLOSSLESS ||
+ tag->id == ST_DEFINEBITSLOSSLESS2) {
+#ifdef HAVE_ZLIB
+ return swf_DefineLosslessBitsTagToImage(tag, dwidth, dheight);
+#else
+ fprintf(stderr, "rfxswf: Error: No JPEG library compiled in");
+ return 0;
+#endif
+ }
+ fprintf(stderr, "rfxswf: Error: Invalid tag (%d, %s)", tag->id, swf_TagGetName(tag));
+ return 0;
+}
+
+#undef OUTBUFFER_SIZE