/*

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

	Copyright (C) 2010 Paul Werelds

*/

namespace OpenHardwareMonitor.Hardware.HDD {

  using System.Collections.Generic;

  [NamePrefix(""), RequireSmart(0xAB), RequireSmart(0xB1)]

  internal class SSDSandforce : AbstractHarddrive {

    private static readonly IEnumerable<SmartAttribute> smartAttributes =

      new List<SmartAttribute> {

      new SmartAttribute(0x01, SmartNames.RawReadErrorRate),

      new SmartAttribute(0x05, SmartNames.RetiredBlockCount, RawToInt),

      new SmartAttribute(0x09, SmartNames.PowerOnHours, RawToInt),

      new SmartAttribute(0x0C, SmartNames.PowerCycleCount, RawToInt),

      new SmartAttribute(0xAB, SmartNames.ProgramFailCount, RawToInt),

      new SmartAttribute(0xAC, SmartNames.EraseFailCount, RawToInt),

      new SmartAttribute(0xAE, SmartNames.UnexpectedPowerLossCount, RawToInt),

      new SmartAttribute(0xB1, SmartNames.WearRangeDelta, RawToInt),

      new SmartAttribute(0xB5, SmartNames.AlternativeProgramFailCount, RawToInt),

      new SmartAttribute(0xB6, SmartNames.AlternativeEraseFailCount, RawToInt),

      new SmartAttribute(0xBB, SmartNames.UncorrectableErrorCount, RawToInt),

      new SmartAttribute(0xC2, SmartNames.Temperature, (byte[] raw, byte value) 

        => { return value; }, SensorType.Temperature, 0, true), 

      new SmartAttribute(0xC3, SmartNames.UnrecoverableEcc), 

      new SmartAttribute(0xC4, SmartNames.ReallocationEventCount, RawToInt),

      new SmartAttribute(0xE7, SmartNames.RemainingLife, null, 

        SensorType.Level, 0),

      new SmartAttribute(0xE9, SmartNames.ControllerWritesToNAND, RawToInt,

        SensorType.Data, 0),

      new SmartAttribute(0xEA, SmartNames.HostWritesToController, RawToInt, 

        SensorType.Data, 1),

      new SmartAttribute(0xF1, SmartNames.HostWrites, RawToInt, 

        SensorType.Data, 1),

      new SmartAttribute(0xF2, SmartNames.HostReads, RawToInt, 

        SensorType.Data, 2)

    };

    private Sensor writeAmplification;

    public SSDSandforce(ISmart smart, string name, string firmwareRevision, 

      int index, ISettings settings) 

      : base(smart, name, firmwareRevision,  index, smartAttributes, settings) 

    {

      this.writeAmplification = new Sensor("Write Amplification", 1, 

        SensorType.Factor, this, settings);    

    }

    public override void UpdateAdditionalSensors(DriveAttributeValue[] values) {

      float? controllerWritesToNAND = null;

      float? hostWritesToController = null;

      foreach (DriveAttributeValue value in values) {

        if (value.Identifier == 0xE9)

          controllerWritesToNAND = RawToInt(value.RawValue, value.AttrValue);

        if (value.Identifier == 0xEA)

          hostWritesToController = RawToInt(value.RawValue, value.AttrValue);

      }

      if (controllerWritesToNAND.HasValue && hostWritesToController.HasValue) {

        if (hostWritesToController.Value > 0)

          writeAmplification.Value =

            controllerWritesToNAND.Value / hostWritesToController.Value;

        else

          writeAmplification.Value = 0;

        ActivateSensor(writeAmplification);

      }

    }

  }

}