mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-13 05:40:42 +01:00
b942d45110
Absent fps allocation imply single layer stream which confuses bitrate adjuster. As a result bitrate adjuster turned off S0T1 and S0T2 layers for the L3T3 structure. Bug: webrtc:12148 Change-Id: I5b3a7b44322f347f41dd8858b3d703827e69dd72 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/201384 Reviewed-by: Erik Språng <sprang@webrtc.org> Commit-Queue: Danil Chapovalov <danilchap@webrtc.org> Cr-Commit-Position: refs/heads/master@{#32952}
123 lines
4.8 KiB
C++
123 lines
4.8 KiB
C++
/*
|
|
* Copyright (c) 2020 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/video_coding/codecs/av1/libaom_av1_encoder.h"
|
|
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include "absl/types/optional.h"
|
|
#include "api/video_codecs/video_codec.h"
|
|
#include "api/video_codecs/video_encoder.h"
|
|
#include "modules/video_coding/codecs/test/encoded_video_frame_producer.h"
|
|
#include "modules/video_coding/include/video_error_codes.h"
|
|
#include "modules/video_coding/svc/scalability_structure_l1t2.h"
|
|
#include "test/gmock.h"
|
|
#include "test/gtest.h"
|
|
|
|
namespace webrtc {
|
|
namespace {
|
|
|
|
using ::testing::ElementsAre;
|
|
using ::testing::IsEmpty;
|
|
using ::testing::SizeIs;
|
|
|
|
VideoCodec DefaultCodecSettings() {
|
|
VideoCodec codec_settings;
|
|
codec_settings.width = 320;
|
|
codec_settings.height = 180;
|
|
codec_settings.maxFramerate = 30;
|
|
codec_settings.maxBitrate = 1000;
|
|
codec_settings.qpMax = 63;
|
|
return codec_settings;
|
|
}
|
|
|
|
VideoEncoder::Settings DefaultEncoderSettings() {
|
|
return VideoEncoder::Settings(
|
|
VideoEncoder::Capabilities(/*loss_notification=*/false),
|
|
/*number_of_cores=*/1, /*max_payload_size=*/1200);
|
|
}
|
|
|
|
TEST(LibaomAv1EncoderTest, CanCreate) {
|
|
std::unique_ptr<VideoEncoder> encoder = CreateLibaomAv1Encoder();
|
|
EXPECT_TRUE(encoder);
|
|
}
|
|
|
|
TEST(LibaomAv1EncoderTest, InitAndRelease) {
|
|
std::unique_ptr<VideoEncoder> encoder = CreateLibaomAv1Encoder();
|
|
ASSERT_TRUE(encoder);
|
|
VideoCodec codec_settings = DefaultCodecSettings();
|
|
EXPECT_EQ(encoder->InitEncode(&codec_settings, DefaultEncoderSettings()),
|
|
WEBRTC_VIDEO_CODEC_OK);
|
|
EXPECT_EQ(encoder->Release(), WEBRTC_VIDEO_CODEC_OK);
|
|
}
|
|
|
|
TEST(LibaomAv1EncoderTest, NoBitrateOnTopLayerRefecltedInActiveDecodeTargets) {
|
|
// Configure encoder with 2 temporal layers.
|
|
std::unique_ptr<VideoEncoder> encoder =
|
|
CreateLibaomAv1Encoder(std::make_unique<ScalabilityStructureL1T2>());
|
|
VideoCodec codec_settings = DefaultCodecSettings();
|
|
ASSERT_EQ(encoder->InitEncode(&codec_settings, DefaultEncoderSettings()),
|
|
WEBRTC_VIDEO_CODEC_OK);
|
|
|
|
VideoEncoder::RateControlParameters rate_parameters;
|
|
rate_parameters.framerate_fps = 30;
|
|
rate_parameters.bitrate.SetBitrate(0, /*temporal_index=*/0, 300'000);
|
|
rate_parameters.bitrate.SetBitrate(0, /*temporal_index=*/1, 0);
|
|
encoder->SetRates(rate_parameters);
|
|
|
|
std::vector<EncodedVideoFrameProducer::EncodedFrame> encoded_frames =
|
|
EncodedVideoFrameProducer(*encoder).SetNumInputFrames(1).Encode();
|
|
ASSERT_THAT(encoded_frames, SizeIs(1));
|
|
ASSERT_NE(encoded_frames[0].codec_specific_info.generic_frame_info,
|
|
absl::nullopt);
|
|
// Assuming L1T2 structure uses 1st decode target for T0 and 2nd decode target
|
|
// for T0+T1 frames, expect only 1st decode target is active.
|
|
EXPECT_EQ(encoded_frames[0]
|
|
.codec_specific_info.generic_frame_info->active_decode_targets,
|
|
0b01);
|
|
}
|
|
|
|
TEST(LibaomAv1EncoderTest, SetsEndOfPictureForLastFrameInTemporalUnit) {
|
|
std::unique_ptr<VideoEncoder> encoder = CreateLibaomAv1Encoder();
|
|
VideoCodec codec_settings = DefaultCodecSettings();
|
|
// Configure encoder with 3 spatial layers.
|
|
codec_settings.SetScalabilityMode("L3T1");
|
|
ASSERT_EQ(encoder->InitEncode(&codec_settings, DefaultEncoderSettings()),
|
|
WEBRTC_VIDEO_CODEC_OK);
|
|
|
|
std::vector<EncodedVideoFrameProducer::EncodedFrame> encoded_frames =
|
|
EncodedVideoFrameProducer(*encoder).SetNumInputFrames(2).Encode();
|
|
ASSERT_THAT(encoded_frames, SizeIs(6));
|
|
EXPECT_FALSE(encoded_frames[0].codec_specific_info.end_of_picture);
|
|
EXPECT_FALSE(encoded_frames[1].codec_specific_info.end_of_picture);
|
|
EXPECT_TRUE(encoded_frames[2].codec_specific_info.end_of_picture);
|
|
EXPECT_FALSE(encoded_frames[3].codec_specific_info.end_of_picture);
|
|
EXPECT_FALSE(encoded_frames[4].codec_specific_info.end_of_picture);
|
|
EXPECT_TRUE(encoded_frames[5].codec_specific_info.end_of_picture);
|
|
}
|
|
|
|
TEST(LibaomAv1EncoderTest, EncoderInfoProvidesFpsAllocation) {
|
|
std::unique_ptr<VideoEncoder> encoder = CreateLibaomAv1Encoder();
|
|
VideoCodec codec_settings = DefaultCodecSettings();
|
|
codec_settings.SetScalabilityMode("L3T3");
|
|
codec_settings.maxFramerate = 60;
|
|
ASSERT_EQ(encoder->InitEncode(&codec_settings, DefaultEncoderSettings()),
|
|
WEBRTC_VIDEO_CODEC_OK);
|
|
|
|
const auto& encoder_info = encoder->GetEncoderInfo();
|
|
EXPECT_THAT(encoder_info.fps_allocation[0], ElementsAre(15, 30, 60));
|
|
EXPECT_THAT(encoder_info.fps_allocation[1], ElementsAre(15, 30, 60));
|
|
EXPECT_THAT(encoder_info.fps_allocation[2], ElementsAre(15, 30, 60));
|
|
EXPECT_THAT(encoder_info.fps_allocation[3], IsEmpty());
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace webrtc
|