tt/AutoJs6
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

6.6.3 - Alpha5 - 新增 images.compressToBytes/downsample; 修复 images.compress; ImageWrapper#saveTo 支持相对路径

Browse Source
This commit is contained in:
SuperMonster003
2025-05-21 00:41:15 +08:00
parent 4c7ba32646
commit 501c01a632
17 changed files with 563 additions and 231 deletions
Download Patch File Download Diff File Expand all files Collapse all files

216
libs/imagequant/PngQuantBridge.cpp
View File

@@ -1,110 +1,156 @@
#include <jni.h>
#include <android/bitmap.h>
#include <android/log.h>
#include <vector>
#include <setjmp.h>
#include "libimagequant.h"
#include "png.h"
#include <android/log.h>
#define ALOGE(fmt, ...) __android_log_print(ANDROID_LOG_ERROR, "PNGQ", fmt, ##__VA_ARGS__)
/* RAII: 保证函数结束时一定会 unlockPixels(). */
class BitmapLocker {
public:
BitmapLocker(JNIEnv *env, jobject bmp)
: mEnv(env), mBmp(bmp), mPixels(nullptr), mLocked(false) {}
bool lock() {
if (AndroidBitmap_lockPixels(mEnv, mBmp, &mPixels) != 0 || !mPixels) {
return false;
}
mLocked = true;
return true;
}
void *pixels() const { return mPixels; }
~BitmapLocker() {
if (mLocked) {
AndroidBitmap_unlockPixels(mEnv, mBmp);
}
}
private:
JNIEnv *mEnv;
jobject mBmp;
void *mPixels;
bool mLocked;
};
extern "C"
JNIEXPORT jbyteArray JNICALL
JNIEXPORT jbyteArray
JNICALL
Java_org_autojs_autojs_runtime_api_PngQuantBridge_quantize(
JNIEnv *env, jclass, jbyteArray srcRgba,
jint width, jint height, jint quality) {
JNIEnv *env, jclass /*cls*/, jobject jbitmap, jint quality) {
/* 读取像素. */
/* 参数检查 & Bitmap 信息. */
if (!jbitmap) {
env->ThrowNew(env->FindClass("java/lang/IllegalArgumentException"), "Bitmap is null");
return nullptr;
}
AndroidBitmapInfo info{};
if (AndroidBitmap_getInfo(env, jbitmap, &info) != 0 ||
info.format != ANDROID_BITMAP_FORMAT_RGBA_8888) {
env->ThrowNew(env->FindClass("java/lang/IllegalArgumentException"),
"Bitmap must be RGBA_8888");
return nullptr;
}
const int width = static_cast<int>(info.width);
const int height = static_cast<int>(info.height);
jsize len = env->GetArrayLength(srcRgba);
std::vector<uint8_t> rgba(len);
env->GetByteArrayRegion(srcRgba, 0, len, reinterpret_cast<jbyte *>(rgba.data()));
/* 提前声明所有资源. */
BitmapLocker locker(env, jbitmap);
liq_attr *attr = nullptr;
liq_image *image = nullptr;
liq_result *res = nullptr;
std::vector <uint8_t> indexed; // 后面 resize()
const liq_palette *pal = nullptr;
std::vector <uint8_t> compressed; // 输出 PNG
if (len != width * height * 4) {
ALOGE("rgba length mismatch, len=%d, expect=%d", len, width * height * 4);
/* 锁定像素. */
if (!locker.lock()) {
ALOGE("AndroidBitmap_lockPixels failed");
env->ThrowNew(env->FindClass("java/io/IOException"),
"AndroidBitmap_lockPixels failed");
return nullptr;
}
/* libimagequant 量化. */
liq_attr *attr = liq_attr_create();
/* libimagequant. */
attr = liq_attr_create();
if (!attr) {
ALOGE("liq_attr_create failed");
return nullptr;
goto fail;
}
if (quality < 0) quality = 0;
if (quality > 100) quality = 100;
liq_set_speed(attr, 8);
liq_set_quality(attr, quality, quality);
liq_set_max_colors(attr, 256);
liq_image *image = liq_image_create_rgba(attr, rgba.data(), width, height, 0);
liq_result *res;
liq_error err = liq_image_quantize(image, attr, &res);
if (err != LIQ_OK) {
ALOGE("liq_image_quantize failed, code=%d", err);
liq_image_destroy(image);
liq_attr_destroy(attr);
return nullptr;
image = liq_image_create_rgba(attr,
static_cast<uint8_t *>(locker.pixels()),
width, height, 0);
if (!image) {
ALOGE("liq_image_create_rgba failed");
goto fail;
}
if (liq_image_quantize(image, attr, &res) != LIQ_OK || !res) {
ALOGE("liq_image_quantize failed");
goto fail;
}
size_t outSize = width * height;
std::vector<uint8_t> indexed(outSize);
liq_write_remapped_image(res, image, indexed.data(), outSize);
const liq_palette *pal = liq_get_palette(res);
/* Remap. */
indexed.resize(static_cast<size_t>(width) * height);
liq_write_remapped_image(res, image, indexed.data(), indexed.size());
/* 写索引色 PNG 到内存. */
pal = liq_get_palette(res);
if (!pal || pal->count == 0 || pal->count > 256) {
ALOGE("invalid palette, count=%d", pal ? pal->count : -1);
goto fail;
}
std::vector<uint8_t> compressed;
do {
/* palette 检查. */
if (!pal || pal->count == 0 || pal->count > 256) {
ALOGE("invalid palette count=%d", pal ? pal->count : -1);
goto png_fail;
}
/* libpng 结构体. */
/* libpng 写 PNG 到内存. */
{
png_structp png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING,
nullptr,
[](png_structp png_ptr, png_const_charp msg) {
ALOGE("libpng error: %s", msg);
longjmp(png_jmpbuf(png_ptr), 1);
},
nullptr);
nullptr,
[](png_structp png_ptr, png_const_charp msg) {
__android_log_print(ANDROID_LOG_ERROR, "PNGQ", "libpng error: %s", msg);
longjmp(png_jmpbuf(png_ptr), 1);
},
nullptr);
if (!png_ptr) {
ALOGE("png_create_write_struct failed");
goto png_fail;
goto fail;
}
png_infop info_ptr = png_create_info_struct(png_ptr);
if (!info_ptr) {
png_destroy_write_struct(&png_ptr, nullptr);
goto png_fail;
ALOGE("png_create_info_struct failed");
goto fail;
}
/* longjmp error handler. */
if (setjmp(png_jmpbuf(png_ptr))) {
if (setjmp(png_jmpbuf(png_ptr))) { // libpng error
png_destroy_write_struct(&png_ptr, &info_ptr);
goto png_fail;
ALOGE("libpng longjmp error");
goto fail;
}
/* custom mem writer. */
/* 自定义写函数 => vector. */
struct Writer {
static void PNGAPI write(png_structp png_ptr, png_bytep data, png_size_t len) {
auto *vec = static_cast<std::vector<uint8_t> *>(png_get_io_ptr(png_ptr));
auto *vec = static_cast<std::vector <uint8_t> *>(png_get_io_ptr(png_ptr));
vec->insert(vec->end(), data, data + len);
}
static void PNGAPI flush(png_structp) {}
};
std::vector<uint8_t> tmpBuf;
png_set_write_fn(png_ptr, &tmpBuf, Writer::write, Writer::flush);
png_set_write_fn(png_ptr, &compressed, Writer::write, Writer::flush);
/* IHDR. */
png_set_IHDR(png_ptr, info_ptr,
static_cast<png_uint_32>(width),
static_cast<png_uint_32>(height),
8, // bit depth
PNG_COLOR_TYPE_PALETTE, PNG_INTERLACE_NONE,
PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
png_set_IHDR(png_ptr, info_ptr, width, height,
8, PNG_COLOR_TYPE_PALETTE, PNG_INTERLACE_NONE,
PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
png_set_compression_level(png_ptr, 3);
png_set_filter(png_ptr, 0, PNG_FILTER_NONE);
@@ -124,36 +170,38 @@ Java_org_autojs_autojs_runtime_api_PngQuantBridge_quantize(
png_write_info(png_ptr, info_ptr);
/* rows. */
std::vector<png_bytep> rows(height);
/* Rows. */
std::vector <png_bytep> rows(height);
for (int y = 0; y < height; ++y) {
rows[y] = indexed.data() + y * width;
}
png_write_image(png_ptr, rows.data());
png_write_end(png_ptr, info_ptr);
compressed.swap(tmpBuf);
/* 无论是否成功, 均销毁. */
png_destroy_write_struct(&png_ptr, &info_ptr);
} while (false);
png_fail:
if (compressed.empty()) {
ALOGE("compress failed, return null to Java");
env->ThrowNew(env->FindClass("java/io/IOException"),
"pngquant native: compress failed");
return nullptr;
}
/* 清理. */
liq_result_destroy(res);
liq_image_destroy(image);
liq_attr_destroy(attr);
/* 成功: 回传 Java. */
if (compressed.empty()) {
ALOGE("compression produced empty data");
goto fail;
}
{
jbyteArray out = env->NewByteArray(static_cast<jsize>(compressed.size()));
env->SetByteArrayRegion(out, 0, static_cast<jsize>(compressed.size()),
reinterpret_cast<const jbyte *>(compressed.data()));
/* 释放 libimagequant 资源. */
liq_result_destroy(res);
liq_image_destroy(image);
liq_attr_destroy(attr);
return out;
}
/* 回 JNI. */
jbyteArray out = env->NewByteArray(compressed.size());
env->SetByteArrayRegion(out, 0, compressed.size(),
reinterpret_cast<jbyte *>(compressed.data()));
return out;
fail:
/* 统一错误处理. */
if (res) liq_result_destroy(res);
if (image) liq_image_destroy(image);
if (attr) liq_attr_destroy(attr);
env->ThrowNew(env->FindClass("java/io/IOException"), "pngquant native failed");
return nullptr;
}

1
libs/imagequant/build.gradle
View File

@@ -40,6 +40,7 @@ android {
externalNativeBuild {
cmake {
cppFlags "-std=c++17"
cFlags "-ljnigraphics"
}
}