OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright (c) 2015 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 "webrtc/modules/rtp_rtcp/source/h264_bitstream_parser.h" | |
12 | |
13 #include "testing/gtest/include/gtest/gtest.h" | |
14 | |
15 namespace webrtc { | |
16 | |
17 // SPS/PPS part of below chunk. | |
18 uint8_t kH264SpsPps[] = {0x00, 0x00, 0x00, 0x01, 0x67, 0x42, 0x80, 0x20, 0xda, | |
19 0x01, 0x40, 0x16, 0xe8, 0x06, 0xd0, 0xa1, 0x35, 0x00, | |
20 0x00, 0x00, 0x01, 0x68, 0xce, 0x06, 0xe2}; | |
21 | |
22 // Contains enough of the image slice to contain slice QP. | |
23 uint8_t kH264BitstreamChunk[] = { | |
24 0x00, 0x00, 0x00, 0x01, 0x67, 0x42, 0x80, 0x20, 0xda, 0x01, 0x40, 0x16, | |
25 0xe8, 0x06, 0xd0, 0xa1, 0x35, 0x00, 0x00, 0x00, 0x01, 0x68, 0xce, 0x06, | |
26 0xe2, 0x00, 0x00, 0x00, 0x01, 0x65, 0xb8, 0x40, 0xf0, 0x8c, 0x03, 0xf2, | |
27 0x75, 0x67, 0xad, 0x41, 0x64, 0x24, 0x0e, 0xa0, 0xb2, 0x12, 0x1e, 0xf8, | |
28 }; | |
29 | |
30 // Contains enough of the image slice to contain slice QP. | |
31 uint8_t kH264BitstreamNextImageSliceChunk[] = { | |
32 0x00, 0x00, 0x00, 0x01, 0x41, 0xe2, 0x01, 0x16, 0x0e, 0x3e, 0x2b, 0x86, | |
33 }; | |
34 | |
35 TEST(H264BitstreamParserTest, ReportsNoQpWithoutParsedSlices) { | |
36 H264BitstreamParser h264_parser; | |
37 int qp; | |
38 EXPECT_FALSE(h264_parser.GetLastSliceQp(&qp)); | |
39 } | |
40 | |
41 TEST(H264BitstreamParserTest, ReportsNoQpWithOnlyParsedPpsAndSpsSlices) { | |
42 H264BitstreamParser h264_parser; | |
43 h264_parser.ParseBitstream(kH264SpsPps, sizeof(kH264SpsPps)); | |
44 int qp; | |
45 EXPECT_FALSE(h264_parser.GetLastSliceQp(&qp)); | |
46 } | |
47 | |
48 TEST(H264BitstreamParserTest, ReportsLastSliceQpForImageSlices) { | |
49 H264BitstreamParser h264_parser; | |
50 h264_parser.ParseBitstream(kH264BitstreamChunk, sizeof(kH264BitstreamChunk)); | |
51 int qp; | |
52 ASSERT_TRUE(h264_parser.GetLastSliceQp(&qp)); | |
53 EXPECT_EQ(35, qp); | |
54 | |
55 // Parse an additional image slice. | |
56 h264_parser.ParseBitstream(kH264BitstreamNextImageSliceChunk, | |
57 sizeof(kH264BitstreamNextImageSliceChunk)); | |
58 ASSERT_TRUE(h264_parser.GetLastSliceQp(&qp)); | |
59 EXPECT_EQ(37, qp); | |
60 } | |
61 | |
62 } // namespace webrtc | |
OLD | NEW |