summaryrefslogtreecommitdiffstats
path: root/third_party/libwebrtc/modules/video_coding/frame_dependencies_calculator_unittest.cc
blob: a09650401a437ee6010c981e614a1e5fbdf5cfd3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
/*
 *  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/frame_dependencies_calculator.h"

#include "common_video/generic_frame_descriptor/generic_frame_info.h"
#include "test/gmock.h"
#include "test/gtest.h"

namespace webrtc {
namespace {

using ::testing::ElementsAre;
using ::testing::IsEmpty;
using ::testing::UnorderedElementsAre;

constexpr CodecBufferUsage ReferenceAndUpdate(int id) {
  return CodecBufferUsage(id, /*referenced=*/true, /*updated=*/true);
}
constexpr CodecBufferUsage Reference(int id) {
  return CodecBufferUsage(id, /*referenced=*/true, /*updated=*/false);
}
constexpr CodecBufferUsage Update(int id) {
  return CodecBufferUsage(id, /*referenced=*/false, /*updated=*/true);
}

TEST(FrameDependenciesCalculatorTest, SingleLayer) {
  CodecBufferUsage pattern[] = {ReferenceAndUpdate(0)};
  FrameDependenciesCalculator calculator;

  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/1, pattern), IsEmpty());
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/3, pattern),
              ElementsAre(1));
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/6, pattern),
              ElementsAre(3));
}

TEST(FrameDependenciesCalculatorTest, TwoTemporalLayers) {
  // Shortened 4-frame pattern:
  // T1:  2---4   6---8 ...
  //      /   /   /   /
  // T0: 1---3---5---7 ...
  CodecBufferUsage pattern0[] = {ReferenceAndUpdate(0)};
  CodecBufferUsage pattern1[] = {Reference(0), Update(1)};
  CodecBufferUsage pattern2[] = {ReferenceAndUpdate(0)};
  CodecBufferUsage pattern3[] = {Reference(0), Reference(1)};
  FrameDependenciesCalculator calculator;

  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/1, pattern0), IsEmpty());
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/2, pattern1),
              ElementsAre(1));
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/3, pattern2),
              ElementsAre(1));
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/4, pattern3),
              UnorderedElementsAre(2, 3));
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/5, pattern0),
              ElementsAre(3));
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/6, pattern1),
              ElementsAre(5));
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/7, pattern2),
              ElementsAre(5));
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/8, pattern3),
              UnorderedElementsAre(6, 7));
}

TEST(FrameDependenciesCalculatorTest, ThreeTemporalLayers4FramePattern) {
  // T2:   2---4   6---8 ...
  //      /   /   /   /
  // T1:  |  3    |  7   ...
  //      /_/     /_/
  // T0: 1-------5-----  ...
  CodecBufferUsage pattern0[] = {ReferenceAndUpdate(0)};
  CodecBufferUsage pattern1[] = {Reference(0), Update(2)};
  CodecBufferUsage pattern2[] = {Reference(0), Update(1)};
  CodecBufferUsage pattern3[] = {Reference(0), Reference(1), Reference(2)};
  FrameDependenciesCalculator calculator;

  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/1, pattern0), IsEmpty());
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/2, pattern1),
              ElementsAre(1));
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/3, pattern2),
              ElementsAre(1));
  // Note that frame#4 references buffer#0 that is updated by frame#1,
  // yet there is no direct dependency from frame#4 to frame#1.
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/4, pattern3),
              UnorderedElementsAre(2, 3));
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/5, pattern0),
              ElementsAre(1));
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/6, pattern1),
              ElementsAre(5));
}

TEST(FrameDependenciesCalculatorTest, SimulcastWith2Layers) {
  // S1: 2---4---6-  ...
  //
  // S0: 1---3---5-  ...
  CodecBufferUsage pattern0[] = {ReferenceAndUpdate(0)};
  CodecBufferUsage pattern1[] = {ReferenceAndUpdate(1)};
  FrameDependenciesCalculator calculator;

  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/1, pattern0), IsEmpty());
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/2, pattern1), IsEmpty());
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/3, pattern0),
              ElementsAre(1));
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/4, pattern1),
              ElementsAre(2));
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/5, pattern0),
              ElementsAre(3));
  EXPECT_THAT(calculator.FromBuffersUsage(/*frame_id=*/6, pattern1),
              ElementsAre(4));
}

}  // namespace
}  // namespace webrtc