/*

  This Source Code Form is subject to the terms of the Mozilla Public

  License, v. 2.0. If a copy of the MPL was not distributed with this

  file, You can obtain one at http://mozilla.org/MPL/2.0/.

  Copyright (C) 2010-2011 Michael Möller <mmoeller@openhardwaremonitor.org>

*/

using System;

using System.Globalization;

using System.IO;

using System.IO.Ports;

using System.Text;

using System.Text.RegularExpressions;

using System.Threading;

namespace OpenHardwareMonitor.Hardware.Heatmaster {

  internal class Heatmaster : Hardware, IDisposable {

    private readonly string portName;

    private SerialPort serialPort;

    private readonly int hardwareRevision;

    private readonly int firmwareRevision;

    private readonly int firmwareCRC;

    private readonly Sensor[] fans;

    private readonly Sensor[] controls;

    private readonly Sensor[] temperatures;

    private readonly Sensor[] flows;

    private readonly Sensor[] relays;

    private readonly bool available;

    private readonly StringBuilder buffer = new StringBuilder();

    private string ReadLine(int timeout) {

      int i = 0;

      StringBuilder builder = new StringBuilder();

      while (i <= timeout) {

        while (serialPort.BytesToRead > 0) {

          byte b = (byte)serialPort.ReadByte();

          switch (b) {

            case 0xAA: return ((char)b).ToString();

            case 0x0D: return builder.ToString();

            default: builder.Append((char)b); break;

          }

        }

        i++;

        Thread.Sleep(1);

      }

      throw new TimeoutException();

    }

    private string ReadField(int device, char field) {

      serialPort.WriteLine("[0:" + device + "]R" + field);

      for (int i = 0; i < 5; i++) {

        string s = ReadLine(200);

        Match match = Regex.Match(s, @"-\[0:" + 

          device.ToString(CultureInfo.InvariantCulture) + @"\]R" +

          Regex.Escape(field.ToString(CultureInfo.InvariantCulture)) + ":(.*)");

        if (match.Success) 

          return match.Groups[1].Value;

      }

      return null;

    }

    protected string ReadString(int device, char field) {

      string s = ReadField(device, field);

      if (s != null && s[0] == '"' && s[s.Length - 1] == '"')

        return s.Substring(1, s.Length - 2);

      else

        return null;

    }

    protected int ReadInteger(int device, char field) {

      string s = ReadField(device, field);      

      int i;

      if (int.TryParse(s, out i))

        return i;

      else

        return 0;

    }

    private bool WriteField(int device, char field, string value) {

      serialPort.WriteLine("[0:" + device + "]W" + field + ":" + value);

      for (int i = 0; i < 5; i++) {

        string s = ReadLine(200);

        Match match = Regex.Match(s, @"-\[0:" + 

          device.ToString(CultureInfo.InvariantCulture) + @"\]W" + 

          Regex.Escape(field.ToString(CultureInfo.InvariantCulture)) +

          ":" + value);

        if (match.Success)

          return true;

      }

      return false;

    }

    protected bool WriteInteger(int device, char field, int value) {

      return WriteField(device, field, 

        value.ToString(CultureInfo.InvariantCulture));

    }

    protected bool WriteString(int device, char field, string value) {

      return WriteField(device, field, '"' + value + '"');

    }

    public Heatmaster(string portName, ISettings settings) 

      : base("Heatmaster", new Identifier("heatmaster",

        portName.TrimStart(new [] {'/'}).ToLowerInvariant()), settings)

