X-Git-Url: http://git.asbjorn.biz/?a=blobdiff_plain;f=lib%2Fmodules%2Fswfbits.c;h=5327d68b30b9eb8c700d8b0d5cd01abe2c872588;hb=f52e48e2f55dd3a724da00a47552b7edba5f4dfa;hp=8852237e54187748448416be069b15a548c0c231;hpb=3ed992bcfe4f55c4c2ca1ccd49d4ab81796d04d3;p=swftools.git diff --git a/lib/modules/swfbits.c b/lib/modules/swfbits.c index 8852237..5327d68 100644 --- a/lib/modules/swfbits.c +++ b/lib/modules/swfbits.c @@ -36,37 +36,140 @@ int swf_ImageHasAlpha(RGBA*img, int width, int height) } return hasalpha; } - -int swf_ImageGetNumberOfPaletteEntries(RGBA*img, int width, int height, RGBA*palette) + +int swf_ImageGetNumberOfPaletteEntries2(RGBA*_img, int width, int height) +{ + int len = width*height; + int t; + U32* img = (U32*)_img; + U32 color1 = img[0]; + U32 color2 = 0; + for(t=1;t> 17) ^ col32; + hash ^= ((hash>>8) + 1) ^ hash; + hash &= 255; + + csize = size[hash]; + cpal = &pal[hash*256]; + for(i=0;icinfo); + jpeg_destroy_compress(&jpeg->cinfo); rfx_free(jpeg); return 0; } @@ -290,6 +394,7 @@ int swf_SetJPEGBits(TAG * t, char *fname, int quality) } } + rfx_free(scanline); swf_SetJPEGBitsFinish(out); jpeg_finish_decompress(&cinfo); fclose(f); @@ -297,11 +402,128 @@ int swf_SetJPEGBits(TAG * t, char *fname, int quality) 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 = 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;yclient_data; - swf_SetTagPos(tag, 2); + if (tag->id == ST_DEFINEBITSJPEG3) { + swf_SetTagPos(tag, 6); + } else { + swf_SetTagPos(tag, 2); + } cinfo->src->bytes_in_buffer = 0; } static boolean tag_fill_input_buffer(struct jpeg_decompress_struct *cinfo) @@ -345,18 +567,24 @@ RGBA *swf_JPEG2TagToImage(TAG * tag, int *width, int *height) struct jpeg_source_mgr mgr; RGBA *dest; int y; + int offset = 0; + int oldtaglen = 0; *width = 0; *height = 0; if (tag->id == ST_DEFINEBITSJPEG) { - fprintf(stderr, - "rfxswf: extracting from definebitsjpeg not yet supported"); + fprintf(stderr, "rfxswf: extracting from definebitsjpeg not yet supported\n"); return 0; } if (tag->id == ST_DEFINEBITSJPEG3) { - fprintf(stderr, - "rfxswf: extracting from definebitsjpeg3 not yet supported"); +#ifdef HAVE_ZLIB + offset = swf_GetU32(tag); + oldtaglen = tag->len; + tag->len = offset+6; +#else + fprintf(stderr, "rfxswf: extracting from definebitsjpeg3 not possible: no zlib\n"); return 0; +#endif } cinfo.err = jpeg_std_error(&jerr); @@ -397,6 +625,30 @@ RGBA *swf_JPEG2TagToImage(TAG * tag, int *width, int *height) jpeg_finish_decompress(&cinfo); jpeg_destroy_decompress(&cinfo); + +#ifdef HAVE_ZLIB + if(offset) { + uLongf datalen = cinfo.output_width*cinfo.output_height; + U8* alphadata = (U8*)rfx_alloc(datalen); + int error; + tag->len = oldtaglen; + swf_SetTagPos(tag, 6+offset); + error = uncompress(alphadata, &datalen, &tag->data[tag->pos], tag->len - tag->pos); + if (error != Z_OK) { + fprintf(stderr, "rfxswf: Zlib error %d while extracting definejpeg3\n", error); + return 0; + } + for(y=0;yid = ST_DEFINEBITSLOSSLESS; } else { tag->id = ST_DEFINEBITSLOSSLESS2; - /* TODO: premultiply alpha? */ + swf_PreMultiplyAlpha(data, width, height); } - int num = swf_ImageGetNumberOfPaletteEntries(data, width, height, 0); + num = swf_ImageGetNumberOfPaletteEntries(data, width, height, 0); if(num>1 && num<=256) { RGBA*palette = (RGBA*)malloc(sizeof(RGBA)*num); - swf_ImageGetNumberOfPaletteEntries(data, width, height, palette); 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>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; @@ -768,6 +1038,8 @@ RGBA *swf_DefineLosslessBitsTagToImage(TAG * tag, int *dwidth, int *dheight) #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; @@ -784,6 +1056,14 @@ int swf_SetJPEGBits3(TAG * tag, U16 width, U16 height, RGBA * bitmap, int qualit 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; @@ -849,11 +1129,53 @@ int swf_SetJPEGBits3(TAG * tag, U16 width, U16 height, RGBA * bitmap, int qualit rfx_free(data); return 0; } + +/* 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 */ + tag1 = swf_InsertTag(0, /*ST_DEFINEBITSLOSSLESS1/2*/0); + swf_SetU16(tag1, bitid); + swf_SetLosslessImage(tag1, mem, width, height); + + /* try jpeg image */ + if(has_alpha) { + tag2 = swf_InsertTag(0, ST_DEFINEBITSJPEG3); + swf_SetU16(tag2, bitid); + swf_SetJPEGBits3(tag2, width, height, mem, quality); + } else { + tag2 = swf_InsertTag(0, ST_DEFINEBITSJPEG2); + swf_SetU16(tag2, bitid); + swf_SetJPEGBits2(tag2, width, height, mem, quality); + } + + if(quality>100 || (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); + } else { + /* use the jpeg version- it's smaller */ + tag2->prev = tag; + if(tag) tag->next = tag2; + tag = tag2; + swf_DeleteTag(tag1); + } + return tag; +} + #endif RGBA *swf_ExtractImage(TAG * tag, int *dwidth, int *dheight) { RGBA *img; + + swf_SetTagPos(tag, 2); // id is 2 bytes + if (tag->id == ST_DEFINEBITSJPEG || tag->id == ST_DEFINEBITSJPEG2 || tag->id == ST_DEFINEBITSJPEG3) { #ifdef HAVE_JPEGLIB @@ -897,11 +1219,13 @@ void swf_RemoveJPEGTables(SWF * swf) tag = swf->firstTag; while (tag) { if (tag->id == ST_DEFINEBITSJPEG) { - void *data = rfx_alloc(tag->len); + int len = tag->len; + void *data = rfx_alloc(len); swf_GetBlock(tag, 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); - swf_SetBlock(tag, data, tag->len); + swf_SetBlock(tag, &((U8*)data)[2], len-2); free(data); } tag = tag->next; @@ -933,7 +1257,6 @@ static scale_lookup_t**make_scale_lookup(int width, int newwidth) if(newwidth<=width) { for(x=0;x=width) tox = width-1; for(xx=fromx;xx<=tox;xx++) { if(xx==fromx && xx==tox) p_x->weight = 256; @@ -956,11 +1280,11 @@ static scale_lookup_t**make_scale_lookup(int width, int newwidth) } } else { for(x=0;x=width) ix2=width-1; double r = px-ix1; + if(ix2>=width) ix2=width-1; + lblockx[x] = p_x; if(bicubic) r = -2*r*r*r+3*r*r; p_x[0].weight = (int)(256*(1-r)); @@ -975,14 +1299,64 @@ static scale_lookup_t**make_scale_lookup(int width, int newwidth) 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> 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; + } +} + RGBA* swf_ImageScale(RGBA*data, int width, int height, int newwidth, int newheight) { - if(newwidth<1 || newheight<1) - return 0; int x,y; - RGBA* newdata= (RGBA*)malloc(newwidth*newheight*sizeof(RGBA)); + RGBA* newdata; scale_lookup_t *p, **lblockx,**lblocky; - rgba_int_t*tmpline = (rgba_int_t*)malloc(width*sizeof(rgba_int_t)); + rgba_int_t*tmpline; + int monochrome = 0; + RGBA monochrome_colors[2]; + + if(newwidth<1 || newheight<1) + return 0; + + if(swf_ImageGetNumberOfPaletteEntries2(data, width, height) == 2) { + monochrome=1; + encodeMonochromeImage(data, width, height, monochrome_colors); + } + + tmpline = (rgba_int_t*)malloc(width*sizeof(rgba_int_t)); + newdata = (RGBA*)malloc(newwidth*newheight*sizeof(RGBA)); lblockx = make_scale_lookup(width, newwidth); lblocky = make_scale_lookup(height, newheight); @@ -995,7 +1369,7 @@ RGBA* swf_ImageScale(RGBA*data, int width, int height, int newwidth, int newheig /* create lookup table for y */ rgba_int_t*l = tmpline; - scale_lookup_t*p_y; + scale_lookup_t*p_y,*p_x; memset(tmpline, 0, width*sizeof(rgba_int_t)); for(p_y=lblocky[y];p_ypos]; @@ -1010,7 +1384,7 @@ RGBA* swf_ImageScale(RGBA*data, int width, int height, int newwidth, int newheig } /* process x direction */ - scale_lookup_t*p_x = lblockx[0]; + p_x = lblockx[0]; for(x=0;x