136 lines
3.8 KiB
C++
136 lines
3.8 KiB
C++
// Copyright 2014 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 <gtest/gtest.h>
|
|
|
|
#include "packager/media/codecs/h26x_bit_reader.h"
|
|
|
|
namespace shaka {
|
|
namespace media {
|
|
|
|
TEST(H26xBitReaderTest, ReadStreamWithoutEscapeAndTrailingZeroBytes) {
|
|
H26xBitReader reader;
|
|
const unsigned char rbsp[] = {0x01, 0x23, 0x45, 0x67, 0x89, 0xa0};
|
|
int dummy = 0;
|
|
|
|
EXPECT_TRUE(reader.Initialize(rbsp, sizeof(rbsp)));
|
|
|
|
EXPECT_TRUE(reader.ReadBits(1, &dummy));
|
|
EXPECT_EQ(dummy, 0x00);
|
|
EXPECT_EQ(reader.NumBitsLeft(), 47);
|
|
EXPECT_TRUE(reader.HasMoreRBSPData());
|
|
|
|
EXPECT_TRUE(reader.ReadBits(8, &dummy));
|
|
EXPECT_EQ(dummy, 0x02);
|
|
EXPECT_EQ(reader.NumBitsLeft(), 39);
|
|
EXPECT_TRUE(reader.HasMoreRBSPData());
|
|
|
|
EXPECT_TRUE(reader.ReadBits(31, &dummy));
|
|
EXPECT_EQ(dummy, 0x23456789);
|
|
EXPECT_EQ(reader.NumBitsLeft(), 8);
|
|
EXPECT_TRUE(reader.HasMoreRBSPData());
|
|
|
|
EXPECT_TRUE(reader.ReadBits(1, &dummy));
|
|
EXPECT_EQ(dummy, 1);
|
|
EXPECT_EQ(reader.NumBitsLeft(), 7);
|
|
EXPECT_TRUE(reader.HasMoreRBSPData());
|
|
|
|
EXPECT_TRUE(reader.ReadBits(1, &dummy));
|
|
EXPECT_EQ(dummy, 0);
|
|
EXPECT_EQ(reader.NumBitsLeft(), 6);
|
|
EXPECT_FALSE(reader.HasMoreRBSPData());
|
|
}
|
|
|
|
TEST(H26xBitReaderTest, ReadPpsWithTrailingZeroByte) {
|
|
H26xBitReader reader;
|
|
|
|
// Data copied from
|
|
// https://github.com/shaka-project/shaka-packager/issues/418.
|
|
const unsigned char pps_rbsp[] = {0xee, 0x3c, 0x80, 0x00};
|
|
EXPECT_TRUE(reader.Initialize(pps_rbsp, sizeof(pps_rbsp)));
|
|
|
|
// Skips all the fields in PPS (kind of simulates ParsePps).
|
|
EXPECT_TRUE(reader.SkipBits(16));
|
|
|
|
EXPECT_EQ(reader.NumBitsLeft(), 16);
|
|
// The remaining data is '80 00'. The trailing null byte is ignored. There
|
|
// are no bits before the stop bit, so there is no more RBSP data.
|
|
EXPECT_FALSE(reader.HasMoreRBSPData());
|
|
}
|
|
|
|
TEST(H26xBitReaderTest, SingleByteStream) {
|
|
H26xBitReader reader;
|
|
const unsigned char rbsp[] = {0x18};
|
|
int dummy = 0;
|
|
|
|
EXPECT_TRUE(reader.Initialize(rbsp, sizeof(rbsp)));
|
|
EXPECT_EQ(reader.NumBitsLeft(), 8);
|
|
EXPECT_TRUE(reader.HasMoreRBSPData());
|
|
|
|
EXPECT_TRUE(reader.ReadBits(4, &dummy));
|
|
EXPECT_EQ(dummy, 0x01);
|
|
EXPECT_EQ(reader.NumBitsLeft(), 4);
|
|
EXPECT_FALSE(reader.HasMoreRBSPData());
|
|
}
|
|
|
|
TEST(H26xBitReaderTest, ReadBool) {
|
|
H26xBitReader reader;
|
|
const unsigned char rbsp[] = {0xc5};
|
|
bool dummy = false;
|
|
|
|
EXPECT_TRUE(reader.Initialize(rbsp, sizeof(rbsp)));
|
|
EXPECT_EQ(reader.NumBitsLeft(), 8);
|
|
|
|
EXPECT_TRUE(reader.ReadBool(&dummy));
|
|
EXPECT_TRUE(dummy);
|
|
EXPECT_TRUE(reader.ReadBool(&dummy));
|
|
EXPECT_TRUE(dummy);
|
|
EXPECT_TRUE(reader.ReadBool(&dummy));
|
|
EXPECT_FALSE(dummy);
|
|
EXPECT_TRUE(reader.ReadBool(&dummy));
|
|
EXPECT_FALSE(dummy);
|
|
|
|
EXPECT_EQ(reader.NumBitsLeft(), 4);
|
|
}
|
|
|
|
TEST(H26xBitReaderTest, SkipBits) {
|
|
H26xBitReader reader;
|
|
const unsigned char rbsp[] = {0xc5, 0x41, 0x51};
|
|
int dummy;
|
|
|
|
EXPECT_TRUE(reader.Initialize(rbsp, sizeof(rbsp)));
|
|
EXPECT_EQ(reader.NumBitsLeft(), 24);
|
|
|
|
EXPECT_TRUE(reader.SkipBits(3));
|
|
EXPECT_EQ(21, reader.NumBitsLeft());
|
|
EXPECT_TRUE(reader.ReadBits(4, &dummy));
|
|
EXPECT_EQ(0x2, dummy);
|
|
EXPECT_TRUE(reader.SkipBits(8));
|
|
EXPECT_EQ(9, reader.NumBitsLeft());
|
|
EXPECT_TRUE(reader.ReadBits(5, &dummy));
|
|
EXPECT_EQ(0x15, dummy);
|
|
EXPECT_EQ(4, reader.NumBitsLeft());
|
|
EXPECT_FALSE(reader.SkipBits(5));
|
|
EXPECT_TRUE(reader.SkipBits(0));
|
|
EXPECT_EQ(4, reader.NumBitsLeft());
|
|
}
|
|
|
|
TEST(H26xBitReaderTest, StopBitOccupyFullByte) {
|
|
H26xBitReader reader;
|
|
const unsigned char rbsp[] = {0xab, 0x80};
|
|
int dummy = 0;
|
|
|
|
EXPECT_TRUE(reader.Initialize(rbsp, sizeof(rbsp)));
|
|
EXPECT_EQ(reader.NumBitsLeft(), 16);
|
|
EXPECT_TRUE(reader.HasMoreRBSPData());
|
|
|
|
EXPECT_TRUE(reader.ReadBits(8, &dummy));
|
|
EXPECT_EQ(dummy, 0xab);
|
|
EXPECT_EQ(reader.NumBitsLeft(), 8);
|
|
EXPECT_FALSE(reader.HasMoreRBSPData());
|
|
}
|
|
|
|
} // namespace media
|
|
} // namespace shaka
|