diff options
| author | gdkchan <gab.dark.100@gmail.com> | 2019-12-31 21:08:02 -0300 |
|---|---|---|
| committer | Thog <thog@protonmail.com> | 2020-01-09 02:13:00 +0100 |
| commit | 0dbfe3c23ee072ec9dbc477f955a163107af2be1 (patch) | |
| tree | f7f3501ef32891e80197717ab18b22a8d88fdb9f /Ryujinx.Graphics.Nvdec/VDec/H264BitStreamWriter.cs | |
| parent | 6e092c05584ccbbd548ac26c056c1a7edfa6c1a0 (diff) | |
Re-add NVDEC project (not integrated)
Diffstat (limited to 'Ryujinx.Graphics.Nvdec/VDec/H264BitStreamWriter.cs')
| -rw-r--r-- | Ryujinx.Graphics.Nvdec/VDec/H264BitStreamWriter.cs | 79 |
1 files changed, 79 insertions, 0 deletions
diff --git a/Ryujinx.Graphics.Nvdec/VDec/H264BitStreamWriter.cs b/Ryujinx.Graphics.Nvdec/VDec/H264BitStreamWriter.cs new file mode 100644 index 00000000..b4fad59b --- /dev/null +++ b/Ryujinx.Graphics.Nvdec/VDec/H264BitStreamWriter.cs @@ -0,0 +1,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; + } + } +}
\ No newline at end of file |
