1 /*
2 * Copyright (c) 2021 The WebRTC project authors. All Rights Reserved.
3 *
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 */
10
11 #include "api/video_codecs/h264_profile_level_id.h"
12
13 #include <map>
14 #include <string>
15
16 #include "absl/types/optional.h"
17 #include "test/gtest.h"
18
19 namespace webrtc {
20
TEST(H264ProfileLevelId,TestParsingInvalid)21 TEST(H264ProfileLevelId, TestParsingInvalid) {
22 // Malformed strings.
23 EXPECT_FALSE(ParseH264ProfileLevelId(""));
24 EXPECT_FALSE(ParseH264ProfileLevelId(" 42e01f"));
25 EXPECT_FALSE(ParseH264ProfileLevelId("4242e01f"));
26 EXPECT_FALSE(ParseH264ProfileLevelId("e01f"));
27 EXPECT_FALSE(ParseH264ProfileLevelId("gggggg"));
28
29 // Invalid level.
30 EXPECT_FALSE(ParseH264ProfileLevelId("42e000"));
31 EXPECT_FALSE(ParseH264ProfileLevelId("42e00f"));
32 EXPECT_FALSE(ParseH264ProfileLevelId("42e0ff"));
33
34 // Invalid profile.
35 EXPECT_FALSE(ParseH264ProfileLevelId("42e11f"));
36 EXPECT_FALSE(ParseH264ProfileLevelId("58601f"));
37 EXPECT_FALSE(ParseH264ProfileLevelId("64e01f"));
38 }
39
TEST(H264ProfileLevelId,TestParsingLevel)40 TEST(H264ProfileLevelId, TestParsingLevel) {
41 EXPECT_EQ(H264Level::kLevel3_1, ParseH264ProfileLevelId("42e01f")->level);
42 EXPECT_EQ(H264Level::kLevel1_1, ParseH264ProfileLevelId("42e00b")->level);
43 EXPECT_EQ(H264Level::kLevel1_b, ParseH264ProfileLevelId("42f00b")->level);
44 EXPECT_EQ(H264Level::kLevel4_2, ParseH264ProfileLevelId("42C02A")->level);
45 EXPECT_EQ(H264Level::kLevel5_2, ParseH264ProfileLevelId("640c34")->level);
46 }
47
TEST(H264ProfileLevelId,TestParsingConstrainedBaseline)48 TEST(H264ProfileLevelId, TestParsingConstrainedBaseline) {
49 EXPECT_EQ(H264Profile::kProfileConstrainedBaseline,
50 ParseH264ProfileLevelId("42e01f")->profile);
51 EXPECT_EQ(H264Profile::kProfileConstrainedBaseline,
52 ParseH264ProfileLevelId("42C02A")->profile);
53 EXPECT_EQ(H264Profile::kProfileConstrainedBaseline,
54 ParseH264ProfileLevelId("4de01f")->profile);
55 EXPECT_EQ(H264Profile::kProfileConstrainedBaseline,
56 ParseH264ProfileLevelId("58f01f")->profile);
57 }
58
TEST(H264ProfileLevelId,TestParsingBaseline)59 TEST(H264ProfileLevelId, TestParsingBaseline) {
60 EXPECT_EQ(H264Profile::kProfileBaseline,
61 ParseH264ProfileLevelId("42a01f")->profile);
62 EXPECT_EQ(H264Profile::kProfileBaseline,
63 ParseH264ProfileLevelId("58A01F")->profile);
64 }
65
TEST(H264ProfileLevelId,TestParsingMain)66 TEST(H264ProfileLevelId, TestParsingMain) {
67 EXPECT_EQ(H264Profile::kProfileMain,
68 ParseH264ProfileLevelId("4D401f")->profile);
69 }
70
TEST(H264ProfileLevelId,TestParsingHigh)71 TEST(H264ProfileLevelId, TestParsingHigh) {
72 EXPECT_EQ(H264Profile::kProfileHigh,
73 ParseH264ProfileLevelId("64001f")->profile);
74 }
75
TEST(H264ProfileLevelId,TestParsingConstrainedHigh)76 TEST(H264ProfileLevelId, TestParsingConstrainedHigh) {
77 EXPECT_EQ(H264Profile::kProfileConstrainedHigh,
78 ParseH264ProfileLevelId("640c1f")->profile);
79 }
80
TEST(H264ProfileLevelId,TestSupportedLevel)81 TEST(H264ProfileLevelId, TestSupportedLevel) {
82 EXPECT_EQ(H264Level::kLevel2_1, *H264SupportedLevel(640 * 480, 25));
83 EXPECT_EQ(H264Level::kLevel3_1, *H264SupportedLevel(1280 * 720, 30));
84 EXPECT_EQ(H264Level::kLevel4_2, *H264SupportedLevel(1920 * 1280, 60));
85 }
86
87 // Test supported level below level 1 requirements.
TEST(H264ProfileLevelId,TestSupportedLevelInvalid)88 TEST(H264ProfileLevelId, TestSupportedLevelInvalid) {
89 EXPECT_FALSE(H264SupportedLevel(0, 0));
90 // All levels support fps > 5.
91 EXPECT_FALSE(H264SupportedLevel(1280 * 720, 5));
92 // All levels support frame sizes > 183 * 137.
93 EXPECT_FALSE(H264SupportedLevel(183 * 137, 30));
94 }
95
TEST(H264ProfileLevelId,TestToString)96 TEST(H264ProfileLevelId, TestToString) {
97 EXPECT_EQ("42e01f", *H264ProfileLevelIdToString(H264ProfileLevelId(
98 H264Profile::kProfileConstrainedBaseline,
99 H264Level::kLevel3_1)));
100 EXPECT_EQ("42000a", *H264ProfileLevelIdToString(H264ProfileLevelId(
101 H264Profile::kProfileBaseline, H264Level::kLevel1)));
102 EXPECT_EQ("4d001f", H264ProfileLevelIdToString(H264ProfileLevelId(
103 H264Profile::kProfileMain, H264Level::kLevel3_1)));
104 EXPECT_EQ("640c2a",
105 *H264ProfileLevelIdToString(H264ProfileLevelId(
106 H264Profile::kProfileConstrainedHigh, H264Level::kLevel4_2)));
107 EXPECT_EQ("64002a", *H264ProfileLevelIdToString(H264ProfileLevelId(
108 H264Profile::kProfileHigh, H264Level::kLevel4_2)));
109 }
110
TEST(H264ProfileLevelId,TestToStringLevel1b)111 TEST(H264ProfileLevelId, TestToStringLevel1b) {
112 EXPECT_EQ("42f00b", *H264ProfileLevelIdToString(H264ProfileLevelId(
113 H264Profile::kProfileConstrainedBaseline,
114 H264Level::kLevel1_b)));
115 EXPECT_EQ("42100b",
116 *H264ProfileLevelIdToString(H264ProfileLevelId(
117 H264Profile::kProfileBaseline, H264Level::kLevel1_b)));
118 EXPECT_EQ("4d100b", *H264ProfileLevelIdToString(H264ProfileLevelId(
119 H264Profile::kProfileMain, H264Level::kLevel1_b)));
120 }
121
TEST(H264ProfileLevelId,TestToStringRoundTrip)122 TEST(H264ProfileLevelId, TestToStringRoundTrip) {
123 EXPECT_EQ("42e01f",
124 *H264ProfileLevelIdToString(*ParseH264ProfileLevelId("42e01f")));
125 EXPECT_EQ("42e01f",
126 *H264ProfileLevelIdToString(*ParseH264ProfileLevelId("42E01F")));
127 EXPECT_EQ("4d100b",
128 *H264ProfileLevelIdToString(*ParseH264ProfileLevelId("4d100b")));
129 EXPECT_EQ("4d100b",
130 *H264ProfileLevelIdToString(*ParseH264ProfileLevelId("4D100B")));
131 EXPECT_EQ("640c2a",
132 *H264ProfileLevelIdToString(*ParseH264ProfileLevelId("640c2a")));
133 EXPECT_EQ("640c2a",
134 *H264ProfileLevelIdToString(*ParseH264ProfileLevelId("640C2A")));
135 }
136
TEST(H264ProfileLevelId,TestToStringInvalid)137 TEST(H264ProfileLevelId, TestToStringInvalid) {
138 EXPECT_FALSE(H264ProfileLevelIdToString(
139 H264ProfileLevelId(H264Profile::kProfileHigh, H264Level::kLevel1_b)));
140 EXPECT_FALSE(H264ProfileLevelIdToString(H264ProfileLevelId(
141 H264Profile::kProfileConstrainedHigh, H264Level::kLevel1_b)));
142 EXPECT_FALSE(H264ProfileLevelIdToString(
143 H264ProfileLevelId(static_cast<H264Profile>(255), H264Level::kLevel3_1)));
144 }
145
TEST(H264ProfileLevelId,TestParseSdpProfileLevelIdEmpty)146 TEST(H264ProfileLevelId, TestParseSdpProfileLevelIdEmpty) {
147 const absl::optional<H264ProfileLevelId> profile_level_id =
148 ParseSdpForH264ProfileLevelId(SdpVideoFormat::Parameters());
149 EXPECT_TRUE(profile_level_id);
150 EXPECT_EQ(H264Profile::kProfileConstrainedBaseline,
151 profile_level_id->profile);
152 EXPECT_EQ(H264Level::kLevel3_1, profile_level_id->level);
153 }
154
TEST(H264ProfileLevelId,TestParseSdpProfileLevelIdConstrainedHigh)155 TEST(H264ProfileLevelId, TestParseSdpProfileLevelIdConstrainedHigh) {
156 SdpVideoFormat::Parameters params;
157 params["profile-level-id"] = "640c2a";
158 const absl::optional<H264ProfileLevelId> profile_level_id =
159 ParseSdpForH264ProfileLevelId(params);
160 EXPECT_TRUE(profile_level_id);
161 EXPECT_EQ(H264Profile::kProfileConstrainedHigh, profile_level_id->profile);
162 EXPECT_EQ(H264Level::kLevel4_2, profile_level_id->level);
163 }
164
TEST(H264ProfileLevelId,TestParseSdpProfileLevelIdInvalid)165 TEST(H264ProfileLevelId, TestParseSdpProfileLevelIdInvalid) {
166 SdpVideoFormat::Parameters params;
167 params["profile-level-id"] = "foobar";
168 EXPECT_FALSE(ParseSdpForH264ProfileLevelId(params));
169 }
170
171 } // namespace webrtc
172