1 files changed, 145 insertions, 0 deletions
diff --git a/source/blender/compositor/operations/COM_DilateErodeOperation.cpp b/source/blender/compositor/operations/COM_DilateErodeOperation.cpp
new file mode 100644
index 00000000000..df3bf996c96
--- /dev/null
+++ b/source/blender/compositor/operations/COM_DilateErodeOperation.cpp
@@ -0,0 +1,145 @@
+/*
+ * 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_DilateErodeOperation.h"
+#include "BLI_math.h"
+
+DilateErodeOperation::DilateErodeOperation(): NodeOperation() {
+	this->addInputSocket(COM_DT_VALUE);
+	this->addOutputSocket(COM_DT_VALUE);
+	this->setComplex(true);
+	this->inputProgram = NULL;
+	this->inset = 0.0f;
+	this->_switch = 0.5f;
+	this->distance = 0.0f;
+}
+void DilateErodeOperation::initExecution() {
+	this->inputProgram = this->getInputSocketReader(0);
+	if (this->distance < 0.0f) {
+		this->scope = - this->distance + this->inset;
+	} else {
+		if (this->inset*2 > this->distance) {
+			this->scope = max(this->inset*2 - this->distance, this->distance);
+		} else {
+			this->scope = distance;
+		}
+	}
+	if (scope < 3) {
+		scope = 3;
+	}
+}
+
+void* DilateErodeOperation::initializeTileData(rcti *rect, MemoryBuffer **memoryBuffers) {
+	void* buffer = inputProgram->initializeTileData(NULL, memoryBuffers);
+	return buffer;
+}
+
+void DilateErodeOperation::executePixel(float* color, int x, int y, MemoryBuffer *inputBuffers[], void* data) {
+	float inputValue[4];
+	const float sw = this->_switch;
+	const float distance = this->distance;
+	float pixelvalue;
+	const float rd = scope * scope;
+	const float inset = this->inset;
+	float mindist = rd*2;
+
+	MemoryBuffer* inputBuffer = (MemoryBuffer*)data;
+	float* buffer = inputBuffer->getBuffer();
+	rcti* rect = inputBuffer->getRect();
+	const int minx = max(x - scope, rect->xmin);
+	const int miny = max(y - scope, rect->ymin);
+	const int maxx = min(x + scope, rect->xmax);
+	const int maxy = min(y + scope, rect->ymax);
+	const int bufferWidth = rect->xmax-rect->xmin;
+	int offset;
+
+	this->inputProgram->read(inputValue, x, y, inputBuffers, NULL);
+	if (inputValue[0]>sw) {
+		for (int yi = miny ; yi<maxy;yi++) {
+			offset = ((yi-rect->ymin)*bufferWidth+(minx-rect->xmin))*4;
+			for (int xi = minx ; xi<maxx;xi++) {
+				if (buffer[offset]<sw) {
+					const float dx = xi-x;
+					const float dy = yi-y;
+					const float dis = dx*dx+dy*dy;
+					mindist = min(mindist, dis);
+				}
+				offset +=4;
+			}
+		}
+		pixelvalue = -sqrtf(mindist);
+	} else {
+		for (int yi = miny ; yi<maxy;yi++) {
+			offset = ((yi-rect->ymin)*bufferWidth+(minx-rect->xmin))*4;
+			for (int xi = minx ; xi<maxx;xi++) {
+				if (buffer[offset]>sw) {
+					const float dx = xi-x;
+					const float dy = yi-y;
+					const float dis = dx*dx+dy*dy;
+					mindist = min(mindist, dis);
+				}
+				offset +=4;
+
+			}
+		}
+		pixelvalue = sqrtf(mindist);
+	}
+
+	if (distance > 0.0f) {
+		const float delta = distance - pixelvalue;
+		if (delta >= 0.0f) {
+			if (delta >= inset) {
+				color[0] = 1.0f;
+			} else {
+				color[0] = delta/inset;
+			}
+		} else {
+			color[0] = 0.0f;
+		}
+	} else {
+		const float delta = -distance+pixelvalue;
+		if (delta < 0.0f) {
+			if (delta < -inset) {
+				color[0] = 1.0f;
+			} else {
+				color[0] = (-delta)/inset;
+			}
+		} else {
+			color[0] = 0.0f;
+		}
+	}
+}
+
+void DilateErodeOperation::deinitExecution() {
+	this->inputProgram = NULL;
+}
+
+bool DilateErodeOperation::determineDependingAreaOfInterest(rcti *input, ReadBufferOperation *readOperation, rcti *output) {
+	rcti newInput;
+
+	newInput.xmax = input->xmax + scope;
+	newInput.xmin = input->xmin - scope;
+	newInput.ymax = input->ymax + scope;
+	newInput.ymin = input->ymin - scope;
+
+	return NodeOperation::determineDependingAreaOfInterest(&newInput, readOperation, output);
+}