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 "media/base/bit_reader.h" |
| 6 |
| 7 #include "testing/gtest/include/gtest/gtest.h" |
| 8 |
| 9 namespace media { |
| 10 |
| 11 TEST(BitReaderTest, EmptyStreamTest) { |
| 12 BitReader reader(NULL, 0); |
| 13 uint8 value8 = 0xff; |
| 14 |
| 15 EXPECT_EQ(reader.NumBitsLeft(), 0); |
| 16 EXPECT_EQ(reader.Tell(), 0); |
| 17 EXPECT_TRUE(reader.ReadBits(0, &value8)); |
| 18 EXPECT_TRUE(reader.SkipBits(0)); |
| 19 EXPECT_EQ(reader.Tell(), 0); |
| 20 EXPECT_EQ(reader.NumBitsLeft(), 0); |
| 21 EXPECT_FALSE(reader.ReadBits(1, &value8)); |
| 22 EXPECT_FALSE(reader.SkipBits(1)); |
| 23 EXPECT_EQ(value8, 0); |
| 24 } |
| 25 |
| 26 TEST(BitReaderTest, NormalOperationTest) { |
| 27 // 0101 0101 1001 1001 repeats 4 times |
| 28 uint8 buffer[] = {0x55, 0x99, 0x55, 0x99, 0x55, 0x99, 0x55, 0x99}; |
| 29 BitReader reader(buffer, 6); // Initialize with 6 bytes only |
| 30 uint8 value8; |
| 31 uint64 value64; |
| 32 |
| 33 EXPECT_EQ(reader.NumBitsLeft(), 48); |
| 34 EXPECT_EQ(reader.Tell(), 0); |
| 35 EXPECT_TRUE(reader.ReadBits(1, &value8)); |
| 36 EXPECT_EQ(value8, 0); |
| 37 EXPECT_TRUE(reader.ReadBits(8, &value8)); |
| 38 EXPECT_EQ(value8, 0xab); // 1010 1011 |
| 39 EXPECT_EQ(reader.Tell(), 9); |
| 40 EXPECT_EQ(reader.NumBitsLeft(), 39); |
| 41 EXPECT_TRUE(reader.SkipBits(7)); |
| 42 EXPECT_EQ(reader.Tell(), 16); |
| 43 EXPECT_TRUE(reader.ReadBits(32, &value64)); |
| 44 EXPECT_EQ(value64, 0x55995599u); |
| 45 EXPECT_EQ(reader.Tell(), 48); |
| 46 EXPECT_EQ(reader.NumBitsLeft(), 0); |
| 47 EXPECT_FALSE(reader.SkipBits(1)); |
| 48 EXPECT_FALSE(reader.ReadBits(1, &value8)); |
| 49 EXPECT_TRUE(reader.SkipBits(0)); |
| 50 value8 = 0xff; |
| 51 EXPECT_TRUE(reader.ReadBits(0, &value8)); |
| 52 EXPECT_EQ(value8, 0); |
| 53 |
| 54 reader.Initialize(buffer, 8); |
| 55 EXPECT_TRUE(reader.ReadBits(64, &value64)); |
| 56 EXPECT_EQ(value64, 0x5599559955995599ull); |
| 57 EXPECT_EQ(reader.NumBitsLeft(), 0); |
| 58 EXPECT_EQ(reader.Tell(), 64); |
| 59 EXPECT_FALSE(reader.ReadBits(1, &value8)); |
| 60 EXPECT_FALSE(reader.SkipBits(1)); |
| 61 EXPECT_TRUE(reader.ReadBits(0, &value8)); |
| 62 EXPECT_TRUE(reader.SkipBits(0)); |
| 63 } |
| 64 |
| 65 TEST(BitReaderTest, LongSkipTest) { |
| 66 uint8 buffer[] = { |
| 67 0x12, 0x34, 0x56, 0x78, 0x12, 0x34, 0x56, 0x78, // 64 * 1 |
| 68 0x12, 0x34, 0x56, 0x78, 0x12, 0x34, 0x56, 0x78, // 64 * 2 |
| 69 0x12, 0x34, 0x56, 0x78, 0x12, 0x34, 0x56, 0x78, // 64 * 3 |
| 70 0x12, 0x34, 0x56, 0x78, 0x12, 0x34, 0x56, 0x78, // 64 * 4 |
| 71 0x12, 0x34, 0x56, 0x78, 0x12, 0x34, 0x56, 0x78, // 64 * 5 |
| 72 0x12, 0x34, 0x56, 0x78, 0x12, 0x34, 0x56, 0x78, // 64 * 6 |
| 73 0x12, 0x34, 0x56, 0x78, 0x12, 0x34, 0x56, 0x78, // 64 * 7 |
| 74 0x12, 0x34, 0x56, 0x78, 0x12, 0x34, 0x56, 0x78, // 64 * 8 |
| 75 0x87, 0x65 |
| 76 }; |
| 77 BitReader reader(buffer, sizeof(buffer)); |
| 78 uint8 value8; |
| 79 |
| 80 EXPECT_TRUE(reader.SkipBits(64 * 8 + 8)); |
| 81 EXPECT_EQ(reader.Tell(), 64 * 8 + 8); |
| 82 EXPECT_TRUE(reader.ReadBits(8, &value8)); |
| 83 EXPECT_EQ(value8, 0x65); |
| 84 EXPECT_EQ(reader.Tell(), 64 * 8 + 16); |
| 85 EXPECT_EQ(reader.NumBitsLeft(), 0); |
| 86 } |
| 87 |
| 88 TEST(BitReaderTest, ReadBeyondEndTest) { |
| 89 uint8 buffer[] = {0x12}; |
| 90 BitReader reader(buffer, sizeof(buffer)); |
| 91 uint8 value8; |
| 92 |
| 93 EXPECT_TRUE(reader.SkipBits(4)); |
| 94 EXPECT_FALSE(reader.ReadBits(5, &value8)); |
| 95 EXPECT_FALSE(reader.ReadBits(1, &value8)); |
| 96 EXPECT_FALSE(reader.SkipBits(1)); |
| 97 EXPECT_TRUE(reader.ReadBits(0, &value8)); |
| 98 EXPECT_TRUE(reader.SkipBits(0)); |
| 99 |
| 100 reader.Initialize(buffer, sizeof(buffer)); |
| 101 EXPECT_TRUE(reader.SkipBits(4)); |
| 102 EXPECT_FALSE(reader.SkipBits(5)); |
| 103 EXPECT_FALSE(reader.ReadBits(1, &value8)); |
| 104 EXPECT_FALSE(reader.SkipBits(1)); |
| 105 EXPECT_TRUE(reader.ReadBits(0, &value8)); |
| 106 EXPECT_TRUE(reader.SkipBits(0)); |
| 107 } |
| 108 |
| 109 } // namespace media |
OLD | NEW |