moel@1: /*
moel@1:   
moel@1:   Version: MPL 1.1/GPL 2.0/LGPL 2.1
moel@1: 
moel@1:   The contents of this file are subject to the Mozilla Public License Version
moel@1:   1.1 (the "License"); you may not use this file except in compliance with
moel@1:   the License. You may obtain a copy of the License at
moel@1:  
moel@1:   http://www.mozilla.org/MPL/
moel@1: 
moel@1:   Software distributed under the License is distributed on an "AS IS" basis,
moel@1:   WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
moel@1:   for the specific language governing rights and limitations under the License.
moel@1: 
moel@1:   The Original Code is the Open Hardware Monitor code.
moel@1: 
moel@1:   The Initial Developer of the Original Code is 
moel@1:   Michael Möller <m.moeller@gmx.ch>.
moel@1:   Portions created by the Initial Developer are Copyright (C) 2009-2010
moel@1:   the Initial Developer. All Rights Reserved.
moel@1: 
paulwerelds@218:   Contributor(s): Paul Werelds
moel@1: 
moel@1:   Alternatively, the contents of this file may be used under the terms of
moel@1:   either the GNU General Public License Version 2 or later (the "GPL"), or
moel@1:   the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
moel@1:   in which case the provisions of the GPL or the LGPL are applicable instead
moel@1:   of those above. If you wish to allow use of your version of this file only
moel@1:   under the terms of either the GPL or the LGPL, and not to allow others to
moel@1:   use your version of this file under the terms of the MPL, indicate your
moel@1:   decision by deleting the provisions above and replace them with the notice
moel@1:   and other provisions required by the GPL or the LGPL. If you do not delete
moel@1:   the provisions above, a recipient may use your version of this file under
moel@1:   the terms of any one of the MPL, the GPL or the LGPL.
moel@1:  
moel@1: */
moel@1: 
moel@1: using System;
paulwerelds@218: using System.Collections.Generic;
moel@166: using System.Globalization;
moel@1: 
moel@1: namespace OpenHardwareMonitor.Hardware.HDD {
moel@165:   internal class HDD : IHardware {
moel@1: 
moel@1:     private const int UPDATE_DIVIDER = 30; // update only every 30s
moel@1: 
moel@195:     private readonly string name;
moel@195:     private readonly IntPtr handle;
moel@195:     private readonly int drive;
moel@1:     private int count;
moel@1: 
paulwerelds@218:     private readonly SMART.AttributeID temperatureID = 0x00;    
paulwerelds@218:     private readonly SMART.SSDLifeID lifeID = 0x00;
paulwerelds@218: 
paulwerelds@218:     private readonly Sensor temperatureSensor;
paulwerelds@218:     private readonly Sensor lifeSensor;
paulwerelds@218: 
paulwerelds@218:     public HDD(string name, IntPtr handle, int drive,
paulwerelds@218:       SMART.AttributeID temperatureID, ISettings settings)
moel@165:     {
moel@1:       this.name = name;
moel@1:       this.handle = handle;
moel@1:       this.drive = drive;
moel@1:       this.count = 0;
paulwerelds@218:       this.temperatureID = temperatureID;
paulwerelds@218:       this.temperatureSensor = new Sensor("HDD", 0, SensorType.Temperature,
paulwerelds@218:         this, settings);
paulwerelds@218: 
moel@1:       Update();
moel@1:     }
moel@1: 
paulwerelds@218:     public HDD(string name, IntPtr handle, int drive, SMART.SSDLifeID lifeID,
paulwerelds@218:       ISettings settings)
paulwerelds@218:     {
paulwerelds@218:       this.name = name;
paulwerelds@218:       this.handle = handle;
paulwerelds@218:       this.drive = drive;
paulwerelds@218:       this.count = 0;
paulwerelds@218:       this.lifeID = lifeID;
paulwerelds@218:       this.lifeSensor = new Sensor("HDD", 0, SensorType.Level, this, settings);
paulwerelds@218: 
paulwerelds@218:       Update();
paulwerelds@218:     }
moel@1: 
moel@1:     public string Name {
moel@1:       get { return name; }
moel@1:     }
moel@1: 
moel@109:     public Identifier Identifier {
moel@166:       get { 
moel@166:         return new Identifier("hdd", 
moel@166:           drive.ToString(CultureInfo.InvariantCulture)); 
moel@166:       }
moel@1:     }
moel@1: 
moel@165:     public HardwareType HardwareType {
moel@165:       get { return HardwareType.HDD; }
moel@1:     }
moel@1: 
moel@64:     public IHardware[] SubHardware {
moel@64:       get { return new IHardware[0]; }
moel@64:     }
moel@64: 
moel@176:     public virtual IHardware Parent {
moel@176:       get { return null; }
moel@176:     }
moel@176: 
moel@1:     public ISensor[] Sensors {
moel@1:       get {
paulwerelds@218:         if (lifeID != SMART.SSDLifeID.None)
paulwerelds@218:           return new ISensor[] { lifeSensor };
paulwerelds@218: 
paulwerelds@218:         if (temperatureID != 0x00)
paulwerelds@218:           return new ISensor[] { temperatureSensor };
paulwerelds@218: 
paulwerelds@218:         return new ISensor[] {};
moel@1:       }
moel@1:     }
moel@1: 
moel@1:     public string GetReport() {
moel@1:       return null;
moel@1:     }
moel@1: 
moel@1:     public void Update() {
moel@1:       if (count == 0) {
paulwerelds@218:         List<SMART.DriveAttribute> attributes = SMART.ReadSmart(handle, drive);
paulwerelds@218:         if (temperatureID != 0x00 &&
paulwerelds@218:           attributes.Exists(attr => (int)attr.ID == (int)temperatureID))
paulwerelds@218:         {
paulwerelds@218:           temperatureSensor.Value = attributes
paulwerelds@218:             .Find(attr => (int)attr.ID == (int)temperatureID)
paulwerelds@218:             .RawValue[0];
paulwerelds@218:         }
paulwerelds@218: 
paulwerelds@218:         if (lifeID != 0x00 &&
paulwerelds@218:           attributes.Exists(attr => (int)attr.ID == (int)lifeID))
paulwerelds@218:         {
paulwerelds@218:           lifeSensor.Value = attributes
paulwerelds@218:             .Find(attr => (int)attr.ID == (int)temperatureID)
paulwerelds@218:             .AttrValue;
paulwerelds@218:         }
moel@1:       } else {
paulwerelds@218:         if (temperatureID != 0x00) {
paulwerelds@218:           temperatureSensor.Value = temperatureSensor.Value;
paulwerelds@218:         }
paulwerelds@218: 
paulwerelds@218:         if (lifeID != 0x00) {
paulwerelds@218:           lifeSensor.Value = lifeSensor.Value;
paulwerelds@218:         }
moel@1:       }
moel@1: 
moel@1:       count++; count %= UPDATE_DIVIDER; 
moel@1:     }
moel@1: 
moel@1:     public void Close() {
moel@1:       SMART.CloseHandle(handle);
moel@1:     }
moel@1: 
moel@1:     #pragma warning disable 67
moel@1:     public event SensorEventHandler SensorAdded;
moel@1:     public event SensorEventHandler SensorRemoved;
moel@110:     #pragma warning restore 67    
moel@110: 
moel@110:     public void Accept(IVisitor visitor) {
moel@167:       if (visitor == null)
moel@167:         throw new ArgumentNullException("visitor");
moel@167:       visitor.VisitHardware(this);
moel@110:     }
moel@110: 
moel@110:     public void Traverse(IVisitor visitor) { }
moel@1:   }
moel@1: }