diff options
Diffstat (limited to 'source/blender/compositor/operations/COM_BokehBlurOperation.cpp')
-rw-r--r-- | source/blender/compositor/operations/COM_BokehBlurOperation.cpp | 159 |
1 files changed, 159 insertions, 0 deletions
diff --git a/source/blender/compositor/operations/COM_BokehBlurOperation.cpp b/source/blender/compositor/operations/COM_BokehBlurOperation.cpp new file mode 100644 index 00000000000..31345dbaeff --- /dev/null +++ b/source/blender/compositor/operations/COM_BokehBlurOperation.cpp @@ -0,0 +1,159 @@ +/* + * 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_BokehBlurOperation.h" +#include "BLI_math.h" + +extern "C" { + #include "RE_pipeline.h" +} + +BokehBlurOperation::BokehBlurOperation() : NodeOperation() { + this->addInputSocket(COM_DT_COLOR); + this->addInputSocket(COM_DT_COLOR, COM_SC_NO_RESIZE); + this->addInputSocket(COM_DT_VALUE); + this->addOutputSocket(COM_DT_COLOR); + this->setComplex(true); + + this->size = .01; + + this->inputProgram = NULL; + this->inputBokehProgram = NULL; + this->inputBoundingBoxReader = NULL; +} + +void* BokehBlurOperation::initializeTileData(rcti *rect, MemoryBuffer **memoryBuffers) { + void* buffer = getInputOperation(0)->initializeTileData(NULL, memoryBuffers); + return buffer; +} + +void BokehBlurOperation::initExecution() { + this->inputProgram = getInputSocketReader(0); + this->inputBokehProgram = getInputSocketReader(1); + this->inputBoundingBoxReader = getInputSocketReader(2); + + int width = inputBokehProgram->getWidth(); + int height = inputBokehProgram->getHeight(); + + float dimension; + if (width<height) { + dimension = width; + } else { + dimension = height; + } + this->bokehMidX = width/2.0f; + this->bokehMidY = height/2.0f; + this->bokehDimension = dimension/2.0f; + QualityStepHelper::initExecution(COM_QH_INCREASE); +} + +void BokehBlurOperation::executePixel(float* color, int x, int y, MemoryBuffer *inputBuffers[], void* data) { + float tempColor[4]; + float tempBoundingBox[4]; + float bokeh[4]; + + inputBoundingBoxReader->read(tempBoundingBox, x, y, COM_PS_NEAREST, inputBuffers); + if (tempBoundingBox[0] >0.0f) { + tempColor[0] = 0; + tempColor[1] = 0; + tempColor[2] = 0; + tempColor[3] = 0; + float overallmultiplyerr = 0; + float overallmultiplyerg = 0; + float overallmultiplyerb = 0; + MemoryBuffer* inputBuffer = (MemoryBuffer*)data; + float* buffer = inputBuffer->getBuffer(); + int bufferwidth = inputBuffer->getWidth(); + int bufferstartx = inputBuffer->getRect()->xmin; + int bufferstarty = inputBuffer->getRect()->ymin; + int pixelSize = this->size*this->getWidth(); + + int miny = y - pixelSize; + int maxy = y + pixelSize; + int minx = x - pixelSize; + int maxx = x + pixelSize; + miny = max(miny, inputBuffer->getRect()->ymin); + minx = max(minx, inputBuffer->getRect()->xmin); + maxy = min(maxy, inputBuffer->getRect()->ymax); + maxx = min(maxx, inputBuffer->getRect()->xmax); + + int step = getStep(); + int offsetadd = getOffsetAdd(); + + float m = this->bokehDimension/pixelSize; + for (int ny = miny ; ny < maxy ; ny +=step) { + int bufferindex = ((minx - bufferstartx)*4)+((ny-bufferstarty)*4*bufferwidth); + for (int nx = minx ; nx < maxx ; nx +=step) { + float u = this->bokehMidX - (nx-x) *m; + float v = this->bokehMidY - (ny-y) *m; + inputBokehProgram->read(bokeh, u, v, COM_PS_NEAREST, inputBuffers); + tempColor[0] += bokeh[0] * buffer[bufferindex]; + tempColor[1] += bokeh[1] * buffer[bufferindex+1]; + tempColor[2] += bokeh[2]* buffer[bufferindex+2]; + overallmultiplyerr += bokeh[0]; + overallmultiplyerg += bokeh[1]; + overallmultiplyerb += bokeh[2]; + bufferindex +=offsetadd; + } + } + color[0] = tempColor[0]*(1.0/overallmultiplyerr); + color[1] = tempColor[1]*(1.0/overallmultiplyerg); + color[2] = tempColor[2]*(1.0/overallmultiplyerb); + color[3] = 1.0f; + } else { + inputProgram->read(color, x, y, COM_PS_NEAREST, inputBuffers); + } +} + +void BokehBlurOperation::deinitExecution() { + this->inputProgram = NULL; + this->inputBokehProgram = NULL; + this->inputBoundingBoxReader = NULL; +} + +bool BokehBlurOperation::determineDependingAreaOfInterest(rcti *input, ReadBufferOperation *readOperation, rcti *output) { + rcti newInput; + rcti bokehInput; + + newInput.xmax = input->xmax + (size*this->getWidth()); + newInput.xmin = input->xmin - (size*this->getWidth()); + newInput.ymax = input->ymax + (size*this->getWidth()); + newInput.ymin = input->ymin - (size*this->getWidth()); + + NodeOperation* operation = getInputOperation(1); + bokehInput.xmax = operation->getWidth(); + bokehInput.xmin = 0; + bokehInput.ymax = operation->getHeight(); + bokehInput.ymin = 0; + if (operation->determineDependingAreaOfInterest(&bokehInput, readOperation, output) ) { + return true; + } + operation = getInputOperation(0); + if (operation->determineDependingAreaOfInterest(&newInput, readOperation, output) ) { + return true; + } + operation = getInputOperation(2); + if (operation->determineDependingAreaOfInterest(input, readOperation, output) ) { + return true; + } + return false; +} |