如何将 3D 元素传递给 JNI？

Question

我正在做一个项目，需要在 C++ 中使用 3D 数组进行一些计算。我需要将这个 3D 数组从 Java 传递到 C++，进行一些计算，然后返回它。我正在使用 JNI 并且对它很陌生，所以我不太了解。我正在尝试制作一个示例程序来测试它并将其用作参考。我已经将 2D 数组传递给 C++，但无法弄清楚 3D 部分。我使用 this solution 构建了 2D 部分。

这是我现在的 C++ 代码（适用于二维数组）：

float*** testFunction(float ***a)
{

printf("Hello from JNI!\n");
    printf("Point at %d,%d,%d is: %f\n", 1, 2, 2, a[1][2]);

    return a[1][2][2];
}
jfloat JNICALL Java_JNIArray_integrateWithTrapezoid(JNIEnv *env, jobject thisObj, jobjectArray jarr)
{
    int sizex = env->GetArrayLength(jarr);
    jfloatArray dim = (jfloatArray)env->GetObjectArrayElement(jarr, 0);
    int sizey = env->GetArrayLength(dim);
    float **localArray;

    localArray = new float *[sizex];

    for (int i = 0; i < sizex; i++)
    {
        jfloatArray oneDim = (jfloatArray)env->GetObjectArrayElement(jarr, i);
        jfloat *element = env->GetFloatArrayElements(oneDim, 0);
        localArray[i] = new float[sizey];

        for (int j = 0; j < sizey; j++)
        {
            localArray[i][j] = element[j];
        }
    }

    return trapezoidalintegral(localArray);
}

任何帮助都会很棒。我对此一无所知。如果我忘记提供任何需要的信息，请告诉我。

谢谢。

编辑：

感谢@Botje，这成功了：

void trapezoidalintegral(float ***a)
{

    printf("Hello from JNI!\n");
    printf("Point at %d, %d, %d is: %f\n", 1, 2, 2, a[1][2][2]);

    return;
}

float *thirdLevel(JNIEnv *env, jfloatArray arr)
{
    jsize len = env->GetArrayLength(arr);
    float *ret = new float[len];
    env->GetFloatArrayRegion(arr, 0, len, ret);
    return ret;
}

float **secondLevel(JNIEnv *env, jobjectArray arr)
{
    jsize len = env->GetArrayLength(arr);
    float **ret = new float *[len];
    for (int i = 0; i < len; i++)
    {
        jobject item = env->GetObjectArrayElement(arr, i);
        ret[i] = thirdLevel(env, (jfloatArray)item);
        env->DeleteLocalRef(item);
    }
    return ret;
}

float ***firstLevel(JNIEnv *env, jobjectArray arr)
{
    jsize len = env->GetArrayLength(arr);
    float ***ret = new float **[len];
    for (int i = 0; i < len; i++)
    {
        jobject item = env->GetObjectArrayElement(arr, i);
        ret[i] = secondLevel(env, (jobjectArray)item);
        env->DeleteLocalRef(item);
    }
    return ret;
}

JNIEXPORT jobjectArray JNICALL Java_JNIArray_integrateWithTrapezoid(JNIEnv *env, jobject thisObj, jobjectArray jarr)
{
    float ***returningArray;

    returningArray = firstLevel(env, jarr);

    trapezoidalintegral(returningArray);

    jclass *pClass;
    jclass cls1;
    jclass jcls1;
    jclass jcls2;
    jobject obj2;

    cls1 = env->GetObjectClass(thisObj);
    // jfieldID fid1 = env->GetFieldID(cls1, "ptr", "J");
    // pClass = (jclass *)env->GetLongField(thisObj, fid1);

    jcls1 = env->FindClass("[[F");
    jcls2 = env->FindClass("[F");

    jobjectArray array1 = env->NewObjectArray(3, jcls1, NULL);

    for (int i = 0; i < 3; i++)
    {
        jobjectArray array2 = env->NewObjectArray(3, jcls2, NULL);

        for (int j = 0; j < 3; j++)
        {
            jfloatArray array3 = env->NewFloatArray(3);
            env->SetFloatArrayRegion(array3, 0, 3, returningArray[i][j]);
            env->SetObjectArrayElement(array2, j, array3);
        }

        env->SetObjectArrayElement(array1, i, array2);
    }

    env->DeleteLocalRef(cls1);
    env->DeleteLocalRef(jcls1);
    env->DeleteLocalRef(jcls2);

    return array1;
}

Answer 1

就像你的数据一样构建你的程序：

float* thirdLevel(JNIEnv *env, jfloatArray arr) {
  jsize len = env->GetArrayLength(arr);
  float* ret = new float[len];
  env->GetFloatArrayRegion(arr, 0, len, ret);
  return ret;
}

float** secondLevel(JNIEnv *env, jobjectArray arr) {
  jsize len = env->GetArrayLength(arr);
  float** ret = new float*[len];
  for (int i = 0; i < len; i++) {
    jobject item = env->GetObjectArrayElement(arr, i);
    ret[i] = thirdLevel(env, (jfloatArray)item);
    env->DeleteLocalRef(item);
  }
  return ret;
}

float*** firstLevel(JNIEnv *env, jobjectArray arr) {
  jsize len = env->GetArrayLength(arr);
  float*** ret = new float**[len];
  for (int i = 0; i < len; i++) {
    jobject item = env->GetObjectArrayElement(arr, i);
    ret[i] = secondLevel(env, (jobjectArray)item);
    env->DeleteLocalRef(item);
  }
  return ret;
}