Adding a small example to the Java wrapper

pavanky · pavanky · commit 23250e669f48 · 2014-03-03T02:38:28.000-05:00
diff --git a/Makefile b/Makefile
@@ -33,14 +33,17 @@ else
 	AF_JAVA_LIB_EXT=$(patsubst %.so, %_cuda.so, $(AF_JAVA_LIB))
 endif
 
+run: all
+	AF_JAVA_PATH=$(AF_JAVA_PATH) make -C examples run
+
 cuda: all
 
 opencl: all
 
 all: $(AF_JAVA_JAR)
 
 $(AF_JAVA_JAR): $(AF_JAVA_LIB) $(AF_JAVA_CLASSES)
-	jar cfm $@ $(AF_JAVA_MANIFEST) $(AF_JAVA_CLASSES) $(AF_JAVA_LIB)
+	jar cfm $@ $(AF_JAVA_MANIFEST) $(AF_JAVA_CLASSES)
 
 %.class: %.java
 	javac $<
diff --git a/examples/.gitignore b/examples/.gitignore
@@ -1 +1,2 @@
 *.jar
+*.class
diff --git a/examples/HelloWorld.java b/examples/HelloWorld.java
@@ -0,0 +1,61 @@
+import java.util.Random;
+import com.arrayfire.Array;
+import com.arrayfire.Util;
+
+public class HelloWorld {
+    public static void main(String[] args) {
+        System.out.println("Demonstrating Array Addition using ArrayFire");
+
+        int[] dims;
+        int total = 1;
+        float[] left, right, res;
+        Random rand = new Random();
+        String str;
+
+        dims = new int[2];
+        dims[0] = 10;
+        dims[1] = 1;
+
+        for (int i = 0; i < dims.length; i++) {
+            total *= dims[i];
+        }
+
+        left = new float[total];
+        right = new float[total];
+
+        for (int i = 0; i < total; i++) {
+            left[i] = (float)i;
+            double tmp = Math.ceil(rand.nextDouble() * 10) / 10;
+            right[i] = (float)(tmp);
+        }
+
+        try {
+            // Send data to ArrayFire
+            Array A = new Array(dims, left);
+            Array B = new Array(dims, right);
+
+            // Create container for output
+            Array C = new Array();
+
+            // Do vector addition on the device
+            C = Array.add(A, B);
+
+            // Get result back to host memory
+            res = C.host();
+
+            for(int i = 0; i < total; i++) {
+                str = Integer.toString(i) + ". ";
+                float val = left[i] + right[i];
+                str = str + Float.toString(left[i] ) + " + ";
+                str = str + Float.toString(right[i]) + " = ";
+                str = str + Float.toString(res[i]);
+                System.out.println(str);
+            }
+
+        } catch (Exception e) {
+            System.out.println("Failed to use ArrayFire");
+            System.out.println(e.getMessage());
+        }
+
+    }
+}
diff --git a/examples/Makefile b/examples/Makefile
@@ -0,0 +1,20 @@
+ifeq ($(shell uname -m), x86_64)
+  LIB:=lib64
+else
+  LIB:=lib
+endif
+
+AF_JAVA_PATH?=../
+AF_JAVA_JAR?=$(AF_JAVA_PATH)/ArrayFire.jar
+AF_JAVA_LIB_PATH?=$(AF_JAVA_PATH)/$(LIB)
+
+CLASSES=$(patsubst %.java, %.class, $(shell ls *.java))
+BINS=$(patsubst %.class, %, $(CLASSES)) 
+
+run: $(BINS)
+
+%: %.class
+	LD_LIBRARY_PATH=$(AF_JAVA_LIB_PATH) java -cp .:$(AF_JAVA_JAR) $@
+
+%.class: %.java
+	javac -cp $(AF_JAVA_PATH)/ArrayFire.jar $<
