amazfitbiptools/WatchFace.Parser/Models/Parameter.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Text;
using NLog;
using NLog.Internal;

namespace WatchFace.Parser.Models
{
    public class Parameter
    {
        private static readonly Logger Logger = LogManager.GetCurrentClassLogger();

        public Parameter(byte id, long value)
        {
            Id = id;
            Value = value;
        }

        public Parameter(byte id, List<Parameter> value)
        {
            Id = id;
            Children = value;
        }

        public byte Id { get; }
        public long Value { get; }
        public List<Parameter> Children { get; }
        public bool HasChildren => Children != null;

        public long Write(Stream stream, int traceOffset = 0)
        {
            var size = (long) 0;
            var flags = HasChildren ? ParameterFlags.HasChildren : 0;
            var rawId = (byte) ((Id << 3) + flags);
            stream.WriteByte(rawId);

            size += 1;
            if (HasChildren)
            {
                Logger.Trace(() => TraceWithOffset($"{Id} ({rawId:X2}):", traceOffset));
                size += WriteList(stream, traceOffset + 1);
                Logger.Trace(() => TraceWithOffset($"{size} bytes", traceOffset));
                return size;
            }
            size += WriteValue(stream, Value, traceOffset);
            Logger.Trace(() => TraceWithOffset($"{Id} ({rawId:X2}): {Value} ({Value:X2})", traceOffset));
            return size;
        }

        public long WriteList(Stream stream, int traceOffset)
        {
            var temporaryStream = new MemoryStream();
            foreach (var parameter in Children) parameter.Write(temporaryStream, traceOffset);
            var size = WriteValue(stream, temporaryStream.Length, traceOffset);
            temporaryStream.Seek(0, SeekOrigin.Begin);
            temporaryStream.Copy(stream);
            size += temporaryStream.Length;
            return size;
        }

        public long WriteValue(Stream stream, long value, int traceOffset)
        {
            var unsignedValue = (ulong) value;
            var size = 0;
            byte currentByte;
            while (unsignedValue >= 0x80)
            {
                currentByte = (byte) ((unsignedValue & 0x7f) | 0x80);
                stream.WriteByte(currentByte);
                size += 1;
                unsignedValue = unsignedValue >> 7;
            }
            currentByte = (byte) (unsignedValue & 0x7f);
            stream.WriteByte(currentByte);
            size += 1;
            return size;
        }

        public static List<Parameter> ReadList(Stream stream, int traceOffset = 0)
        {
            var result = new List<Parameter>();
            while (stream.Position < stream.Length)
            {
                var parameter = ReadFrom(stream, traceOffset);
                result.Add(parameter);
            }
            return result;
        }

        public static Parameter ReadFrom(Stream fileStream, int traceOffset = 0)
        {
            var rawId = ReadByte(fileStream);
            var id = (byte) ((rawId & 0xf8) >> 3);
            var flags = (ParameterFlags) (rawId & 0x7);

            if (id == 0)
                throw new ArgumentException("Parameter with zero Id is invalid.");

            var value = ReadValue(fileStream);
            if (flags.HasFlag(ParameterFlags.HasChildren))
            {
                Logger.Trace(() => TraceWithOffset($"{id} ({rawId:X2}): {value} bytes", traceOffset));
                var buffer = new byte[value];
                fileStream.Read(buffer, 0, (int) value);
                var stream = new MemoryStream(buffer);

                var list = ReadList(stream, traceOffset + 1);
                return new Parameter(id, list);
            }
            Logger.Trace(() => TraceWithOffset($"{id} ({rawId:X2}): {value} ({value:X2})", traceOffset));
            return new Parameter(id, value);
        }

        private static long ReadValue(Stream fileStream)
        {
            var bytesLength = 0;
            long value = 0;
            var offset = 0;

            var currentByte = ReadByte(fileStream);
            bytesLength += 1;
            while ((currentByte & 0x80) > 0)
            {
                if (bytesLength > 9)
                    throw new ArgumentException("Value of the parameter too long.");
                value = value | ((long) (currentByte & 0x7f) << offset);
                offset += 7;
                currentByte = ReadByte(fileStream);
                bytesLength += 1;
            }
            value = value | ((long) (currentByte & 0x7f) << offset);
            return value;
        }

        private static byte ReadByte(Stream stream)
        {
            var currentByte = stream.ReadByte();
            if (currentByte == -1)
                throw new ArgumentException("Reading buffer is empty.");
            return (byte) currentByte;
        }

        private static string TraceWithOffset(string message, int offset)
        {
            var builder = new StringBuilder();
            for (var i = 0; i < offset; i++) builder.Append("    ");
            builder.Append(message);
            return builder.ToString();
        }
    }
}