vp8_bool_decoder_unittest.cc 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125
  1. // Copyright 2015 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/parsers/vp8_bool_decoder.h"
  5. #include <stddef.h>
  6. #include <stdint.h>
  7. #include <limits>
  8. #include "testing/gtest/include/gtest/gtest.h"
  9. namespace media {
  10. const static size_t NUM_BITS_TO_TEST = 100;
  11. namespace {
  12. // 100 zeros with probability of 0x80.
  13. const uint8_t kDataZerosAndEvenProbabilities[] = {
  14. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  15. 0x00, 0x00, 0x00};
  16. // 100 ones with probability of 0x80.
  17. const uint8_t kDataOnesAndEvenProbabilities[] = {
  18. 0xfe, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
  19. 0xff, 0xf0, 0x20};
  20. // [0, 1, 0, 1, ..., 1] with probability [0, 1, 2, 3, ..., 99].
  21. const uint8_t kDataParitiesAndIncreasingProbabilities[] = {
  22. 0x00, 0x02, 0x08, 0x31, 0x8e, 0xca, 0xab, 0xe2, 0xc8, 0x31, 0x12,
  23. 0xb3, 0x2c, 0x19, 0x90, 0xc6, 0x6a, 0xeb, 0x17, 0x52, 0x30};
  24. } // namespace
  25. class Vp8BoolDecoderTest : public ::testing::Test {
  26. public:
  27. Vp8BoolDecoderTest() = default;
  28. Vp8BoolDecoderTest(const Vp8BoolDecoderTest&) = delete;
  29. Vp8BoolDecoderTest& operator=(const Vp8BoolDecoderTest&) = delete;
  30. protected:
  31. // Fixture member, the bool decoder to be tested.
  32. Vp8BoolDecoder bd_;
  33. };
  34. #define INITIALIZE(data) ASSERT_TRUE(bd_.Initialize(data, sizeof(data)));
  35. TEST_F(Vp8BoolDecoderTest, DecodeBoolsWithZerosAndEvenProbabilities) {
  36. INITIALIZE(kDataZerosAndEvenProbabilities);
  37. ASSERT_EQ(0u, bd_.BitOffset());
  38. for (size_t i = 0; i < NUM_BITS_TO_TEST; ++i) {
  39. bool out = true;
  40. ASSERT_TRUE(bd_.ReadBool(&out, 0x80));
  41. ASSERT_FALSE(out);
  42. ASSERT_EQ(i, bd_.BitOffset());
  43. }
  44. }
  45. TEST_F(Vp8BoolDecoderTest, DecodeLiteralsWithZerosAndEvenProbabilities) {
  46. INITIALIZE(kDataZerosAndEvenProbabilities);
  47. int value = 1;
  48. ASSERT_TRUE(bd_.ReadLiteral(1, &value));
  49. ASSERT_EQ(0, value);
  50. value = 1;
  51. ASSERT_TRUE(bd_.ReadLiteral(32, &value));
  52. ASSERT_EQ(0, value);
  53. value = 1;
  54. ASSERT_TRUE(bd_.ReadLiteralWithSign(1, &value));
  55. ASSERT_EQ(0, value);
  56. value = 1;
  57. ASSERT_TRUE(bd_.ReadLiteralWithSign(31, &value));
  58. ASSERT_EQ(0, value);
  59. }
  60. TEST_F(Vp8BoolDecoderTest, DecodeBoolsWithOnesAndEvenProbabilities) {
  61. INITIALIZE(kDataOnesAndEvenProbabilities);
  62. ASSERT_EQ(0u, bd_.BitOffset());
  63. for (size_t i = 0; i < NUM_BITS_TO_TEST; ++i) {
  64. bool out = false;
  65. ASSERT_TRUE(bd_.ReadBool(&out, 0x80));
  66. ASSERT_TRUE(out);
  67. ASSERT_EQ(i + 1, bd_.BitOffset());
  68. }
  69. }
  70. TEST_F(Vp8BoolDecoderTest, DecodeLiteralsWithOnesAndEvenProbabilities) {
  71. INITIALIZE(kDataOnesAndEvenProbabilities);
  72. int value = 0;
  73. ASSERT_TRUE(bd_.ReadLiteral(1, &value));
  74. EXPECT_EQ(1, value);
  75. value = 0;
  76. ASSERT_TRUE(bd_.ReadLiteral(31, &value));
  77. EXPECT_EQ(0x7FFFFFFF, value);
  78. value = 0;
  79. ASSERT_TRUE(bd_.ReadLiteralWithSign(1, &value));
  80. EXPECT_EQ(-1, value);
  81. value = 0;
  82. ASSERT_TRUE(bd_.ReadLiteralWithSign(31, &value));
  83. EXPECT_EQ(-0x7FFFFFFF, value);
  84. }
  85. TEST_F(Vp8BoolDecoderTest, DecodeBoolsWithParitiesAndIncreasingProbabilities) {
  86. INITIALIZE(kDataParitiesAndIncreasingProbabilities);
  87. for (size_t i = 0; i < NUM_BITS_TO_TEST; ++i) {
  88. bool out = !(i & 1);
  89. ASSERT_LE(i, std::numeric_limits<uint8_t>::max());
  90. ASSERT_TRUE(bd_.ReadBool(&out, static_cast<uint8_t>(i)));
  91. EXPECT_EQ(out, !!(i & 1));
  92. }
  93. }
  94. } // namespace media