parent
60f10bb604
commit
03282a7c86
@ -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); |
||||
} |
||||
|
||||
} |
Loading…
Reference in new issue