| /* |
| * 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 <cstring> |
| #include <map> |
| #include <set> |
| #include <utility> |
| #include <vector> |
| |
| #include "modules/video_coding/frame_object.h" |
| #include "modules/video_coding/rtp_vp9_ref_finder.h" |
| #include "rtc_base/random.h" |
| #include "test/gtest.h" |
| |
| namespace webrtc { |
| namespace video_coding { |
| |
| namespace { |
| std::unique_ptr<RtpFrameObject> CreateFrame( |
| uint16_t seq_num_start, |
| uint16_t seq_num_end, |
| bool keyframe, |
| VideoCodecType codec, |
| const RTPVideoTypeHeader& video_type_header) { |
| RTPVideoHeader video_header; |
| video_header.frame_type = keyframe ? VideoFrameType::kVideoFrameKey |
| : VideoFrameType::kVideoFrameDelta; |
| video_header.video_type_header = video_type_header; |
| |
| // clang-format off |
| return std::make_unique<RtpFrameObject>( |
| seq_num_start, |
| seq_num_end, |
| /*markerBit=*/true, |
| /*times_nacked=*/0, |
| /*first_packet_received_time=*/0, |
| /*last_packet_received_time=*/0, |
| /*rtp_timestamp=*/0, |
| /*ntp_time_ms=*/0, |
| VideoSendTiming(), |
| /*payload_type=*/0, |
| codec, |
| kVideoRotation_0, |
| VideoContentType::UNSPECIFIED, |
| video_header, |
| /*color_space=*/absl::nullopt, |
| RtpPacketInfos(), |
| EncodedImageBuffer::Create(/*size=*/0)); |
| // clang-format on |
| } |
| } // namespace |
| |
| class RtpVp9RefFinderTest : public ::testing::Test { |
| protected: |
| RtpVp9RefFinderTest() |
| : rand_(0x8739211), |
| reference_finder_(std::make_unique<RtpVp9RefFinder>()), |
| frames_from_callback_(FrameComp()) {} |
| |
| uint16_t Rand() { return rand_.Rand<uint16_t>(); } |
| |
| void OnCompleteFrame(std::unique_ptr<EncodedFrame> frame) { |
| int64_t pid = frame->id.picture_id; |
| uint16_t sidx = frame->id.spatial_layer; |
| auto frame_it = frames_from_callback_.find(std::make_pair(pid, sidx)); |
| if (frame_it != frames_from_callback_.end()) { |
| ADD_FAILURE() << "Already received frame with (pid:sidx): (" << pid << ":" |
| << sidx << ")"; |
| return; |
| } |
| |
| frames_from_callback_.insert( |
| std::make_pair(std::make_pair(pid, sidx), std::move(frame))); |
| } |
| |
| void InsertVp9Gof(uint16_t seq_num_start, |
| uint16_t seq_num_end, |
| bool keyframe, |
| int32_t pid = kNoPictureId, |
| uint8_t sid = kNoSpatialIdx, |
| uint8_t tid = kNoTemporalIdx, |
| int32_t tl0 = kNoTl0PicIdx, |
| bool up_switch = false, |
| bool inter_pic_predicted = true, |
| GofInfoVP9* ss = nullptr) { |
| RTPVideoHeaderVP9 vp9_header{}; |
| vp9_header.flexible_mode = false; |
| vp9_header.picture_id = pid % (1 << 15); |
| vp9_header.temporal_idx = tid; |
| vp9_header.spatial_idx = sid; |
| vp9_header.tl0_pic_idx = tl0; |
| vp9_header.temporal_up_switch = up_switch; |
| vp9_header.inter_pic_predicted = inter_pic_predicted && !keyframe; |
| if (ss != nullptr) { |
| vp9_header.ss_data_available = true; |
| vp9_header.gof = *ss; |
| } |
| |
| std::unique_ptr<RtpFrameObject> frame = CreateFrame( |
| seq_num_start, seq_num_end, keyframe, kVideoCodecVP9, vp9_header); |
| |
| for (auto& f : reference_finder_->ManageFrame(std::move(frame))) { |
| OnCompleteFrame(std::move(f)); |
| } |
| } |
| |
| void InsertVp9Flex(uint16_t seq_num_start, |
| uint16_t seq_num_end, |
| bool keyframe, |
| int32_t pid = kNoPictureId, |
| uint8_t sid = kNoSpatialIdx, |
| uint8_t tid = kNoTemporalIdx, |
| bool inter = false, |
| std::vector<uint8_t> refs = std::vector<uint8_t>()) { |
| RTPVideoHeaderVP9 vp9_header{}; |
| vp9_header.inter_layer_predicted = inter; |
| vp9_header.flexible_mode = true; |
| vp9_header.picture_id = pid % (1 << 15); |
| vp9_header.temporal_idx = tid; |
| vp9_header.spatial_idx = sid; |
| vp9_header.tl0_pic_idx = kNoTl0PicIdx; |
| vp9_header.num_ref_pics = refs.size(); |
| for (size_t i = 0; i < refs.size(); ++i) |
| vp9_header.pid_diff[i] = refs[i]; |
| |
| std::unique_ptr<RtpFrameObject> frame = CreateFrame( |
| seq_num_start, seq_num_end, keyframe, kVideoCodecVP9, vp9_header); |
| for (auto& f : reference_finder_->ManageFrame(std::move(frame))) { |
| OnCompleteFrame(std::move(f)); |
| } |
| } |
| |
| // Check if a frame with picture id |pid| and spatial index |sidx| has been |
| // delivered from the packet buffer, and if so, if it has the references |
| // specified by |refs|. |
| template <typename... T> |
| void CheckReferences(int64_t picture_id_offset, |
| uint16_t sidx, |
| T... refs) const { |
| int64_t pid = picture_id_offset; |
| auto frame_it = frames_from_callback_.find(std::make_pair(pid, sidx)); |
| if (frame_it == frames_from_callback_.end()) { |
| ADD_FAILURE() << "Could not find frame with (pid:sidx): (" << pid << ":" |
| << sidx << ")"; |
| return; |
| } |
| |
| std::set<int64_t> actual_refs; |
| for (uint8_t r = 0; r < frame_it->second->num_references; ++r) |
| actual_refs.insert(frame_it->second->references[r]); |
| |
| std::set<int64_t> expected_refs; |
| RefsToSet(&expected_refs, refs...); |
| |
| ASSERT_EQ(expected_refs, actual_refs); |
| } |
| |
| template <typename... T> |
| void CheckReferencesVp9(int64_t pid, uint8_t sidx, T... refs) const { |
| CheckReferences(pid, sidx, refs...); |
| } |
| |
| template <typename... T> |
| void RefsToSet(std::set<int64_t>* m, int64_t ref, T... refs) const { |
| m->insert(ref); |
| RefsToSet(m, refs...); |
| } |
| |
| void RefsToSet(std::set<int64_t>* m) const {} |
| |
| Random rand_; |
| std::unique_ptr<RtpVp9RefFinder> reference_finder_; |
| struct FrameComp { |
| bool operator()(const std::pair<int64_t, uint8_t> f1, |
| const std::pair<int64_t, uint8_t> f2) const { |
| if (f1.first == f2.first) |
| return f1.second < f2.second; |
| return f1.first < f2.first; |
| } |
| }; |
| std:: |
| map<std::pair<int64_t, uint8_t>, std::unique_ptr<EncodedFrame>, FrameComp> |
| frames_from_callback_; |
| }; |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofInsertOneFrame) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.SetGofInfoVP9(kTemporalStructureMode1); |
| |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 0, false, false, &ss); |
| |
| CheckReferencesVp9(pid, 0); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofTemporalLayers_0) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.SetGofInfoVP9(kTemporalStructureMode1); // Only 1 spatial layer. |
| |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 0, false, false, &ss); |
| InsertVp9Gof(sn + 1, sn + 1, false, pid + 1, 0, 0, 1, false); |
| InsertVp9Gof(sn + 2, sn + 2, false, pid + 2, 0, 0, 2, false); |
| InsertVp9Gof(sn + 3, sn + 3, false, pid + 3, 0, 0, 3, false); |
| InsertVp9Gof(sn + 4, sn + 4, false, pid + 4, 0, 0, 4, false); |
| InsertVp9Gof(sn + 5, sn + 5, false, pid + 5, 0, 0, 5, false); |
| InsertVp9Gof(sn + 6, sn + 6, false, pid + 6, 0, 0, 6, false); |
| InsertVp9Gof(sn + 7, sn + 7, false, pid + 7, 0, 0, 7, false); |
| InsertVp9Gof(sn + 8, sn + 8, false, pid + 8, 0, 0, 8, false); |
| InsertVp9Gof(sn + 9, sn + 9, false, pid + 9, 0, 0, 9, false); |
| InsertVp9Gof(sn + 10, sn + 10, false, pid + 10, 0, 0, 10, false); |
| InsertVp9Gof(sn + 11, sn + 11, false, pid + 11, 0, 0, 11, false); |
| InsertVp9Gof(sn + 12, sn + 12, false, pid + 12, 0, 0, 12, false); |
| InsertVp9Gof(sn + 13, sn + 13, false, pid + 13, 0, 0, 13, false); |
| InsertVp9Gof(sn + 14, sn + 14, false, pid + 14, 0, 0, 14, false); |
| InsertVp9Gof(sn + 15, sn + 15, false, pid + 15, 0, 0, 15, false); |
| InsertVp9Gof(sn + 16, sn + 16, false, pid + 16, 0, 0, 16, false); |
| InsertVp9Gof(sn + 17, sn + 17, false, pid + 17, 0, 0, 17, false); |
| InsertVp9Gof(sn + 18, sn + 18, false, pid + 18, 0, 0, 18, false); |
| InsertVp9Gof(sn + 19, sn + 19, false, pid + 19, 0, 0, 19, false); |
| |
| ASSERT_EQ(20UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| CheckReferencesVp9(pid + 1, 0, pid); |
| CheckReferencesVp9(pid + 2, 0, pid + 1); |
| CheckReferencesVp9(pid + 3, 0, pid + 2); |
| CheckReferencesVp9(pid + 4, 0, pid + 3); |
| CheckReferencesVp9(pid + 5, 0, pid + 4); |
| CheckReferencesVp9(pid + 6, 0, pid + 5); |
| CheckReferencesVp9(pid + 7, 0, pid + 6); |
| CheckReferencesVp9(pid + 8, 0, pid + 7); |
| CheckReferencesVp9(pid + 9, 0, pid + 8); |
| CheckReferencesVp9(pid + 10, 0, pid + 9); |
| CheckReferencesVp9(pid + 11, 0, pid + 10); |
| CheckReferencesVp9(pid + 12, 0, pid + 11); |
| CheckReferencesVp9(pid + 13, 0, pid + 12); |
| CheckReferencesVp9(pid + 14, 0, pid + 13); |
| CheckReferencesVp9(pid + 15, 0, pid + 14); |
| CheckReferencesVp9(pid + 16, 0, pid + 15); |
| CheckReferencesVp9(pid + 17, 0, pid + 16); |
| CheckReferencesVp9(pid + 18, 0, pid + 17); |
| CheckReferencesVp9(pid + 19, 0, pid + 18); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofSpatialLayers_2) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.SetGofInfoVP9(kTemporalStructureMode1); // Only 1 spatial layer. |
| |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 0, false, false, &ss); |
| InsertVp9Gof(sn + 1, sn + 1, false, pid + 1, 0, 0, 1, false, true); |
| // Not inter_pic_predicted because it's the first frame with this layer. |
| InsertVp9Gof(sn + 2, sn + 2, false, pid + 1, 1, 0, 1, false, false); |
| InsertVp9Gof(sn + 3, sn + 3, false, pid + 2, 0, 0, 1, false, true); |
| InsertVp9Gof(sn + 4, sn + 4, false, pid + 2, 1, 0, 1, false, true); |
| |
| ASSERT_EQ(5UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| CheckReferencesVp9(pid + 1, 0, pid); |
| CheckReferencesVp9(pid + 1, 1); |
| CheckReferencesVp9(pid + 2, 0, pid + 1); |
| CheckReferencesVp9(pid + 2, 1, pid + 1); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofTemporalLayersReordered_0) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.SetGofInfoVP9(kTemporalStructureMode1); // Only 1 spatial layer. |
| |
| InsertVp9Gof(sn + 2, sn + 2, false, pid + 2, 0, 0, 2, false); |
| InsertVp9Gof(sn + 1, sn + 1, false, pid + 1, 0, 0, 1, false); |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 0, false, false, &ss); |
| InsertVp9Gof(sn + 4, sn + 4, false, pid + 4, 0, 0, 4, false); |
| InsertVp9Gof(sn + 3, sn + 3, false, pid + 3, 0, 0, 3, false); |
| InsertVp9Gof(sn + 5, sn + 5, false, pid + 5, 0, 0, 5, false); |
| InsertVp9Gof(sn + 7, sn + 7, false, pid + 7, 0, 0, 7, false); |
| InsertVp9Gof(sn + 6, sn + 6, false, pid + 6, 0, 0, 6, false); |
| InsertVp9Gof(sn + 8, sn + 8, false, pid + 8, 0, 0, 8, false); |
| InsertVp9Gof(sn + 10, sn + 10, false, pid + 10, 0, 0, 10, false); |
| InsertVp9Gof(sn + 13, sn + 13, false, pid + 13, 0, 0, 13, false); |
| InsertVp9Gof(sn + 11, sn + 11, false, pid + 11, 0, 0, 11, false); |
| InsertVp9Gof(sn + 9, sn + 9, false, pid + 9, 0, 0, 9, false); |
| InsertVp9Gof(sn + 16, sn + 16, false, pid + 16, 0, 0, 16, false); |
| InsertVp9Gof(sn + 14, sn + 14, false, pid + 14, 0, 0, 14, false); |
| InsertVp9Gof(sn + 15, sn + 15, false, pid + 15, 0, 0, 15, false); |
| InsertVp9Gof(sn + 12, sn + 12, false, pid + 12, 0, 0, 12, false); |
| InsertVp9Gof(sn + 17, sn + 17, false, pid + 17, 0, 0, 17, false); |
| InsertVp9Gof(sn + 19, sn + 19, false, pid + 19, 0, 0, 19, false); |
| InsertVp9Gof(sn + 18, sn + 18, false, pid + 18, 0, 0, 18, false); |
| |
| ASSERT_EQ(20UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| CheckReferencesVp9(pid + 1, 0, pid); |
| CheckReferencesVp9(pid + 2, 0, pid + 1); |
| CheckReferencesVp9(pid + 3, 0, pid + 2); |
| CheckReferencesVp9(pid + 4, 0, pid + 3); |
| CheckReferencesVp9(pid + 5, 0, pid + 4); |
| CheckReferencesVp9(pid + 6, 0, pid + 5); |
| CheckReferencesVp9(pid + 7, 0, pid + 6); |
| CheckReferencesVp9(pid + 8, 0, pid + 7); |
| CheckReferencesVp9(pid + 9, 0, pid + 8); |
| CheckReferencesVp9(pid + 10, 0, pid + 9); |
| CheckReferencesVp9(pid + 11, 0, pid + 10); |
| CheckReferencesVp9(pid + 12, 0, pid + 11); |
| CheckReferencesVp9(pid + 13, 0, pid + 12); |
| CheckReferencesVp9(pid + 14, 0, pid + 13); |
| CheckReferencesVp9(pid + 15, 0, pid + 14); |
| CheckReferencesVp9(pid + 16, 0, pid + 15); |
| CheckReferencesVp9(pid + 17, 0, pid + 16); |
| CheckReferencesVp9(pid + 18, 0, pid + 17); |
| CheckReferencesVp9(pid + 19, 0, pid + 18); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofSkipFramesTemporalLayers_01) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.SetGofInfoVP9(kTemporalStructureMode2); // 0101 pattern |
| |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 0, false, false, &ss); |
| InsertVp9Gof(sn + 1, sn + 1, false, pid + 1, 0, 1, 0, false); |
| // Skip GOF with tl0 1 |
| InsertVp9Gof(sn + 4, sn + 4, true, pid + 4, 0, 0, 2, false, true, &ss); |
| InsertVp9Gof(sn + 5, sn + 5, false, pid + 5, 0, 1, 2, false); |
| // Skip GOF with tl0 3 |
| // Skip GOF with tl0 4 |
| InsertVp9Gof(sn + 10, sn + 10, false, pid + 10, 0, 0, 5, false, true, &ss); |
| InsertVp9Gof(sn + 11, sn + 11, false, pid + 11, 0, 1, 5, false); |
| |
| ASSERT_EQ(6UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| CheckReferencesVp9(pid + 1, 0, pid); |
| CheckReferencesVp9(pid + 4, 0); |
| CheckReferencesVp9(pid + 5, 0, pid + 4); |
| CheckReferencesVp9(pid + 10, 0, pid + 8); |
| CheckReferencesVp9(pid + 11, 0, pid + 10); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofSkipFramesTemporalLayers_0212) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.SetGofInfoVP9(kTemporalStructureMode3); // 02120212 pattern |
| |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 0, false, false, &ss); |
| InsertVp9Gof(sn + 1, sn + 1, false, pid + 1, 0, 2, 0, false); |
| InsertVp9Gof(sn + 2, sn + 2, false, pid + 2, 0, 1, 0, false); |
| InsertVp9Gof(sn + 3, sn + 3, false, pid + 3, 0, 2, 0, false); |
| |
| ASSERT_EQ(4UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| CheckReferencesVp9(pid + 1, 0, pid); |
| CheckReferencesVp9(pid + 2, 0, pid); |
| CheckReferencesVp9(pid + 3, 0, pid + 2); |
| |
| // Skip frames with tl0 = 1 |
| |
| InsertVp9Gof(sn + 8, sn + 8, true, pid + 8, 0, 0, 2, false, false, &ss); |
| InsertVp9Gof(sn + 9, sn + 9, false, pid + 9, 0, 2, 2, false); |
| InsertVp9Gof(sn + 10, sn + 10, false, pid + 10, 0, 1, 2, false); |
| InsertVp9Gof(sn + 11, sn + 11, false, pid + 11, 0, 2, 2, false); |
| |
| ASSERT_EQ(8UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid + 8, 0); |
| CheckReferencesVp9(pid + 9, 0, pid + 8); |
| CheckReferencesVp9(pid + 10, 0, pid + 8); |
| CheckReferencesVp9(pid + 11, 0, pid + 10); |
| |
| // Now insert frames with tl0 = 1 |
| InsertVp9Gof(sn + 4, sn + 4, true, pid + 4, 0, 0, 1, false, true, &ss); |
| InsertVp9Gof(sn + 7, sn + 7, false, pid + 7, 0, 2, 1, false); |
| |
| ASSERT_EQ(9UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid + 4, 0); |
| |
| // Rest of frames belonging to tl0 = 1 |
| InsertVp9Gof(sn + 5, sn + 5, false, pid + 5, 0, 2, 1, false); |
| InsertVp9Gof(sn + 6, sn + 6, false, pid + 6, 0, 1, 1, true); // up-switch |
| |
| ASSERT_EQ(12UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid + 5, 0, pid + 4); |
| CheckReferencesVp9(pid + 6, 0, pid + 4); |
| CheckReferencesVp9(pid + 7, 0, pid + 6); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofTemporalLayers_01) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.SetGofInfoVP9(kTemporalStructureMode2); // 0101 pattern |
| |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 0, false, false, &ss); |
| InsertVp9Gof(sn + 1, sn + 1, false, pid + 1, 0, 1, 0, false); |
| InsertVp9Gof(sn + 2, sn + 2, false, pid + 2, 0, 0, 1, false); |
| InsertVp9Gof(sn + 3, sn + 3, false, pid + 3, 0, 1, 1, false); |
| InsertVp9Gof(sn + 4, sn + 4, false, pid + 4, 0, 0, 2, false); |
| InsertVp9Gof(sn + 5, sn + 5, false, pid + 5, 0, 1, 2, false); |
| InsertVp9Gof(sn + 6, sn + 6, false, pid + 6, 0, 0, 3, false); |
| InsertVp9Gof(sn + 7, sn + 7, false, pid + 7, 0, 1, 3, false); |
| InsertVp9Gof(sn + 8, sn + 8, false, pid + 8, 0, 0, 4, false); |
| InsertVp9Gof(sn + 9, sn + 9, false, pid + 9, 0, 1, 4, false); |
| InsertVp9Gof(sn + 10, sn + 10, false, pid + 10, 0, 0, 5, false); |
| InsertVp9Gof(sn + 11, sn + 11, false, pid + 11, 0, 1, 5, false); |
| InsertVp9Gof(sn + 12, sn + 12, false, pid + 12, 0, 0, 6, false); |
| InsertVp9Gof(sn + 13, sn + 13, false, pid + 13, 0, 1, 6, false); |
| InsertVp9Gof(sn + 14, sn + 14, false, pid + 14, 0, 0, 7, false); |
| InsertVp9Gof(sn + 15, sn + 15, false, pid + 15, 0, 1, 7, false); |
| InsertVp9Gof(sn + 16, sn + 16, false, pid + 16, 0, 0, 8, false); |
| InsertVp9Gof(sn + 17, sn + 17, false, pid + 17, 0, 1, 8, false); |
| InsertVp9Gof(sn + 18, sn + 18, false, pid + 18, 0, 0, 9, false); |
| InsertVp9Gof(sn + 19, sn + 19, false, pid + 19, 0, 1, 9, false); |
| |
| ASSERT_EQ(20UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| CheckReferencesVp9(pid + 1, 0, pid); |
| CheckReferencesVp9(pid + 2, 0, pid); |
| CheckReferencesVp9(pid + 3, 0, pid + 2); |
| CheckReferencesVp9(pid + 4, 0, pid + 2); |
| CheckReferencesVp9(pid + 5, 0, pid + 4); |
| CheckReferencesVp9(pid + 6, 0, pid + 4); |
| CheckReferencesVp9(pid + 7, 0, pid + 6); |
| CheckReferencesVp9(pid + 8, 0, pid + 6); |
| CheckReferencesVp9(pid + 9, 0, pid + 8); |
| CheckReferencesVp9(pid + 10, 0, pid + 8); |
| CheckReferencesVp9(pid + 11, 0, pid + 10); |
| CheckReferencesVp9(pid + 12, 0, pid + 10); |
| CheckReferencesVp9(pid + 13, 0, pid + 12); |
| CheckReferencesVp9(pid + 14, 0, pid + 12); |
| CheckReferencesVp9(pid + 15, 0, pid + 14); |
| CheckReferencesVp9(pid + 16, 0, pid + 14); |
| CheckReferencesVp9(pid + 17, 0, pid + 16); |
| CheckReferencesVp9(pid + 18, 0, pid + 16); |
| CheckReferencesVp9(pid + 19, 0, pid + 18); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofTemporalLayersReordered_01) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.SetGofInfoVP9(kTemporalStructureMode2); // 01 pattern |
| |
| InsertVp9Gof(sn + 1, sn + 1, false, pid + 1, 0, 1, 0, false); |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 0, false, false, &ss); |
| InsertVp9Gof(sn + 2, sn + 2, false, pid + 2, 0, 0, 1, false); |
| InsertVp9Gof(sn + 4, sn + 4, false, pid + 4, 0, 0, 2, false); |
| InsertVp9Gof(sn + 3, sn + 3, false, pid + 3, 0, 1, 1, false); |
| InsertVp9Gof(sn + 5, sn + 5, false, pid + 5, 0, 1, 2, false); |
| InsertVp9Gof(sn + 7, sn + 7, false, pid + 7, 0, 1, 3, false); |
| InsertVp9Gof(sn + 6, sn + 6, false, pid + 6, 0, 0, 3, false); |
| InsertVp9Gof(sn + 10, sn + 10, false, pid + 10, 0, 0, 5, false); |
| InsertVp9Gof(sn + 8, sn + 8, false, pid + 8, 0, 0, 4, false); |
| InsertVp9Gof(sn + 9, sn + 9, false, pid + 9, 0, 1, 4, false); |
| InsertVp9Gof(sn + 11, sn + 11, false, pid + 11, 0, 1, 5, false); |
| InsertVp9Gof(sn + 13, sn + 13, false, pid + 13, 0, 1, 6, false); |
| InsertVp9Gof(sn + 16, sn + 16, false, pid + 16, 0, 0, 8, false); |
| InsertVp9Gof(sn + 12, sn + 12, false, pid + 12, 0, 0, 6, false); |
| InsertVp9Gof(sn + 14, sn + 14, false, pid + 14, 0, 0, 7, false); |
| InsertVp9Gof(sn + 17, sn + 17, false, pid + 17, 0, 1, 8, false); |
| InsertVp9Gof(sn + 19, sn + 19, false, pid + 19, 0, 1, 9, false); |
| InsertVp9Gof(sn + 15, sn + 15, false, pid + 15, 0, 1, 7, false); |
| InsertVp9Gof(sn + 18, sn + 18, false, pid + 18, 0, 0, 9, false); |
| |
| ASSERT_EQ(20UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| CheckReferencesVp9(pid + 1, 0, pid); |
| CheckReferencesVp9(pid + 2, 0, pid); |
| CheckReferencesVp9(pid + 3, 0, pid + 2); |
| CheckReferencesVp9(pid + 4, 0, pid + 2); |
| CheckReferencesVp9(pid + 5, 0, pid + 4); |
| CheckReferencesVp9(pid + 6, 0, pid + 4); |
| CheckReferencesVp9(pid + 7, 0, pid + 6); |
| CheckReferencesVp9(pid + 8, 0, pid + 6); |
| CheckReferencesVp9(pid + 9, 0, pid + 8); |
| CheckReferencesVp9(pid + 10, 0, pid + 8); |
| CheckReferencesVp9(pid + 11, 0, pid + 10); |
| CheckReferencesVp9(pid + 12, 0, pid + 10); |
| CheckReferencesVp9(pid + 13, 0, pid + 12); |
| CheckReferencesVp9(pid + 14, 0, pid + 12); |
| CheckReferencesVp9(pid + 15, 0, pid + 14); |
| CheckReferencesVp9(pid + 16, 0, pid + 14); |
| CheckReferencesVp9(pid + 17, 0, pid + 16); |
| CheckReferencesVp9(pid + 18, 0, pid + 16); |
| CheckReferencesVp9(pid + 19, 0, pid + 18); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofTemporalLayers_0212) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.SetGofInfoVP9(kTemporalStructureMode3); // 0212 pattern |
| |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 0, false, false, &ss); |
| InsertVp9Gof(sn + 1, sn + 1, false, pid + 1, 0, 2, 0, false); |
| InsertVp9Gof(sn + 2, sn + 2, false, pid + 2, 0, 1, 0, false); |
| InsertVp9Gof(sn + 3, sn + 3, false, pid + 3, 0, 2, 0, false); |
| InsertVp9Gof(sn + 4, sn + 4, false, pid + 4, 0, 0, 1, false); |
| InsertVp9Gof(sn + 5, sn + 5, false, pid + 5, 0, 2, 1, false); |
| InsertVp9Gof(sn + 6, sn + 6, false, pid + 6, 0, 1, 1, false); |
| InsertVp9Gof(sn + 7, sn + 7, false, pid + 7, 0, 2, 1, false); |
| InsertVp9Gof(sn + 8, sn + 8, false, pid + 8, 0, 0, 2, false); |
| InsertVp9Gof(sn + 9, sn + 9, false, pid + 9, 0, 2, 2, false); |
| InsertVp9Gof(sn + 10, sn + 10, false, pid + 10, 0, 1, 2, false); |
| InsertVp9Gof(sn + 11, sn + 11, false, pid + 11, 0, 2, 2, false); |
| InsertVp9Gof(sn + 12, sn + 12, false, pid + 12, 0, 0, 3, false); |
| InsertVp9Gof(sn + 13, sn + 13, false, pid + 13, 0, 2, 3, false); |
| InsertVp9Gof(sn + 14, sn + 14, false, pid + 14, 0, 1, 3, false); |
| InsertVp9Gof(sn + 15, sn + 15, false, pid + 15, 0, 2, 3, false); |
| InsertVp9Gof(sn + 16, sn + 16, false, pid + 16, 0, 0, 4, false); |
| InsertVp9Gof(sn + 17, sn + 17, false, pid + 17, 0, 2, 4, false); |
| InsertVp9Gof(sn + 18, sn + 18, false, pid + 18, 0, 1, 4, false); |
| InsertVp9Gof(sn + 19, sn + 19, false, pid + 19, 0, 2, 4, false); |
| |
| ASSERT_EQ(20UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| CheckReferencesVp9(pid + 1, 0, pid); |
| CheckReferencesVp9(pid + 2, 0, pid); |
| CheckReferencesVp9(pid + 3, 0, pid + 2); |
| CheckReferencesVp9(pid + 4, 0, pid); |
| CheckReferencesVp9(pid + 5, 0, pid + 4); |
| CheckReferencesVp9(pid + 6, 0, pid + 4); |
| CheckReferencesVp9(pid + 7, 0, pid + 6); |
| CheckReferencesVp9(pid + 8, 0, pid + 4); |
| CheckReferencesVp9(pid + 9, 0, pid + 8); |
| CheckReferencesVp9(pid + 10, 0, pid + 8); |
| CheckReferencesVp9(pid + 11, 0, pid + 10); |
| CheckReferencesVp9(pid + 12, 0, pid + 8); |
| CheckReferencesVp9(pid + 13, 0, pid + 12); |
| CheckReferencesVp9(pid + 14, 0, pid + 12); |
| CheckReferencesVp9(pid + 15, 0, pid + 14); |
| CheckReferencesVp9(pid + 16, 0, pid + 12); |
| CheckReferencesVp9(pid + 17, 0, pid + 16); |
| CheckReferencesVp9(pid + 18, 0, pid + 16); |
| CheckReferencesVp9(pid + 19, 0, pid + 18); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofTemporalLayersReordered_0212) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.SetGofInfoVP9(kTemporalStructureMode3); // 0212 pattern |
| |
| InsertVp9Gof(sn + 2, sn + 2, false, pid + 2, 0, 1, 0, false); |
| InsertVp9Gof(sn + 1, sn + 1, false, pid + 1, 0, 2, 0, false); |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 0, false, false, &ss); |
| InsertVp9Gof(sn + 3, sn + 3, false, pid + 3, 0, 2, 0, false); |
| InsertVp9Gof(sn + 6, sn + 6, false, pid + 6, 0, 1, 1, false); |
| InsertVp9Gof(sn + 5, sn + 5, false, pid + 5, 0, 2, 1, false); |
| InsertVp9Gof(sn + 4, sn + 4, false, pid + 4, 0, 0, 1, false); |
| InsertVp9Gof(sn + 9, sn + 9, false, pid + 9, 0, 2, 2, false); |
| InsertVp9Gof(sn + 7, sn + 7, false, pid + 7, 0, 2, 1, false); |
| InsertVp9Gof(sn + 8, sn + 8, false, pid + 8, 0, 0, 2, false); |
| InsertVp9Gof(sn + 11, sn + 11, false, pid + 11, 0, 2, 2, false); |
| InsertVp9Gof(sn + 10, sn + 10, false, pid + 10, 0, 1, 2, false); |
| InsertVp9Gof(sn + 13, sn + 13, false, pid + 13, 0, 2, 3, false); |
| InsertVp9Gof(sn + 12, sn + 12, false, pid + 12, 0, 0, 3, false); |
| InsertVp9Gof(sn + 14, sn + 14, false, pid + 14, 0, 1, 3, false); |
| InsertVp9Gof(sn + 16, sn + 16, false, pid + 16, 0, 0, 4, false); |
| InsertVp9Gof(sn + 15, sn + 15, false, pid + 15, 0, 2, 3, false); |
| InsertVp9Gof(sn + 17, sn + 17, false, pid + 17, 0, 2, 4, false); |
| InsertVp9Gof(sn + 19, sn + 19, false, pid + 19, 0, 2, 4, false); |
| InsertVp9Gof(sn + 18, sn + 18, false, pid + 18, 0, 1, 4, false); |
| |
| ASSERT_EQ(20UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| CheckReferencesVp9(pid + 1, 0, pid); |
| CheckReferencesVp9(pid + 2, 0, pid); |
| CheckReferencesVp9(pid + 3, 0, pid + 2); |
| CheckReferencesVp9(pid + 4, 0, pid); |
| CheckReferencesVp9(pid + 5, 0, pid + 4); |
| CheckReferencesVp9(pid + 6, 0, pid + 4); |
| CheckReferencesVp9(pid + 7, 0, pid + 6); |
| CheckReferencesVp9(pid + 8, 0, pid + 4); |
| CheckReferencesVp9(pid + 9, 0, pid + 8); |
| CheckReferencesVp9(pid + 10, 0, pid + 8); |
| CheckReferencesVp9(pid + 11, 0, pid + 10); |
| CheckReferencesVp9(pid + 12, 0, pid + 8); |
| CheckReferencesVp9(pid + 13, 0, pid + 12); |
| CheckReferencesVp9(pid + 14, 0, pid + 12); |
| CheckReferencesVp9(pid + 15, 0, pid + 14); |
| CheckReferencesVp9(pid + 16, 0, pid + 12); |
| CheckReferencesVp9(pid + 17, 0, pid + 16); |
| CheckReferencesVp9(pid + 18, 0, pid + 16); |
| CheckReferencesVp9(pid + 19, 0, pid + 18); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofTemporalLayersUpSwitch_02120212) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.SetGofInfoVP9(kTemporalStructureMode4); // 02120212 pattern |
| |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 0, false, false, &ss); |
| InsertVp9Gof(sn + 1, sn + 1, false, pid + 1, 0, 2, 0, false); |
| InsertVp9Gof(sn + 2, sn + 2, false, pid + 2, 0, 1, 0, false); |
| InsertVp9Gof(sn + 3, sn + 3, false, pid + 3, 0, 2, 0, false); |
| InsertVp9Gof(sn + 4, sn + 4, false, pid + 4, 0, 0, 1, false); |
| InsertVp9Gof(sn + 5, sn + 5, false, pid + 5, 0, 2, 1, false); |
| InsertVp9Gof(sn + 6, sn + 6, false, pid + 6, 0, 1, 1, true); |
| InsertVp9Gof(sn + 7, sn + 7, false, pid + 7, 0, 2, 1, false); |
| InsertVp9Gof(sn + 8, sn + 8, false, pid + 8, 0, 0, 2, true); |
| InsertVp9Gof(sn + 9, sn + 9, false, pid + 9, 0, 2, 2, false); |
| InsertVp9Gof(sn + 10, sn + 10, false, pid + 10, 0, 1, 2, false); |
| InsertVp9Gof(sn + 11, sn + 11, false, pid + 11, 0, 2, 2, true); |
| InsertVp9Gof(sn + 12, sn + 12, false, pid + 12, 0, 0, 3, false); |
| InsertVp9Gof(sn + 13, sn + 13, false, pid + 13, 0, 2, 3, false); |
| InsertVp9Gof(sn + 14, sn + 14, false, pid + 14, 0, 1, 3, false); |
| InsertVp9Gof(sn + 15, sn + 15, false, pid + 15, 0, 2, 3, false); |
| |
| ASSERT_EQ(16UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| CheckReferencesVp9(pid + 1, 0, pid); |
| CheckReferencesVp9(pid + 2, 0, pid); |
| CheckReferencesVp9(pid + 3, 0, pid + 1, pid + 2); |
| CheckReferencesVp9(pid + 4, 0, pid); |
| CheckReferencesVp9(pid + 5, 0, pid + 3, pid + 4); |
| CheckReferencesVp9(pid + 6, 0, pid + 2, pid + 4); |
| CheckReferencesVp9(pid + 7, 0, pid + 6); |
| CheckReferencesVp9(pid + 8, 0, pid + 4); |
| CheckReferencesVp9(pid + 9, 0, pid + 8); |
| CheckReferencesVp9(pid + 10, 0, pid + 8); |
| CheckReferencesVp9(pid + 11, 0, pid + 9, pid + 10); |
| CheckReferencesVp9(pid + 12, 0, pid + 8); |
| CheckReferencesVp9(pid + 13, 0, pid + 11, pid + 12); |
| CheckReferencesVp9(pid + 14, 0, pid + 10, pid + 12); |
| CheckReferencesVp9(pid + 15, 0, pid + 13, pid + 14); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofTemporalLayersUpSwitchReordered_02120212) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.SetGofInfoVP9(kTemporalStructureMode4); // 02120212 pattern |
| |
| InsertVp9Gof(sn + 1, sn + 1, false, pid + 1, 0, 2, 0, false); |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 0, false, false, &ss); |
| InsertVp9Gof(sn + 4, sn + 4, false, pid + 4, 0, 0, 1, false); |
| InsertVp9Gof(sn + 2, sn + 2, false, pid + 2, 0, 1, 0, false); |
| InsertVp9Gof(sn + 5, sn + 5, false, pid + 5, 0, 2, 1, false); |
| InsertVp9Gof(sn + 3, sn + 3, false, pid + 3, 0, 2, 0, false); |
| InsertVp9Gof(sn + 7, sn + 7, false, pid + 7, 0, 2, 1, false); |
| InsertVp9Gof(sn + 9, sn + 9, false, pid + 9, 0, 2, 2, false); |
| InsertVp9Gof(sn + 6, sn + 6, false, pid + 6, 0, 1, 1, true); |
| InsertVp9Gof(sn + 12, sn + 12, false, pid + 12, 0, 0, 3, false); |
| InsertVp9Gof(sn + 10, sn + 10, false, pid + 10, 0, 1, 2, false); |
| InsertVp9Gof(sn + 8, sn + 8, false, pid + 8, 0, 0, 2, true); |
| InsertVp9Gof(sn + 11, sn + 11, false, pid + 11, 0, 2, 2, true); |
| InsertVp9Gof(sn + 13, sn + 13, false, pid + 13, 0, 2, 3, false); |
| InsertVp9Gof(sn + 15, sn + 15, false, pid + 15, 0, 2, 3, false); |
| InsertVp9Gof(sn + 14, sn + 14, false, pid + 14, 0, 1, 3, false); |
| |
| ASSERT_EQ(16UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| CheckReferencesVp9(pid + 1, 0, pid); |
| CheckReferencesVp9(pid + 2, 0, pid); |
| CheckReferencesVp9(pid + 3, 0, pid + 1, pid + 2); |
| CheckReferencesVp9(pid + 4, 0, pid); |
| CheckReferencesVp9(pid + 5, 0, pid + 3, pid + 4); |
| CheckReferencesVp9(pid + 6, 0, pid + 2, pid + 4); |
| CheckReferencesVp9(pid + 7, 0, pid + 6); |
| CheckReferencesVp9(pid + 8, 0, pid + 4); |
| CheckReferencesVp9(pid + 9, 0, pid + 8); |
| CheckReferencesVp9(pid + 10, 0, pid + 8); |
| CheckReferencesVp9(pid + 11, 0, pid + 9, pid + 10); |
| CheckReferencesVp9(pid + 12, 0, pid + 8); |
| CheckReferencesVp9(pid + 13, 0, pid + 11, pid + 12); |
| CheckReferencesVp9(pid + 14, 0, pid + 10, pid + 12); |
| CheckReferencesVp9(pid + 15, 0, pid + 13, pid + 14); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofTemporalLayersReordered_01_0212) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.SetGofInfoVP9(kTemporalStructureMode2); // 01 pattern |
| |
| InsertVp9Gof(sn + 1, sn + 1, false, pid + 1, 0, 1, 0, false); |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 0, false, false, &ss); |
| InsertVp9Gof(sn + 3, sn + 3, false, pid + 3, 0, 1, 1, false); |
| InsertVp9Gof(sn + 6, sn + 6, false, pid + 6, 0, 1, 2, false); |
| ss.SetGofInfoVP9(kTemporalStructureMode3); // 0212 pattern |
| InsertVp9Gof(sn + 4, sn + 4, false, pid + 4, 0, 0, 2, false, true, &ss); |
| InsertVp9Gof(sn + 2, sn + 2, false, pid + 2, 0, 0, 1, false); |
| InsertVp9Gof(sn + 5, sn + 5, false, pid + 5, 0, 2, 2, false); |
| InsertVp9Gof(sn + 8, sn + 8, false, pid + 8, 0, 0, 3, false); |
| InsertVp9Gof(sn + 10, sn + 10, false, pid + 10, 0, 1, 3, false); |
| InsertVp9Gof(sn + 7, sn + 7, false, pid + 7, 0, 2, 2, false); |
| InsertVp9Gof(sn + 11, sn + 11, false, pid + 11, 0, 2, 3, false); |
| InsertVp9Gof(sn + 9, sn + 9, false, pid + 9, 0, 2, 3, false); |
| |
| ASSERT_EQ(12UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| CheckReferencesVp9(pid + 1, 0, pid); |
| CheckReferencesVp9(pid + 2, 0, pid); |
| CheckReferencesVp9(pid + 3, 0, pid + 2); |
| CheckReferencesVp9(pid + 4, 0, pid); |
| CheckReferencesVp9(pid + 5, 0, pid + 4); |
| CheckReferencesVp9(pid + 6, 0, pid + 4); |
| CheckReferencesVp9(pid + 7, 0, pid + 6); |
| CheckReferencesVp9(pid + 8, 0, pid + 4); |
| CheckReferencesVp9(pid + 9, 0, pid + 8); |
| CheckReferencesVp9(pid + 10, 0, pid + 8); |
| CheckReferencesVp9(pid + 11, 0, pid + 10); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9FlexibleModeOneFrame) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| |
| InsertVp9Flex(sn, sn, true, pid, 0, 0, false); |
| |
| ASSERT_EQ(1UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9FlexibleModeTwoSpatialLayers) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| |
| InsertVp9Flex(sn, sn, true, pid, 0, 0, false); |
| InsertVp9Flex(sn + 1, sn + 1, true, pid, 1, 0, true); |
| InsertVp9Flex(sn + 2, sn + 2, false, pid + 1, 1, 0, false, {1}); |
| InsertVp9Flex(sn + 3, sn + 3, false, pid + 2, 0, 0, false, {2}); |
| InsertVp9Flex(sn + 4, sn + 4, false, pid + 2, 1, 0, false, {1}); |
| InsertVp9Flex(sn + 5, sn + 5, false, pid + 3, 1, 0, false, {1}); |
| InsertVp9Flex(sn + 6, sn + 6, false, pid + 4, 0, 0, false, {2}); |
| InsertVp9Flex(sn + 7, sn + 7, false, pid + 4, 1, 0, false, {1}); |
| InsertVp9Flex(sn + 8, sn + 8, false, pid + 5, 1, 0, false, {1}); |
| InsertVp9Flex(sn + 9, sn + 9, false, pid + 6, 0, 0, false, {2}); |
| InsertVp9Flex(sn + 10, sn + 10, false, pid + 6, 1, 0, false, {1}); |
| InsertVp9Flex(sn + 11, sn + 11, false, pid + 7, 1, 0, false, {1}); |
| InsertVp9Flex(sn + 12, sn + 12, false, pid + 8, 0, 0, false, {2}); |
| InsertVp9Flex(sn + 13, sn + 13, false, pid + 8, 1, 0, false, {1}); |
| |
| ASSERT_EQ(14UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| CheckReferencesVp9(pid, 1); |
| CheckReferencesVp9(pid + 1, 1, pid); |
| CheckReferencesVp9(pid + 2, 0, pid); |
| CheckReferencesVp9(pid + 2, 1, pid + 1); |
| CheckReferencesVp9(pid + 3, 1, pid + 2); |
| CheckReferencesVp9(pid + 4, 0, pid + 2); |
| CheckReferencesVp9(pid + 4, 1, pid + 3); |
| CheckReferencesVp9(pid + 5, 1, pid + 4); |
| CheckReferencesVp9(pid + 6, 0, pid + 4); |
| CheckReferencesVp9(pid + 6, 1, pid + 5); |
| CheckReferencesVp9(pid + 7, 1, pid + 6); |
| CheckReferencesVp9(pid + 8, 0, pid + 6); |
| CheckReferencesVp9(pid + 8, 1, pid + 7); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9FlexibleModeTwoSpatialLayersReordered) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| |
| InsertVp9Flex(sn + 1, sn + 1, true, pid, 1, 0, true); |
| InsertVp9Flex(sn + 2, sn + 2, false, pid + 1, 1, 0, false, {1}); |
| InsertVp9Flex(sn, sn, true, pid, 0, 0, false); |
| InsertVp9Flex(sn + 4, sn + 4, false, pid + 2, 1, 0, false, {1}); |
| InsertVp9Flex(sn + 5, sn + 5, false, pid + 3, 1, 0, false, {1}); |
| InsertVp9Flex(sn + 3, sn + 3, false, pid + 2, 0, 0, false, {2}); |
| InsertVp9Flex(sn + 7, sn + 7, false, pid + 4, 1, 0, false, {1}); |
| InsertVp9Flex(sn + 6, sn + 6, false, pid + 4, 0, 0, false, {2}); |
| InsertVp9Flex(sn + 8, sn + 8, false, pid + 5, 1, 0, false, {1}); |
| InsertVp9Flex(sn + 9, sn + 9, false, pid + 6, 0, 0, false, {2}); |
| InsertVp9Flex(sn + 11, sn + 11, false, pid + 7, 1, 0, false, {1}); |
| InsertVp9Flex(sn + 10, sn + 10, false, pid + 6, 1, 0, false, {1}); |
| InsertVp9Flex(sn + 13, sn + 13, false, pid + 8, 1, 0, false, {1}); |
| InsertVp9Flex(sn + 12, sn + 12, false, pid + 8, 0, 0, false, {2}); |
| |
| ASSERT_EQ(14UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| CheckReferencesVp9(pid, 1); |
| CheckReferencesVp9(pid + 1, 1, pid); |
| CheckReferencesVp9(pid + 2, 0, pid); |
| CheckReferencesVp9(pid + 2, 1, pid + 1); |
| CheckReferencesVp9(pid + 3, 1, pid + 2); |
| CheckReferencesVp9(pid + 4, 0, pid + 2); |
| CheckReferencesVp9(pid + 4, 1, pid + 3); |
| CheckReferencesVp9(pid + 5, 1, pid + 4); |
| CheckReferencesVp9(pid + 6, 0, pid + 4); |
| CheckReferencesVp9(pid + 6, 1, pid + 5); |
| CheckReferencesVp9(pid + 7, 1, pid + 6); |
| CheckReferencesVp9(pid + 8, 0, pid + 6); |
| CheckReferencesVp9(pid + 8, 1, pid + 7); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, WrappingFlexReference) { |
| InsertVp9Flex(0, 0, false, 0, 0, 0, false, {1}); |
| |
| ASSERT_EQ(1UL, frames_from_callback_.size()); |
| const EncodedFrame& frame = *frames_from_callback_.begin()->second; |
| ASSERT_EQ(frame.id.picture_id - frame.references[0], 1); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofPidJump) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.SetGofInfoVP9(kTemporalStructureMode3); |
| |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 0, false, false, &ss); |
| InsertVp9Gof(sn + 1, sn + 1, false, pid + 1000, 0, 0, 1); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofTl0Jump) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.SetGofInfoVP9(kTemporalStructureMode3); |
| |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 125, true, false, &ss); |
| InsertVp9Gof(sn + 1, sn + 1, false, pid + 1, 0, 0, 0, false, true, &ss); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofTidTooHigh) { |
| const int kMaxTemporalLayers = 5; |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.SetGofInfoVP9(kTemporalStructureMode2); |
| ss.temporal_idx[1] = kMaxTemporalLayers; |
| |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 0, false, false, &ss); |
| InsertVp9Gof(sn + 1, sn + 1, false, pid + 1, 0, 0, 1); |
| |
| ASSERT_EQ(1UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| } |
| |
| TEST_F(RtpVp9RefFinderTest, Vp9GofZeroFrames) { |
| uint16_t pid = Rand(); |
| uint16_t sn = Rand(); |
| GofInfoVP9 ss; |
| ss.num_frames_in_gof = 0; |
| |
| InsertVp9Gof(sn, sn, true, pid, 0, 0, 0, false, false, &ss); |
| InsertVp9Gof(sn + 1, sn + 1, false, pid + 1, 0, 0, 1); |
| |
| ASSERT_EQ(2UL, frames_from_callback_.size()); |
| CheckReferencesVp9(pid, 0); |
| CheckReferencesVp9(pid + 1, 0, pid); |
| } |
| |
| } // namespace video_coding |
| } // namespace webrtc |
