diff options
Diffstat (limited to 'turbojpeg-jni.c')
-rw-r--r-- | turbojpeg-jni.c | 737 |
1 files changed, 737 insertions, 0 deletions
diff --git a/turbojpeg-jni.c b/turbojpeg-jni.c new file mode 100644 index 0000000..634bedf --- /dev/null +++ b/turbojpeg-jni.c @@ -0,0 +1,737 @@ +/* + * Copyright (C)2011-2013 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: + * + * - Redistributions of source code must retain the above copyright notice, + * this list of conditions and the following disclaimer. + * - Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following disclaimer in the documentation + * and/or other materials provided with the distribution. + * - Neither the name of the libjpeg-turbo Project nor the names of its + * contributors may be used to endorse or promote products derived from this + * software without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS", + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE + * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN + * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE + * POSSIBILITY OF SUCH DAMAGE. + */ + +#include <stdlib.h> +#include <string.h> +#include "turbojpeg.h" +#ifdef WIN32 +#include "tjutil.h" +#endif +#include <jni.h> +#include "java/org_libjpegturbo_turbojpeg_TJCompressor.h" +#include "java/org_libjpegturbo_turbojpeg_TJDecompressor.h" +#include "java/org_libjpegturbo_turbojpeg_TJ.h" + +#define _throw(msg) { \ + jclass _exccls=(*env)->FindClass(env, "java/lang/Exception"); \ + if(!_exccls) goto bailout; \ + (*env)->ThrowNew(env, _exccls, msg); \ + goto bailout; \ +} + +#define bailif0(f) {if(!(f)) { \ + char temps[80]; \ + snprintf(temps, 80, "Unexpected NULL condition in line %d", __LINE__); \ + _throw(temps); \ +}} + +#define gethandle() \ + jclass _cls=(*env)->GetObjectClass(env, obj); \ + jfieldID _fid; \ + if(!_cls) goto bailout; \ + bailif0(_fid=(*env)->GetFieldID(env, _cls, "handle", "J")); \ + handle=(tjhandle)(jlong)(*env)->GetLongField(env, obj, _fid); \ + +JNIEXPORT jint JNICALL Java_org_libjpegturbo_turbojpeg_TJ_bufSize + (JNIEnv *env, jclass cls, jint width, jint height, jint jpegSubsamp) +{ + jint retval=(jint)tjBufSize(width, height, jpegSubsamp); + if(retval==-1) _throw(tjGetErrorStr()); + + bailout: + return retval; +} + +JNIEXPORT jint JNICALL Java_org_libjpegturbo_turbojpeg_TJ_bufSizeYUV + (JNIEnv *env, jclass cls, jint width, jint height, jint subsamp) +{ + jint retval=(jint)tjBufSizeYUV(width, height, subsamp); + if(retval==-1) _throw(tjGetErrorStr()); + + bailout: + return retval; +} + +JNIEXPORT void JNICALL Java_org_libjpegturbo_turbojpeg_TJCompressor_init + (JNIEnv *env, jobject obj) +{ + jclass cls; + jfieldID fid; + tjhandle handle; + + if((handle=tjInitCompress())==NULL) + _throw(tjGetErrorStr()); + + bailif0(cls=(*env)->GetObjectClass(env, obj)); + bailif0(fid=(*env)->GetFieldID(env, cls, "handle", "J")); + (*env)->SetLongField(env, obj, fid, (jlong)handle); + + bailout: + return; +} + +JNIEXPORT jint JNICALL Java_org_libjpegturbo_turbojpeg_TJCompressor_compress___3BIIIIII_3BIII + (JNIEnv *env, jobject obj, jbyteArray src, jint x, jint y, jint width, + jint pitch, jint height, jint pf, jbyteArray dst, jint jpegSubsamp, + jint jpegQual, jint flags) +{ + tjhandle handle=0; + unsigned long jpegSize=0; + jsize arraySize=0, actualPitch; + unsigned char *srcBuf=NULL, *jpegBuf=NULL; + + gethandle(); + + if(pf<0 || pf>=org_libjpegturbo_turbojpeg_TJ_NUMPF || width<1 || height<1 + || pitch<0) + _throw("Invalid argument in compress()"); + if(org_libjpegturbo_turbojpeg_TJ_NUMPF!=TJ_NUMPF) + _throw("Mismatch between Java and C API"); + + actualPitch=(pitch==0)? width*tjPixelSize[pf]:pitch; + arraySize=(y+height-1)*actualPitch + x+width; + if((*env)->GetArrayLength(env, src)<arraySize) + _throw("Source buffer is not large enough"); + jpegSize=tjBufSize(width, height, jpegSubsamp); + if((*env)->GetArrayLength(env, dst)<(jsize)jpegSize) + _throw("Destination buffer is not large enough"); + + bailif0(srcBuf=(*env)->GetPrimitiveArrayCritical(env, src, 0)); + bailif0(jpegBuf=(*env)->GetPrimitiveArrayCritical(env, dst, 0)); + + if(tjCompress2(handle, &srcBuf[y*actualPitch + x*tjPixelSize[pf]], width, + pitch, height, pf, &jpegBuf, &jpegSize, jpegSubsamp, jpegQual, + flags|TJFLAG_NOREALLOC)==-1) + { + (*env)->ReleasePrimitiveArrayCritical(env, dst, jpegBuf, 0); + (*env)->ReleasePrimitiveArrayCritical(env, src, srcBuf, 0); + jpegBuf=srcBuf=NULL; + _throw(tjGetErrorStr()); + } + + bailout: + if(jpegBuf) (*env)->ReleasePrimitiveArrayCritical(env, dst, jpegBuf, 0); + if(srcBuf) (*env)->ReleasePrimitiveArrayCritical(env, src, srcBuf, 0); + return (jint)jpegSize; +} + +JNIEXPORT jint JNICALL Java_org_libjpegturbo_turbojpeg_TJCompressor_compress___3BIIII_3BIII + (JNIEnv *env, jobject obj, jbyteArray src, jint width, jint pitch, + jint height, jint pf, jbyteArray dst, jint jpegSubsamp, jint jpegQual, + jint flags) +{ + return Java_org_libjpegturbo_turbojpeg_TJCompressor_compress___3BIIIIII_3BIII( + env, obj, src, 0, 0, width, pitch, height, pf, dst, jpegSubsamp, jpegQual, + flags); +} + +JNIEXPORT jint JNICALL Java_org_libjpegturbo_turbojpeg_TJCompressor_compress___3IIIIIII_3BIII + (JNIEnv *env, jobject obj, jintArray src, jint x, jint y, jint width, + jint stride, jint height, jint pf, jbyteArray dst, jint jpegSubsamp, + jint jpegQual, jint flags) +{ + tjhandle handle=0; + unsigned long jpegSize=0; + jsize arraySize=0, actualStride; + unsigned char *srcBuf=NULL, *jpegBuf=NULL; + + gethandle(); + + if(pf<0 || pf>=org_libjpegturbo_turbojpeg_TJ_NUMPF || width<1 || height<1 + || stride<0) + _throw("Invalid argument in compress()"); + if(org_libjpegturbo_turbojpeg_TJ_NUMPF!=TJ_NUMPF) + _throw("Mismatch between Java and C API"); + if(tjPixelSize[pf]!=sizeof(jint)) + _throw("Pixel format must be 32-bit when compressing from an integer buffer."); + + actualStride=(stride==0)? width:stride; + arraySize=(y+height-1)*actualStride + x+width; + if((*env)->GetArrayLength(env, src)<arraySize) + _throw("Source buffer is not large enough"); + jpegSize=tjBufSize(width, height, jpegSubsamp); + if((*env)->GetArrayLength(env, dst)<(jsize)jpegSize) + _throw("Destination buffer is not large enough"); + + bailif0(srcBuf=(*env)->GetPrimitiveArrayCritical(env, src, 0)); + bailif0(jpegBuf=(*env)->GetPrimitiveArrayCritical(env, dst, 0)); + + if(tjCompress2(handle, &srcBuf[(y*actualStride + x)*sizeof(int)], width, + stride*sizeof(jint), height, pf, &jpegBuf, &jpegSize, jpegSubsamp, + jpegQual, flags|TJFLAG_NOREALLOC)==-1) + { + (*env)->ReleasePrimitiveArrayCritical(env, dst, jpegBuf, 0); + (*env)->ReleasePrimitiveArrayCritical(env, src, srcBuf, 0); + jpegBuf=srcBuf=NULL; + _throw(tjGetErrorStr()); + } + + bailout: + if(jpegBuf) (*env)->ReleasePrimitiveArrayCritical(env, dst, jpegBuf, 0); + if(srcBuf) (*env)->ReleasePrimitiveArrayCritical(env, src, srcBuf, 0); + return (jint)jpegSize; +} + +JNIEXPORT jint JNICALL Java_org_libjpegturbo_turbojpeg_TJCompressor_compress___3IIIII_3BIII + (JNIEnv *env, jobject obj, jintArray src, jint width, jint pitch, + jint height, jint pf, jbyteArray dst, jint jpegSubsamp, jint jpegQual, + jint flags) +{ + return Java_org_libjpegturbo_turbojpeg_TJCompressor_compress___3IIIIIII_3BIII( + env, obj, src, 0, 0, width, pitch, height, pf, dst, jpegSubsamp, jpegQual, + flags); +} + +JNIEXPORT void JNICALL Java_org_libjpegturbo_turbojpeg_TJCompressor_encodeYUV___3BIIII_3BII + (JNIEnv *env, jobject obj, jbyteArray src, jint width, jint pitch, + jint height, jint pf, jbyteArray dst, jint subsamp, jint flags) +{ + tjhandle handle=0; + jsize arraySize=0; + unsigned char *srcBuf=NULL, *dstBuf=NULL; + + gethandle(); + + if(pf<0 || pf>=org_libjpegturbo_turbojpeg_TJ_NUMPF || width<1 || height<1 + || pitch<0) + _throw("Invalid argument in encodeYUV()"); + if(org_libjpegturbo_turbojpeg_TJ_NUMPF!=TJ_NUMPF) + _throw("Mismatch between Java and C API"); + + arraySize=(pitch==0)? width*tjPixelSize[pf]*height:pitch*height; + if((*env)->GetArrayLength(env, src)<arraySize) + _throw("Source buffer is not large enough"); + if((*env)->GetArrayLength(env, dst) + <(jsize)tjBufSizeYUV(width, height, subsamp)) + _throw("Destination buffer is not large enough"); + + bailif0(srcBuf=(*env)->GetPrimitiveArrayCritical(env, src, 0)); + bailif0(dstBuf=(*env)->GetPrimitiveArrayCritical(env, dst, 0)); + + if(tjEncodeYUV2(handle, srcBuf, width, pitch, height, pf, dstBuf, subsamp, + flags)==-1) + { + (*env)->ReleasePrimitiveArrayCritical(env, dst, dstBuf, 0); + (*env)->ReleasePrimitiveArrayCritical(env, src, srcBuf, 0); + dstBuf=srcBuf=NULL; + _throw(tjGetErrorStr()); + } + + bailout: + if(dstBuf) (*env)->ReleasePrimitiveArrayCritical(env, dst, dstBuf, 0); + if(srcBuf) (*env)->ReleasePrimitiveArrayCritical(env, src, srcBuf, 0); + return; +} + +JNIEXPORT void JNICALL Java_org_libjpegturbo_turbojpeg_TJCompressor_encodeYUV___3IIIII_3BII + (JNIEnv *env, jobject obj, jintArray src, jint width, jint stride, + jint height, jint pf, jbyteArray dst, jint subsamp, jint flags) +{ + tjhandle handle=0; + jsize arraySize=0; + unsigned char *srcBuf=NULL, *dstBuf=NULL; + + gethandle(); + + if(pf<0 || pf>=org_libjpegturbo_turbojpeg_TJ_NUMPF || width<1 || height<1 + || stride<0) + _throw("Invalid argument in encodeYUV()"); + if(org_libjpegturbo_turbojpeg_TJ_NUMPF!=TJ_NUMPF) + _throw("Mismatch between Java and C API"); + if(tjPixelSize[pf]!=sizeof(jint)) + _throw("Pixel format must be 32-bit when encoding from an integer buffer."); + + arraySize=(stride==0)? width*height:stride*height; + if((*env)->GetArrayLength(env, src)<arraySize) + _throw("Source buffer is not large enough"); + if((*env)->GetArrayLength(env, dst) + <(jsize)tjBufSizeYUV(width, height, subsamp)) + _throw("Destination buffer is not large enough"); + + bailif0(srcBuf=(*env)->GetPrimitiveArrayCritical(env, src, 0)); + bailif0(dstBuf=(*env)->GetPrimitiveArrayCritical(env, dst, 0)); + + if(tjEncodeYUV2(handle, srcBuf, width, stride*sizeof(jint), height, pf, + dstBuf, subsamp, flags)==-1) + { + (*env)->ReleasePrimitiveArrayCritical(env, dst, dstBuf, 0); + (*env)->ReleasePrimitiveArrayCritical(env, src, srcBuf, 0); + dstBuf=srcBuf=NULL; + _throw(tjGetErrorStr()); + } + + bailout: + if(dstBuf) (*env)->ReleasePrimitiveArrayCritical(env, dst, dstBuf, 0); + if(srcBuf) (*env)->ReleasePrimitiveArrayCritical(env, src, srcBuf, 0); + return; +} + +JNIEXPORT void JNICALL Java_org_libjpegturbo_turbojpeg_TJCompressor_destroy + (JNIEnv *env, jobject obj) +{ + tjhandle handle=0; + + gethandle(); + + if(tjDestroy(handle)==-1) _throw(tjGetErrorStr()); + (*env)->SetLongField(env, obj, _fid, 0); + + bailout: + return; +} + +JNIEXPORT void JNICALL Java_org_libjpegturbo_turbojpeg_TJDecompressor_init + (JNIEnv *env, jobject obj) +{ + jclass cls; + jfieldID fid; + tjhandle handle; + + if((handle=tjInitDecompress())==NULL) _throw(tjGetErrorStr()); + + bailif0(cls=(*env)->GetObjectClass(env, obj)); + bailif0(fid=(*env)->GetFieldID(env, cls, "handle", "J")); + (*env)->SetLongField(env, obj, fid, (jlong)handle); + + bailout: + return; +} + +JNIEXPORT jobjectArray JNICALL Java_org_libjpegturbo_turbojpeg_TJ_getScalingFactors + (JNIEnv *env, jclass cls) +{ + jclass sfcls=NULL; jfieldID fid=0; + tjscalingfactor *sf=NULL; int n=0, i; + jobject sfobj=NULL; + jobjectArray sfjava=NULL; + + if((sf=tjGetScalingFactors(&n))==NULL || n==0) + _throw(tjGetErrorStr()); + + bailif0(sfcls=(*env)->FindClass(env, "org/libjpegturbo/turbojpeg/TJScalingFactor")); + bailif0(sfjava=(jobjectArray)(*env)->NewObjectArray(env, n, sfcls, 0)); + + for(i=0; i<n; i++) + { + bailif0(sfobj=(*env)->AllocObject(env, sfcls)); + bailif0(fid=(*env)->GetFieldID(env, sfcls, "num", "I")); + (*env)->SetIntField(env, sfobj, fid, sf[i].num); + bailif0(fid=(*env)->GetFieldID(env, sfcls, "denom", "I")); + (*env)->SetIntField(env, sfobj, fid, sf[i].denom); + (*env)->SetObjectArrayElement(env, sfjava, i, sfobj); + } + + bailout: + return sfjava; +} + +JNIEXPORT void JNICALL Java_org_libjpegturbo_turbojpeg_TJDecompressor_decompressHeader + (JNIEnv *env, jobject obj, jbyteArray src, jint jpegSize) +{ + tjhandle handle=0; + unsigned char *jpegBuf=NULL; + int width=0, height=0, jpegSubsamp=-1; + + gethandle(); + + if((*env)->GetArrayLength(env, src)<jpegSize) + _throw("Source buffer is not large enough"); + + bailif0(jpegBuf=(*env)->GetPrimitiveArrayCritical(env, src, 0)); + + if(tjDecompressHeader2(handle, jpegBuf, (unsigned long)jpegSize, + &width, &height, &jpegSubsamp)==-1) + { + (*env)->ReleasePrimitiveArrayCritical(env, src, jpegBuf, 0); + _throw(tjGetErrorStr()); + } + (*env)->ReleasePrimitiveArrayCritical(env, src, jpegBuf, 0); jpegBuf=NULL; + + bailif0(_fid=(*env)->GetFieldID(env, _cls, "jpegSubsamp", "I")); + (*env)->SetIntField(env, obj, _fid, jpegSubsamp); + bailif0(_fid=(*env)->GetFieldID(env, _cls, "jpegWidth", "I")); + (*env)->SetIntField(env, obj, _fid, width); + bailif0(_fid=(*env)->GetFieldID(env, _cls, "jpegHeight", "I")); + (*env)->SetIntField(env, obj, _fid, height); + + bailout: + return; +} + +JNIEXPORT void JNICALL Java_org_libjpegturbo_turbojpeg_TJDecompressor_decompress___3BI_3BIIIIIII + (JNIEnv *env, jobject obj, jbyteArray src, jint jpegSize, jbyteArray dst, + jint x, jint y, jint width, jint pitch, jint height, jint pf, jint flags) +{ + tjhandle handle=0; + jsize arraySize=0, actualPitch; + unsigned char *jpegBuf=NULL, *dstBuf=NULL; + + gethandle(); + + if(pf<0 || pf>=org_libjpegturbo_turbojpeg_TJ_NUMPF) + _throw("Invalid argument in decompress()"); + if(org_libjpegturbo_turbojpeg_TJ_NUMPF!=TJ_NUMPF) + _throw("Mismatch between Java and C API"); + + if((*env)->GetArrayLength(env, src)<jpegSize) + _throw("Source buffer is not large enough"); + actualPitch=(pitch==0)? width*tjPixelSize[pf]:pitch; + arraySize=(y+height-1)*actualPitch + (x+width)*tjPixelSize[pf]; + if((*env)->GetArrayLength(env, dst)<arraySize) + _throw("Destination buffer is not large enough"); + + bailif0(jpegBuf=(*env)->GetPrimitiveArrayCritical(env, src, 0)); + bailif0(dstBuf=(*env)->GetPrimitiveArrayCritical(env, dst, 0)); + + if(tjDecompress2(handle, jpegBuf, (unsigned long)jpegSize, + &dstBuf[y*actualPitch + x*tjPixelSize[pf]], width, pitch, height, pf, + flags)==-1) + { + (*env)->ReleasePrimitiveArrayCritical(env, dst, dstBuf, 0); + (*env)->ReleasePrimitiveArrayCritical(env, src, jpegBuf, 0); + dstBuf=jpegBuf=NULL; + _throw(tjGetErrorStr()); + } + + bailout: + if(dstBuf) (*env)->ReleasePrimitiveArrayCritical(env, dst, dstBuf, 0); + if(jpegBuf) (*env)->ReleasePrimitiveArrayCritical(env, src, jpegBuf, 0); + return; +} + +JNIEXPORT void JNICALL Java_org_libjpegturbo_turbojpeg_TJDecompressor_decompress___3BI_3BIIIII + (JNIEnv *env, jobject obj, jbyteArray src, jint jpegSize, jbyteArray dst, + jint width, jint pitch, jint height, jint pf, jint flags) +{ + Java_org_libjpegturbo_turbojpeg_TJDecompressor_decompress___3BI_3BIIIIIII + (env, obj, src, jpegSize, dst, 0, 0, width, pitch, height, pf, flags); +} + +JNIEXPORT void JNICALL Java_org_libjpegturbo_turbojpeg_TJDecompressor_decompress___3BI_3IIIIIIII + (JNIEnv *env, jobject obj, jbyteArray src, jint jpegSize, jintArray dst, + jint x, jint y, jint width, jint stride, jint height, jint pf, jint flags) +{ + tjhandle handle=0; + jsize arraySize=0, actualStride; + unsigned char *jpegBuf=NULL, *dstBuf=NULL; + + gethandle(); + + if(pf<0 || pf>=org_libjpegturbo_turbojpeg_TJ_NUMPF) + _throw("Invalid argument in decompress()"); + if(org_libjpegturbo_turbojpeg_TJ_NUMPF!=TJ_NUMPF) + _throw("Mismatch between Java and C API"); + if(tjPixelSize[pf]!=sizeof(jint)) + _throw("Pixel format must be 32-bit when decompressing to an integer buffer."); + + if((*env)->GetArrayLength(env, src)<jpegSize) + _throw("Source buffer is not large enough"); + actualStride=(stride==0)? width:stride; + arraySize=(y+height-1)*actualStride + x+width; + if((*env)->GetArrayLength(env, dst)<arraySize) + _throw("Destination buffer is not large enough"); + + bailif0(jpegBuf=(*env)->GetPrimitiveArrayCritical(env, src, 0)); + bailif0(dstBuf=(*env)->GetPrimitiveArrayCritical(env, dst, 0)); + + if(tjDecompress2(handle, jpegBuf, (unsigned long)jpegSize, + &dstBuf[(y*actualStride + x)*sizeof(int)], width, stride*sizeof(jint), + height, pf, flags)==-1) + { + (*env)->ReleasePrimitiveArrayCritical(env, dst, dstBuf, 0); + (*env)->ReleasePrimitiveArrayCritical(env, src, jpegBuf, 0); + dstBuf=jpegBuf=NULL; + _throw(tjGetErrorStr()); + } + + bailout: + if(dstBuf) (*env)->ReleasePrimitiveArrayCritical(env, dst, dstBuf, 0); + if(jpegBuf) (*env)->ReleasePrimitiveArrayCritical(env, src, jpegBuf, 0); + return; +} + +JNIEXPORT void JNICALL Java_org_libjpegturbo_turbojpeg_TJDecompressor_decompress___3BI_3IIIIII + (JNIEnv *env, jobject obj, jbyteArray src, jint jpegSize, jintArray dst, + jint width, jint stride, jint height, jint pf, jint flags) +{ + Java_org_libjpegturbo_turbojpeg_TJDecompressor_decompress___3BI_3IIIIIIII + (env, obj, src, jpegSize, dst, 0, 0, width, stride, height, pf, flags); + +} + +JNIEXPORT void JNICALL Java_org_libjpegturbo_turbojpeg_TJDecompressor_decompressToYUV + (JNIEnv *env, jobject obj, jbyteArray src, jint jpegSize, jbyteArray dst, + jint flags) +{ + tjhandle handle=0; + unsigned char *jpegBuf=NULL, *dstBuf=NULL; + int jpegSubsamp=-1, jpegWidth=0, jpegHeight=0; + + gethandle(); + + if((*env)->GetArrayLength(env, src)<jpegSize) + _throw("Source buffer is not large enough"); + bailif0(_fid=(*env)->GetFieldID(env, _cls, "jpegSubsamp", "I")); + jpegSubsamp=(int)(*env)->GetIntField(env, obj, _fid); + bailif0(_fid=(*env)->GetFieldID(env, _cls, "jpegWidth", "I")); + jpegWidth=(int)(*env)->GetIntField(env, obj, _fid); + bailif0(_fid=(*env)->GetFieldID(env, _cls, "jpegHeight", "I")); + jpegHeight=(int)(*env)->GetIntField(env, obj, _fid); + if((*env)->GetArrayLength(env, dst) + <(jsize)tjBufSizeYUV(jpegWidth, jpegHeight, jpegSubsamp)) + _throw("Destination buffer is not large enough"); + + bailif0(jpegBuf=(*env)->GetPrimitiveArrayCritical(env, src, 0)); + bailif0(dstBuf=(*env)->GetPrimitiveArrayCritical(env, dst, 0)); + + if(tjDecompressToYUV(handle, jpegBuf, (unsigned long)jpegSize, dstBuf, + flags)==-1) + { + (*env)->ReleasePrimitiveArrayCritical(env, dst, dstBuf, 0); + (*env)->ReleasePrimitiveArrayCritical(env, src, jpegBuf, 0); + dstBuf=jpegBuf=NULL; + _throw(tjGetErrorStr()); + } + + bailout: + if(dstBuf) (*env)->ReleasePrimitiveArrayCritical(env, dst, dstBuf, 0); + if(jpegBuf) (*env)->ReleasePrimitiveArrayCritical(env, src, jpegBuf, 0); + return; +} + +JNIEXPORT void JNICALL Java_org_libjpegturbo_turbojpeg_TJTransformer_init + (JNIEnv *env, jobject obj) +{ + jclass cls; + jfieldID fid; + tjhandle handle; + + if((handle=tjInitTransform())==NULL) _throw(tjGetErrorStr()); + + bailif0(cls=(*env)->GetObjectClass(env, obj)); + bailif0(fid=(*env)->GetFieldID(env, cls, "handle", "J")); + (*env)->SetLongField(env, obj, fid, (jlong)handle); + + bailout: + return; +} + +typedef struct _JNICustomFilterParams +{ + JNIEnv *env; + jobject tobj; + jobject cfobj; +} JNICustomFilterParams; + +static int JNICustomFilter(short *coeffs, tjregion arrayRegion, + tjregion planeRegion, int componentIndex, int transformIndex, + tjtransform *transform) +{ + JNICustomFilterParams *params=(JNICustomFilterParams *)transform->data; + JNIEnv *env=params->env; + jobject tobj=params->tobj, cfobj=params->cfobj; + jobject arrayRegionObj, planeRegionObj, bufobj, borobj; + jclass cls; jmethodID mid; jfieldID fid; + + bailif0(bufobj=(*env)->NewDirectByteBuffer(env, coeffs, + sizeof(short)*arrayRegion.w*arrayRegion.h)); + bailif0(cls=(*env)->FindClass(env, "java/nio/ByteOrder")); + bailif0(mid=(*env)->GetStaticMethodID(env, cls, "nativeOrder", + "()Ljava/nio/ByteOrder;")); + bailif0(borobj=(*env)->CallStaticObjectMethod(env, cls, mid)); + bailif0(cls=(*env)->GetObjectClass(env, bufobj)); + bailif0(mid=(*env)->GetMethodID(env, cls, "order", + "(Ljava/nio/ByteOrder;)Ljava/nio/ByteBuffer;")); + (*env)->CallObjectMethod(env, bufobj, mid, borobj); + bailif0(mid=(*env)->GetMethodID(env, cls, "asShortBuffer", + "()Ljava/nio/ShortBuffer;")); + bailif0(bufobj=(*env)->CallObjectMethod(env, bufobj, mid)); + + bailif0(cls=(*env)->FindClass(env, "java/awt/Rectangle")); + bailif0(arrayRegionObj=(*env)->AllocObject(env, cls)); + bailif0(fid=(*env)->GetFieldID(env, cls, "x", "I")); + (*env)->SetIntField(env, arrayRegionObj, fid, arrayRegion.x); + bailif0(fid=(*env)->GetFieldID(env, cls, "y", "I")); + (*env)->SetIntField(env, arrayRegionObj, fid, arrayRegion.y); + bailif0(fid=(*env)->GetFieldID(env, cls, "width", "I")); + (*env)->SetIntField(env, arrayRegionObj, fid, arrayRegion.w); + bailif0(fid=(*env)->GetFieldID(env, cls, "height", "I")); + (*env)->SetIntField(env, arrayRegionObj, fid, arrayRegion.h); + + bailif0(planeRegionObj=(*env)->AllocObject(env, cls)); + bailif0(fid=(*env)->GetFieldID(env, cls, "x", "I")); + (*env)->SetIntField(env, planeRegionObj, fid, planeRegion.x); + bailif0(fid=(*env)->GetFieldID(env, cls, "y", "I")); + (*env)->SetIntField(env, planeRegionObj, fid, planeRegion.y); + bailif0(fid=(*env)->GetFieldID(env, cls, "width", "I")); + (*env)->SetIntField(env, planeRegionObj, fid, planeRegion.w); + bailif0(fid=(*env)->GetFieldID(env, cls, "height", "I")); + (*env)->SetIntField(env, planeRegionObj, fid, planeRegion.h); + + bailif0(cls=(*env)->GetObjectClass(env, cfobj)); + bailif0(mid=(*env)->GetMethodID(env, cls, "customFilter", + "(Ljava/nio/ShortBuffer;Ljava/awt/Rectangle;Ljava/awt/Rectangle;IILorg/libjpegturbo/turbojpeg/TJTransform;)V")); + (*env)->CallVoidMethod(env, cfobj, mid, bufobj, arrayRegionObj, + planeRegionObj, componentIndex, transformIndex, tobj); + + return 0; + + bailout: + return -1; +} + +JNIEXPORT jintArray JNICALL Java_org_libjpegturbo_turbojpeg_TJTransformer_transform + (JNIEnv *env, jobject obj, jbyteArray jsrcBuf, jint jpegSize, + jobjectArray dstobjs, jobjectArray tobjs, jint flags) +{ + tjhandle handle=0; int i; + unsigned char *jpegBuf=NULL, **dstBufs=NULL; jsize n=0; + unsigned long *dstSizes=NULL; tjtransform *t=NULL; + jbyteArray *jdstBufs=NULL; + int jpegWidth=0, jpegHeight=0, jpegSubsamp; + jintArray jdstSizes=0; jint *dstSizesi=NULL; + JNICustomFilterParams *params=NULL; + + gethandle(); + + if((*env)->GetArrayLength(env, jsrcBuf)<jpegSize) + _throw("Source buffer is not large enough"); + bailif0(_fid=(*env)->GetFieldID(env, _cls, "jpegWidth", "I")); + jpegWidth=(int)(*env)->GetIntField(env, obj, _fid); + bailif0(_fid=(*env)->GetFieldID(env, _cls, "jpegHeight", "I")); + jpegHeight=(int)(*env)->GetIntField(env, obj, _fid); + bailif0(_fid=(*env)->GetFieldID(env, _cls, "jpegSubsamp", "I")); + jpegSubsamp=(int)(*env)->GetIntField(env, obj, _fid); + + n=(*env)->GetArrayLength(env, dstobjs); + if(n!=(*env)->GetArrayLength(env, tobjs)) + _throw("Mismatch between size of transforms array and destination buffers array"); + + if((dstBufs=(unsigned char **)malloc(sizeof(unsigned char *)*n))==NULL) + _throw("Memory allocation failure"); + if((jdstBufs=(jbyteArray *)malloc(sizeof(jbyteArray)*n))==NULL) + _throw("Memory allocation failure"); + if((dstSizes=(unsigned long *)malloc(sizeof(unsigned long)*n))==NULL) + _throw("Memory allocation failure"); + if((t=(tjtransform *)malloc(sizeof(tjtransform)*n))==NULL) + _throw("Memory allocation failure"); + if((params=(JNICustomFilterParams *)malloc(sizeof(JNICustomFilterParams)*n)) + ==NULL) + _throw("Memory allocation failure"); + for(i=0; i<n; i++) + { + dstBufs[i]=NULL; jdstBufs[i]=NULL; dstSizes[i]=0; + memset(&t[i], 0, sizeof(tjtransform)); + memset(¶ms[i], 0, sizeof(JNICustomFilterParams)); + } + + for(i=0; i<n; i++) + { + jobject tobj, cfobj; + + bailif0(tobj=(*env)->GetObjectArrayElement(env, tobjs, i)); + bailif0(_cls=(*env)->GetObjectClass(env, tobj)); + bailif0(_fid=(*env)->GetFieldID(env, _cls, "op", "I")); + t[i].op=(*env)->GetIntField(env, tobj, _fid); + bailif0(_fid=(*env)->GetFieldID(env, _cls, "options", "I")); + t[i].options=(*env)->GetIntField(env, tobj, _fid); + bailif0(_fid=(*env)->GetFieldID(env, _cls, "x", "I")); + t[i].r.x=(*env)->GetIntField(env, tobj, _fid); + bailif0(_fid=(*env)->GetFieldID(env, _cls, "y", "I")); + t[i].r.y=(*env)->GetIntField(env, tobj, _fid); + bailif0(_fid=(*env)->GetFieldID(env, _cls, "width", "I")); + t[i].r.w=(*env)->GetIntField(env, tobj, _fid); + bailif0(_fid=(*env)->GetFieldID(env, _cls, "height", "I")); + t[i].r.h=(*env)->GetIntField(env, tobj, _fid); + + bailif0(_fid=(*env)->GetFieldID(env, _cls, "cf", + "Lorg/libjpegturbo/turbojpeg/TJCustomFilter;")); + cfobj=(*env)->GetObjectField(env, tobj, _fid); + if(cfobj) + { + params[i].env=env; + params[i].tobj=tobj; + params[i].cfobj=cfobj; + t[i].customFilter=JNICustomFilter; + t[i].data=(void *)¶ms[i]; + } + } + + bailif0(jpegBuf=(*env)->GetPrimitiveArrayCritical(env, jsrcBuf, 0)); + for(i=0; i<n; i++) + { + int w=jpegWidth, h=jpegHeight; + if(t[i].r.w!=0) w=t[i].r.w; + if(t[i].r.h!=0) h=t[i].r.h; + bailif0(jdstBufs[i]=(*env)->GetObjectArrayElement(env, dstobjs, i)); + if((unsigned long)(*env)->GetArrayLength(env, jdstBufs[i]) + <tjBufSize(w, h, jpegSubsamp)) + _throw("Destination buffer is not large enough"); + bailif0(dstBufs[i]=(*env)->GetPrimitiveArrayCritical(env, jdstBufs[i], 0)); + } + + if(tjTransform(handle, jpegBuf, jpegSize, n, dstBufs, dstSizes, t, + flags|TJFLAG_NOREALLOC)==-1) + { + (*env)->ReleasePrimitiveArrayCritical(env, jsrcBuf, jpegBuf, 0); + jpegBuf=NULL; + for(i=0; i<n; i++) + { + (*env)->ReleasePrimitiveArrayCritical(env, jdstBufs[i], dstBufs[i], 0); + dstBufs[i]=NULL; + } + _throw(tjGetErrorStr()); + } + + jdstSizes=(*env)->NewIntArray(env, n); + bailif0(dstSizesi=(*env)->GetIntArrayElements(env, jdstSizes, 0)); + for(i=0; i<n; i++) dstSizesi[i]=(int)dstSizes[i]; + + bailout: + if(jpegBuf) (*env)->ReleasePrimitiveArrayCritical(env, jsrcBuf, jpegBuf, 0); + if(dstBufs) + { + for(i=0; i<n; i++) + { + if(dstBufs[i] && jdstBufs && jdstBufs[i]) + (*env)->ReleasePrimitiveArrayCritical(env, jdstBufs[i], dstBufs[i], 0); + } + free(dstBufs); + } + if(jdstBufs) free(jdstBufs); + if(dstSizes) free(dstSizes); + if(dstSizesi) (*env)->ReleaseIntArrayElements(env, jdstSizes, dstSizesi, 0); + if(t) free(t); + return jdstSizes; +} + +JNIEXPORT void JNICALL Java_org_libjpegturbo_turbojpeg_TJDecompressor_destroy + (JNIEnv *env, jobject obj) +{ + Java_org_libjpegturbo_turbojpeg_TJCompressor_destroy(env, obj); +} |