Add blind-norm

Signed-off-by: Mattias Andrée <maandree@kth.se>

1 files changed, 86 insertions, 0 deletions

diff --git a/src/blind-norm.c b/src/blind-norm.c
new file mode 100644
index 0000000..c4d04a0
--- /dev/null
+++ b/src/blind-norm.c
@@ -0,0 +1,86 @@
+/* See LICENSE file for copyright and license details. */
+#include "common.h"
+
+USAGE("[-axyz]")
+
+static int skip_a = 0;
+static int skip_x = 0;
+static int skip_y = 0;
+static int skip_z = 0;
+
+
+#define PROCESS(TYPE, SUFFIX)\
+	static void\
+	process_##SUFFIX(struct stream *stream)\
+	{\
+		size_t i, n;\
+		TYPE x, y, z, a, norm;\
+		do {\
+			n = stream->ptr / stream->pixel_size;\
+			for (i = 0; i < n; i++) {\
+				x = ((TYPE *)(stream->buf))[4 * i + 0];\
+				y = ((TYPE *)(stream->buf))[4 * i + 1];\
+				z = ((TYPE *)(stream->buf))[4 * i + 2];\
+				a = ((TYPE *)(stream->buf))[4 * i + 3];\
+				norm = sqrt(x * x + y * y + z * z + a * a);\
+				if (!skip_x)\
+					((TYPE *)(stream->buf))[4 * i + 0] = norm;\
+				if (!skip_y)\
+					((TYPE *)(stream->buf))[4 * i + 1] = norm;\
+				if (!skip_z)\
+					((TYPE *)(stream->buf))[4 * i + 2] = norm;\
+				if (!skip_a)\
+					((TYPE *)(stream->buf))[4 * i + 3] = norm;\
+			}\
+			n *= stream->pixel_size;\
+			ewriteall(STDOUT_FILENO, stream->buf, n, "<stdout>");\
+			memmove(stream->buf, stream->buf + n, stream->ptr -= n);\
+		} while (eread_stream(stream, SIZE_MAX));\
+		if (stream->ptr)\
+			eprintf("%s: incomplete frame\n", stream->file);\
+	}
+
+PROCESS(double, lf)
+PROCESS(float, f)
+
+
+int
+main(int argc, char *argv[])
+{
+	struct stream stream;
+	void (*process)(struct stream *stream);
+
+	ARGBEGIN {
+	case 'a':
+		skip_a = 1;
+		break;
+	case 'x':
+		skip_x = 1;
+		break;
+	case 'y':
+		skip_y = 1;
+		break;
+	case 'z':
+		skip_z = 1;
+		break;
+	default:
+		usage();
+	} ARGEND;
+
+	if (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);
+
+	fprint_stream_head(stdout, &stream);
+	efflush(stdout, "<stdout>");
+	process(&stream);
+	return 0;
+}