using System.IO.Ports;
using System.Threading;
using UnityEngine;
using System.Collections.Generic;

/// <summary>
/// Modbus RTU 控制脚本
/// </summary>
public class ModbusMultiDeviceController : MonoBehaviour
{
    [System.Serializable]
    public class DeviceInfo
    {
        public string deviceName;      // 设备名称
        public ushort registerAddress; // 寄存器地址
        public bool isOn;             // 当前状态
    }

    [Header("串口配置")]
    public string portName = "COM3";
    public int baudRate = 9600;
    public Parity parity = Parity.None;
    public int dataBits = 8;
    public StopBits stopBits = StopBits.One;

    [Header("设备列表")]
    public List<DeviceInfo> devices = new List<DeviceInfo>()
    {
        new DeviceInfo { deviceName = "第一路红色管路", registerAddress = 0x0000, isOn = false },
        new DeviceInfo { deviceName = "第二路绿色管路", registerAddress = 0x0001, isOn = false },
        new DeviceInfo { deviceName = "第三路蓝色管路", registerAddress = 0x0002, isOn = false },
        new DeviceInfo { deviceName = "第四路罐体", registerAddress = 0x0003, isOn = false },
        new DeviceInfo { deviceName = "第五路照明", registerAddress = 0x0004, isOn = false },
        new DeviceInfo { deviceName = "第六路第二个罐亮", registerAddress = 0x0005, isOn = false },
        new DeviceInfo { deviceName = "第七路第二个汽轮机工作", registerAddress = 0x0006, isOn = false }
    };

    private SerialPort serialPort;
    private byte deviceAddress = 0x01; // 设备地址
    private byte functionCode = 0x05;   // 功能码：写单个线圈

    // 初始化串口
    public bool OpenPort()
    {
        if (serialPort != null && serialPort.IsOpen)
        {
            Debug.LogWarning("串口已经打开。");
            return true;
        }

        try
        {
            serialPort = new SerialPort(portName, baudRate, parity, dataBits, stopBits);
            serialPort.ReadTimeout = 500;
            serialPort.WriteTimeout = 500;
            serialPort.Open();
            Debug.Log($"串口 {portName} 打开成功，波特率：{baudRate}");
            return true;
        }
        catch (System.Exception ex)
        {
            Debug.LogError($"打开串口失败: {ex.Message}");
            return false;
        }
    }

    // 计算 Modbus RTU 的 CRC16 校验码 (低位在前)
    public static byte[] CalculateCRC(byte[] data, int length)
    {
        ushort crc = 0xFFFF;
        for (int i = 0; i < length; i++)
        {
            crc ^= data[i];
            for (int j = 0; j < 8; j++)
            {
                if ((crc & 0x0001) != 0)
                {
                    crc >>= 1;
                    crc ^= 0xA001;
                }
                else
                {
                    crc >>= 1;
                }
            }
        }
        return new byte[] { (byte)(crc & 0xFF), (byte)((crc >> 8) & 0xFF) };
    }

    // 构建完整指令帧
    public byte[] BuildCommand(ushort registerAddress, bool turnOn)
    {
        ushort value = turnOn ? (ushort)0xFF00 : (ushort)0x0000;
        
        byte[] frame = new byte[8];
        frame[0] = deviceAddress;
        frame[1] = functionCode;
        frame[2] = (byte)(registerAddress >> 8);
        frame[3] = (byte)(registerAddress & 0xFF);
        frame[4] = (byte)(value >> 8);
        frame[5] = (byte)(value & 0xFF);
        
        byte[] crc = CalculateCRC(frame, 6);
        frame[6] = crc[0];
        frame[7] = crc[1];
        
        return frame;
    }

    // 发送指令到指定设备
    public bool SendCommand(int deviceIndex, bool turnOn)
    {
        if (deviceIndex < 0 || deviceIndex >= devices.Count)
        {
            Debug.LogError($"设备索引 {deviceIndex} 无效，有效范围：0-{devices.Count - 1}");
            return false;
        }

        if (serialPort == null || !serialPort.IsOpen)
        {
            Debug.LogError("串口未打开，请先调用 OpenPort()");
            return false;
        }

        DeviceInfo device = devices[deviceIndex];
        byte[] command = BuildCommand(device.registerAddress, turnOn);
        
        try
        {
            serialPort.DiscardInBuffer();
            serialPort.Write(command, 0, command.Length);
            Debug.Log($"[{device.deviceName}] 发送指令: {(turnOn ? "打开" : "关闭")} - {ByteArrayToString(command)}");

            // 等待设备响应
            Thread.Sleep(50);
            
            // 读取响应
            int bytesToRead = serialPort.BytesToRead;
            if (bytesToRead > 0)
            {
                byte[] response = new byte[bytesToRead];
                int bytesRead = serialPort.Read(response, 0, bytesToRead);
                
                // 验证响应
                if (bytesRead == command.Length && ByteArraysEqual(command, response))
                {
                    device.isOn = turnOn;
                    Debug.Log($"[{device.deviceName}] {(turnOn ? "打开" : "关闭")} 成功");
                    return true;
                }
                else
                {
                    Debug.LogWarning($"[{device.deviceName}] 响应验证失败");
                    return false;
                }
            }
            else
            {
                Debug.LogWarning($"[{device.deviceName}] 设备无响应");
                return false;
            }
        }
        catch (System.Exception ex)
        {
            Debug.LogError($"[{device.deviceName}] 通信错误: {ex.Message}");
            return false;
        }
    }

