From e76229ac4f95ac8c6e4e38e60689a09a091e4048 Mon Sep 17 00:00:00 2001
From: Mattias Andrée <maandree@kth.se>
Date: Sun, 2 Jul 2017 23:58:01 +0200
Subject: Add gradients
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Mattias Andrée <maandree@kth.se>
---
 src/blind-sinc-wave.c | 114 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 114 insertions(+)
 create mode 100644 src/blind-sinc-wave.c

(limited to 'src/blind-sinc-wave.c')

diff --git a/src/blind-sinc-wave.c b/src/blind-sinc-wave.c
new file mode 100644
index 0000000..02c053d
--- /dev/null
+++ b/src/blind-sinc-wave.c
@@ -0,0 +1,114 @@
+/* See LICENSE file for copyright and license details. */
+#include "common.h"
+
+USAGE("[-e] [theta0-stream]")
+
+static int equal = 0;
+
+
+#define PROCESS(TYPE, SUFFIX)\
+	static void\
+	process_##SUFFIX(struct stream *grad, struct stream *theta0)\
+	{\
+		size_t i, n, m = 0;\
+		TYPE *theta0xyza;\
+		TYPE x, theta0x = 0;\
+		TYPE y, theta0y = 0;\
+		TYPE z, theta0z = 0;\
+		TYPE a, theta0a = 0;\
+		echeck_dimensions(grad, WIDTH | HEIGHT, NULL);\
+		do {\
+			if (!m) {\
+				m = grad->frame_size;\
+				if (theta0) {\
+					while (theta0->ptr < theta0->frame_size)\
+						if (!eread_stream(theta0, theta0->frame_size - theta0->ptr))\
+							return;\
+					theta0xyza = (TYPE *)theta0->buf;\
+					theta0x = (theta0xyza)[0];\
+					theta0y = (theta0xyza)[1];\
+					theta0z = (theta0xyza)[2];\
+					theta0a = (theta0xyza)[3];\
+					memmove(theta0->buf, theta0->buf + theta0->frame_size,\
+						theta0->ptr -= theta0->frame_size);\
+				}\
+			}\
+			n = MIN(grad->ptr, m) / grad->pixel_size;\
+			if (equal) {\
+				for (i = 0; i < n; i++) {\
+					a = ((TYPE *)(grad->buf))[4 * i + 3];\
+					a = (a ? sin(a + theta0y) / a : sin(a + theta0y)) / 2 + 0.5;\
+					((TYPE *)(grad->buf))[4 * i + 0] = a;\
+					((TYPE *)(grad->buf))[4 * i + 1] = a;\
+					((TYPE *)(grad->buf))[4 * i + 2] = a;\
+					((TYPE *)(grad->buf))[4 * i + 3] = a;\
+				}\
+			} else {\
+				for (i = 0; i < n; i++) {\
+					x = ((TYPE *)(grad->buf))[4 * i + 0];\
+					y = ((TYPE *)(grad->buf))[4 * i + 1];\
+					z = ((TYPE *)(grad->buf))[4 * i + 2];\
+					a = ((TYPE *)(grad->buf))[4 * i + 3];\
+					x = (x ? sin(x + theta0x) / x : sin(x + theta0x)) / 2 + 0.5;\
+					y = (y ? sin(y + theta0y) / y : sin(y + theta0y)) / 2 + 0.5;\
+					z = (z ? sin(z + theta0z) / z : sin(z + theta0z)) / 2 + 0.5;\
+					a = (a ? sin(a + theta0a) / a : sin(a + theta0a)) / 2 + 0.5;\
+					((TYPE *)(grad->buf))[4 * i + 0] = x;\
+					((TYPE *)(grad->buf))[4 * i + 1] = y;\
+					((TYPE *)(grad->buf))[4 * i + 2] = z;\
+					((TYPE *)(grad->buf))[4 * i + 3] = a;\
+				}\
+			}\
+			n *= grad->pixel_size;\
+			m -= n;\
+			ewriteall(STDOUT_FILENO, grad->buf, n, "<stdout>");\
+			memmove(grad->buf, grad->buf + n, grad->ptr -= n);\
+		} while (eread_stream(grad, SIZE_MAX));\
+		if (grad->ptr)\
+			eprintf("%s: incomplete frame\n", grad->file);\
+	}
+
+PROCESS(double, lf)
+PROCESS(float, f)
+
+
+int
+main(int argc, char *argv[])
+{
+	struct stream stream, theta0;
+	int have_theta0;
+	void (*process)(struct stream *grad, struct stream *theta0);
+
+	ARGBEGIN {
+	case 'e':
+		equal = 1;
+		break;
+	default:
+		usage();
+	} ARGEND;
+
+	if (argc > 1)
+		usage();
+
+	eopen_stream(&stream, NULL);
+	if ((have_theta0 = argc == 1)) {
+		eopen_stream(&theta0, argv[0]);
+		if (theta0.width != 1 || theta0.height != 1)
+			eprintf("theta0-stream must be of dimension 1x1\n");
+	}
+
+	if (!strcmp(stream.pixfmt, "xyza"))
+		process = process_lf;
+	else if (!strcmp(stream.pixfmt, "xyza f"))
+		process = process_f;
+	else
+		eprintf("pixel format %s is not supported, try xyza\n", stream.pixfmt);
+
+	if (have_theta0 && strcmp(stream.pixfmt, theta0.pixfmt))
+		eprintf("videos use incompatible pixel formats\n");
+
+	fprint_stream_head(stdout, &stream);
+	efflush(stdout, "<stdout>");
+	process(&stream, have_theta0 ? &theta0 : NULL);
+	return 0;
+}
-- 
cgit v1.2.3-70-g09d2