    {

      this.portName = portName;

      try {

        serialPort = new SerialPort(portName, 38400, Parity.None, 8,

          StopBits.One);

        serialPort.Open();

        serialPort.NewLine = ((char)0x0D).ToString();

        hardwareRevision = ReadInteger(0, 'H');

        firmwareRevision = ReadInteger(0, 'V');

        firmwareCRC = ReadInteger(0, 'C');

        int fanCount = Math.Min(ReadInteger(32, '?'), 4);

        int temperatureCount = Math.Min(ReadInteger(48, '?'), 6);

        int flowCount = Math.Min(ReadInteger(64, '?'), 1);

        int relayCount =  Math.Min(ReadInteger(80, '?'), 1);

        fans = new Sensor[fanCount];

        controls = new Sensor[fanCount];

        for (int i = 0; i < fanCount; i++) {

          int device = 33 + i;

          string name = ReadString(device, 'C');

          fans[i] = new Sensor(name, device, SensorType.Fan, this, settings);          

          fans[i].Value = ReadInteger(device, 'R');

          ActivateSensor(fans[i]);

          controls[i] =

            new Sensor(name, device, SensorType.Control, this, settings);

          controls[i].Value = (100 / 255.0f) * ReadInteger(device, 'P');

          ActivateSensor(controls[i]);

        }       

        temperatures = new Sensor[temperatureCount];

        for (int i = 0; i < temperatureCount; i++) {

          int device = 49 + i;

          string name = ReadString(device, 'C');

          temperatures[i] =

            new Sensor(name, device, SensorType.Temperature, this, settings);

          int value = ReadInteger(device, 'T');

          temperatures[i].Value = 0.1f * value;

          if (value != -32768)

            ActivateSensor(temperatures[i]);

        }

        flows = new Sensor[flowCount];

        for (int i = 0; i < flowCount; i++) {

          int device = 65 + i;

          string name = ReadString(device, 'C');

          flows[i] = new Sensor(name, device, SensorType.Flow, this, settings);

          flows[i].Value = 0.1f * ReadInteger(device, 'L');

          ActivateSensor(flows[i]);

        }

        relays = new Sensor[relayCount];

        for (int i = 0; i < relayCount; i++) {

          int device = 81 + i;

          string name = ReadString(device, 'C');

          relays[i] = 

            new Sensor(name, device, SensorType.Control, this, settings);

          relays[i].Value = 100 * ReadInteger(device, 'S');

          ActivateSensor(relays[i]);

        }

        // set the update rate to 2 Hz

        WriteInteger(0, 'L', 2);

        available = true;

      } catch (IOException) { } catch (TimeoutException) { }      

    }

    public override HardwareType HardwareType {

      get { return HardwareType.Heatmaster; }

    }

    private void ProcessUpdateLine(string line) {

      Match match = Regex.Match(line, @">\[0:(\d+)\]([0-9:\|-]+)");

      if (match.Success) {

        int device;

        if (int.TryParse(match.Groups[1].Value, out device)) {

          foreach (string s in match.Groups[2].Value.Split('|')) {

            string[] strings = s.Split(':');

            int[] ints = new int[strings.Length];            

            bool valid = true;

            for (int i = 0; i < ints.Length; i++)

              if (!int.TryParse(strings[i], out ints[i])) {

                valid = false;

                break;

              }

            if (!valid)

              continue; 

            switch (device) {

              case 32:

                if (ints.Length == 3 && ints[0] <= fans.Length) {

                  fans[ints[0] - 1].Value = ints[1];

                  controls[ints[0] - 1].Value = (100 / 255.0f) * ints[2];

                }

                break;

              case 48:

                if (ints.Length == 2 && ints[0] <= temperatures.Length)

                  temperatures[ints[0] - 1].Value = 0.1f * ints[1];

                break;

              case 64:

                if (ints.Length == 3 && ints[0] <= flows.Length)

                  flows[ints[0] - 1].Value = 0.1f * ints[1];

                break;

              case 80:

                if (ints.Length == 2 && ints[0] <= relays.Length)

                  relays[ints[0] - 1].Value = 100 * ints[1];

                break;

            }

          }

        }

      }

    }

    public override void Update() {

      if (!available)

        return;

      while (serialPort.IsOpen &&  serialPort.BytesToRead > 0) {

        byte b = (byte)serialPort.ReadByte();

        if (b == 0x0D) {

          ProcessUpdateLine(buffer.ToString());

          buffer.Length = 0;

        } else {

          buffer.Append((char)b);

        }

      }

    }

    public override string GetReport() {

      StringBuilder r = new StringBuilder();

      r.AppendLine("Heatmaster");

      r.AppendLine();

      r.Append("Port: ");

      r.AppendLine(portName);

      r.Append("Hardware Revision: ");

      r.AppendLine(hardwareRevision.ToString(CultureInfo.InvariantCulture));

      r.Append("Firmware Revision: ");

      r.AppendLine(firmwareRevision.ToString(CultureInfo.InvariantCulture));

      r.Append("Firmware CRC: ");

      r.AppendLine(firmwareCRC.ToString(CultureInfo.InvariantCulture));

      r.AppendLine();

      return r.ToString();

    }

    public override void Close() {

      serialPort.Close();

      serialPort.Dispose();

      serialPort = null;

      base.Close();

    }

    public void Dispose() {

      if (serialPort != null) {

        serialPort.Dispose();

        serialPort = null;

      }

    }

  }

}