d93b7b91e0
Bug: webrtc:14763 Change-Id: I9615a9ce41c9b577c4ebd4cdcc9885bfbc5dac48 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/293040 Reviewed-by: Sam Zackrisson <saza@webrtc.org> Commit-Queue: Jesus de Vicente Pena <devicentepena@webrtc.org> Reviewed-by: Björn Terelius <terelius@webrtc.org> Reviewed-by: Jakob Ivarsson <jakobi@webrtc.org> Cr-Commit-Position: refs/heads/main@{#39706}
107 lines
3.7 KiB
C++
107 lines
3.7 KiB
C++
/*
|
|
* 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 "modules/audio_coding/neteq/tools/neteq_replacement_input.h"
|
|
|
|
#include "modules/audio_coding/neteq/tools/fake_decode_from_file.h"
|
|
#include "rtc_base/checks.h"
|
|
|
|
namespace webrtc {
|
|
namespace test {
|
|
|
|
NetEqReplacementInput::NetEqReplacementInput(
|
|
std::unique_ptr<NetEqInput> source,
|
|
uint8_t replacement_payload_type,
|
|
const std::set<uint8_t>& comfort_noise_types,
|
|
const std::set<uint8_t>& forbidden_types)
|
|
: source_(std::move(source)),
|
|
replacement_payload_type_(replacement_payload_type),
|
|
comfort_noise_types_(comfort_noise_types),
|
|
forbidden_types_(forbidden_types) {
|
|
RTC_CHECK(source_);
|
|
event_ = source_->PopEvent();
|
|
ReplaceIfPacketEvent();
|
|
}
|
|
|
|
std::unique_ptr<NetEqInput::Event> NetEqReplacementInput::PopEvent() {
|
|
std::unique_ptr<NetEqInput::Event> event_to_return = std::move(event_);
|
|
while (true) {
|
|
event_ = source_->PopEvent();
|
|
if (event_ == nullptr || event_->type() != Event::Type::kPacketData) {
|
|
break;
|
|
}
|
|
PacketData& packet = static_cast<PacketData&>(*event_);
|
|
if (packet.payload.size() > packet.header.paddingLength) {
|
|
// Not padding only. Good to go. Skip this packet otherwise.
|
|
break;
|
|
}
|
|
}
|
|
ReplaceIfPacketEvent();
|
|
return event_to_return;
|
|
}
|
|
|
|
bool NetEqReplacementInput::ended() const {
|
|
return source_->ended();
|
|
}
|
|
|
|
absl::optional<RTPHeader> NetEqReplacementInput::NextHeader() const {
|
|
return source_->NextHeader();
|
|
}
|
|
|
|
void NetEqReplacementInput::ReplaceIfPacketEvent() {
|
|
if (event_ == nullptr || event_->type() != Event::Type::kPacketData) {
|
|
return;
|
|
}
|
|
|
|
PacketData& packet = static_cast<PacketData&>(*event_);
|
|
|
|
RTC_CHECK_EQ(forbidden_types_.count(packet.header.payloadType), 0)
|
|
<< "Payload type " << static_cast<int>(packet.header.payloadType)
|
|
<< " is forbidden.";
|
|
|
|
// Check if this packet is comfort noise.
|
|
if (comfort_noise_types_.count(packet.header.payloadType) != 0) {
|
|
// If CNG, simply insert a zero-energy one-byte payload.
|
|
uint8_t cng_payload[1] = {127}; // Max attenuation of CNG.
|
|
packet.payload.SetData(cng_payload);
|
|
return;
|
|
}
|
|
|
|
absl::optional<RTPHeader> next_hdr = source_->NextHeader();
|
|
if (!next_hdr) {
|
|
// End of input. Cannot do proper replacement on the very last packet, so we
|
|
// delete it instead.
|
|
event_ = nullptr;
|
|
return;
|
|
}
|
|
RTC_DCHECK(next_hdr);
|
|
uint8_t payload[12];
|
|
RTC_DCHECK_LE(last_frame_size_timestamps_, 120 * 48);
|
|
uint32_t input_frame_size_timestamps = last_frame_size_timestamps_;
|
|
const uint32_t timestamp_diff = next_hdr->timestamp - packet.header.timestamp;
|
|
if (next_hdr->sequenceNumber == packet.header.sequenceNumber + 1 &&
|
|
timestamp_diff <= 120 * 48) {
|
|
// Packets are in order and the timestamp diff is less than 5760 samples.
|
|
// Accept the timestamp diff as a valid frame size.
|
|
input_frame_size_timestamps = timestamp_diff;
|
|
last_frame_size_timestamps_ = input_frame_size_timestamps;
|
|
}
|
|
RTC_DCHECK_LE(input_frame_size_timestamps, 120 * 48);
|
|
FakeDecodeFromFile::PrepareEncoded(packet.header.timestamp,
|
|
input_frame_size_timestamps,
|
|
packet.payload.size(), payload);
|
|
packet.payload.SetData(payload);
|
|
packet.header.payloadType = replacement_payload_type_;
|
|
return;
|
|
}
|
|
|
|
} // namespace test
|
|
} // namespace webrtc
|