diff options
author | Tom Hudson <tomhudson@google.com> | 2016-05-04 13:22:56 -0400 |
---|---|---|
committer | Tom Hudson <tomhudson@google.com> | 2016-05-04 13:22:56 -0400 |
commit | 0d47d2d3a728e78676a15b1d818cc668cb7e5a9c (patch) | |
tree | 044a430eeaa2dec4d6de7b624da15fda1b8ed25f /turbojpeg.c | |
parent | 9d35298a6223278a66423f828a949d93d94d5911 (diff) |
Update to libjpeg_turbo 1.4.90
(Duplicate of https://codereview.chromium.org/1939823002/ for landing.)
TBR=noel@chromium.org,thakis@chromium.org
BUG=608347, 398235, 591927
Review URL: https://codereview.chromium.org/1953443002 .
Diffstat (limited to 'turbojpeg.c')
-rw-r--r-- | turbojpeg.c | 995 |
1 files changed, 888 insertions, 107 deletions
diff --git a/turbojpeg.c b/turbojpeg.c index 9117273..f51df78 100644 --- a/turbojpeg.c +++ b/turbojpeg.c @@ -1,5 +1,5 @@ /* - * Copyright (C)2009-2012, 2014 D. R. Commander. All Rights Reserved. + * Copyright (C)2009-2016 D. R. Commander. All Rights Reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: @@ -31,6 +31,7 @@ #include <stdio.h> #include <stdlib.h> +#include <ctype.h> #include <jinclude.h> #define JPEG_INTERNALS #include <jpeglib.h> @@ -39,12 +40,15 @@ #include "./turbojpeg.h" #include "./tjutil.h" #include "transupp.h" +#include "./jpegcomp.h" extern void jpeg_mem_dest_tj(j_compress_ptr, unsigned char **, unsigned long *, boolean); -extern void jpeg_mem_src_tj(j_decompress_ptr, unsigned char *, unsigned long); +extern void jpeg_mem_src_tj(j_decompress_ptr, const unsigned char *, + unsigned long); #define PAD(v, p) ((v+(p)-1)&(~((p)-1))) +#define isPow2(x) (((x)&(x-1))==0) /* Error handling (based on example in example.c) */ @@ -55,6 +59,8 @@ struct my_error_mgr { struct jpeg_error_mgr pub; jmp_buf setjmp_buffer; + void (*emit_message)(j_common_ptr, int); + boolean warning; }; typedef struct my_error_mgr *my_error_ptr; @@ -72,6 +78,13 @@ static void my_output_message(j_common_ptr cinfo) (*cinfo->err->format_message)(cinfo, errStr); } +static void my_emit_message(j_common_ptr cinfo, int msg_level) +{ + my_error_ptr myerr=(my_error_ptr)cinfo->err; + myerr->emit_message(cinfo, msg_level); + if(msg_level<0) myerr->warning=TRUE; +} + /* Global structures, macros, etc. */ @@ -82,10 +95,10 @@ typedef struct _tjinstance struct jpeg_compress_struct cinfo; struct jpeg_decompress_struct dinfo; struct my_error_mgr jerr; - int init; + int init, headerRead; } tjinstance; -static const int pixelsize[TJ_NUMSAMP]={3, 3, 3, 1, 3}; +static const int pixelsize[TJ_NUMSAMP]={3, 3, 3, 1, 3, 3}; static const JXFORM_CODE xformtypes[TJ_NUMXOP]= { @@ -119,7 +132,20 @@ static const tjscalingfactor sf[NUMSF]={ j_compress_ptr cinfo=NULL; j_decompress_ptr dinfo=NULL; \ if(!this) {snprintf(errStr, JMSG_LENGTH_MAX, "Invalid handle"); \ return -1;} \ - cinfo=&this->cinfo; dinfo=&this->dinfo; + cinfo=&this->cinfo; dinfo=&this->dinfo; \ + this->jerr.warning=FALSE; +#define getcinstance(handle) tjinstance *this=(tjinstance *)handle; \ + j_compress_ptr cinfo=NULL; \ + if(!this) {snprintf(errStr, JMSG_LENGTH_MAX, "Invalid handle"); \ + return -1;} \ + cinfo=&this->cinfo; \ + this->jerr.warning=FALSE; +#define getdinstance(handle) tjinstance *this=(tjinstance *)handle; \ + j_decompress_ptr dinfo=NULL; \ + if(!this) {snprintf(errStr, JMSG_LENGTH_MAX, "Invalid handle"); \ + return -1;} \ + dinfo=&this->dinfo; \ + this->jerr.warning=FALSE; static int getPixelFormat(int pixelSize, int flags) { @@ -149,6 +175,7 @@ static int setCompDefaults(struct jpeg_compress_struct *cinfo, int pixelFormat, int subsamp, int jpegQual, int flags) { int retval=0; + char *env=NULL; switch(pixelFormat) { @@ -185,10 +212,34 @@ static int setCompDefaults(struct jpeg_compress_struct *cinfo, cinfo->in_color_space=JCS_RGB; pixelFormat=TJPF_RGB; break; #endif + case TJPF_CMYK: + cinfo->in_color_space=JCS_CMYK; break; } cinfo->input_components=tjPixelSize[pixelFormat]; jpeg_set_defaults(cinfo); + +#ifndef NO_GETENV + if((env=getenv("TJ_OPTIMIZE"))!=NULL && strlen(env)>0 && !strcmp(env, "1")) + cinfo->optimize_coding=TRUE; + if((env=getenv("TJ_ARITHMETIC"))!=NULL && strlen(env)>0 && !strcmp(env, "1")) + cinfo->arith_code=TRUE; + if((env=getenv("TJ_RESTART"))!=NULL && strlen(env)>0) + { + int temp=-1; char tempc=0; + if(sscanf(env, "%d%c", &temp, &tempc)>=1 && temp>=0 && temp<=65535) + { + if(toupper(tempc)=='B') + { + cinfo->restart_interval=temp; + cinfo->restart_in_rows=0; + } + else + cinfo->restart_in_rows=temp; + } + } +#endif + if(jpegQual>=0) { jpeg_set_quality(cinfo, jpegQual, TRUE); @@ -197,15 +248,26 @@ static int setCompDefaults(struct jpeg_compress_struct *cinfo, } if(subsamp==TJSAMP_GRAY) jpeg_set_colorspace(cinfo, JCS_GRAYSCALE); - else - jpeg_set_colorspace(cinfo, JCS_YCbCr); + else if(pixelFormat==TJPF_CMYK) + jpeg_set_colorspace(cinfo, JCS_YCCK); + else jpeg_set_colorspace(cinfo, JCS_YCbCr); + +#ifndef NO_GETENV + if((env=getenv("TJ_PROGRESSIVE"))!=NULL && strlen(env)>0 + && !strcmp(env, "1")) + jpeg_simple_progression(cinfo); +#endif cinfo->comp_info[0].h_samp_factor=tjMCUWidth[subsamp]/8; cinfo->comp_info[1].h_samp_factor=1; cinfo->comp_info[2].h_samp_factor=1; + if(cinfo->num_components>3) + cinfo->comp_info[3].h_samp_factor=tjMCUWidth[subsamp]/8; cinfo->comp_info[0].v_samp_factor=tjMCUHeight[subsamp]/8; cinfo->comp_info[1].v_samp_factor=1; cinfo->comp_info[2].v_samp_factor=1; + if(cinfo->num_components>3) + cinfo->comp_info[3].v_samp_factor=tjMCUHeight[subsamp]/8; return retval; } @@ -255,6 +317,8 @@ static int setDecompDefaults(struct jpeg_decompress_struct *dinfo, case TJPF_ABGR: dinfo->out_color_space=JCS_RGB; break; #endif + case TJPF_CMYK: + dinfo->out_color_space=JCS_CMYK; break; default: _throw("Unsupported pixel format"); } @@ -269,9 +333,20 @@ static int setDecompDefaults(struct jpeg_decompress_struct *dinfo, static int getSubsamp(j_decompress_ptr dinfo) { int retval=-1, i, k; + + /* The sampling factors actually have no meaning with grayscale JPEG files, + and in fact it's possible to generate grayscale JPEGs with sampling + factors > 1 (even though those sampling factors are ignored by the + decompressor.) Thus, we need to treat grayscale as a special case. */ + if(dinfo->num_components==1 && dinfo->jpeg_color_space==JCS_GRAYSCALE) + return TJSAMP_GRAY; + for(i=0; i<NUMSUBOPT; i++) { - if(dinfo->num_components==pixelsize[i]) + if(dinfo->num_components==pixelsize[i] + || ((dinfo->jpeg_color_space==JCS_YCCK + || dinfo->jpeg_color_space==JCS_CMYK) + && pixelsize[i]==3 && dinfo->num_components==4)) { if(dinfo->comp_info[0].h_samp_factor==tjMCUWidth[i]/8 && dinfo->comp_info[0].v_samp_factor==tjMCUHeight[i]/8) @@ -279,8 +354,13 @@ static int getSubsamp(j_decompress_ptr dinfo) int match=0; for(k=1; k<dinfo->num_components; k++) { - if(dinfo->comp_info[k].h_samp_factor==1 - && dinfo->comp_info[k].v_samp_factor==1) + int href=1, vref=1; + if(dinfo->jpeg_color_space==JCS_YCCK && k==3) + { + href=tjMCUWidth[i]/8; vref=tjMCUHeight[i]/8; + } + if(dinfo->comp_info[k].h_samp_factor==href + && dinfo->comp_info[k].v_samp_factor==vref) match++; } if(match==dinfo->num_components-1) @@ -477,12 +557,15 @@ DLLEXPORT unsigned char *DLLCALL tjAlloc(int bytes) static tjhandle _tjInitCompress(tjinstance *this) { - unsigned char buffer[1], *buf=buffer; unsigned long size=1; + static unsigned char buffer[1]; + unsigned char *buf=buffer; unsigned long size=1; /* This is also straight out of example.c */ this->cinfo.err=jpeg_std_error(&this->jerr.pub); this->jerr.pub.error_exit=my_error_exit; this->jerr.pub.output_message=my_output_message; + this->jerr.emit_message=this->jerr.pub.emit_message; + this->jerr.pub.emit_message=my_emit_message; if(setjmp(this->jerr.setjmp_buffer)) { @@ -547,22 +630,33 @@ DLLEXPORT unsigned long DLLCALL TJBUFSIZE(int width, int height) } -DLLEXPORT unsigned long DLLCALL tjBufSizeYUV(int width, int height, +DLLEXPORT unsigned long DLLCALL tjBufSizeYUV2(int width, int pad, int height, int subsamp) { - unsigned long retval=0; - int pw, ph, cw, ch; - if(width<1 || height<1 || subsamp<0 || subsamp>=NUMSUBOPT) - _throw("tjBufSizeYUV(): Invalid argument"); - pw=PAD(width, tjMCUWidth[subsamp]/8); - ph=PAD(height, tjMCUHeight[subsamp]/8); - cw=pw*8/tjMCUWidth[subsamp]; ch=ph*8/tjMCUHeight[subsamp]; - retval=PAD(pw, 4)*ph + (subsamp==TJSAMP_GRAY? 0:PAD(cw, 4)*ch*2); + int retval=0, nc, i; + + if(subsamp<0 || subsamp>=NUMSUBOPT) + _throw("tjBufSizeYUV2(): Invalid argument"); + + nc=(subsamp==TJSAMP_GRAY? 1:3); + for(i=0; i<nc; i++) + { + int pw=tjPlaneWidth(i, width, subsamp); + int stride=PAD(pw, pad); + int ph=tjPlaneHeight(i, height, subsamp); + if(pw<0 || ph<0) return -1; + else retval+=stride*ph; + } bailout: return retval; } +DLLEXPORT unsigned long DLLCALL tjBufSizeYUV(int width, int height, + int subsamp) +{ + return tjBufSizeYUV2(width, 4, height, subsamp); +} DLLEXPORT unsigned long DLLCALL TJBUFSIZEYUV(int width, int height, int subsamp) @@ -571,7 +665,72 @@ DLLEXPORT unsigned long DLLCALL TJBUFSIZEYUV(int width, int height, } -DLLEXPORT int DLLCALL tjCompress2(tjhandle handle, unsigned char *srcBuf, +DLLEXPORT int tjPlaneWidth(int componentID, int width, int subsamp) +{ + int pw, nc, retval=0; + + if(width<1 || subsamp<0 || subsamp>=TJ_NUMSAMP) + _throw("tjPlaneWidth(): Invalid argument"); + nc=(subsamp==TJSAMP_GRAY? 1:3); + if(componentID<0 || componentID>=nc) + _throw("tjPlaneWidth(): Invalid argument"); + + pw=PAD(width, tjMCUWidth[subsamp]/8); + if(componentID==0) + retval=pw; + else + retval=pw*8/tjMCUWidth[subsamp]; + + bailout: + return retval; +} + + +DLLEXPORT int tjPlaneHeight(int componentID, int height, int subsamp) +{ + int ph, nc, retval=0; + + if(height<1 || subsamp<0 || subsamp>=TJ_NUMSAMP) + _throw("tjPlaneHeight(): Invalid argument"); + nc=(subsamp==TJSAMP_GRAY? 1:3); + if(componentID<0 || componentID>=nc) + _throw("tjPlaneHeight(): Invalid argument"); + + ph=PAD(height, tjMCUHeight[subsamp]/8); + if(componentID==0) + retval=ph; + else + retval=ph*8/tjMCUHeight[subsamp]; + + bailout: + return retval; +} + + +DLLEXPORT unsigned long DLLCALL tjPlaneSizeYUV(int componentID, int width, + int stride, int height, int subsamp) +{ + unsigned long retval=0; + int pw, ph; + + if(width<1 || height<1 || subsamp<0 || subsamp>=NUMSUBOPT) + _throw("tjPlaneSizeYUV(): Invalid argument"); + + pw=tjPlaneWidth(componentID, width, subsamp); + ph=tjPlaneHeight(componentID, height, subsamp); + if(pw<0 || ph<0) return -1; + + if(stride==0) stride=pw; + else stride=abs(stride); + + retval=stride*(ph-1)+pw; + + bailout: + return retval; +} + + +DLLEXPORT int DLLCALL tjCompress2(tjhandle handle, const unsigned char *srcBuf, int width, int pitch, int height, int pixelFormat, unsigned char **jpegBuf, unsigned long *jpegSize, int jpegSubsamp, int jpegQual, int flags) { @@ -580,7 +739,7 @@ DLLEXPORT int DLLCALL tjCompress2(tjhandle handle, unsigned char *srcBuf, unsigned char *rgbBuf=NULL; #endif - getinstance(handle) + getcinstance(handle) if((this->init&COMPRESS)==0) _throw("tjCompress2(): Instance has not been initialized for compression"); @@ -599,7 +758,7 @@ DLLEXPORT int DLLCALL tjCompress2(tjhandle handle, unsigned char *srcBuf, if(pitch==0) pitch=width*tjPixelSize[pixelFormat]; #ifndef JCS_EXTENSIONS - if(pixelFormat!=TJPF_GRAY) + if(pixelFormat!=TJPF_GRAY && pixelFormat!=TJPF_CMYK) { rgbBuf=(unsigned char *)malloc(width*height*RGB_PIXELSIZE); if(!rgbBuf) _throw("tjCompress2(): Memory allocation failure"); @@ -628,8 +787,9 @@ DLLEXPORT int DLLCALL tjCompress2(tjhandle handle, unsigned char *srcBuf, _throw("tjCompress2(): Memory allocation failure"); for(i=0; i<height; i++) { - if(flags&TJFLAG_BOTTOMUP) row_pointer[i]=&srcBuf[(height-i-1)*pitch]; - else row_pointer[i]=&srcBuf[i*pitch]; + if(flags&TJFLAG_BOTTOMUP) + row_pointer[i]=(JSAMPROW)&srcBuf[(height-i-1)*pitch]; + else row_pointer[i]=(JSAMPROW)&srcBuf[i*pitch]; } while(cinfo->next_scanline<cinfo->image_height) { @@ -644,6 +804,7 @@ DLLEXPORT int DLLCALL tjCompress2(tjhandle handle, unsigned char *srcBuf, if(rgbBuf) free(rgbBuf); #endif if(row_pointer) free(row_pointer); + if(this->jerr.warning) retval=-1; return retval; } @@ -669,23 +830,23 @@ DLLEXPORT int DLLCALL tjCompress(tjhandle handle, unsigned char *srcBuf, } -DLLEXPORT int DLLCALL tjEncodeYUV2(tjhandle handle, unsigned char *srcBuf, - int width, int pitch, int height, int pixelFormat, unsigned char *dstBuf, - int subsamp, int flags) +DLLEXPORT int DLLCALL tjEncodeYUVPlanes(tjhandle handle, + const unsigned char *srcBuf, int width, int pitch, int height, + int pixelFormat, unsigned char **dstPlanes, int *strides, int subsamp, + int flags) { int i, retval=0; JSAMPROW *row_pointer=NULL; JSAMPLE *_tmpbuf[MAX_COMPONENTS], *_tmpbuf2[MAX_COMPONENTS]; JSAMPROW *tmpbuf[MAX_COMPONENTS], *tmpbuf2[MAX_COMPONENTS]; JSAMPROW *outbuf[MAX_COMPONENTS]; - int row, pw, ph, cw[MAX_COMPONENTS], ch[MAX_COMPONENTS]; - JSAMPLE *ptr=dstBuf; - unsigned long yuvsize=0; + int row, pw0, ph0, pw[MAX_COMPONENTS], ph[MAX_COMPONENTS]; + JSAMPLE *ptr; jpeg_component_info *compptr; #ifndef JCS_EXTENSIONS unsigned char *rgbBuf=NULL; #endif - getinstance(handle); + getcinstance(handle); for(i=0; i<MAX_COMPONENTS; i++) { @@ -694,12 +855,14 @@ DLLEXPORT int DLLCALL tjEncodeYUV2(tjhandle handle, unsigned char *srcBuf, } if((this->init&COMPRESS)==0) - _throw("tjEncodeYUV2(): Instance has not been initialized for compression"); + _throw("tjEncodeYUVPlanes(): Instance has not been initialized for compression"); if(srcBuf==NULL || width<=0 || pitch<0 || height<=0 || pixelFormat<0 - || pixelFormat>=TJ_NUMPF || dstBuf==NULL || subsamp<0 + || pixelFormat>=TJ_NUMPF || !dstPlanes || !dstPlanes[0] || subsamp<0 || subsamp>=NUMSUBOPT) - _throw("tjEncodeYUV2(): Invalid argument"); + _throw("tjEncodeYUVPlanes(): Invalid argument"); + if(subsamp!=TJSAMP_GRAY && (!dstPlanes[1] || !dstPlanes[2])) + _throw("tjEncodeYUVPlanes(): Invalid argument"); if(setjmp(this->jerr.setjmp_buffer)) { @@ -708,13 +871,16 @@ DLLEXPORT int DLLCALL tjEncodeYUV2(tjhandle handle, unsigned char *srcBuf, goto bailout; } + if(pixelFormat==TJPF_CMYK) + _throw("tjEncodeYUVPlanes(): Cannot generate YUV images from CMYK pixels"); + if(pitch==0) pitch=width*tjPixelSize[pixelFormat]; #ifndef JCS_EXTENSIONS - if(pixelFormat!=TJPF_GRAY) + if(pixelFormat!=TJPF_GRAY && pixelFormat!=TJPF_CMYK) { rgbBuf=(unsigned char *)malloc(width*height*RGB_PIXELSIZE); - if(!rgbBuf) _throw("tjEncodeYUV2(): Memory allocation failure"); + if(!rgbBuf) _throw("tjEncodeYUVPlanes(): Memory allocation failure"); srcBuf=toRGB(srcBuf, width, pitch, height, pixelFormat, rgbBuf); pitch=width*RGB_PIXELSIZE; } @@ -727,7 +893,6 @@ DLLEXPORT int DLLCALL tjEncodeYUV2(tjhandle handle, unsigned char *srcBuf, else if(flags&TJFLAG_FORCESSE) putenv("JSIMD_FORCESSE=1"); else if(flags&TJFLAG_FORCESSE2) putenv("JSIMD_FORCESSE2=1"); - yuvsize=tjBufSizeYUV(width, height, subsamp); if(setCompDefaults(cinfo, pixelFormat, subsamp, -1, flags)==-1) return -1; /* Execute only the parts of jpeg_start_compress() that we need. If we @@ -735,25 +900,26 @@ DLLEXPORT int DLLCALL tjEncodeYUV2(tjhandle handle, unsigned char *srcBuf, to write the file headers, which could overflow the output buffer if the YUV image were very small. */ if(cinfo->global_state!=CSTATE_START) - _throw("tjEncodeYUV3(): libjpeg API is in the wrong state"); + _throw("tjEncodeYUVPlanes(): libjpeg API is in the wrong state"); (*cinfo->err->reset_error_mgr)((j_common_ptr)cinfo); jinit_c_master_control(cinfo, FALSE); jinit_color_converter(cinfo); jinit_downsampler(cinfo); (*cinfo->cconvert->start_pass)(cinfo); - pw=PAD(width, cinfo->max_h_samp_factor); - ph=PAD(height, cinfo->max_v_samp_factor); + pw0=PAD(width, cinfo->max_h_samp_factor); + ph0=PAD(height, cinfo->max_v_samp_factor); - if((row_pointer=(JSAMPROW *)malloc(sizeof(JSAMPROW)*ph))==NULL) - _throw("tjEncodeYUV2(): Memory allocation failure"); + if((row_pointer=(JSAMPROW *)malloc(sizeof(JSAMPROW)*ph0))==NULL) + _throw("tjEncodeYUVPlanes(): Memory allocation failure"); for(i=0; i<height; i++) { - if(flags&TJFLAG_BOTTOMUP) row_pointer[i]=&srcBuf[(height-i-1)*pitch]; - else row_pointer[i]=&srcBuf[i*pitch]; + if(flags&TJFLAG_BOTTOMUP) + row_pointer[i]=(JSAMPROW)&srcBuf[(height-i-1)*pitch]; + else row_pointer[i]=(JSAMPROW)&srcBuf[i*pitch]; } - if(height<ph) - for(i=height; i<ph; i++) row_pointer[i]=row_pointer[height-1]; + if(height<ph0) + for(i=height; i<ph0; i++) row_pointer[i]=row_pointer[height-1]; for(i=0; i<cinfo->num_components; i++) { @@ -761,9 +927,9 @@ DLLEXPORT int DLLCALL tjEncodeYUV2(tjhandle handle, unsigned char *srcBuf, _tmpbuf[i]=(JSAMPLE *)malloc( PAD((compptr->width_in_blocks*cinfo->max_h_samp_factor*DCTSIZE) /compptr->h_samp_factor, 16) * cinfo->max_v_samp_factor + 16); - if(!_tmpbuf[i]) _throw("tjEncodeYUV2(): Memory allocation failure"); + if(!_tmpbuf[i]) _throw("tjEncodeYUVPlanes(): Memory allocation failure"); tmpbuf[i]=(JSAMPROW *)malloc(sizeof(JSAMPROW)*cinfo->max_v_samp_factor); - if(!tmpbuf[i]) _throw("tjEncodeYUV2(): Memory allocation failure"); + if(!tmpbuf[i]) _throw("tjEncodeYUVPlanes(): Memory allocation failure"); for(row=0; row<cinfo->max_v_samp_factor; row++) { unsigned char *_tmpbuf_aligned= @@ -774,9 +940,9 @@ DLLEXPORT int DLLCALL tjEncodeYUV2(tjhandle handle, unsigned char *srcBuf, } _tmpbuf2[i]=(JSAMPLE *)malloc(PAD(compptr->width_in_blocks*DCTSIZE, 16) * compptr->v_samp_factor + 16); - if(!_tmpbuf2[i]) _throw("tjEncodeYUV2(): Memory allocation failure"); + if(!_tmpbuf2[i]) _throw("tjEncodeYUVPlanes(): Memory allocation failure"); tmpbuf2[i]=(JSAMPROW *)malloc(sizeof(JSAMPROW)*compptr->v_samp_factor); - if(!tmpbuf2[i]) _throw("tjEncodeYUV2(): Memory allocation failure"); + if(!tmpbuf2[i]) _throw("tjEncodeYUVPlanes(): Memory allocation failure"); for(row=0; row<compptr->v_samp_factor; row++) { unsigned char *_tmpbuf2_aligned= @@ -784,20 +950,19 @@ DLLEXPORT int DLLCALL tjEncodeYUV2(tjhandle handle, unsigned char *srcBuf, tmpbuf2[i][row]=&_tmpbuf2_aligned[ PAD(compptr->width_in_blocks*DCTSIZE, 16) * row]; } - cw[i]=pw*compptr->h_samp_factor/cinfo->max_h_samp_factor; - ch[i]=ph*compptr->v_samp_factor/cinfo->max_v_samp_factor; - outbuf[i]=(JSAMPROW *)malloc(sizeof(JSAMPROW)*ch[i]); - if(!outbuf[i]) _throw("tjEncodeYUV2(): Memory allocation failure"); - for(row=0; row<ch[i]; row++) + pw[i]=pw0*compptr->h_samp_factor/cinfo->max_h_samp_factor; + ph[i]=ph0*compptr->v_samp_factor/cinfo->max_v_samp_factor; + outbuf[i]=(JSAMPROW *)malloc(sizeof(JSAMPROW)*ph[i]); + if(!outbuf[i]) _throw("tjEncodeYUVPlanes(): Memory allocation failure"); + ptr=dstPlanes[i]; + for(row=0; row<ph[i]; row++) { outbuf[i][row]=ptr; - ptr+=PAD(cw[i], 4); + ptr+=(strides && strides[i]!=0)? strides[i]:pw[i]; } } - if(yuvsize!=(unsigned long)(ptr-dstBuf)) - _throw("tjEncodeYUV2(): Generated image is not the correct size"); - for(row=0; row<ph; row+=cinfo->max_v_samp_factor) + for(row=0; row<ph0; row+=cinfo->max_v_samp_factor) { (*cinfo->cconvert->color_convert)(cinfo, &row_pointer[row], tmpbuf, 0, cinfo->max_v_samp_factor); @@ -805,7 +970,7 @@ DLLEXPORT int DLLCALL tjEncodeYUV2(tjhandle handle, unsigned char *srcBuf, for(i=0, compptr=cinfo->comp_info; i<cinfo->num_components; i++, compptr++) jcopy_sample_rows(tmpbuf2[i], 0, outbuf[i], row*compptr->v_samp_factor/cinfo->max_v_samp_factor, - compptr->v_samp_factor, cw[i]); + compptr->v_samp_factor, pw[i]); } cinfo->next_scanline+=height; jpeg_abort_compress(cinfo); @@ -824,9 +989,54 @@ DLLEXPORT int DLLCALL tjEncodeYUV2(tjhandle handle, unsigned char *srcBuf, if(_tmpbuf2[i]!=NULL) free(_tmpbuf2[i]); if(outbuf[i]!=NULL) free(outbuf[i]); } + if(this->jerr.warning) retval=-1; return retval; } +DLLEXPORT int DLLCALL tjEncodeYUV3(tjhandle handle, + const unsigned char *srcBuf, int width, int pitch, int height, + int pixelFormat, unsigned char *dstBuf, int pad, int subsamp, int flags) +{ + unsigned char *dstPlanes[3]; + int pw0, ph0, strides[3], retval=-1; + + if(width<=0 || height<=0 || dstBuf==NULL || pad<0 || !isPow2(pad) + || subsamp<0 || subsamp>=NUMSUBOPT) + _throw("tjEncodeYUV3(): Invalid argument"); + + pw0=tjPlaneWidth(0, width, subsamp); + ph0=tjPlaneHeight(0, height, subsamp); + dstPlanes[0]=dstBuf; + strides[0]=PAD(pw0, pad); + if(subsamp==TJSAMP_GRAY) + { + strides[1]=strides[2]=0; + dstPlanes[1]=dstPlanes[2]=NULL; + } + else + { + int pw1=tjPlaneWidth(1, width, subsamp); + int ph1=tjPlaneHeight(1, height, subsamp); + strides[1]=strides[2]=PAD(pw1, pad); + dstPlanes[1]=dstPlanes[0]+strides[0]*ph0; + dstPlanes[2]=dstPlanes[1]+strides[1]*ph1; + } + + return tjEncodeYUVPlanes(handle, srcBuf, width, pitch, height, pixelFormat, + dstPlanes, strides, subsamp, flags); + + bailout: + return retval; +} + +DLLEXPORT int DLLCALL tjEncodeYUV2(tjhandle handle, unsigned char *srcBuf, + int width, int pitch, int height, int pixelFormat, unsigned char *dstBuf, + int subsamp, int flags) +{ + return tjEncodeYUV3(handle, srcBuf, width, pitch, height, pixelFormat, + dstBuf, 4, subsamp, flags); +} + DLLEXPORT int DLLCALL tjEncodeYUV(tjhandle handle, unsigned char *srcBuf, int width, int pitch, int height, int pixelSize, unsigned char *dstBuf, int subsamp, int flags) @@ -836,16 +1046,187 @@ DLLEXPORT int DLLCALL tjEncodeYUV(tjhandle handle, unsigned char *srcBuf, } +DLLEXPORT int DLLCALL tjCompressFromYUVPlanes(tjhandle handle, + const unsigned char **srcPlanes, int width, const int *strides, int height, + int subsamp, unsigned char **jpegBuf, unsigned long *jpegSize, int jpegQual, + int flags) +{ + int i, row, retval=0, alloc=1; JSAMPROW *inbuf[MAX_COMPONENTS]; + int pw[MAX_COMPONENTS], ph[MAX_COMPONENTS], iw[MAX_COMPONENTS], + tmpbufsize=0, usetmpbuf=0, th[MAX_COMPONENTS]; + JSAMPLE *_tmpbuf=NULL, *ptr; JSAMPROW *tmpbuf[MAX_COMPONENTS]; + + getcinstance(handle) + + for(i=0; i<MAX_COMPONENTS; i++) + { + tmpbuf[i]=NULL; inbuf[i]=NULL; + } + + if((this->init&COMPRESS)==0) + _throw("tjCompressFromYUVPlanes(): Instance has not been initialized for compression"); + + if(!srcPlanes || !srcPlanes[0] || width<=0 || height<=0 || subsamp<0 + || subsamp>=NUMSUBOPT || jpegBuf==NULL || jpegSize==NULL || jpegQual<0 + || jpegQual>100) + _throw("tjCompressFromYUVPlanes(): Invalid argument"); + if(subsamp!=TJSAMP_GRAY && (!srcPlanes[1] || !srcPlanes[2])) + _throw("tjCompressFromYUVPlanes(): Invalid argument"); + + if(setjmp(this->jerr.setjmp_buffer)) + { + /* If we get here, the JPEG code has signaled an error. */ + retval=-1; + goto bailout; + } + + cinfo->image_width=width; + cinfo->image_height=height; + + if(flags&TJFLAG_FORCEMMX) putenv("JSIMD_FORCEMMX=1"); + else if(flags&TJFLAG_FORCESSE) putenv("JSIMD_FORCESSE=1"); + else if(flags&TJFLAG_FORCESSE2) putenv("JSIMD_FORCESSE2=1"); + + if(flags&TJFLAG_NOREALLOC) + { + alloc=0; *jpegSize=tjBufSize(width, height, subsamp); + } + jpeg_mem_dest_tj(cinfo, jpegBuf, jpegSize, alloc); + if(setCompDefaults(cinfo, TJPF_RGB, subsamp, jpegQual, flags)==-1) + return -1; + cinfo->raw_data_in=TRUE; + + jpeg_start_compress(cinfo, TRUE); + for(i=0; i<cinfo->num_components; i++) + { + jpeg_component_info *compptr=&cinfo->comp_info[i]; + int ih; + iw[i]=compptr->width_in_blocks*DCTSIZE; + ih=compptr->height_in_blocks*DCTSIZE; + pw[i]=PAD(cinfo->image_width, cinfo->max_h_samp_factor) + *compptr->h_samp_factor/cinfo->max_h_samp_factor; + ph[i]=PAD(cinfo->image_height, cinfo->max_v_samp_factor) + *compptr->v_samp_factor/cinfo->max_v_samp_factor; + if(iw[i]!=pw[i] || ih!=ph[i]) usetmpbuf=1; + th[i]=compptr->v_samp_factor*DCTSIZE; + tmpbufsize+=iw[i]*th[i]; + if((inbuf[i]=(JSAMPROW *)malloc(sizeof(JSAMPROW)*ph[i]))==NULL) + _throw("tjCompressFromYUVPlanes(): Memory allocation failure"); + ptr=(JSAMPLE *)srcPlanes[i]; + for(row=0; row<ph[i]; row++) + { + inbuf[i][row]=ptr; + ptr+=(strides && strides[i]!=0)? strides[i]:pw[i]; + } + } + if(usetmpbuf) + { + if((_tmpbuf=(JSAMPLE *)malloc(sizeof(JSAMPLE)*tmpbufsize))==NULL) + _throw("tjCompressFromYUVPlanes(): Memory allocation failure"); + ptr=_tmpbuf; + for(i=0; i<cinfo->num_components; i++) + { + if((tmpbuf[i]=(JSAMPROW *)malloc(sizeof(JSAMPROW)*th[i]))==NULL) + _throw("tjCompressFromYUVPlanes(): Memory allocation failure"); + for(row=0; row<th[i]; row++) + { + tmpbuf[i][row]=ptr; + ptr+=iw[i]; + } + } + } + + for(row=0; row<(int)cinfo->image_height; + row+=cinfo->max_v_samp_factor*DCTSIZE) + { + JSAMPARRAY yuvptr[MAX_COMPONENTS]; + int crow[MAX_COMPONENTS]; + for(i=0; i<cinfo->num_components; i++) + { + jpeg_component_info *compptr=&cinfo->comp_info[i]; + crow[i]=row*compptr->v_samp_factor/cinfo->max_v_samp_factor; + if(usetmpbuf) + { + int j, k; + for(j=0; j<min(th[i], ph[i]-crow[i]); j++) + { + memcpy(tmpbuf[i][j], inbuf[i][crow[i]+j], pw[i]); + /* Duplicate last sample in row to fill out MCU */ + for(k=pw[i]; k<iw[i]; k++) tmpbuf[i][j][k]=tmpbuf[i][j][pw[i]-1]; + } + /* Duplicate last row to fill out MCU */ + for(j=ph[i]-crow[i]; j<th[i]; j++) + memcpy(tmpbuf[i][j], tmpbuf[i][ph[i]-crow[i]-1], iw[i]); + yuvptr[i]=tmpbuf[i]; + } + else + yuvptr[i]=&inbuf[i][crow[i]]; + } + jpeg_write_raw_data(cinfo, yuvptr, cinfo->max_v_samp_factor*DCTSIZE); + } + jpeg_finish_compress(cinfo); + + bailout: + if(cinfo->global_state>CSTATE_START) jpeg_abort_compress(cinfo); + for(i=0; i<MAX_COMPONENTS; i++) + { + if(tmpbuf[i]) free(tmpbuf[i]); + if(inbuf[i]) free(inbuf[i]); + } + if(_tmpbuf) free(_tmpbuf); + if(this->jerr.warning) retval=-1; + return retval; +} + +DLLEXPORT int DLLCALL tjCompressFromYUV(tjhandle handle, + const unsigned char *srcBuf, int width, int pad, int height, int subsamp, + unsigned char **jpegBuf, unsigned long *jpegSize, int jpegQual, int flags) +{ + const unsigned char *srcPlanes[3]; + int pw0, ph0, strides[3], retval=-1; + + if(srcBuf==NULL || width<=0 || pad<1 || height<=0 || subsamp<0 + || subsamp>=NUMSUBOPT) + _throw("tjCompressFromYUV(): Invalid argument"); + + pw0=tjPlaneWidth(0, width, subsamp); + ph0=tjPlaneHeight(0, height, subsamp); + srcPlanes[0]=srcBuf; + strides[0]=PAD(pw0, pad); + if(subsamp==TJSAMP_GRAY) + { + strides[1]=strides[2]=0; + srcPlanes[1]=srcPlanes[2]=NULL; + } + else + { + int pw1=tjPlaneWidth(1, width, subsamp); + int ph1=tjPlaneHeight(1, height, subsamp); + strides[1]=strides[2]=PAD(pw1, pad); + srcPlanes[1]=srcPlanes[0]+strides[0]*ph0; + srcPlanes[2]=srcPlanes[1]+strides[1]*ph1; + } + + return tjCompressFromYUVPlanes(handle, srcPlanes, width, strides, height, + subsamp, jpegBuf, jpegSize, jpegQual, flags); + + bailout: + return retval; +} + + /* Decompressor */ static tjhandle _tjInitDecompress(tjinstance *this) { - unsigned char buffer[1]; + static unsigned char buffer[1]; /* This is also straight out of example.c */ this->dinfo.err=jpeg_std_error(&this->jerr.pub); this->jerr.pub.error_exit=my_error_exit; this->jerr.pub.output_message=my_output_message; + this->jerr.emit_message=this->jerr.pub.emit_message; + this->jerr.pub.emit_message=my_emit_message; if(setjmp(this->jerr.setjmp_buffer)) { @@ -875,19 +1256,19 @@ DLLEXPORT tjhandle DLLCALL tjInitDecompress(void) } -DLLEXPORT int DLLCALL tjDecompressHeader2(tjhandle handle, - unsigned char *jpegBuf, unsigned long jpegSize, int *width, int *height, - int *jpegSubsamp) +DLLEXPORT int DLLCALL tjDecompressHeader3(tjhandle handle, + const unsigned char *jpegBuf, unsigned long jpegSize, int *width, + int *height, int *jpegSubsamp, int *jpegColorspace) { int retval=0; - getinstance(handle); + getdinstance(handle); if((this->init&DECOMPRESS)==0) - _throw("tjDecompressHeader2(): Instance has not been initialized for decompression"); + _throw("tjDecompressHeader3(): Instance has not been initialized for decompression"); if(jpegBuf==NULL || jpegSize<=0 || width==NULL || height==NULL - || jpegSubsamp==NULL) - _throw("tjDecompressHeader2(): Invalid argument"); + || jpegSubsamp==NULL || jpegColorspace==NULL) + _throw("tjDecompressHeader3(): Invalid argument"); if(setjmp(this->jerr.setjmp_buffer)) { @@ -901,18 +1282,39 @@ DLLEXPORT int DLLCALL tjDecompressHeader2(tjhandle handle, *width=dinfo->image_width; *height=dinfo->image_height; *jpegSubsamp=getSubsamp(dinfo); + switch(dinfo->jpeg_color_space) + { + case JCS_GRAYSCALE: *jpegColorspace=TJCS_GRAY; break; + case JCS_RGB: *jpegColorspace=TJCS_RGB; break; + case JCS_YCbCr: *jpegColorspace=TJCS_YCbCr; break; + case JCS_CMYK: *jpegColorspace=TJCS_CMYK; break; + case JCS_YCCK: *jpegColorspace=TJCS_YCCK; break; + default: *jpegColorspace=-1; break; + } jpeg_abort_decompress(dinfo); if(*jpegSubsamp<0) - _throw("tjDecompressHeader2(): Could not determine subsampling type for JPEG image"); + _throw("tjDecompressHeader3(): Could not determine subsampling type for JPEG image"); + if(*jpegColorspace<0) + _throw("tjDecompressHeader3(): Could not determine colorspace of JPEG image"); if(*width<1 || *height<1) - _throw("tjDecompressHeader2(): Invalid data returned in header"); + _throw("tjDecompressHeader3(): Invalid data returned in header"); bailout: + if(this->jerr.warning) retval=-1; return retval; } +DLLEXPORT int DLLCALL tjDecompressHeader2(tjhandle handle, + unsigned char *jpegBuf, unsigned long jpegSize, int *width, int *height, + int *jpegSubsamp) +{ + int jpegColorspace; + return tjDecompressHeader3(handle, jpegBuf, jpegSize, width, height, + jpegSubsamp, &jpegColorspace); +} + DLLEXPORT int DLLCALL tjDecompressHeader(tjhandle handle, unsigned char *jpegBuf, unsigned long jpegSize, int *width, int *height) { @@ -936,9 +1338,9 @@ DLLEXPORT tjscalingfactor* DLLCALL tjGetScalingFactors(int *numscalingfactors) } -DLLEXPORT int DLLCALL tjDecompress2(tjhandle handle, unsigned char *jpegBuf, - unsigned long jpegSize, unsigned char *dstBuf, int width, int pitch, - int height, int pixelFormat, int flags) +DLLEXPORT int DLLCALL tjDecompress2(tjhandle handle, + const unsigned char *jpegBuf, unsigned long jpegSize, unsigned char *dstBuf, + int width, int pitch, int height, int pixelFormat, int flags) { int i, retval=0; JSAMPROW *row_pointer=NULL; int jpegwidth, jpegheight, scaledw, scaledh; @@ -947,7 +1349,7 @@ DLLEXPORT int DLLCALL tjDecompress2(tjhandle handle, unsigned char *jpegBuf, unsigned char *_dstBuf=NULL; int _pitch=0; #endif - getinstance(handle); + getdinstance(handle); if((this->init&DECOMPRESS)==0) _throw("tjDecompress2(): Instance has not been initialized for decompression"); @@ -985,7 +1387,7 @@ DLLEXPORT int DLLCALL tjDecompress2(tjhandle handle, unsigned char *jpegBuf, if(scaledw<=width && scaledh<=height) break; } - if(scaledw>width || scaledh>height) + if(i>=NUMSF) _throw("tjDecompress2(): Could not scale down to desired image dimensions"); width=scaledw; height=scaledh; dinfo->scale_num=sf[i].num; @@ -995,7 +1397,7 @@ DLLEXPORT int DLLCALL tjDecompress2(tjhandle handle, unsigned char *jpegBuf, if(pitch==0) pitch=dinfo->output_width*tjPixelSize[pixelFormat]; #ifndef JCS_EXTENSIONS - if(pixelFormat!=TJPF_GRAY && + if(pixelFormat!=TJPF_GRAY && pixelFormat!=TJPF_CMYK && (RGB_RED!=tjRedOffset[pixelFormat] || RGB_GREEN!=tjGreenOffset[pixelFormat] || RGB_BLUE!=tjBlueOffset[pixelFormat] || @@ -1034,6 +1436,7 @@ DLLEXPORT int DLLCALL tjDecompress2(tjhandle handle, unsigned char *jpegBuf, if(rgbBuf) free(rgbBuf); #endif if(row_pointer) free(row_pointer); + if(this->jerr.warning) retval=-1; return retval; } @@ -1049,16 +1452,270 @@ DLLEXPORT int DLLCALL tjDecompress(tjhandle handle, unsigned char *jpegBuf, } -DLLEXPORT int DLLCALL tjDecompressToYUV(tjhandle handle, - unsigned char *jpegBuf, unsigned long jpegSize, unsigned char *dstBuf, +static int setDecodeDefaults(struct jpeg_decompress_struct *dinfo, + int pixelFormat, int subsamp, int flags) +{ + int i; + + dinfo->scale_num=dinfo->scale_denom=1; + + if(subsamp==TJSAMP_GRAY) + { + dinfo->num_components=dinfo->comps_in_scan=1; + dinfo->jpeg_color_space=JCS_GRAYSCALE; + } + else + { + dinfo->num_components=dinfo->comps_in_scan=3; + dinfo->jpeg_color_space=JCS_YCbCr; + } + + dinfo->comp_info=(jpeg_component_info *) + (*dinfo->mem->alloc_small)((j_common_ptr)dinfo, JPOOL_IMAGE, + dinfo->num_components*sizeof(jpeg_component_info)); + + for(i=0; i<dinfo->num_components; i++) + { + jpeg_component_info *compptr=&dinfo->comp_info[i]; + compptr->h_samp_factor=(i==0)? tjMCUWidth[subsamp]/8:1; + compptr->v_samp_factor=(i==0)? tjMCUHeight[subsamp]/8:1; + compptr->component_index=i; + compptr->component_id=i+1; + compptr->quant_tbl_no=compptr->dc_tbl_no=compptr->ac_tbl_no= + (i==0)? 0:1; + dinfo->cur_comp_info[i]=compptr; + } + dinfo->data_precision=8; + for(i=0; i<2; i++) + { + if(dinfo->quant_tbl_ptrs[i]==NULL) + dinfo->quant_tbl_ptrs[i]=jpeg_alloc_quant_table((j_common_ptr)dinfo); + } + + return 0; +} + + +int my_read_markers(j_decompress_ptr dinfo) +{ + return JPEG_REACHED_SOS; +} + +void my_reset_marker_reader(j_decompress_ptr dinfo) +{ +} + +DLLEXPORT int DLLCALL tjDecodeYUVPlanes(tjhandle handle, + const unsigned char **srcPlanes, const int *strides, int subsamp, + unsigned char *dstBuf, int width, int pitch, int height, int pixelFormat, int flags) { - int i, row, retval=0; JSAMPROW *outbuf[MAX_COMPONENTS]; - int cw[MAX_COMPONENTS], ch[MAX_COMPONENTS], iw[MAX_COMPONENTS], + int i, retval=0; JSAMPROW *row_pointer=NULL; + JSAMPLE *_tmpbuf[MAX_COMPONENTS]; + JSAMPROW *tmpbuf[MAX_COMPONENTS], *inbuf[MAX_COMPONENTS]; + int row, pw0, ph0, pw[MAX_COMPONENTS], ph[MAX_COMPONENTS]; + JSAMPLE *ptr; + jpeg_component_info *compptr; + #ifndef JCS_EXTENSIONS + unsigned char *rgbBuf=NULL; + unsigned char *_dstBuf=NULL; int _pitch=0; + #endif + int (*old_read_markers)(j_decompress_ptr); + void (*old_reset_marker_reader)(j_decompress_ptr); + + getdinstance(handle); + + for(i=0; i<MAX_COMPONENTS; i++) + { + tmpbuf[i]=NULL; _tmpbuf[i]=NULL; inbuf[i]=NULL; + } + + if((this->init&DECOMPRESS)==0) + _throw("tjDecodeYUVPlanes(): Instance has not been initialized for decompression"); + + if(!srcPlanes || !srcPlanes[0] || subsamp<0 || subsamp>=NUMSUBOPT + || dstBuf==NULL || width<=0 || pitch<0 || height<=0 || pixelFormat<0 + || pixelFormat>=TJ_NUMPF) + _throw("tjDecodeYUVPlanes(): Invalid argument"); + if(subsamp!=TJSAMP_GRAY && (!srcPlanes[1] || !srcPlanes[2])) + _throw("tjDecodeYUVPlanes(): Invalid argument"); + + if(setjmp(this->jerr.setjmp_buffer)) + { + /* If we get here, the JPEG code has signaled an error. */ + retval=-1; + goto bailout; + } + + if(pixelFormat==TJPF_CMYK) + _throw("tjDecodeYUVPlanes(): Cannot decode YUV images into CMYK pixels."); + + if(pitch==0) pitch=width*tjPixelSize[pixelFormat]; + dinfo->image_width=width; + dinfo->image_height=height; + + if(flags&TJFLAG_FORCEMMX) putenv("JSIMD_FORCEMMX=1"); + else if(flags&TJFLAG_FORCESSE) putenv("JSIMD_FORCESSE=1"); + else if(flags&TJFLAG_FORCESSE2) putenv("JSIMD_FORCESSE2=1"); + + if(setDecodeDefaults(dinfo, pixelFormat, subsamp, flags)==-1) + { + retval=-1; goto bailout; + } + old_read_markers=dinfo->marker->read_markers; + dinfo->marker->read_markers=my_read_markers; + old_reset_marker_reader=dinfo->marker->reset_marker_reader; + dinfo->marker->reset_marker_reader=my_reset_marker_reader; + jpeg_read_header(dinfo, TRUE); + dinfo->marker->read_markers=old_read_markers; + dinfo->marker->reset_marker_reader=old_reset_marker_reader; + + if(setDecompDefaults(dinfo, pixelFormat, flags)==-1) + { + retval=-1; goto bailout; + } + dinfo->do_fancy_upsampling=FALSE; + dinfo->Se=DCTSIZE2-1; + jinit_master_decompress(dinfo); + (*dinfo->upsample->start_pass)(dinfo); + + pw0=PAD(width, dinfo->max_h_samp_factor); + ph0=PAD(height, dinfo->max_v_samp_factor); + + if(pitch==0) pitch=dinfo->output_width*tjPixelSize[pixelFormat]; + + #ifndef JCS_EXTENSIONS + if(pixelFormat!=TJPF_GRAY && pixelFormat!=TJPF_CMYK && + (RGB_RED!=tjRedOffset[pixelFormat] || + RGB_GREEN!=tjGreenOffset[pixelFormat] || + RGB_BLUE!=tjBlueOffset[pixelFormat] || + RGB_PIXELSIZE!=tjPixelSize[pixelFormat])) + { + rgbBuf=(unsigned char *)malloc(width*height*3); + if(!rgbBuf) _throw("tjDecodeYUVPlanes(): Memory allocation failure"); + _pitch=pitch; pitch=width*3; + _dstBuf=dstBuf; dstBuf=rgbBuf; + } + #endif + + if((row_pointer=(JSAMPROW *)malloc(sizeof(JSAMPROW)*ph0))==NULL) + _throw("tjDecodeYUVPlanes(): Memory allocation failure"); + for(i=0; i<height; i++) + { + if(flags&TJFLAG_BOTTOMUP) row_pointer[i]=&dstBuf[(height-i-1)*pitch]; + else row_pointer[i]=&dstBuf[i*pitch]; + } + if(height<ph0) + for(i=height; i<ph0; i++) row_pointer[i]=row_pointer[height-1]; + + for(i=0; i<dinfo->num_components; i++) + { + compptr=&dinfo->comp_info[i]; + _tmpbuf[i]=(JSAMPLE *)malloc(PAD(compptr->width_in_blocks*DCTSIZE, 16) + * compptr->v_samp_factor + 16); + if(!_tmpbuf[i]) _throw("tjDecodeYUVPlanes(): Memory allocation failure"); + tmpbuf[i]=(JSAMPROW *)malloc(sizeof(JSAMPROW)*compptr->v_samp_factor); + if(!tmpbuf[i]) _throw("tjDecodeYUVPlanes(): Memory allocation failure"); + for(row=0; row<compptr->v_samp_factor; row++) + { + unsigned char *_tmpbuf_aligned= + (unsigned char *)PAD((size_t)_tmpbuf[i], 16); + tmpbuf[i][row]=&_tmpbuf_aligned[ + PAD(compptr->width_in_blocks*DCTSIZE, 16) * row]; + } + pw[i]=pw0*compptr->h_samp_factor/dinfo->max_h_samp_factor; + ph[i]=ph0*compptr->v_samp_factor/dinfo->max_v_samp_factor; + inbuf[i]=(JSAMPROW *)malloc(sizeof(JSAMPROW)*ph[i]); + if(!inbuf[i]) _throw("tjDecodeYUVPlanes(): Memory allocation failure"); + ptr=(JSAMPLE *)srcPlanes[i]; + for(row=0; row<ph[i]; row++) + { + inbuf[i][row]=ptr; + ptr+=(strides && strides[i]!=0)? strides[i]:pw[i]; + } + } + + for(row=0; row<ph0; row+=dinfo->max_v_samp_factor) + { + JDIMENSION inrow=0, outrow=0; + for(i=0, compptr=dinfo->comp_info; i<dinfo->num_components; i++, compptr++) + jcopy_sample_rows(inbuf[i], + row*compptr->v_samp_factor/dinfo->max_v_samp_factor, tmpbuf[i], 0, + compptr->v_samp_factor, pw[i]); + (dinfo->upsample->upsample)(dinfo, tmpbuf, &inrow, + dinfo->max_v_samp_factor, &row_pointer[row], &outrow, + dinfo->max_v_samp_factor); + } + jpeg_abort_decompress(dinfo); + + #ifndef JCS_EXTENSIONS + fromRGB(rgbBuf, _dstBuf, width, _pitch, height, pixelFormat); + #endif + + bailout: + if(dinfo->global_state>DSTATE_START) jpeg_abort_decompress(dinfo); + #ifndef JCS_EXTENSIONS + if(rgbBuf) free(rgbBuf); + #endif + if(row_pointer) free(row_pointer); + for(i=0; i<MAX_COMPONENTS; i++) + { + if(tmpbuf[i]!=NULL) free(tmpbuf[i]); + if(_tmpbuf[i]!=NULL) free(_tmpbuf[i]); + if(inbuf[i]!=NULL) free(inbuf[i]); + } + if(this->jerr.warning) retval=-1; + return retval; +} + +DLLEXPORT int DLLCALL tjDecodeYUV(tjhandle handle, const unsigned char *srcBuf, + int pad, int subsamp, unsigned char *dstBuf, int width, int pitch, + int height, int pixelFormat, int flags) +{ + const unsigned char *srcPlanes[3]; + int pw0, ph0, strides[3], retval=-1; + + if(srcBuf==NULL || pad<0 || !isPow2(pad) || subsamp<0 || subsamp>=NUMSUBOPT + || width<=0 || height<=0) + _throw("tjDecodeYUV(): Invalid argument"); + + pw0=tjPlaneWidth(0, width, subsamp); + ph0=tjPlaneHeight(0, height, subsamp); + srcPlanes[0]=srcBuf; + strides[0]=PAD(pw0, pad); + if(subsamp==TJSAMP_GRAY) + { + strides[1]=strides[2]=0; + srcPlanes[1]=srcPlanes[2]=NULL; + } + else + { + int pw1=tjPlaneWidth(1, width, subsamp); + int ph1=tjPlaneHeight(1, height, subsamp); + strides[1]=strides[2]=PAD(pw1, pad); + srcPlanes[1]=srcPlanes[0]+strides[0]*ph0; + srcPlanes[2]=srcPlanes[1]+strides[1]*ph1; + } + + return tjDecodeYUVPlanes(handle, srcPlanes, strides, subsamp, dstBuf, width, + pitch, height, pixelFormat, flags); + + bailout: + return retval; +} + +DLLEXPORT int DLLCALL tjDecompressToYUVPlanes(tjhandle handle, + const unsigned char *jpegBuf, unsigned long jpegSize, + unsigned char **dstPlanes, int width, int *strides, int height, int flags) +{ + int i, sfi, row, retval=0; JSAMPROW *outbuf[MAX_COMPONENTS]; + int jpegwidth, jpegheight, jpegSubsamp, scaledw, scaledh; + int pw[MAX_COMPONENTS], ph[MAX_COMPONENTS], iw[MAX_COMPONENTS], tmpbufsize=0, usetmpbuf=0, th[MAX_COMPONENTS]; - JSAMPLE *_tmpbuf=NULL, *ptr=dstBuf; JSAMPROW *tmpbuf[MAX_COMPONENTS]; + JSAMPLE *_tmpbuf=NULL, *ptr; JSAMPROW *tmpbuf[MAX_COMPONENTS]; + int dctsize; - getinstance(handle); + getdinstance(handle); for(i=0; i<MAX_COMPONENTS; i++) { @@ -1066,10 +1723,11 @@ DLLEXPORT int DLLCALL tjDecompressToYUV(tjhandle handle, } if((this->init&DECOMPRESS)==0) - _throw("tjDecompressToYUV(): Instance has not been initialized for decompression"); + _throw("tjDecompressToYUVPlanes(): Instance has not been initialized for decompression"); - if(jpegBuf==NULL || jpegSize<=0 || dstBuf==NULL) - _throw("tjDecompressToYUV(): Invalid argument"); + if(jpegBuf==NULL || jpegSize<=0 || !dstPlanes || !dstPlanes[0] || width<0 + || height<0) + _throw("tjDecompressToYUVPlanes(): Invalid argument"); if(flags&TJFLAG_FORCEMMX) putenv("JSIMD_FORCEMMX=1"); else if(flags&TJFLAG_FORCESSE) putenv("JSIMD_FORCESSE=1"); @@ -1082,39 +1740,73 @@ DLLEXPORT int DLLCALL tjDecompressToYUV(tjhandle handle, goto bailout; } - jpeg_mem_src_tj(dinfo, jpegBuf, jpegSize); - jpeg_read_header(dinfo, TRUE); + if(!this->headerRead) + { + jpeg_mem_src_tj(dinfo, jpegBuf, jpegSize); + jpeg_read_header(dinfo, TRUE); + } + this->headerRead=0; + jpegSubsamp=getSubsamp(dinfo); + if(jpegSubsamp<0) + _throw("tjDecompressToYUVPlanes(): Could not determine subsampling type for JPEG image"); + + if(jpegSubsamp!=TJSAMP_GRAY && (!dstPlanes[1] || !dstPlanes[2])) + _throw("tjDecompressToYUVPlanes(): Invalid argument"); + + jpegwidth=dinfo->image_width; jpegheight=dinfo->image_height; + if(width==0) width=jpegwidth; + if(height==0) height=jpegheight; + for(i=0; i<NUMSF; i++) + { + scaledw=TJSCALED(jpegwidth, sf[i]); + scaledh=TJSCALED(jpegheight, sf[i]); + if(scaledw<=width && scaledh<=height) + break; + } + if(i>=NUMSF) + _throw("tjDecompressToYUVPlanes(): Could not scale down to desired image dimensions"); + if(dinfo->num_components>3) + _throw("tjDecompressToYUVPlanes(): JPEG image must have 3 or fewer components"); + + width=scaledw; height=scaledh; + dinfo->scale_num=sf[i].num; + dinfo->scale_denom=sf[i].denom; + sfi=i; + jpeg_calc_output_dimensions(dinfo); + + dctsize=DCTSIZE*sf[sfi].num/sf[sfi].denom; for(i=0; i<dinfo->num_components; i++) { jpeg_component_info *compptr=&dinfo->comp_info[i]; int ih; - iw[i]=compptr->width_in_blocks*DCTSIZE; - ih=compptr->height_in_blocks*DCTSIZE; - cw[i]=PAD(dinfo->image_width, dinfo->max_h_samp_factor) + iw[i]=compptr->width_in_blocks*dctsize; + ih=compptr->height_in_blocks*dctsize; + pw[i]=PAD(dinfo->output_width, dinfo->max_h_samp_factor) *compptr->h_samp_factor/dinfo->max_h_samp_factor; - ch[i]=PAD(dinfo->image_height, dinfo->max_v_samp_factor) + ph[i]=PAD(dinfo->output_height, dinfo->max_v_samp_factor) *compptr->v_samp_factor/dinfo->max_v_samp_factor; - if(iw[i]!=cw[i] || ih!=ch[i]) usetmpbuf=1; - th[i]=compptr->v_samp_factor*DCTSIZE; + if(iw[i]!=pw[i] || ih!=ph[i]) usetmpbuf=1; + th[i]=compptr->v_samp_factor*dctsize; tmpbufsize+=iw[i]*th[i]; - if((outbuf[i]=(JSAMPROW *)malloc(sizeof(JSAMPROW)*ch[i]))==NULL) - _throw("tjDecompressToYUV(): Memory allocation failure"); - for(row=0; row<ch[i]; row++) + if((outbuf[i]=(JSAMPROW *)malloc(sizeof(JSAMPROW)*ph[i]))==NULL) + _throw("tjDecompressToYUVPlanes(): Memory allocation failure"); + ptr=dstPlanes[i]; + for(row=0; row<ph[i]; row++) { outbuf[i][row]=ptr; - ptr+=PAD(cw[i], 4); + ptr+=(strides && strides[i]!=0)? strides[i]:pw[i]; } } if(usetmpbuf) { if((_tmpbuf=(JSAMPLE *)malloc(sizeof(JSAMPLE)*tmpbufsize))==NULL) - _throw("tjDecompressToYUV(): Memory allocation failure"); + _throw("tjDecompressToYUVPlanes(): Memory allocation failure"); ptr=_tmpbuf; for(i=0; i<dinfo->num_components; i++) { if((tmpbuf[i]=(JSAMPROW *)malloc(sizeof(JSAMPROW)*th[i]))==NULL) - _throw("tjDecompressToYUV(): Memory allocation failure"); + _throw("tjDecompressToYUVPlanes(): Memory allocation failure"); for(row=0; row<th[i]; row++) { tmpbuf[i][row]=ptr; @@ -1129,26 +1821,45 @@ DLLEXPORT int DLLCALL tjDecompressToYUV(tjhandle handle, jpeg_start_decompress(dinfo); for(row=0; row<(int)dinfo->output_height; - row+=dinfo->max_v_samp_factor*DCTSIZE) + row+=dinfo->max_v_samp_factor*dinfo->_min_DCT_scaled_size) { JSAMPARRAY yuvptr[MAX_COMPONENTS]; int crow[MAX_COMPONENTS]; for(i=0; i<dinfo->num_components; i++) { jpeg_component_info *compptr=&dinfo->comp_info[i]; + if(jpegSubsamp==TJ_420) + { + /* When 4:2:0 subsampling is used with IDCT scaling, libjpeg will try + to be clever and use the IDCT to perform upsampling on the U and V + planes. For instance, if the output image is to be scaled by 1/2 + relative to the JPEG image, then the scaling factor and upsampling + effectively cancel each other, so a normal 8x8 IDCT can be used. + However, this is not desirable when using the decompress-to-YUV + functionality in TurboJPEG, since we want to output the U and V + planes in their subsampled form. Thus, we have to override some + internal libjpeg parameters to force it to use the "scaled" IDCT + functions on the U and V planes. */ + compptr->_DCT_scaled_size=dctsize; + compptr->MCU_sample_width=tjMCUWidth[jpegSubsamp]* + sf[sfi].num/sf[sfi].denom* + compptr->v_samp_factor/dinfo->max_v_samp_factor; + dinfo->idct->inverse_DCT[i] = dinfo->idct->inverse_DCT[0]; + } crow[i]=row*compptr->v_samp_factor/dinfo->max_v_samp_factor; if(usetmpbuf) yuvptr[i]=tmpbuf[i]; else yuvptr[i]=&outbuf[i][crow[i]]; } - jpeg_read_raw_data(dinfo, yuvptr, dinfo->max_v_samp_factor*DCTSIZE); + jpeg_read_raw_data(dinfo, yuvptr, + dinfo->max_v_samp_factor*dinfo->_min_DCT_scaled_size); if(usetmpbuf) { int j; for(i=0; i<dinfo->num_components; i++) { - for(j=0; j<min(th[i], ch[i]-crow[i]); j++) + for(j=0; j<min(th[i], ph[i]-crow[i]); j++) { - memcpy(outbuf[i][crow[i]+j], tmpbuf[i][j], cw[i]); + memcpy(outbuf[i][crow[i]+j], tmpbuf[i][j], pw[i]); } } } @@ -1163,7 +1874,76 @@ DLLEXPORT int DLLCALL tjDecompressToYUV(tjhandle handle, if(outbuf[i]) free(outbuf[i]); } if(_tmpbuf) free(_tmpbuf); + if(this->jerr.warning) retval=-1; + return retval; +} + +DLLEXPORT int DLLCALL tjDecompressToYUV2(tjhandle handle, + const unsigned char *jpegBuf, unsigned long jpegSize, unsigned char *dstBuf, + int width, int pad, int height, int flags) +{ + unsigned char *dstPlanes[3]; + int pw0, ph0, strides[3], retval=-1, jpegSubsamp=-1; + int i, jpegwidth, jpegheight, scaledw, scaledh; + + getdinstance(handle); + + if(jpegBuf==NULL || jpegSize<=0 || dstBuf==NULL || width<0 || pad<1 + || !isPow2(pad) || height<0) + _throw("tjDecompressToYUV2(): Invalid argument"); + + jpeg_mem_src_tj(dinfo, jpegBuf, jpegSize); + jpeg_read_header(dinfo, TRUE); + jpegSubsamp=getSubsamp(dinfo); + if(jpegSubsamp<0) + _throw("tjDecompressToYUV2(): Could not determine subsampling type for JPEG image"); + + jpegwidth=dinfo->image_width; jpegheight=dinfo->image_height; + if(width==0) width=jpegwidth; + if(height==0) height=jpegheight; + + for(i=0; i<NUMSF; i++) + { + scaledw=TJSCALED(jpegwidth, sf[i]); + scaledh=TJSCALED(jpegheight, sf[i]); + if(scaledw<=width && scaledh<=height) + break; + } + if(i>=NUMSF) + _throw("tjDecompressToYUV2(): Could not scale down to desired image dimensions"); + + pw0=tjPlaneWidth(0, width, jpegSubsamp); + ph0=tjPlaneHeight(0, height, jpegSubsamp); + dstPlanes[0]=dstBuf; + strides[0]=PAD(pw0, pad); + if(jpegSubsamp==TJSAMP_GRAY) + { + strides[1]=strides[2]=0; + dstPlanes[1]=dstPlanes[2]=NULL; + } + else + { + int pw1=tjPlaneWidth(1, width, jpegSubsamp); + int ph1=tjPlaneHeight(1, height, jpegSubsamp); + strides[1]=strides[2]=PAD(pw1, pad); + dstPlanes[1]=dstPlanes[0]+strides[0]*ph0; + dstPlanes[2]=dstPlanes[1]+strides[1]*ph1; + } + + this->headerRead=1; + return tjDecompressToYUVPlanes(handle, jpegBuf, jpegSize, dstPlanes, width, + strides, height, flags); + + bailout: return retval; + +} + +DLLEXPORT int DLLCALL tjDecompressToYUV(tjhandle handle, + unsigned char *jpegBuf, unsigned long jpegSize, unsigned char *dstBuf, + int flags) +{ + return tjDecompressToYUV2(handle, jpegBuf, jpegSize, dstBuf, 0, 4, 0, flags); } @@ -1186,9 +1966,9 @@ DLLEXPORT tjhandle DLLCALL tjInitTransform(void) } -DLLEXPORT int DLLCALL tjTransform(tjhandle handle, unsigned char *jpegBuf, - unsigned long jpegSize, int n, unsigned char **dstBufs, - unsigned long *dstSizes, tjtransform *t, int flags) +DLLEXPORT int DLLCALL tjTransform(tjhandle handle, + const unsigned char *jpegBuf, unsigned long jpegSize, int n, + unsigned char **dstBufs, unsigned long *dstSizes, tjtransform *t, int flags) { jpeg_transform_info *xinfo=NULL; jvirt_barray_ptr *srccoefs, *dstcoefs; @@ -1336,5 +2116,6 @@ DLLEXPORT int DLLCALL tjTransform(tjhandle handle, unsigned char *jpegBuf, if(cinfo->global_state>CSTATE_START) jpeg_abort_compress(cinfo); if(dinfo->global_state>DSTATE_START) jpeg_abort_decompress(dinfo); if(xinfo) free(xinfo); + if(this->jerr.warning) retval=-1; return retval; } |