|
@@ -0,0 +1,144 @@
|
|
|
|
+/*
|
|
|
|
+ * 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.material.Material;
|
|
|
|
+import com.jme3.math.ColorRGBA;
|
|
|
|
+import com.jme3.opencl.*;
|
|
|
|
+import com.jme3.scene.Geometry;
|
|
|
|
+import com.jme3.scene.VertexBuffer;
|
|
|
|
+import com.jme3.scene.shape.Box;
|
|
|
|
+import com.jme3.system.AppSettings;
|
|
|
|
+import java.util.logging.Logger;
|
|
|
|
+
|
|
|
|
+/**
|
|
|
|
+ * This test class tests the capability to read and modify an OpenGL vertex buffer.
|
|
|
|
+ *
|
|
|
|
+ * @author Sebastian Weiss
|
|
|
|
+ */
|
|
|
|
+public class TestVertexBufferSharing extends SimpleApplication {
|
|
|
|
+ private static final Logger LOG = Logger.getLogger(TestVertexBufferSharing.class.getName());
|
|
|
|
+
|
|
|
|
+ private int initCounter;
|
|
|
|
+ private Context clContext;
|
|
|
|
+ private CommandQueue clQueue;
|
|
|
|
+ private Geometry geom;
|
|
|
|
+ private Buffer buffer;
|
|
|
|
+ private Kernel kernel;
|
|
|
|
+ private WorkSize ws;
|
|
|
|
+ private float time;
|
|
|
|
+
|
|
|
|
+ public static void main(String[] args){
|
|
|
|
+ TestVertexBufferSharing app = new TestVertexBufferSharing();
|
|
|
|
+ AppSettings settings = new AppSettings(true);
|
|
|
|
+ settings.setOpenCLSupport(true);
|
|
|
|
+ settings.setVSync(false);
|
|
|
|
+ app.setSettings(settings);
|
|
|
|
+ app.start(); // start the game
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ @Override
|
|
|
|
+ public void simpleInitApp() {
|
|
|
|
+ initOpenCL1();
|
|
|
|
+
|
|
|
|
+ Box b = new Box(1, 1, 1); // create cube shape
|
|
|
|
+ geom = new Geometry("Box", b); // create cube geometry from the shape
|
|
|
|
+ Material mat = new Material(assetManager,
|
|
|
|
+ "Common/MatDefs/Misc/Unshaded.j3md"); // create a simple material
|
|
|
|
+ mat.setColor("Color", ColorRGBA.Blue); // set color of material to blue
|
|
|
|
+ geom.setMaterial(mat); // set the cube's material
|
|
|
|
+ rootNode.attachChild(geom); // make the cube appear in the scene
|
|
|
|
+
|
|
|
|
+ initCounter = 0;
|
|
|
|
+ time = 0;
|
|
|
|
+
|
|
|
|
+ flyCam.setDragToRotate(true);
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ @Override
|
|
|
|
+ public void simpleUpdate(float tpf) {
|
|
|
|
+ super.simpleUpdate(tpf);
|
|
|
|
+
|
|
|
|
+ if (initCounter < 2) {
|
|
|
|
+ initCounter++;
|
|
|
|
+ } else if (initCounter == 2) {
|
|
|
|
+ //when initCounter reaches 2, the scene was drawn once and the texture was uploaded to the GPU
|
|
|
|
+ //then we can bind the texture to OpenCL
|
|
|
|
+ initOpenCL2();
|
|
|
|
+ updateOpenCL(tpf);
|
|
|
|
+ initCounter = 3;
|
|
|
|
+ } else {
|
|
|
|
+ updateOpenCL(tpf);
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ private void initOpenCL1() {
|
|
|
|
+ clContext = context.getOpenCLContext();
|
|
|
|
+ clQueue = clContext.createQueue();
|
|
|
|
+ //create kernel
|
|
|
|
+ String source = ""
|
|
|
|
+ + "__kernel void ScaleKernel(__global float* vb, float scale)\n"
|
|
|
|
+ + "{\n"
|
|
|
|
+ + " int idx = get_global_id(0);\n"
|
|
|
|
+ + " float3 pos = vload3(idx, vb);\n"
|
|
|
|
+ + " pos *= scale;\n"
|
|
|
|
+ + " vstore3(pos, idx, vb);\n"
|
|
|
|
+ + "}\n";
|
|
|
|
+ Program program = clContext.createProgramFromSourceCode(source);
|
|
|
|
+ program.build();
|
|
|
|
+ kernel = program.createKernel("ScaleKernel");
|
|
|
|
+ }
|
|
|
|
+ private void initOpenCL2() {
|
|
|
|
+ //bind vertex buffer to OpenCL
|
|
|
|
+ VertexBuffer vb = geom.getMesh().getBuffer(VertexBuffer.Type.Position);
|
|
|
|
+ buffer = clContext.bindVertexBuffer(vb, MemoryAccess.READ_WRITE);
|
|
|
|
+ ws = new WorkSize(geom.getMesh().getVertexCount());
|
|
|
|
+ }
|
|
|
|
+ private void updateOpenCL(float tpf) {
|
|
|
|
+ //advect time
|
|
|
|
+ time += tpf;
|
|
|
|
+
|
|
|
|
+ //aquire resource
|
|
|
|
+ buffer.acquireBufferForSharingAsync(clQueue);
|
|
|
|
+ //no need to wait for the returned event, since the kernel implicitely waits for it (same command queue)
|
|
|
|
+
|
|
|
|
+ //execute kernel
|
|
|
|
+ float scale = (float) Math.pow(1.1, (1.0 - time%2) / 16.0);
|
|
|
|
+ kernel.Run1(clQueue, ws, buffer, scale);
|
|
|
|
+
|
|
|
|
+ //release resource
|
|
|
|
+ buffer.releaseBufferForSharingAsync(clQueue);
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+}
|