source: trunk/sources/HeuristicLab.Analysis.Views/3.3/HeatMapView.cs @ 4739

#region License Information
/* HeuristicLab
 * Copyright (C) 2002-2010 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
 *
 * This file is part of HeuristicLab.
 *
 * HeuristicLab is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * HeuristicLab is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
 */
#endregion

using System;
using System.Drawing;
using System.Windows.Forms;
using System.Windows.Forms.DataVisualization.Charting;
using HeuristicLab.Core.Views;
using HeuristicLab.MainForm;

namespace HeuristicLab.Analysis.Views {
  [View("HeatMap View")]
  [Content(typeof(HeatMap), true)]
  public partial class HeatMapView : ItemView {
    protected static Color[] colors = new Color[256];
    protected static Color[] grayscaleColors = new Color[256];

    #region Initialize Colors
    static HeatMapView() {
      int stepWidth = (256 * 4) / colors.Length;
      int currentValue;
      int currentClass;
      for (int i = 0; i < colors.Length; i++) {
        currentValue = (i * stepWidth) % 256;
        currentClass = (i * stepWidth) / 256;
        switch (currentClass) {
          case 0: { colors[i] = Color.FromArgb(0, currentValue, 255); break; }        // blue -> cyan
          case 1: { colors[i] = Color.FromArgb(0, 255, 255 - currentValue); break; }  // cyan -> green
          case 2: { colors[i] = Color.FromArgb(currentValue, 255, 0); break; }        // green -> yellow
          case 3: { colors[i] = Color.FromArgb(255, 255 - currentValue, 0); break; }  // yellow -> red
        }
      }
      for (int i = 0; i < 256; i++)
        grayscaleColors[i] = Color.FromArgb(255 - i, 255 - i, 255 - i);  // white -> black
    }
    #endregion

    public new HeatMap Content {
      get { return (HeatMap)base.Content; }
      set { base.Content = value; }
    }

    public HeatMapView() {
      InitializeComponent();
      chart.CustomizeAllChartAreas();
    }

    protected override void DeregisterContentEvents() {
      Content.TitleChanged -= new EventHandler(Content_TitleChanged);
      Content.MinimumChanged -= new EventHandler(Content_MinimumChanged);
      Content.MaximumChanged -= new EventHandler(Content_MaximumChanged);
      base.DeregisterContentEvents();
    }
    protected override void RegisterContentEvents() {
      base.RegisterContentEvents();
      Content.TitleChanged += new EventHandler(Content_TitleChanged);
      Content.MinimumChanged += new EventHandler(Content_MinimumChanged);
      Content.MaximumChanged += new EventHandler(Content_MaximumChanged);
    }

    protected override void OnContentChanged() {
      base.OnContentChanged();
      if (Content == null) {
        chart.Series.Clear();
        chart.Titles[0].Text = "Heat Map";
        minimumLabel.Text = "0.0";
        maximumLabel.Text = "1.0";
      } else {
        chart.Titles[0].Text = Content.Title;
        minimumLabel.Text = Content.Minimum.ToString();
        maximumLabel.Text = Content.Maximum.ToString();
        UpdatePoints();
      }
    }

    protected override void SetEnabledStateOfControls() {
      base.SetEnabledStateOfControls();
      chart.Enabled = Content != null;
      grayscaleCheckBox.Enabled = Content != null;
    }

    protected virtual void UpdatePoints() {
      chart.Series.Clear();
      Series series = new Series();
      series.ChartType = SeriesChartType.Point;
      series.XValueType = ChartValueType.Int32;
      series.YValueType = ChartValueType.Int32;
      series.YAxisType = AxisType.Primary;
      for (int i = 1; i < Content.Rows + 1; i++)
        for (int j = 1; j < Content.Columns + 1; j++)
          series.Points.Add(CreateDataPoint(j, i, Content[i - 1, j - 1]));
      chart.ChartAreas[0].AxisX.Minimum = 0;
      chart.ChartAreas[0].AxisX.Maximum = Content.Columns + 1;
      chart.ChartAreas[0].AxisY.Minimum = 0;
      chart.ChartAreas[0].AxisY.Maximum = Content.Rows + 1;
      chart.Series.Add(series);
    }

    protected virtual DataPoint CreateDataPoint(int index1, int index2, double value) {
      DataPoint p = new DataPoint(index1, index2);
      p.Color = GetDataPointColor(value, Content.Minimum, Content.Maximum, grayscaleCheckBox.Checked);
      p.MarkerStyle = MarkerStyle.Square;
      //p.MarkerSize = 10;
      //string nl = Environment.NewLine;
      //p.ToolTip = string.Format("Row: {0}{3}Column: {1}{3}Value: {2}", index2, index1, value, nl);
      return p;
    }

    protected virtual Color GetDataPointColor(double value, double min, double max, bool grayscale) {
      int count = grayscale ? grayscaleColors.Length : colors.Length;
      int index = (int)((count - 1) * (value - min) / (max - min));
      if (index >= count) index = count - 1;
      if (index < 0) index = 0;
      return grayscale ? grayscaleColors[index] : colors[index];
    }

    #region Content Events
    protected virtual void Content_TitleChanged(object sender, EventArgs e) {
      if (InvokeRequired)
        Invoke(new EventHandler(Content_TitleChanged), sender, e);
      else {
        chart.Titles[0].Text = Content.Title;
      }
    }
    protected virtual void Content_MinimumChanged(object sender, EventArgs e) {
      if (InvokeRequired)
        Invoke(new EventHandler(Content_MinimumChanged), sender, e);
      else {
        minimumLabel.Text = Content.Minimum.ToString();
        UpdatePoints();
      }
    }
    protected virtual void Content_MaximumChanged(object sender, EventArgs e) {
      if (InvokeRequired)
        Invoke(new EventHandler(Content_MaximumChanged), sender, e);
      else {
        maximumLabel.Text = Content.Maximum.ToString();
        UpdatePoints();
      }
    }
    #endregion

    #region Control Events
    protected virtual void grayscaledImagesCheckBox_CheckedChanged(object sender, EventArgs e) {
      grayscalesPictureBox.Visible = grayscaleCheckBox.Checked;
      colorsPictureBox.Visible = !grayscaleCheckBox.Checked;
      UpdatePoints();
    }
    #endregion
  }
}