gidrolock-configurator/Datasheet.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.Diagnostics;
using System.Drawing;
using System.IO;
using System.IO.Ports;
using System.Linq;
using System.Security.Cryptography;
using System.Threading;
using System.Threading.Tasks;
using System.Windows.Forms;

namespace Gidrolock_Modbus_Scanner
{
    public partial class Datasheet : Form
    {
        byte modbusID;
        Device device;

        public static ModbusResponseEventArgs latestMessage;

        SerialPort port = Modbus.port;
        bool isPolling = false;
        static bool isAwaitingResponse = false;
        static bool responseReceived = false;


        bool isValveClosed = false;
        bool alarmStatus = false;
        bool cleaningStatus = false;

        List<WiredSensor> wiredSensors = new List<WiredSensor>();
        List<WirelessSensor> wirelessSensors;
        public static string firmwarePath;

        Stopwatch stopwatch = new Stopwatch();
        Thread fileThread = new Thread((ThreadStart)delegate
        {
            OpenFileDialog ofd = new OpenFileDialog();
            ofd.InitialDirectory = Application.StartupPath;
            ofd.RestoreDirectory = true;

            if (ofd.ShowDialog() == DialogResult.OK)
                firmwarePath = ofd.FileName;
        });

        public Datasheet(byte modbusID, Device device) : base()
        {
            InitializeComponent();
            firmwareProgressBar.Minimum = 0;
            firmwareProgressBar.Maximum = 100;
            nudModbusID.Minimum = 1;
            nudModbusID.Maximum = 246;
            nudModbusID.Value = modbusID;
            this.modbusID = modbusID;
            this.device = device;

            cBoxSpeed.Items.Add("1200");
            cBoxSpeed.Items.Add("2400");
            cBoxSpeed.Items.Add("4800");
            cBoxSpeed.Items.Add("9600");
            cBoxSpeed.Items.Add("14400");
            cBoxSpeed.Items.Add("19200");
            cBoxSpeed.Items.Add("38400");
            cBoxSpeed.Items.Add("57600");
            cBoxSpeed.Items.Add("115200");
            cBoxSpeed.Text = port.BaudRate.ToString();

            labelModel.Text = device.name;
            labelFirmware.Text = "v???";

            if (!device.hasCleaningMode)
            {
                labelCleaning.Text = "Недоступна.";
                buttonCleaning.Enabled = false;
            }
            if (!device.hasBattery)
                labelBattery.Text = "Нет";
            else labelBattery.Text = "???%";

            Modbus.ResponseReceived += (sndr, msg) => { responseReceived = true; latestMessage = msg; isAwaitingResponse = false; };

            for (int i = 0; i < device.wiredSensors; i++)
            {
                WiredSensor ws = new WiredSensor(i,device) { Width = 495, Height = 24 };
                sensorPanel.Controls.Add(ws);
                ws.Visible = true;

                if (device.modelName == "PRPLS1")
                {
                    ws.wspPlusEntry = device.wspPlusMode[i];
                    ws.wspPlusCheckbox.CheckedChanged += (s, e) =>
                    {
                        try
                        {
                            ushort value = ws.wspPlusCheckbox.Checked ? (ushort)0xFF00 : (ushort)0x0000;
                            SetEntry(ws.wspPlusEntry, value);
                        }
                        catch (Exception err) { MessageBox.Show(err.Message, "WSP+ toggle error"); }                        
                    };
                }
            }
            if (device.hasScenarioSensor)
            {
                ScenarioSensor scenSen = new ScenarioSensor() { Width = 495, Height = 24 };
                sensorPanel.Controls.Add(scenSen);
                scenSen.Visible = true;
            }
            if (device.wiredSensors < device.sensorAlarm.length)
            {
                wirelessSensors = new List<WirelessSensor>();
                int wsrIndex = device.sensorAlarm.length - device.wiredSensors - (device.hasScenarioSensor ? 1 : 0);
                for (int i = 0; i < wsrIndex; i++)
                {
                    WirelessSensor wsr = new WirelessSensor(i) { Width = 495, Height = 24 };
                    sensorPanel.Controls.Add(wsr);
                    wsr.Visible = true;
                }
            }

            for (int i = 0; i < sensorPanel.Controls.Count; i++)
                sensorPanel.Controls[i].BackColor = i % 2 == 0 ? Color.White : Color.LightGray;


            sensorPanel.Update();
        }

