1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
2 /* This Source Code Form is subject to the terms of the Mozilla Public
3 * License, v. 2.0. If a copy of the MPL was not distributed with this
4 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
5
6 #include "gtest/gtest.h"
7 #include "BitReader.h"
8 #include "BitWriter.h"
9 #include "H264.h"
10
11 using namespace mozilla;
12
TEST(BitWriter,BitWriter)13 TEST(BitWriter, BitWriter)
14 {
15 RefPtr<MediaByteBuffer> test = new MediaByteBuffer();
16 BitWriter b(test);
17 b.WriteBit(false);
18 b.WriteBits(~1ULL, 1); // ensure that extra bits don't modify byte buffer.
19 b.WriteBits(3, 1);
20 b.WriteUE(1280 / 16 - 1);
21 b.WriteUE(720 / 16 - 1);
22 b.WriteUE(1280);
23 b.WriteUE(720);
24 b.WriteBit(true);
25 b.WriteBit(false);
26 b.WriteBit(true);
27 b.WriteU8(7);
28 b.WriteU32(16356);
29 b.WriteU64(116356);
30 b.WriteBits(~(0ULL) & ~1ULL, 16);
31 const uint32_t length = b.BitCount();
32 b.CloseWithRbspTrailing();
33
34 BitReader c(test);
35
36 EXPECT_EQ(c.ReadBit(), false);
37 EXPECT_EQ(c.ReadBit(), false);
38 EXPECT_EQ(c.ReadBit(), true);
39 EXPECT_EQ(c.ReadUE(), 1280u / 16 - 1);
40 EXPECT_EQ(c.ReadUE(), 720u / 16 - 1);
41 EXPECT_EQ(c.ReadUE(), 1280u);
42 EXPECT_EQ(c.ReadUE(), 720u);
43 EXPECT_EQ(c.ReadBit(), true);
44 EXPECT_EQ(c.ReadBit(), false);
45 EXPECT_EQ(c.ReadBit(), true);
46 EXPECT_EQ(c.ReadBits(8), 7u);
47 EXPECT_EQ(c.ReadU32(), 16356u);
48 EXPECT_EQ(c.ReadU64(), 116356u);
49 EXPECT_EQ(c.ReadBits(16), 0xfffeu);
50 EXPECT_EQ(length, BitReader::GetBitLength(test));
51 }
52
TEST(BitWriter,SPS)53 TEST(BitWriter, SPS)
54 {
55 uint8_t sps_pps[] = {0x01, 0x4d, 0x40, 0x0c, 0xff, 0xe1, 0x00, 0x1b, 0x67,
56 0x4d, 0x40, 0x0c, 0xe8, 0x80, 0x80, 0x9d, 0x80, 0xb5,
57 0x01, 0x01, 0x01, 0x40, 0x00, 0x00, 0x03, 0x00, 0x40,
58 0x00, 0x00, 0x0f, 0x03, 0xc5, 0x0a, 0x44, 0x80, 0x01,
59 0x00, 0x04, 0x68, 0xeb, 0xef, 0x20};
60
61 RefPtr<MediaByteBuffer> extraData = new MediaByteBuffer();
62 extraData->AppendElements(sps_pps, sizeof(sps_pps));
63 SPSData spsdata1;
64 bool success = H264::DecodeSPSFromExtraData(extraData, spsdata1);
65 EXPECT_EQ(success, true);
66
67 RefPtr<MediaByteBuffer> extraData2 =
68 H264::CreateExtraData(0x42, 0xc0, 0x1e, {1280, 720});
69 SPSData spsdata2;
70 success = H264::DecodeSPSFromExtraData(extraData2, spsdata2);
71 EXPECT_EQ(success, true);
72 }
73