h264_bit_reader_unittest.cc 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. // Copyright 2014 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. #include "media/video/h264_bit_reader.h"
  5. #include "testing/gtest/include/gtest/gtest.h"
  6. namespace media {
  7. TEST(H264BitReaderTest, ReadStreamWithoutEscapeAndTrailingZeroBytes) {
  8. H264BitReader reader;
  9. const unsigned char rbsp[] = {0x01, 0x23, 0x45, 0x67, 0x89, 0xa0};
  10. int dummy = 0;
  11. EXPECT_TRUE(reader.Initialize(rbsp, sizeof(rbsp)));
  12. EXPECT_TRUE(reader.ReadBits(1, &dummy));
  13. EXPECT_EQ(dummy, 0x00);
  14. EXPECT_EQ(reader.NumBitsLeft(), 47);
  15. EXPECT_TRUE(reader.HasMoreRBSPData());
  16. EXPECT_TRUE(reader.ReadBits(8, &dummy));
  17. EXPECT_EQ(dummy, 0x02);
  18. EXPECT_EQ(reader.NumBitsLeft(), 39);
  19. EXPECT_TRUE(reader.HasMoreRBSPData());
  20. EXPECT_TRUE(reader.ReadBits(31, &dummy));
  21. EXPECT_EQ(dummy, 0x23456789);
  22. EXPECT_EQ(reader.NumBitsLeft(), 8);
  23. EXPECT_TRUE(reader.HasMoreRBSPData());
  24. EXPECT_TRUE(reader.ReadBits(1, &dummy));
  25. EXPECT_EQ(dummy, 1);
  26. EXPECT_EQ(reader.NumBitsLeft(), 7);
  27. EXPECT_TRUE(reader.HasMoreRBSPData());
  28. EXPECT_TRUE(reader.ReadBits(1, &dummy));
  29. EXPECT_EQ(dummy, 0);
  30. EXPECT_EQ(reader.NumBitsLeft(), 6);
  31. EXPECT_FALSE(reader.HasMoreRBSPData());
  32. }
  33. TEST(H264BitReaderTest, SingleByteStream) {
  34. H264BitReader reader;
  35. const unsigned char rbsp[] = {0x18};
  36. int dummy = 0;
  37. EXPECT_TRUE(reader.Initialize(rbsp, sizeof(rbsp)));
  38. EXPECT_EQ(reader.NumBitsLeft(), 8);
  39. EXPECT_TRUE(reader.HasMoreRBSPData());
  40. EXPECT_TRUE(reader.ReadBits(4, &dummy));
  41. EXPECT_EQ(dummy, 0x01);
  42. EXPECT_EQ(reader.NumBitsLeft(), 4);
  43. EXPECT_FALSE(reader.HasMoreRBSPData());
  44. }
  45. TEST(H264BitReaderTest, StopBitOccupyFullByte) {
  46. H264BitReader reader;
  47. const unsigned char rbsp[] = {0xab, 0x80};
  48. int dummy = 0;
  49. EXPECT_TRUE(reader.Initialize(rbsp, sizeof(rbsp)));
  50. EXPECT_EQ(reader.NumBitsLeft(), 16);
  51. EXPECT_TRUE(reader.HasMoreRBSPData());
  52. EXPECT_TRUE(reader.ReadBits(8, &dummy));
  53. EXPECT_EQ(dummy, 0xab);
  54. EXPECT_EQ(reader.NumBitsLeft(), 8);
  55. EXPECT_FALSE(reader.HasMoreRBSPData());
  56. }
  57. } // namespace media