using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace DL.Com.Protocol.Modbus
{
    /// <summary>
    /// 数据帧:报文头+帧结构
    /// </summary>
    public class DataFrame
    {
        /// <summary>
        /// 报文头
        /// </summary>
        public MBAP MBAP;
        /// <summary>
        /// 帧结构
        /// </summary>
        public PDU PDU;
        public DataFrame()
        {
            MBAP = new MBAP();
            PDU = new PDU();
        }
        /// <summary>
        /// 设置帧数据
        /// </summary>
        /// <param name="functionCode">功能码</param>
        /// <param name="addr">开始地址</param>
        /// <param name="value">数量</param>
        /// <param name="bytelenght">字节长度，写多个线圈/寄存器使用</param>
        /// <param name="valueofmulti">写入的值,写多个线圈/寄存器使用</param>
        public void SetPDU(byte functionCode, ushort addr, ushort value, byte bytelenght = 0, byte[] valueofmulti = null)
        {
            PDU.SetPDUFunctionCode(functionCode);
            PDU.SetPDUData(addr, value, bytelenght, valueofmulti);
            MBAP.SetDataLength(PDU.GetLength());
        }
        /// <summary>
        /// 设置帧数据
        /// </summary>
        /// <param name="functionCode">功能码</param>
        /// <param name="addr">开始地址</param>
        /// <param name="value">数量</param>
        /// <param name="bytelenght">字节长度，写多个线圈/寄存器使用</param>
        /// <param name="valueofmulti">写入的值,写多个线圈/寄存器使用</param>
        public void SetPDU(byte functionCode, Byte2 addr, Byte2 value, byte bytelenght = 0x00, byte[] valueofmulti = null)
        {
            PDU.SetPDUFunctionCode(functionCode);
            PDU.SetPDUData(addr, value, bytelenght, valueofmulti);
            MBAP.SetDataLength(PDU.GetLength());
        }
        /// <summary>
        /// 设置报文头
        /// </summary>
        public void SetMBAP(ushort serialnum, byte equipAddr, ushort protocolid = 0)
        {
            MBAP.SetSerialNum(serialnum);
            MBAP.SetEquipAddr(equipAddr);
            MBAP.SetProtocolId(protocolid);
        }
        /// <summary>
        /// 设置报文头
        /// </summary>
        public void SetMBAP(Byte2 serialnum, byte equipAddr, ushort protocolid = 0)
        {
            MBAP.SetSerialNum(serialnum);
            MBAP.SetEquipAddr(equipAddr);
            MBAP.SetProtocolId(protocolid);
        }
        /// <summary>
        /// 数据帧字节数据
        /// </summary>
        /// <returns></returns>
        public byte[] ToBytes()
        {
            List<byte> tmp = new List<byte>();
            tmp.AddRange(MBAP.ToBytes());
            tmp.AddRange(PDU.ToBytes());
            return tmp.ToArray();
        }
        public byte[] ToMbapFuncCodeBytes()
        {
            List<byte> tmp = new List<byte>();
            tmp.AddRange(MBAP.ToBytes());
            tmp.Add(PDU.FuncCode);
            return tmp.ToArray();
        }
    }

    /// <summary>
    /// 报文头
    /// </summary>
    public class MBAP
    {
        /// <summary>
        /// 报文序列号，事务处理标识符
        /// 长度：2字节
        /// </summary>
        public Byte2 SerialNum;
        /// <summary>
        /// 协议标识符
        /// 0000标识ModbusTCP协议
        /// 长度：2字节
        /// </summary>
        public Byte2 ProtocolId;
        /// <summary>
        /// 接下来的数据长度
        /// 长度：2字节
        /// </summary>
        public Byte2 DataLength;
        /// <summary>
        /// 设备地址
        /// 长度：1字节
        /// </summary>
        public byte EquipAddr;

        public MBAP()
        {
            SerialNum = new Byte2();
            ProtocolId = new Byte2();
            DataLength = new Byte2();
        }
        public void SetData(byte[] data)
        {
            if (data.Length > 7)
            {
                SerialNum.Set(data[0], data[1]);
                ProtocolId.Set(data[2], data[3]);
                DataLength.Set(data[4], data[5]);
                EquipAddr = data[6];
            }
        }
        public void SetSerialNum(UInt16 serialnum)
        {
            SerialNum.SetByInt(serialnum);
        }
        public void SetSerialNum(byte Hi, byte Low)
        {
            SerialNum.Set(Hi, Low);
        }
        public void SetSerialNum(Byte2 stru2Byte)
        {
            SerialNum.Set(stru2Byte);
        }
        /// <summary>
        /// 设置协议ID，默认TCP协议
        /// </summary>
        /// <param name="protocolid"></param>
        public void SetProtocolId(UInt16 protocolid = 0)
        {
            ProtocolId.SetByInt(protocolid);
        }
        /// <summary>
        ///  设置协议ID，默认TCP协议
        /// </summary>
        /// <param name="Hi"></param>
        /// <param name="Low"></param>
        public void SetProtocolId(byte Hi = 0x00, byte Low = 0x00)
        {
            ProtocolId.Set(Hi, Low);
        }
        public void SetProtocolId(Byte2 stru2Byte)
        {
            ProtocolId.Set(stru2Byte);
        }
        /// <summary>
        /// 设置协议地址
        /// </summary>
        /// <param name="equipAddr"></param>
        public void SetEquipAddr(byte equipAddr)
        {
            EquipAddr = equipAddr;
        }
        /// <summary>
        /// 设置接下来的数据长度
        /// </summary>
        /// <param name="datalength">PDU里的数据长度</param>
        public void SetDataLength(ushort datalength)
        {
            DataLength.SetByInt((ushort)(datalength+1));
        }
        /// <summary>
        ///  设置接下来的数据长度
        /// </summary>
        /// <param name="Hi"></param>
        /// <param name="Low"></param>
        public void SetDataLength(byte Hi, byte Low)
        {
            DataLength.Set(Hi, Low);
        }
        public override bool Equals(object obj)
        {
            MBAP cMBAP = obj as MBAP;
            if (cMBAP == null)
                return false;
            if (cMBAP.SerialNum.Equals(this.SerialNum) &&
                cMBAP.ProtocolId.Equals(this.ProtocolId) &&
                cMBAP.EquipAddr.Equals(this.EquipAddr))
                return true;
            else
                return false;
        }
        /// <summary>
        /// 报文头字节数据
        /// </summary>
        /// <returns></returns>
        public byte[] ToBytes()
        {
            List<byte> tmp = new List<byte>();
            tmp.AddRange(SerialNum.ToBytes());
            tmp.AddRange(ProtocolId.ToBytes());
            tmp.AddRange(DataLength.ToBytes());
            tmp.Add(EquipAddr);
            return tmp.ToArray();
        }
    }
    /// <summary>
    /// 数据帧
    /// </summary>
    public class PDU
    {
        /// <summary>
        /// 功能码
        /// </summary>
        public byte FuncCode;
        /// <summary>
        /// 数据
        /// </summary>
        public DataInPDU Data;

        public PDU()
        {
            Data = new DataInPDU();
        }
        /// <summary>
        /// 设置帧结构功能码
        /// </summary>
        /// <param name="funcCode">功能码</param>
        public void SetPDUFunctionCode(byte funcCode)
        {
            FuncCode = funcCode;
        }
        public void SetPDUData(ushort addr, ushort value, byte bytelenght = 0, byte[] valueofmulti = null)
        {
            Data.SetData(addr, value, bytelenght, valueofmulti);
        }
        public void SetPDUData(Byte2 addr, Byte2 value, byte bytelenght = 0x00, byte[] valueofmulti = null)
        {
            Data.SetData(addr, value, bytelenght, valueofmulti);
        }
        /// <summary>
        /// 帧结构的数据长度
        /// </summary>
        /// <returns></returns>
        public ushort GetLength()
        {
            return (ushort)(Data.Length() + 1);
        }
        public byte[] ToBytes()
        {
            List<byte> tmp = new List<byte>();
            tmp.Add(FuncCode);
            tmp.AddRange(Data.ToBytes());
            return tmp.ToArray();
        }
    }

    public class DataInPDU
    {
        /// <summary>
        /// 起始地址/寄存器地址
        /// </summary>
        public Byte2 Addr;
        /// <summary>
        /// 数量/值
        /// </summary>
        public Byte2 Value;
        /// <summary>
        /// 字节长度，写多个线圈/保持寄存器时使用
        /// </summary>
        public byte ByteLength;
        /// <summary>
        /// 输出值/寄存器值，写多个线圈/保持寄存器时使用
        /// </summary>
        public List<byte> ValueOfMulti;

        public DataInPDU()
        {
            Addr = new Byte2();
            Value = new Byte2();
            ValueOfMulti = new List<byte>();
        }
        public void SetData(ushort addr, ushort value, byte bytelenght = 0, byte[] valueofmulti = null)
        {
            Addr.SetByInt(addr);
            Value.SetByInt(value);
            ByteLength = bytelenght;
            ValueOfMulti.Clear();
            if (valueofmulti != null)
                foreach (var item in valueofmulti)
                {
                    ValueOfMulti.Add(item);
                }

        }
        public void SetData(Byte2 addr, Byte2 value, byte bytelenght = 0x00, byte[] valueofmulti = null)
        {
            Addr.Set(addr);
            Value.Set(value);
            ByteLength = bytelenght;
            ValueOfMulti.Clear();
            if (valueofmulti != null)
                foreach (var item in valueofmulti)
                {
                    ValueOfMulti.Add(item);
                }

        }
        public byte[] ToBytes()
        {
            List<byte> tmp = new List<byte>();
            tmp.AddRange(Addr.ToBytes());
            tmp.AddRange(Value.ToBytes());
            if (ByteLength != 0)
            {
                tmp.Add(ByteLength);
                foreach (var item in ValueOfMulti)
                {
                    tmp.Add(item);
                }
            }
            return tmp.ToArray();
        }
        public byte[] AddrValueToBytes()
        {
            List<byte> tmp = new List<byte>();
            tmp.AddRange(Addr.ToBytes());
            tmp.AddRange(Value.ToBytes());
            return tmp.ToArray();
        }
        public int Length()
        {
            List<byte> tmp = new List<byte>();
            tmp.AddRange(Addr.ToBytes());
            tmp.AddRange(Value.ToBytes());
            if (ByteLength != 0)
            {
                tmp.Add(ByteLength);
                foreach (var item in ValueOfMulti)
                {
                    tmp.Add(item);
                }
            }
            return tmp.Count;
        }
    }
    /// <summary>
    /// 两字节组成的值
    /// </summary>
    public struct Byte2
    {
        public byte ByteH { get; set; }
        public byte ByteL { get; set; }

        public static Byte2 GetStru2Byte(ushort val)
        {
            return new Byte2
            {
                ByteH = (byte)((val >> 8) & 0xff),
                ByteL = (byte)((val & 0xff))
            };
        }
        public static ushort GetUInt16(byte H,byte L)
        {
            return (ushort)(((H & 0xff) << 8) | (L & 0xff));
        }
        public static short GetInt16(byte H, byte L)
        {
            return (short)((H << 8) | (L & 0xff));
        }
        public static int GetInt32(byte H, byte L)
        {
            return ((H << 8) | (L & 0xff));
        }
        public static uint GetUInt32(byte H, byte L)
        {
            return (uint)(((H & 0xff) << 8) | (L & 0xff));
        }
        public void SetByInt(ushort val)
        {
            ByteH = (byte)((val >> 8) & 0xff);
            ByteL = (byte)((val & 0xff));
        }
        public void Set(byte High, byte Low)
        {
            ByteH = High;
            ByteL = Low;
        }
        public void Set(Byte2 stru2Byte)
        {
            ByteH = stru2Byte.ByteH;
            ByteL = stru2Byte.ByteL;
        }
        public ushort GetIntValue()
        {
            return (ushort)(((ByteH & 0xff) << 8) | (ByteL & 0xff));
        }
        public override string ToString()
        {
            return $"{ByteH.ToString("x2")} {ByteL.ToString("x2")}";
        }
        public override bool Equals(object obj)
        {
            Byte2 stru2Byte = (Byte2)obj;

            return ByteH.Equals(stru2Byte.ByteH) && ByteL.Equals(stru2Byte.ByteL);
        }
        public byte[] ToBytes()
        {
            return new byte[] { ByteH, ByteL };
        }
    }
    /// <summary>
    /// 四字节组成的值
    /// </summary>
    public struct Byte4
    {
        public byte ByteH1 { get; set; }
        public byte ByteH2 { get; set; }
        public byte ByteL1 { get; set; }
        public byte ByteL2 { get; set; }
        public static Byte4 GetStru2Byte(int val)
        {
            return new Byte4
            {
                ByteH1 = (byte)((val >> 24) & 0xff),
                ByteH2 = (byte)((val >> 16) & 0xff),
                ByteL1 = (byte)((val >> 8) & 0xff),
                ByteL2 = (byte)((val & 0xff))
            };
        }
        public void SetByInt(uint val)
        {
            ByteH1 = (byte)((val >> 24) & 0xff);
            ByteH2 = (byte)((val >> 16) & 0xff);
            ByteL1 = (byte)((val >> 8) & 0xff);
            ByteL2 = (byte)((val & 0xff));
        }
        public void Set(Byte4 stru4Byte)
        {
            ByteH1 = stru4Byte.ByteH1;
            ByteH2 = stru4Byte.ByteH2;
            ByteL1 = stru4Byte.ByteL1;
            ByteL2 = stru4Byte.ByteL2;
        }
        public static uint GetUInt32(byte h1,byte h2,byte l1,byte l2)
        {
            return (uint)(((h1 & 0xff) << 24) | ((h2 & 0xff) << 16) | ((l1 & 0xff) << 8) | (l2 & 0xff));
        }
        public static int GetInt32(byte h1, byte h2, byte l1, byte l2)
        {
            return (int)((h1 << 24) | ((h2 & 0xff) << 16) | ((l1 & 0xff) << 8) | (l2 & 0xff));
        }
        /// <summary>
        /// 大终端转float
        /// </summary>
        /// <param name="h1"></param>
        /// <param name="h2"></param>
        /// <param name="l1"></param>
        /// <param name="l2"></param>
        /// <returns></returns>
        public static float GetFloat(byte h1, byte h2, byte l1, byte l2)
        {
            byte[] tmp = new byte[4] { h1,h2,l1,l2};
            return (float)(GetInt32(h1, h2, l1, l2));
            //float f = 0;
            //byte[] x = tmp;
            //byte i;
            //unsafe
            //{
            //    void* pf;
            //    fixed(byte* pb = x)
            //    {
            //        pf = &f;
            //        for (i = 0; i < tmp.Length; i++)
            //        {
            //            *((byte*)pf + i) = *(pb+i);
            //        }
            //    }
            //}
            //return f;
        }
        public uint GetIntValue()
        {
            return (uint)(((ByteH1 & 0xff) << 24) | ((ByteH2 & 0xff) << 16) | ((ByteL1 & 0xff) << 8) | (ByteL2 & 0xff));
        }
        public override string ToString()
        {
            return $"{ByteH1.ToString("x2")} {ByteH2.ToString("x2")} {ByteL1.ToString("x2")} {ByteL2.ToString("x2")}";
        }
        public override bool Equals(object obj)
        {
            Byte4 stru4Byte = (Byte4)obj;

            return ByteH1.Equals(stru4Byte.ByteH1) && ByteH2.Equals(stru4Byte.ByteH2) && ByteL1.Equals(stru4Byte.ByteL1) && ByteL2.Equals(stru4Byte.ByteL2);
        }
        public byte[] ToBytes()
        {
            return new byte[] { ByteH1, ByteH2, ByteL1, ByteL2 };
        }
    }
    public class ResponseData
    {
        public MBAP MBAP { get; set; }
        public byte FuncCode { get; set; }
        public byte[] OtherData { get; set; }
        public ResponseData()
        {
            MBAP = new MBAP();
        }
        public void Set(byte[] data)
        {
            if (data == null)
                return;
            MBAP.SetData(data);
            FuncCode = data[7];
            OtherData = new byte[data.Length - 8];
            Array.Copy(data, 8, OtherData, 0, data.Length - 8);
        }
        public byte[] ToBytes()
        {
            List<byte> vs = new List<byte>();
            vs.AddRange(MBAP.ToBytes());
            vs.Add(FuncCode);
            vs.AddRange(OtherData);
            return vs.ToArray();
        }

    }
}