        private async void buttonPoll_Click(object sender, EventArgs e)
        {
            if (isPolling)
                return;
            // hardcoded for now, probably easier to keep it like this in the future
            try
            {
                await Task.Run(() =>
                {

                    bool res = PollEntry(device.firmware);
                    Console.WriteLine("Polling for alarm status, poll success: " + res);
                    if (res)
                    {
                        this.Invoke(new MethodInvoker(delegate { labelFirmware.Text = App.ByteArrayToUnicode(latestMessage.Data); }));
                    }

                    if (device.hasBattery)
                    {
                        res = PollEntry(device.batteryCharge);
                        if (res)
                        {
                            this.Invoke(new MethodInvoker(delegate { labelBattery.Text = latestMessage.Data.Last().ToString(); }));
                        }
                    }

                    res = PollEntry(device.valveStatus);
                    Console.WriteLine("Polling for valve status, poll success: " + res);
                    if (res)
                    {
                        if (latestMessage.Data.Last() > 0)
                        {
                            isValveClosed = true;
                            this.Invoke(new MethodInvoker(delegate { labelValve.Text = "Закрыт"; }));
                            this.Invoke(new MethodInvoker(delegate { buttonValve.Text = "Открыть"; }));
                        }
                        else
                        {
                            isValveClosed = false;
                            this.Invoke(new MethodInvoker(delegate { labelValve.Text = "Открыт"; }));
                            this.Invoke(new MethodInvoker(delegate { buttonValve.Text = "Закрыть"; }));
                        }
                    }

                    res = PollEntry(device.alarmStatus);
                    Console.WriteLine("Polling for alarm status, poll success: " + res);
                    if (res)
                    {
                        Console.WriteLine("Alarm data: " + Modbus.ByteArrayToString(latestMessage.Data));
                        Console.WriteLine("Alarm data.last: " + latestMessage.Data.Last().ToString());
                        if (latestMessage.Data.Last() > 0)
                        {
                            alarmStatus = true;
                            this.Invoke(new MethodInvoker(delegate { buttonAlarm.Text = "Выключить"; }));
                            this.Invoke(new MethodInvoker(delegate { labelAlarm.Text = "Протечка!"; }));
                        }
                        else
                        {
                            alarmStatus = false;
                            this.Invoke(new MethodInvoker(delegate { buttonAlarm.Text = "Авария"; }));
                            this.Invoke(new MethodInvoker(delegate { labelAlarm.Text = "нет"; }));
                        }
                    }

                    if (device.hasCleaningMode)
                    {
                        res = PollEntry(device.cleaningMode);
                        if (res)
                        {
                            if (latestMessage.Data.Last() > 0)
                            {
                                cleaningStatus = true;
                                this.Invoke(new MethodInvoker(delegate { buttonCleaning.Text = "Выключить"; }));
                                this.Invoke(new MethodInvoker(delegate { labelCleaning.Text = "вкл"; }));
                            }
                            else
                            {
                                cleaningStatus = false;
                                this.Invoke(new MethodInvoker(delegate { buttonCleaning.Text = "Включить"; }));
                                this.Invoke(new MethodInvoker(delegate { labelCleaning.Text = "выкл"; }));
                            }
                        }
                    }

                    if (device.wspPlusMode != null || device.wspPlusMode.Count > 0)
                    {
                        for(int i = 0; i < device.wspPlusMode.Count; i++)
                        {
                            res = PollEntry(device.wspPlusMode[i]);
                            if (res)
                            {
                                bool value = latestMessage.Data[0] > 0x00 ? true : false;
                                WiredSensor snsr = sensorPanel.Controls[i] as WiredSensor;
                                snsr.Invoke(new MethodInvoker(delegate { snsr.wspPlusCheckbox.Checked = value; }));
                            }
                        }
                    }

                    if (device.wiredLineBreak != null || device.wiredLineBreak.Count > 0)
                    {
                        for (int i = 0; i < device.wiredLineBreak.Count; i++)
                        {
                            res = PollEntry(device.wiredLineBreak[i]);
                            if (res)
                            {
                                bool value = latestMessage.Data[0] > 0x00 ? true : false;
                                WiredSensor snsr = sensorPanel.Controls[i] as WiredSensor;
                                snsr.Invoke(new MethodInvoker(delegate { snsr.labelBreak.Text = value ? "Обрыв!" : "ОК"; }));

                            }
                        }
                    }

                    res = PollEntry(device.sensorAlarm);
                    if (res)
                    {
                        BitArray bArray = new BitArray(latestMessage.Data);
                        bool[] bools = new bool[bArray.Length];
                        bArray.CopyTo(bools, 0);
                        for (int i = 0; i < sensorPanel.Controls.Count; i++)
                        {
                            Sensor snsr = sensorPanel.Controls[i] as Sensor;
                            snsr.Invoke(new MethodInvoker(delegate { snsr.labelLeak.Text = bools[i] ? "Протечка!" : "нет"; }));
                        }
                    }

                    Console.WriteLine("Polling for radio status");
                    res = PollEntry(device.radioStatus);
                    if (res)
                    {
                        List<byte> values = new List<byte>(latestMessage.Data.Length / 2);
                        for (int i = 1; i < latestMessage.Data.Length; i += 2)
                            values.Add(latestMessage.Data[i]);
                        int add = device.wiredSensors + (device.hasScenarioSensor ? 1 : 0);
                        for (int i = 0; i < sensorPanel.Controls.Count - add; i++)
                        {

                            WirelessSensor snsr = sensorPanel.Controls[i + add] as WirelessSensor;
                            string txt = "нет";
                            switch (values[i])
                            {
                                case 1:
                                    txt = "норма";
                                    break;
                                case 2:
                                    txt = "протечка";
                                    break;
                                case 3:
                                    txt = "разряжен";
                                    break;
                                case 4:
                                    txt = "потеря";
                                    break;
                            }
                            snsr.Invoke(new MethodInvoker(delegate { snsr.labelStatus.Text = txt; }));
                        }
                    }
                });
            }
            catch (Exception err) { MessageBox.Show(err.Message); }
        }

