modules/audio_coding/codecs/builtin_audio_decoder_factory_unittest.cc

/*
 *  Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */

#include "api/audio_codecs/builtin_audio_decoder_factory.h"

#include <memory>
#include <optional>
#include <string>
#include <utility>
#include <vector>

#include "api/audio_codecs/audio_decoder.h"
#include "api/audio_codecs/audio_decoder_factory.h"
#include "api/audio_codecs/audio_format.h"
#include "api/environment/environment.h"
#include "api/rtp_parameters.h"
#include "api/scoped_refptr.h"
#include "test/create_test_environment.h"
#include "test/gtest.h"

namespace webrtc {

TEST(AudioDecoderFactoryTest, CreateUnknownDecoder) {
  const Environment env = CreateTestEnvironment();
  scoped_refptr<AudioDecoderFactory> adf = CreateBuiltinAudioDecoderFactory();
  ASSERT_TRUE(adf);
  EXPECT_FALSE(adf->Create(env, SdpAudioFormat("rey", 8000, 1), std::nullopt));
}

TEST(AudioDecoderFactoryTest, CreatePcmu) {
  const Environment env = CreateTestEnvironment();
  scoped_refptr<AudioDecoderFactory> adf = CreateBuiltinAudioDecoderFactory();
  ASSERT_TRUE(adf);
  // PCMu supports 8 kHz, and any number of channels.
  EXPECT_FALSE(adf->Create(env, SdpAudioFormat("pcmu", 8000, 0), std::nullopt));
  EXPECT_TRUE(adf->Create(env, SdpAudioFormat("pcmu", 8000, 1), std::nullopt));
  EXPECT_TRUE(adf->Create(env, SdpAudioFormat("pcmu", 8000, 2), std::nullopt));
  EXPECT_TRUE(adf->Create(env, SdpAudioFormat("pcmu", 8000, 3), std::nullopt));
  EXPECT_FALSE(
      adf->Create(env, SdpAudioFormat("pcmu", 16000, 1), std::nullopt));
}

TEST(AudioDecoderFactoryTest, CreatePcma) {
  const Environment env = CreateTestEnvironment();
  scoped_refptr<AudioDecoderFactory> adf = CreateBuiltinAudioDecoderFactory();
  ASSERT_TRUE(adf);
  // PCMa supports 8 kHz, and any number of channels.
  EXPECT_FALSE(adf->Create(env, SdpAudioFormat("pcma", 8000, 0), std::nullopt));
  EXPECT_TRUE(adf->Create(env, SdpAudioFormat("pcma", 8000, 1), std::nullopt));
  EXPECT_TRUE(adf->Create(env, SdpAudioFormat("pcma", 8000, 2), std::nullopt));
  EXPECT_TRUE(adf->Create(env, SdpAudioFormat("pcma", 8000, 3), std::nullopt));
  EXPECT_FALSE(
      adf->Create(env, SdpAudioFormat("pcma", 16000, 1), std::nullopt));
}

TEST(AudioDecoderFactoryTest, CreateL16) {
  const Environment env = CreateTestEnvironment();
  scoped_refptr<AudioDecoderFactory> adf = CreateBuiltinAudioDecoderFactory();
  ASSERT_TRUE(adf);
  // L16 supports any clock rate and any number of channels up to 16.
  const int clockrates[] = {8000, 16000, 32000, 48000};
  const int num_channels[] = {1, 2, 3, 16};
  for (int clockrate : clockrates) {
    EXPECT_FALSE(
        adf->Create(env, SdpAudioFormat("l16", clockrate, 0), std::nullopt));
    for (int channels : num_channels) {
      EXPECT_TRUE(adf->Create(env, SdpAudioFormat("l16", clockrate, channels),
                              std::nullopt));
    }
  }
}

// Tests that using more channels than the maximum does not work
TEST(AudioDecoderFactoryTest, MaxNrOfChannels) {
  const Environment env = CreateTestEnvironment();
  scoped_refptr<AudioDecoderFactory> adf = CreateBuiltinAudioDecoderFactory();
  std::vector<std::string> codecs = {
#ifdef WEBRTC_CODEC_OPUS
      "opus",
#endif
      "pcmu", "pcma", "l16", "G722", "G711",
  };

  for (auto codec : codecs) {
    EXPECT_FALSE(adf->Create(
        env,
        SdpAudioFormat(codec, 32000, AudioDecoder::kMaxNumberOfChannels + 1),
        std::nullopt));
  }
}

TEST(AudioDecoderFactoryTest, CreateG722) {
  const Environment env = CreateTestEnvironment();
  scoped_refptr<AudioDecoderFactory> adf = CreateBuiltinAudioDecoderFactory();
  ASSERT_TRUE(adf);
  // g722 supports 8 kHz, 1-2 channels.
  EXPECT_FALSE(adf->Create(env, SdpAudioFormat("g722", 8000, 0), std::nullopt));
  EXPECT_TRUE(adf->Create(env, SdpAudioFormat("g722", 8000, 1), std::nullopt));
  EXPECT_TRUE(adf->Create(env, SdpAudioFormat("g722", 8000, 2), std::nullopt));
  EXPECT_FALSE(adf->Create(env, SdpAudioFormat("g722", 8000, 3), std::nullopt));
  EXPECT_FALSE(
      adf->Create(env, SdpAudioFormat("g722", 16000, 1), std::nullopt));
  EXPECT_FALSE(
      adf->Create(env, SdpAudioFormat("g722", 32000, 1), std::nullopt));

  // g722 actually uses a 16 kHz sample rate instead of the nominal 8 kHz.
  std::unique_ptr<AudioDecoder> dec =
      adf->Create(env, SdpAudioFormat("g722", 8000, 1), std::nullopt);
  EXPECT_EQ(16000, dec->SampleRateHz());
}

TEST(AudioDecoderFactoryTest, CreateOpus) {
  const Environment env = CreateTestEnvironment();
  scoped_refptr<AudioDecoderFactory> adf = CreateBuiltinAudioDecoderFactory();
  ASSERT_TRUE(adf);
  // Opus supports 48 kHz and 2 channels. It is possible to specify a "stereo"
  // parameter whose value is either "0" or "1".
  for (int hz : {8000, 16000, 32000, 48000}) {
    for (int channels : {0, 1, 2, 3}) {
      for (std::string stereo : {"XX", "0", "1", "2"}) {
        CodecParameterMap params;
        if (stereo != "XX") {
          params["stereo"] = stereo;
        }
        const bool good = (hz == 48000 && channels == 2 &&
                           (stereo == "XX" || stereo == "0" || stereo == "1"));
        EXPECT_EQ(
            good,
            static_cast<bool>(adf->Create(
                env, SdpAudioFormat("opus", hz, channels, std::move(params)),
                std::nullopt)));
      }
    }
  }
}

}  // namespace webrtc