| | | 1 | | using System.Collections; |
| | | 2 | | using System.Runtime.Serialization; |
| | | 3 | | |
| | | 4 | | namespace NLightning.Infrastructure.Serialization.Node; |
| | | 5 | | |
| | | 6 | | using Converters; |
| | | 7 | | using Domain.Node; |
| | | 8 | | using Interfaces; |
| | | 9 | | |
| | | 10 | | public class FeatureSetSerializer : IFeatureSetSerializer |
| | | 11 | | { |
| | | 12 | | /// <summary> |
| | | 13 | | /// Serializes the features to a binary writer. |
| | | 14 | | /// </summary> |
| | | 15 | | /// <param name="featureSet">The features to serialize.</param> |
| | | 16 | | /// <param name="stream">The stream to write to.</param> |
| | | 17 | | /// <param name="asGlobal">If the features should be serialized as a global feature set.</param> |
| | | 18 | | /// <param name="includeLength">If the length of the byte array should be included.</param> |
| | | 19 | | /// <remarks> |
| | | 20 | | /// If the features are serialized as a global feature set, only the first 13 bits are serialized. |
| | | 21 | | /// </remarks> |
| | | 22 | | /// <remarks> |
| | | 23 | | /// If the length of the byte array is included, the first 2 bytes are written as the length of the byte array. |
| | | 24 | | /// </remarks> |
| | | 25 | | public async Task SerializeAsync(FeatureSet featureSet, Stream stream, bool asGlobal = false, |
| | | 26 | | bool includeLength = true) |
| | | 27 | | { |
| | | 28 | | // If it's a global feature, cut out any bit greater than 13 |
| | 192 | 29 | | if (asGlobal) |
| | 12 | 30 | | featureSet.FeatureFlags.Length = 13; |
| | | 31 | | |
| | | 32 | | // Convert BitArray to byte array |
| | 192 | 33 | | var bytes = new byte[(featureSet.FeatureFlags.Length + 7) / 8]; |
| | 192 | 34 | | featureSet.FeatureFlags.CopyTo(bytes, 0); |
| | | 35 | | |
| | | 36 | | // Set bytes as big endian |
| | 192 | 37 | | if (BitConverter.IsLittleEndian) |
| | 192 | 38 | | Array.Reverse(bytes); |
| | | 39 | | |
| | | 40 | | // Trim leading zero bytes |
| | 192 | 41 | | var leadingZeroBytes = 0; |
| | 4740 | 42 | | foreach (var t in bytes) |
| | | 43 | | { |
| | 2272 | 44 | | if (t == 0) |
| | | 45 | | { |
| | 2084 | 46 | | leadingZeroBytes++; |
| | | 47 | | } |
| | | 48 | | else |
| | | 49 | | { |
| | | 50 | | break; |
| | | 51 | | } |
| | | 52 | | } |
| | | 53 | | |
| | 192 | 54 | | var trimmedBytes = bytes[leadingZeroBytes..]; |
| | | 55 | | |
| | | 56 | | // Write the length of the byte array or 1 if all bytes are zero |
| | 192 | 57 | | if (includeLength) |
| | 132 | 58 | | await stream.WriteAsync(EndianBitConverter.GetBytesBigEndian((ushort)trimmedBytes.Length)); |
| | | 59 | | |
| | | 60 | | // Otherwise, return the array starting from the first non-zero byte |
| | 192 | 61 | | await stream.WriteAsync(trimmedBytes); |
| | 192 | 62 | | } |
| | | 63 | | |
| | | 64 | | /// <summary> |
| | | 65 | | /// Deserializes the features from a binary reader. |
| | | 66 | | /// </summary> |
| | | 67 | | /// <param name="stream">The stream to read from.</param> |
| | | 68 | | /// <param name="includeLength">If the length of the byte array is included.</param> |
| | | 69 | | /// <remarks> |
| | | 70 | | /// If the length of the byte array is included, the first 2 bytes are read as the length of the byte array. |
| | | 71 | | /// </remarks> |
| | | 72 | | /// <returns>The deserialized features.</returns> |
| | | 73 | | /// <exception cref="SerializationException">Error deserializing Features</exception> |
| | | 74 | | public async Task<FeatureSet> DeserializeAsync(Stream stream, bool includeLength = true) |
| | | 75 | | { |
| | | 76 | | try |
| | | 77 | | { |
| | 88 | 78 | | var length = 8; |
| | | 79 | | |
| | 88 | 80 | | var bytes = new byte[2]; |
| | 88 | 81 | | if (includeLength) |
| | | 82 | | { |
| | | 83 | | // Read the length of the byte array |
| | 84 | 84 | | await stream.ReadExactlyAsync(bytes); |
| | 84 | 85 | | length = EndianBitConverter.ToUInt16BigEndian(bytes); |
| | | 86 | | } |
| | | 87 | | |
| | | 88 | | // Read the byte array |
| | 88 | 89 | | bytes = new byte[length]; |
| | 88 | 90 | | await stream.ReadExactlyAsync(bytes); |
| | | 91 | | |
| | 88 | 92 | | if (BitConverter.IsLittleEndian) |
| | 88 | 93 | | Array.Reverse(bytes); |
| | | 94 | | |
| | | 95 | | // Convert the byte array to BitArray |
| | 88 | 96 | | return new FeatureSet { FeatureFlags = new BitArray(bytes) }; |
| | | 97 | | } |
| | 0 | 98 | | catch (Exception e) |
| | | 99 | | { |
| | 0 | 100 | | throw new SerializationException("Error deserializing Features", e); |
| | | 101 | | } |
| | 88 | 102 | | } |
| | | 103 | | } |