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