|
@@ -0,0 +1,257 @@
|
|
|
+/*
|
|
|
+ * Copyright (c) 2009-2012 jMonkeyEngine
|
|
|
+ * All rights reserved.
|
|
|
+ *
|
|
|
+ * Redistribution and use in source and binary forms, with or without
|
|
|
+ * modification, are permitted provided that the following conditions are
|
|
|
+ * met:
|
|
|
+ *
|
|
|
+ * * Redistributions of source code must retain the above copyright
|
|
|
+ * notice, this list of conditions and the following disclaimer.
|
|
|
+ *
|
|
|
+ * * Redistributions in binary form must reproduce the above copyright
|
|
|
+ * notice, this list of conditions and the following disclaimer in the
|
|
|
+ * documentation and/or other materials provided with the distribution.
|
|
|
+ *
|
|
|
+ * * Neither the name of 'jMonkeyEngine' nor the names of its contributors
|
|
|
+ * may be used to endorse or promote products derived from this software
|
|
|
+ * without specific prior written permission.
|
|
|
+ *
|
|
|
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
|
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
|
|
|
+ * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
|
|
+ * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
|
|
|
+ * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
|
|
+ * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
|
|
+ * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
|
|
+ * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
|
|
+ * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
|
|
+ * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
|
+ * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
+ */
|
|
|
+
|
|
|
+package jme3test.opencl;
|
|
|
+
|
|
|
+import com.jme3.app.SimpleApplication;
|
|
|
+import com.jme3.font.BitmapFont;
|
|
|
+import com.jme3.font.BitmapText;
|
|
|
+import com.jme3.math.ColorRGBA;
|
|
|
+import com.jme3.opencl.*;
|
|
|
+import com.jme3.system.AppSettings;
|
|
|
+import com.jme3.util.BufferUtils;
|
|
|
+import java.nio.*;
|
|
|
+import java.util.Arrays;
|
|
|
+import java.util.Objects;
|
|
|
+import java.util.Random;
|
|
|
+import java.util.logging.Level;
|
|
|
+import java.util.logging.Logger;
|
|
|
+
|
|
|
+/**
|
|
|
+ * Test class for the build in libraries
|
|
|
+ * @author shaman
|
|
|
+ */
|
|
|
+public class TestOpenCLLibraries extends SimpleApplication {
|
|
|
+ private static final Logger LOG = Logger.getLogger(TestOpenCLLibraries.class.getName());
|
|
|
+
|
|
|
+ public static void main(String[] args){
|
|
|
+ TestOpenCLLibraries app = new TestOpenCLLibraries();
|
|
|
+ AppSettings settings = new AppSettings(true);
|
|
|
+ settings.setOpenCLSupport(true);
|
|
|
+ settings.setVSync(true);
|
|
|
+// settings.setRenderer(AppSettings.JOGL_OPENGL_FORWARD_COMPATIBLE);
|
|
|
+ app.setSettings(settings);
|
|
|
+ app.start(); // start the game
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public void simpleInitApp() {
|
|
|
+ BitmapFont fnt = assetManager.loadFont("Interface/Fonts/Default.fnt");
|
|
|
+ Context clContext = context.getOpenCLContext();
|
|
|
+ if (clContext == null) {
|
|
|
+ BitmapText txt = new BitmapText(fnt);
|
|
|
+ txt.setText("No OpenCL Context created!\nSee output log for details.");
|
|
|
+ txt.setLocalTranslation(5, settings.getHeight() - 5, 0);
|
|
|
+ guiNode.attachChild(txt);
|
|
|
+ return;
|
|
|
+ }
|
|
|
+ CommandQueue clQueue = clContext.createQueue(clContext.getDevices().get(0));
|
|
|
+
|
|
|
+ StringBuilder str = new StringBuilder();
|
|
|
+ str.append("OpenCL Context created:\n Platform: ")
|
|
|
+ .append(clContext.getDevices().get(0).getPlatform().getName())
|
|
|
+ .append("\n Devices: ").append(clContext.getDevices());
|
|
|
+ str.append("\nTests:");
|
|
|
+ str.append("\n Random numbers: ").append(testRandom(clContext, clQueue));
|
|
|
+
|
|
|
+ clQueue.release();
|
|
|
+
|
|
|
+ BitmapText txt1 = new BitmapText(fnt);
|
|
|
+ txt1.setText(str.toString());
|
|
|
+ txt1.setLocalTranslation(5, settings.getHeight() - 5, 0);
|
|
|
+ guiNode.attachChild(txt1);
|
|
|
+
|
|
|
+ flyCam.setEnabled(false);
|
|
|
+ inputManager.setCursorVisible(true);
|
|
|
+ }
|
|
|
+
|
|
|
+ private static void assertEquals(byte expected, byte actual, String message) {
|
|
|
+ if (expected != actual) {
|
|
|
+ System.err.println(message+": expected="+expected+", actual="+actual);
|
|
|
+ throw new AssertionError();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ private static void assertEquals(long expected, long actual, String message) {
|
|
|
+ if (expected != actual) {
|
|
|
+ System.err.println(message+": expected="+expected+", actual="+actual);
|
|
|
+ throw new AssertionError();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ private static void assertEquals(double expected, double actual, String message) {
|
|
|
+ if (Math.abs(expected - actual) >= 0.00001) {
|
|
|
+ System.err.println(message+": expected="+expected+", actual="+actual);
|
|
|
+ throw new AssertionError();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ private static void assertEquals(Object expected, Object actual, String message) {
|
|
|
+ if (!Objects.equals(expected, actual)) {
|
|
|
+ System.err.println(message+": expected="+expected+", actual="+actual);
|
|
|
+ throw new AssertionError();
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ private boolean testRandom(Context clContext, CommandQueue clQueue) {
|
|
|
+ try {
|
|
|
+ //test for doubles
|
|
|
+ boolean supportsDoubles = clContext.getDevices().get(0).hasDouble();
|
|
|
+
|
|
|
+ //create code
|
|
|
+ String code = ""
|
|
|
+ + "#import \"Common/OpenCL/Random.clh\"\n"
|
|
|
+ + "__kernel void TestBool(__global ulong* seeds, __global uchar* results) {\n"
|
|
|
+ + " results[get_global_id(0)] = randBool(seeds + get_global_id(0)) ? 1 : 0;\n"
|
|
|
+ + "}\n"
|
|
|
+ + "__kernel void TestInt(__global ulong* seeds, __global int* results) {\n"
|
|
|
+ + " results[get_global_id(0)] = randInt(seeds + get_global_id(0));\n"
|
|
|
+ + "}\n"
|
|
|
+ + "__kernel void TestIntN(__global ulong* seeds, int n, __global int* results) {\n"
|
|
|
+ + " results[get_global_id(0)] = randIntN(n, seeds + get_global_id(0));\n"
|
|
|
+ + "}\n"
|
|
|
+ + "__kernel void TestLong(__global ulong* seeds, __global long* results) {\n"
|
|
|
+ + " results[get_global_id(0)] = randLong(seeds + get_global_id(0));\n"
|
|
|
+ + "}\n"
|
|
|
+ + "__kernel void TestFloat(__global ulong* seeds, __global float* results) {\n"
|
|
|
+ + " results[get_global_id(0)] = randFloat(seeds + get_global_id(0));\n"
|
|
|
+ + "}\n"
|
|
|
+ + "#ifdef RANDOM_DOUBLES\n"
|
|
|
+ + "__kernel void TestDouble(__global ulong* seeds, __global double* results) {\n"
|
|
|
+ + " results[get_global_id(0)] = randDouble(seeds + get_global_id(0));\n"
|
|
|
+ + "}\n"
|
|
|
+ + "#endif\n";
|
|
|
+ if (supportsDoubles) {
|
|
|
+ code = "#define RANDOM_DOUBLES\n" + code;
|
|
|
+ }
|
|
|
+ Program program = clContext.createProgramFromSourceCodeWithDependencies(code, assetManager);
|
|
|
+ program.build();
|
|
|
+
|
|
|
+ int count = 256;
|
|
|
+ Kernel.WorkSize ws = new Kernel.WorkSize(count);
|
|
|
+
|
|
|
+ //create seeds
|
|
|
+ Random initRandom = new Random();
|
|
|
+ long[] seeds = new long[count];
|
|
|
+ Random[] randoms = new Random[count];
|
|
|
+ for (int i=0; i<count; ++i) {
|
|
|
+ seeds[i] = initRandom.nextLong();
|
|
|
+ randoms[i] = new Random(seeds[i]);
|
|
|
+ seeds[i] = (seeds[i] ^ 0x5DEECE66DL) & ((1L << 48) - 1); //needed because the Random constructor scrambles the initial seed
|
|
|
+ }
|
|
|
+ com.jme3.opencl.Buffer seedsBuffer = clContext.createBuffer(8 * count);
|
|
|
+ ByteBuffer tmpByteBuffer = BufferUtils.createByteBuffer(8 * count);
|
|
|
+ tmpByteBuffer.asLongBuffer().put(seeds);
|
|
|
+ seedsBuffer.write(clQueue, tmpByteBuffer);
|
|
|
+
|
|
|
+ //test it
|
|
|
+ ByteBuffer resultByteBuffer = BufferUtils.createByteBuffer(8 * count);
|
|
|
+ IntBuffer resultIntBuffer = resultByteBuffer.asIntBuffer();
|
|
|
+ LongBuffer resultLongBuffer = resultByteBuffer.asLongBuffer();
|
|
|
+ FloatBuffer resultFloatBuffer = resultByteBuffer.asFloatBuffer();
|
|
|
+ DoubleBuffer resultDoubleBuffer = resultByteBuffer.asDoubleBuffer();
|
|
|
+ com.jme3.opencl.Buffer resultBuffer = clContext.createBuffer(8 * count);
|
|
|
+ //boolean
|
|
|
+ Kernel testBoolKernel = program.createKernel("TestBool");
|
|
|
+ testBoolKernel.Run1NoEvent(clQueue, ws, seedsBuffer, resultBuffer);
|
|
|
+ resultByteBuffer.rewind();
|
|
|
+ resultBuffer.read(clQueue, resultByteBuffer);
|
|
|
+ for (int i=0; i<count; ++i) {
|
|
|
+ assertEquals(randoms[i].nextBoolean() ? 1 : 0, resultByteBuffer.get(i), "randBool at i="+i);
|
|
|
+ }
|
|
|
+ testBoolKernel.release();
|
|
|
+ //int
|
|
|
+ Kernel testIntKernel = program.createKernel("TestInt");
|
|
|
+ testIntKernel.Run1NoEvent(clQueue, ws, seedsBuffer, resultBuffer);
|
|
|
+ resultByteBuffer.rewind();
|
|
|
+ resultBuffer.read(clQueue, resultByteBuffer);
|
|
|
+ for (int i=0; i<count; ++i) {
|
|
|
+ assertEquals(randoms[i].nextInt(), resultIntBuffer.get(i), "randInt at i="+i);
|
|
|
+ }
|
|
|
+ testIntKernel.release();
|
|
|
+ //int n
|
|
|
+ Kernel testIntNKernel = program.createKernel("TestIntN");
|
|
|
+ testIntNKernel.Run1NoEvent(clQueue, ws, seedsBuffer, 186, resultBuffer);
|
|
|
+ resultByteBuffer.rewind();
|
|
|
+ resultBuffer.read(clQueue, resultByteBuffer);
|
|
|
+ for (int i=0; i<count; ++i) {
|
|
|
+ assertEquals(randoms[i].nextInt(186), resultIntBuffer.get(i), "randInt at i="+i+" with n="+186);
|
|
|
+ }
|
|
|
+ testIntNKernel.Run1NoEvent(clQueue, ws, seedsBuffer, 97357, resultBuffer);
|
|
|
+ resultByteBuffer.rewind();
|
|
|
+ resultBuffer.read(clQueue, resultByteBuffer);
|
|
|
+ for (int i=0; i<count; ++i) {
|
|
|
+ assertEquals(randoms[i].nextInt(97357), resultIntBuffer.get(i), "randInt at i="+i+" with n="+97357);
|
|
|
+ }
|
|
|
+ testIntNKernel.release();
|
|
|
+ //long
|
|
|
+ Kernel testLongKernel = program.createKernel("TestLong");
|
|
|
+ testLongKernel.Run1NoEvent(clQueue, ws, seedsBuffer, resultBuffer);
|
|
|
+ resultByteBuffer.rewind();
|
|
|
+ resultBuffer.read(clQueue, resultByteBuffer);
|
|
|
+ for (int i=0; i<count; ++i) {
|
|
|
+ assertEquals(randoms[i].nextLong(), resultLongBuffer.get(i), "randLong at i="+i);
|
|
|
+ }
|
|
|
+ testLongKernel.release();
|
|
|
+ //float
|
|
|
+ Kernel testFloatKernel = program.createKernel("TestFloat");
|
|
|
+ testFloatKernel.Run1NoEvent(clQueue, ws, seedsBuffer, resultBuffer);
|
|
|
+ resultByteBuffer.rewind();
|
|
|
+ resultBuffer.read(clQueue, resultByteBuffer);
|
|
|
+ for (int i=0; i<count; ++i) {
|
|
|
+ assertEquals(randoms[i].nextFloat(), resultFloatBuffer.get(i), "randFloat at i="+i);
|
|
|
+ }
|
|
|
+ testFloatKernel.release();
|
|
|
+ //double
|
|
|
+ if (supportsDoubles) {
|
|
|
+ Kernel testDoubleKernel = program.createKernel("TestDouble");
|
|
|
+ testDoubleKernel.Run1NoEvent(clQueue, ws, seedsBuffer, resultBuffer);
|
|
|
+ resultByteBuffer.rewind();
|
|
|
+ resultBuffer.read(clQueue, resultByteBuffer);
|
|
|
+ for (int i=0; i<count; ++i) {
|
|
|
+ assertEquals(randoms[i].nextDouble(), resultDoubleBuffer.get(i), "randLong at i="+i);
|
|
|
+ }
|
|
|
+ testDoubleKernel.release();
|
|
|
+ }
|
|
|
+
|
|
|
+ seedsBuffer.release();
|
|
|
+ resultBuffer.release();
|
|
|
+ program.release();
|
|
|
+
|
|
|
+ } catch (AssertionError ex) {
|
|
|
+ LOG.log(Level.SEVERE, "kernel test failed with an assertion error");
|
|
|
+ return false;
|
|
|
+ } catch (Exception ex) {
|
|
|
+ LOG.log(Level.SEVERE, "kernel test failed with:", ex);
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+ return true;
|
|
|
+ }
|
|
|
+
|
|
|
+}
|