GUI/SensorNotifyIcon.cs
author moel.mich
Tue, 14 Sep 2010 19:23:11 +0000
changeset 185 edb59f3745e8
parent 166 fa9dfbfc4145
child 202 551243a66b32
permissions -rw-r--r--
Fixed an UnauthorizedAccessException when modifying the auto-startup registry entry.
moel@40
     1
/*
moel@40
     2
  
moel@40
     3
  Version: MPL 1.1/GPL 2.0/LGPL 2.1
moel@40
     4
moel@40
     5
  The contents of this file are subject to the Mozilla Public License Version
moel@40
     6
  1.1 (the "License"); you may not use this file except in compliance with
moel@40
     7
  the License. You may obtain a copy of the License at
moel@40
     8
 
moel@40
     9
  http://www.mozilla.org/MPL/
moel@40
    10
moel@40
    11
  Software distributed under the License is distributed on an "AS IS" basis,
moel@40
    12
  WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
moel@40
    13
  for the specific language governing rights and limitations under the License.
moel@40
    14
moel@40
    15
  The Original Code is the Open Hardware Monitor code.
moel@40
    16
moel@40
    17
  The Initial Developer of the Original Code is 
moel@40
    18
  Michael Möller <m.moeller@gmx.ch>.
moel@40
    19
  Portions created by the Initial Developer are Copyright (C) 2009-2010
moel@40
    20
  the Initial Developer. All Rights Reserved.
moel@40
    21
moel@40
    22
  Contributor(s):
moel@40
    23
moel@40
    24
  Alternatively, the contents of this file may be used under the terms of
moel@40
    25
  either the GNU General Public License Version 2 or later (the "GPL"), or
moel@40
    26
  the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
moel@40
    27
  in which case the provisions of the GPL or the LGPL are applicable instead
moel@40
    28
  of those above. If you wish to allow use of your version of this file only
moel@40
    29
  under the terms of either the GPL or the LGPL, and not to allow others to
moel@40
    30
  use your version of this file under the terms of the MPL, indicate your
moel@40
    31
  decision by deleting the provisions above and replace them with the notice
moel@40
    32
  and other provisions required by the GPL or the LGPL. If you do not delete
moel@40
    33
  the provisions above, a recipient may use your version of this file under
moel@40
    34
  the terms of any one of the MPL, the GPL or the LGPL.
moel@40
    35
 
moel@40
    36
*/
moel@40
    37
moel@40
    38
using System;
moel@40
    39
using System.Drawing;
moel@40
    40
using System.Drawing.Drawing2D;
moel@40
    41
using System.Drawing.Imaging;
moel@40
    42
using System.Drawing.Text;
moel@176
    43
using System.Runtime.InteropServices;
moel@40
    44
using System.Windows.Forms;
moel@40
    45
using OpenHardwareMonitor.Hardware;
moel@40
    46
using OpenHardwareMonitor.Utilities;
moel@40
    47
moel@40
    48