        bool PollEntry(Entry entry)
        {
            latestMessage = null;
            bool res = false;
            isAwaitingResponse = true;
            isPolling = true;

            Modbus.ReadRegAsync(modbusID, (FunctionCode)entry.registerType, entry.address, entry.length);
            stopwatch.Restart();

            while (isAwaitingResponse && latestMessage == null)
            {
                if (stopwatch.ElapsedMilliseconds > 5000)
                {
                    Console.WriteLine("Response timed out.");
                    break;
                }
            }

            if (latestMessage != null && latestMessage.Status != ModbusStatus.Error)
                res = true;

            Console.WriteLine("Poll attempt finished");
            isPolling = false;
            return res;
        }
        bool SetEntry(Entry entry, ushort value)
        {
            latestMessage = null;
            bool res = false;
            isAwaitingResponse = true;
            isPolling = true;
            FunctionCode fc = FunctionCode.WriteCoil;
            if (entry.registerType == RegisterType.Holding)
                fc = FunctionCode.WriteRegister;

            Modbus.WriteSingleAsync(fc, modbusID, entry.address, value);
            stopwatch.Restart();

            while (isAwaitingResponse && latestMessage == null)
            {
                if (stopwatch.ElapsedMilliseconds > port.ReadTimeout)
                {
                    Console.WriteLine("Response timed out.");
                    break;
                }   
            }

            if (latestMessage != null && latestMessage.Status == ModbusStatus.WriteSuccess)
                res = true;
            isPolling = false;
            return res;
        }

