add tools/mux_ivf

refers to GH-434

Author: MartinPulec

tools/.gitignore

@@ -4,5 +4,6 @@ astat_test
 benchmark_ff_convs
 convert
 decklink_temperature
+mux_ivf
 thumbnailgen
 uyvy2yuv422p

tools/Makefile

@@ -11,6 +11,7 @@ ifeq ($(ARCH), x86_64)
 endif
 
 TARGETS=astat_lib astat_test benchmark_ff_convs convert decklink_temperature \
+	mux_ivf \
 	thumbnailgen uyvy2yuv422p
 
 COMMON_OBJS = src/color.o src/debug.o src/video_codec.o src/pixfmt_conv.o \
@@ -54,6 +55,9 @@ convert: convert.o $(COMMON_OBJS)
 decklink_temperature: decklink_temperature.cpp ext-deps/DeckLink/Linux/DeckLinkAPIDispatch.o
 	$(CXX) $^ -o $@
 
+mux_ivf: mux_ivf.o
+	$(CC) $^ -o $@
+
 uyvy2yuv422p: uyvy2yuv422p.c
 	$(CC) -g -std=c99 -Wall $< -o $@
 

tools/mux_ivf.c

@@ -0,0 +1,90 @@
+/**
+ * Simple Duck IVF mixer for VP8/VP9/AV1.
+ *
+ * Can be used eg. to process exported (--record) VP9 files, that cannot be
+ * simply concatenated as eg. H.264 frames. Specification:
+ * <https://wiki.multimedia.cx/index.php/Duck_IVF>
+ */
+
+#include <assert.h>
+#include <errno.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+const char *
+get_fcc(const char *ext)
+{
+        if (strcasecmp(ext, "vp8") == 0) {
+                return "VP80";
+        }
+        if (strcasecmp(ext, "vp9") == 0) {
+                return "VP90";
+        }
+        if (strcasecmp(ext, "av1") == 0) {
+                return "av01";
+        }
+        fprintf(stderr, "Unsupported extension: %s\n", ext);
+        exit(EXIT_FAILURE);
+}
+
+#define OUTPUT(val) fwrite(&(val), sizeof (val), 1, stdout)
+int
+main(int argc, char *argv[])
+{
+        if (argc <= 4) {
+                fprintf(
+                    stderr,
+                    "Usage:\n\t%s <width> <height> <fps> INFILES > out.ivf\n",
+                    argv[0]);
+                return EXIT_FAILURE;
+        }
+        argv += 1; // skip progname
+
+        // file header
+        printf("DKIF");
+        const uint16_t version = 0;
+        OUTPUT(version);
+        const uint16_t hdr_len = 32;
+        OUTPUT(hdr_len);
+        printf("%s", get_fcc(strchr(argv[3], '.') + 1)); // codec FourCC
+        const uint16_t width  = atoi(*argv++);
+        const uint16_t height = atoi(*argv++);
+        const uint32_t fps    = atoi(*argv++);
+        assert(width * height * fps != 0);
+        OUTPUT(width);
+        OUTPUT(height);
+        const uint32_t fps_den = fps;
+        OUTPUT(fps_den);
+        const uint32_t fps_num = 1;
+        OUTPUT(fps_num);
+        const uint32_t nr_frames = argc - 3;
+        OUTPUT(nr_frames);
+        const uint32_t unused = 0;
+        OUTPUT(unused);
+
+        uint64_t pts = 0;
+        while (*argv != NULL) { // iterate over files
+                FILE *f = fopen(*argv++, "rb");
+                assert(f != NULL);
+
+                fseek(f, 0, SEEK_END);
+                const uint32_t file_len = ftell(f);
+                fseek(f, 0, SEEK_SET);
+
+                // frame header
+                OUTPUT(file_len);
+                OUTPUT(pts);
+                pts += 1;
+
+                // copy the content of the file to output
+                int c = 0;
+                while ((c = fgetc(f)) != EOF) {
+                        putc(c, stdout);
+                }
+                fclose(f);
+        }
+        // we do not do much error checking so at least return errno
+        return errno;
+}