/*

  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) 2009-2010 Michael Möller <mmoeller@openhardwaremonitor.org>

*/

using System;

using System.Collections.Generic;

using System.Globalization;

using System.Text;

using System.Threading;

namespace OpenHardwareMonitor.Hardware.TBalancer {

  internal class TBalancerGroup : IGroup {

    private readonly List<TBalancer> hardware = new List<TBalancer>();

    private readonly StringBuilder report = new StringBuilder();

    public TBalancerGroup(ISettings settings) {

      uint numDevices;

      try {

        if (FTD2XX.FT_CreateDeviceInfoList(out numDevices) != FT_STATUS.FT_OK) {

          report.AppendLine("Status: FT_CreateDeviceInfoList failed");

          return;

        }

      } catch (DllNotFoundException) { return; } 

        catch (ArgumentNullException) { return; }

        catch (EntryPointNotFoundException) { return; }

        catch (BadImageFormatException) { return; }

      FT_DEVICE_INFO_NODE[] info = new FT_DEVICE_INFO_NODE[numDevices];

      if (FTD2XX.FT_GetDeviceInfoList(info, ref numDevices) != FT_STATUS.FT_OK) 

      {

        report.AppendLine("Status: FT_GetDeviceInfoList failed");        

        return;

      }

      // make sure numDevices is not larger than the info array

      if (numDevices > info.Length)

        numDevices = (uint)info.Length;

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

        report.Append("Device Index: ");

        report.AppendLine(i.ToString(CultureInfo.InvariantCulture));

        report.Append("Device Type: ");

        report.AppendLine(info[i].Type.ToString());

        // the T-Balancer always uses an FT232BM

        if (info[i].Type != FT_DEVICE.FT_DEVICE_232BM) {

          report.AppendLine("Status: Wrong device type");

          continue;

        }

        FT_HANDLE handle;

        FT_STATUS status = FTD2XX.FT_Open(i, out handle);

        if (status != FT_STATUS.FT_OK) {

          report.AppendLine("Open Status: " + status);

          continue;

        }

        FTD2XX.FT_SetBaudRate(handle, 19200);

        FTD2XX.FT_SetDataCharacteristics(handle, 8, 1, 0);

        FTD2XX.FT_SetFlowControl(handle, FT_FLOW_CONTROL.FT_FLOW_RTS_CTS, 0x11, 

          0x13);

        FTD2XX.FT_SetTimeouts(handle, 1000, 1000);

        FTD2XX.FT_Purge(handle, FT_PURGE.FT_PURGE_ALL);

        status = FTD2XX.Write(handle, new byte[] { 0x38 });

        if (status != FT_STATUS.FT_OK) {

          report.AppendLine("Write Status: " + status);

          FTD2XX.FT_Close(handle);

          continue;

        }

        bool isValid = false;

        byte protocolVersion = 0;

        int j = 0;

        while (FTD2XX.BytesToRead(handle) == 0 && j < 2) {

          Thread.Sleep(100);

          j++;

        }

        if (FTD2XX.BytesToRead(handle) > 0) {

          if (FTD2XX.ReadByte(handle) == TBalancer.STARTFLAG) {

            while (FTD2XX.BytesToRead(handle) < 284 && j < 5) {

              Thread.Sleep(100);

              j++;

            }

            int length = FTD2XX.BytesToRead(handle);

            if (length >= 284) {

              byte[] data = new byte[285];

              data[0] = TBalancer.STARTFLAG;

              for (int k = 1; k < data.Length; k++)

                data[k] = FTD2XX.ReadByte(handle);

              // check protocol version 2X (protocols seen: 2C, 2A, 28)

              isValid = (data[274] & 0xF0) == 0x20;

              protocolVersion = data[274];

              if (!isValid) {

                report.Append("Status: Wrong Protocol Version: 0x");

                report.AppendLine(

                  protocolVersion.ToString("X", CultureInfo.InvariantCulture));

              }

            } else {

              report.AppendLine("Status: Wrong Message Length: " + length);

            }

          } else {

            report.AppendLine("Status: Wrong Startflag");

          }

        } else {

          report.AppendLine("Status: No Response");

        }

        FTD2XX.FT_Purge(handle, FT_PURGE.FT_PURGE_ALL);

        FTD2XX.FT_Close(handle);

        if (isValid) {

          report.AppendLine("Status: OK");

          hardware.Add(new TBalancer(i, protocolVersion, settings));

          return;

        }

        report.AppendLine();

      }

    }

    public IHardware[] Hardware {

      get {

        return hardware.ToArray();

      }

    }

    public string GetReport() {

      if (report.Length > 0) {

        StringBuilder r = new StringBuilder();

        r.AppendLine("FTD2XX");

        r.AppendLine();

        r.Append(report);

        r.AppendLine();

        return r.ToString();

      } else

        return null;

    }

    public void Close() {

      foreach (TBalancer tbalancer in hardware)

        tbalancer.Close();

    }

  }

}