namespace OpenHardwareMonitor.GUI {
moel@40
    49
  public class SensorNotifyIcon : IDisposable {
moel@40
    50
moel@40
    51
    private ISensor sensor;
moel@40
    52
    private NotifyIcon notifyIcon;
moel@40
    53
    private Bitmap bitmap;
moel@40
    54
    private Graphics graphics;
moel@42
    55
    private Color color;
moel@43
    56
    private Color darkColor;
moel@43
    57
    private Brush brush;
moel@43
    58
    private Brush darkBrush;
moel@43
    59
    private Pen pen;
moel@70
    60
    private Font font;
moel@40
    61
moel@133
    62
    public SensorNotifyIcon(SystemTray sensorSystemTray, ISensor sensor,
moel@165
    63
      bool balloonTip, PersistentSettings settings) 
moel@42
    64
    {
moel@40
    65
      this.sensor = sensor;
moel@40
    66
      this.notifyIcon = new NotifyIcon();
moel@43
    67
moel@43
    68
      Color defaultColor = Color.Black;
moel@43
    69
      if (sensor.SensorType == SensorType.Load) {
moel@43
    70
        defaultColor = Color.FromArgb(0xff, 0x70, 0x8c, 0xf1);
moel@43
    71
      }
moel@166
    72
      Color = settings.GetValue(new Identifier(sensor.Identifier, 
moel@109
    73
        "traycolor").ToString(), defaultColor);      
moel@40
    74
      
moel@43
    75
      this.pen = new Pen(Color.FromArgb(96, Color.Black));
moel@117
    76
      this.font = SystemFonts.MessageBoxFont;
moel@43
    77
moel@156
    78
      ContextMenu contextMenu = new ContextMenu();
moel@156
    79
      MenuItem hideShowItem = new MenuItem("Hide/Show");
moel@133
    80
      hideShowItem.Click += delegate(object obj, EventArgs args) {
moel@133
    81
        sensorSystemTray.SendHideShowCommand();
moel@133
    82
      };
moel@156
    83
      contextMenu.MenuItems.Add(hideShowItem);
moel@156
    84
      contextMenu.MenuItems.Add(new MenuItem("-"));
moel@156
    85
      MenuItem removeItem = new MenuItem("Remove Sensor");
moel@42
    86
      removeItem.Click += delegate(object obj, EventArgs args) {
moel@42
    87
        sensorSystemTray.Remove(this.sensor);
moel@40
    88
      };
moel@156
    89
      contextMenu.MenuItems.Add(removeItem);
moel@156
    90
      MenuItem colorItem = new MenuItem("Change Color...");
moel@42
    91
      colorItem.Click += delegate(object obj, EventArgs args) {
moel@42
    92
        ColorDialog dialog = new ColorDialog();
moel@43
    93
        dialog.Color = Color;
moel@42
    94
        if (dialog.ShowDialog() == DialogResult.OK) {
moel@43
    95
          Color = dialog.Color;
moel@166
    96
          settings.SetValue(new Identifier(sensor.Identifier,
moel@109
    97
            "traycolor").ToString(), Color);
moel@42
    98
        }
moel@42
    99
      };
moel@156
   100
      contextMenu.MenuItems.Add(colorItem);
moel@156
   101
      contextMenu.MenuItems.Add(new MenuItem("-"));
moel@156
   102
      MenuItem exitItem = new MenuItem("Exit");
moel@133
   103
      exitItem.Click += delegate(object obj, EventArgs args) {
moel@133
   104
        sensorSystemTray.SendExitCommand();
moel@133
   105
      };
moel@156
   106
      contextMenu.MenuItems.Add(exitItem);
moel@156
   107
      this.notifyIcon.ContextMenu = contextMenu;
moel@133
   108
      this.notifyIcon.DoubleClick += delegate(object obj, EventArgs args) {
moel@133
   109
        sensorSystemTray.SendHideShowCommand();
moel@133
   110
      };
moel@40
   111
moel@40
   112
      this.bitmap = new Bitmap(16, 16, PixelFormat.Format32bppArgb);
moel@40
   113
      this.graphics = Graphics.FromImage(this.bitmap);
moel@117
   114
moel@117
   115
      if (Environment.OSVersion.Version.Major > 5) {
moel@117
   116
        this.graphics.TextRenderingHint = TextRenderingHint.ClearTypeGridFit;
moel@117
   117
        this.graphics.SmoothingMode = SmoothingMode.HighQuality;
moel@117
   118
      }
moel@40
   119
    }
moel@40
   120
moel@40
   121
    public ISensor Sensor {
moel@40
   122
      get { return sensor; }
moel@40
   123
    }
moel@40
   124
moel@42
   125
    public Color Color {
moel@42
   126
      get { return color; }
moel@43
   127
      set { 
moel@43
   128
        this.color = value;
moel@43
   129
        this.darkColor = Color.FromArgb(255,
moel@44
   130
          this.color.R / 3,
moel@44
   131
          this.color.G / 3,
moel@44
   132
          this.color.B / 3);
moel@43
   133
        Brush brush = this.brush;
moel@43
   134
        this.brush = new SolidBrush(this.color);
moel@43
   135
        if (brush != null)
moel@43
   136
          brush.Dispose();
moel@43
   137
        Brush darkBrush = this.darkBrush;
moel@43
   138
        this.darkBrush = new SolidBrush(this.darkColor);
moel@43
   139
        if (darkBrush != null)
moel@43
   140
          darkBrush.Dispose();
moel@43
   141
      }
moel@42
   142
    }
moel@42
   143
moel@40
   144
    public void Dispose() {      
moel@40
   145
      Icon icon = notifyIcon.Icon;
moel@40
   146
      notifyIcon.Icon = null;
moel@40
   147
      if (icon != null)
moel@40
   148
        icon.Dispose();      
moel@40
   149
      notifyIcon.Dispose();
moel@40
   150
moel@43
   151
      if (brush != null)
moel@43
   152
        brush.Dispose();
moel@43
   153
      if (darkBrush != null)
moel@43
   154
        darkBrush.Dispose();
moel@43
   155
      pen.Dispose();
moel@85
   156
      graphics.Dispose();      
moel@85
   157
      bitmap.Dispose();      
moel@40
   158
    }
moel@40
   159
moel@40
   160
    private string GetString() {
moel@40
   161
      switch (sensor.SensorType) {
moel@40
   162
        case SensorType.Voltage:
moel@40
   163
          return string.Format("{0:F11}", sensor.Value);
moel@40
   164
        case SensorType.Clock:
moel@40
   165
          return string.Format("{0:F11}", 1e-3f * sensor.Value);
moel@40
   166
        case SensorType.Load: 
moel@118
   167
          return string.Format("{0:F0}", sensor.Value);
moel@40
   168
        case SensorType.Temperature: 
moel@40
   169
          return string.Format("{0:F0}", sensor.Value);
moel@40
   170
        case SensorType.Fan: 
moel@40
   171
          return string.Format("{0:F11}", 1e-3f * sensor.Value);
moel@57
   172
        case SensorType.Flow:
moel@57
   173
          return string.Format("{0:F11}", 1e-3f * sensor.Value);
moel@118
   174
        case SensorType.Control:
moel@118
   175
          return string.Format("{0:F0}", sensor.Value);
moel@40
   176
      }
moel@40
   177
      return "-";
moel@40
   178
    }
moel@40
   179
moel@40
   180
    private Icon CreateTransparentIcon() {
moel@40
   181
moel@40
   182
      graphics.Clear(Color.Black);
moel@70
   183
      TextRenderer.DrawText(graphics, GetString(), font,
moel@117
   184
        new Point(-2, 0), Color.White, Color.Black);        
moel@40
   185
moel@40
   186
      BitmapData data = bitmap.LockBits(
moel@40
   187
        new Rectangle(0, 0, bitmap.Width, bitmap.Height),
moel@40
   188
        ImageLockMode.ReadOnly, PixelFormat.Format32bppArgb);
moel@40
   189
moel@40
   190
      int stride = data.Stride;
moel@40
   191
      IntPtr Scan0 = data.Scan0;
moel@40
   192
moel@40
   193
      int numBytes = bitmap.Width * bitmap.Height * 4;
moel@40
   194
      byte[] bytes = new byte[numBytes];
moel@40
   195
      Marshal.Copy(Scan0, bytes, 0, numBytes);
moel@40
   196
      bitmap.UnlockBits(data);
moel@40
   197
moel@40
   198
      byte red, green, blue;
moel@40
   199
      for (int i = 0; i < bytes.Length; i += 4) {
moel@40
   200
        blue = bytes[i];
moel@40
   201
        green = bytes[i + 1];
moel@40
   202
        red = bytes[i + 2];
moel@40
   203
moel@42
   204
        bytes[i] = color.B;
moel@42
   205
        bytes[i + 1] = color.G;
moel@42
   206
        bytes[i + 2] = color.R;
moel@40
   207
        bytes[i + 3] = (byte)(0.3 * red + 0.59 * green + 0.11 * blue);
moel@40
   208
      }
moel@40
   209
moel@40
   210
      return IconFactory.Create(bytes, 16, 16, PixelFormat.Format32bppArgb);
moel@40
   211
    }
moel@40
   212
moel@86
   213
    private Icon CreateLoadIcon() {      
moel@86
   214
      try {
moel@86
   215
        graphics.Clear(Color.Transparent);
moel@86
   216
      } catch (ArgumentException) {
moel@86
   217
        graphics.Clear(Color.Black);
moel@86
   218
      }
moel@43
   219
      graphics.FillRectangle(darkBrush, 0.5f, -0.5f, 14, 16);
moel@43
   220
      float y = 0.16f * (100 - sensor.Value.Value);
moel@43
   221
      graphics.FillRectangle(brush, 0.5f, -0.5f + y, 14, 16 - y);
moel@43
   222
      graphics.DrawRectangle(pen, 1, 0, 13, 15);
moel@43
   223
moel@43
   224
      BitmapData data = bitmap.LockBits(
moel@43
   225
        new Rectangle(0, 0, bitmap.Width, bitmap.Height),
moel@43
   226
        ImageLockMode.ReadOnly, PixelFormat.Format32bppArgb);
moel@43
   227
      byte[] bytes = new byte[bitmap.Width * bitmap.Height * 4];
moel@43
   228
      Marshal.Copy(data.Scan0, bytes, 0, bytes.Length);
moel@43
   229
      bitmap.UnlockBits(data);
moel@43
   230
moel@43
   231
      return IconFactory.Create(bytes, 16, 16, PixelFormat.Format32bppArgb);
moel@43
   232
    }
moel@43
   233
moel@40
   234
    public void Update() {
moel@40
   235
      Icon icon = notifyIcon.Icon;
moel@40
   236
moel@43
   237
      if (sensor.SensorType == SensorType.Load) {
moel@43
   238
        notifyIcon.Icon = CreateLoadIcon();
moel@40
   239
      } else {
moel@40
   240
        notifyIcon.Icon = CreateTransparentIcon();
moel@40
   241
      }
moel@40
   242
      if (icon != null) 
moel@40
   243
        icon.Dispose();
moel@40
   244
moel@40
   245
      string format = "";
moel@40
   246
      switch (sensor.SensorType) {
moel@116
   247
        case SensorType.Voltage: format = "\n{0}: {1:F2} V"; break;
moel@116
   248
        case SensorType.Clock: format = "\n{0}: {1:F0} MHz"; break;
moel@116
   249
        case SensorType.Load: format = "\n{0}: {1:F1} %"; break;
moel@116
   250
        case SensorType.Temperature: format = "\n{0}: {1:F1} °C"; break;
moel@116
   251
        case SensorType.Fan: format = "\n{0}: {1:F0} RPM"; break;
moel@116
   252
        case SensorType.Flow: format = "\n{0}: {1:F0} L/h"; break;
moel@118
   253
        case SensorType.Control: format = "\n{0}: {1:F1} %"; break;
moel@40
   254
      }
moel@116
   255
      string formattedValue = string.Format(format, sensor.Name, sensor.Value);
moel@116
   256
      string hardwareName = sensor.Hardware.Name;
moel@116
   257
      hardwareName = hardwareName.Substring(0, 
moel@116
   258
        Math.Min(63 - formattedValue.Length, hardwareName.Length));
moel@116
   259
      string text = hardwareName + formattedValue;
moel@116
   260
      if (text.Length > 63)
moel@116
   261
        text = null;
moel@40
   262
moel@116
   263
      notifyIcon.Text = text;
moel@42
   264
      notifyIcon.Visible = true;         
moel@40
   265
    }
moel@40
   266
  }
moel@40
   267
}