        // Задать новый Slave ID для устройства
        private async void buttonSetID_Click(object sender, EventArgs e)
        {
            byte newID = (byte)nudModbusID.Value; // should prevent assigning wrong ID if UpDown is fiddled with in the middle of request 
            isAwaitingResponse = true;
            latestMessage = null;
            Modbus.WriteSingleAsync(FunctionCode.WriteRegister, modbusID, 128, newID);

            stopwatch.Restart();
            while (isAwaitingResponse)
            {
                if (stopwatch.ElapsedMilliseconds > 10000)
                {
                    Console.WriteLine("Response timed out.");
                    break;
                }
            }

            if (latestMessage != null && latestMessage.Status != ModbusStatus.Error)
                modbusID = newID;
        }

        // Кран
        private async void buttonValve_Click(object sender, EventArgs e)
        {
            try
            {
                await Task.Run(() =>
                {
                    ushort value = isValveClosed ? (ushort)0 : (ushort)0xFF00;
                    if (SetEntry(device.valveStatus, value))
                    {
                        isValveClosed = !isValveClosed;
                        labelValve.Invoke(new MethodInvoker(delegate { labelValve.Text = isValveClosed ? "Закрыт" : "Открыт"; }));
                        buttonValve.Invoke(new MethodInvoker(delegate { buttonValve.Text = isValveClosed ? "Открыть" : "Закрыть"; }));
                    }
                });
            }
            catch (Exception err)
            {
                MessageBox.Show(err.Message, "Valve Set Error");
            }
        }

        // Авария
        private void buttonAlarm_Click(object sender, EventArgs e)
        {
            ushort value = alarmStatus ? (ushort)0 : (ushort)0xFF00;
            if (SetEntry(device.alarmStatus, value))
            {
                alarmStatus = !alarmStatus;
                labelAlarm.Text = alarmStatus ? "Протечка!" : "Нет";
                buttonAlarm.Text = alarmStatus ? "Выключить" : "Авария";
            }
        }

        // Режим уборки
        private void buttonCleaning_Click(object sender, EventArgs e)
        {
            ushort value = cleaningStatus ? (ushort)0 : (ushort)0xFF00;
            if (SetEntry(device.cleaningMode, value))
            {
                cleaningStatus = !cleaningStatus;
                labelCleaning.Text = cleaningStatus ? "вкл" : "выкл";
                buttonCleaning.Text = cleaningStatus ? "Выключить" : "Включить";
            }
        }

        // Задать скорость передачи данных для устройства
        private void buttonSetSpeed_Click(object sender, EventArgs e)
        {
            try
            {
                string str = cBoxSpeed.Items[cBoxSpeed.SelectedIndex].ToString();
                str = str.Substring(0, str.Length - 2); //clip off two zeroes at the end
                ushort newSpeed = (ushort)Int16.Parse(str);
                Console.WriteLine("new speed: " + newSpeed);
                if (SetEntry(device.baudRate, newSpeed))
                {
                    //port.Close();
                    port.BaudRate = newSpeed * 100;
                    //port.Open();
                }
            }
            catch (Exception ex) { MessageBox.Show(ex.Message); }
        }

        // Выбор файла прошивки
        private void BrowseFirmware_Click(object sender, EventArgs e)
        {
            try
            {
                fileThread.SetApartmentState(ApartmentState.STA);
                fileThread.Start();
                while (!fileThread.IsAlive) { Thread.Sleep(1); }
                Thread.Sleep(1);
                fileThread.Join();
            }
            catch (Exception err) { MessageBox.Show(err.Message); }

            firmwarePathLabel.Invoke(new MethodInvoker(delegate { firmwarePathLabel.Text = firmwarePath; }));
        }

