| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "testing/gtest/include/gtest/gtest.h" | |
| 6 | |
| 7 #include "content/common/gpu/media/h264_bit_reader.h" | |
| 8 | |
| 9 using content::H264BitReader; | |
| 10 | |
| 11 TEST(H264BitReaderTest, ReadStreamWithoutEscapeAndTrailingZeroBytes) { | |
| 12 H264BitReader reader; | |
| 13 const unsigned char rbsp[] = {0x01, 0x23, 0x45, 0x67, 0x89, 0xa0}; | |
| 14 int dummy = 0; | |
| 15 | |
| 16 EXPECT_TRUE(reader.Initialize(rbsp, sizeof(rbsp))); | |
| 17 | |
| 18 EXPECT_TRUE(reader.ReadBits(1, &dummy)); | |
| 19 EXPECT_EQ(dummy, 0x00); | |
| 20 EXPECT_EQ(reader.NumBitsLeft(), 47); | |
| 21 EXPECT_TRUE(reader.HasMoreRBSPData()); | |
| 22 | |
| 23 EXPECT_TRUE(reader.ReadBits(8, &dummy)); | |
| 24 EXPECT_EQ(dummy, 0x02); | |
| 25 EXPECT_EQ(reader.NumBitsLeft(), 39); | |
| 26 EXPECT_TRUE(reader.HasMoreRBSPData()); | |
| 27 | |
| 28 EXPECT_TRUE(reader.ReadBits(31, &dummy)); | |
| 29 EXPECT_EQ(dummy, 0x23456789); | |
| 30 EXPECT_EQ(reader.NumBitsLeft(), 8); | |
| 31 EXPECT_TRUE(reader.HasMoreRBSPData()); | |
| 32 | |
| 33 EXPECT_TRUE(reader.ReadBits(1, &dummy)); | |
| 34 EXPECT_EQ(dummy, 1); | |
| 35 EXPECT_EQ(reader.NumBitsLeft(), 7); | |
| 36 EXPECT_TRUE(reader.HasMoreRBSPData()); | |
| 37 | |
| 38 EXPECT_TRUE(reader.ReadBits(1, &dummy)); | |
| 39 EXPECT_EQ(dummy, 0); | |
| 40 EXPECT_EQ(reader.NumBitsLeft(), 6); | |
| 41 EXPECT_FALSE(reader.HasMoreRBSPData()); | |
| 42 } | |
| 43 | |
| 44 TEST(H264BitReaderTest, SingleByteStream) { | |
| 45 H264BitReader reader; | |
| 46 const unsigned char rbsp[] = {0x18}; | |
| 47 int dummy = 0; | |
| 48 | |
| 49 EXPECT_TRUE(reader.Initialize(rbsp, sizeof(rbsp))); | |
| 50 EXPECT_EQ(reader.NumBitsLeft(), 8); | |
| 51 EXPECT_TRUE(reader.HasMoreRBSPData()); | |
| 52 | |
| 53 EXPECT_TRUE(reader.ReadBits(4, &dummy)); | |
| 54 EXPECT_EQ(dummy, 0x01); | |
| 55 EXPECT_EQ(reader.NumBitsLeft(), 4); | |
| 56 EXPECT_FALSE(reader.HasMoreRBSPData()); | |
| 57 } | |
| 58 | |
| 59 TEST(H264BitReaderTest, StopBitOccupyFullByte) { | |
| 60 H264BitReader reader; | |
| 61 const unsigned char rbsp[] = {0xab, 0x80}; | |
| 62 int dummy = 0; | |
| 63 | |
| 64 EXPECT_TRUE(reader.Initialize(rbsp, sizeof(rbsp))); | |
| 65 EXPECT_EQ(reader.NumBitsLeft(), 16); | |
| 66 EXPECT_TRUE(reader.HasMoreRBSPData()); | |
| 67 | |
| 68 EXPECT_TRUE(reader.ReadBits(8, &dummy)); | |
| 69 EXPECT_EQ(dummy, 0xab); | |
| 70 EXPECT_EQ(reader.NumBitsLeft(), 8); | |
| 71 EXPECT_FALSE(reader.HasMoreRBSPData()); | |
| 72 } | |
| OLD | NEW |