1 files changed, 161 insertions, 0 deletions
diff --git a/source/blender/compositor/operations/COM_WriteBufferOperation.cpp b/source/blender/compositor/operations/COM_WriteBufferOperation.cpp
new file mode 100644
index 00000000000..44a769ee947
--- /dev/null
+++ b/source/blender/compositor/operations/COM_WriteBufferOperation.cpp
@@ -0,0 +1,161 @@
+/*
+ * Copyright 2011, Blender Foundation.
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * as published by the Free Software Foundation; either version 2
+ * of the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ *
+ * Contributor: 
+ *		Jeroen Bakker 
+ *		Monique Dewanchand
+ */
+
+#include "COM_WriteBufferOperation.h"
+#include "COM_defines.h"
+#include "COM_MemoryManager.h"
+#include <stdio.h>
+
+/// @TODO: writebuffers don't have an actual data type set.
+WriteBufferOperation::WriteBufferOperation() :NodeOperation() {
+	this->addInputSocket(COM_DT_COLOR);
+	this->memoryProxy = new MemoryProxy();
+	this->memoryProxy->setWriteBufferOperation(this);
+	this->memoryProxy->setExecutor(NULL);
+	this->tree = NULL;
+}
+WriteBufferOperation::~WriteBufferOperation() {
+	if (this->memoryProxy) {
+		delete this->memoryProxy;
+		this->memoryProxy = NULL;
+	}
+}
+
+void WriteBufferOperation::executePixel(float* color, float x, float y, PixelSampler sampler, MemoryBuffer *inputBuffers[]) {
+	input->read(color, x, y, sampler, inputBuffers);
+}
+void WriteBufferOperation::initExecution() {
+		this->input = this->getInputOperation(0);
+	MemoryManager::addMemoryProxy(this->memoryProxy);
+}
+void WriteBufferOperation::deinitExecution() {
+	this->input = NULL;
+}
+
+
+void WriteBufferOperation::executeRegion(rcti *rect, unsigned int tileNumber, MemoryBuffer** memoryBuffers) {
+	MemoryBuffer* memoryBuffer = MemoryManager::getMemoryBuffer(this->getMemoryProxy(), tileNumber);
+	float* buffer = memoryBuffer->getBuffer();
+	if (this->input->isComplex()) {
+		void* data = this->input->initializeTileData(rect, memoryBuffers);
+		int x1 = rect->xmin;
+		int y1 = rect->ymin;
+		int x2 = rect->xmax;
+		int y2 = rect->ymax;
+		int offset4 = 0;
+		int x;
+		int y;
+		bool breaked = false;
+		for (y = y1 ; y < y2 && (!breaked) ; y++) {
+			for (x = x1 ; x < x2; x++) {
+				input->read(&(buffer[offset4]), x, y, memoryBuffers, data);
+				offset4 +=4;
+
+			}
+			if (tree->test_break && tree->test_break(tree->tbh)) {
+				breaked = true;
+			}
+
+		}
+		if (data) {
+			this->input->deinitializeTileData(rect, memoryBuffers, data);
+			data = NULL;
+		}
+	} else {
+		int x1 = rect->xmin;
+		int y1 = rect->ymin;
+		int x2 = rect->xmax;
+		int y2 = rect->ymax;
+		int offset4 = 0;
+		int x;
+		int y;
+		bool breaked = false;
+		for (y = y1 ; y < y2 && (!breaked) ; y++) {
+			for (x = x1 ; x < x2 ; x++) {
+				input->read(&(buffer[offset4]), x, y, COM_PS_NEAREST, memoryBuffers);
+				offset4 +=4;
+			}
+			if (tree->test_break && tree->test_break(tree->tbh)) {
+				breaked = true;
+			}
+		}
+	}
+	memoryBuffer->setCreatedState();
+}
+
+void WriteBufferOperation::executeOpenCLRegion(cl_context context, cl_program program, cl_command_queue queue, rcti *rect, unsigned int chunkNumber, MemoryBuffer** inputMemoryBuffers) {
+	MemoryBuffer* outputMemoryBuffer = MemoryManager::getMemoryBuffer(this->getMemoryProxy(), chunkNumber);
+	float* outputFloatBuffer = outputMemoryBuffer->getBuffer();
+	cl_int error;
+	/*
+	  1. create cl_mem from outputbuffer
+	  2. call NodeOperation (input) executeOpenCLChunk(.....)
+	  3. schedule readback from opencl to main device (outputbuffer)
+	  4. schedule native callback
+
+	  note: list of cl_mem will be filled by 2, and needs to be cleaned up by 4
+	  */
+	// STEP 1
+	const unsigned int outputBufferWidth = outputMemoryBuffer->getWidth();
+	const unsigned int outputBufferHeight = outputMemoryBuffer->getHeight();
+
+	const cl_image_format imageFormat = {
+		CL_RGBA,
+		CL_FLOAT
+	};
+
+	cl_mem clOutputBuffer = clCreateImage2D(context, CL_MEM_WRITE_ONLY|CL_MEM_USE_HOST_PTR, &imageFormat, outputBufferWidth, outputBufferHeight, 0, outputFloatBuffer, &error);
+	if (error != CL_SUCCESS) { printf("CLERROR[%d]: %s\n", error, clewErrorString(error));	}
+	
+	// STEP 2
+	list<cl_mem> * clMemToCleanUp = new list<cl_mem>();
+	clMemToCleanUp->push_back(clOutputBuffer);
+	list<cl_kernel> *clKernelsToCleanUp = new list<cl_kernel>();
+
+	this->input->executeOpenCL(context, program, queue, outputMemoryBuffer, clOutputBuffer, inputMemoryBuffers, clMemToCleanUp, clKernelsToCleanUp);
+
+	// STEP 3
+
+	size_t origin[3] = {0,0,0};
+	size_t region[3] = {outputBufferWidth,outputBufferHeight,1};
+
+	error = clEnqueueBarrier(queue);
+	if (error != CL_SUCCESS) { printf("CLERROR[%d]: %s\n", error, clewErrorString(error));	}
+	error = clEnqueueReadImage(queue, clOutputBuffer, CL_TRUE, origin, region, 0, 0, outputFloatBuffer, 0, NULL, NULL);
+	if (error != CL_SUCCESS) { printf("CLERROR[%d]: %s\n", error, clewErrorString(error));	}
+	// STEP 4
+
+	while (clMemToCleanUp->size()>0) {
+		cl_mem mem = clMemToCleanUp->front();
+		error = clReleaseMemObject(mem);
+		if (error != CL_SUCCESS) { printf("CLERROR[%d]: %s\n", error, clewErrorString(error));	}
+		clMemToCleanUp->pop_front();
+	}
+
+	while (clKernelsToCleanUp->size()>0) {
+		cl_kernel kernel = clKernelsToCleanUp->front();
+		error = clReleaseKernel(kernel);
+		if (error != CL_SUCCESS) { printf("CLERROR[%d]: %s\n", error, clewErrorString(error));	}
+		clKernelsToCleanUp->pop_front();
+	}
+	delete clKernelsToCleanUp;
+}