1 files changed, 147 insertions, 0 deletions

diff --git a/source/blender/compositor/operations/COM_DisplaceOperation.cpp b/source/blender/compositor/operations/COM_DisplaceOperation.cpp
new file mode 100644
index 00000000000..b30ce5ac5b5
--- /dev/null
+++ b/source/blender/compositor/operations/COM_DisplaceOperation.cpp
@@ -0,0 +1,147 @@
+/*
+ * 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:
+ *		Dalai Felinto
+ */
+
+#include "COM_DisplaceOperation.h"
+#include "BLI_math.h"
+#include "BLI_utildefines.h"
+
+DisplaceOperation::DisplaceOperation(): NodeOperation() {
+	this->addInputSocket(COM_DT_COLOR);
+	this->addInputSocket(COM_DT_VECTOR);
+	this->addInputSocket(COM_DT_VALUE);
+	this->addInputSocket(COM_DT_VALUE);
+	this->addOutputSocket(COM_DT_COLOR);
+	this->setComplex(true);
+
+	this->inputColorProgram = NULL;
+	this->inputVectorProgram = NULL;
+	this->inputScaleXProgram = NULL;
+	this->inputScaleYProgram = NULL;
+}
+
+void DisplaceOperation::initExecution() {
+	this->inputColorProgram = this->getInputSocketReader(0);
+	this->inputVectorProgram = this->getInputSocketReader(1);
+	this->inputScaleXProgram = this->getInputSocketReader(2);
+	this->inputScaleYProgram = this->getInputSocketReader(3);
+
+	width_x4 = this->getWidth() * 4;
+	height_x4 = this->getHeight() * 4;
+}
+
+
+/* minimum distance (in pixels) a pixel has to be displaced
+ * in order to take effect */
+#define DISPLACE_EPSILON	0.01f
+
+void DisplaceOperation::executePixel(float* color, int x, int y, MemoryBuffer *inputBuffers[], void* data)
+{
+	float inVector[4];
+	float inScale[4];
+
+	float p_dx, p_dy;	/* main displacement in pixel space */
+	float d_dx, d_dy;
+	float dxt, dyt;
+	float u, v;
+
+	this->inputScaleXProgram->read(inScale, x, y, COM_PS_NEAREST, inputBuffers);
+	float xs = inScale[0];
+	this->inputScaleYProgram->read(inScale, x, y, COM_PS_NEAREST, inputBuffers);
+	float ys = inScale[0];
+
+	/* clamp x and y displacement to triple image resolution - 
+	 * to prevent hangs from huge values mistakenly plugged in eg. z buffers */
+	CLAMP(xs, -width_x4, width_x4);
+	CLAMP(ys, -height_x4, height_x4);
+
+	this->inputVectorProgram->read(inVector, x, y, COM_PS_NEAREST, inputBuffers);
+	p_dx = inVector[0] * xs;
+	p_dy = inVector[1] * ys;
+
+	/* displaced pixel in uv coords, for image sampling */
+	u = x - p_dx + 0.5f;
+	v = y - p_dy + 0.5f;
+
+	/* calc derivatives */
+	this->inputVectorProgram->read(inVector, x+1, y, COM_PS_NEAREST, inputBuffers);
+	d_dx = inVector[0] * xs;
+	this->inputVectorProgram->read(inVector, x, y+1, COM_PS_NEAREST, inputBuffers);
+	d_dy = inVector[0] * ys;
+
+	/* clamp derivatives to minimum displacement distance in UV space */
+	dxt = p_dx - d_dx;
+	dyt = p_dy - d_dy;
+
+	dxt = signf(dxt)*maxf(fabsf(dxt), DISPLACE_EPSILON)/this->getWidth();
+	dyt = signf(dyt)*maxf(fabsf(dyt), DISPLACE_EPSILON)/this->getHeight();
+
+	/* EWA filtering */
+	this->inputColorProgram->read(color, u, v, dxt, dyt, inputBuffers);
+}
+
+void DisplaceOperation::deinitExecution() {
+	this->inputColorProgram = NULL;
+	this->inputVectorProgram = NULL;
+	this->inputScaleXProgram = NULL;
+	this->inputScaleYProgram = NULL;
+}
+
+bool DisplaceOperation::determineDependingAreaOfInterest(rcti *input, ReadBufferOperation *readOperation, rcti *output) {
+	rcti colorInput;
+	rcti vectorInput;
+	NodeOperation *operation=NULL;
+
+	/* the vector buffer only needs a 2x2 buffer. The image needs whole buffer */
+	/* image */
+	operation = getInputOperation(0);
+	colorInput.xmax = operation->getWidth();
+	colorInput.xmin = 0;
+	colorInput.ymax = operation->getHeight();
+	colorInput.ymin = 0;
+	if (operation->determineDependingAreaOfInterest(&colorInput, readOperation, output)) {
+		return true;
+	}
+
+	/* vector */
+	operation = getInputOperation(1);
+	vectorInput.xmax = input->xmax + 2;
+	vectorInput.xmin = input->xmin;
+	vectorInput.ymax = input->ymax + 2;
+	vectorInput.ymin = input->ymin;
+	if (operation->determineDependingAreaOfInterest(&vectorInput, readOperation, output)) {
+		return true;
+	}
+
+	/* scale x */
+	operation = getInputOperation(2);
+	if (operation->determineDependingAreaOfInterest(input, readOperation, output) ) {
+		return true;
+	}
+
+	/* scale y */
+	operation = getInputOperation(3);
+	if (operation->determineDependingAreaOfInterest(input, readOperation, output) ) {
+		return true;
+	}
+
+	return false;
+}
+