Add internal utility to log audio streams to disk from real-time audio callbacks

1 files changed, 74 insertions, 0 deletions

diff --git a/test/test_audio_dump.cpp b/test/test_audio_dump.cpp
new file mode 100644
index 0000000..f01c609
--- /dev/null
+++ b/test/test_audio_dump.cpp
@@ -0,0 +1,74 @@
+/*
+ * Copyright © 2023 Mozilla Foundation
+ *
+ * This program is made available under an ISC-style license.  See the
+ * accompanying file LICENSE for details.
+ */
+
+#define NOMINMAX
+#define _USE_MATH_DEFINES
+
+#include "cubeb/cubeb.h"
+#include <ratio>
+
+#include "cubeb_audio_dump.h"
+#include "gtest/gtest.h"
+#include <chrono>
+#include <cmath>
+#include <fstream>
+#include <iostream>
+#include <thread>
+
+TEST(cubeb, audio_dump)
+{
+  cubeb_audio_dump_session_t session;
+  int rv = cubeb_audio_dump_init(&session);
+  ASSERT_EQ(rv, 0);
+
+  cubeb_stream_params params;
+  params.rate = 44100;
+  params.channels = 2;
+  params.format = CUBEB_SAMPLE_FLOAT32NE;
+
+  cubeb_audio_dump_stream_t dump_stream;
+  rv = cubeb_audio_dump_stream_init(session, &dump_stream, params, "test.wav");
+  ASSERT_EQ(rv, 0);
+
+  rv = cubeb_audio_dump_start(session);
+  ASSERT_EQ(rv, 0);
+
+  float phase = 0;
+  const size_t buf_sz = 2 * 44100 / 50;
+  float buf[buf_sz];
+  for (uint32_t iteration = 0; iteration < 50; iteration++) {
+    uint32_t write_idx = 0;
+    for (uint32_t i = 0; i < buf_sz / params.channels; i++) {
+      for (uint32_t j = 0; j < params.channels; j++) {
+        buf[write_idx++] = sin(phase);
+      }
+      phase += 440 * M_PI * 2 / 44100;
+      if (phase > 2 * M_PI) {
+        phase -= 2 * M_PI;
+      }
+    }
+    rv = cubeb_audio_dump_write(dump_stream, buf, 2 * 44100 / 50);
+    std::this_thread::sleep_for(std::chrono::milliseconds(100));
+    ASSERT_EQ(rv, 0);
+  }
+
+  std::this_thread::sleep_for(std::chrono::milliseconds(100));
+
+  rv = cubeb_audio_dump_stop(session);
+  ASSERT_EQ(rv, 0);
+
+  rv = cubeb_audio_dump_stream_shutdown(session, dump_stream);
+  ASSERT_EQ(rv, 0);
+
+  rv = cubeb_audio_dump_shutdown(session);
+  ASSERT_EQ(rv, 0);
+
+  std::ifstream file("test.wav");
+  ASSERT_TRUE(file.good());
+}
+
+#undef NOMINMAX