aboutsummaryrefslogtreecommitdiff
path: root/Ryujinx.Graphics/VDec/H264BitStreamWriter.cs
blob: b388a2aafb1f1f74105a6ea84b054191bd572828 (plain) (blame)
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
using System.IO;

namespace Ryujinx.Graphics.VDec
{
    class H264BitStreamWriter : BitStreamWriter
    {
        public H264BitStreamWriter(Stream BaseStream) : base(BaseStream) { }

        public void WriteU(int Value, int ValueSize)
        {
            WriteBits(Value, ValueSize);
        }

        public void WriteSe(int Value)
        {
            WriteExpGolombCodedInt(Value);
        }

        public void WriteUe(int Value)
        {
            WriteExpGolombCodedUInt((uint)Value);
        }

        public void End()
        {
            WriteBit(true);

            Flush();
        }

        private void WriteExpGolombCodedInt(int Value)
        {
            int Sign = Value <= 0 ? 0 : 1;

            if (Value < 0)
            {
                Value = -Value;
            }

            Value = (Value << 1) - Sign;

            WriteExpGolombCodedUInt((uint)Value);
        }

        private void WriteExpGolombCodedUInt(uint Value)
        {
            int Size = 32 - CountLeadingZeros((int)Value + 1);

            WriteBits(1, Size);

            Value -= (1u << (Size - 1)) - 1;

            WriteBits((int)Value, Size - 1);
        }

        private static readonly byte[] ClzNibbleTbl = { 4, 3, 2, 2, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0 };

        private static int CountLeadingZeros(int Value)
        {
            if (Value == 0)
            {
                return 32;
            }

            int NibbleIdx = 32;
            int PreCount, Count = 0;

            do
            {
                NibbleIdx -= 4;
                PreCount = ClzNibbleTbl[(Value >> NibbleIdx) & 0b1111];
                Count += PreCount;
            }
            while (PreCount == 4);

            return Count;
        }
    }
}