tools: add sofa2wavs

Signed-off-by: Paul B Mahol <onemda@gmail.com>

3 files changed, 81 insertions, 0 deletions

diff --git a/Makefile b/Makefile
index 685b613a1c..a2df8b9d8d 100644
--- a/Makefile
+++ b/Makefile
@@ -82,6 +82,7 @@ target_dec_%_fuzzer$(EXESUF): target_dec_%_fuzzer.o $(FF_DEP_LIBS)
 	$(LD) $(LDFLAGS) $(LDEXEFLAGS) $(LD_O) $^ $(ELIBS) $(FF_EXTRALIBS) $(LIBFUZZER_PATH)
 
 tools/cws2fws$(EXESUF): ELIBS = $(ZLIB)
+tools/sofa2wavs$(EXESUF): ELIBS = $(FF_EXTRALIBS)
 tools/uncoded_frame$(EXESUF): $(FF_DEP_LIBS)
 tools/uncoded_frame$(EXESUF): ELIBS = $(FF_EXTRALIBS)
 tools/target_dec_%_fuzzer$(EXESUF): $(FF_DEP_LIBS)
diff --git a/tools/Makefile b/tools/Makefile
index 3ebd3619ca..3909e7cfdd 100644
--- a/tools/Makefile
+++ b/tools/Makefile
@@ -1,4 +1,5 @@
 TOOLS = qt-faststart trasher uncoded_frame
+TOOLS-$(CONFIG_LIBMYSOFA) += sofa2wavs
 TOOLS-$(CONFIG_ZLIB) += cws2fws
 
 tools/target_dec_%_fuzzer.o: tools/target_dec_fuzzer.c
diff --git a/tools/sofa2wavs.c b/tools/sofa2wavs.c
new file mode 100644
index 0000000000..f85e54f771
--- /dev/null
+++ b/tools/sofa2wavs.c
@@ -0,0 +1,79 @@
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <unistd.h>
+#include <stdio.h>
+#include <mysofa.h>
+
+int main(int argc, char **argv)
+{
+    struct MYSOFA_HRTF *hrtf;
+    int sample_rate;
+    int err, i, j;
+
+    if (argc < 3) {
+        printf("usage: %s input_SOFA_file output_directory\n", argv[0]);
+        return 1;
+    }
+
+    hrtf = mysofa_load(argv[1], &err);
+    if (!hrtf || err) {
+        printf("invalid input SOFA file: %s\n", argv[1]);
+        return 1;
+    }
+
+    if (hrtf->DataSamplingRate.elements != 1)
+        goto fail;
+    sample_rate = hrtf->DataSamplingRate.values[0];
+
+    err = mkdir(argv[2], 0744);
+    if (err)
+        goto fail;
+
+    err = chdir(argv[2]);
+    if (err)
+        goto fail;
+
+    for (i = 0; i < hrtf->M; i++) {
+        FILE *file;
+        int bps = 32;
+        int blkalign = 8;
+        int bytespersec = blkalign * sample_rate;
+        char filename[1024];
+        int azi = hrtf->SourcePosition.values[i * 3];
+        int ele = hrtf->SourcePosition.values[i * 3 + 1];
+        int dis = hrtf->SourcePosition.values[i * 3 + 2];
+        int size = 8 * hrtf->N;
+        int offset = i * 2 * hrtf->N;
+
+        snprintf(filename, sizeof(filename), "azi_%d_ele_%d_dis_%d.wav", azi, ele, dis);
+        file = fopen(filename, "w+");
+        fwrite("RIFF", 4, 1, file);
+        fwrite("\xFF\xFF\xFF\xFF", 4, 1, file);
+        fwrite("WAVE", 4, 1, file);
+        fwrite("fmt ", 4, 1, file);
+        fwrite("\x10\x00\00\00", 4, 1, file);
+        fwrite("\x03\x00", 2, 1, file);
+        fwrite("\x02\x00", 2, 1, file);
+        fwrite(&sample_rate, 4, 1, file);
+        fwrite(&bytespersec, 4, 1, file);
+        fwrite(&blkalign, 2, 1, file);
+        fwrite(&bps, 2, 1, file);
+        fwrite("data", 4, 1, file);
+        fwrite(&size, 4, 1, file);
+
+        for (j = 0; j < hrtf->N; j++) {
+            float l, r;
+
+            l = hrtf->DataIR.values[offset + j];
+            r = hrtf->DataIR.values[offset + j + hrtf->N];
+            fwrite(&l, 4, 1, file);
+            fwrite(&r, 4, 1, file);
+        }
+        fclose(file);
+    }
+
+fail:
+    mysofa_free(hrtf);
+
+    return 0;
+}