diff options
Diffstat (limited to 'src/com/mbien/opencl/demos/fractal/MultiDeviceFractal.java')
| -rw-r--r-- | src/com/mbien/opencl/demos/fractal/MultiDeviceFractal.java | 486 |
1 files changed, 0 insertions, 486 deletions
diff --git a/src/com/mbien/opencl/demos/fractal/MultiDeviceFractal.java b/src/com/mbien/opencl/demos/fractal/MultiDeviceFractal.java deleted file mode 100644 index 265add8..0000000 --- a/src/com/mbien/opencl/demos/fractal/MultiDeviceFractal.java +++ /dev/null @@ -1,486 +0,0 @@ -package com.mbien.opencl.demos.fractal; - -import com.mbien.opencl.CLBuffer; -import com.mbien.opencl.CLCommandQueue; -import com.mbien.opencl.CLDevice; -import com.mbien.opencl.CLEvent; -import com.mbien.opencl.CLEventList; -import com.mbien.opencl.CLException; -import com.mbien.opencl.gl.CLGLBuffer; -import com.mbien.opencl.gl.CLGLContext; -import com.mbien.opencl.CLKernel; -import com.mbien.opencl.CLPlatform; -import com.mbien.opencl.CLProgram; -import com.mbien.opencl.CLProgram.CompilerOptions; -import com.jogamp.opengl.util.awt.TextRenderer; -import java.awt.Color; -import java.awt.Dimension; -import java.awt.Font; -import java.awt.Point; -import java.awt.event.KeyAdapter; -import java.awt.event.KeyEvent; -import java.awt.event.MouseAdapter; -import java.awt.event.MouseEvent; -import java.awt.event.MouseWheelEvent; -import java.io.IOException; -import java.nio.IntBuffer; -import java.util.logging.Level; -import java.util.logging.Logger; -import javax.media.opengl.DebugGL2; -import javax.media.opengl.GL; -import javax.media.opengl.GL2; -import javax.media.opengl.GLAutoDrawable; -import javax.media.opengl.GLCapabilities; -import javax.media.opengl.GLContext; -import javax.media.opengl.GLEventListener; -import javax.media.opengl.GLProfile; -import javax.media.opengl.awt.GLCanvas; -import javax.swing.JFrame; -import javax.swing.SwingUtilities; - -import static com.jogamp.common.nio.Buffers.*; -import static javax.media.opengl.GL2.*; -import static com.mbien.opencl.CLMemory.Mem.*; -import static com.mbien.opencl.CLEvent.ProfilingCommand.*; -import static com.mbien.opencl.CLCommandQueue.Mode.*; -import static com.mbien.opencl.CLDevice.Type.*; -import static java.lang.Math.*; - -/** - * Computes the Mandelbrot set with OpenCL using multiple GPUs and renders the result with OpenGL. - * A shared PBO is used as storage for the fractal image.<br/> - * http://en.wikipedia.org/wiki/Mandelbrot_set - * <p> - * controls:<br/> - * keys 1-9 control parallelism level<br/> - * space enables/disables slice seperator<br/> - * 'd' toggles between 32/64bit floatingpoint precision<br/> - * mouse/mousewheel to drag and zoom<br/> - * </p> - * @author Michael Bien - */ -public class MultiDeviceFractal implements GLEventListener { - - // max number of used GPUs - private static final int MAX_PARRALLELISM_LEVEL = 8; - - // max per pixel iterations to compute the fractal - private static final int MAX_ITERATIONS = 500; - - private GLCanvas canvas; - - private CLGLContext clContext; - private CLCommandQueue[] queues; - private CLKernel[] kernels; - private CLProgram program; - private CLEventList probes; - private CLGLBuffer<?>[] pboBuffers; - private CLBuffer<IntBuffer>[] colorMap; - - private int width = 0; - private int height = 0; - - private double minX = -2f; - private double minY = -1.2f; - private double maxX = 0.6f; - private double maxY = 1.3f; - - private int slices; - - private boolean drawSeperator; - private boolean doublePrecision; - private boolean buffersInitialized; - private boolean rebuild; - - private final TextRenderer textRenderer; - - public MultiDeviceFractal(int width, int height) { - - this.width = width; - this.height = height; - - canvas = new GLCanvas(new GLCapabilities(GLProfile.get(GLProfile.GL2))); - canvas.addGLEventListener(this); - initSceneInteraction(); - - JFrame frame = new JFrame("JOCL Multi GPU Mandelbrot Set"); - frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE); - canvas.setPreferredSize(new Dimension(width, height)); - frame.add(canvas); - frame.pack(); - - frame.setVisible(true); - - textRenderer = new TextRenderer(frame.getFont().deriveFont(Font.BOLD, 14), true, true, null, false); - } - - public void init(GLAutoDrawable drawable) { - - if(clContext == null) { - // enable GL error checking using the composable pipeline - drawable.setGL(new DebugGL2(drawable.getGL().getGL2())); - - drawable.getGL().glFinish(); - initCL(drawable.getContext()); - - GL2 gl = drawable.getGL().getGL2(); - - gl.setSwapInterval(0); - gl.glDisable(GL_DEPTH_TEST); - gl.glClearColor(0.0f, 0.0f, 0.0f, 1.0f); - - initView(gl, drawable.getWidth(), drawable.getHeight()); - - initPBO(gl); - drawable.getGL().glFinish(); - - setKernelConstants(); - } - } - - private void initCL(GLContext glCtx){ - try { - // create context managing all available GPUs -// clContext = CLGLContext.create(glCtx, GPU); - clContext = CLGLContext.create(glCtx, CLPlatform.getDefault().listCLDevices()[0]); - - - CLDevice[] devices = clContext.getDevices(); - - slices = min(devices.length, MAX_PARRALLELISM_LEVEL); - - // create command queues for every GPU, setup colormap and init kernels - queues = new CLCommandQueue[slices]; - kernels = new CLKernel[slices]; - probes = new CLEventList(slices); - colorMap = new CLBuffer[slices]; - - for (int i = 0; i < slices; i++) { - - colorMap[i] = clContext.createIntBuffer(32*2, READ_ONLY); - initColorMap(colorMap[i].getBuffer(), 32, Color.BLUE, Color.GREEN, Color.RED); - - // create command queue and upload color map buffer on each used device - queues[i] = devices[i].createCommandQueue(PROFILING_MODE).putWriteBuffer(colorMap[i], true); // blocking upload - - } - - // load and build program - program = clContext.createProgram(getClass().getResourceAsStream("Mandelbrot.cl")); - buildProgram(); - - } catch (IOException ex) { - Logger.getLogger(getClass().getName()).log(Level.SEVERE, "can not find 'Mandelbrot.cl' in classpath.", ex); - } catch (CLException ex) { - Logger.getLogger(getClass().getName()).log(Level.SEVERE, "something went wrong, hopefully no one got hurt", ex); - } - - } - - private void initColorMap(IntBuffer colorMap, int stepSize, Color... colors) { - - for (int n = 0; n < colors.length - 1; n++) { - - Color color = colors[n]; - int r0 = color.getRed(); - int g0 = color.getGreen(); - int b0 = color.getBlue(); - - color = colors[n + 1]; - int r1 = color.getRed(); - int g1 = color.getGreen(); - int b1 = color.getBlue(); - - int deltaR = r1 - r0; - int deltaG = g1 - g0; - int deltaB = b1 - b0; - - for (int step = 0; step < stepSize; step++) { - float alpha = (float) step / (stepSize - 1); - int r = (int) (r0 + alpha * deltaR); - int g = (int) (g0 + alpha * deltaG); - int b = (int) (b0 + alpha * deltaB); - colorMap.put((r << 16) | (g << 8) | (b << 0)); - } - } - colorMap.rewind(); - - } - - private void initView(GL2 gl, int width, int height) { - - gl.glViewport(0, 0, width, height); - - gl.glMatrixMode(GL_MODELVIEW); - gl.glLoadIdentity(); - - gl.glMatrixMode(GL_PROJECTION); - gl.glLoadIdentity(); - gl.glOrtho(0.0, width, 0.0, height, 0.0, 1.0); - } - - @SuppressWarnings("unchecked") - private void initPBO(GL gl) { - - if(pboBuffers != null) { - int[] oldPbos = new int[pboBuffers.length]; - for (int i = 0; i < pboBuffers.length; i++) { - CLGLBuffer<?> buffer = pboBuffers[i]; - oldPbos[i] = buffer.GLID; - buffer.release(); - } - gl.glDeleteBuffers(oldPbos.length, oldPbos, 0); - } - - pboBuffers = new CLGLBuffer[slices]; - - int[] pbo = new int[slices]; - gl.glGenBuffers(slices, pbo, 0); - - // setup one empty PBO per slice - for (int i = 0; i < slices; i++) { - - gl.glBindBuffer(GL_PIXEL_UNPACK_BUFFER, pbo[i]); - gl.glBufferData(GL_PIXEL_UNPACK_BUFFER, width*height * SIZEOF_INT / slices, null, GL_STREAM_DRAW); - gl.glBindBuffer(GL_PIXEL_UNPACK_BUFFER, 0); - - pboBuffers[i] = clContext.createFromGLBuffer(pbo[i], WRITE_ONLY); - - } - - buffersInitialized = true; - } - - private void buildProgram() { - - /* - * workaround: The driver keeps using the old binaries for some reason. - * to solve this we simple create a new program and release the old. - * however rebuilding programs should be possible -> remove when drivers are fixed. - */ - if(program != null && rebuild) { - String source = program.getSource(); - program.release(); - program = clContext.createProgram(source); - } - - // disable 64bit floating point math if not available - if(doublePrecision) { - for (CLDevice device : program.getCLDevices()) { - if(!device.isDoubleFPAvailable()) { - doublePrecision = false; - break; - } - } - } - - if(doublePrecision) { - program.build(CompilerOptions.FAST_RELAXED_MATH, "-D DOUBLE_FP"); - }else{ - program.build(CompilerOptions.FAST_RELAXED_MATH); - } - rebuild = false; - - for (int i = 0; i < kernels.length; i++) { - // init kernel with constants - kernels[i] = program.createCLKernel("mandelbrot"); - } - - } - - // init kernels with constants - private void setKernelConstants() { - for (int i = 0; i < slices; i++) { - kernels[i].setForce32BitArgs(!doublePrecision) - .setArg(6, pboBuffers[i]) - .setArg(7, colorMap[i]) - .setArg(8, colorMap[i].getBuffer().capacity()) - .setArg(9, MAX_ITERATIONS); - } - } - - // rendering cycle - public void display(GLAutoDrawable drawable) { - GL gl = drawable.getGL(); - - // make sure GL does not use our objects before we start computeing - gl.glFinish(); - if(!buffersInitialized) { - initPBO(gl); - setKernelConstants(); - } - if(rebuild) { - buildProgram(); - setKernelConstants(); - } - compute(); - - render(gl.getGL2()); - } - - // OpenCL - private void compute() { - - int sliceWidth = width / slices; - double rangeX = (maxX - minX) / slices; - double rangeY = (maxY - minY); - - // release all old events, you can't reuse events in OpenCL - probes.release(); - - // start computation - for (int i = 0; i < slices; i++) { - - kernels[i].putArg( sliceWidth).putArg(height) - .putArg(minX + rangeX*i).putArg( minY) - .putArg( rangeX ).putArg(rangeY) - .rewind(); - - // aquire GL objects, and enqueue a kernel with a probe from the list - queues[i].putAcquireGLObject(pboBuffers[i].ID) - .put2DRangeKernel(kernels[i], 0, 0, sliceWidth, height, 0, 0, probes) - .putReleaseGLObject(pboBuffers[i].ID); - - } - - // block until done (important: finish before doing further gl work) - for (int i = 0; i < slices; i++) { - queues[i].finish(); - } - - } - - // OpenGL - private void render(GL2 gl) { - - gl.glClear(GL_COLOR_BUFFER_BIT); - - //draw slices - int sliceWidth = width / slices; - - for (int i = 0; i < slices; i++) { - - int seperatorOffset = drawSeperator?i:0; - - gl.glBindBuffer(GL_PIXEL_UNPACK_BUFFER, pboBuffers[i].GLID); - gl.glRasterPos2i(sliceWidth*i + seperatorOffset, 0); - - gl.glDrawPixels(sliceWidth, height, GL_BGRA, GL_UNSIGNED_BYTE, 0); - - } - gl.glBindBuffer(GL_PIXEL_UNPACK_BUFFER, 0); - - //draw info text - textRenderer.beginRendering(width, height, false); - - textRenderer.draw("precision: "+ (doublePrecision?"64bit":"32bit"), 10, height-15); - - for (int i = 0; i < slices; i++) { - CLEvent event = probes.getEvent(i); - long start = event.getProfilingInfo(START); - long end = event.getProfilingInfo(END); - textRenderer.draw("GPU"+i +" "+(int)((end-start)/1000000.0f)+"ms", 10, height-(20+16*(slices-i))); - } - - textRenderer.endRendering(); - } - - public void reshape(GLAutoDrawable drawable, int x, int y, int width, int height) { - - if(this.width == width && this.height == height) - return; - - this.width = width; - this.height = height; - - initPBO(drawable.getGL()); - - initView(drawable.getGL().getGL2(), drawable.getWidth(), drawable.getHeight()); - } - - private void initSceneInteraction() { - - MouseAdapter mouseAdapter = new MouseAdapter() { - - Point lastpos = new Point(); - - @Override - public void mouseDragged(MouseEvent e) { - - double offsetX = (lastpos.x - e.getX()) * (maxX - minX) / width; - double offsetY = (lastpos.y - e.getY()) * (maxY - minY) / height; - - minX += offsetX; - minY -= offsetY; - - maxX += offsetX; - maxY -= offsetY; - - lastpos = e.getPoint(); - - canvas.display(); - - } - - @Override - public void mouseMoved(MouseEvent e) { - lastpos = e.getPoint(); - } - - @Override - public void mouseWheelMoved(MouseWheelEvent e) { - float rotation = e.getWheelRotation() / 25.0f; - - double deltaX = rotation * (maxX - minX); - double deltaY = rotation * (maxY - minY); - - // offset for "zoom to cursor" - double offsetX = (e.getX() / (float)width - 0.5f) * deltaX * 2; - double offsetY = (e.getY() / (float)height- 0.5f) * deltaY * 2; - - minX += deltaX+offsetX; - minY += deltaY-offsetY; - - maxX +=-deltaX+offsetX; - maxY +=-deltaY-offsetY; - - canvas.display(); - } - }; - - KeyAdapter keyAdapter = new KeyAdapter() { - - @Override - public void keyPressed(KeyEvent e) { - if(e.getKeyCode() == KeyEvent.VK_SPACE) { - drawSeperator = !drawSeperator; - }else if(e.getKeyChar() > '0' && e.getKeyChar() < '9') { - int number = e.getKeyChar()-'0'; - slices = min(number, min(queues.length, MAX_PARRALLELISM_LEVEL)); - buffersInitialized = false; - }else if(e.getKeyCode() == KeyEvent.VK_D) { - doublePrecision = !doublePrecision; - rebuild = true; - } - canvas.display(); - } - - }; - - canvas.addMouseMotionListener(mouseAdapter); - canvas.addMouseWheelListener(mouseAdapter); - canvas.addKeyListener(keyAdapter); - } - - public void dispose(GLAutoDrawable drawable) { - } - - public static void main(String args[]) { - SwingUtilities.invokeLater(new Runnable() { - public void run() { - new MultiDeviceFractal(512, 512); - } - }); - } - -} |