| 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 |