test_audio_dump.cpp

Enable keyboard shortcuts

/*

 * 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