#include "jni_helper.h"

BEGIN_EXTERN_C

#define ARRAY_FUNC(FUNC) AF_MANGLE(Array, FUNC)

JNIEXPORT void JNICALL ARRAY_FUNC(destroyArray)(JNIEnv *env, jclass clazz, jlong ref)

{

AF_TO_JAVA(af_release_array(ARRAY(ref)));

}

JNIEXPORT jintArray JNICALL ARRAY_FUNC(getDims)(JNIEnv *env, jclass clazz, jlong ref)

{

jintArray result = env->NewIntArray(MaxDimSupported);

if (result == NULL) {

return NULL;

}

dim_t dims[4];

AF_TO_JAVA(af_get_dims(dims + 0,

dims + 1,

dims + 2,

dims + 3,

ARRAY(ref)));

jint* dimsf = env->GetIntArrayElements(result, 0);

for(int k=0; k<MaxDimSupported; ++k) {

dimsf[k] = dims[k];

}

env->ReleaseIntArrayElements(result, dimsf, 0);

return result;

}

JNIEXPORT jint JNICALL ARRAY_FUNC(getType)(JNIEnv *env, jclass clazz, jlong ref)

{

af_dtype ty = f32;

AF_TO_JAVA(af_get_type(&ty, ARRAY(ref)));

return ty;

}

END_EXTERN_C