| | 1 | | using System.Buffers.Binary; |
| | 2 | | using System.Runtime.CompilerServices; |
| | 3 | |
|
| | 4 | | namespace NLightning.Common.Utils; |
| | 5 | |
|
| | 6 | | using Domain.Serialization; |
| | 7 | |
|
| 244 | 8 | | public class BitReader(byte[] buffer) : IBitReader |
| | 9 | | { |
| 244 | 10 | | private readonly byte[] _buffer = buffer ?? throw new ArgumentNullException(nameof(buffer)); |
| 244 | 11 | | private readonly int _totalBits = buffer.Length * 8; |
| | 12 | |
|
| | 13 | | private int _bitOffset; |
| | 14 | |
|
| | 15 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | 16 | | public int ReadBits(Span<byte> value, int bitLength) |
| | 17 | | { |
| 2268 | 18 | | return ReadBits(value, 0, bitLength); |
| | 19 | | } |
| | 20 | |
|
| | 21 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | 22 | | private int ReadBits(Span<byte> value, int valueOffset, int bitLength) |
| | 23 | | { |
| 2268 | 24 | | if (_bitOffset + bitLength > _totalBits) |
| | 25 | | { |
| 0 | 26 | | throw new InvalidOperationException($"Not enough bits left to read. Requested {bitLength}, but only {_totalB |
| | 27 | | } |
| | 28 | |
|
| 2268 | 29 | | var byteOffset = _bitOffset / 8; |
| 2268 | 30 | | var shift = (int)((uint)_bitOffset % 8); |
| | 31 | |
|
| 2268 | 32 | | if (shift == 0) |
| | 33 | | { |
| | 34 | | // copy bytes to value |
| 524 | 35 | | _buffer.AsSpan(byteOffset, bitLength / 8 + (bitLength % 8 == 0 ? 0 : 1)).CopyTo(value[valueOffset..]); |
| | 36 | |
|
| | 37 | | // mask extra bits |
| 524 | 38 | | if (bitLength % 8 != 0) |
| | 39 | | { |
| 460 | 40 | | value[valueOffset + bitLength / 8] &= (byte)(0xFF << (8 - bitLength % 8)); |
| | 41 | | } |
| | 42 | | } |
| | 43 | | else |
| | 44 | | { |
| 1744 | 45 | | var bytesToRead = bitLength / 8 + (shift > 0 ? 0 : -1); |
| 1744 | 46 | | var maxBytesToRead = value.Length - 1 - valueOffset; |
| 1744 | 47 | | if (bytesToRead > maxBytesToRead) |
| | 48 | | { |
| 52 | 49 | | bytesToRead = maxBytesToRead; |
| | 50 | | } |
| | 51 | |
|
| 19656 | 52 | | for (var i = 0; i <= bytesToRead; i++) |
| | 53 | | { |
| 8084 | 54 | | var left = (byte)(_buffer[byteOffset + i] << shift); |
| 8084 | 55 | | var right = (byte)((_buffer[byteOffset + i + 1] & |
| 8084 | 56 | | (i == bytesToRead + 1 ? 0xFF << (8 - bitLength % 8) : 0xFF)) >> (8 - shift)); |
| | 57 | |
|
| 8084 | 58 | | value[valueOffset + i] = (byte)(left | right); |
| | 59 | | } |
| | 60 | | } |
| | 61 | |
|
| 2268 | 62 | | _bitOffset += bitLength; |
| 2268 | 63 | | return bitLength; |
| | 64 | | } |
| | 65 | |
|
| | 66 | | public bool ReadBit() |
| | 67 | | { |
| 2848 | 68 | | if (_bitOffset >= _totalBits) |
| | 69 | | { |
| 4 | 70 | | throw new InvalidOperationException("No more bits to read."); |
| | 71 | | } |
| | 72 | |
|
| 2844 | 73 | | var byteIndex = _bitOffset / 8; |
| 2844 | 74 | | var bitIndex = _bitOffset % 8; |
| | 75 | |
|
| | 76 | | // Extract the bit at the current BitOffset |
| 2844 | 77 | | var bit = (_buffer[byteIndex] >> (7 - bitIndex)) & 1; |
| | 78 | |
|
| | 79 | | // Increment the BitOffset |
| 2844 | 80 | | _bitOffset++; |
| | 81 | |
|
| 2844 | 82 | | return bit == 1; |
| | 83 | | } |
| | 84 | |
|
| | 85 | | public byte ReadByteFromBits(int bits) |
| | 86 | | { |
| 920 | 87 | | Span<byte> bytes = stackalloc byte[sizeof(byte)]; |
| 920 | 88 | | ReadBits(bytes, bits); |
| 920 | 89 | | var mask = (1 << bits) - 1; |
| 920 | 90 | | return (byte)((bytes[0] >> (sizeof(byte) * 8 - bits)) & mask); |
| | 91 | | } |
| | 92 | |
|
| | 93 | | public short ReadInt16FromBits(int bits, bool bigEndian = true) |
| | 94 | | { |
| 876 | 95 | | Span<byte> bytes = stackalloc byte[sizeof(short)]; |
| 876 | 96 | | ReadBits(bytes, bits); |
| | 97 | |
|
| 876 | 98 | | if ((bigEndian && BitConverter.IsLittleEndian) || (!bigEndian && !BitConverter.IsLittleEndian)) |
| | 99 | | { |
| 872 | 100 | | bytes.Reverse(); |
| | 101 | | } |
| | 102 | |
|
| 876 | 103 | | var mask = (1 << bits) - 1; |
| 876 | 104 | | return (short)((BinaryPrimitives.ReadInt16LittleEndian(bytes) >> (sizeof(short) * 8 - bits)) & mask); |
| | 105 | | } |
| | 106 | |
|
| | 107 | | public ushort ReadUInt16FromBits(int bits, bool bigEndian = true) |
| | 108 | | { |
| 8 | 109 | | Span<byte> bytes = stackalloc byte[sizeof(ushort)]; |
| 8 | 110 | | ReadBits(bytes, bits); |
| | 111 | |
|
| 8 | 112 | | if ((bigEndian && BitConverter.IsLittleEndian) || (!bigEndian && !BitConverter.IsLittleEndian)) |
| | 113 | | { |
| 8 | 114 | | bytes.Reverse(); |
| | 115 | | } |
| | 116 | |
|
| 8 | 117 | | var mask = (1 << bits) - 1; |
| 8 | 118 | | return (ushort)((BinaryPrimitives.ReadUInt16LittleEndian(bytes) >> (sizeof(ushort) * 8 - bits)) & mask); |
| | 119 | | } |
| | 120 | |
|
| | 121 | | public int ReadInt32FromBits(int bits, bool bigEndian = true) |
| | 122 | | { |
| 64 | 123 | | Span<byte> bytes = stackalloc byte[sizeof(int)]; |
| 64 | 124 | | ReadBits(bytes, bits); |
| | 125 | |
|
| 64 | 126 | | if ((bigEndian && BitConverter.IsLittleEndian) || (!bigEndian && !BitConverter.IsLittleEndian)) |
| | 127 | | { |
| 64 | 128 | | bytes.Reverse(); |
| | 129 | | } |
| | 130 | |
|
| 64 | 131 | | var mask = bits != 32 ? (1 << bits) - 1 : -1; |
| 64 | 132 | | return (BinaryPrimitives.ReadInt32LittleEndian(bytes) >> (sizeof(int) * 8 - bits)) & mask; |
| | 133 | | } |
| | 134 | |
|
| | 135 | | public long ReadInt64FromBits(int bits, bool bigEndian = true) |
| | 136 | | { |
| 68 | 137 | | Span<byte> bytes = stackalloc byte[sizeof(long)]; |
| 68 | 138 | | ReadBits(bytes, bits); |
| | 139 | |
|
| 68 | 140 | | if ((bigEndian && BitConverter.IsLittleEndian) || (!bigEndian && !BitConverter.IsLittleEndian)) |
| | 141 | | { |
| 68 | 142 | | bytes.Reverse(); |
| | 143 | | } |
| | 144 | |
|
| 68 | 145 | | return BinaryPrimitives.ReadInt64LittleEndian(bytes) >> (sizeof(long) * 8 - bits); |
| | 146 | | } |
| | 147 | |
|
| | 148 | | public bool HasMoreBits(int requiredBits) |
| | 149 | | { |
| 1288 | 150 | | return _bitOffset + requiredBits <= _totalBits; |
| | 151 | | } |
| | 152 | |
|
| | 153 | | public void SkipBits(int v) |
| | 154 | | { |
| 32 | 155 | | if (_bitOffset + v > _totalBits) |
| | 156 | | { |
| 0 | 157 | | throw new InvalidOperationException($"Cannot skip {v} bits, only {_totalBits - _bitOffset} remain."); |
| | 158 | | } |
| | 159 | |
|
| 32 | 160 | | _bitOffset += v; |
| 32 | 161 | | } |
| | 162 | | } |