Add vu-frame-to-image

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

1 files changed, 67 insertions, 0 deletions

diff --git a/src/vu-frame-to-image.c b/src/vu-frame-to-image.c
new file mode 100644
index 0000000..e4466cf
--- /dev/null
+++ b/src/vu-frame-to-image.c
@@ -0,0 +1,67 @@
+/* See LICENSE file for copyright and license details. */
+#include "arg.h"
+#include "util.h"
+
+#include <inttypes.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+
+static void
+usage(void)
+{
+	eprintf("usage: %s -w width -h height\n", argv0);
+}
+
+int
+main(int argc, char *argv[])
+{
+	size_t width = 0, height = 0;
+	size_t p, n;
+	ssize_t r;
+	char buf[8096];
+
+	ARGBEGIN {
+	case 'w':
+		if (tozu(EARGF(usage()), 1, SIZE_MAX, &width))
+			eprintf("argument of -w must be an integer in [1, %zu]\n", SIZE_MAX);
+		break;
+	case 'h':
+		if (tozu(EARGF(usage()), 1, SIZE_MAX, &height))
+			eprintf("argument of -h must be an integer in [1, %zu]\n", SIZE_MAX);
+		break;
+	default:
+		usage();
+	} ARGEND;
+
+	if (!width || !height || argc)
+		usage();
+
+	printf("P7\n"
+	       "WIDTH %zu\n"
+	       "HEIGHT %zu\n"
+	       "DEPTH 4\n"
+	       "MAXVAL 255\n"
+	       "TUPLTYPE RGB_ALPHA\n"
+	       "ENDHDR\n", width, height);
+	fflush(stdout);
+	if (ferror(stdout))
+		eprintf("<stdout>:");
+
+	for (n = 0;;) {
+		for (p = 0; p < n;) {
+			r = write(STDOUT_FILENO, buf + p, n - p);
+			if (r < 0)
+				eprintf("write <stdin>:");
+			p += (size_t)r;
+		}
+		r = read(STDIN_FILENO, buf, sizeof(buf));
+		if (r < 0)
+			 eprintf("read <stdin>:");
+		if (r == 0)
+			break;
+		n = (size_t)r;
+	}
+
+	return 0;
+}