along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */
+#include <stdlib.h>
+#include <stdio.h>
+#include <memory.h>
+#include "../../config.h"
+#ifdef HAVE_ZLIB
+#include <zconf.h>
+#include <zlib.h>
+#endif
+#include <fcntl.h>
+#include <ctype.h>
+
+#ifdef HAVE_JPEGLIB
+#define HAVE_BOOLEAN
+#ifdef __cplusplus
+extern "C" {
+#endif
+#include <jpeglib.h>
+#ifdef __cplusplus
+}
+#endif
+#endif // HAVE_JPEGLIB
+
+#include "../rfxswf.h"
+
#define OUTBUFFER_SIZE 0x8000
int swf_ImageHasAlpha(RGBA*img, int width, int height)
int palette_overflow = 0;
U32 lastcol32 = 0;
- pal = malloc(65536*sizeof(U32));
+ pal = (U32*)malloc(65536*sizeof(U32));
memset(size, 0, sizeof(size));
if (!jpeg)
return -1;
jpeg_finish_compress(&jpeg->cinfo);
+ jpeg_destroy_compress(&jpeg->cinfo);
rfx_free(jpeg);
return 0;
}
-void swf_SetJPEGBits2(TAG * tag, U16 width, U16 height, RGBA * bitmap,
- int quality)
+#if defined(HAVE_JPEGLIB)
+void swf_SetJPEGBits2(TAG * tag, U16 width, U16 height, RGBA * bitmap, int quality)
{
JPEGBITS *jpeg;
int y;
jpeg = swf_SetJPEGBitsStart(tag, width, height, quality);
+ U8 *scanline = (U8*)rfx_alloc(3 * width);
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;
}
swf_SetJPEGBitsLine(jpeg, scanline);
}
+ rfx_free(scanline);
swf_SetJPEGBitsFinish(jpeg);
}
+#else
+void swf_SetJPEGBits2(TAG * tag, U16 width, U16 height, RGBA * bitmap, int quality)
+{
+ fprintf(stderr, "Error: swftools compiled without jpeglib\n");
+ return -1;
+}
+#endif
-void swf_GetJPEGSize(char *fname, int *width, int *height)
+void swf_GetJPEGSize(const char *fname, int *width, int *height)
{
struct jpeg_decompress_struct cinfo;
struct jpeg_error_mgr jerr;
fclose(fi);
}
-int swf_SetJPEGBits(TAG * t, char *fname, int quality)
+int swf_SetJPEGBits(TAG * t, const char *fname, int quality)
{
struct jpeg_decompress_struct cinfo;
struct jpeg_error_mgr jerr;
}
}
+ rfx_free(scanline);
swf_SetJPEGBitsFinish(out);
jpeg_finish_decompress(&cinfo);
fclose(f);
return 0;
}
+typedef struct _JPEGFILEMGR {
+ struct jpeg_destination_mgr mgr;
+ JOCTET *buffer;
+ struct jpeg_compress_struct* cinfo;
+ struct jpeg_error_mgr* jerr;
+ FILE*fi;
+} JPEGFILEMGR;
+
+static void file_init_destination(j_compress_ptr cinfo)
+{
+ JPEGFILEMGR*fmgr = (JPEGFILEMGR*)(cinfo->dest);
+ struct jpeg_destination_mgr*dmgr = &fmgr->mgr;
+
+ fmgr->buffer = (JOCTET*)rfx_alloc(OUTBUFFER_SIZE);
+ if(!fmgr->buffer) {
+ perror("malloc");
+ fprintf(stderr, "Out of memory!\n");
+ exit(1);
+ }
+
+ dmgr->next_output_byte = fmgr->buffer;
+ dmgr->free_in_buffer = OUTBUFFER_SIZE;
+}
+
+static boolean file_empty_output_buffer(j_compress_ptr cinfo)
+{
+ JPEGFILEMGR*fmgr = (JPEGFILEMGR*)(cinfo->dest);
+ struct jpeg_destination_mgr*dmgr = &fmgr->mgr;
+
+ if(fmgr->fi)
+ fwrite(fmgr->buffer, OUTBUFFER_SIZE, 1, fmgr->fi);
+
+ dmgr->next_output_byte = fmgr->buffer;
+ dmgr->free_in_buffer = OUTBUFFER_SIZE;
+ return 1;
+}
+
+static void file_term_destination(j_compress_ptr cinfo)
+{
+ JPEGFILEMGR*fmgr = (JPEGFILEMGR*)(cinfo->dest);
+ struct jpeg_destination_mgr*dmgr = &fmgr->mgr;
+
+ if(fmgr->fi)
+ fwrite(fmgr->buffer, OUTBUFFER_SIZE-dmgr->free_in_buffer, 1, fmgr->fi);
+
+ rfx_free(fmgr->buffer);
+ fmgr->buffer = 0;
+ dmgr->free_in_buffer = 0;
+ dmgr->next_output_byte = 0;
+}
+
+void swf_SaveJPEG(char*filename, RGBA*pixels, int width, int height, int quality)
+{
+ JPEGFILEMGR fmgr;
+ struct jpeg_compress_struct cinfo;
+ struct jpeg_error_mgr jerr;
+ unsigned char*data2 = 0;
+ int y;
+
+ FILE*fi = fopen(filename, "wb");
+ if(!fi) {
+ char buf[256];
+ sprintf(buf, "rfxswf: Couldn't create %s", filename);
+ perror(buf);
+ return;
+ }
+ data2 = (unsigned char *)rfx_calloc(width*3);
+
+ memset(&cinfo, 0, sizeof(cinfo));
+ memset(&jerr, 0, sizeof(jerr));
+ memset(&fmgr, 0, sizeof(fmgr));
+ cinfo.err = jpeg_std_error(&jerr);
+ jpeg_create_compress(&cinfo);
+
+ fmgr.mgr.init_destination = file_init_destination;
+ fmgr.mgr.empty_output_buffer = file_empty_output_buffer;
+ fmgr.mgr.term_destination = file_term_destination;
+ fmgr.fi = fi;
+ fmgr.cinfo = &cinfo;
+ fmgr.jerr = &jerr;
+ cinfo.dest = (struct jpeg_destination_mgr*)&fmgr;
+
+ // init compression
+
+ cinfo.image_width = width;
+ cinfo.image_height = height;
+ cinfo.input_components = 3;
+ cinfo.in_color_space = JCS_RGB;
+ jpeg_set_defaults(&cinfo);
+ cinfo.dct_method = JDCT_IFAST;
+ jpeg_set_quality(&cinfo,quality,TRUE);
+
+ //jpeg_write_tables(&cinfo);
+ //jpeg_suppress_tables(&cinfo, TRUE);
+ jpeg_start_compress(&cinfo, FALSE);
+
+ for(y=0;y<height;y++) {
+ int x;
+ RGBA*src = &pixels[y*width];
+ for(x=0;x<width;x++) {
+ data2[x*3+0] = src[x].r;
+ data2[x*3+1] = src[x].g;
+ data2[x*3+2] = src[x].b;
+ }
+ jpeg_write_scanlines(&cinfo, &data2, 1);
+ }
+ rfx_free(data2);
+ jpeg_finish_compress(&cinfo);
+ jpeg_destroy_compress(&cinfo);
+
+ fclose(fi);
+}
+
/* jpeg_source_mgr functions */
static void tag_init_source(struct jpeg_decompress_struct *cinfo)
{
static boolean tag_fill_input_buffer(struct jpeg_decompress_struct *cinfo)
{
TAG *tag = (TAG *) cinfo->client_data;
- if (tag->data[tag->pos + 0] == 0xff &&
+ if (tag->pos + 4 <= tag->len &&
+ tag->data[tag->pos + 0] == 0xff &&
tag->data[tag->pos + 1] == 0xd9 &&
tag->data[tag->pos + 2] == 0xff &&
tag->data[tag->pos + 3] == 0xd8) {
jpeg_read_header(&cinfo, TRUE);
*width = cinfo.image_width;
*height = cinfo.image_height;
- dest =
+ dest = (RGBA*)
rfx_alloc(sizeof(RGBA) * cinfo.image_width * cinfo.image_height);
jpeg_start_decompress(&cinfo);
#ifdef HAVE_ZLIB
if(offset) {
- U32 datalen = cinfo.output_width*cinfo.output_height;
+ uLongf datalen = cinfo.output_width*cinfo.output_height;
U8* alphadata = (U8*)rfx_alloc(datalen);
int error;
tag->len = oldtaglen;
int RFXSWF_deflate_wraper(TAG * t, z_stream * zs, boolean finish)
{
- U8 *data = rfx_alloc(OUTBUFFER_SIZE);
+ U8 *data = (U8*)rfx_alloc(OUTBUFFER_SIZE);
zs->next_out = data;
zs->avail_out = OUTBUFFER_SIZE;
while (1) {
switch (bitmap_flags) {
case BMF_8BIT:
- return swf_SetLosslessBitsIndexed(t, width, height, bitmap, NULL, 256);
+ return swf_SetLosslessBitsIndexed(t, width, height, (U8*)bitmap, NULL, 256);
case BMF_16BIT:
bps = BYTES_PER_SCANLINE(sizeof(U16) * width);
break;
if (deflateInit(&zs, Z_DEFAULT_COMPRESSION) == Z_OK) {
zs.avail_in = bps * height;
- zs.next_in = bitmap;
+ zs.next_in = (Bytef *)bitmap;
if (RFXSWF_deflate_wraper(t, &zs, TRUE) < 0)
res = -3;
if (!pal) // create default palette for grayscale images
{
int i;
- pal = rfx_alloc(256 * sizeof(RGBA));
+ pal = (RGBA*)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;
if (deflateInit(&zs, Z_DEFAULT_COMPRESSION) == Z_OK) {
U8 *zpal; // compress palette
- if ((zpal = rfx_alloc(ncolors * 4))) {
+ if ((zpal = (U8*)rfx_alloc(ncolors * 4))) {
U8 *pp = zpal;
int i;
}
}
+/* expects mem to be non-premultiplied */
void swf_SetLosslessImage(TAG*tag, RGBA*data, int width, int height)
{
int hasalpha = swf_ImageHasAlpha(data, width, height);
RGBA *swf_DefineLosslessBitsTagToImage(TAG * tag, int *dwidth, int *dheight)
{
int id, format, height, width, pos;
- U32 datalen, datalen2;
+ uLongf datalen, datalen2;
int error;
int bpp = 1;
int cols = 0;
*dwidth = width = swf_GetU16(tag);
*dheight = height = swf_GetU16(tag);
- dest = rfx_alloc(sizeof(RGBA) * width * height);
+ dest = (RGBA*)rfx_alloc(sizeof(RGBA) * width * height);
if (format == 3)
cols = swf_GetU8(tag) + 1;
if (data)
rfx_free(data);
datalen += 4096;
- data = rfx_alloc(datalen);
+ data = (U8*)rfx_alloc(datalen);
error =
uncompress(data, &datalen, &tag->data[tag->pos],
tag->len - tag->pos);
}
} else {
for (x = 0; x < width; x++) {
- /* TODO: is un-premultiplying alpha the right thing to do?
- dest[pos2].r = data[pos + 1];
- dest[pos2].g = data[pos + 2];
- dest[pos2].b = data[pos + 3];*/
+ /* remove premultiplication */
int alpha = data[pos+0];
- if(alpha) {
- dest[pos2].r = ((int)data[pos + 1]*255)/alpha;
- dest[pos2].g = ((int)data[pos + 2]*255)/alpha;
- dest[pos2].b = ((int)data[pos + 3]*255)/alpha;
- } else {
- dest[pos2].r = data[pos + 1];
- dest[pos2].g = data[pos + 2];
- dest[pos2].b = data[pos + 3];
- }
+ 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;
#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;
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++) {
- U8 scanline[3 * width];
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 = rfx_alloc(OUTBUFFER_SIZE);
+ data = (U8*)rfx_alloc(OUTBUFFER_SIZE);
memset(&zs, 0x00, sizeof(z_stream));
if (deflateInit(&zs, Z_DEFAULT_COMPRESSION) != Z_OK) {
zs.next_out = data;
zs.avail_out = OUTBUFFER_SIZE;
+ scanline = (U8*)rfx_alloc(width);
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;
}
}
+ rfx_free(scanline);
+
while (1) {
int ret = deflate(&zs, Z_FINISH);
if (ret != Z_OK && ret != Z_STREAM_END) {
return 0;
}
+#else
+int swf_SetJPEGBits3(TAG * tag, U16 width, U16 height, RGBA * bitmap, int quality)
+{
+ fprintf(stderr, "Error: swftools compiled without jpeglib\n");
+ return -1;
+}
+#endif
+
+
+/* expects mem to be non-premultiplied */
TAG* swf_AddImage(TAG*tag, int bitid, RGBA*mem, int width, int height, int quality)
{
TAG *tag1 = 0, *tag2 = 0;
int has_alpha = swf_ImageHasAlpha(mem,width,height);
/* try lossless image */
+
+#ifdef NO_LOSSLESS
+ tag1 = swf_InsertTag(0, /*ST_DEFINEBITSLOSSLESS1/2*/0);
+ tag1->len = 0x7fffffff;
+#else
tag1 = swf_InsertTag(0, /*ST_DEFINEBITSLOSSLESS1/2*/0);
swf_SetU16(tag1, bitid);
swf_SetLosslessImage(tag1, mem, width, height);
+#endif
+#if defined(HAVE_JPEGLIB)
/* try jpeg image */
if(has_alpha) {
tag2 = swf_InsertTag(0, ST_DEFINEBITSJPEG3);
swf_SetU16(tag2, bitid);
swf_SetJPEGBits2(tag2, width, height, mem, quality);
}
+#endif
- if(tag1 && tag1->len < tag2->len) {
+ if(quality>100 || !tag2 || (tag1 && tag1->len < tag2->len)) {
/* use the zlib version- it's smaller */
tag1->prev = tag;
if(tag) tag->next = tag1;
tag = tag1;
- swf_DeleteTag(tag2);
+ swf_DeleteTag(0, tag2);
} else {
/* use the jpeg version- it's smaller */
tag2->prev = tag;
if(tag) tag->next = tag2;
tag = tag2;
- swf_DeleteTag(tag1);
+ swf_DeleteTag(0, tag1);
}
return tag;
}
-#endif
-
RGBA *swf_ExtractImage(TAG * tag, int *dwidth, int *dheight)
{
RGBA *img;
if (tag->id == ST_DEFINEBITSJPEG) {
int len = tag->len;
void *data = rfx_alloc(len);
- swf_GetBlock(tag, data, tag->len);
+ swf_GetBlock(tag, (U8*)data, tag->len);
swf_ResetTag(tag, ST_DEFINEBITSJPEG2);
swf_SetBlock(tag, &((U8*)data)[0], 2); //id
swf_SetBlock(tag, tables_tag->data, tables_tag->len);
}
if (swf->firstTag == tables_tag)
swf->firstTag = tables_tag->next;
- swf_DeleteTag(tables_tag);
+ swf_DeleteTag(swf, tables_tag);
}
typedef struct scale_lookup {
static scale_lookup_t**make_scale_lookup(int width, int newwidth)
{
- scale_lookup_t*lookupx = malloc((width>newwidth?width:newwidth)*2*sizeof(scale_lookup_t));
+ scale_lookup_t*lookupx = (scale_lookup_t*)malloc((width>newwidth?width:newwidth)*2*sizeof(scale_lookup_t));
scale_lookup_t**lblockx = (scale_lookup_t**)malloc((newwidth+1)*sizeof(scale_lookup_t**));
double fx = ((double)width)/((double)newwidth);
double px = 0;
lblockx[newwidth] = p_x;
return lblockx;
}
+
+static void encodeMonochromeImage(RGBA*data, int width, int height, RGBA*colors)
+{
+ int t;
+ int len = width*height;
+
+ U32* img = (U32*)data;
+ U32 color1 = img[0];
+ U32 color2 = 0;
+ for(t=1;t<len;t++) {
+ if(img[t] != color1) {
+ color2 = img[t];
+ break;
+ }
+ }
+ *(U32*)&colors[0] = color1;
+ *(U32*)&colors[1] = color2;
+ for(t=0;t<len;t++) {
+ if(img[t] == color1) {
+ img[t] = 0;
+ } else {
+ img[t] = 0xffffffff;
+ }
+ }
+}
+
+static void decodeMonochromeImage(RGBA*data, int width, int height, RGBA*colors)
+{
+ int t;
+ int len = width*height;
+
+ for(t=0;t<len;t++) {
+ U32 m = data[t].r;
+ data[t].r = (colors[0].r * (255-m) + colors[1].r * m) >> 8;
+ data[t].g = (colors[0].g * (255-m) + colors[1].g * m) >> 8;
+ data[t].b = (colors[0].b * (255-m) + colors[1].b * m) >> 8;
+ data[t].a = (colors[0].a * (255-m) + colors[1].a * m) >> 8;
+ }
+}
+
+static void blurImage(RGBA*src, int width, int height, int r)
+{
+ int e = 2; // r times e is the sampling interval
+ double*gauss = (double*)malloc(r*e*sizeof(double));
+ double sum=0;
+ int x;
+ for(x=0;x<r*e;x++) {
+ double t = (x - r*e/2.0)/r;
+ gauss[x] = exp(-0.5*t*t);
+ sum += gauss[x];
+ }
+ int*weights = (int*)malloc(r*e*sizeof(int));
+ for(x=0;x<r*e;x++) {
+ weights[x] = (int)(gauss[x]*65536.0001/sum);
+ }
+ int range = r*e/2;
+
+ RGBA*tmp = malloc(sizeof(RGBA)*width*height);
+
+ int y;
+ for(y=0;y<height;y++) {
+ RGBA*s = &src[y*width];
+ RGBA*d = &tmp[y*width];
+ for(x=0;x<range;x++) {
+ d[x] = s[x];
+ }
+ for(x=range;x<width-range;x++) {
+ int r=0;
+ int g=0;
+ int b=0;
+ int a=0;
+ int*f = weights;
+ int xx;
+ for(xx=x-range;xx<x+range;xx++) {
+ r += s[xx].r * f[0];
+ g += s[xx].g * f[0];
+ b += s[xx].b * f[0];
+ a += s[xx].a * f[0];
+ f++;
+ }
+ d[x].r = r >> 16;
+ d[x].g = g >> 16;
+ d[x].b = b >> 16;
+ d[x].a = a >> 16;
+ }
+ for(x=width-range;x<width;x++) {
+ d[x] = s[x];
+ }
+ }
+
+ for(x=0;x<width;x++) {
+ RGBA*s = &tmp[x];
+ RGBA*d = &src[x];
+ int yy=0;
+ for(y=0;y<range;y++) {
+ d[yy] = s[yy];
+ yy+=width;
+ }
+ for(y=range;y<height-range;y++) {
+ int r=0;
+ int g=0;
+ int b=0;
+ int a=0;
+ int*f = weights;
+ int cy,cyy=yy-range*width;
+ for(cy=y-range;cy<y+range;cy++) {
+ r += s[cyy].r * f[0];
+ g += s[cyy].g * f[0];
+ b += s[cyy].b * f[0];
+ a += s[cyy].a * f[0];
+ cyy += width;
+ f++;
+ }
+ d[yy].r = r >> 16;
+ d[yy].g = g >> 16;
+ d[yy].b = b >> 16;
+ d[yy].a = a >> 16;
+ yy += width;
+ }
+ for(y=0;y<range;y++) {
+ d[yy] = s[yy];
+ yy += width;
+ }
+ }
+
+ free(tmp);
+ free(weights);
+ free(gauss);
+}
+
+
RGBA* swf_ImageScale(RGBA*data, int width, int height, int newwidth, int newheight)
{
int x,y;
RGBA* newdata;
scale_lookup_t *p, **lblockx,**lblocky;
rgba_int_t*tmpline;
+ int monochrome = 0;
+ RGBA monochrome_colors[2];
if(newwidth<1 || newheight<1)
return 0;
- /* this is bad because this scaler doesn't yet handle monochrome
- images with 2 colors in a way that the final image hasn't more
- than 256 colors */
- if(swf_ImageGetNumberOfPaletteEntries2(data, width, height) == 2)
- fprintf(stderr, "Warning: scaling monochrome image\n");
+ if(swf_ImageGetNumberOfPaletteEntries2(data, width, height) == 2) {
+ monochrome=1;
+ encodeMonochromeImage(data, width, height, monochrome_colors);
+ int r1 = width / newwidth;
+ int r2 = height / newheight;
+ int r = r1<r2?r1:r2;
+ if(r>4) {
+ /* high-resolution monochrome images are usually dithered, so
+ low-pass filter them first to get rid of any moire patterns */
+ blurImage(data, width, height, r+1);
+ }
+ }
tmpline = (rgba_int_t*)malloc(width*sizeof(rgba_int_t));
newdata = (RGBA*)malloc(newwidth*newheight*sizeof(RGBA));
destline++;
}
}
+
+ if(monochrome)
+ decodeMonochromeImage(newdata, newwidth, newheight, monochrome_colors);
+
free(tmpline);
free(*lblockx);
free(lblockx);
return newdata;
}
+
git.asbjorn.biz / swftools.git / blobdiff