blob: 31da98bde482432e3e82efd764c5ab66b60e566f [file] [log] [blame]
kjellander@webrtc.org35a17562011-10-06 06:44:54 +00001/*
pwestin@webrtc.orgce330352012-04-12 06:59:14 +00002 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
kjellander@webrtc.org35a17562011-10-06 06:44:54 +00003 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
kjellander@webrtc.org5b97b122011-12-08 07:42:18 +000010
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020011#include "modules/video_coding/codecs/test/videoprocessor.h"
kjellander@webrtc.org35a17562011-10-06 06:44:54 +000012
Yves Gerey3e707812018-11-28 16:47:49 +010013#include <string.h>
ssilkin612f8582017-09-28 09:23:17 -070014#include <algorithm>
Yves Gerey3e707812018-11-28 16:47:49 +010015#include <cstddef>
kjellander@webrtc.org35a17562011-10-06 06:44:54 +000016#include <limits>
Erik Språng08127a92016-11-16 16:41:30 +010017#include <utility>
kjellander@webrtc.org35a17562011-10-06 06:44:54 +000018
Yves Gerey3e707812018-11-28 16:47:49 +010019#include "absl/memory/memory.h"
Mirko Bonadeid9708072019-01-25 20:26:48 +010020#include "api/scoped_refptr.h"
Jiawei Ouc2ebe212018-11-08 10:02:56 -080021#include "api/video/builtin_video_bitrate_allocator_factory.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020022#include "api/video/i420_buffer.h"
Yves Gerey3e707812018-11-28 16:47:49 +010023#include "api/video/video_bitrate_allocator_factory.h"
24#include "api/video/video_frame_buffer.h"
25#include "api/video/video_rotation.h"
26#include "api/video_codecs/video_codec.h"
Mirko Bonadei71207422017-09-15 13:58:09 +020027#include "common_types.h" // NOLINT(build/include)
ssilkin612f8582017-09-28 09:23:17 -070028#include "common_video/h264/h264_common.h"
Rasmus Brandt5f7a8912018-02-28 17:17:15 +010029#include "common_video/libyuv/include/webrtc_libyuv.h"
Sergey Silkin3be2a552018-01-17 15:11:44 +010030#include "modules/rtp_rtcp/include/rtp_rtcp_defines.h"
Yves Gerey3e707812018-11-28 16:47:49 +010031#include "modules/video_coding/codecs/interface/common_constants.h"
Rasmus Brandtd00c8952018-03-14 12:29:57 +010032#include "modules/video_coding/include/video_error_codes.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020033#include "rtc_base/checks.h"
Steve Anton10542f22019-01-11 09:11:00 -080034#include "rtc_base/time_utils.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020035#include "test/gtest.h"
Sergey Silkin8d3758e2018-03-14 11:28:15 +010036#include "third_party/libyuv/include/libyuv/compare.h"
Sergey Silkin10d9d592018-02-01 13:25:17 +010037#include "third_party/libyuv/include/libyuv/scale.h"
kjellander@webrtc.org35a17562011-10-06 06:44:54 +000038
39namespace webrtc {
40namespace test {
41
Kári Tristan Helgason169005d2018-05-22 13:34:14 +020042using FrameStatistics = VideoCodecTestStats::FrameStatistics;
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +010043
Kári Tristan Helgason169005d2018-05-22 13:34:14 +020044namespace {
Åsa Persson91af24a2018-01-24 17:20:18 +010045const int kMsToRtpTimestamp = kVideoPayloadTypeFrequency / 1000;
Sami Kalliomäkic75a5e82018-07-09 13:27:42 +020046const int kMaxBufferedInputFrames = 20;
brandtr17b958c2017-03-07 01:41:43 -080047
Sergey Silkin3be2a552018-01-17 15:11:44 +010048size_t GetMaxNaluSizeBytes(const EncodedImage& encoded_frame,
Kári Tristan Helgason169005d2018-05-22 13:34:14 +020049 const VideoCodecTestFixture::Config& config) {
ssilkin612f8582017-09-28 09:23:17 -070050 if (config.codec_settings.codecType != kVideoCodecH264)
Sergey Silkin3be2a552018-01-17 15:11:44 +010051 return 0;
ssilkin612f8582017-09-28 09:23:17 -070052
53 std::vector<webrtc::H264::NaluIndex> nalu_indices =
Niels Möller77536a22019-01-15 08:50:01 +010054 webrtc::H264::FindNaluIndices(encoded_frame.data(), encoded_frame.size());
ssilkin612f8582017-09-28 09:23:17 -070055
56 RTC_CHECK(!nalu_indices.empty());
57
Sergey Silkin3be2a552018-01-17 15:11:44 +010058 size_t max_size = 0;
ssilkin612f8582017-09-28 09:23:17 -070059 for (const webrtc::H264::NaluIndex& index : nalu_indices)
Sergey Silkin3be2a552018-01-17 15:11:44 +010060 max_size = std::max(max_size, index.payload_size);
ssilkin612f8582017-09-28 09:23:17 -070061
Sergey Silkin3be2a552018-01-17 15:11:44 +010062 return max_size;
ssilkin612f8582017-09-28 09:23:17 -070063}
64
Niels Möllerd3b8c632018-08-27 15:33:42 +020065size_t GetTemporalLayerIndex(const CodecSpecificInfo& codec_specific) {
66 size_t temporal_idx = 0;
Rasmus Brandtd062a3c2018-03-08 16:45:54 +010067 if (codec_specific.codecType == kVideoCodecVP8) {
Niels Möllerd3b8c632018-08-27 15:33:42 +020068 temporal_idx = codec_specific.codecSpecific.VP8.temporalIdx;
Rasmus Brandtd062a3c2018-03-08 16:45:54 +010069 } else if (codec_specific.codecType == kVideoCodecVP9) {
Niels Möllerd3b8c632018-08-27 15:33:42 +020070 temporal_idx = codec_specific.codecSpecific.VP9.temporal_idx;
Rasmus Brandtd062a3c2018-03-08 16:45:54 +010071 }
Niels Möllerd3b8c632018-08-27 15:33:42 +020072 if (temporal_idx == kNoTemporalIdx) {
73 temporal_idx = 0;
Rasmus Brandtd062a3c2018-03-08 16:45:54 +010074 }
Niels Möllerd3b8c632018-08-27 15:33:42 +020075 return temporal_idx;
Rasmus Brandtd062a3c2018-03-08 16:45:54 +010076}
77
asaperssonae9ba042017-03-07 00:25:38 -080078int GetElapsedTimeMicroseconds(int64_t start_ns, int64_t stop_ns) {
79 int64_t diff_us = (stop_ns - start_ns) / rtc::kNumNanosecsPerMicrosec;
80 RTC_DCHECK_GE(diff_us, std::numeric_limits<int>::min());
81 RTC_DCHECK_LE(diff_us, std::numeric_limits<int>::max());
82 return static_cast<int>(diff_us);
83}
84
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +010085void ExtractI420BufferWithSize(const VideoFrame& image,
86 int width,
87 int height,
88 rtc::Buffer* buffer) {
Åsa Perssonf0c44672017-10-24 16:03:39 +020089 if (image.width() != width || image.height() != height) {
90 EXPECT_DOUBLE_EQ(static_cast<double>(width) / height,
91 static_cast<double>(image.width()) / image.height());
92 // Same aspect ratio, no cropping needed.
93 rtc::scoped_refptr<I420Buffer> scaled(I420Buffer::Create(width, height));
94 scaled->ScaleFrom(*image.video_frame_buffer()->ToI420());
95
96 size_t length =
97 CalcBufferSize(VideoType::kI420, scaled->width(), scaled->height());
98 buffer->SetSize(length);
99 RTC_CHECK_NE(ExtractBuffer(scaled, length, buffer->data()), -1);
100 return;
101 }
102
103 // No resize.
104 size_t length =
105 CalcBufferSize(VideoType::kI420, image.width(), image.height());
106 buffer->SetSize(length);
107 RTC_CHECK_NE(ExtractBuffer(image, length, buffer->data()), -1);
108}
109
Sergey Silkin8d3758e2018-03-14 11:28:15 +0100110void CalculateFrameQuality(const I420BufferInterface& ref_buffer,
111 const I420BufferInterface& dec_buffer,
Sergey Silkin6e1402b2019-02-13 09:33:00 +0100112 FrameStatistics* frame_stat,
113 bool calc_ssim) {
Sergey Silkin8d3758e2018-03-14 11:28:15 +0100114 if (ref_buffer.width() != dec_buffer.width() ||
115 ref_buffer.height() != dec_buffer.height()) {
116 RTC_CHECK_GE(ref_buffer.width(), dec_buffer.width());
117 RTC_CHECK_GE(ref_buffer.height(), dec_buffer.height());
118 // Downscale reference frame.
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100119 rtc::scoped_refptr<I420Buffer> scaled_buffer =
Sergey Silkin8d3758e2018-03-14 11:28:15 +0100120 I420Buffer::Create(dec_buffer.width(), dec_buffer.height());
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100121 I420Scale(ref_buffer.DataY(), ref_buffer.StrideY(), ref_buffer.DataU(),
122 ref_buffer.StrideU(), ref_buffer.DataV(), ref_buffer.StrideV(),
123 ref_buffer.width(), ref_buffer.height(),
124 scaled_buffer->MutableDataY(), scaled_buffer->StrideY(),
125 scaled_buffer->MutableDataU(), scaled_buffer->StrideU(),
126 scaled_buffer->MutableDataV(), scaled_buffer->StrideV(),
127 scaled_buffer->width(), scaled_buffer->height(),
128 libyuv::kFilterBox);
Sergey Silkin8d3758e2018-03-14 11:28:15 +0100129
Sergey Silkin6e1402b2019-02-13 09:33:00 +0100130 CalculateFrameQuality(*scaled_buffer, dec_buffer, frame_stat, calc_ssim);
Sergey Silkin8d3758e2018-03-14 11:28:15 +0100131 } else {
132 const uint64_t sse_y = libyuv::ComputeSumSquareErrorPlane(
133 dec_buffer.DataY(), dec_buffer.StrideY(), ref_buffer.DataY(),
134 ref_buffer.StrideY(), dec_buffer.width(), dec_buffer.height());
135
136 const uint64_t sse_u = libyuv::ComputeSumSquareErrorPlane(
137 dec_buffer.DataU(), dec_buffer.StrideU(), ref_buffer.DataU(),
138 ref_buffer.StrideU(), dec_buffer.width() / 2, dec_buffer.height() / 2);
139
140 const uint64_t sse_v = libyuv::ComputeSumSquareErrorPlane(
141 dec_buffer.DataV(), dec_buffer.StrideV(), ref_buffer.DataV(),
142 ref_buffer.StrideV(), dec_buffer.width() / 2, dec_buffer.height() / 2);
143
144 const size_t num_y_samples = dec_buffer.width() * dec_buffer.height();
145 const size_t num_u_samples =
146 dec_buffer.width() / 2 * dec_buffer.height() / 2;
147
148 frame_stat->psnr_y = libyuv::SumSquareErrorToPsnr(sse_y, num_y_samples);
149 frame_stat->psnr_u = libyuv::SumSquareErrorToPsnr(sse_u, num_u_samples);
150 frame_stat->psnr_v = libyuv::SumSquareErrorToPsnr(sse_v, num_u_samples);
151 frame_stat->psnr = libyuv::SumSquareErrorToPsnr(
152 sse_y + sse_u + sse_v, num_y_samples + 2 * num_u_samples);
Sergey Silkin6e1402b2019-02-13 09:33:00 +0100153
154 if (calc_ssim) {
155 frame_stat->ssim = I420SSIM(ref_buffer, dec_buffer);
156 }
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100157 }
158}
159
brandtrb78bc752017-02-22 01:26:59 -0800160} // namespace
161
brandtrc4095522017-08-07 08:12:33 -0700162VideoProcessor::VideoProcessor(webrtc::VideoEncoder* encoder,
Sergey Silkin10d9d592018-02-01 13:25:17 +0100163 VideoDecoderList* decoders,
164 FrameReader* input_frame_reader,
Kári Tristan Helgason169005d2018-05-22 13:34:14 +0200165 const VideoCodecTestFixture::Config& config,
166 VideoCodecTestStats* stats,
Sergey Silkin10d9d592018-02-01 13:25:17 +0100167 IvfFileWriterList* encoded_frame_writers,
168 FrameWriterList* decoded_frame_writers)
Åsa Perssonf0c44672017-10-24 16:03:39 +0200169 : config_(config),
Sergey Silkin10d9d592018-02-01 13:25:17 +0100170 num_simulcast_or_spatial_layers_(
171 std::max(config_.NumberOfSimulcastStreams(),
172 config_.NumberOfSpatialLayers())),
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100173 stats_(stats),
brandtr07734a52017-08-08 08:35:53 -0700174 encoder_(encoder),
Sergey Silkin10d9d592018-02-01 13:25:17 +0100175 decoders_(decoders),
Jiawei Ouc2ebe212018-11-08 10:02:56 -0800176 bitrate_allocator_(
177 CreateBuiltinVideoBitrateAllocatorFactory()
178 ->CreateVideoBitrateAllocator(config_.codec_settings)),
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100179 framerate_fps_(0),
brandtrbdd555c2017-08-21 01:34:04 -0700180 encode_callback_(this),
Sergey Silkin10d9d592018-02-01 13:25:17 +0100181 input_frame_reader_(input_frame_reader),
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +0100182 merged_encoded_frames_(num_simulcast_or_spatial_layers_),
Sergey Silkin10d9d592018-02-01 13:25:17 +0100183 encoded_frame_writers_(encoded_frame_writers),
184 decoded_frame_writers_(decoded_frame_writers),
Sergey Silkin3be2a552018-01-17 15:11:44 +0100185 last_inputed_frame_num_(0),
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100186 last_inputed_timestamp_(0),
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +0100187 first_encoded_frame_(num_simulcast_or_spatial_layers_, true),
188 last_encoded_frame_num_(num_simulcast_or_spatial_layers_),
189 first_decoded_frame_(num_simulcast_or_spatial_layers_, true),
Sergey Silkinc89eed92018-04-01 23:57:51 +0200190 last_decoded_frame_num_(num_simulcast_or_spatial_layers_),
Sergey Silkin56138792018-05-02 10:50:55 +0200191 decoded_frame_buffer_(num_simulcast_or_spatial_layers_),
Sergey Silkinc89eed92018-04-01 23:57:51 +0200192 post_encode_time_ns_(0) {
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100193 // Sanity checks.
Rasmus Brandt4b381af2018-02-07 13:56:16 +0100194 RTC_CHECK(rtc::TaskQueue::Current())
195 << "VideoProcessor must be run on a task queue.";
Sergey Silkin10d9d592018-02-01 13:25:17 +0100196 RTC_CHECK(encoder);
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100197 RTC_CHECK(decoders);
198 RTC_CHECK_EQ(decoders->size(), num_simulcast_or_spatial_layers_);
Sergey Silkin10d9d592018-02-01 13:25:17 +0100199 RTC_CHECK(input_frame_reader);
200 RTC_CHECK(stats);
201 RTC_CHECK(!encoded_frame_writers ||
202 encoded_frame_writers->size() == num_simulcast_or_spatial_layers_);
203 RTC_CHECK(!decoded_frame_writers ||
204 decoded_frame_writers->size() == num_simulcast_or_spatial_layers_);
brandtr17b958c2017-03-07 01:41:43 -0800205
Sergey Silkin10d9d592018-02-01 13:25:17 +0100206 // Setup required callbacks for the encoder and decoder and initialize them.
brandtrbdd555c2017-08-21 01:34:04 -0700207 RTC_CHECK_EQ(encoder_->RegisterEncodeCompleteCallback(&encode_callback_),
Åsa Perssonf0c44672017-10-24 16:03:39 +0200208 WEBRTC_VIDEO_CODEC_OK);
asapersson654d54c2017-02-10 00:16:07 -0800209
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100210 // Initialize codecs so that they are ready to receive frames.
Sergey Silkin1723cf92018-01-22 15:49:55 +0100211 RTC_CHECK_EQ(encoder_->InitEncode(&config_.codec_settings,
212 static_cast<int>(config_.NumberOfCores()),
213 config_.max_payload_size_bytes),
214 WEBRTC_VIDEO_CODEC_OK);
Sergey Silkin645e2e02018-04-06 09:42:13 +0200215
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200216 for (size_t i = 0; i < num_simulcast_or_spatial_layers_; ++i) {
Sergey Silkin645e2e02018-04-06 09:42:13 +0200217 decode_callback_.push_back(
Karl Wiberg918f50c2018-07-05 11:40:33 +0200218 absl::make_unique<VideoProcessorDecodeCompleteCallback>(this, i));
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200219 RTC_CHECK_EQ(
220 decoders_->at(i)->InitDecode(&config_.codec_settings,
221 static_cast<int>(config_.NumberOfCores())),
222 WEBRTC_VIDEO_CODEC_OK);
223 RTC_CHECK_EQ(decoders_->at(i)->RegisterDecodeCompleteCallback(
224 decode_callback_.at(i).get()),
Sergey Silkin10d9d592018-02-01 13:25:17 +0100225 WEBRTC_VIDEO_CODEC_OK);
226 }
kjellander@webrtc.org35a17562011-10-06 06:44:54 +0000227}
228
Åsa Perssonf0c44672017-10-24 16:03:39 +0200229VideoProcessor::~VideoProcessor() {
brandtrc8c59052017-08-21 06:44:16 -0700230 RTC_DCHECK_CALLED_SEQUENTIALLY(&sequence_checker_);
231
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100232 // Explicitly reset codecs, in case they don't do that themselves when they
233 // go out of scope.
brandtr77920a42017-08-11 07:48:15 -0700234 RTC_CHECK_EQ(encoder_->Release(), WEBRTC_VIDEO_CODEC_OK);
brandtrbdd555c2017-08-21 01:34:04 -0700235 encoder_->RegisterEncodeCompleteCallback(nullptr);
Sergey Silkin10d9d592018-02-01 13:25:17 +0100236 for (auto& decoder : *decoders_) {
237 RTC_CHECK_EQ(decoder->Release(), WEBRTC_VIDEO_CODEC_OK);
238 decoder->RegisterDecodeCompleteCallback(nullptr);
239 }
240
Rasmus Brandtd00c8952018-03-14 12:29:57 +0100241 // Sanity check.
242 RTC_CHECK_LE(input_frames_.size(), kMaxBufferedInputFrames);
brandtr77920a42017-08-11 07:48:15 -0700243}
244
brandtr8935d972017-09-06 01:53:22 -0700245void VideoProcessor::ProcessFrame() {
brandtrc8c59052017-08-21 06:44:16 -0700246 RTC_DCHECK_CALLED_SEQUENTIALLY(&sequence_checker_);
Sergey Silkin3be2a552018-01-17 15:11:44 +0100247 const size_t frame_number = last_inputed_frame_num_++;
asapersson654d54c2017-02-10 00:16:07 -0800248
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100249 // Get input frame and store for future quality calculation.
250 rtc::scoped_refptr<I420BufferInterface> buffer =
251 input_frame_reader_->ReadFrame();
brandtrbdd555c2017-08-21 01:34:04 -0700252 RTC_CHECK(buffer) << "Tried to read too many frames from the file.";
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100253 const size_t timestamp =
254 last_inputed_timestamp_ + kVideoPayloadTypeFrequency / framerate_fps_;
Artem Titov1ebfb6a2019-01-03 23:49:37 +0100255 VideoFrame input_frame =
256 VideoFrame::Builder()
257 .set_video_frame_buffer(buffer)
258 .set_timestamp_rtp(static_cast<uint32_t>(timestamp))
259 .set_timestamp_ms(static_cast<int64_t>(timestamp / kMsToRtpTimestamp))
260 .set_rotation(webrtc::kVideoRotation_0)
261 .build();
Rasmus Brandtd00c8952018-03-14 12:29:57 +0100262 // Store input frame as a reference for quality calculations.
263 if (config_.decode && !config_.measure_cpu) {
Rasmus Brandt7b92ceb2018-06-26 12:34:59 +0200264 if (input_frames_.size() == kMaxBufferedInputFrames) {
265 input_frames_.erase(input_frames_.begin());
266 }
Rasmus Brandtd00c8952018-03-14 12:29:57 +0100267 input_frames_.emplace(frame_number, input_frame);
268 }
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100269 last_inputed_timestamp_ = timestamp;
brandtr17b958c2017-03-07 01:41:43 -0800270
Sergey Silkinc89eed92018-04-01 23:57:51 +0200271 post_encode_time_ns_ = 0;
272
Rasmus Brandt5f7a8912018-02-28 17:17:15 +0100273 // Create frame statistics object for all simulcast/spatial layers.
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200274 for (size_t i = 0; i < num_simulcast_or_spatial_layers_; ++i) {
Sergey Silkin02fed022018-09-25 13:48:19 +0200275 FrameStatistics frame_stat(frame_number, timestamp, i);
276 stats_->AddFrame(frame_stat);
Sergey Silkin10d9d592018-02-01 13:25:17 +0100277 }
brandtr17b958c2017-03-07 01:41:43 -0800278
279 // For the highest measurement accuracy of the encode time, the start/stop
280 // time recordings should wrap the Encode call as tightly as possible.
Sergey Silkin10d9d592018-02-01 13:25:17 +0100281 const int64_t encode_start_ns = rtc::TimeNanos();
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200282 for (size_t i = 0; i < num_simulcast_or_spatial_layers_; ++i) {
283 FrameStatistics* frame_stat = stats_->GetFrame(frame_number, i);
Sergey Silkin10d9d592018-02-01 13:25:17 +0100284 frame_stat->encode_start_ns = encode_start_ns;
285 }
286
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100287 // Encode.
288 const std::vector<FrameType> frame_types =
Rasmus Brandt7b3f4a22019-02-26 15:12:27 +0100289 (frame_number == 0) ? std::vector<FrameType>{kVideoFrameKey}
290 : std::vector<FrameType>{kVideoFrameDelta};
Sergey Silkin10d9d592018-02-01 13:25:17 +0100291 const int encode_return_code =
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100292 encoder_->Encode(input_frame, nullptr, &frame_types);
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200293 for (size_t i = 0; i < num_simulcast_or_spatial_layers_; ++i) {
294 FrameStatistics* frame_stat = stats_->GetFrame(frame_number, i);
Sergey Silkin10d9d592018-02-01 13:25:17 +0100295 frame_stat->encode_return_code = encode_return_code;
296 }
kjellander@webrtc.org35a17562011-10-06 06:44:54 +0000297}
298
Sergey Silkin3be2a552018-01-17 15:11:44 +0100299void VideoProcessor::SetRates(size_t bitrate_kbps, size_t framerate_fps) {
brandtrc8c59052017-08-21 06:44:16 -0700300 RTC_DCHECK_CALLED_SEQUENTIALLY(&sequence_checker_);
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100301 framerate_fps_ = static_cast<uint32_t>(framerate_fps);
Sergey Silkin3be2a552018-01-17 15:11:44 +0100302 bitrate_allocation_ = bitrate_allocator_->GetAllocation(
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100303 static_cast<uint32_t>(bitrate_kbps * 1000), framerate_fps_);
304 const int set_rates_result =
305 encoder_->SetRateAllocation(bitrate_allocation_, framerate_fps_);
brandtrbea36fd2017-08-07 03:36:54 -0700306 RTC_DCHECK_GE(set_rates_result, 0)
brandtrbdd555c2017-08-21 01:34:04 -0700307 << "Failed to update encoder with new rate " << bitrate_kbps << ".";
brandtrbea36fd2017-08-07 03:36:54 -0700308}
309
Sami Kalliomäki451b29c2018-07-04 14:33:51 +0200310int32_t VideoProcessor::VideoProcessorDecodeCompleteCallback::Decoded(
311 VideoFrame& image) {
312 // Post the callback to the right task queue, if needed.
313 if (!task_queue_->IsCurrent()) {
314 // There might be a limited amount of output buffers, make a copy to make
315 // sure we don't block the decoder.
Artem Titov1ebfb6a2019-01-03 23:49:37 +0100316 VideoFrame copy = VideoFrame::Builder()
317 .set_video_frame_buffer(I420Buffer::Copy(
318 *image.video_frame_buffer()->ToI420()))
319 .set_rotation(image.rotation())
320 .set_timestamp_us(image.timestamp_us())
321 .set_id(image.id())
322 .build();
Sami Kalliomäki451b29c2018-07-04 14:33:51 +0200323 copy.set_timestamp(image.timestamp());
324
325 task_queue_->PostTask([this, copy]() {
326 video_processor_->FrameDecoded(copy, simulcast_svc_idx_);
327 });
328 return 0;
329 }
330 video_processor_->FrameDecoded(image, simulcast_svc_idx_);
331 return 0;
332}
333
Sergey Silkin10d9d592018-02-01 13:25:17 +0100334void VideoProcessor::FrameEncoded(
335 const webrtc::EncodedImage& encoded_image,
336 const webrtc::CodecSpecificInfo& codec_specific) {
brandtrc8c59052017-08-21 06:44:16 -0700337 RTC_DCHECK_CALLED_SEQUENTIALLY(&sequence_checker_);
338
brandtr32e0d262017-02-15 05:29:38 -0800339 // For the highest measurement accuracy of the encode time, the start/stop
340 // time recordings should wrap the Encode call as tightly as possible.
Rasmus Brandt5f7a8912018-02-28 17:17:15 +0100341 const int64_t encode_stop_ns = rtc::TimeNanos();
brandtr32e0d262017-02-15 05:29:38 -0800342
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100343 const VideoCodecType codec_type = codec_specific.codecType;
Rasmus Brandtf7a35582017-10-24 10:16:33 +0200344 if (config_.encoded_frame_checker) {
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100345 config_.encoded_frame_checker->CheckEncodedFrame(codec_type, encoded_image);
Rasmus Brandtf7a35582017-10-24 10:16:33 +0200346 }
brandtrb78bc752017-02-22 01:26:59 -0800347
Rasmus Brandtd062a3c2018-03-08 16:45:54 +0100348 // Layer metadata.
Niels Möllerd3b8c632018-08-27 15:33:42 +0200349 size_t spatial_idx = encoded_image.SpatialIndex().value_or(0);
350 size_t temporal_idx = GetTemporalLayerIndex(codec_specific);
Sergey Silkin10d9d592018-02-01 13:25:17 +0100351
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200352 FrameStatistics* frame_stat =
Niels Möller23775882018-08-16 10:24:12 +0200353 stats_->GetFrameWithTimestamp(encoded_image.Timestamp(), spatial_idx);
Åsa Perssona6e7b882018-01-19 14:57:10 +0100354 const size_t frame_number = frame_stat->frame_number;
Sergey Silkin10d9d592018-02-01 13:25:17 +0100355
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +0100356 // Ensure that the encode order is monotonically increasing, within this
357 // simulcast/spatial layer.
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200358 RTC_CHECK(first_encoded_frame_[spatial_idx] ||
359 last_encoded_frame_num_[spatial_idx] < frame_number);
Sergey Silkin10d9d592018-02-01 13:25:17 +0100360
361 // Ensure SVC spatial layers are delivered in ascending order.
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200362 if (!first_encoded_frame_[spatial_idx] &&
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +0100363 config_.NumberOfSpatialLayers() > 1) {
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200364 for (size_t i = 0; i < spatial_idx; ++i) {
Sergey Silkin122ba6c2018-03-27 14:32:21 +0200365 RTC_CHECK_LE(last_encoded_frame_num_[i], frame_number);
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +0100366 }
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200367 for (size_t i = spatial_idx + 1; i < num_simulcast_or_spatial_layers_;
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +0100368 ++i) {
369 RTC_CHECK_GT(frame_number, last_encoded_frame_num_[i]);
370 }
Sergey Silkin3be2a552018-01-17 15:11:44 +0100371 }
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200372 first_encoded_frame_[spatial_idx] = false;
373 last_encoded_frame_num_[spatial_idx] = frame_number;
brandtr17b958c2017-03-07 01:41:43 -0800374
brandtr8935d972017-09-06 01:53:22 -0700375 // Update frame statistics.
Sergey Silkin10d9d592018-02-01 13:25:17 +0100376 frame_stat->encoding_successful = true;
Sergey Silkinc89eed92018-04-01 23:57:51 +0200377 frame_stat->encode_time_us = GetElapsedTimeMicroseconds(
378 frame_stat->encode_start_ns, encode_stop_ns - post_encode_time_ns_);
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200379 frame_stat->target_bitrate_kbps =
Sergey Silkinbfd54ef2018-04-13 23:41:11 +0200380 bitrate_allocation_.GetTemporalLayerSum(spatial_idx, temporal_idx) / 1000;
Niels Möller77536a22019-01-15 08:50:01 +0100381 frame_stat->length_bytes = encoded_image.size();
brandtr17b958c2017-03-07 01:41:43 -0800382 frame_stat->frame_type = encoded_image._frameType;
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200383 frame_stat->temporal_idx = temporal_idx;
Sergey Silkin3be2a552018-01-17 15:11:44 +0100384 frame_stat->max_nalu_size_bytes = GetMaxNaluSizeBytes(encoded_image, config_);
Sergey Silkin10d9d592018-02-01 13:25:17 +0100385 frame_stat->qp = encoded_image.qp_;
ssilkin612f8582017-09-28 09:23:17 -0700386
Sergey Silkin645e2e02018-04-06 09:42:13 +0200387 const size_t num_spatial_layers = config_.NumberOfSpatialLayers();
Sergey Silkinbc0f0d32018-04-24 21:29:14 +0200388 bool end_of_picture = false;
Sergey Silkin645e2e02018-04-06 09:42:13 +0200389 if (codec_type == kVideoCodecVP9) {
390 const CodecSpecificInfoVP9& vp9_info = codec_specific.codecSpecific.VP9;
391 frame_stat->inter_layer_predicted = vp9_info.inter_layer_predicted;
Sergey Silkin3c30c9c2018-05-02 09:18:48 +0200392 frame_stat->non_ref_for_inter_layer_pred =
393 vp9_info.non_ref_for_inter_layer_pred;
Sergey Silkinbc0f0d32018-04-24 21:29:14 +0200394 end_of_picture = vp9_info.end_of_picture;
Sergey Silkin3c30c9c2018-05-02 09:18:48 +0200395 } else {
396 frame_stat->inter_layer_predicted = false;
397 frame_stat->non_ref_for_inter_layer_pred = true;
Sergey Silkin645e2e02018-04-06 09:42:13 +0200398 }
399
Sergey Silkin122ba6c2018-03-27 14:32:21 +0200400 const webrtc::EncodedImage* encoded_image_for_decode = &encoded_image;
Sergey Silkin645e2e02018-04-06 09:42:13 +0200401 if (config_.decode || encoded_frame_writers_) {
402 if (num_spatial_layers > 1) {
403 encoded_image_for_decode = BuildAndStoreSuperframe(
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200404 encoded_image, codec_type, frame_number, spatial_idx,
Sergey Silkin645e2e02018-04-06 09:42:13 +0200405 frame_stat->inter_layer_predicted);
Rasmus Brandtd00c8952018-03-14 12:29:57 +0100406 }
Sergey Silkin645e2e02018-04-06 09:42:13 +0200407 }
408
409 if (config_.decode) {
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200410 DecodeFrame(*encoded_image_for_decode, spatial_idx);
Sergey Silkin645e2e02018-04-06 09:42:13 +0200411
Sergey Silkin3c30c9c2018-05-02 09:18:48 +0200412 if (end_of_picture && num_spatial_layers > 1) {
Sergey Silkin645e2e02018-04-06 09:42:13 +0200413 // If inter-layer prediction is enabled and upper layer was dropped then
414 // base layer should be passed to upper layer decoder. Otherwise decoder
415 // won't be able to decode next superframe.
416 const EncodedImage* base_image = nullptr;
Sergey Silkin3c30c9c2018-05-02 09:18:48 +0200417 const FrameStatistics* base_stat = nullptr;
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200418 for (size_t i = 0; i < num_spatial_layers; ++i) {
Sergey Silkin3c30c9c2018-05-02 09:18:48 +0200419 const bool layer_dropped = (first_decoded_frame_[i] ||
420 last_decoded_frame_num_[i] < frame_number);
Sergey Silkin645e2e02018-04-06 09:42:13 +0200421
422 // Ensure current layer was decoded.
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200423 RTC_CHECK(layer_dropped == false || i != spatial_idx);
Sergey Silkin645e2e02018-04-06 09:42:13 +0200424
425 if (!layer_dropped) {
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200426 base_image = &merged_encoded_frames_[i];
Sergey Silkin3c30c9c2018-05-02 09:18:48 +0200427 base_stat =
Niels Möller23775882018-08-16 10:24:12 +0200428 stats_->GetFrameWithTimestamp(encoded_image.Timestamp(), i);
Sergey Silkin3c30c9c2018-05-02 09:18:48 +0200429 } else if (base_image && !base_stat->non_ref_for_inter_layer_pred) {
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200430 DecodeFrame(*base_image, i);
Sergey Silkin645e2e02018-04-06 09:42:13 +0200431 }
432 }
433 }
Rasmus Brandtd00c8952018-03-14 12:29:57 +0100434 } else {
435 frame_stat->decode_return_code = WEBRTC_VIDEO_CODEC_NO_OUTPUT;
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +0100436 }
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +0100437
438 if (encoded_frame_writers_) {
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200439 RTC_CHECK(encoded_frame_writers_->at(spatial_idx)
Sergey Silkin122ba6c2018-03-27 14:32:21 +0200440 ->WriteFrame(*encoded_image_for_decode,
441 config_.codec_settings.codecType));
brandtr8935d972017-09-06 01:53:22 -0700442 }
Sergey Silkinc89eed92018-04-01 23:57:51 +0200443
Kári Tristan Helgasonf1677622018-08-24 13:21:26 +0200444 if (!config_.encode_in_real_time) {
Sergey Silkinc89eed92018-04-01 23:57:51 +0200445 // To get pure encode time for next layers, measure time spent in encode
446 // callback and subtract it from encode time of next layers.
447 post_encode_time_ns_ += rtc::TimeNanos() - encode_stop_ns;
448 }
kjellander@webrtc.org35a17562011-10-06 06:44:54 +0000449}
450
Sergey Silkin645e2e02018-04-06 09:42:13 +0200451void VideoProcessor::FrameDecoded(const VideoFrame& decoded_frame,
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200452 size_t spatial_idx) {
brandtrc8c59052017-08-21 06:44:16 -0700453 RTC_DCHECK_CALLED_SEQUENTIALLY(&sequence_checker_);
454
brandtr32e0d262017-02-15 05:29:38 -0800455 // For the highest measurement accuracy of the decode time, the start/stop
456 // time recordings should wrap the Decode call as tightly as possible.
Rasmus Brandt5f7a8912018-02-28 17:17:15 +0100457 const int64_t decode_stop_ns = rtc::TimeNanos();
brandtr8bc93852017-02-15 05:19:51 -0800458
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200459 FrameStatistics* frame_stat =
460 stats_->GetFrameWithTimestamp(decoded_frame.timestamp(), spatial_idx);
Åsa Perssona6e7b882018-01-19 14:57:10 +0100461 const size_t frame_number = frame_stat->frame_number;
Sergey Silkin64eaa992017-11-17 14:47:32 +0100462
Sergey Silkin56138792018-05-02 10:50:55 +0200463 if (decoded_frame_writers_ && !first_decoded_frame_[spatial_idx]) {
464 // Fill drops with last decoded frame to make them look like freeze at
465 // playback and to keep decoded layers in sync.
466 for (size_t i = last_decoded_frame_num_[spatial_idx] + 1; i < frame_number;
467 ++i) {
468 RTC_CHECK(decoded_frame_writers_->at(spatial_idx)
469 ->WriteFrame(decoded_frame_buffer_[spatial_idx].data()));
470 }
471 }
472
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +0100473 // Ensure that the decode order is monotonically increasing, within this
474 // simulcast/spatial layer.
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200475 RTC_CHECK(first_decoded_frame_[spatial_idx] ||
476 last_decoded_frame_num_[spatial_idx] < frame_number);
477 first_decoded_frame_[spatial_idx] = false;
478 last_decoded_frame_num_[spatial_idx] = frame_number;
brandtr17b958c2017-03-07 01:41:43 -0800479
Sergey Silkin10d9d592018-02-01 13:25:17 +0100480 // Update frame statistics.
481 frame_stat->decoding_successful = true;
482 frame_stat->decode_time_us =
483 GetElapsedTimeMicroseconds(frame_stat->decode_start_ns, decode_stop_ns);
484 frame_stat->decoded_width = decoded_frame.width();
485 frame_stat->decoded_height = decoded_frame.height();
486
Sergey Silkin64eaa992017-11-17 14:47:32 +0100487 // Skip quality metrics calculation to not affect CPU usage.
488 if (!config_.measure_cpu) {
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +0100489 const auto reference_frame = input_frames_.find(frame_number);
490 RTC_CHECK(reference_frame != input_frames_.cend())
491 << "The codecs are either buffering too much, dropping too much, or "
492 "being too slow relative the input frame rate.";
Sergey Silkin6e1402b2019-02-13 09:33:00 +0100493
494 // SSIM calculation is not optimized. Skip it in real-time mode.
495 const bool calc_ssim = !config_.encode_in_real_time;
Sergey Silkin8d3758e2018-03-14 11:28:15 +0100496 CalculateFrameQuality(
497 *reference_frame->second.video_frame_buffer()->ToI420(),
Sergey Silkin6e1402b2019-02-13 09:33:00 +0100498 *decoded_frame.video_frame_buffer()->ToI420(), frame_stat, calc_ssim);
Niels Möller718a7632016-06-13 13:06:01 +0200499
Rasmus Brandtd00c8952018-03-14 12:29:57 +0100500 // Erase all buffered input frames that we have moved past for all
501 // simulcast/spatial layers. Never buffer more than
502 // |kMaxBufferedInputFrames| frames, to protect against long runs of
503 // consecutive frame drops for a particular layer.
504 const auto min_last_decoded_frame_num = std::min_element(
505 last_decoded_frame_num_.cbegin(), last_decoded_frame_num_.cend());
506 const size_t min_buffered_frame_num = std::max(
507 0, static_cast<int>(frame_number) - kMaxBufferedInputFrames + 1);
508 RTC_CHECK(min_last_decoded_frame_num != last_decoded_frame_num_.cend());
509 const auto input_frames_erase_before = input_frames_.lower_bound(
510 std::max(*min_last_decoded_frame_num, min_buffered_frame_num));
511 input_frames_.erase(input_frames_.cbegin(), input_frames_erase_before);
512 }
Sergey Silkin64eaa992017-11-17 14:47:32 +0100513
Sergey Silkin10d9d592018-02-01 13:25:17 +0100514 if (decoded_frame_writers_) {
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +0100515 ExtractI420BufferWithSize(decoded_frame, config_.codec_settings.width,
Sergey Silkin56138792018-05-02 10:50:55 +0200516 config_.codec_settings.height,
517 &decoded_frame_buffer_[spatial_idx]);
518 RTC_CHECK_EQ(decoded_frame_buffer_[spatial_idx].size(),
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200519 decoded_frame_writers_->at(spatial_idx)->FrameLength());
520 RTC_CHECK(decoded_frame_writers_->at(spatial_idx)
Sergey Silkin56138792018-05-02 10:50:55 +0200521 ->WriteFrame(decoded_frame_buffer_[spatial_idx].data()));
Sergey Silkin64eaa992017-11-17 14:47:32 +0100522 }
Åsa Perssonf0c44672017-10-24 16:03:39 +0200523}
brandtr17b958c2017-03-07 01:41:43 -0800524
Sergey Silkin645e2e02018-04-06 09:42:13 +0200525void VideoProcessor::DecodeFrame(const EncodedImage& encoded_image,
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200526 size_t spatial_idx) {
Sergey Silkin645e2e02018-04-06 09:42:13 +0200527 RTC_DCHECK_CALLED_SEQUENTIALLY(&sequence_checker_);
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200528 FrameStatistics* frame_stat =
Niels Möller23775882018-08-16 10:24:12 +0200529 stats_->GetFrameWithTimestamp(encoded_image.Timestamp(), spatial_idx);
Sergey Silkin645e2e02018-04-06 09:42:13 +0200530
531 frame_stat->decode_start_ns = rtc::TimeNanos();
532 frame_stat->decode_return_code =
Niels Möller401d0762018-05-08 11:54:29 +0200533 decoders_->at(spatial_idx)->Decode(encoded_image, false, nullptr, 0);
Sergey Silkin645e2e02018-04-06 09:42:13 +0200534}
535
536const webrtc::EncodedImage* VideoProcessor::BuildAndStoreSuperframe(
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +0100537 const EncodedImage& encoded_image,
538 const VideoCodecType codec,
539 size_t frame_number,
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200540 size_t spatial_idx,
Sergey Silkin645e2e02018-04-06 09:42:13 +0200541 bool inter_layer_predicted) {
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +0100542 // Should only be called for SVC.
543 RTC_CHECK_GT(config_.NumberOfSpatialLayers(), 1);
Sergey Silkin10d9d592018-02-01 13:25:17 +0100544
545 EncodedImage base_image;
Niels Möller77536a22019-01-15 08:50:01 +0100546 RTC_CHECK_EQ(base_image.size(), 0);
Sergey Silkin10d9d592018-02-01 13:25:17 +0100547
Sergey Silkin122ba6c2018-03-27 14:32:21 +0200548 // Each SVC layer is decoded with dedicated decoder. Find the nearest
549 // non-dropped base frame and merge it and current frame into superframe.
Sergey Silkin645e2e02018-04-06 09:42:13 +0200550 if (inter_layer_predicted) {
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200551 for (int base_idx = static_cast<int>(spatial_idx) - 1; base_idx >= 0;
Sergey Silkin122ba6c2018-03-27 14:32:21 +0200552 --base_idx) {
553 EncodedImage lower_layer = merged_encoded_frames_.at(base_idx);
Niels Möller23775882018-08-16 10:24:12 +0200554 if (lower_layer.Timestamp() == encoded_image.Timestamp()) {
Sergey Silkin122ba6c2018-03-27 14:32:21 +0200555 base_image = lower_layer;
556 break;
557 }
558 }
Sergey Silkin10d9d592018-02-01 13:25:17 +0100559 }
Niels Möller77536a22019-01-15 08:50:01 +0100560 const size_t payload_size_bytes = base_image.size() + encoded_image.size();
Sergey Silkin10d9d592018-02-01 13:25:17 +0100561 const size_t buffer_size_bytes =
562 payload_size_bytes + EncodedImage::GetBufferPaddingBytes(codec);
563
Niels Möller663844d2019-02-14 16:15:54 +0100564 EncodedImage copied_image = encoded_image;
565 copied_image.Allocate(buffer_size_bytes);
Niels Möller77536a22019-01-15 08:50:01 +0100566 if (base_image.size()) {
Niels Möller24871e42019-01-17 11:31:13 +0100567 RTC_CHECK(base_image.data());
Niels Möller663844d2019-02-14 16:15:54 +0100568 memcpy(copied_image.data(), base_image.data(), base_image.size());
Sergey Silkin10d9d592018-02-01 13:25:17 +0100569 }
Niels Möller663844d2019-02-14 16:15:54 +0100570 memcpy(copied_image.data() + base_image.size(), encoded_image.data(),
Niels Möller77536a22019-01-15 08:50:01 +0100571 encoded_image.size());
Sergey Silkin10d9d592018-02-01 13:25:17 +0100572
Niels Möller77536a22019-01-15 08:50:01 +0100573 copied_image.set_size(payload_size_bytes);
Sergey Silkin10d9d592018-02-01 13:25:17 +0100574
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +0100575 // Replace previous EncodedImage for this spatial layer.
Niels Möller663844d2019-02-14 16:15:54 +0100576 merged_encoded_frames_.at(spatial_idx) = std::move(copied_image);
Rasmus Brandt0f1c0bd2018-03-12 10:01:16 +0100577
Sergey Silkinbc20fe12018-04-15 21:06:16 +0200578 return &merged_encoded_frames_.at(spatial_idx);
Sergey Silkin10d9d592018-02-01 13:25:17 +0100579}
580
kjellander@webrtc.org35a17562011-10-06 06:44:54 +0000581} // namespace test
582} // namespace webrtc