/*

   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>

 */

 using System;

 using System.Drawing;

 using System.Drawing.Drawing2D;

 using System.Drawing.Imaging;

 using System.Drawing.Text;

 using System.Runtime.InteropServices;

 using System.Windows.Forms;

 using OpenHardwareMonitor.Hardware;

 using OpenHardwareMonitor.Utilities;

 namespace OpenHardwareMonitor.GUI {

   public class SensorNotifyIcon : IDisposable {

     private UnitManager unitManager;

     private ISensor sensor;

     private NotifyIcon notifyIcon;

     private Bitmap bitmap;

     private Graphics graphics;

     private Color color;

     private Color darkColor;

     private Brush brush;

     private Brush darkBrush;

     private Pen pen;

     private Font font;

     private Font smallFont;

     public SensorNotifyIcon(SystemTray sensorSystemTray, ISensor sensor,

       bool balloonTip, PersistentSettings settings, UnitManager unitManager) 

     {

       this.unitManager = unitManager;

       this.sensor = sensor;

       this.notifyIcon = new NotifyIcon();

       Color defaultColor = Color.Black;

       if (sensor.SensorType == SensorType.Load ||

           sensor.SensorType == SensorType.Control ||

           sensor.SensorType == SensorType.Level) 

       {

         defaultColor = Color.FromArgb(0xff, 0x70, 0x8c, 0xf1);

       }

       Color = settings.GetValue(new Identifier(sensor.Identifier, 

         "traycolor").ToString(), defaultColor);      

       this.pen = new Pen(Color.FromArgb(96, Color.Black));

       this.font = SystemFonts.MessageBoxFont;

       this.smallFont = new Font(font.FontFamily, font.Size * 0.8f);

       ContextMenu contextMenu = new ContextMenu();

       MenuItem hideShowItem = new MenuItem("Hide/Show");

       hideShowItem.Click += delegate(object obj, EventArgs args) {

         sensorSystemTray.SendHideShowCommand();

       };

       contextMenu.MenuItems.Add(hideShowItem);

       contextMenu.MenuItems.Add(new MenuItem("-"));

       MenuItem removeItem = new MenuItem("Remove Sensor");

       removeItem.Click += delegate(object obj, EventArgs args) {

         sensorSystemTray.Remove(this.sensor);

       };

       contextMenu.MenuItems.Add(removeItem);

       MenuItem colorItem = new MenuItem("Change Color...");

       colorItem.Click += delegate(object obj, EventArgs args) {

         ColorDialog dialog = new ColorDialog();

         dialog.Color = Color;

         if (dialog.ShowDialog() == DialogResult.OK) {

           Color = dialog.Color;

           settings.SetValue(new Identifier(sensor.Identifier,

             "traycolor").ToString(), Color);

         }

       };

       contextMenu.MenuItems.Add(colorItem);

       contextMenu.MenuItems.Add(new MenuItem("-"));

       MenuItem exitItem = new MenuItem("Exit");

       exitItem.Click += delegate(object obj, EventArgs args) {

         sensorSystemTray.SendExitCommand();

       };

       contextMenu.MenuItems.Add(exitItem);

       this.notifyIcon.ContextMenu = contextMenu;

       this.notifyIcon.DoubleClick += delegate(object obj, EventArgs args) {

         sensorSystemTray.SendHideShowCommand();

       };

       // get the default dpi to create an icon with the correct size

       float dpiX, dpiY;

       using (Bitmap b = new Bitmap(1, 1, PixelFormat.Format32bppArgb)) {

         dpiX = b.HorizontalResolution;

         dpiY = b.VerticalResolution;

       }

       // adjust the size of the icon to current dpi (default is 16x16 at 96 dpi) 

       int width = (int)Math.Round(16 * dpiX / 96);

       int height = (int)Math.Round(16 * dpiY / 96);

       // make sure it does never get smaller than 16x16

       width = width < 16 ? 16: width;

       height = height < 16 ? 16: height;

       this.bitmap = new Bitmap(width, height, PixelFormat.Format32bppArgb);      

       this.graphics = Graphics.FromImage(this.bitmap);

       if (Environment.OSVersion.Version.Major > 5) {

         this.graphics.TextRenderingHint = TextRenderingHint.ClearTypeGridFit;

         this.graphics.SmoothingMode = SmoothingMode.HighQuality;

       }

     }

     public ISensor Sensor {

       get { return sensor; }

     }

     public Color Color {

       get { return color; }

       set { 

         this.color = value;

         this.darkColor = Color.FromArgb(255,

           this.color.R / 3,

           this.color.G / 3,

           this.color.B / 3);

         Brush brush = this.brush;

         this.brush = new SolidBrush(this.color);

         if (brush != null)

           brush.Dispose();

         Brush darkBrush = this.darkBrush;

         this.darkBrush = new SolidBrush(this.darkColor);

         if (darkBrush != null)

           darkBrush.Dispose();

       }

     }

     public void Dispose() {      

       Icon icon = notifyIcon.Icon;

       notifyIcon.Icon = null;

       if (icon != null)

         icon.Dispose();      

       notifyIcon.Dispose();

       if (brush != null)

         brush.Dispose();

       if (darkBrush != null)

         darkBrush.Dispose();

       pen.Dispose();

       graphics.Dispose();      

       bitmap.Dispose();

       smallFont.Dispose();

     }

     private string GetString() {

       if (!sensor.Value.HasValue)

         return "-";

       switch (sensor.SensorType) {

         case SensorType.Voltage:

           return string.Format("{0:F1}", sensor.Value);

         case SensorType.Clock:

           return string.Format("{0:F1}", 1e-3f * sensor.Value);

         case SensorType.Load: 

           return string.Format("{0:F0}", sensor.Value);

         case SensorType.Temperature:

           if (unitManager.TemperatureUnit == TemperatureUnit.Fahrenheit)

             return string.Format("{0:F0}", 

               UnitManager.CelsiusToFahrenheit(sensor.Value));

           else 

             return string.Format("{0:F0}", sensor.Value);

         case SensorType.Fan: 

           return string.Format("{0:F1}", 1e-3f * sensor.Value);

         case SensorType.Flow:

           return string.Format("{0:F1}", 1e-3f * sensor.Value);

         case SensorType.Control:

           return string.Format("{0:F0}", sensor.Value);

         case SensorType.Level:

           return string.Format("{0:F0}", sensor.Value);

         case SensorType.Power:

           return string.Format("{0:F0}", sensor.Value);

         case SensorType.Data:

           return string.Format("{0:F0}", sensor.Value);

         case SensorType.Factor:

           return string.Format("{0:F1}", sensor.Value);

       }

       return "-";

     }

     private Icon CreateTransparentIcon() {

       string text = GetString();

       int count = 0;

       for (int i = 0; i < text.Length; i++)

         if ((text[i] >= '0' && text[i] <= '9') || text[i] == '-')

           count++;

       bool small = count > 2;

       graphics.Clear(Color.Black);

       TextRenderer.DrawText(graphics, text, small ? smallFont : font,

         new Point(-2, small ? 1 : 0), Color.White, Color.Black);        

       BitmapData data = bitmap.LockBits(

         new Rectangle(0, 0, bitmap.Width, bitmap.Height),

         ImageLockMode.ReadOnly, PixelFormat.Format32bppArgb);

       IntPtr Scan0 = data.Scan0;

       int numBytes = bitmap.Width * bitmap.Height * 4;

       byte[] bytes = new byte[numBytes];

       Marshal.Copy(Scan0, bytes, 0, numBytes);

       bitmap.UnlockBits(data);

       byte red, green, blue;

       for (int i = 0; i < bytes.Length; i += 4) {

         blue = bytes[i];

         green = bytes[i + 1];

         red = bytes[i + 2];

         bytes[i] = color.B;

         bytes[i + 1] = color.G;

         bytes[i + 2] = color.R;

         bytes[i + 3] = (byte)(0.3 * red + 0.59 * green + 0.11 * blue);

       }

       return IconFactory.Create(bytes, bitmap.Width, bitmap.Height, 

         PixelFormat.Format32bppArgb);

     }

     private Icon CreatePercentageIcon() {      

       try {

         graphics.Clear(Color.Transparent);

       } catch (ArgumentException) {

         graphics.Clear(Color.Black);

       }

       graphics.FillRectangle(darkBrush, 0.5f, -0.5f, bitmap.Width - 2, bitmap.Height);

       float value = sensor.Value.GetValueOrDefault();

       float y = 0.16f * (100 - value);

       graphics.FillRectangle(brush, 0.5f, -0.5f + y, bitmap.Width - 2, bitmap.Height - y);

       graphics.DrawRectangle(pen, 1, 0, bitmap.Width - 3, bitmap.Height - 1);

       BitmapData data = bitmap.LockBits(

         new Rectangle(0, 0, bitmap.Width, bitmap.Height),

         ImageLockMode.ReadOnly, PixelFormat.Format32bppArgb);

       byte[] bytes = new byte[bitmap.Width * bitmap.Height * 4];

       Marshal.Copy(data.Scan0, bytes, 0, bytes.Length);

       bitmap.UnlockBits(data);

       return IconFactory.Create(bytes, bitmap.Width, bitmap.Height, 

         PixelFormat.Format32bppArgb);

     }

     public void Update() {

       Icon icon = notifyIcon.Icon;

       switch (sensor.SensorType) {

         case SensorType.Load:

         case SensorType.Control:

         case SensorType.Level:

           notifyIcon.Icon = CreatePercentageIcon();

           break;

         default:

           notifyIcon.Icon = CreateTransparentIcon();

           break;

       }

       if (icon != null) 

         icon.Dispose();

       string format = "";

       switch (sensor.SensorType) {

         case SensorType.Voltage: format = "\n{0}: {1:F2} V"; break;

         case SensorType.Clock: format = "\n{0}: {1:F0} MHz"; break;

         case SensorType.Load: format = "\n{0}: {1:F1} %"; break;

         case SensorType.Temperature: format = "\n{0}: {1:F1} °C"; break;

         case SensorType.Fan: format = "\n{0}: {1:F0} RPM"; break;

         case SensorType.Flow: format = "\n{0}: {1:F0} L/h"; break;

         case SensorType.Control: format = "\n{0}: {1:F1} %"; break;

         case SensorType.Level: format = "\n{0}: {1:F1} %"; break;

         case SensorType.Power: format = "\n{0}: {1:F0} W"; break;

         case SensorType.Data: format = "\n{0}: {1:F0} GB"; break;

         case SensorType.Factor: format = "\n{0}: {1:F3} GB"; break;

       }

       string formattedValue = string.Format(format, sensor.Name, sensor.Value);

       if (sensor.SensorType == SensorType.Temperature &&

         unitManager.TemperatureUnit == TemperatureUnit.Fahrenheit) 

       {

         format = "\n{0}: {1:F1} °F";

         formattedValue = string.Format(format, sensor.Name,

           UnitManager.CelsiusToFahrenheit(sensor.Value));

       }

       string hardwareName = sensor.Hardware.Name;

       hardwareName = hardwareName.Substring(0, 

         Math.Min(63 - formattedValue.Length, hardwareName.Length));

       string text = hardwareName + formattedValue;

       if (text.Length > 63)

         text = null;

       notifyIcon.Text = text;

       notifyIcon.Visible = true;         

     }

   }

 }