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: 
moel@1:   Contributor(s):
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;
moel@1: using System.Collections.Generic;
moel@1: using System.Drawing;
moel@1: using Aga.Controls.Tree;
moel@1: using OpenHardwareMonitor.Hardware;
moel@1: 
moel@1: namespace OpenHardwareMonitor.GUI {
moel@1:   public class HardwareNode : Node {
moel@1: 
moel@1:     private IHardware hardware;
moel@1: 
moel@1:     private List<TypeNode> typeNodes = new List<TypeNode>();
moel@1: 
moel@1:     public HardwareNode(IHardware hardware) : base(hardware.Name) {
moel@1:       
moel@1:       this.hardware = hardware;
moel@1:       this.Image = hardware.Icon;
moel@1: 
moel@1:       typeNodes.Add(new TypeNode(SensorType.Voltage));
moel@1:       typeNodes.Add(new TypeNode(SensorType.Clock));
moel@1:       typeNodes.Add(new TypeNode(SensorType.Temperature));
moel@1:       typeNodes.Add(new TypeNode(SensorType.Fan));
moel@1:       
moel@1:       foreach (ISensor sensor in hardware.Sensors)
moel@1:         SensorAdded(sensor);
moel@1: 
moel@1:       hardware.SensorAdded +=new SensorEventHandler(SensorAdded);
moel@1:       hardware.SensorRemoved += new SensorEventHandler(SensorRemoved);
moel@1:     }
moel@1: 
moel@1:     public IHardware Hardware {
moel@1:       get { return hardware; }
moel@1:     }
moel@1: 
moel@1:     public void SetVisible(SensorType sensorType, bool visible) {
moel@1:       foreach (TypeNode node in typeNodes)
moel@1:         if (node.SensorType == sensorType) {
moel@1:           node.IsVisible = visible;
moel@1:           UpdateNode(node);
moel@1:         }
moel@1:     }
moel@1: 
moel@1:     private void UpdateNode(TypeNode node) {      
moel@1:       if (node.IsVisible && node.Nodes.Count > 0) {
moel@1:         if (!Nodes.Contains(node)) {
moel@1:           int i = 0;
moel@1:           while (i < Nodes.Count &&
moel@1:             ((TypeNode)Nodes[i]).SensorType < node.SensorType)
moel@1:             i++;
moel@1:           Nodes.Insert(i, node);  
moel@1:         }
moel@1:       } else {
moel@1:         if (Nodes.Contains(node))
moel@1:           Nodes.Remove(node);
moel@1:       }
moel@1:     }
moel@1: 
moel@1:     private void SensorRemoved(ISensor sensor) {
moel@1:       foreach (TypeNode node in typeNodes)
moel@1:         if (node.SensorType == sensor.SensorType) {
moel@1:           node.Nodes.Remove(new SensorNode(sensor));
moel@1:           UpdateNode(node);
moel@1:         }
moel@1:     }
moel@1: 
moel@1:     private void InsertSorted(Node node, ISensor sensor) {
moel@1:       int i = 0;
moel@1:       while (i < node.Nodes.Count &&
moel@1:         ((SensorNode)node.Nodes[i]).Sensor.Index < sensor.Index)
moel@1:         i++;
moel@1:       node.Nodes.Insert(i, new SensorNode(sensor));        
moel@1:     }
moel@1: 
moel@1:     private void SensorAdded(ISensor sensor) {
moel@1:       foreach (TypeNode node in typeNodes)
moel@1:         if (node.SensorType == sensor.SensorType) {
moel@1:           InsertSorted(node, sensor);
moel@1:           UpdateNode(node);
moel@1:         }
moel@1:     }
moel@1:   }
moel@1: }