        // Запись прошивки
        private async void WriteFirmware_Click(object sender, EventArgs e)
        {
            if (firmwarePath is null || firmwarePath.Length == 0)
            {
                MessageBox.Show("Выберите файл прошивки.");
                return;
            }
            int cntr = 0;

            FileStream fileStream = File.OpenRead(firmwarePath);
            long bytesLeft = fileStream.Length;
            long bytesTotal = fileStream.Length;
            int count = 64;
            byte[] buffer = new byte[count];
            byte[] bdma;
            short _flashAddr = 0;
            byte[] flashAddr = new byte[2];
            byte[] CRC;
            List<byte> message;
            bool firstMessageSent = false;
            long bytesWritten = 0;
            await Task.Run(() =>
            {
                while (bytesLeft > 0)
                {
                    if (firstMessageSent && port.BaudRate != 9600) // after first message the device is sent into recovery mode which only supports 9600 bps
                    {
                        port.Close();
                        port.BaudRate = 9600;
                        port.Open();
                    }

                    count = bytesLeft > 64 ? 64 : (int)bytesLeft;
                    buffer = new byte[count];
                    fileStream.Read(buffer, 0, count);


                    bdma = new byte[2];
                    bdma[0] = (byte)((bytesLeft & 0xFF_00) >> 8);
                    bdma[1] = (byte)(bytesLeft & 0x00_FF);

                    flashAddr[0] = (byte)((_flashAddr & 0xFF_00) >> 8);
                    flashAddr[1] = (byte)(_flashAddr & 0x00_FF);

                    message = new List<byte>();
                    message.Add(modbusID);  // device ID
                    message.Add(0x10);      // function code
                    message.Add(0xFF);      // register address
                    message.Add(0xFF);      // register address
                    message.Add(0x00);      // regCnt (?)
                    message.Add(0x21);      // regCnt (?)
                    message.Add(0x42);      // data bytecount

                    message.Add(flashAddr[0]);
                    message.Add(flashAddr[1]);

                    try
                    {
                        for (int i = 0; i < buffer.Length; i++)
                        {
                            message.Add(buffer[i]);
                        }
                        message.Add(0x00);
                        message.Add(0x00);
                        CRC = new byte[2];
                        Modbus.GetCRC(message.ToArray(), ref CRC);
                        message[message.Count - 2] = CRC[0];
                        message[message.Count - 1] = CRC[1];

                        responseReceived = false;


                        while (true)
                        {
                            if (cntr > 3)
                            {
                                Console.WriteLine("Response timed out 4 times in a row, aborting. Check connection.");
                                return;
                            }
                            isAwaitingResponse = true;
                            latestMessage = null;
                            Console.WriteLine("Outgoing firmware message: " + Modbus.ByteArrayToString(message.ToArray()));
                            port.Write(message.ToArray(), 0, message.Count);
                            stopwatch.Restart();

                            while (isAwaitingResponse)
                            {
                                if (stopwatch.ElapsedMilliseconds > port.ReadTimeout)
                                {
                                    Console.WriteLine("Response timed out.");
                                    cntr++;
                                    break;
                                }
                            }

                            if (responseReceived)
                            {
                                if (latestMessage.Status == ModbusStatus.Error)
                                    Console.WriteLine("Response received: Error!");
                                else
                                {
                                    Console.WriteLine("Response received: all good;");
                                    break;
                                }
                            }
                        }
                        cntr = 0;
                        bytesLeft -= count;
                        bytesWritten += count;
                        firmwareProgressBar.Invoke((MethodInvoker)delegate { firmwareProgressBar.Increment((int)(bytesWritten / bytesTotal) * 100); });

                        _flashAddr += (short)count;
                        if (port.BaudRate != 9600)
                            firstMessageSent = true;

                        if (bytesLeft <= 0)
                            Console.WriteLine("Reached the end of firmware file.");
                    }
                    catch (Exception ex)
                    {
                        MessageBox.Show(ex.Message, "Firmware writing error");
                        return;
                    }

                }

                /* Final Message */
                message = new List<byte>();
                message.Add(modbusID);  // device ID
                message.Add(0x10);      // function code
                message.Add(0xFF);      // register address
                message.Add(0xFF);      // register address
                message.Add(0x00);      // regCnt (?)
                message.Add(0x21);      // regCnt (?)
                message.Add(0x00);      // data bytecount
                message.Add(0x00);      // CRC
                message.Add(0x00);      // CRC
                CRC = new byte[2];
                Modbus.GetCRC(message.ToArray(), ref CRC);
                message[message.Count - 2] = CRC[0];
                message[message.Count - 1] = CRC[1];
                while (true)
                {
                    isAwaitingResponse = true;
                    Console.WriteLine("Outgoing firmware message: " + Modbus.ByteArrayToString(message.ToArray()));
                    port.Write(message.ToArray(), 0, message.Count);
                    stopwatch.Restart();

                    while (isAwaitingResponse)
                    {
                        if (stopwatch.ElapsedMilliseconds > 1000)
                        {
                            Console.WriteLine("Response timed out.");
                            break;
                        }
                    }

                    if (responseReceived)
                    {
                        if (latestMessage.Status == ModbusStatus.Error)
                            Console.WriteLine("Response received: Error!");
                        else
                        {
                            Console.WriteLine("Response received: all good;");
                            break;
                        }
                    }
                }
            });

        }
    }

#region Sensor Classes
    public class Sensor : FlowLayoutPanel
    {
        public Label labelName = new Label() { Width = 60, Height = 24 };
        public Label labelLeakFluff = new Label() { Width = 60, Height = 24 };
        public Label labelLeak = new Label() { Width = 50, Height = 24 };
    }
    public class WiredSensor : Sensor
    {
        public Label labelBreakFluff = new Label() { Width = 45, Height = 24 };
        public Label labelBreak = new Label() { Width = 55, Height = 24 }; // обрыв линии для WSP+

