bundles/org.simantics.fastlz/native/jniWrapper.c

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
   3  * in Industry THTH ry.\r
   4  * All rights reserved. This program and the accompanying materials\r
   5  * are made available under the terms of the Eclipse Public License v1.0\r
   6  * which accompanies this distribution, and is available at\r
   7  * http://www.eclipse.org/legal/epl-v10.html\r
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 #include "fastlz.h"\r
  13 #include "lz4.h"\r
  14 #include <stdio.h>\r
  15 #include "jni.h"\r
  16 #include <string.h>\r
  17 #include <stdlib.h>\r
  18 \r
  19 JNIEXPORT jint JNICALL Java_org_simantics_fastlz_FastLZ_compress(JNIEnv* env, jclass clazz, \r
  20                                                                  jobject input, jint inputOffset, jint length, \r
  21                                                                  jobject output, jint outputOffset) {\r
  22     void* inputAddress = (char*)(*env)->GetDirectBufferAddress(env, input) + inputOffset;\r
  23     void* outputAddress = (char*)(*env)->GetDirectBufferAddress(env, output) + outputOffset;\r
  24     return fastlz_compress(inputAddress, length, outputAddress);\r
  25 }\r
  26 \r
  27 JNIEXPORT jint JNICALL Java_org_simantics_fastlz_FastLZ_decompress(JNIEnv* env, jclass clazz, \r
  28                                                                    jobject input, jint inputOffset, jint length, \r
  29                                                                    jobject output, jint outputOffset, jint maxout) {\r
  30     void* inputAddress = (char*)(*env)->GetDirectBufferAddress(env, input) + inputOffset;\r
  31     void* outputAddress = (char*)(*env)->GetDirectBufferAddress(env, output) + outputOffset;\r
  32     return fastlz_decompress(inputAddress, length, outputAddress, maxout);\r
  33 }\r
  34 \r
  35 #define INITIAL_SIZE 2000000\r
  36 \r
  37 JNIEXPORT jint JNICALL Java_org_simantics_fastlz_FastLZ_decompressCluster(JNIEnv* env, jclass clazz, jobject deflated, jint deflatedSize, jint inflatedSize, jobjectArray arrays) {\r
  38 \r
  39         static char *inflateBuffer = 0;\r
  40         static int inflateBufferSize = 0;\r
  41 \r
  42         int ll, il, bl;\r
  43 \r
  44         jlongArray longs;\r
  45         jintArray ints;\r
  46         jbyteArray bytes;\r
  47 \r
  48         char *input = (char*)(*env)->GetDirectBufferAddress(env, deflated);\r
  49         char *address;\r
  50 \r
  51         if(inflateBufferSize < inflatedSize) {\r
  52                 if(!inflateBuffer) {\r
  53                         if(inflatedSize < INITIAL_SIZE) inflatedSize = INITIAL_SIZE;\r
  54                         inflateBuffer = malloc(inflatedSize);\r
  55                         inflateBufferSize = inflatedSize;\r
  56                 } else {\r
  57                         if(inflateBuffer) free(inflateBuffer);\r
  58                         inflateBuffer = malloc(inflatedSize);\r
  59                         inflateBufferSize = inflatedSize;\r
  60                 }\r
  61         }\r
  62 \r
  63         address = inflateBuffer;\r
  64 \r
  65         fastlz_decompress(input, deflatedSize, inflateBuffer, inflateBufferSize);\r
  66 \r
  67         ll = *(int *)address;\r
  68         longs = (*env)->NewLongArray(env, ll);\r
  69         (*env)->SetLongArrayRegion(env, longs, 0, ll, (const jlong *)(address + 4));\r
  70         (*env)->SetObjectArrayElement(env, arrays, 0, longs);\r
  71 \r
  72         address += 4 + 8 * ll;\r
  73 \r
  74         il = *(int *)address;\r
  75         ints = (*env)->NewIntArray(env, il);\r
  76         (*env)->SetIntArrayRegion(env, ints, 0, il, (const jint *)(address + 4));\r
  77         (*env)->SetObjectArrayElement(env, arrays, 1, ints);\r
  78 \r
  79         address += 4 * il + 4;\r
  80 \r
  81         bl = *(int *)address;\r
  82         bytes = (*env)->NewByteArray(env, bl);\r
  83         (*env)->SetByteArrayRegion(env, bytes, 0, bl, (const jbyte *)(address + 4));\r
  84         (*env)->SetObjectArrayElement(env, arrays, 2, bytes);\r
  85 \r
  86         return 0;\r
  87 \r
  88 }\r
  89 \r
  90 JNIEXPORT jint JNICALL Java_org_simantics_fastlz_LZ4_compress(JNIEnv* env, jclass clazz, \r
  91                                                               jobject input, jint inputOffset, jint length, \r
  92                                                               jobject output, jint outputOffset) {\r
  93     void* inputAddress = (char*)(*env)->GetDirectBufferAddress(env, input) + inputOffset;\r
  94     void* outputAddress = (char*)(*env)->GetDirectBufferAddress(env, output) + outputOffset;\r
  95     return LZ4_compress(inputAddress, outputAddress, length);\r
  96 }\r
  97 \r
  98 JNIEXPORT jint JNICALL Java_org_simantics_fastlz_LZ4_decompress(JNIEnv* env, jclass clazz, \r
  99                                                                 jobject input, jint inputOffset, jint length, \r
 100                                                                 jobject output, jint outputOffset, jint maxout) {\r
 101     void* inputAddress = (char*)(*env)->GetDirectBufferAddress(env, input) + inputOffset;\r
 102     void* outputAddress = (char*)(*env)->GetDirectBufferAddress(env, output) + outputOffset;\r
 103         return LZ4_uncompress_unknownOutputSize(inputAddress, outputAddress, length, maxout);\r
 104 }\r
 105 \r
 106 JNIEXPORT jint JNICALL Java_org_simantics_fastlz_LZ4_decompressCluster(JNIEnv* env, jclass clazz, jobject deflated, jint deflatedSize, jint inflatedSize, jobjectArray arrays) {\r
 107 \r
 108         static char *inflateBuffer = 0;\r
 109         static int inflateBufferSize = 0;\r
 110 \r
 111         int ll, il, bl;\r
 112 \r
 113         jlongArray longs;\r
 114         jintArray ints;\r
 115         jbyteArray bytes;\r
 116 \r
 117         char *input = (char*)(*env)->GetDirectBufferAddress(env, deflated);\r
 118         char *address;\r
 119 \r
 120         if(inflateBufferSize < inflatedSize) {\r
 121                 if(!inflateBuffer) {\r
 122                         if(inflatedSize < INITIAL_SIZE) inflatedSize = INITIAL_SIZE;\r
 123                         inflateBuffer = malloc(inflatedSize);\r
 124                         inflateBufferSize = inflatedSize;\r
 125                 } else {\r
 126                         if(inflateBuffer) free(inflateBuffer);\r
 127                         inflateBuffer = malloc(inflatedSize);\r
 128                         inflateBufferSize = inflatedSize;\r
 129                 }\r
 130         }\r
 131 \r
 132         address = inflateBuffer;\r
 133 \r
 134         LZ4_uncompress_unknownOutputSize(input, inflateBuffer, deflatedSize, inflateBufferSize);\r
 135 \r
 136         ll = *(int *)address;\r
 137         longs = (*env)->NewLongArray(env, ll);\r
 138         (*env)->SetLongArrayRegion(env, longs, 0, ll, (const jlong *)(address + 4));\r
 139         (*env)->SetObjectArrayElement(env, arrays, 0, longs);\r
 140 \r
 141         address += 4 + 8 * ll;\r
 142 \r
 143         il = *(int *)address;\r
 144         ints = (*env)->NewIntArray(env, il);\r
 145         (*env)->SetIntArrayRegion(env, ints, 0, il, (const jint *)(address + 4));\r
 146         (*env)->SetObjectArrayElement(env, arrays, 1, ints);\r
 147 \r
 148         address += 4 * il + 4;\r
 149 \r
 150         bl = *(int *)address;\r
 151         bytes = (*env)->NewByteArray(env, bl);\r
 152         (*env)->SetByteArrayRegion(env, bytes, 0, bl, (const jbyte *)(address + 4));\r
 153         (*env)->SetObjectArrayElement(env, arrays, 2, bytes);\r
 154 \r
 155         return 0;\r
 156 \r
 157 }\r