    // 通过设备名称控制
    public bool SendCommandByName(string deviceName, bool turnOn)
    {
        int deviceIndex = devices.FindIndex(d => d.deviceName == deviceName);
        if (deviceIndex == -1)
        {
            Debug.LogError($"未找到设备: {deviceName}");
            return false;
        }
        return SendCommand(deviceIndex, turnOn);
    }

    // 控制所有设备
    public void TurnAllOn()
    {
        Debug.Log("打开所有设备...");
        for (int i = 0; i < devices.Count; i++)
        {
            SendCommand(i, true);
            Thread.Sleep(100); // 设备间间隔
        }
    }

    public void TurnAllOff()
    {
        Debug.Log("关闭所有设备...");
        for (int i = devices.Count - 1; i >= 0; i--)
        {
            SendCommand(i, false);
            Thread.Sleep(100); // 设备间间隔
        }
    }

    // 切换设备状态
    public void ToggleDevice(int deviceIndex)
    {
        if (deviceIndex >= 0 && deviceIndex < devices.Count)
        {
            DeviceInfo device = devices[deviceIndex];
            SendCommand(deviceIndex, !device.isOn);
        }
    }

    // 获取设备状态
    public bool GetDeviceStatus(int deviceIndex)
    {
        if (deviceIndex >= 0 && deviceIndex < devices.Count)
        {
            return devices[deviceIndex].isOn;
        }
        return false;
    }

    public bool GetDeviceStatusByName(string deviceName)
    {
        DeviceInfo device = devices.Find(d => d.deviceName == deviceName);
        return device != null ? device.isOn : false;
    }

    // 工具方法
    private string ByteArrayToString(byte[] bytes)
    {
        return System.BitConverter.ToString(bytes).Replace("-", " ");
    }

    private bool ByteArraysEqual(byte[] a, byte[] b)
    {
        if (a.Length != b.Length) return false;
        for (int i = 0; i < a.Length; i++)
        {
            if (a[i] != b[i]) return false;
        }
        return true;
    }

    // 测试功能 - 按顺序控制所有设备
    public void TestAllDevices()
    {
        if (!OpenPort())
        {
            Debug.LogError("无法打开串口，测试中止");
            return;
        }

        Debug.Log("开始测试所有设备...");
        StartCoroutine(TestSequence());
    }

    private System.Collections.IEnumerator TestSequence()
    {
        // 打开所有设备
        for (int i = 0; i < devices.Count; i++)
        {
            SendCommand(i, true);
            yield return new WaitForSeconds(1.0f); // 等待1秒
        }

        yield return new WaitForSeconds(2.0f); // 等待2秒

        // 关闭所有设备
        for (int i = devices.Count - 1; i >= 0; i--)
        {
            SendCommand(i, false);
            yield return new WaitForSeconds(1.0f); // 等待1秒
        }

        Debug.Log("设备测试完成");
    }

    // 直接控制特定设备（快捷方法）
    public void ControlRedPipe(bool turnOn) => SendCommandByName("第一路红色管路", turnOn);
    public void ControlGreenPipe(bool turnOn) => SendCommandByName("第二路绿色管路", turnOn);
    public void ControlBluePipe(bool turnOn) => SendCommandByName("第三路蓝色管路", turnOn);
    public void ControlTank(bool turnOn) => SendCommandByName("第四路罐体", turnOn);
    public void ControlLighting(bool turnOn) => SendCommandByName("第五路照明", turnOn);
    public void ControlSecondTankLight(bool turnOn) => SendCommandByName("第六路第二个罐亮", turnOn);
    public void ControlSecondTurbine(bool turnOn) => SendCommandByName("第七路第二个汽轮机工作", turnOn);

    // 关闭串口
    public void ClosePort()
    {
        if (serialPort != null && serialPort.IsOpen)
        {
            TurnAllOff(); // 先关闭所有设备
            Thread.Sleep(500);
            serialPort.Close();
            Debug.Log("串口已关闭");
        }
    }

    // Unity 生命周期
    private void Start()
    {
        // 自动打开串口
        OpenPort();
    }

    private void OnDestroy()
    {
        ClosePort();
    }

    private void OnApplicationQuit()
    {
        ClosePort();
    }
}