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-square-gradient.c | 109 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 109 insertions(+)
 create mode 100644 src/blind-square-gradient.c

(limited to 'src/blind-square-gradient.c')

diff --git a/src/blind-square-gradient.c b/src/blind-square-gradient.c
new file mode 100644
index 0000000..392a62f
--- /dev/null
+++ b/src/blind-square-gradient.c
@@ -0,0 +1,109 @@
+/* See LICENSE file for copyright and license details. */
+#include "common.h"
+
+USAGE("-w width -h height")
+
+static size_t width = 0;
+static size_t height = 0;
+static int with_multiplier;
+
+
+#define PROCESS(TYPE, SUFFIX)\
+	static void\
+	process_##SUFFIX(struct stream *stream)\
+	{\
+		typedef TYPE pixel_t[4];\
+		pixel_t buf[BUFSIZ / sizeof(pixel_t)];\
+		TYPE *params, x1, y1, x2, y2, norm, rd = 1; \
+		TYPE x, y, p, r, rx, ry;\
+		size_t ix, iy, ptr = 0;\
+		for (;;) {\
+			while (stream->ptr < stream->frame_size) {\
+				if (!eread_stream(stream, stream->frame_size - stream->ptr)) {\
+					ewriteall(STDOUT_FILENO, buf, ptr * sizeof(*buf), "<stdout>");\
+					return;\
+				}\
+			}\
+			params = (TYPE *)stream->buf;\
+			x1 = (params)[0];\
+			y1 = (params)[1];\
+			x2 = (params)[4];\
+			y2 = (params)[5];\
+			if (with_multiplier)\
+				rd = (params)[9];\
+			memmove(stream->buf, stream->buf + stream->frame_size,\
+				stream->ptr -= stream->frame_size);\
+			\
+			x2 -= x1;\
+			y2 -= y1;\
+			norm = sqrt(x2 * x2 + y2 * y2);\
+			x2 /= norm;\
+			y2 /= norm;\
+			\
+			for (iy = 0; iy < height; iy++) {\
+				y = (TYPE)iy - y1;\
+				for (ix = 0; ix < width; ix++) {\
+					x = (TYPE)ix - x1;\
+					p = x * x2 + y * y2;\
+					rx = x - p * x2;\
+					ry = y - p * y2;\
+					r = sqrt(rx * rx + ry * ry) / rd;\
+					p = abs(p);\
+					x = MAX(p, r) / norm;\
+					buf[ptr][0] = buf[ptr][1] = buf[ptr][2] = buf[ptr][3] = x;\
+					if (++ptr == ELEMENTSOF(buf)) {\
+						ewriteall(STDOUT_FILENO, buf, sizeof(buf), "<stdout>");\
+						ptr = 0;\
+					}\
+				}\
+			}\
+		}\
+	}
+
+PROCESS(double, lf)
+PROCESS(float, f)
+
+
+int
+main(int argc, char *argv[])
+{
+	struct stream stream;
+	void (*process)(struct stream *stream);
+
+	ARGBEGIN {
+	case 'w':
+		width = etozu_flag('w', UARGF(), 1, SIZE_MAX);
+		break;
+	case 'h':
+		height = etozu_flag('h', UARGF(), 1, SIZE_MAX);
+		break;
+	default:
+		usage();
+	} ARGEND;
+
+	if (!width || !height || argc)
+		usage();
+
+	eopen_stream(&stream, NULL);
+
+	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 (stream.width > 3 || stream.height > 3 ||
+	    stream.width * stream.height < 2 ||
+	    stream.width * stream.height > 3)
+		eprintf("<stdin>: each frame must contain exactly 2 or 3 pixels\n");
+
+	with_multiplier = stream.width * stream.height == 3;
+
+	stream.width = width;
+	stream.height = height;
+	fprint_stream_head(stdout, &stream);
+	efflush(stdout, "<stdout>");
+	process(&stream);
+	return 0;
+}
-- 
cgit v1.2.3-70-g09d2