mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-13 13:50:40 +01:00

vp9 encoder wrapper rely on that behavioue when generates vp9-specific temporal references Bug: webrtc:11999 Change-Id: Ie1b4cb60adf290992cc3307b56397a88eda78be4 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/211661 Reviewed-by: Philip Eliasson <philipel@webrtc.org> Commit-Queue: Danil Chapovalov <danilchap@webrtc.org> Cr-Commit-Position: refs/heads/master@{#33446}
102 lines
3.7 KiB
C++
102 lines
3.7 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/svc/scalability_structure_full_svc.h"
|
|
|
|
#include <vector>
|
|
|
|
#include "modules/video_coding/svc/scalability_structure_test_helpers.h"
|
|
#include "test/gmock.h"
|
|
#include "test/gtest.h"
|
|
|
|
namespace webrtc {
|
|
namespace {
|
|
|
|
using ::testing::IsEmpty;
|
|
using ::testing::SizeIs;
|
|
|
|
TEST(ScalabilityStructureL3T3Test, SkipS1T1FrameKeepsStructureValid) {
|
|
ScalabilityStructureL3T3 structure;
|
|
ScalabilityStructureWrapper wrapper(structure);
|
|
|
|
structure.OnRatesUpdated(EnableTemporalLayers(/*s0=*/3, /*s1=*/3));
|
|
auto frames = wrapper.GenerateFrames(/*num_temporal_units=*/1);
|
|
EXPECT_THAT(frames, SizeIs(2));
|
|
EXPECT_EQ(frames[0].temporal_id, 0);
|
|
|
|
frames = wrapper.GenerateFrames(/*num_temporal_units=*/1);
|
|
EXPECT_THAT(frames, SizeIs(2));
|
|
EXPECT_EQ(frames[0].temporal_id, 2);
|
|
|
|
structure.OnRatesUpdated(EnableTemporalLayers(/*s0=*/3, /*s1=*/0));
|
|
frames = wrapper.GenerateFrames(/*num_temporal_units=*/1);
|
|
EXPECT_THAT(frames, SizeIs(1));
|
|
EXPECT_EQ(frames[0].temporal_id, 1);
|
|
|
|
structure.OnRatesUpdated(EnableTemporalLayers(/*s0=*/3, /*s1=*/3));
|
|
// Rely on checks inside GenerateFrames frame references are valid.
|
|
frames = wrapper.GenerateFrames(/*num_temporal_units=*/1);
|
|
EXPECT_THAT(frames, SizeIs(2));
|
|
EXPECT_EQ(frames[0].temporal_id, 2);
|
|
}
|
|
|
|
TEST(ScalabilityStructureL3T3Test, SkipT1FrameByEncoderKeepsReferencesValid) {
|
|
std::vector<GenericFrameInfo> frames;
|
|
ScalabilityStructureL3T3 structure;
|
|
ScalabilityStructureWrapper wrapper(structure);
|
|
|
|
// 1st 2 temporal units (T0 and T2)
|
|
wrapper.GenerateFrames(/*num_temporal_units=*/2, frames);
|
|
// Simulate T1 frame dropped by the encoder,
|
|
// i.e. retrieve config, but skip calling OnEncodeDone.
|
|
structure.NextFrameConfig(/*restart=*/false);
|
|
// one more temporal units (T2)
|
|
wrapper.GenerateFrames(/*num_temporal_units=*/1, frames);
|
|
|
|
EXPECT_TRUE(wrapper.FrameReferencesAreValid(frames));
|
|
}
|
|
|
|
TEST(ScalabilityStructureL3T3Test,
|
|
SkippingFrameReusePreviousFrameConfiguration) {
|
|
std::vector<GenericFrameInfo> frames;
|
|
ScalabilityStructureL3T3 structure;
|
|
ScalabilityStructureWrapper wrapper(structure);
|
|
|
|
// 1st 2 temporal units (T0 and T2)
|
|
wrapper.GenerateFrames(/*num_temporal_units=*/2, frames);
|
|
ASSERT_THAT(frames, SizeIs(6));
|
|
ASSERT_EQ(frames[0].temporal_id, 0);
|
|
ASSERT_EQ(frames[3].temporal_id, 2);
|
|
|
|
// Simulate a frame dropped by the encoder,
|
|
// i.e. retrieve config, but skip calling OnEncodeDone.
|
|
structure.NextFrameConfig(/*restart=*/false);
|
|
// two more temporal unit, expect temporal pattern continues
|
|
wrapper.GenerateFrames(/*num_temporal_units=*/2, frames);
|
|
ASSERT_THAT(frames, SizeIs(12));
|
|
// Expect temporal pattern continues as if there were no dropped frames.
|
|
EXPECT_EQ(frames[6].temporal_id, 1);
|
|
EXPECT_EQ(frames[9].temporal_id, 2);
|
|
}
|
|
|
|
TEST(ScalabilityStructureL3T3Test, SwitchSpatialLayerBeforeT1Frame) {
|
|
ScalabilityStructureL3T3 structure;
|
|
ScalabilityStructureWrapper wrapper(structure);
|
|
|
|
structure.OnRatesUpdated(EnableTemporalLayers(/*s0=*/2, /*s1=*/0));
|
|
EXPECT_THAT(wrapper.GenerateFrames(1), SizeIs(1));
|
|
structure.OnRatesUpdated(EnableTemporalLayers(/*s0=*/0, /*s1=*/2));
|
|
auto frames = wrapper.GenerateFrames(1);
|
|
ASSERT_THAT(frames, SizeIs(1));
|
|
EXPECT_THAT(frames[0].frame_diffs, IsEmpty());
|
|
EXPECT_EQ(frames[0].temporal_id, 0);
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace webrtc
|