        public Label labelWSPPlusFluff;
        public CheckBox wspPlusCheckbox;
        public Entry wspPlusEntry; // for WSP+ control
        public WiredSensor(int count, Device device)
        {
            this.Margin = Padding.Empty;
            this.Padding = new Padding(0, 5, 0, 0);
            this.WrapContents = false;
            this.BackColor = Color.White;
            this.Height = 15;
            this.FlowDirection = FlowDirection.LeftToRight;
            this.Controls.Add(labelName);

            this.Controls.Add(labelBreakFluff);
            this.Controls.Add(labelBreak);

            this.Controls.Add(labelLeakFluff);
            this.Controls.Add(labelLeak);

            labelName.Text = "WSP " + (count + 1);

            labelLeakFluff.Text = "Протечка:";
            labelLeak.Text = "неизвестно";

            labelBreakFluff.Text = "Обрыв:";
            labelBreak.Text = "неизвестно";
            if (device.wspPlusMode != null || device.wspPlusMode.Count > 0)
            {
                labelWSPPlusFluff = new Label() { Width = 45, Height = 24, Text = "WSP+:" };
                wspPlusCheckbox = new CheckBox() { Width = 20, Height = 14, Margin = Padding.Empty };

                this.Controls.Add(labelWSPPlusFluff);
                this.Controls.Add(wspPlusCheckbox);
            }

        }
    }

    public class WirelessSensor : Sensor
    {

        public Label labelStatusFluff = new Label() { Width = 45, Height = 24 };
        public Label labelStatus = new Label() { Width = 55, Height = 24 };

        public WirelessSensor(int count)
        {
            this.Margin = Padding.Empty;
            this.Padding = new Padding(0, 5, 0, 0);
            this.BackColor = Color.White;
            this.FlowDirection = FlowDirection.LeftToRight;
            this.WrapContents = false;

            this.Controls.Add(labelName);
            this.Controls.Add(labelStatusFluff);
            this.Controls.Add(labelStatus);
            this.Controls.Add(labelLeakFluff);
            this.Controls.Add(labelLeak);

            labelName.Text = "WSR " + (count + 1);

            labelLeakFluff.Text = "Протечка:";
            labelLeak.Text = "неизвестно";

            labelStatusFluff.Text = "Статус:";
            labelStatus.Text = "неизвестно";
        }
    }

    public class ScenarioSensor : Sensor
    {
        public ScenarioSensor()
        {
            labelName.Width = 172;
            this.Margin = Padding.Empty;
            this.Padding = new Padding(0, 5, 0, 0);
            this.FlowDirection = FlowDirection.LeftToRight;
            this.WrapContents = false;

            this.Controls.Add(labelName);
            this.Controls.Add(labelLeakFluff);
            this.Controls.Add(labelLeak);

            labelName.Text = "Сценарный датчик";

            labelLeakFluff.Text = "Протечка:";
            labelLeak.Text = "неизвестно";
        }
    }
}
#endregion