diff options
Diffstat (limited to 'source/blender/compositor/operations/COM_GaussianBokehBlurOperation.cpp')
-rw-r--r-- | source/blender/compositor/operations/COM_GaussianBokehBlurOperation.cpp | 174 |
1 files changed, 174 insertions, 0 deletions
diff --git a/source/blender/compositor/operations/COM_GaussianBokehBlurOperation.cpp b/source/blender/compositor/operations/COM_GaussianBokehBlurOperation.cpp new file mode 100644 index 00000000000..9b7df61143e --- /dev/null +++ b/source/blender/compositor/operations/COM_GaussianBokehBlurOperation.cpp @@ -0,0 +1,174 @@ +/* + * 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_GaussianBokehBlurOperation.h" +#include "BLI_math.h" + +extern "C" { + #include "RE_pipeline.h" +} + +GaussianBokehBlurOperation::GaussianBokehBlurOperation(): BlurBaseOperation() { + this->gausstab = NULL; +} + +void* GaussianBokehBlurOperation::initializeTileData(rcti *rect, MemoryBuffer **memoryBuffers) { + updateGauss(memoryBuffers); + void* buffer = getInputOperation(0)->initializeTileData(NULL, memoryBuffers); + return buffer; +} + +void GaussianBokehBlurOperation::updateGauss(MemoryBuffer **memoryBuffers) { + if (this->gausstab == NULL) { + float radxf; + float radyf; + int n; + float* dgauss; + float* ddgauss; + float val; + int j, i; + const float width = this->getWidth(); + const float height = this->getHeight(); + updateSize(memoryBuffers); + + radxf = size*(float)this->data->sizex; + if(radxf>width/2.0f) + radxf= width/2.0f; + else if(radxf<1.0f) + radxf= 1.0f; + + /* vertical */ + radyf = size*(float)this->data->sizey; + if(radyf>height/2.0f) + radyf= height/2.0f; + else if(radyf<1.0f) + radyf= 1.0f; + + radx= ceil(radxf); + rady= ceil(radyf); + + n = (2*radx+1)*(2*rady+1); + + /* create a full filter image */ + ddgauss= new float[n]; + dgauss= ddgauss; + val= 0.0f; + for(j=-rady; j<=rady; j++) { + for(i=-radx; i<=radx; i++, dgauss++) { + float fj= (float)j/radyf; + float fi= (float)i/radxf; + float dist= sqrt(fj*fj + fi*fi); + *dgauss= RE_filter_value(this->data->filtertype, dist); + + val+= *dgauss; + } + } + if(val!=0.0f) { + val= 1.0f/val; + for(j= n -1; j>=0; j--) + ddgauss[j]*= val; + } + else ddgauss[4]= 1.0f; + + gausstab = ddgauss; + } +} + +void GaussianBokehBlurOperation::executePixel(float* color, int x, int y, MemoryBuffer *inputBuffers[], void* data) { + float tempColor[4]; + tempColor[0] = 0; + tempColor[1] = 0; + tempColor[2] = 0; + tempColor[3] = 0; + float overallmultiplyer = 0; + MemoryBuffer* inputBuffer = (MemoryBuffer*)data; + float* buffer = inputBuffer->getBuffer(); + int bufferwidth = inputBuffer->getWidth(); + int bufferstartx = inputBuffer->getRect()->xmin; + int bufferstarty = inputBuffer->getRect()->ymin; + + int miny = y - this->rady; + int maxy = y + this->rady; + int minx = x - this->radx; + int maxx = x + this->radx; + 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 index = 0; + int step = QualityStepHelper::getStep(); + int offsetadd = QualityStepHelper::getOffsetAdd(); + 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 multiplyer = gausstab[index]; + tempColor[0] += multiplyer * buffer[bufferindex]; + tempColor[1] += multiplyer * buffer[bufferindex+1]; + tempColor[2] += multiplyer * buffer[bufferindex+2]; + tempColor[3] += multiplyer * buffer[bufferindex+3]; + overallmultiplyer += multiplyer; + index += step; + bufferindex +=offsetadd; + } + } + float divider = 1.0/overallmultiplyer; + color[0] = tempColor[0]*divider; + color[1] = tempColor[1]*divider; + color[2] = tempColor[2]*divider; + color[3] = tempColor[3]*divider; +} + +void GaussianBokehBlurOperation::deinitExecution() { + BlurBaseOperation::deinitExecution(); + delete this->gausstab; + this->gausstab = NULL; +} + +bool GaussianBokehBlurOperation::determineDependingAreaOfInterest(rcti *input, ReadBufferOperation *readOperation, rcti *output) { + rcti newInput; + rcti sizeInput; + sizeInput.xmin = 0; + sizeInput.ymin = 0; + sizeInput.xmax = 5; + sizeInput.ymax = 5; + NodeOperation * operation = this->getInputOperation(1); + + if (operation->determineDependingAreaOfInterest(&sizeInput, readOperation, output)) { + return true; + } else { + if (this->gausstab) { + int addx = radx; + int addy = rady; + newInput.xmax = input->xmax + addx; + newInput.xmin = input->xmin - addx; + newInput.ymax = input->ymax + addy; + newInput.ymin = input->ymin - addy; + } else { + newInput.xmin = 0; + newInput.ymin = 0; + newInput.xmax = this->getWidth(); + newInput.ymax = this->getHeight(); + } + return BlurBaseOperation::determineDependingAreaOfInterest(&newInput, readOperation, output); + } +} |