1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "media/base/bit_reader.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace media {
TEST(BitReaderTest, NormalOperationTest) {
uint8 value8;
uint64 value64;
// 0101 0101 1001 1001 repeats 4 times
uint8 buffer[] = {0x55, 0x99, 0x55, 0x99, 0x55, 0x99, 0x55, 0x99};
BitReader reader1(buffer, 6); // Initialize with 6 bytes only
EXPECT_TRUE(reader1.ReadBits(1, &value8));
EXPECT_EQ(value8, 0);
EXPECT_TRUE(reader1.ReadBits(8, &value8));
EXPECT_EQ(value8, 0xab); // 1010 1011
EXPECT_TRUE(reader1.ReadBits(7, &value64));
EXPECT_TRUE(reader1.ReadBits(32, &value64));
EXPECT_EQ(value64, 0x55995599u);
EXPECT_FALSE(reader1.ReadBits(1, &value8));
value8 = 0xff;
EXPECT_TRUE(reader1.ReadBits(0, &value8));
EXPECT_EQ(value8, 0);
BitReader reader2(buffer, 8);
EXPECT_TRUE(reader2.ReadBits(64, &value64));
EXPECT_EQ(value64, 0x5599559955995599ull);
EXPECT_FALSE(reader2.ReadBits(1, &value8));
EXPECT_TRUE(reader2.ReadBits(0, &value8));
}
TEST(BitReaderTest, ReadBeyondEndTest) {
uint8 value8;
uint8 buffer[] = {0x12};
BitReader reader1(buffer, sizeof(buffer));
EXPECT_TRUE(reader1.ReadBits(4, &value8));
EXPECT_FALSE(reader1.ReadBits(5, &value8));
EXPECT_FALSE(reader1.ReadBits(1, &value8));
EXPECT_TRUE(reader1.ReadBits(0, &value8));
}
TEST(BitReaderTest, SkipBitsTest) {
uint8 value8;
uint8 buffer[] = { 0x0a, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 };
BitReader reader1(buffer, sizeof(buffer));
EXPECT_TRUE(reader1.SkipBits(2));
EXPECT_TRUE(reader1.ReadBits(3, &value8));
EXPECT_EQ(value8, 1);
EXPECT_TRUE(reader1.SkipBits(11));
EXPECT_TRUE(reader1.ReadBits(8, &value8));
EXPECT_EQ(value8, 3);
EXPECT_TRUE(reader1.SkipBits(76));
EXPECT_TRUE(reader1.ReadBits(4, &value8));
EXPECT_EQ(value8, 13);
EXPECT_FALSE(reader1.SkipBits(100));
EXPECT_TRUE(reader1.SkipBits(0));
EXPECT_FALSE(reader1.SkipBits(1));
}
TEST(BitReaderTest, BitsReadTest) {
int value;
bool flag;
uint8 buffer[] = { 0x0a, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 };
BitReader reader1(buffer, sizeof(buffer));
EXPECT_EQ(reader1.bits_available(), 120);
EXPECT_TRUE(reader1.SkipBits(2));
EXPECT_EQ(reader1.bits_read(), 2);
EXPECT_EQ(reader1.bits_available(), 118);
EXPECT_TRUE(reader1.ReadBits(3, &value));
EXPECT_EQ(reader1.bits_read(), 5);
EXPECT_EQ(reader1.bits_available(), 115);
EXPECT_TRUE(reader1.ReadFlag(&flag));
EXPECT_EQ(reader1.bits_read(), 6);
EXPECT_EQ(reader1.bits_available(), 114);
EXPECT_TRUE(reader1.SkipBits(76));
EXPECT_EQ(reader1.bits_read(), 82);
EXPECT_EQ(reader1.bits_available(), 38);
}
} // namespace media
|
