Hide the AudioEncoderCng class behind a create function
And put codecs/cng/webrtc_cng.h in a non-public build target while we're at it. Bug: webrtc:8396 Change-Id: I9f51dffadfb645cd1454617fad30e09d639ff53c Reviewed-on: https://webrtc-review.googlesource.com/c/108782 Reviewed-by: Ivo Creusen <ivoc@webrtc.org> Commit-Queue: Karl Wiberg <kwiberg@webrtc.org> Cr-Commit-Position: refs/heads/master@{#25486}
This commit is contained in:
@ -23,29 +23,58 @@ namespace {
|
||||
|
||||
const int kMaxFrameSizeMs = 60;
|
||||
|
||||
} // namespace
|
||||
class AudioEncoderCng final : public AudioEncoder {
|
||||
public:
|
||||
explicit AudioEncoderCng(AudioEncoderCngConfig&& config);
|
||||
~AudioEncoderCng() override;
|
||||
|
||||
AudioEncoderCng::Config::Config() = default;
|
||||
AudioEncoderCng::Config::Config(Config&&) = default;
|
||||
AudioEncoderCng::Config::~Config() = default;
|
||||
// Not copyable or moveable.
|
||||
AudioEncoderCng(const AudioEncoderCng&) = delete;
|
||||
AudioEncoderCng(AudioEncoderCng&&) = delete;
|
||||
AudioEncoderCng& operator=(const AudioEncoderCng&) = delete;
|
||||
AudioEncoderCng& operator=(AudioEncoderCng&&) = delete;
|
||||
|
||||
bool AudioEncoderCng::Config::IsOk() const {
|
||||
if (num_channels != 1)
|
||||
return false;
|
||||
if (!speech_encoder)
|
||||
return false;
|
||||
if (num_channels != speech_encoder->NumChannels())
|
||||
return false;
|
||||
if (sid_frame_interval_ms <
|
||||
static_cast<int>(speech_encoder->Max10MsFramesInAPacket() * 10))
|
||||
return false;
|
||||
if (num_cng_coefficients > WEBRTC_CNG_MAX_LPC_ORDER ||
|
||||
num_cng_coefficients <= 0)
|
||||
return false;
|
||||
return true;
|
||||
}
|
||||
int SampleRateHz() const override;
|
||||
size_t NumChannels() const override;
|
||||
int RtpTimestampRateHz() const override;
|
||||
size_t Num10MsFramesInNextPacket() const override;
|
||||
size_t Max10MsFramesInAPacket() const override;
|
||||
int GetTargetBitrate() const override;
|
||||
EncodedInfo EncodeImpl(uint32_t rtp_timestamp,
|
||||
rtc::ArrayView<const int16_t> audio,
|
||||
rtc::Buffer* encoded) override;
|
||||
void Reset() override;
|
||||
bool SetFec(bool enable) override;
|
||||
bool SetDtx(bool enable) override;
|
||||
bool SetApplication(Application application) override;
|
||||
void SetMaxPlaybackRate(int frequency_hz) override;
|
||||
rtc::ArrayView<std::unique_ptr<AudioEncoder>> ReclaimContainedEncoders()
|
||||
override;
|
||||
void OnReceivedUplinkPacketLossFraction(
|
||||
float uplink_packet_loss_fraction) override;
|
||||
void OnReceivedUplinkRecoverablePacketLossFraction(
|
||||
float uplink_recoverable_packet_loss_fraction) override;
|
||||
void OnReceivedUplinkBandwidth(
|
||||
int target_audio_bitrate_bps,
|
||||
absl::optional<int64_t> bwe_period_ms) override;
|
||||
|
||||
AudioEncoderCng::AudioEncoderCng(Config&& config)
|
||||
private:
|
||||
EncodedInfo EncodePassive(size_t frames_to_encode, rtc::Buffer* encoded);
|
||||
EncodedInfo EncodeActive(size_t frames_to_encode, rtc::Buffer* encoded);
|
||||
size_t SamplesPer10msFrame() const;
|
||||
|
||||
std::unique_ptr<AudioEncoder> speech_encoder_;
|
||||
const int cng_payload_type_;
|
||||
const int num_cng_coefficients_;
|
||||
const int sid_frame_interval_ms_;
|
||||
std::vector<int16_t> speech_buffer_;
|
||||
std::vector<uint32_t> rtp_timestamps_;
|
||||
bool last_frame_active_;
|
||||
std::unique_ptr<Vad> vad_;
|
||||
std::unique_ptr<ComfortNoiseEncoder> cng_encoder_;
|
||||
};
|
||||
|
||||
AudioEncoderCng::AudioEncoderCng(AudioEncoderCngConfig&& config)
|
||||
: speech_encoder_((static_cast<void>([&] {
|
||||
RTC_CHECK(config.IsOk()) << "Invalid configuration.";
|
||||
}()),
|
||||
@ -263,4 +292,31 @@ size_t AudioEncoderCng::SamplesPer10msFrame() const {
|
||||
return rtc::CheckedDivExact(10 * SampleRateHz(), 1000);
|
||||
}
|
||||
|
||||
} // namespace
|
||||
|
||||
AudioEncoderCngConfig::AudioEncoderCngConfig() = default;
|
||||
AudioEncoderCngConfig::AudioEncoderCngConfig(AudioEncoderCngConfig&&) = default;
|
||||
AudioEncoderCngConfig::~AudioEncoderCngConfig() = default;
|
||||
|
||||
bool AudioEncoderCngConfig::IsOk() const {
|
||||
if (num_channels != 1)
|
||||
return false;
|
||||
if (!speech_encoder)
|
||||
return false;
|
||||
if (num_channels != speech_encoder->NumChannels())
|
||||
return false;
|
||||
if (sid_frame_interval_ms <
|
||||
static_cast<int>(speech_encoder->Max10MsFramesInAPacket() * 10))
|
||||
return false;
|
||||
if (num_cng_coefficients > WEBRTC_CNG_MAX_LPC_ORDER ||
|
||||
num_cng_coefficients <= 0)
|
||||
return false;
|
||||
return true;
|
||||
}
|
||||
|
||||
std::unique_ptr<AudioEncoder> CreateComfortNoiseEncoder(
|
||||
AudioEncoderCngConfig&& config) {
|
||||
return absl::make_unique<AudioEncoderCng>(std::move(config));
|
||||
}
|
||||
|
||||
} // namespace webrtc
|
||||
|
||||
@ -12,89 +12,37 @@
|
||||
#define MODULES_AUDIO_CODING_CODECS_CNG_AUDIO_ENCODER_CNG_H_
|
||||
|
||||
#include <stddef.h>
|
||||
#include <stdint.h>
|
||||
#include <memory>
|
||||
#include <vector>
|
||||
|
||||
#include "absl/types/optional.h"
|
||||
#include "api/array_view.h"
|
||||
#include "api/audio_codecs/audio_encoder.h"
|
||||
#include "common_audio/vad/include/vad.h"
|
||||
#include "rtc_base/buffer.h"
|
||||
#include "rtc_base/constructormagic.h"
|
||||
|
||||
namespace webrtc {
|
||||
|
||||
class ComfortNoiseEncoder;
|
||||
struct AudioEncoderCngConfig {
|
||||
// Moveable, not copyable.
|
||||
AudioEncoderCngConfig();
|
||||
AudioEncoderCngConfig(AudioEncoderCngConfig&&);
|
||||
~AudioEncoderCngConfig();
|
||||
|
||||
class AudioEncoderCng final : public AudioEncoder {
|
||||
public:
|
||||
struct Config {
|
||||
Config();
|
||||
Config(Config&&);
|
||||
~Config();
|
||||
bool IsOk() const;
|
||||
bool IsOk() const;
|
||||
|
||||
size_t num_channels = 1;
|
||||
int payload_type = 13;
|
||||
std::unique_ptr<AudioEncoder> speech_encoder;
|
||||
Vad::Aggressiveness vad_mode = Vad::kVadNormal;
|
||||
int sid_frame_interval_ms = 100;
|
||||
int num_cng_coefficients = 8;
|
||||
// The Vad pointer is mainly for testing. If a NULL pointer is passed, the
|
||||
// AudioEncoderCng creates (and destroys) a Vad object internally. If an
|
||||
// object is passed, the AudioEncoderCng assumes ownership of the Vad
|
||||
// object.
|
||||
Vad* vad = nullptr;
|
||||
};
|
||||
|
||||
explicit AudioEncoderCng(Config&& config);
|
||||
~AudioEncoderCng() override;
|
||||
|
||||
int SampleRateHz() const override;
|
||||
size_t NumChannels() const override;
|
||||
int RtpTimestampRateHz() const override;
|
||||
size_t Num10MsFramesInNextPacket() const override;
|
||||
size_t Max10MsFramesInAPacket() const override;
|
||||
int GetTargetBitrate() const override;
|
||||
EncodedInfo EncodeImpl(uint32_t rtp_timestamp,
|
||||
rtc::ArrayView<const int16_t> audio,
|
||||
rtc::Buffer* encoded) override;
|
||||
void Reset() override;
|
||||
bool SetFec(bool enable) override;
|
||||
bool SetDtx(bool enable) override;
|
||||
bool SetApplication(Application application) override;
|
||||
void SetMaxPlaybackRate(int frequency_hz) override;
|
||||
rtc::ArrayView<std::unique_ptr<AudioEncoder>> ReclaimContainedEncoders()
|
||||
override;
|
||||
void OnReceivedUplinkPacketLossFraction(
|
||||
float uplink_packet_loss_fraction) override;
|
||||
void OnReceivedUplinkRecoverablePacketLossFraction(
|
||||
float uplink_recoverable_packet_loss_fraction) override;
|
||||
void OnReceivedUplinkBandwidth(
|
||||
int target_audio_bitrate_bps,
|
||||
absl::optional<int64_t> bwe_period_ms) override;
|
||||
|
||||
private:
|
||||
EncodedInfo EncodePassive(size_t frames_to_encode,
|
||||
rtc::Buffer* encoded);
|
||||
EncodedInfo EncodeActive(size_t frames_to_encode,
|
||||
rtc::Buffer* encoded);
|
||||
size_t SamplesPer10msFrame() const;
|
||||
|
||||
std::unique_ptr<AudioEncoder> speech_encoder_;
|
||||
const int cng_payload_type_;
|
||||
const int num_cng_coefficients_;
|
||||
const int sid_frame_interval_ms_;
|
||||
std::vector<int16_t> speech_buffer_;
|
||||
std::vector<uint32_t> rtp_timestamps_;
|
||||
bool last_frame_active_;
|
||||
std::unique_ptr<Vad> vad_;
|
||||
std::unique_ptr<ComfortNoiseEncoder> cng_encoder_;
|
||||
|
||||
RTC_DISALLOW_COPY_AND_ASSIGN(AudioEncoderCng);
|
||||
size_t num_channels = 1;
|
||||
int payload_type = 13;
|
||||
std::unique_ptr<AudioEncoder> speech_encoder;
|
||||
Vad::Aggressiveness vad_mode = Vad::kVadNormal;
|
||||
int sid_frame_interval_ms = 100;
|
||||
int num_cng_coefficients = 8;
|
||||
// The Vad pointer is mainly for testing. If a NULL pointer is passed, the
|
||||
// AudioEncoderCng creates (and destroys) a Vad object internally. If an
|
||||
// object is passed, the AudioEncoderCng assumes ownership of the Vad
|
||||
// object.
|
||||
Vad* vad = nullptr;
|
||||
};
|
||||
|
||||
std::unique_ptr<AudioEncoder> CreateComfortNoiseEncoder(
|
||||
AudioEncoderCngConfig&& config);
|
||||
|
||||
} // namespace webrtc
|
||||
|
||||
#endif // MODULES_AUDIO_CODING_CODECS_CNG_AUDIO_ENCODER_CNG_H_
|
||||
|
||||
@ -50,8 +50,8 @@ class AudioEncoderCngTest : public ::testing::Test {
|
||||
cng_.reset();
|
||||
}
|
||||
|
||||
AudioEncoderCng::Config MakeCngConfig() {
|
||||
AudioEncoderCng::Config config;
|
||||
AudioEncoderCngConfig MakeCngConfig() {
|
||||
AudioEncoderCngConfig config;
|
||||
config.speech_encoder = std::move(mock_encoder_owner_);
|
||||
EXPECT_TRUE(config.speech_encoder);
|
||||
|
||||
@ -63,7 +63,7 @@ class AudioEncoderCngTest : public ::testing::Test {
|
||||
return config;
|
||||
}
|
||||
|
||||
void CreateCng(AudioEncoderCng::Config&& config) {
|
||||
void CreateCng(AudioEncoderCngConfig&& config) {
|
||||
num_audio_samples_10ms_ = static_cast<size_t>(10 * sample_rate_hz_ / 1000);
|
||||
ASSERT_LE(num_audio_samples_10ms_, kMaxNumSamples);
|
||||
if (config.speech_encoder) {
|
||||
@ -75,7 +75,7 @@ class AudioEncoderCngTest : public ::testing::Test {
|
||||
EXPECT_CALL(*mock_encoder_, Max10MsFramesInAPacket())
|
||||
.WillOnce(Return(1u));
|
||||
}
|
||||
cng_.reset(new AudioEncoderCng(std::move(config)));
|
||||
cng_ = CreateComfortNoiseEncoder(std::move(config));
|
||||
}
|
||||
|
||||
void Encode() {
|
||||
@ -193,7 +193,7 @@ class AudioEncoderCngTest : public ::testing::Test {
|
||||
return encoded_info_.payload_type != kCngPayloadType;
|
||||
}
|
||||
|
||||
std::unique_ptr<AudioEncoderCng> cng_;
|
||||
std::unique_ptr<AudioEncoder> cng_;
|
||||
std::unique_ptr<MockAudioEncoder> mock_encoder_owner_;
|
||||
MockAudioEncoder* mock_encoder_;
|
||||
MockVad* mock_vad_; // Ownership is transferred to |cng_|.
|
||||
@ -432,7 +432,7 @@ class AudioEncoderCngDeathTest : public AudioEncoderCngTest {
|
||||
// deleted.
|
||||
void TearDown() override { cng_.reset(); }
|
||||
|
||||
AudioEncoderCng::Config MakeCngConfig() {
|
||||
AudioEncoderCngConfig MakeCngConfig() {
|
||||
// Don't provide a Vad mock object, since it would leak when the test dies.
|
||||
auto config = AudioEncoderCngTest::MakeCngConfig();
|
||||
config.vad = nullptr;
|
||||
|
||||
Reference in New Issue
Block a user