aboutsummaryrefslogtreecommitdiffstats
path: root/src/main/jni/image.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/jni/image.c')
-rw-r--r--src/main/jni/image.c569
1 files changed, 569 insertions, 0 deletions
diff --git a/src/main/jni/image.c b/src/main/jni/image.c
new file mode 100644
index 000000000..85a859007
--- /dev/null
+++ b/src/main/jni/image.c
@@ -0,0 +1,569 @@
+#include <jni.h>
+#include <stdio.h>
+#include <setjmp.h>
+#include <libjpeg/jpeglib.h>
+#include <android/bitmap.h>
+#include <libwebp/webp/decode.h>
+#include <libwebp/webp/encode.h>
+#include "utils.h"
+#include "image.h"
+
+jclass jclass_NullPointerException;
+jclass jclass_RuntimeException;
+
+jclass jclass_Options;
+jfieldID jclass_Options_inJustDecodeBounds;
+jfieldID jclass_Options_outHeight;
+jfieldID jclass_Options_outWidth;
+
+jclass jclass_Bitmap;
+jmethodID jclass_Bitmap_createBitmap;
+jclass jclass_Config;
+jfieldID jclass_Config_ARGB_8888;
+
+const uint32_t PGPhotoEnhanceHistogramBins = 256;
+const uint32_t PGPhotoEnhanceSegments = 4;
+
+jclass createGlobarRef(JNIEnv *env, jclass class) {
+ if (class) {
+ return (*env)->NewGlobalRef(env, class);
+ }
+ return 0;
+}
+
+jint imageOnJNILoad(JavaVM *vm, void *reserved, JNIEnv *env) {
+ jclass_NullPointerException = createGlobarRef(env, (*env)->FindClass(env, "java/lang/NullPointerException"));
+ if (jclass_NullPointerException == 0) {
+ return -1;
+ }
+ jclass_RuntimeException = createGlobarRef(env, (*env)->FindClass(env, "java/lang/RuntimeException"));
+ if (jclass_RuntimeException == 0) {
+ return -1;
+ }
+
+ jclass_Options = createGlobarRef(env, (*env)->FindClass(env, "android/graphics/BitmapFactory$Options"));
+ if (jclass_Options == 0) {
+ return -1;
+ }
+ jclass_Options_inJustDecodeBounds = (*env)->GetFieldID(env, jclass_Options, "inJustDecodeBounds", "Z");
+ if (jclass_Options_inJustDecodeBounds == 0) {
+ return -1;
+ }
+ jclass_Options_outHeight = (*env)->GetFieldID(env, jclass_Options, "outHeight", "I");
+ if (jclass_Options_outHeight == 0) {
+ return -1;
+ }
+ jclass_Options_outWidth = (*env)->GetFieldID(env, jclass_Options, "outWidth", "I");
+ if (jclass_Options_outWidth == 0) {
+ return -1;
+ }
+
+ jclass_Bitmap = createGlobarRef(env, (*env)->FindClass(env, "android/graphics/Bitmap"));
+ if (jclass_Bitmap == 0) {
+ return -1;
+ }
+ jclass_Bitmap_createBitmap = (*env)->GetStaticMethodID(env, jclass_Bitmap, "createBitmap", "(IILandroid/graphics/Bitmap$Config;)Landroid/graphics/Bitmap;");
+ if (jclass_Bitmap_createBitmap == 0) {
+ return -1;
+ }
+
+ jclass_Config = createGlobarRef(env, (*env)->FindClass(env, "android/graphics/Bitmap$Config"));
+ if (jclass_Config == 0) {
+ return -1;
+ }
+ jclass_Config_ARGB_8888 = (*env)->GetStaticFieldID(env, jclass_Config, "ARGB_8888", "Landroid/graphics/Bitmap$Config;");
+ if (jclass_Config_ARGB_8888 == 0) {
+ return -1;
+ }
+
+ return JNI_VERSION_1_6;
+}
+
+static inline uint64_t get_colors (const uint8_t *p) {
+ return p[0] + (p[1] << 16) + ((uint64_t)p[2] << 32);
+}
+
+static void fastBlurMore(int imageWidth, int imageHeight, int imageStride, void *pixels, int radius) {
+ uint8_t *pix = (uint8_t *)pixels;
+ const int w = imageWidth;
+ const int h = imageHeight;
+ const int stride = imageStride;
+ const int r1 = radius + 1;
+ const int div = radius * 2 + 1;
+
+ if (radius > 15 || div >= w || div >= h || w * h > 128 * 128 || imageStride > imageWidth * 4) {
+ return;
+ }
+
+ uint64_t *rgb = malloc(imageWidth * imageHeight * sizeof(uint64_t));
+ if (rgb == NULL) {
+ return;
+ }
+
+ int x, y, i;
+
+ int yw = 0;
+ const int we = w - r1;
+ for (y = 0; y < h; y++) {
+ uint64_t cur = get_colors (&pix[yw]);
+ uint64_t rgballsum = -radius * cur;
+ uint64_t rgbsum = cur * ((r1 * (r1 + 1)) >> 1);
+
+ for (i = 1; i <= radius; i++) {
+ uint64_t cur = get_colors (&pix[yw + i * 4]);
+ rgbsum += cur * (r1 - i);
+ rgballsum += cur;
+ }
+
+ x = 0;
+
+ #define update(start, middle, end) \
+ rgb[y * w + x] = (rgbsum >> 6) & 0x00FF00FF00FF00FF; \
+ rgballsum += get_colors (&pix[yw + (start) * 4]) - 2 * get_colors (&pix[yw + (middle) * 4]) + get_colors (&pix[yw + (end) * 4]); \
+ rgbsum += rgballsum; \
+ x++; \
+
+ while (x < r1) {
+ update (0, x, x + r1);
+ }
+ while (x < we) {
+ update (x - r1, x, x + r1);
+ }
+ while (x < w) {
+ update (x - r1, x, w - 1);
+ }
+ #undef update
+
+ yw += stride;
+ }
+
+ const int he = h - r1;
+ for (x = 0; x < w; x++) {
+ uint64_t rgballsum = -radius * rgb[x];
+ uint64_t rgbsum = rgb[x] * ((r1 * (r1 + 1)) >> 1);
+ for (i = 1; i <= radius; i++) {
+ rgbsum += rgb[i * w + x] * (r1 - i);
+ rgballsum += rgb[i * w + x];
+ }
+
+ y = 0;
+ int yi = x * 4;
+
+ #define update(start, middle, end) \
+ int64_t res = rgbsum >> 6; \
+ pix[yi] = res; \
+ pix[yi + 1] = res >> 16; \
+ pix[yi + 2] = res >> 32; \
+ rgballsum += rgb[x + (start) * w] - 2 * rgb[x + (middle) * w] + rgb[x + (end) * w]; \
+ rgbsum += rgballsum; \
+ y++; \
+ yi += stride;
+
+ while (y < r1) {
+ update (0, y, y + r1);
+ }
+ while (y < he) {
+ update (y - r1, y, y + r1);
+ }
+ while (y < h) {
+ update (y - r1, y, h - 1);
+ }
+ #undef update
+ }
+}
+
+static void fastBlur(int imageWidth, int imageHeight, int imageStride, void *pixels, int radius) {
+ uint8_t *pix = (uint8_t *)pixels;
+ const int w = imageWidth;
+ const int h = imageHeight;
+ const int stride = imageStride;
+ const int r1 = radius + 1;
+ const int div = radius * 2 + 1;
+ int shift;
+ if (radius == 1) {
+ shift = 2;
+ } else if (radius == 3) {
+ shift = 4;
+ } else if (radius == 7) {
+ shift = 6;
+ } else if (radius == 15) {
+ shift = 8;
+ } else {
+ return;
+ }
+
+ if (radius > 15 || div >= w || div >= h || w * h > 128 * 128 || imageStride > imageWidth * 4) {
+ return;
+ }
+
+ uint64_t *rgb = malloc(imageWidth * imageHeight * sizeof(uint64_t));
+ if (rgb == NULL) {
+ return;
+ }
+
+ int x, y, i;
+
+ int yw = 0;
+ const int we = w - r1;
+ for (y = 0; y < h; y++) {
+ uint64_t cur = get_colors (&pix[yw]);
+ uint64_t rgballsum = -radius * cur;
+ uint64_t rgbsum = cur * ((r1 * (r1 + 1)) >> 1);
+
+ for (i = 1; i <= radius; i++) {
+ uint64_t cur = get_colors (&pix[yw + i * 4]);
+ rgbsum += cur * (r1 - i);
+ rgballsum += cur;
+ }
+
+ x = 0;
+
+ #define update(start, middle, end) \
+ rgb[y * w + x] = (rgbsum >> shift) & 0x00FF00FF00FF00FFLL; \
+ rgballsum += get_colors (&pix[yw + (start) * 4]) - 2 * get_colors (&pix[yw + (middle) * 4]) + get_colors (&pix[yw + (end) * 4]); \
+ rgbsum += rgballsum; \
+ x++; \
+
+ while (x < r1) {
+ update (0, x, x + r1);
+ }
+ while (x < we) {
+ update (x - r1, x, x + r1);
+ }
+ while (x < w) {
+ update (x - r1, x, w - 1);
+ }
+
+ #undef update
+
+ yw += stride;
+ }
+
+ const int he = h - r1;
+ for (x = 0; x < w; x++) {
+ uint64_t rgballsum = -radius * rgb[x];
+ uint64_t rgbsum = rgb[x] * ((r1 * (r1 + 1)) >> 1);
+ for (i = 1; i <= radius; i++) {
+ rgbsum += rgb[i * w + x] * (r1 - i);
+ rgballsum += rgb[i * w + x];
+ }
+
+ y = 0;
+ int yi = x * 4;
+
+ #define update(start, middle, end) \
+ int64_t res = rgbsum >> shift; \
+ pix[yi] = res; \
+ pix[yi + 1] = res >> 16; \
+ pix[yi + 2] = res >> 32; \
+ rgballsum += rgb[x + (start) * w] - 2 * rgb[x + (middle) * w] + rgb[x + (end) * w]; \
+ rgbsum += rgballsum; \
+ y++; \
+ yi += stride;
+
+ while (y < r1) {
+ update (0, y, y + r1);
+ }
+ while (y < he) {
+ update (y - r1, y, y + r1);
+ }
+ while (y < h) {
+ update (y - r1, y, h - 1);
+ }
+ #undef update
+ }
+
+ free(rgb);
+}
+
+typedef struct my_error_mgr {
+ struct jpeg_error_mgr pub;
+ jmp_buf setjmp_buffer;
+} *my_error_ptr;
+
+
+METHODDEF(void) my_error_exit(j_common_ptr cinfo) {
+ my_error_ptr myerr = (my_error_ptr) cinfo->err;
+ (*cinfo->err->output_message) (cinfo);
+ longjmp(myerr->setjmp_buffer, 1);
+}
+
+JNIEXPORT void Java_org_telegram_messenger_Utilities_blurBitmap(JNIEnv *env, jclass class, jobject bitmap, int radius) {
+ if (!bitmap) {
+ return;
+ }
+
+ AndroidBitmapInfo info;
+
+ if (AndroidBitmap_getInfo(env, bitmap, &info) < 0) {
+ return;
+ }
+
+ if (info.format != ANDROID_BITMAP_FORMAT_RGBA_8888 || !info.width || !info.height || !info.stride) {
+ return;
+ }
+
+ void *pixels = 0;
+ if (AndroidBitmap_lockPixels(env, bitmap, &pixels) < 0) {
+ return;
+ }
+ if (radius <= 3) {
+ fastBlur(info.width, info.height, info.stride, pixels, radius);
+ } else {
+ fastBlurMore(info.width, info.height, info.stride, pixels, radius);
+ }
+ AndroidBitmap_unlockPixels(env, bitmap);
+}
+
+JNIEXPORT void Java_org_telegram_messenger_Utilities_calcCDT(JNIEnv *env, jclass class, jobject hsvBuffer, int width, int height, jobject buffer) {
+ float imageWidth = width;
+ float imageHeight = height;
+ float _clipLimit = 1.25f;
+
+ uint32_t totalSegments = PGPhotoEnhanceSegments * PGPhotoEnhanceSegments;
+ uint32_t tileArea = (uint32_t)(floorf(imageWidth / PGPhotoEnhanceSegments) * floorf(imageHeight / PGPhotoEnhanceSegments));
+ uint32_t clipLimit = (uint32_t)max(1, _clipLimit * tileArea / (float) PGPhotoEnhanceHistogramBins);
+ float scale = 255.0f / (float) tileArea;
+
+
+ unsigned char *bytes = (*env)->GetDirectBufferAddress(env, hsvBuffer);
+
+ uint32_t **hist = calloc(totalSegments, sizeof(uint32_t *));
+ uint32_t **cdfs = calloc(totalSegments, sizeof(uint32_t *));
+ uint32_t *cdfsMin = calloc(totalSegments, sizeof(uint32_t));
+ uint32_t *cdfsMax = calloc(totalSegments, sizeof(uint32_t));
+
+ for (uint32_t a = 0; a < totalSegments; a++) {
+ hist[a] = calloc(PGPhotoEnhanceHistogramBins, sizeof(uint32_t));
+ cdfs[a] = calloc(PGPhotoEnhanceHistogramBins, sizeof(uint32_t));
+ }
+
+ float xMul = PGPhotoEnhanceSegments / imageWidth;
+ float yMul = PGPhotoEnhanceSegments / imageHeight;
+
+ for (uint32_t y = 0; y < imageHeight; y++) {
+ uint32_t yOffset = y * width * 4;
+ for (uint32_t x = 0; x < imageWidth; x++) {
+ uint32_t index = x * 4 + yOffset;
+
+ uint32_t tx = (uint32_t)(x * xMul);
+ uint32_t ty = (uint32_t)(y * yMul);
+ uint32_t t = ty * PGPhotoEnhanceSegments + tx;
+
+ hist[t][bytes[index + 2]]++;
+ }
+ }
+
+ for (uint32_t i = 0; i < totalSegments; i++) {
+ if (clipLimit > 0) {
+ uint32_t clipped = 0;
+ for (uint32_t j = 0; j < PGPhotoEnhanceHistogramBins; ++j) {
+ if (hist[i][j] > clipLimit) {
+ clipped += hist[i][j] - clipLimit;
+ hist[i][j] = clipLimit;
+ }
+ }
+
+ uint32_t redistBatch = clipped / PGPhotoEnhanceHistogramBins;
+ uint32_t residual = clipped - redistBatch * PGPhotoEnhanceHistogramBins;
+
+ for (uint32_t j = 0; j < PGPhotoEnhanceHistogramBins; ++j) {
+ hist[i][j] += redistBatch;
+ }
+
+ for (uint32_t j = 0; j < residual; ++j) {
+ hist[i][j]++;
+ }
+ }
+ memcpy(cdfs[i], hist[i], PGPhotoEnhanceHistogramBins * sizeof(uint32_t));
+
+ uint32_t hMin = PGPhotoEnhanceHistogramBins - 1;
+ for (uint32_t j = 0; j < hMin; ++j) {
+ if (cdfs[j] != 0) {
+ hMin = j;
+ }
+ }
+
+ uint32_t cdf = 0;
+ for (uint32_t j = hMin; j < PGPhotoEnhanceHistogramBins; ++j) {
+ cdf += cdfs[i][j];
+ cdfs[i][j] = (uint8_t) min(255, cdf * scale);
+ }
+
+ cdfsMin[i] = cdfs[i][hMin];
+ cdfsMax[i] = cdfs[i][PGPhotoEnhanceHistogramBins - 1];
+ }
+
+ uint32_t resultSize = 4 * PGPhotoEnhanceHistogramBins * totalSegments;
+ uint32_t resultBytesPerRow = 4 * PGPhotoEnhanceHistogramBins;
+
+ unsigned char *result = (*env)->GetDirectBufferAddress(env, buffer);
+ for (uint32_t tile = 0; tile < totalSegments; tile++) {
+ uint32_t yOffset = tile * resultBytesPerRow;
+ for (uint32_t i = 0; i < PGPhotoEnhanceHistogramBins; i++) {
+ uint32_t index = i * 4 + yOffset;
+ result[index] = (uint8_t)cdfs[tile][i];
+ result[index + 1] = (uint8_t)cdfsMin[tile];
+ result[index + 2] = (uint8_t)cdfsMax[tile];
+ result[index + 3] = 255;
+ }
+ }
+
+ for (uint32_t a = 0; a < totalSegments; a++) {
+ free(hist[a]);
+ free(cdfs[a]);
+ }
+ free(hist);
+ free(cdfs);
+ free(cdfsMax);
+ free(cdfsMin);
+}
+
+JNIEXPORT int Java_org_telegram_messenger_Utilities_pinBitmap(JNIEnv *env, jclass class, jobject bitmap) {
+ unsigned char *pixels;
+ return AndroidBitmap_lockPixels(env, bitmap, &pixels) >= 0 ? 1 : 0;
+}
+
+JNIEXPORT void Java_org_telegram_messenger_Utilities_loadBitmap(JNIEnv *env, jclass class, jstring path, jobject bitmap, int scale, int width, int height, int stride) {
+
+ AndroidBitmapInfo info;
+ int i;
+
+ if ((i = AndroidBitmap_getInfo(env, bitmap, &info)) >= 0) {
+ char *fileName = (*env)->GetStringUTFChars(env, path, NULL);
+ FILE *infile;
+
+ if ((infile = fopen(fileName, "rb"))) {
+ struct my_error_mgr jerr;
+ struct jpeg_decompress_struct cinfo;
+
+ cinfo.err = jpeg_std_error(&jerr.pub);
+ jerr.pub.error_exit = my_error_exit;
+
+ if (!setjmp(jerr.setjmp_buffer)) {
+ jpeg_create_decompress(&cinfo);
+ jpeg_stdio_src(&cinfo, infile);
+
+ jpeg_read_header(&cinfo, TRUE);
+
+ cinfo.scale_denom = scale;
+ cinfo.scale_num = 1;
+
+ jpeg_start_decompress(&cinfo);
+ int row_stride = cinfo.output_width * cinfo.output_components;
+ JSAMPARRAY buffer = (*cinfo.mem->alloc_sarray) ((j_common_ptr) &cinfo, JPOOL_IMAGE, row_stride, 1);
+
+ unsigned char *pixels;
+ if ((i = AndroidBitmap_lockPixels(env, bitmap, &pixels)) >= 0) {
+ int rowCount = min(cinfo.output_height, height);
+ int colCount = min(cinfo.output_width, width);
+
+ while (cinfo.output_scanline < rowCount) {
+ jpeg_read_scanlines(&cinfo, buffer, 1);
+
+ //if (info.format == ANDROID_BITMAP_FORMAT_RGBA_8888) {
+ if (cinfo.out_color_space == JCS_GRAYSCALE) {
+ for (i = 0; i < colCount; i++) {
+ float alpha = buffer[0][i] / 255.0f;
+ pixels[i * 4] *= alpha;
+ pixels[i * 4 + 1] *= alpha;
+ pixels[i * 4 + 2] *= alpha;
+ pixels[i * 4 + 3] = buffer[0][i];
+ }
+ } else {
+ int c = 0;
+ for (i = 0; i < colCount; i++) {
+ pixels[i * 4] = buffer[0][i * 3];
+ pixels[i * 4 + 1] = buffer[0][i * 3 + 1];
+ pixels[i * 4 + 2] = buffer[0][i * 3 + 2];
+ pixels[i * 4 + 3] = 255;
+ c += 4;
+ }
+ }
+ //} else if (info.format == ANDROID_BITMAP_FORMAT_RGB_565) {
+
+ //}
+
+ pixels += stride;
+ }
+
+ AndroidBitmap_unlockPixels(env, bitmap);
+ } else {
+ throwException(env, "AndroidBitmap_lockPixels() failed ! error=%d", i);
+ }
+
+ jpeg_finish_decompress(&cinfo);
+ } else {
+ throwException(env, "the JPEG code has signaled an error");
+ }
+
+ jpeg_destroy_decompress(&cinfo);
+ fclose(infile);
+ } else {
+ throwException(env, "can't open %s", fileName);
+ }
+
+ (*env)->ReleaseStringUTFChars(env, path, fileName);
+ } else {
+ throwException(env, "AndroidBitmap_getInfo() failed ! error=%d", i);
+ }
+}
+
+JNIEXPORT jobject Java_org_telegram_messenger_Utilities_loadWebpImage(JNIEnv *env, jclass class, jobject buffer, int len, jobject options) {
+ if (!buffer) {
+ (*env)->ThrowNew(env, jclass_NullPointerException, "Input buffer can not be null");
+ return 0;
+ }
+
+ jbyte *inputBuffer = (*env)->GetDirectBufferAddress(env, buffer);
+
+ int bitmapWidth = 0;
+ int bitmapHeight = 0;
+ if (!WebPGetInfo((uint8_t*)inputBuffer, len, &bitmapWidth, &bitmapHeight)) {
+ (*env)->ThrowNew(env, jclass_RuntimeException, "Invalid WebP format");
+ return 0;
+ }
+
+ if (options && (*env)->GetBooleanField(env, options, jclass_Options_inJustDecodeBounds) == JNI_TRUE) {
+ (*env)->SetIntField(env, options, jclass_Options_outWidth, bitmapWidth);
+ (*env)->SetIntField(env, options, jclass_Options_outHeight, bitmapHeight);
+ return 0;
+ }
+
+ jobject value__ARGB_8888 = (*env)->GetStaticObjectField(env, jclass_Config, jclass_Config_ARGB_8888);
+ jobject outputBitmap = (*env)->CallStaticObjectMethod(env, jclass_Bitmap, jclass_Bitmap_createBitmap, (jint)bitmapWidth, (jint)bitmapHeight, value__ARGB_8888);
+ if (!outputBitmap) {
+ (*env)->ThrowNew(env, jclass_RuntimeException, "Failed to allocate Bitmap");
+ return 0;
+ }
+ outputBitmap = (*env)->NewLocalRef(env, outputBitmap);
+
+ AndroidBitmapInfo bitmapInfo;
+ if (AndroidBitmap_getInfo(env, outputBitmap, &bitmapInfo) != ANDROID_BITMAP_RESUT_SUCCESS) {
+ (*env)->DeleteLocalRef(env, outputBitmap);
+ (*env)->ThrowNew(env, jclass_RuntimeException, "Failed to get Bitmap information");
+ return 0;
+ }
+
+ void *bitmapPixels = 0;
+ if (AndroidBitmap_lockPixels(env, outputBitmap, &bitmapPixels) != ANDROID_BITMAP_RESUT_SUCCESS) {
+ (*env)->DeleteLocalRef(env, outputBitmap);
+ (*env)->ThrowNew(env, jclass_RuntimeException, "Failed to lock Bitmap pixels");
+ return 0;
+ }
+
+ if (!WebPDecodeRGBAInto((uint8_t*)inputBuffer, len, (uint8_t*)bitmapPixels, bitmapInfo.height * bitmapInfo.stride, bitmapInfo.stride)) {
+ AndroidBitmap_unlockPixels(env, outputBitmap);
+ (*env)->DeleteLocalRef(env, outputBitmap);
+ (*env)->ThrowNew(env, jclass_RuntimeException, "Failed to unlock Bitmap pixels");
+ return 0;
+ }
+
+ if (AndroidBitmap_unlockPixels(env, outputBitmap) != ANDROID_BITMAP_RESUT_SUCCESS) {
+ (*env)->DeleteLocalRef(env, outputBitmap);
+ (*env)->ThrowNew(env, jclass_RuntimeException, "Failed to unlock Bitmap pixels");
+ return 0;
+ }
+
+ return outputBitmap;
+}
generated by cgit v1.2.3 (git 2.25.1) at 2024-06-10 13:48:18 +0000