diff options
Diffstat (limited to 'source/blender/compositor/operations/COM_WriteBufferOperation.cpp')
-rw-r--r-- | source/blender/compositor/operations/COM_WriteBufferOperation.cpp | 161 |
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; +} |