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@258:   Portions created by the Initial Developer are Copyright (C) 2009-2011
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@275:   internal class HDD : Hardware {
moel@1: 
moel@1:     private const int UPDATE_DIVIDER = 30; // update only every 30s
moel@1: 
moel@195:     private readonly IntPtr handle;
moel@195:     private readonly int drive;
moel@1:     private int count;
moel@1: 
moel@231:     private readonly SMART.AttributeID temperatureID = SMART.AttributeID.None;
moel@231:     private readonly SMART.AttributeID lifeID = SMART.AttributeID.None;
paulwerelds@218: 
paulwerelds@218:     private readonly Sensor temperatureSensor;
paulwerelds@218:     private readonly Sensor lifeSensor;
paulwerelds@218: 
moel@231:     public HDD(string name, IntPtr handle, int drive, 
moel@231:       SMART.AttributeID temperatureID, SMART.AttributeID lifeID, 
moel@275:       ISettings settings) : base(name, new Identifier("hdd", 
moel@275:           drive.ToString(CultureInfo.InvariantCulture)), settings)
paulwerelds@218:     {
paulwerelds@218:       this.handle = handle;
paulwerelds@218:       this.drive = drive;
paulwerelds@218:       this.count = 0;
moel@231:       if (temperatureID != SMART.AttributeID.None) {
moel@231:         this.temperatureID = temperatureID;
moel@231:         this.temperatureSensor = new Sensor("HDD", 0, SensorType.Temperature,
moel@231:           this, settings);
moel@231:       }
moel@231: 
moel@231:       if (lifeID != SMART.AttributeID.None) {
moel@231:         this.lifeID = lifeID;
moel@231:         this.lifeSensor = new Sensor("Remaining life", 0, SensorType.Level,
moel@231:           this, settings);
moel@231:       }
paulwerelds@218: 
paulwerelds@218:       Update();
paulwerelds@218:     }
moel@1: 
moel@275:     public override HardwareType HardwareType {
moel@165:       get { return HardwareType.HDD; }
moel@1:     }
moel@1: 
moel@275:     public override ISensor[] Sensors {
moel@1:       get {
moel@231:         if (lifeID != SMART.AttributeID.None)
paulwerelds@218:           return new ISensor[] { lifeSensor };
paulwerelds@218: 
moel@231:         if (temperatureID != SMART.AttributeID.None)
paulwerelds@218:           return new ISensor[] { temperatureSensor };
paulwerelds@218: 
paulwerelds@218:         return new ISensor[] {};
moel@1:       }
moel@1:     }
moel@1: 
moel@275:     public override void Update() {
moel@1:       if (count == 0) {
paulwerelds@233:         SMART.DriveAttribute[] attributes = SMART.ReadSmart(handle, drive);
paulwerelds@233: 
moel@231:         if (temperatureID != SMART.AttributeID.None &&
paulwerelds@233:           Array.Exists(attributes, attr => attr.ID == temperatureID))
paulwerelds@218:         {
paulwerelds@233:           temperatureSensor.Value = Array
paulwerelds@233:             .Find(attributes, attr => attr.ID == temperatureID)
paulwerelds@218:             .RawValue[0];
paulwerelds@218:         }
paulwerelds@218: 
moel@231:         if (lifeID != SMART.AttributeID.None &&
paulwerelds@233:           Array.Exists(attributes, attr => attr.ID == lifeID))
paulwerelds@218:         {
paulwerelds@233:           lifeSensor.Value = Array
paulwerelds@233:             .Find(attributes, attr => attr.ID == lifeID)
paulwerelds@218:             .AttrValue;
paulwerelds@218:         }
moel@1:       } else {
paulwerelds@233:         if (temperatureID != SMART.AttributeID.None)
paulwerelds@218:           temperatureSensor.Value = temperatureSensor.Value;
paulwerelds@218: 
paulwerelds@233:         if (lifeID != SMART.AttributeID.None)
paulwerelds@218:           lifeSensor.Value = lifeSensor.Value;
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@275:     public override void Traverse(IVisitor visitor) {
moel@258:       foreach (ISensor sensor in Sensors)
moel@258:         sensor.Accept(visitor);
moel@258:     }
moel@1:   }
moel@1: }