Added wrapper for af_where in Algorithm

ShadyBoukhary · umar456 · commit b65f7a34ec0f · 2019-07-31T17:34:44.000-04:00
diff --git a/com/arrayfire/Algorithm.java b/com/arrayfire/Algorithm.java
@@ -15,6 +15,8 @@ class Algorithm {
 
   private native static long min(long a, int dim);
 
+  private native static long afWhere(long ref);
+
   // Scalar return operations
   public static double sumAll(Array a) throws Exception {
     return sumAll(a.ref);
@@ -54,4 +56,8 @@ public static void max(Array res, Array a) throws Exception {
   public static void min(Array res, Array a) throws Exception {
     min(res, a, 0);
   }
+
+  public static Array where(final Array in) throws Exception {
+      return new Array(afWhere(in.ref));
+  }
 }
diff --git a/src/algorithm.cpp b/src/algorithm.cpp
@@ -28,4 +28,10 @@ ARRAY_RET_OP_DEF(sum)
 ARRAY_RET_OP_DEF(max)
 ARRAY_RET_OP_DEF(min)
 
+JNIEXPORT jlong JNICALL ALGO_FUNC(afWhere)(JNIEnv *env, jclass clazz, jlong arr) {
+    af_array ret = 0;
+    AF_CHECK(af_where(&ret, ARRAY(arr)));
+    return JLONG(ret);
+}
+
 END_EXTERN_C
