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;
+}