/* * Copyright (C) 2011-2012 The Android Open Source Project * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #define LOG_TAG "libRS_jni" #include #include #include #include #include #include #include #include #include #include #include "rsDispatch.h" //#define LOG_API ALOG #define LOG_API(...) #define LOG_ERR(...) __android_log_print(ANDROID_LOG_ERROR, "RenderScript JNI", __VA_ARGS__); #define RS_JNI_VERSION 2301 #define NELEM(m) (sizeof(m) / sizeof((m)[0])) template void UNUSED(T... t) {} #define PER_ARRAY_TYPE(flag, fnc, readonly, ...) { \ jint len = 0; \ void *ptr = nullptr; \ void *srcPtr = nullptr; \ size_t typeBytes = 0; \ jint relFlag = 0; \ if (readonly) { \ /* The on-release mode should only be JNI_ABORT for read-only accesses. */ \ /* readonly = true, also indicates we are copying to the allocation . */ \ relFlag = JNI_ABORT; \ } \ switch(dataType) { \ case RS_TYPE_FLOAT_32: \ len = _env->GetArrayLength((jfloatArray)data); \ ptr = _env->GetFloatArrayElements((jfloatArray)data, flag); \ typeBytes = 4; \ if (usePadding) { \ srcPtr = ptr; \ len = len / 3 * 4; \ if (count == 0) { \ count = len / 4; \ } \ ptr = malloc (len * typeBytes); \ if (readonly) { \ copyWithPadding(ptr, srcPtr, mSize, count); \ fnc(__VA_ARGS__); \ } else { \ fnc(__VA_ARGS__); \ copyWithUnPadding(srcPtr, ptr, mSize, count); \ } \ free(ptr); \ ptr = srcPtr; \ } else { \ fnc(__VA_ARGS__); \ } \ _env->ReleaseFloatArrayElements((jfloatArray)data, (jfloat *)ptr, relFlag); \ return; \ case RS_TYPE_FLOAT_64: \ len = _env->GetArrayLength((jdoubleArray)data); \ ptr = _env->GetDoubleArrayElements((jdoubleArray)data, flag); \ typeBytes = 8; \ if (usePadding) { \ srcPtr = ptr; \ len = len / 3 * 4; \ if (count == 0) { \ count = len / 4; \ } \ ptr = malloc (len * typeBytes); \ if (readonly) { \ copyWithPadding(ptr, srcPtr, mSize, count); \ fnc(__VA_ARGS__); \ } else { \ fnc(__VA_ARGS__); \ copyWithUnPadding(srcPtr, ptr, mSize, count); \ } \ free(ptr); \ ptr = srcPtr; \ } else { \ fnc(__VA_ARGS__); \ } \ _env->ReleaseDoubleArrayElements((jdoubleArray)data, (jdouble *)ptr, relFlag); \ return; \ case RS_TYPE_SIGNED_8: \ case RS_TYPE_UNSIGNED_8: \ len = _env->GetArrayLength((jbyteArray)data); \ ptr = _env->GetByteArrayElements((jbyteArray)data, flag); \ typeBytes = 1; \ if (usePadding) { \ srcPtr = ptr; \ len = len / 3 * 4; \ if (count == 0) { \ count = len / 4; \ } \ ptr = malloc (len * typeBytes); \ if (readonly) { \ copyWithPadding(ptr, srcPtr, mSize, count); \ fnc(__VA_ARGS__); \ } else { \ fnc(__VA_ARGS__); \ copyWithUnPadding(srcPtr, ptr, mSize, count); \ } \ free(ptr); \ ptr = srcPtr; \ } else { \ fnc(__VA_ARGS__); \ } \ _env->ReleaseByteArrayElements((jbyteArray)data, (jbyte*)ptr, relFlag); \ return; \ case RS_TYPE_SIGNED_16: \ case RS_TYPE_UNSIGNED_16: \ len = _env->GetArrayLength((jshortArray)data); \ ptr = _env->GetShortArrayElements((jshortArray)data, flag); \ typeBytes = 2; \ if (usePadding) { \ srcPtr = ptr; \ len = len / 3 * 4; \ if (count == 0) { \ count = len / 4; \ } \ ptr = malloc (len * typeBytes); \ if (readonly) { \ copyWithPadding(ptr, srcPtr, mSize, count); \ fnc(__VA_ARGS__); \ } else { \ fnc(__VA_ARGS__); \ copyWithUnPadding(srcPtr, ptr, mSize, count); \ } \ free(ptr); \ ptr = srcPtr; \ } else { \ fnc(__VA_ARGS__); \ } \ _env->ReleaseShortArrayElements((jshortArray)data, (jshort *)ptr, relFlag); \ return; \ case RS_TYPE_SIGNED_32: \ case RS_TYPE_UNSIGNED_32: \ len = _env->GetArrayLength((jintArray)data); \ ptr = _env->GetIntArrayElements((jintArray)data, flag); \ typeBytes = 4; \ if (usePadding) { \ srcPtr = ptr; \ len = len / 3 * 4; \ if (count == 0) { \ count = len / 4; \ } \ ptr = malloc (len * typeBytes); \ if (readonly) { \ copyWithPadding(ptr, srcPtr, mSize, count); \ fnc(__VA_ARGS__); \ } else { \ fnc(__VA_ARGS__); \ copyWithUnPadding(srcPtr, ptr, mSize, count); \ } \ free(ptr); \ ptr = srcPtr; \ } else { \ fnc(__VA_ARGS__); \ } \ _env->ReleaseIntArrayElements((jintArray)data, (jint *)ptr, relFlag); \ return; \ case RS_TYPE_SIGNED_64: \ case RS_TYPE_UNSIGNED_64: \ len = _env->GetArrayLength((jlongArray)data); \ ptr = _env->GetLongArrayElements((jlongArray)data, flag); \ typeBytes = 8; \ if (usePadding) { \ srcPtr = ptr; \ len = len / 3 * 4; \ if (count == 0) { \ count = len / 4; \ } \ ptr = malloc (len * typeBytes); \ if (readonly) { \ copyWithPadding(ptr, srcPtr, mSize, count); \ fnc(__VA_ARGS__); \ } else { \ fnc(__VA_ARGS__); \ copyWithUnPadding(srcPtr, ptr, mSize, count); \ } \ free(ptr); \ ptr = srcPtr; \ } else { \ fnc(__VA_ARGS__); \ } \ _env->ReleaseLongArrayElements((jlongArray)data, (jlong *)ptr, relFlag); \ return; \ default: \ break; \ } \ UNUSED(len, ptr, srcPtr, typeBytes, relFlag); \ } class AutoJavaStringToUTF8 { public: AutoJavaStringToUTF8(JNIEnv* env, jstring str) : fEnv(env), fJStr(str) { fCStr = env->GetStringUTFChars(str, NULL); fLength = env->GetStringUTFLength(str); } ~AutoJavaStringToUTF8() { fEnv->ReleaseStringUTFChars(fJStr, fCStr); } const char* c_str() const { return fCStr; } jsize length() const { return fLength; } private: JNIEnv* fEnv; jstring fJStr; const char* fCStr; jsize fLength; }; class AutoJavaStringArrayToUTF8 { public: AutoJavaStringArrayToUTF8(JNIEnv* env, jobjectArray strings, jsize stringsLength) : mEnv(env), mStrings(strings), mStringsLength(stringsLength) { mCStrings = NULL; mSizeArray = NULL; if (stringsLength > 0) { mCStrings = (const char **)calloc(stringsLength, sizeof(char *)); mSizeArray = (size_t*)calloc(stringsLength, sizeof(size_t)); for (jsize ct = 0; ct < stringsLength; ct ++) { jstring s = (jstring)mEnv->GetObjectArrayElement(mStrings, ct); mCStrings[ct] = mEnv->GetStringUTFChars(s, NULL); mSizeArray[ct] = mEnv->GetStringUTFLength(s); } } } ~AutoJavaStringArrayToUTF8() { for (jsize ct=0; ct < mStringsLength; ct++) { jstring s = (jstring)mEnv->GetObjectArrayElement(mStrings, ct); mEnv->ReleaseStringUTFChars(s, mCStrings[ct]); } free(mCStrings); free(mSizeArray); } const char **c_str() const { return mCStrings; } size_t *c_str_len() const { return mSizeArray; } jsize length() const { return mStringsLength; } private: JNIEnv *mEnv; jobjectArray mStrings; const char **mCStrings; size_t *mSizeArray; jsize mStringsLength; }; // --------------------------------------------------------------------------- static dispatchTable dispatchTab; // Incremental Support lib static dispatchTable dispatchTabInc; static jboolean nLoadSO(JNIEnv *_env, jobject _this, jboolean useNative, jint targetApi, jstring libPath) { void* handle = NULL; if (useNative) { handle = dlopen("libRS.so", RTLD_LAZY | RTLD_LOCAL); } else { // For API 9+, dlopen the full path of libRSSupport. if (libPath != NULL) { const char * libPathJni = _env->GetStringUTFChars(libPath, JNI_FALSE); handle = dlopen(libPathJni, RTLD_LAZY | RTLD_LOCAL); _env->ReleaseStringUTFChars(libPath, libPathJni); } else { handle = dlopen("libRSSupport.so", RTLD_LAZY | RTLD_LOCAL); } } if (handle == NULL) { LOG_ERR("couldn't dlopen %s; librsjni version: %d", dlerror(), RS_JNI_VERSION); return false; } if (loadSymbols(handle, dispatchTab, targetApi) == false) { LOG_ERR("Dispatch table init failed! librsjni version: %d", RS_JNI_VERSION); dlclose(handle); return false; } LOG_API("Successfully loaded runtime"); return true; } static ioSuppDT ioDispatch; static jboolean nLoadIOSO(JNIEnv *_env, jobject _this) { void* handleIO = NULL; handleIO = dlopen("libRSSupportIO.so", RTLD_LAZY | RTLD_LOCAL); if (handleIO == NULL) { LOG_ERR("Couldn't load libRSSupportIO.so, librsjni version: %d", RS_JNI_VERSION); return false; } if (loadIOSuppSyms(handleIO, ioDispatch) == false) { LOG_ERR("libRSSupportIO init failed! librsjni version: %d", RS_JNI_VERSION); return false; } return true; } // --------------------------------------------------------------------------- static void copyWithPadding(void* ptr, void* srcPtr, int mSize, int count) { int sizeBytesPad = mSize * 4; int sizeBytes = mSize * 3; uint8_t *dst = static_cast(ptr); uint8_t *src = static_cast(srcPtr); for (int i = 0; i < count; i++) { memcpy(dst, src, sizeBytes); dst += sizeBytesPad; src += sizeBytes; } } static void copyWithUnPadding(void* ptr, void* srcPtr, int mSize, int count) { int sizeBytesPad = mSize * 4; int sizeBytes = mSize * 3; uint8_t *dst = static_cast(ptr); uint8_t *src = static_cast(srcPtr); for (int i = 0; i < count; i++) { memcpy(dst, src, sizeBytes); dst += sizeBytes; src += sizeBytesPad; } } // --------------------------------------------------------------------------- static void nContextFinish(JNIEnv *_env, jobject _this, jlong con) { LOG_API("nContextFinish, con(%p)", (RsContext)con); dispatchTab.ContextFinish((RsContext)con); } static jlong nClosureCreate(JNIEnv *_env, jobject _this, jlong con, jlong kernelID, jlong returnValue, jlongArray fieldIDArray, jlongArray valueArray, jintArray sizeArray, jlongArray depClosureArray, jlongArray depFieldIDArray) { jlong ret = 0; jlong* jFieldIDs = _env->GetLongArrayElements(fieldIDArray, nullptr); jsize fieldIDs_length = _env->GetArrayLength(fieldIDArray); jlong* jValues = _env->GetLongArrayElements(valueArray, nullptr); jsize values_length = _env->GetArrayLength(valueArray); jint* jSizes = _env->GetIntArrayElements(sizeArray, nullptr); jsize sizes_length = _env->GetArrayLength(sizeArray); jlong* jDepClosures = _env->GetLongArrayElements(depClosureArray, nullptr); jsize depClosures_length = _env->GetArrayLength(depClosureArray); jlong* jDepFieldIDs = _env->GetLongArrayElements(depFieldIDArray, nullptr); jsize depFieldIDs_length = _env->GetArrayLength(depFieldIDArray); size_t numValues, numDependencies; RsScriptFieldID* fieldIDs; RsClosure* depClosures; RsScriptFieldID* depFieldIDs; if (fieldIDs_length != values_length || values_length != sizes_length) { LOG_ERR("Unmatched field IDs, values, and sizes in closure creation."); goto exit; } numValues = (size_t)fieldIDs_length; if (depClosures_length != depFieldIDs_length) { LOG_ERR("Unmatched closures and field IDs for dependencies in closure creation."); goto exit; } numDependencies = (size_t)depClosures_length; if (numDependencies > numValues) { LOG_ERR("Unexpected number of dependencies in closure creation"); goto exit; } if (numValues > RS_CLOSURE_MAX_NUMBER_ARGS_AND_BINDINGS) { LOG_ERR("Too many arguments or globals in closure creation"); goto exit; } if (numValues > 0) { fieldIDs = (RsScriptFieldID*)alloca(sizeof(RsScriptFieldID) * numValues); if (fieldIDs == nullptr) { goto exit; } } else { // numValues == 0 // alloca(0) implementation is platform dependent fieldIDs = nullptr; } for (size_t i = 0; i < numValues; i++) { fieldIDs[i] = (RsScriptFieldID)jFieldIDs[i]; } if (numDependencies > 0) { depClosures = (RsClosure*)alloca(sizeof(RsClosure) * numDependencies); if (depClosures == nullptr) { goto exit; } for (size_t i = 0; i < numDependencies; i++) { depClosures[i] = (RsClosure)jDepClosures[i]; } depFieldIDs = (RsScriptFieldID*)alloca(sizeof(RsScriptFieldID) * numDependencies); if (depFieldIDs == nullptr) { goto exit; } for (size_t i = 0; i < numDependencies; i++) { depFieldIDs[i] = (RsClosure)jDepFieldIDs[i]; } } else { // numDependencies == 0 // alloca(0) implementation is platform dependent depClosures = nullptr; depFieldIDs = nullptr; } ret = (jlong)(uintptr_t)dispatchTab.ClosureCreate( (RsContext)con, (RsScriptKernelID)kernelID, (RsAllocation)returnValue, fieldIDs, numValues, jValues, numValues, (int*)jSizes, numValues, depClosures, numDependencies, depFieldIDs, numDependencies); exit: _env->ReleaseLongArrayElements(depFieldIDArray, jDepFieldIDs, JNI_ABORT); _env->ReleaseLongArrayElements(depClosureArray, jDepClosures, JNI_ABORT); _env->ReleaseIntArrayElements (sizeArray, jSizes, JNI_ABORT); _env->ReleaseLongArrayElements(valueArray, jValues, JNI_ABORT); _env->ReleaseLongArrayElements(fieldIDArray, jFieldIDs, JNI_ABORT); return ret; } static jlong nInvokeClosureCreate(JNIEnv *_env, jobject _this, jlong con, jlong invokeID, jbyteArray paramArray, jlongArray fieldIDArray, jlongArray valueArray, jintArray sizeArray) { jlong ret = 0; jbyte* jParams = _env->GetByteArrayElements(paramArray, nullptr); jsize jParamLength = _env->GetArrayLength(paramArray); jlong* jFieldIDs = _env->GetLongArrayElements(fieldIDArray, nullptr); jsize fieldIDs_length = _env->GetArrayLength(fieldIDArray); jlong* jValues = _env->GetLongArrayElements(valueArray, nullptr); jsize values_length = _env->GetArrayLength(valueArray); jint* jSizes = _env->GetIntArrayElements(sizeArray, nullptr); jsize sizes_length = _env->GetArrayLength(sizeArray); size_t numValues; RsScriptFieldID* fieldIDs; if (fieldIDs_length != values_length || values_length != sizes_length) { LOG_ERR("Unmatched field IDs, values, and sizes in closure creation."); goto exit; } numValues = (size_t) fieldIDs_length; if (numValues > RS_CLOSURE_MAX_NUMBER_ARGS_AND_BINDINGS) { LOG_ERR("Too many arguments or globals in closure creation"); goto exit; } fieldIDs = (RsScriptFieldID*)alloca(sizeof(RsScriptFieldID) * numValues); if (fieldIDs == nullptr) { goto exit; } for (size_t i = 0; i < numValues; i++) { fieldIDs[i] = (RsScriptFieldID)jFieldIDs[i]; } ret = (jlong)(uintptr_t)dispatchTab.InvokeClosureCreate( (RsContext)con, (RsScriptInvokeID)invokeID, jParams, jParamLength, fieldIDs, numValues, jValues, numValues, (int*)jSizes, numValues); exit: _env->ReleaseIntArrayElements (sizeArray, jSizes, JNI_ABORT); _env->ReleaseLongArrayElements(valueArray, jValues, JNI_ABORT); _env->ReleaseLongArrayElements(fieldIDArray, jFieldIDs, JNI_ABORT); _env->ReleaseByteArrayElements(paramArray, jParams, JNI_ABORT); return ret; } static void nClosureSetArg(JNIEnv *_env, jobject _this, jlong con, jlong closureID, jint index, jlong value, jint size) { // Size is signed with -1 indicating the values is an Allocation dispatchTab.ClosureSetArg((RsContext)con, (RsClosure)closureID, (uint32_t)index, (uintptr_t)value, size); } static void nClosureSetGlobal(JNIEnv *_env, jobject _this, jlong con, jlong closureID, jlong fieldID, jlong value, jint size) { // Size is signed with -1 indicating the values is an Allocation dispatchTab.ClosureSetGlobal((RsContext)con, (RsClosure)closureID, (RsScriptFieldID)fieldID, (int64_t)value, size); } static long nScriptGroup2Create(JNIEnv *_env, jobject _this, jlong con, jstring name, jstring cacheDir, jlongArray closureArray) { jlong ret = 0; AutoJavaStringToUTF8 nameUTF(_env, name); AutoJavaStringToUTF8 cacheDirUTF(_env, cacheDir); jlong* jClosures = _env->GetLongArrayElements(closureArray, nullptr); jsize numClosures = _env->GetArrayLength(closureArray); RsClosure* closures; if (numClosures > (jsize) RS_SCRIPT_GROUP_MAX_NUMBER_CLOSURES) { LOG_ERR("Too many closures in script group"); goto exit; } closures = (RsClosure*)alloca(sizeof(RsClosure) * numClosures); if (closures == nullptr) { goto exit; } for (int i = 0; i < numClosures; i++) { closures[i] = (RsClosure)jClosures[i]; } ret = (jlong)(uintptr_t)dispatchTab.ScriptGroup2Create( (RsContext)con, nameUTF.c_str(), nameUTF.length(), cacheDirUTF.c_str(), cacheDirUTF.length(), closures, numClosures); exit: _env->ReleaseLongArrayElements(closureArray, jClosures, JNI_ABORT); return ret; } static void nScriptGroup2Execute(JNIEnv *_env, jobject _this, jlong con, jlong groupID) { dispatchTab.ScriptGroupExecute((RsContext)con, (RsScriptGroup2)groupID); } static void nObjDestroy(JNIEnv *_env, jobject _this, jlong con, jlong obj) { LOG_API("nObjDestroy, con(%p) obj(%p)", (RsContext)con, (void *)obj); dispatchTab.ObjDestroy((RsContext)con, (void *)obj); } static void nScriptIntrinsicBLAS_Single(JNIEnv *_env, jobject _this, jlong con, jlong incCon, jlong id, jint func, jint TransA, jint TransB, jint Side, jint Uplo, jint Diag, jint M, jint N, jint K, jfloat alpha, jlong A, jlong B, jfloat beta, jlong C, jint incX, jint incY, jint KL, jint KU, jboolean mUseInc) { RsBlasCall call; memset(&call, 0, sizeof(call)); call.func = (RsBlasFunction)func; call.transA = (RsBlasTranspose)TransA; call.transB = (RsBlasTranspose)TransB; call.side = (RsBlasSide)Side; call.uplo = (RsBlasUplo)Uplo; call.diag = (RsBlasDiag)Diag; call.M = M; call.N = N; call.K = K; call.alpha.f = alpha; call.beta.f = beta; call.incX = incX; call.incY = incY; call.KL = KL; call.KU = KU; RsAllocation in_allocs[3]; in_allocs[0] = (RsAllocation)A; in_allocs[1] = (RsAllocation)B; in_allocs[2] = (RsAllocation)C; if (mUseInc) { dispatchTab.ContextFinish((RsContext)con); dispatchTabInc.ScriptForEachMulti((RsContext)incCon, (RsScript)id, 0, in_allocs, NELEM(in_allocs), nullptr, &call, sizeof(call), nullptr, 0); } else { dispatchTab.ScriptForEachMulti((RsContext)con, (RsScript)id, 0, in_allocs, NELEM(in_allocs), nullptr, &call, sizeof(call), nullptr, 0); } } static void nScriptIntrinsicBLAS_Double(JNIEnv *_env, jobject _this, jlong con, jlong incCon, jlong id, jint func, jint TransA, jint TransB, jint Side, jint Uplo, jint Diag, jint M, jint N, jint K, jdouble alpha, jlong A, jlong B, jdouble beta, jlong C, jint incX, jint incY, jint KL, jint KU, jboolean mUseInc) { RsBlasCall call; memset(&call, 0, sizeof(call)); call.func = (RsBlasFunction)func; call.transA = (RsBlasTranspose)TransA; call.transB = (RsBlasTranspose)TransB; call.side = (RsBlasSide)Side; call.uplo = (RsBlasUplo)Uplo; call.diag = (RsBlasDiag)Diag; call.M = M; call.N = N; call.K = K; call.alpha.d = alpha; call.beta.d = beta; call.incX = incX; call.incY = incY; call.KL = KL; call.KU = KU; RsAllocation in_allocs[3]; in_allocs[0] = (RsAllocation)A; in_allocs[1] = (RsAllocation)B; in_allocs[2] = (RsAllocation)C; if (mUseInc) { dispatchTab.ContextFinish((RsContext)con); dispatchTabInc.ScriptForEachMulti((RsContext)incCon, (RsScript)id, 0, in_allocs, NELEM(in_allocs), nullptr, &call, sizeof(call), nullptr, 0); } else { dispatchTab.ScriptForEachMulti((RsContext)con, (RsScript)id, 0, in_allocs, NELEM(in_allocs), nullptr, &call, sizeof(call), nullptr, 0); } } static void nScriptIntrinsicBLAS_Complex(JNIEnv *_env, jobject _this, jlong con, jlong incCon, jlong id, jint func, jint TransA, jint TransB, jint Side, jint Uplo, jint Diag, jint M, jint N, jint K, jfloat alphaX, jfloat alphaY, jlong A, jlong B, jfloat betaX, jfloat betaY, jlong C, jint incX, jint incY, jint KL, jint KU, jboolean mUseInc) { RsBlasCall call; memset(&call, 0, sizeof(call)); call.func = (RsBlasFunction)func; call.transA = (RsBlasTranspose)TransA; call.transB = (RsBlasTranspose)TransB; call.side = (RsBlasSide)Side; call.uplo = (RsBlasUplo)Uplo; call.diag = (RsBlasDiag)Diag; call.M = M; call.N = N; call.K = K; call.alpha.c.r = alphaX; call.alpha.c.i = alphaY; call.beta.c.r = betaX; call.beta.c.i = betaY; call.incX = incX; call.incY = incY; call.KL = KL; call.KU = KU; RsAllocation in_allocs[3]; in_allocs[0] = (RsAllocation)A; in_allocs[1] = (RsAllocation)B; in_allocs[2] = (RsAllocation)C; if (mUseInc) { dispatchTab.ContextFinish((RsContext)con); dispatchTabInc.ScriptForEachMulti((RsContext)incCon, (RsScript)id, 0, in_allocs, NELEM(in_allocs), nullptr, &call, sizeof(call), nullptr, 0); } else { dispatchTab.ScriptForEachMulti((RsContext)con, (RsScript)id, 0, in_allocs, NELEM(in_allocs), nullptr, &call, sizeof(call), nullptr, 0); } } static void nScriptIntrinsicBLAS_Z(JNIEnv *_env, jobject _this, jlong con, jlong incCon, jlong id, jint func, jint TransA, jint TransB, jint Side, jint Uplo, jint Diag, jint M, jint N, jint K, jdouble alphaX, jdouble alphaY, jlong A, jlong B, jdouble betaX, jdouble betaY, jlong C, jint incX, jint incY, jint KL, jint KU, jboolean mUseInc) { RsBlasCall call; memset(&call, 0, sizeof(call)); call.func = (RsBlasFunction)func; call.transA = (RsBlasTranspose)TransA; call.transB = (RsBlasTranspose)TransB; call.side = (RsBlasSide)Side; call.uplo = (RsBlasUplo)Uplo; call.diag = (RsBlasDiag)Diag; call.M = M; call.N = N; call.K = K; call.alpha.z.r = alphaX; call.alpha.z.i = alphaY; call.beta.z.r = betaX; call.beta.z.i = betaY; call.incX = incX; call.incY = incY; call.KL = KL; call.KU = KU; RsAllocation in_allocs[3]; in_allocs[0] = (RsAllocation)A; in_allocs[1] = (RsAllocation)B; in_allocs[2] = (RsAllocation)C; if (mUseInc) { dispatchTab.ContextFinish((RsContext)con); dispatchTabInc.ScriptForEachMulti((RsContext)incCon, (RsScript)id, 0, in_allocs, NELEM(in_allocs), nullptr, &call, sizeof(call), nullptr, 0); } else { dispatchTab.ScriptForEachMulti((RsContext)con, (RsScript)id, 0, in_allocs, NELEM(in_allocs), nullptr, &call, sizeof(call), nullptr, 0); } } static void nScriptIntrinsicBLAS_BNNM(JNIEnv *_env, jobject _this, jlong con, jlong incCon, jlong id, jint M, jint N, jint K, jlong A, jint a_offset, jlong B, jint b_offset, jlong C, jint c_offset, jint c_mult_int, jboolean mUseInc) { RsBlasCall call; memset(&call, 0, sizeof(call)); call.func = RsBlas_bnnm; call.M = M; call.N = N; call.K = K; call.a_offset = a_offset & 0xFF; call.b_offset = b_offset & 0xFF; call.c_offset = c_offset; call.c_mult_int = c_mult_int; RsAllocation in_allocs[3]; in_allocs[0] = (RsAllocation)A; in_allocs[1] = (RsAllocation)B; in_allocs[2] = (RsAllocation)C; if (mUseInc) { dispatchTab.ContextFinish((RsContext)con); dispatchTabInc.ScriptForEachMulti((RsContext)incCon, (RsScript)id, 0, in_allocs, NELEM(in_allocs), nullptr, &call, sizeof(call), nullptr, 0); } else { dispatchTab.ScriptForEachMulti((RsContext)con, (RsScript)id, 0, in_allocs, NELEM(in_allocs), nullptr, &call, sizeof(call), nullptr, 0); } } // --------------------------------------------------------------------------- static jlong nDeviceCreate(JNIEnv *_env, jobject _this) { LOG_API("nDeviceCreate"); return (jlong)(uintptr_t)dispatchTab.DeviceCreate(); } static void nDeviceDestroy(JNIEnv *_env, jobject _this, jlong dev) { LOG_API("nDeviceDestroy"); return dispatchTab.DeviceDestroy((RsDevice)dev); } static void nDeviceSetConfig(JNIEnv *_env, jobject _this, jlong dev, jint p, jint value) { LOG_API("nDeviceSetConfig dev(%p), param(%i), value(%i)", (void *)dev, p, value); return dispatchTab.DeviceSetConfig((RsDevice)dev, (RsDeviceParam)p, value); } static jlong nContextCreate(JNIEnv *_env, jobject _this, jlong dev, jint ver, jint sdkVer, jint ct, jstring nativeLibDirJava) { LOG_API("nContextCreate"); // Access the NativeLibDir in the Java Context. const char * nativeLibDir = _env->GetStringUTFChars(nativeLibDirJava, JNI_FALSE); size_t length = (size_t)_env->GetStringUTFLength(nativeLibDirJava); jlong id = (jlong)(uintptr_t)dispatchTab.ContextCreate((RsDevice)dev, ver, sdkVer, (RsContextType)ct, 0); if (dispatchTab.SetNativeLibDir) { dispatchTab.SetNativeLibDir((RsContext)id, nativeLibDir, length); } _env->ReleaseStringUTFChars(nativeLibDirJava, nativeLibDir); return id; } static void nContextSetPriority(JNIEnv *_env, jobject _this, jlong con, jint p) { LOG_API("ContextSetPriority, con(%p), priority(%i)", (RsContext)con, p); dispatchTab.ContextSetPriority((RsContext)con, p); } static void nContextDestroy(JNIEnv *_env, jobject _this, jlong con) { LOG_API("nContextDestroy, con(%p)", (RsContext)con); dispatchTab.ContextDestroy((RsContext)con); } static void nContextDump(JNIEnv *_env, jobject _this, jlong con, jint bits) { LOG_API("nContextDump, con(%p) bits(%i)", (RsContext)con, bits); dispatchTab.ContextDump((RsContext)con, bits); } static jstring nContextGetErrorMessage(JNIEnv *_env, jobject _this, jlong con) { LOG_API("nContextGetErrorMessage, con(%p)", (RsContext)con); char buf[1024]; size_t receiveLen; uint32_t subID; int id = dispatchTab.ContextGetMessage((RsContext)con, buf, sizeof(buf), &receiveLen, sizeof(receiveLen), &subID, sizeof(subID)); if (!id && receiveLen) { // __android_log_print(ANDROID_LOG_VERBOSE, LOG_TAG, // "message receive buffer too small. %zu", receiveLen); } return _env->NewStringUTF(buf); } static jint nContextGetUserMessage(JNIEnv *_env, jobject _this, jlong con, jintArray data) { jint len = _env->GetArrayLength(data); LOG_API("nContextGetMessage, con(%p), len(%i)", (RsContext)con, len); jint *ptr = _env->GetIntArrayElements(data, NULL); size_t receiveLen; uint32_t subID; int id = dispatchTab.ContextGetMessage((RsContext)con, ptr, len * 4, &receiveLen, sizeof(receiveLen), &subID, sizeof(subID)); if (!id && receiveLen) { // __android_log_print(ANDROID_LOG_VERBOSE, LOG_TAG, // "message receive buffer too small. %zu", receiveLen); } _env->ReleaseIntArrayElements(data, ptr, 0); return (jint)id; } static jint nContextPeekMessage(JNIEnv *_env, jobject _this, jlong con, jintArray auxData) { LOG_API("nContextPeekMessage, con(%p)", (RsContext)con); jint *auxDataPtr = _env->GetIntArrayElements(auxData, NULL); size_t receiveLen; uint32_t subID; int id = dispatchTab.ContextPeekMessage((RsContext)con, &receiveLen, sizeof(receiveLen), &subID, sizeof(subID)); auxDataPtr[0] = (jint)subID; auxDataPtr[1] = (jint)receiveLen; _env->ReleaseIntArrayElements(auxData, auxDataPtr, 0); return (jint)id; } static void nContextInitToClient(JNIEnv *_env, jobject _this, jlong con) { LOG_API("nContextInitToClient, con(%p)", (RsContext)con); dispatchTab.ContextInitToClient((RsContext)con); } static void nContextDeinitToClient(JNIEnv *_env, jobject _this, jlong con) { LOG_API("nContextDeinitToClient, con(%p)", (RsContext)con); dispatchTab.ContextDeinitToClient((RsContext)con); } static void nContextSendMessage(JNIEnv *_env, jobject _this, jlong con, jint id, jintArray data) { jint *ptr = NULL; jint len = 0; if (data) { len = _env->GetArrayLength(data); ptr = _env->GetIntArrayElements(data, NULL); } LOG_API("nContextSendMessage, con(%p), id(%i), len(%i)", (RsContext)con, id, len); dispatchTab.ContextSendMessage((RsContext)con, id, (const uint8_t *)ptr, len * sizeof(int)); if (data) { _env->ReleaseIntArrayElements(data, ptr, JNI_ABORT); } } static jlong nElementCreate(JNIEnv *_env, jobject _this, jlong con, jlong type, jint kind, jboolean norm, jint size) { LOG_API("nElementCreate, con(%p), type(%i), kind(%i), norm(%i), size(%i)", (RsContext)con, type, kind, norm, size); return (jlong)(uintptr_t)dispatchTab.ElementCreate((RsContext)con, (RsDataType)type, (RsDataKind)kind, norm, size); } static jlong nElementCreate2(JNIEnv *_env, jobject _this, jlong con, jlongArray _ids, jobjectArray _names, jintArray _arraySizes) { int fieldCount = _env->GetArrayLength(_ids); LOG_API("nElementCreate2, con(%p)", (RsContext)con); jlong *jIds = _env->GetLongArrayElements(_ids, NULL); jint *jArraySizes = _env->GetIntArrayElements(_arraySizes, NULL); RsElement *ids = (RsElement*)malloc(fieldCount * sizeof(RsElement)); uint32_t *arraySizes = (uint32_t *)malloc(fieldCount * sizeof(uint32_t)); for(int i = 0; i < fieldCount; i ++) { ids[i] = (RsElement)jIds[i]; arraySizes[i] = (uint32_t)jArraySizes[i]; } AutoJavaStringArrayToUTF8 names(_env, _names, fieldCount); const char **nameArray = names.c_str(); size_t *sizeArray = names.c_str_len(); jlong id = (jlong)(uintptr_t)dispatchTab.ElementCreate2((RsContext)con, (RsElement *)ids, fieldCount, nameArray, fieldCount * sizeof(size_t), sizeArray, (const uint32_t *)arraySizes, fieldCount); free(ids); free(arraySizes); _env->ReleaseLongArrayElements(_ids, jIds, JNI_ABORT); _env->ReleaseIntArrayElements(_arraySizes, jArraySizes, JNI_ABORT); return id; } static void nElementGetSubElements(JNIEnv *_env, jobject _this, jlong con, jlong id, jlongArray _IDs, jobjectArray _names, jintArray _arraySizes) { uint32_t dataSize = _env->GetArrayLength(_IDs); LOG_API("nElementGetSubElements, con(%p)", (RsContext)con); uintptr_t *ids = (uintptr_t *)malloc(dataSize * sizeof(uintptr_t)); const char **names = (const char **)malloc((uint32_t)dataSize * sizeof(const char *)); size_t *arraySizes = (size_t *)malloc(dataSize * sizeof(size_t)); dispatchTab.ElementGetSubElements((RsContext)con, (RsElement)id, ids, names, arraySizes, (uint32_t)dataSize); for(uint32_t i = 0; i < dataSize; i++) { const jlong id = (jlong)(uintptr_t)ids[i]; const jint arraySize = (jint)arraySizes[i]; _env->SetObjectArrayElement(_names, i, _env->NewStringUTF(names[i])); _env->SetLongArrayRegion(_IDs, i, 1, &id); _env->SetIntArrayRegion(_arraySizes, i, 1, &arraySize); } free(ids); free(names); free(arraySizes); } // ----------------------------------- static jlong nTypeCreate(JNIEnv *_env, jobject _this, jlong con, jlong eid, jint dimx, jint dimy, jint dimz, jboolean mips, jboolean faces, jint yuv) { LOG_API("nTypeCreate, con(%p) eid(%p), x(%i), y(%i), z(%i), mips(%i), faces(%i), yuv(%i)", (RsContext)con, eid, dimx, dimy, dimz, mips, faces, yuv); return (jlong)(uintptr_t)dispatchTab.TypeCreate((RsContext)con, (RsElement)eid, dimx, dimy, dimz, mips, faces, yuv); } // ----------------------------------- static jlong nAllocationCreateTyped(JNIEnv *_env, jobject _this, jlong con, jlong type, jint mips, jint usage, jlong pointer) { LOG_API("nAllocationCreateTyped, con(%p), type(%p), mip(%i), usage(%i), ptr(%p)", (RsContext)con, (RsElement)type, mips, usage, (void *)pointer); return (jlong)(uintptr_t) dispatchTab.AllocationCreateTyped((RsContext)con, (RsType)type, (RsAllocationMipmapControl)mips, (uint32_t)usage, (uintptr_t)pointer); } static void nAllocationSyncAll(JNIEnv *_env, jobject _this, jlong con, jlong a, jint bits) { LOG_API("nAllocationSyncAll, con(%p), a(%p), bits(0x%08x)", (RsContext)con, (RsAllocation)a, bits); dispatchTab.AllocationSyncAll((RsContext)con, (RsAllocation)a, (RsAllocationUsageType)bits); } static void nAllocationSetSurface(JNIEnv *_env, jobject _this, jlong con, jlong alloc, jobject sur) { ioDispatch.sAllocationSetSurface(_env, _this, (RsContext)con, (RsAllocation)alloc, sur, dispatchTab); } static void nAllocationIoSend(JNIEnv *_env, jobject _this, jlong con, jlong alloc) { dispatchTab.AllocationIoSend((RsContext)con, (RsAllocation)alloc); } static void nAllocationGenerateMipmaps(JNIEnv *_env, jobject _this, jlong con, jlong alloc) { LOG_API("nAllocationGenerateMipmaps, con(%p), a(%p)", (RsContext)con, (RsAllocation)alloc); dispatchTab.AllocationGenerateMipmaps((RsContext)con, (RsAllocation)alloc); } static size_t GetBitmapSize(JNIEnv *env, jobject jbitmap) { AndroidBitmapInfo info; memset(&info, 0, sizeof(info)); AndroidBitmap_getInfo(env, jbitmap, &info); size_t s = info.width * info.height; switch (info.format) { case ANDROID_BITMAP_FORMAT_RGBA_8888: s *= 4; break; case ANDROID_BITMAP_FORMAT_RGB_565: s *= 2; break; case ANDROID_BITMAP_FORMAT_RGBA_4444: s *= 2; break; } return s; } static jlong nAllocationCreateFromBitmap(JNIEnv *_env, jobject _this, jlong con, jlong type, jint mip, jobject jbitmap, jint usage) { jlong id = 0; void *pixels = NULL; AndroidBitmap_lockPixels(_env, jbitmap, &pixels); if (pixels != NULL) { id = (jlong)(uintptr_t)dispatchTab.AllocationCreateFromBitmap((RsContext)con, (RsType)type, (RsAllocationMipmapControl)mip, pixels, GetBitmapSize(_env, jbitmap), usage); AndroidBitmap_unlockPixels(_env, jbitmap); } return id; } static jlong nAllocationCreateBitmapBackedAllocation(JNIEnv *_env, jobject _this, jlong con, jlong type, jint mip, jobject jbitmap, jint usage) { jlong id = 0; void *pixels = NULL; AndroidBitmap_lockPixels(_env, jbitmap, &pixels); if (pixels != NULL) { id = (jlong)(uintptr_t)dispatchTab.AllocationCreateTyped((RsContext)con, (RsType)type, (RsAllocationMipmapControl)mip, (uint32_t)usage, (uintptr_t)pixels); AndroidBitmap_unlockPixels(_env, jbitmap); } return id; } static jlong nAllocationCubeCreateFromBitmap(JNIEnv *_env, jobject _this, jlong con, jlong type, jint mip, jobject jbitmap, jint usage) { void *pixels = NULL; AndroidBitmap_lockPixels(_env, jbitmap, &pixels); jlong id = 0; if (pixels != NULL) { id = (jlong)(uintptr_t)dispatchTab.AllocationCubeCreateFromBitmap((RsContext)con, (RsType)type, (RsAllocationMipmapControl)mip, pixels, GetBitmapSize(_env, jbitmap), usage); AndroidBitmap_unlockPixels(_env, jbitmap); } return id; } static void nAllocationCopyFromBitmap(JNIEnv *_env, jobject _this, jlong con, jlong alloc, jobject jbitmap) { AndroidBitmapInfo info; memset(&info, 0, sizeof(info)); AndroidBitmap_getInfo(_env, jbitmap, &info); void *pixels = NULL; AndroidBitmap_lockPixels(_env, jbitmap, &pixels); if (pixels != NULL) { dispatchTab.Allocation2DData((RsContext)con, (RsAllocation)alloc, 0, 0, 0, RS_ALLOCATION_CUBEMAP_FACE_POSITIVE_X, info.width, info.height, pixels, GetBitmapSize(_env, jbitmap), 0); AndroidBitmap_unlockPixels(_env, jbitmap); } } static void nAllocationCopyToBitmap(JNIEnv *_env, jobject _this, jlong con, jlong alloc, jobject jbitmap) { AndroidBitmapInfo info; memset(&info, 0, sizeof(info)); AndroidBitmap_getInfo(_env, jbitmap, &info); void *pixels = NULL; AndroidBitmap_lockPixels(_env, jbitmap, &pixels); if (pixels != NULL) { dispatchTab.AllocationCopyToBitmap((RsContext)con, (RsAllocation)alloc, pixels, GetBitmapSize(_env, jbitmap)); AndroidBitmap_unlockPixels(_env, jbitmap); } //bitmap.notifyPixelsChanged(); } // Copies from the Java object data into the Allocation pointed to by _alloc. static void nAllocationData1D(JNIEnv *_env, jobject _this, jlong con, jlong _alloc, jint offset, jint lod, jint count, jobject data, jint sizeBytes, jint dataType, jint mSize, jboolean usePadding) { RsAllocation *alloc = (RsAllocation *)_alloc; LOG_API("nAllocation1DData, con(%p), adapter(%p), offset(%i), count(%i), sizeBytes(%i), " "dataType(%i)", (RsContext)con, (RsAllocation)alloc, offset, count, sizeBytes, dataType); PER_ARRAY_TYPE(nullptr, dispatchTab.Allocation1DData, true, (RsContext)con, alloc, offset, lod, count, ptr, sizeBytes); } static void nAllocationElementData1D(JNIEnv *_env, jobject _this, jlong con, jlong alloc, jint xoff, jint lod, jint compIdx, jbyteArray data, jint sizeBytes) { LOG_API("nAllocationElementData1D, con(%p), alloc(%p), xoff(%i), comp(%i), len(%i), " "sizeBytes(%i)", (RsContext)con, (RsAllocation)alloc, xoff, compIdx, _env->GetArrayLength(data), sizeBytes); jbyte *ptr = _env->GetByteArrayElements(data, nullptr); dispatchTab.Allocation1DElementData((RsContext)con, (RsAllocation)alloc, xoff, lod, ptr, sizeBytes, compIdx); _env->ReleaseByteArrayElements(data, ptr, JNI_ABORT); } /* static void nAllocationElementData(JNIEnv *_env, jobject _this, jlong con, jlong alloc, jint xoff, jint yoff, jint zoff, jint lod, jint compIdx, jbyteArray data, jint sizeBytes) { jint len = _env->GetArrayLength(data); LOG_API("nAllocationElementData, con(%p), alloc(%p), xoff(%i), yoff(%i), zoff(%i), comp(%i), len(%i), " "sizeBytes(%i)", (RsContext)con, (RsAllocation)alloc, xoff, yoff, zoff, compIdx, len, sizeBytes); jbyte *ptr = _env->GetByteArrayElements(data, nullptr); dispatchTab.AllocationElementData((RsContext)con, (RsAllocation)alloc, xoff, yoff, zoff, lod, ptr, sizeBytes, compIdx); _env->ReleaseByteArrayElements(data, ptr, JNI_ABORT); } */ // Copies from the Java object data into the Allocation pointed to by _alloc. static void nAllocationData2D(JNIEnv *_env, jobject _this, jlong con, jlong _alloc, jint xoff, jint yoff, jint lod, jint _face, jint w, jint h, jobject data, jint sizeBytes, jint dataType, jint mSize, jboolean usePadding) { RsAllocation *alloc = (RsAllocation *)_alloc; RsAllocationCubemapFace face = (RsAllocationCubemapFace)_face; LOG_API("nAllocation2DData, con(%p), adapter(%p), xoff(%i), yoff(%i), w(%i), h(%i), len(%i) " "type(%i)", (RsContext)con, alloc, xoff, yoff, w, h, sizeBytes, dataType); int count = w * h; PER_ARRAY_TYPE(nullptr, dispatchTab.Allocation2DData, true, (RsContext)con, alloc, xoff, yoff, lod, face, w, h, ptr, sizeBytes, 0); } static void nAllocationData2D_alloc(JNIEnv *_env, jobject _this, jlong con, jlong dstAlloc, jint dstXoff, jint dstYoff, jint dstMip, jint dstFace, jint width, jint height, jlong srcAlloc, jint srcXoff, jint srcYoff, jint srcMip, jint srcFace) { LOG_API("nAllocation2DData_s, con(%p), dstAlloc(%p), dstXoff(%i), dstYoff(%i)," " dstMip(%i), dstFace(%i), width(%i), height(%i)," " srcAlloc(%p), srcXoff(%i), srcYoff(%i), srcMip(%i), srcFace(%i)", (RsContext)con, (RsAllocation)dstAlloc, dstXoff, dstYoff, dstMip, dstFace, width, height, (RsAllocation)srcAlloc, srcXoff, srcYoff, srcMip, srcFace); dispatchTab.AllocationCopy2DRange((RsContext)con, (RsAllocation)dstAlloc, dstXoff, dstYoff, dstMip, dstFace, width, height, (RsAllocation)srcAlloc, srcXoff, srcYoff, srcMip, srcFace); } // Copies from the Java object data into the Allocation pointed to by _alloc. static void nAllocationData3D(JNIEnv *_env, jobject _this, jlong con, jlong _alloc, jint xoff, jint yoff, jint zoff, jint lod, jint w, jint h, jint d, jobject data, jint sizeBytes, jint dataType, jint mSize, jboolean usePadding) { RsAllocation *alloc = (RsAllocation *)_alloc; LOG_API("nAllocation3DData, con(%p), alloc(%p), xoff(%i), yoff(%i), zoff(%i), lod(%i), w(%i)," " h(%i), d(%i), sizeBytes(%i)", (RsContext)con, (RsAllocation)alloc, xoff, yoff, zoff, lod, w, h, d, sizeBytes); int count = w * h * d; PER_ARRAY_TYPE(nullptr, dispatchTab.Allocation3DData, true, (RsContext)con, alloc, xoff, yoff, zoff, lod, w, h, d, ptr, sizeBytes, 0); } static void nAllocationData3D_alloc(JNIEnv *_env, jobject _this, jlong con, jlong dstAlloc, jint dstXoff, jint dstYoff, jint dstZoff, jint dstMip, jint width, jint height, jint depth, jlong srcAlloc, jint srcXoff, jint srcYoff, jint srcZoff, jint srcMip) { LOG_API("nAllocationData3D_alloc, con(%p), dstAlloc(%p), dstXoff(%i), dstYoff(%i)," " dstMip(%i), width(%i), height(%i)," " srcAlloc(%p), srcXoff(%i), srcYoff(%i), srcMip(%i)", (RsContext)con, (RsAllocation)dstAlloc, dstXoff, dstYoff, dstMip, dstFace, width, height, (RsAllocation)srcAlloc, srcXoff, srcYoff, srcMip, srcFace); dispatchTab.AllocationCopy3DRange((RsContext)con, (RsAllocation)dstAlloc, dstXoff, dstYoff, dstZoff, dstMip, width, height, depth, (RsAllocation)srcAlloc, srcXoff, srcYoff, srcZoff, srcMip); } // Copies from the Allocation pointed to by _alloc into the Java object data. static void nAllocationRead(JNIEnv *_env, jobject _this, jlong con, jlong _alloc, jobject data, jint dataType, jint mSize, jboolean usePadding) { RsAllocation *alloc = (RsAllocation *)_alloc; LOG_API("nAllocationRead, con(%p), alloc(%p)", (RsContext)con, (RsAllocation)alloc); int count = 0; PER_ARRAY_TYPE(0, dispatchTab.AllocationRead, false, (RsContext)con, alloc, ptr, len * typeBytes); } // Copies from the Allocation pointed to by _alloc into the Java object data. static void nAllocationRead1D(JNIEnv *_env, jobject _this, jlong con, jlong _alloc, jint offset, jint lod, jint count, jobject data, jint sizeBytes, jint dataType, jint mSize, jboolean usePadding) { RsAllocation *alloc = (RsAllocation *)_alloc; LOG_API("nAllocation1DRead, con(%p), adapter(%p), offset(%i), count(%i), sizeBytes(%i), " "dataType(%i)", (RsContext)con, alloc, offset, count, sizeBytes, dataType); PER_ARRAY_TYPE(0, dispatchTab.Allocation1DRead, false, (RsContext)con, alloc, offset, lod, count, ptr, sizeBytes); } // Copies from the Element in the Allocation pointed to by _alloc into the Java array data. /* static void nAllocationElementRead(JNIEnv *_env, jobject _this, jlong con, jlong _alloc, jint xoff, jint yoff, jint zoff, jint lod, jint compIdx, jobject data, jint sizeBytes) { jint len = _env->GetArrayLength(data); LOG_API("nAllocationElementRead, con(%p), alloc(%p), xoff(%i), yoff(%i), zoff(%i), comp(%i), len(%i), " "sizeBytes(%i)", (RsContext)con, (RsAllocation)alloc, xoff, yoff, zoff, compIdx, len, sizeBytes); jbyte *ptr = _env->GetByteArrayElements(data, nullptr); dispatchTab.AllocationElementRead((RsContext)con, (RsAllocation)alloc, xoff, yoff, zoff, lod, ptr, sizeBytes, compIdx); _env->ReleaseByteArrayElements(data, ptr, JNI_ABORT); } */ // Copies from the Allocation pointed to by _alloc into the Java object data. static void nAllocationRead2D(JNIEnv *_env, jobject _this, jlong con, jlong _alloc, jint xoff, jint yoff, jint lod, jint _face, jint w, jint h, jobject data, jint sizeBytes, jint dataType, jint mSize, jboolean usePadding) { RsAllocation *alloc = (RsAllocation *)_alloc; RsAllocationCubemapFace face = (RsAllocationCubemapFace)_face; LOG_API("nAllocation2DRead, con(%p), adapter(%p), xoff(%i), yoff(%i), w(%i), h(%i), len(%i) " "type(%i)", (RsContext)con, alloc, xoff, yoff, w, h, sizeBytes, dataType); int count = w * h; PER_ARRAY_TYPE(0, dispatchTab.Allocation2DRead, false, (RsContext)con, alloc, xoff, yoff, lod, face, w, h, ptr, sizeBytes, 0); } // Copies from the Allocation pointed to by _alloc into the Java object data. /* static void nAllocationRead3D(JNIEnv *_env, jobject _this, jlong con, jlong _alloc, jint xoff, jint yoff, jint zoff, jint lod, jint w, jint h, jint d, jobject data, int sizeBytes, int dataType, jint mSize, jboolean usePadding) { RsAllocation *alloc = (RsAllocation *)_alloc; LOG_API("nAllocation3DRead, con(%p), alloc(%p), xoff(%i), yoff(%i), zoff(%i), lod(%i), w(%i)," " h(%i), d(%i), sizeBytes(%i)", (RsContext)con, (RsAllocation)alloc, xoff, yoff, zoff, lod, w, h, d, sizeBytes); int count = w * h * d; PER_ARRAY_TYPE(nullptr, dispatchTab.Allocation3DRead, false, (RsContext)con, alloc, xoff, yoff, zoff, lod, w, h, d, ptr, sizeBytes, 0); } */ static jlong nAllocationGetType(JNIEnv *_env, jobject _this, jlong con, jlong a) { LOG_API("nAllocationGetType, con(%p), a(%p)", (RsContext)con, (RsAllocation)a); return (jlong)(uintptr_t) dispatchTab.AllocationGetType((RsContext)con, (RsAllocation)a); } static void nAllocationResize1D(JNIEnv *_env, jobject _this, jlong con, jlong alloc, jint dimX) { LOG_API("nAllocationResize1D, con(%p), alloc(%p), sizeX(%i)", (RsContext)con, (RsAllocation)alloc, dimX); dispatchTab.AllocationResize1D((RsContext)con, (RsAllocation)alloc, dimX); } // ----------------------------------- static void nScriptBindAllocation(JNIEnv *_env, jobject _this, jlong con, jlong script, jlong alloc, jint slot, jboolean mUseInc) { LOG_API("nScriptBindAllocation, con(%p), script(%p), alloc(%p), slot(%i)", (RsContext)con, (RsScript)script, (RsAllocation)alloc, slot); if (mUseInc) { dispatchTabInc.ScriptBindAllocation((RsContext)con, (RsScript)script, (RsAllocation)alloc, slot); } else { dispatchTab.ScriptBindAllocation((RsContext)con, (RsScript)script, (RsAllocation)alloc, slot); } } static void nScriptSetVarI(JNIEnv *_env, jobject _this, jlong con, jlong script, jint slot, jint val, jboolean mUseInc) { LOG_API("nScriptSetVarI, con(%p), s(%p), slot(%i), val(%i)", (RsContext)con, (void *)script, slot, val); if (mUseInc) { dispatchTabInc.ScriptSetVarI((RsContext)con, (RsScript)script, slot, val); } else { dispatchTab.ScriptSetVarI((RsContext)con, (RsScript)script, slot, val); } } static void nScriptSetVarObj(JNIEnv *_env, jobject _this, jlong con, jlong script, jint slot, jlong val, jboolean mUseInc) { LOG_API("nScriptSetVarObj, con(%p), s(%p), slot(%i), val(%i)", (RsContext)con, (void *)script, slot, val); if (mUseInc) { dispatchTabInc.ScriptSetVarObj((RsContext)con, (RsScript)script, slot, (RsObjectBase)val); } else { dispatchTab.ScriptSetVarObj((RsContext)con, (RsScript)script, slot, (RsObjectBase)val); } } static void nScriptSetVarJ(JNIEnv *_env, jobject _this, jlong con, jlong script, jint slot, jlong val, jboolean mUseInc) { LOG_API("nScriptSetVarJ, con(%p), s(%p), slot(%i), val(%lli)", (RsContext)con, (void *)script, slot, val); if (mUseInc) { dispatchTabInc.ScriptSetVarJ((RsContext)con, (RsScript)script, slot, val); } else { dispatchTab.ScriptSetVarJ((RsContext)con, (RsScript)script, slot, val); } } static void nScriptSetVarF(JNIEnv *_env, jobject _this, jlong con, jlong script, jint slot, float val, jboolean mUseInc) { LOG_API("nScriptSetVarF, con(%p), s(%p), slot(%i), val(%f)", (RsContext)con, (void *)script, slot, val); if (mUseInc) { dispatchTabInc.ScriptSetVarF((RsContext)con, (RsScript)script, slot, val); } else { dispatchTab.ScriptSetVarF((RsContext)con, (RsScript)script, slot, val); } } static void nScriptSetVarD(JNIEnv *_env, jobject _this, jlong con, jlong script, jint slot, double val, jboolean mUseInc) { LOG_API("nScriptSetVarD, con(%p), s(%p), slot(%i), val(%lf)", (RsContext)con, (void *)script, slot, val); if (mUseInc) { dispatchTabInc.ScriptSetVarD((RsContext)con, (RsScript)script, slot, val); } else { dispatchTab.ScriptSetVarD((RsContext)con, (RsScript)script, slot, val); } } static void nScriptSetVarV(JNIEnv *_env, jobject _this, jlong con, jlong script, jint slot, jbyteArray data, jboolean mUseInc) { LOG_API("nScriptSetVarV, con(%p), s(%p), slot(%i)", (RsContext)con, (void *)script, slot); jint len = _env->GetArrayLength(data); jbyte *ptr = _env->GetByteArrayElements(data, NULL); if (mUseInc) { dispatchTabInc.ScriptSetVarV((RsContext)con, (RsScript)script, slot, ptr, len); } else { dispatchTab.ScriptSetVarV((RsContext)con, (RsScript)script, slot, ptr, len); } _env->ReleaseByteArrayElements(data, ptr, JNI_ABORT); } static void nScriptSetVarVE(JNIEnv *_env, jobject _this, jlong con, jlong script, jint slot, jbyteArray data, jlong elem, jintArray dims, jboolean mUseInc) { LOG_API("nScriptSetVarVE, con(%p), s(%p), slot(%i)", (RsContext)con, (void *)script, slot); jint len = _env->GetArrayLength(data); jbyte *ptr = _env->GetByteArrayElements(data, NULL); jint dimsLen = _env->GetArrayLength(dims) * sizeof(int); jint *dimsPtr = _env->GetIntArrayElements(dims, NULL); if (mUseInc) { dispatchTabInc.ScriptSetVarVE((RsContext)con, (RsScript)script, slot, ptr, len, (RsElement)elem, (const uint32_t *)dimsPtr, dimsLen); } else { dispatchTab.ScriptSetVarVE((RsContext)con, (RsScript)script, slot, ptr, len, (RsElement)elem, (const uint32_t *)dimsPtr, dimsLen); } _env->ReleaseByteArrayElements(data, ptr, JNI_ABORT); _env->ReleaseIntArrayElements(dims, dimsPtr, JNI_ABORT); } static void nScriptSetTimeZone(JNIEnv *_env, jobject _this, jlong con, jlong script, jbyteArray timeZone, jboolean mUseInc) { LOG_API("nScriptCSetTimeZone, con(%p), s(%p), timeZone(%s)", (RsContext)con, (void *)script, (const char *)timeZone); jint length = _env->GetArrayLength(timeZone); jbyte* timeZone_ptr; timeZone_ptr = (jbyte *) _env->GetPrimitiveArrayCritical(timeZone, (jboolean *)0); if (mUseInc) { dispatchTabInc.ScriptSetTimeZone((RsContext)con, (RsScript)script, (const char *)timeZone_ptr, length); } else { dispatchTab.ScriptSetTimeZone((RsContext)con, (RsScript)script, (const char *)timeZone_ptr, length); } if (timeZone_ptr) { _env->ReleasePrimitiveArrayCritical(timeZone, timeZone_ptr, 0); } } static void nScriptInvoke(JNIEnv *_env, jobject _this, jlong con, jlong obj, jint slot, jboolean mUseInc) { LOG_API("nScriptInvoke, con(%p), script(%p)", (RsContext)con, (void *)obj); if (mUseInc) { dispatchTabInc.ScriptInvoke((RsContext)con, (RsScript)obj, slot); } else { dispatchTab.ScriptInvoke((RsContext)con, (RsScript)obj, slot); } } static void nScriptInvokeV(JNIEnv *_env, jobject _this, jlong con, jlong script, jint slot, jbyteArray data, jboolean mUseInc) { LOG_API("nScriptInvokeV, con(%p), s(%p), slot(%i)", (RsContext)con, (void *)script, slot); jint len = _env->GetArrayLength(data); jbyte *ptr = _env->GetByteArrayElements(data, NULL); if (mUseInc) { dispatchTabInc.ScriptInvokeV((RsContext)con, (RsScript)script, slot, ptr, len); } else { dispatchTab.ScriptInvokeV((RsContext)con, (RsScript)script, slot, ptr, len); } _env->ReleaseByteArrayElements(data, ptr, JNI_ABORT); } static void nScriptForEach(JNIEnv *_env, jobject _this, jlong con, jlong incCon, jlong script, jint slot, jlong ain, jlong aout, jboolean mUseInc) { LOG_API("nScriptForEach, con(%p), s(%p), slot(%i)", (RsContext)con, (void *)script, slot); if (mUseInc) { dispatchTab.ContextFinish((RsContext)con); dispatchTabInc.ScriptForEach((RsContext)incCon, (RsScript)script, slot, (RsAllocation)ain, (RsAllocation)aout, NULL, 0, NULL, 0); } else { dispatchTab.ScriptForEach((RsContext)con, (RsScript)script, slot, (RsAllocation)ain, (RsAllocation)aout, NULL, 0, NULL, 0); } } static void nScriptForEachV(JNIEnv *_env, jobject _this, jlong con, jlong incCon, jlong script, jint slot, jlong ain, jlong aout, jbyteArray params, jboolean mUseInc) { LOG_API("nScriptForEach, con(%p), s(%p), slot(%i)", (RsContext)con, (void *)script, slot); jint len = _env->GetArrayLength(params); jbyte *ptr = _env->GetByteArrayElements(params, NULL); if (mUseInc) { dispatchTab.ContextFinish((RsContext)con); dispatchTabInc.ScriptForEach((RsContext)incCon, (RsScript)script, slot, (RsAllocation)ain, (RsAllocation)aout, ptr, len, NULL, 0); } else { dispatchTab.ScriptForEach((RsContext)con, (RsScript)script, slot, (RsAllocation)ain, (RsAllocation)aout, ptr, len, NULL, 0); } _env->ReleaseByteArrayElements(params, ptr, JNI_ABORT); } static void nScriptForEachClipped(JNIEnv *_env, jobject _this, jlong con, jlong incCon, jlong script, jint slot, jlong ain, jlong aout, jint xstart, jint xend, jint ystart, jint yend, jint zstart, jint zend, jboolean mUseInc) { LOG_API("nScriptForEachClipped, con(%p), s(%p), slot(%i)", (RsContext)con, (void *)script, slot); RsScriptCall sc; sc.xStart = xstart; sc.xEnd = xend; sc.yStart = ystart; sc.yEnd = yend; sc.zStart = zstart; sc.zEnd = zend; sc.strategy = RS_FOR_EACH_STRATEGY_DONT_CARE; sc.arrayStart = 0; sc.arrayEnd = 0; sc.array2Start = 0; sc.array2End = 0; sc.array3Start = 0; sc.array3End = 0; sc.array4Start = 0; sc.array4End = 0; if (mUseInc) { dispatchTab.ContextFinish((RsContext)con); dispatchTabInc.ScriptForEach((RsContext)incCon, (RsScript)script, slot, (RsAllocation)ain, (RsAllocation)aout, NULL, 0, &sc, sizeof(sc)); } else { dispatchTab.ScriptForEach((RsContext)con, (RsScript)script, slot, (RsAllocation)ain, (RsAllocation)aout, NULL, 0, &sc, sizeof(sc)); } } static void nScriptForEachClippedV(JNIEnv *_env, jobject _this, jlong con, jlong incCon, jlong script, jint slot, jlong ain, jlong aout, jbyteArray params, jint xstart, jint xend, jint ystart, jint yend, jint zstart, jint zend, jboolean mUseInc) { LOG_API("nScriptForEachClipped, con(%p), s(%p), slot(%i)", (RsContext)con, (void *)script, slot); jint len = _env->GetArrayLength(params); jbyte *ptr = _env->GetByteArrayElements(params, NULL); RsScriptCall sc; sc.xStart = xstart; sc.xEnd = xend; sc.yStart = ystart; sc.yEnd = yend; sc.zStart = zstart; sc.zEnd = zend; sc.strategy = RS_FOR_EACH_STRATEGY_DONT_CARE; sc.arrayStart = 0; sc.arrayEnd = 0; sc.array2Start = 0; sc.array2End = 0; sc.array3Start = 0; sc.array3End = 0; sc.array4Start = 0; sc.array4End = 0; if (mUseInc) { dispatchTab.ContextFinish((RsContext)con); dispatchTabInc.ScriptForEach((RsContext)incCon, (RsScript)script, slot, (RsAllocation)ain, (RsAllocation)aout, ptr, len, &sc, sizeof(sc)); } else { dispatchTab.ScriptForEach((RsContext)con, (RsScript)script, slot, (RsAllocation)ain, (RsAllocation)aout, ptr, len, &sc, sizeof(sc)); } _env->ReleaseByteArrayElements(params, ptr, JNI_ABORT); } static void nScriptForEachMulti(JNIEnv *_env, jobject _this, jlong con, jlong script, jint slot, jlongArray ains, jlong aout, jbyteArray params, jintArray limits) { LOG_API("nScriptForEach, con(%p), s(%p), slot(%i) ains(%p) aout(%" PRId64 ")", (RsContext)con, (void *)script, slot, ains, aout); jint in_len = 0; jlong *in_ptr = nullptr; RsAllocation *in_allocs = nullptr; if (ains != nullptr) { in_len = _env->GetArrayLength(ains); if (in_len > (jint)RS_KERNEL_MAX_ARGUMENTS) { LOG_ERR("Too many arguments in kernel launch."); // TODO (b/20758983): Report back to Java and throw an exception return; } // TODO (b/20760800): Check in_ptr is not null in_ptr = _env->GetLongArrayElements(ains, nullptr); if (sizeof(RsAllocation) == sizeof(jlong)) { in_allocs = (RsAllocation*)in_ptr; } else { // Convert from 64-bit jlong types to the native pointer type. in_allocs = (RsAllocation*)alloca(in_len * sizeof(RsAllocation)); if (in_allocs == nullptr) { LOG_ERR("Failed launching kernel for lack of memory."); _env->ReleaseLongArrayElements(ains, in_ptr, JNI_ABORT); return; } for (int index = in_len; --index >= 0;) { in_allocs[index] = (RsAllocation)in_ptr[index]; } } } jint param_len = 0; jbyte *param_ptr = nullptr; if (params != nullptr) { param_len = _env->GetArrayLength(params); param_ptr = _env->GetByteArrayElements(params, nullptr); } RsScriptCall sc, *sca = nullptr; uint32_t sc_size = 0; jint limit_len = 0; jint *limit_ptr = nullptr; if (limits != nullptr) { limit_len = _env->GetArrayLength(limits); limit_ptr = _env->GetIntArrayElements(limits, nullptr); if (limit_len != 6) { LOG_ERR("LaunchOptions cannot be recognized."); goto exit; } sc.xStart = limit_ptr[0]; sc.xEnd = limit_ptr[1]; sc.yStart = limit_ptr[2]; sc.yEnd = limit_ptr[3]; sc.zStart = limit_ptr[4]; sc.zEnd = limit_ptr[5]; sc.strategy = RS_FOR_EACH_STRATEGY_DONT_CARE; sc.arrayStart = 0; sc.arrayEnd = 0; sc.array2Start = 0; sc.array2End = 0; sc.array3Start = 0; sc.array3End = 0; sc.array4Start = 0; sc.array4End = 0; sca = ≻ } dispatchTab.ScriptForEachMulti((RsContext)con, (RsScript)script, slot, in_allocs, in_len, (RsAllocation)aout, param_ptr, param_len, sca, sc_size); exit: if (ains != nullptr) { _env->ReleaseLongArrayElements(ains, in_ptr, JNI_ABORT); } if (params != nullptr) { _env->ReleaseByteArrayElements(params, param_ptr, JNI_ABORT); } if (limits != nullptr) { _env->ReleaseIntArrayElements(limits, limit_ptr, JNI_ABORT); } } static void nScriptReduce(JNIEnv *_env, jobject _this, jlong con, jlong script, jint slot, jlongArray ains, jlong aout, jintArray limits) { LOG_API("nScriptReduce, con(%p), s(%p), slot(%i) ains(%p) aout(%" PRId64 ")", (RsContext)con, (void *)script, slot, ains, aout); if (ains == nullptr) { LOG_ERR("At least one input required."); // TODO (b/20758983): Report back to Java and throw an exception return; } jint in_len = _env->GetArrayLength(ains); if (in_len > (jint)RS_KERNEL_MAX_ARGUMENTS) { LOG_ERR("Too many arguments in kernel launch."); // TODO (b/20758983): Report back to Java and throw an exception return; } jlong *in_ptr = _env->GetLongArrayElements(ains, nullptr); if (in_ptr == nullptr) { LOG_ERR("Failed to get Java array elements"); // TODO (b/20758983): Report back to Java and throw an exception return; } RsAllocation *in_allocs = nullptr; if (sizeof(RsAllocation) == sizeof(jlong)) { in_allocs = (RsAllocation*)in_ptr; } else { // Convert from 64-bit jlong types to the native pointer type. in_allocs = (RsAllocation*)alloca(in_len * sizeof(RsAllocation)); if (in_allocs == nullptr) { LOG_ERR("Failed launching kernel for lack of memory."); // TODO (b/20758983): Report back to Java and throw an exception _env->ReleaseLongArrayElements(ains, in_ptr, JNI_ABORT); return; } for (int index = in_len; --index >= 0;) { in_allocs[index] = (RsAllocation)in_ptr[index]; } } RsScriptCall sc, *sca = nullptr; uint32_t sc_size = 0; jint limit_len = 0; jint *limit_ptr = nullptr; if (limits != nullptr) { limit_len = _env->GetArrayLength(limits); limit_ptr = _env->GetIntArrayElements(limits, nullptr); if (limit_ptr == nullptr) { LOG_ERR("Failed to get Java array elements"); // TODO (b/20758983): Report back to Java and throw an exception _env->ReleaseLongArrayElements(ains, in_ptr, JNI_ABORT); return; } if (limit_len != 6) { LOG_ERR("LaunchOptions cannot be recognized"); // TODO (b/20758983): Report back to Java and throw an exception _env->ReleaseLongArrayElements(ains, in_ptr, JNI_ABORT); return; } sc.xStart = limit_ptr[0]; sc.xEnd = limit_ptr[1]; sc.yStart = limit_ptr[2]; sc.yEnd = limit_ptr[3]; sc.zStart = limit_ptr[4]; sc.zEnd = limit_ptr[5]; sc.strategy = RS_FOR_EACH_STRATEGY_DONT_CARE; sc.arrayStart = 0; sc.arrayEnd = 0; sc.array2Start = 0; sc.array2End = 0; sc.array3Start = 0; sc.array3End = 0; sc.array4Start = 0; sc.array4End = 0; sca = ≻ sc_size = sizeof(sc); } dispatchTab.ScriptReduce((RsContext)con, (RsScript)script, slot, in_allocs, in_len, (RsAllocation)aout, sca, sc_size); _env->ReleaseLongArrayElements(ains, in_ptr, JNI_ABORT); if (limits != nullptr) { _env->ReleaseIntArrayElements(limits, limit_ptr, JNI_ABORT); } } // ----------------------------------- static jlong nScriptCCreate(JNIEnv *_env, jobject _this, jlong con, jstring resName, jstring cacheDir, jbyteArray scriptRef, jint length) { LOG_API("nScriptCCreate, con(%p)", (RsContext)con); AutoJavaStringToUTF8 resNameUTF(_env, resName); AutoJavaStringToUTF8 cacheDirUTF(_env, cacheDir); jlong ret = 0; jbyte* script_ptr = NULL; jint _exception = 0; jint remaining; if (!scriptRef) { _exception = 1; //jniThrowException(_env, "java/lang/IllegalArgumentException", "script == null"); goto exit; } if (length < 0) { _exception = 1; //jniThrowException(_env, "java/lang/IllegalArgumentException", "length < 0"); goto exit; } remaining = _env->GetArrayLength(scriptRef); if (remaining < length) { _exception = 1; //jniThrowException(_env, "java/lang/IllegalArgumentException", // "length > script.length - offset"); goto exit; } script_ptr = (jbyte *) _env->GetPrimitiveArrayCritical(scriptRef, (jboolean *)0); //rsScriptCSetText(con, (const char *)script_ptr, length); ret = (jlong)(uintptr_t)dispatchTab.ScriptCCreate((RsContext)con, resNameUTF.c_str(), resNameUTF.length(), cacheDirUTF.c_str(), cacheDirUTF.length(), (const char *)script_ptr, length); exit: if (script_ptr) { _env->ReleasePrimitiveArrayCritical(scriptRef, script_ptr, _exception ? JNI_ABORT: 0); } return (jlong)(uintptr_t)ret; } static jlong nScriptIntrinsicCreate(JNIEnv *_env, jobject _this, jlong con, jint id, jlong eid, jboolean mUseInc) { LOG_API("nScriptIntrinsicCreate, con(%p) id(%i) element(%p)", (RsContext)con, id, (void *)eid); if (mUseInc) { return (jlong)(uintptr_t)dispatchTabInc.ScriptIntrinsicCreate((RsContext)con, id, (RsElement)eid); } else { return (jlong)(uintptr_t)dispatchTab.ScriptIntrinsicCreate((RsContext)con, id, (RsElement)eid); } } static jlong nScriptKernelIDCreate(JNIEnv *_env, jobject _this, jlong con, jlong sid, jint slot, jint sig, jboolean mUseInc) { LOG_API("nScriptKernelIDCreate, con(%p) script(%p), slot(%i), sig(%i)", (RsContext)con, (void *)sid, slot, sig); if (mUseInc) { return (jlong)(uintptr_t)dispatchTabInc.ScriptKernelIDCreate((RsContext)con, (RsScript)sid, slot, sig); } else { return (jlong)(uintptr_t)dispatchTab.ScriptKernelIDCreate((RsContext)con, (RsScript)sid, slot, sig); } } static jlong nScriptInvokeIDCreate(JNIEnv *_env, jobject _this, jlong con, jlong sid, jint slot) { LOG_API("nScriptInvokeIDCreate, con(%p) script(%p), slot(%i), sig(%i)", con, (void *)sid, slot); return (jlong)dispatchTab.ScriptInvokeIDCreate((RsContext)con, (RsScript)sid, slot); } static jlong nScriptFieldIDCreate(JNIEnv *_env, jobject _this, jlong con, jlong sid, jint slot, jboolean mUseInc) { LOG_API("nScriptFieldIDCreate, con(%p) script(%p), slot(%i)", (RsContext)con, (void *)sid, slot); if (mUseInc) { return (jlong)(uintptr_t)dispatchTabInc.ScriptFieldIDCreate((RsContext)con, (RsScript)sid, slot); } else { return (jlong)(uintptr_t)dispatchTab.ScriptFieldIDCreate((RsContext)con, (RsScript)sid, slot); } } static jlong nScriptGroupCreate(JNIEnv *_env, jobject _this, jlong con, jlongArray _kernels, jlongArray _src, jlongArray _dstk, jlongArray _dstf, jlongArray _types) { LOG_API("nScriptGroupCreate, con(%p)", (RsContext)con); jlong id = 0; RsScriptKernelID* kernelsPtr; jint kernelsLen = _env->GetArrayLength(_kernels); jlong *jKernelsPtr = _env->GetLongArrayElements(_kernels, nullptr); RsScriptKernelID* srcPtr; jint srcLen = _env->GetArrayLength(_src); jlong *jSrcPtr = _env->GetLongArrayElements(_src, nullptr); RsScriptKernelID* dstkPtr; jint dstkLen = _env->GetArrayLength(_dstk); jlong *jDstkPtr = _env->GetLongArrayElements(_dstk, nullptr); RsScriptKernelID* dstfPtr; jint dstfLen = _env->GetArrayLength(_dstf); jlong *jDstfPtr = _env->GetLongArrayElements(_dstf, nullptr); RsType* typesPtr; jint typesLen = _env->GetArrayLength(_types); jlong *jTypesPtr = _env->GetLongArrayElements(_types, nullptr); if (jKernelsPtr == nullptr) { LOG_ERR("Failed to get Java array elements: kernels"); goto cleanup; } if (jSrcPtr == nullptr) { LOG_ERR("Failed to get Java array elements: src"); goto cleanup; } if (jDstkPtr == nullptr) { LOG_ERR("Failed to get Java array elements: dstk"); goto cleanup; } if (jDstfPtr == nullptr) { LOG_ERR("Failed to get Java array elements: dstf"); goto cleanup; } if (jTypesPtr == nullptr) { LOG_ERR("Failed to get Java array elements: types"); goto cleanup; } kernelsPtr = (RsScriptKernelID*) malloc(sizeof(RsScriptKernelID) * kernelsLen); for(int i = 0; i < kernelsLen; ++i) { kernelsPtr[i] = (RsScriptKernelID)jKernelsPtr[i]; } srcPtr = (RsScriptKernelID*) malloc(sizeof(RsScriptKernelID) * srcLen); for(int i = 0; i < srcLen; ++i) { srcPtr[i] = (RsScriptKernelID)jSrcPtr[i]; } dstkPtr = (RsScriptKernelID*) malloc(sizeof(RsScriptKernelID) * dstkLen); for(int i = 0; i < dstkLen; ++i) { dstkPtr[i] = (RsScriptKernelID)jDstkPtr[i]; } dstfPtr = (RsScriptKernelID*) malloc(sizeof(RsScriptKernelID) * dstfLen); for(int i = 0; i < dstfLen; ++i) { dstfPtr[i] = (RsScriptKernelID)jDstfPtr[i]; } typesPtr = (RsType*) malloc(sizeof(RsType) * typesLen); for(int i = 0; i < typesLen; ++i) { typesPtr[i] = (RsType)jTypesPtr[i]; } id = (jlong)(uintptr_t) dispatchTab.ScriptGroupCreate((RsContext)con, (RsScriptKernelID *)kernelsPtr, kernelsLen * sizeof(RsScriptKernelID), (RsScriptKernelID *)srcPtr, srcLen * sizeof(RsScriptKernelID), (RsScriptKernelID *)dstkPtr, dstkLen * sizeof(RsScriptKernelID), (RsScriptFieldID *)dstfPtr, dstfLen * sizeof(RsScriptKernelID), (RsType *)typesPtr, typesLen * sizeof(RsType)); free(kernelsPtr); free(srcPtr); free(dstkPtr); free(dstfPtr); free(typesPtr); cleanup: if (jKernelsPtr != nullptr) { _env->ReleaseLongArrayElements(_kernels, jKernelsPtr, 0); } if (jSrcPtr != nullptr) { _env->ReleaseLongArrayElements(_src, jSrcPtr, 0); } if (jDstkPtr != nullptr) { _env->ReleaseLongArrayElements(_dstk, jDstkPtr, 0); } if (jDstfPtr != nullptr) { _env->ReleaseLongArrayElements(_dstf, jDstfPtr, 0); } if (jTypesPtr != nullptr) { _env->ReleaseLongArrayElements(_types, jTypesPtr, 0); } return id; } static void nScriptGroupSetInput(JNIEnv *_env, jobject _this, jlong con, jlong gid, jlong kid, jlong alloc) { LOG_API("nScriptGroupSetInput, con(%p) group(%p), kernelId(%p), alloc(%p)", (RsContext)con, (void *)gid, (void *)kid, (void *)alloc); dispatchTab.ScriptGroupSetInput((RsContext)con, (RsScriptGroup)gid, (RsScriptKernelID)kid, (RsAllocation)alloc); } static void nScriptGroupSetOutput(JNIEnv *_env, jobject _this, jlong con, jlong gid, jlong kid, jlong alloc) { LOG_API("nScriptGroupSetOutput, con(%p) group(%p), kernelId(%p), alloc(%p)", (RsContext)con, (void *)gid, (void *)kid, (void *)alloc); dispatchTab.ScriptGroupSetOutput((RsContext)con, (RsScriptGroup)gid, (RsScriptKernelID)kid, (RsAllocation)alloc); } static void nScriptGroupExecute(JNIEnv *_env, jobject _this, jlong con, jlong gid) { LOG_API("nScriptGroupSetOutput, con(%p) group(%p)", (RsContext)con, (void *)gid); dispatchTab.ScriptGroupExecute((RsContext)con, (RsScriptGroup)gid); } // --------------------------------------------------------------------------- static jlong nSamplerCreate(JNIEnv *_env, jobject _this, jlong con, jint magFilter, jint minFilter, jint wrapS, jint wrapT, jint wrapR, jfloat aniso) { LOG_API("nSamplerCreate, con(%p)", (RsContext)con); return (jlong)(uintptr_t)dispatchTab.SamplerCreate((RsContext)con, (RsSamplerValue)magFilter, (RsSamplerValue)minFilter, (RsSamplerValue)wrapS, (RsSamplerValue)wrapT, (RsSamplerValue)wrapR, aniso); } static jint nSystemGetPointerSize(JNIEnv *_env, jobject _this) { return (jint)sizeof(void*); } // --------------------------------------------------------------------------- // For Incremental Intrinsic Support static jboolean nIncLoadSO(JNIEnv *_env, jobject _this, jint deviceApi, jstring libPath) { void* handle = NULL; // For API 9+, dlopen the full path of libRSSupport. if (libPath != NULL) { const char * libPathJni = _env->GetStringUTFChars(libPath, JNI_FALSE); handle = dlopen(libPathJni, RTLD_LAZY | RTLD_LOCAL); _env->ReleaseStringUTFChars(libPath, libPathJni); } else { handle = dlopen("libRSSupport.so", RTLD_LAZY | RTLD_LOCAL); } if (handle == NULL) { LOG_ERR("couldn't dlopen %s; librsjni version: %d", dlerror(), RS_JNI_VERSION); return false; } if (loadSymbols(handle, dispatchTabInc, deviceApi) == false) { LOG_ERR("Dispatch Table init failed! librsjni version: %d", RS_JNI_VERSION); dlclose(handle); return false; } dispatchTabInc.AllocationCreateStrided = (AllocationCreateStridedFnPtr)dlsym(handle, "rsAllocationCreateStrided"); if (dispatchTabInc.AllocationCreateStrided == NULL) { LOG_ERR("Couldn't initialize dispatchTabInc.AllocationCreateStrided"); dlclose(handle); return false; } LOG_API("Successfully loaded compat runtime"); return true; } // ----------------------------------- // To create/destroy a dummy context static void nIncObjDestroy(JNIEnv *_env, jobject _this, jlong con, jlong obj) { LOG_API("nObjDestroy, con(%p) obj(%p)", (RsContext)con, (void *)obj); dispatchTabInc.ObjDestroy((RsContext)con, (void *)obj); } static jlong nIncDeviceCreate(JNIEnv *_env, jobject _this) { LOG_API("nDeviceCreate"); return (jlong)(uintptr_t)dispatchTabInc.DeviceCreate(); } static void nIncDeviceDestroy(JNIEnv *_env, jobject _this, jlong dev) { LOG_API("nDeviceDestroy"); return dispatchTabInc.DeviceDestroy((RsDevice)dev); } static jlong nIncContextCreate(JNIEnv *_env, jobject _this, jlong dev, jint ver, jint sdkVer, jint ct) { LOG_API("nContextCreate"); //The compat context for incremental support will be synchronous. return (jlong)(uintptr_t)dispatchTabInc.ContextCreate((RsDevice)dev, ver, sdkVer, (RsContextType)ct, RS_CONTEXT_SYNCHRONOUS); } static void nIncContextFinish(JNIEnv *_env, jobject _this, jlong con) { LOG_API("nContextFinish, con(%p)", (RsContext)con); dispatchTabInc.ContextFinish((RsContext)con); } static void nIncContextDestroy(JNIEnv *_env, jobject _this, jlong con) { LOG_API("nContextDestroy, con(%p)", (RsContext)con); dispatchTabInc.ContextDestroy((RsContext)con); } // ----------------------------------- // Create dummy Element static jlong nIncElementCreate(JNIEnv *_env, jobject _this, jlong con, jlong type, jint kind, jboolean norm, jint size) { LOG_API("nElementCreate, con(%p), type(%i), kind(%i), norm(%i), size(%i)", (RsContext)con, type, kind, norm, size); return (jlong)(uintptr_t)dispatchTabInc.ElementCreate((RsContext)con, (RsDataType)type, (RsDataKind)kind, norm, size); } // ----------------------------------- // Create dummy Type static jlong nIncTypeCreate(JNIEnv *_env, jobject _this, jlong con, jlong eid, jint dimx, jint dimy, jint dimz, jboolean mips, jboolean faces, jint yuv) { LOG_API("nTypeCreate, con(%p) eid(%p), x(%i), y(%i), z(%i), mips(%i), faces(%i), yuv(%i)", incCon, eid, dimx, dimy, dimz, mips, faces, yuv); return (jlong)(uintptr_t)dispatchTabInc.TypeCreate((RsContext)con, (RsElement)eid, dimx, dimy, dimz, mips, faces, yuv); } // ----------------------------------- // Create Allocation from pointer static jlong nIncAllocationCreateTyped(JNIEnv *_env, jobject _this, jlong con, jlong incCon, jlong alloc, jlong type, jint xBytesSize) { LOG_API("nAllocationCreateTyped, con(%p), type(%p), mip(%i), usage(%i), ptr(%p)", incCon, (RsElement)type, mips, usage, (void *)pointer); size_t strideIn; void* pIn = NULL; RsAllocation ainI = NULL; if (alloc != 0) { pIn = dispatchTab.AllocationGetPointer((RsContext)con, (RsAllocation)alloc, 0, RS_ALLOCATION_CUBEMAP_FACE_POSITIVE_X, 0, 0, &strideIn, sizeof(size_t)); /* * By definition stride is a roundup of xBytesSize with requiredAlignment, so requiredAlignment must * be strictly larger than the difference of (stride - xBytesSize). * * We can prove that as long as requiredAlignment satisfies the following two conditions, the * memory layout will be identical : * 1. Smaller or equal than stride; * 2. Larger than minRequiredAlignment. * * In this case we can simply choose the first power of 2 that satisfies both conditions. */ size_t requiredAlignment = 16; size_t minRequiredAlignment = strideIn - xBytesSize; while (requiredAlignment <= minRequiredAlignment) { requiredAlignment <<= 1; } ainI = dispatchTabInc.AllocationCreateStrided((RsContext)incCon, (RsType)type, RS_ALLOCATION_MIPMAP_NONE, RS_ALLOCATION_USAGE_INCREMENTAL_SUPPORT | RS_ALLOCATION_USAGE_SHARED, (uintptr_t)pIn, requiredAlignment); } return (jlong)(uintptr_t) ainI; } static jobject nAllocationGetByteBuffer(JNIEnv *_env, jobject _this, jlong con, jlong alloc, jint xBytesSize, jint dimY, jint dimZ) { LOG_API("nAllocationGetByteBuffer, con(%p), alloc(%p)", (RsContext)con, (RsAllocation)alloc); size_t strideIn = xBytesSize; void* ptr = NULL; if (alloc != 0 && dispatchTab.AllocationGetPointer != nullptr) { ptr = dispatchTab.AllocationGetPointer((RsContext)con, (RsAllocation)alloc, 0, RS_ALLOCATION_CUBEMAP_FACE_POSITIVE_X, dimZ, 0, &strideIn, sizeof(size_t)); } if (ptr != NULL) { size_t bufferSize = strideIn; if (dimY > 0) { bufferSize *= dimY; } if (dimZ > 0) { bufferSize *= dimZ; } jobject byteBuffer = _env->NewDirectByteBuffer(ptr, (jlong) bufferSize); return byteBuffer; } else { return NULL; } } static jlong nAllocationGetStride(JNIEnv *_env, jobject _this, jlong con, jlong alloc) { LOG_API("nAllocationGetStride, con(%p), alloc(%p)", (RsContext)con, (RsAllocation)alloc); size_t strideIn = 0; if (alloc != 0 && dispatchTab.AllocationGetPointer != nullptr) { dispatchTab.AllocationGetPointer((RsContext)con, (RsAllocation)alloc, 0, RS_ALLOCATION_CUBEMAP_FACE_POSITIVE_X, 0, 0, &strideIn, sizeof(size_t)); } return (jlong)strideIn; } // --------------------------------------------------------------------------- static const char *classPathName = "androidx/renderscript/RenderScript"; static JNINativeMethod methods[] = { {"nLoadSO", "(ZILjava/lang/String;)Z", (bool*)nLoadSO }, {"nLoadIOSO", "()Z", (bool*)nLoadIOSO }, {"nDeviceCreate", "()J", (void*)nDeviceCreate }, {"nDeviceDestroy", "(J)V", (void*)nDeviceDestroy }, {"nDeviceSetConfig", "(JII)V", (void*)nDeviceSetConfig }, {"nContextGetUserMessage", "(J[I)I", (void*)nContextGetUserMessage }, {"nContextGetErrorMessage", "(J)Ljava/lang/String;", (void*)nContextGetErrorMessage }, {"nContextPeekMessage", "(J[I)I", (void*)nContextPeekMessage }, {"nContextInitToClient", "(J)V", (void*)nContextInitToClient }, {"nContextDeinitToClient", "(J)V", (void*)nContextDeinitToClient }, // All methods below are thread protected in java. {"rsnContextCreate", "(JIIILjava/lang/String;)J", (void*)nContextCreate }, {"rsnContextFinish", "(J)V", (void*)nContextFinish }, {"rsnContextSetPriority", "(JI)V", (void*)nContextSetPriority }, {"rsnContextDestroy", "(J)V", (void*)nContextDestroy }, {"rsnContextDump", "(JI)V", (void*)nContextDump }, {"rsnContextSendMessage", "(JI[I)V", (void*)nContextSendMessage }, {"rsnClosureCreate", "(JJJ[J[J[I[J[J)J", (void*)nClosureCreate }, {"rsnInvokeClosureCreate", "(JJ[B[J[J[I)J", (void*)nInvokeClosureCreate }, {"rsnClosureSetArg", "(JJIJI)V", (void*)nClosureSetArg }, {"rsnClosureSetGlobal", "(JJJJI)V", (void*)nClosureSetGlobal }, {"rsnObjDestroy", "(JJ)V", (void*)nObjDestroy }, {"rsnElementCreate", "(JJIZI)J", (void*)nElementCreate }, {"rsnElementCreate2", "(J[J[Ljava/lang/String;[I)J", (void*)nElementCreate2 }, {"rsnElementGetSubElements", "(JJ[J[Ljava/lang/String;[I)V", (void*)nElementGetSubElements }, {"rsnTypeCreate", "(JJIIIZZI)J", (void*)nTypeCreate }, {"rsnAllocationCreateTyped", "(JJIIJ)J", (void*)nAllocationCreateTyped }, {"rsnAllocationCreateFromBitmap", "(JJILandroid/graphics/Bitmap;I)J", (void*)nAllocationCreateFromBitmap }, {"rsnAllocationCreateBitmapBackedAllocation", "(JJILandroid/graphics/Bitmap;I)J", (void*)nAllocationCreateBitmapBackedAllocation }, {"rsnAllocationCubeCreateFromBitmap","(JJILandroid/graphics/Bitmap;I)J", (void*)nAllocationCubeCreateFromBitmap }, {"rsnAllocationCopyFromBitmap", "(JJLandroid/graphics/Bitmap;)V", (void*)nAllocationCopyFromBitmap }, {"rsnAllocationCopyToBitmap", "(JJLandroid/graphics/Bitmap;)V", (void*)nAllocationCopyToBitmap }, {"rsnAllocationSyncAll", "(JJI)V", (void*)nAllocationSyncAll }, {"rsnAllocationSetSurface", "(JJLandroid/view/Surface;)V", (void*)nAllocationSetSurface }, {"rsnAllocationIoSend", "(JJ)V", (void*)nAllocationIoSend }, {"rsnAllocationData1D", "(JJIIILjava/lang/Object;IIIZ)V", (void*)nAllocationData1D }, {"rsnAllocationElementData1D", "(JJIII[BI)V", (void*)nAllocationElementData1D }, //{"rsnAllocationElementData", "(JJIIIII[BI)V", (void*)nAllocationElementData }, {"rsnAllocationData2D", "(JJIIIIIILjava/lang/Object;IIIZ)V", (void*)nAllocationData2D }, {"rsnAllocationData2D", "(JJIIIIIIJIIII)V", (void*)nAllocationData2D_alloc }, {"rsnAllocationData3D", "(JJIIIIIIILjava/lang/Object;IIIZ)V", (void*)nAllocationData3D }, {"rsnAllocationData3D", "(JJIIIIIIIJIIII)V", (void*)nAllocationData3D_alloc }, {"rsnAllocationRead", "(JJLjava/lang/Object;IIZ)V", (void*)nAllocationRead }, {"rsnAllocationRead1D", "(JJIIILjava/lang/Object;IIIZ)V", (void*)nAllocationRead1D }, //{"rsnAllocationElementRead", "(JJIIIII[BI)V", (void*)nAllocationElementRead }, {"rsnAllocationRead2D", "(JJIIIIIILjava/lang/Object;IIIZ)V", (void*)nAllocationRead2D }, //{"rsnAllocationRead3D", "(JJIIIIIIILjava/lang/Object;IIIZ)V", (void*)nAllocationRead3D }, {"rsnAllocationGetType", "(JJ)J", (void*)nAllocationGetType}, {"rsnAllocationResize1D", "(JJI)V", (void*)nAllocationResize1D }, {"rsnAllocationGenerateMipmaps", "(JJ)V", (void*)nAllocationGenerateMipmaps }, {"rsnScriptBindAllocation", "(JJJIZ)V", (void*)nScriptBindAllocation }, {"rsnScriptSetTimeZone", "(JJ[BZ)V", (void*)nScriptSetTimeZone }, {"rsnScriptInvoke", "(JJIZ)V", (void*)nScriptInvoke }, {"rsnScriptInvokeV", "(JJI[BZ)V", (void*)nScriptInvokeV }, {"rsnScriptForEach", "(JJJIJJZ)V", (void*)nScriptForEach }, {"rsnScriptForEach", "(JJJIJJ[BZ)V", (void*)nScriptForEachV }, {"rsnScriptForEach", "(JJI[JJ[B[I)V", (void*)nScriptForEachMulti }, {"rsnScriptForEachClipped", "(JJJIJJIIIIIIZ)V", (void*)nScriptForEachClipped }, {"rsnScriptForEachClipped", "(JJJIJJ[BIIIIIIZ)V", (void*)nScriptForEachClippedV }, {"rsnScriptReduce", "(JJI[JJ[I)V", (void*)nScriptReduce }, {"rsnScriptSetVarI", "(JJIIZ)V", (void*)nScriptSetVarI }, {"rsnScriptSetVarJ", "(JJIJZ)V", (void*)nScriptSetVarJ }, {"rsnScriptSetVarF", "(JJIFZ)V", (void*)nScriptSetVarF }, {"rsnScriptSetVarD", "(JJIDZ)V", (void*)nScriptSetVarD }, {"rsnScriptSetVarV", "(JJI[BZ)V", (void*)nScriptSetVarV }, {"rsnScriptSetVarVE", "(JJI[BJ[IZ)V", (void*)nScriptSetVarVE }, {"rsnScriptSetVarObj", "(JJIJZ)V", (void*)nScriptSetVarObj }, {"rsnScriptCCreate", "(JLjava/lang/String;Ljava/lang/String;[BI)J", (void*)nScriptCCreate }, {"rsnScriptIntrinsicCreate", "(JIJZ)J", (void*)nScriptIntrinsicCreate }, {"rsnScriptKernelIDCreate", "(JJIIZ)J", (void*)nScriptKernelIDCreate }, {"rsnScriptInvokeIDCreate", "(JJI)J", (void*)nScriptInvokeIDCreate }, {"rsnScriptFieldIDCreate", "(JJIZ)J", (void*)nScriptFieldIDCreate }, {"rsnScriptGroupCreate", "(J[J[J[J[J[J)J", (void*)nScriptGroupCreate }, {"rsnScriptGroup2Create", "(JLjava/lang/String;Ljava/lang/String;[J)J", (void*)nScriptGroup2Create }, {"rsnScriptGroupSetInput", "(JJJJ)V", (void*)nScriptGroupSetInput }, {"rsnScriptGroupSetOutput", "(JJJJ)V", (void*)nScriptGroupSetOutput }, {"rsnScriptGroupExecute", "(JJ)V", (void*)nScriptGroupExecute }, {"rsnScriptGroup2Execute", "(JJ)V", (void*)nScriptGroup2Execute }, {"rsnScriptIntrinsicBLAS_Single", "(JJJIIIIIIIIIFJJFJIIIIZ)V", (void*)nScriptIntrinsicBLAS_Single }, {"rsnScriptIntrinsicBLAS_Double", "(JJJIIIIIIIIIDJJDJIIIIZ)V", (void*)nScriptIntrinsicBLAS_Double }, {"rsnScriptIntrinsicBLAS_Complex", "(JJJIIIIIIIIIFFJJFFJIIIIZ)V", (void*)nScriptIntrinsicBLAS_Complex }, {"rsnScriptIntrinsicBLAS_Z", "(JJJIIIIIIIIIDDJJDDJIIIIZ)V", (void*)nScriptIntrinsicBLAS_Z }, {"rsnScriptIntrinsicBLAS_BNNM", "(JJJIIIJIJIJIIZ)V", (void*)nScriptIntrinsicBLAS_BNNM }, {"rsnSamplerCreate", "(JIIIIIF)J", (void*)nSamplerCreate }, {"rsnSystemGetPointerSize", "()I", (void*)nSystemGetPointerSize }, // Entry points for Inc libRSSupport {"nIncLoadSO", "(ILjava/lang/String;)Z", (bool*)nIncLoadSO }, {"nIncDeviceCreate", "()J", (void*)nIncDeviceCreate }, {"nIncDeviceDestroy", "(J)V", (void*)nIncDeviceDestroy }, {"rsnIncContextCreate", "(JIII)J", (void*)nIncContextCreate }, {"rsnIncContextFinish", "(J)V", (void*)nIncContextFinish }, {"rsnIncContextDestroy", "(J)V", (void*)nIncContextDestroy }, {"rsnIncObjDestroy", "(JJ)V", (void*)nIncObjDestroy }, {"rsnIncElementCreate", "(JJIZI)J", (void*)nIncElementCreate }, {"rsnIncTypeCreate", "(JJIIIZZI)J", (void*)nIncTypeCreate }, {"rsnIncAllocationCreateTyped", "(JJJJI)J", (void*)nIncAllocationCreateTyped }, {"rsnAllocationGetByteBuffer", "(JJIII)Ljava/nio/ByteBuffer;", (void*)nAllocationGetByteBuffer }, {"rsnAllocationGetStride", "(JJ)J", (void*)nAllocationGetStride }, }; // --------------------------------------------------------------------------- jint JNI_OnLoad(JavaVM* vm, void* reserved) { JNIEnv* env = NULL; jclass clazz = NULL; jint result = -1; if (vm->GetEnv((void**) &env, JNI_VERSION_1_4) != JNI_OK) { // __android_log_print(ANDROID_LOG_ERROR, LOG_TAG, // "ERROR: GetEnv failed\n"); goto bail; } if (env == NULL) { // __android_log_print(ANDROID_LOG_ERROR, LOG_TAG, "ERROR: env == NULL"); goto bail; } clazz = env->FindClass(classPathName); if (clazz == NULL) { goto bail; } if (env->RegisterNatives(clazz, methods, NELEM(methods)) < 0) { // __android_log_print(ANDROID_LOG_ERROR, LOG_TAG, // "ERROR: MediaPlayer native registration failed\n"); goto bail; } /* success -- return valid version number */ result = JNI_VERSION_1_4; bail: return result; }