source: branches/HeuristicLab.Visualization/HeuristicLab.Visualization/3.3/ChartControl.cs @ 9491

#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.Collections.Generic;
using System.Drawing;
using System.Drawing.Drawing2D;
using System.Drawing.Imaging;
using System.Windows.Forms;

namespace HeuristicLab.Visualization {
  public partial class ChartControl : UserControl {
    private Bitmap bitmap;
    public Bitmap Bitmap { get; private set; }
    private bool renderingRequired;
    private Point buttonDownPoint;
    private Point mousePosition;
    private int mouseClickCount;

    private Chart myChart;
    public Chart Chart {
      get { return myChart; }
      set {
        if (myChart != null) myChart.Update -= new EventHandler(myChart_Update);
        myChart = value;
        if (myChart != null) {
          myChart.Update += new EventHandler(myChart_Update);
          SetMode(Chart.Mode);
        }
        GenerateImage();
      }
    }
    private bool myScaleOnResize;
    public bool ScaleOnResize {
      get { return myScaleOnResize; }
      set { myScaleOnResize = value; }
    }

    public ChartControl() {
      InitializeComponent();
      myScaleOnResize = true;
      GenerateImage();
    }

    private void myChart_Update(object sender, EventArgs e) {
      GenerateImage();
    }
    private void pictureBox_SizeChanged(object sender, EventArgs e) {
      if (ScaleOnResize) {
        if ((pictureBox.Width > 0) && (pictureBox.Height > 0) && (Chart != null)) {
          PointD point = Chart.TransformPixelToWorld(new Point(pictureBox.Width, Chart.SizeInPixels.Height - pictureBox.Height));
          Chart.SetPosition(Chart.LowerLeft, point);
        }
      }
      GenerateImage();
    }
    private void pictureBox_VisibleChanged(object sender, EventArgs e) {
      if (pictureBox.Visible && renderingRequired) {
        GenerateImage();
      }
    }

    private void pictureBox_MouseDown(object sender, MouseEventArgs e) {
      buttonDownPoint = e.Location;
      mouseClickCount = e.Clicks;
    }
    protected virtual void pictureBox_MouseUp(object sender, MouseEventArgs e) {
      if (e.Button == MouseButtons.Left) {
        if (Chart.Mode == ChartMode.Zoom) {
          pictureBox.Refresh();
          // some code to preserve the aspect ratio when zooming
          //       +---------------------+ (Width,Height)
          //       |  (x1,y1)            |
          //       |      +-------+      |
          //       |      |       |      | 
          //       |      |       |      |
          //       |      +-------+      |
          //       |             (x2,y2) |
          // (0,0) +---------------------+
          // 
          double aspectRatio = (double)Width / Height;
          double x1 = Math.Min(e.X, buttonDownPoint.X);
          double y1 = Math.Max(e.Y, buttonDownPoint.Y);
          double x2 = Math.Max(e.X, buttonDownPoint.X);
          double y2 = Math.Min(e.Y, buttonDownPoint.Y);
          // consider the relative ratios between the X and Y dimensions of the selection 
          // area and the Width and Height, respectively. The "dominant" dimension is the 
          // one with a ratio closer to 1, and it stays fixed while the other is adjusted 
          // in order to preserve the aspect ratio
          if ((x2 - x1) / Width > (y1 - y2) / Height) {
            y1 = y2 + (x2 - x1) / aspectRatio;
          } else {
            x2 = aspectRatio * (y1 - y2) + x1;
          }
          var lowerLeft = new Point((int)x1, (int)y1);
          var upperRight = new Point((int)x2, (int)y2);
          if ((lowerLeft.X != upperRight.X) && (lowerLeft.Y != upperRight.Y)) {
            Chart.ZoomIn(lowerLeft, upperRight);
          }
        }
      } else if (e.Button == MouseButtons.Right) {
        propertiesToolStripMenuItem.Enabled = Chart.Group.SelectedPrimitives.Count == 1;
        oneLayerUpToolStripMenuItem.Enabled = Chart.Group.SelectedPrimitives.Count == 1;
        oneLayerDownToolStripMenuItem.Enabled = Chart.Group.SelectedPrimitives.Count == 1;
        intoForegroundToolStripMenuItem.Enabled = Chart.Group.SelectedPrimitives.Count == 1;
        intoBackgroundToolStripMenuItem.Enabled = Chart.Group.SelectedPrimitives.Count == 1;
      } else if (e.Button == MouseButtons.Middle) {
        if (Chart.Mode == ChartMode.Zoom) {
          if (mouseClickCount == 1) Chart.ZoomOut();
          else if (mouseClickCount == 2) Chart.Unzoom();
        }
      }
      if (buttonDownPoint == e.Location) {
        if (mouseClickCount == 1)
          Chart.MouseClick(e.Location, e.Button);
        else if (mouseClickCount == 2)
          Chart.MouseDoubleClick(e.Location, e.Button);
      }
    }

    protected virtual void pictureBox_MouseMove(object sender, MouseEventArgs e) {
      var text = Chart.GetToolTipText(e.Location);
      if (toolTip.GetToolTip(pictureBox) != text)
        toolTip.SetToolTip(pictureBox, text);
      Cursor cursor = Chart.GetCursor(e.Location);
      if (cursor != null) pictureBox.Cursor = cursor;
      else pictureBox.Cursor = Cursors.Default;

      if (e.Button != MouseButtons.None) {
        if ((Chart.Mode == ChartMode.Zoom) && (e.Button == MouseButtons.Left)) {
          pictureBox.Refresh();
          Graphics graphics = pictureBox.CreateGraphics();
          Pen pen = new Pen(Color.Gray);
          pen.DashStyle = DashStyle.Dash;
          graphics.DrawRectangle(pen,
                                 Math.Min(e.X, buttonDownPoint.X),
                                 Math.Min(e.Y, buttonDownPoint.Y),
                                 Math.Abs(e.X - buttonDownPoint.X),
                                 Math.Abs(e.Y - buttonDownPoint.Y));
          pen.Dispose();
          graphics.Dispose();
        }
        Chart.MouseDrag(mousePosition, e.Location, e.Button);
      } else {
        Chart.MouseMove(mousePosition, e.Location);
      }
      mousePosition = e.Location;
    }

    protected virtual void pictureBox_MouseWheel(object sender, MouseEventArgs e) {
      if (e.Delta > 0) {
        // zoom in
        double x1 = Chart.LowerLeft.X + Chart.Size.Width / 8;
        double y1 = Chart.LowerLeft.Y + Chart.Size.Height / 8;
        double x2 = Chart.UpperRight.X - Chart.Size.Width / 8;
        double y2 = Chart.UpperRight.Y - Chart.Size.Height / 8;
        Chart.ZoomIn(x1, y1, x2, y2);
      } else if (e.Delta < 0) {
        // zoom out
        double x1 = Chart.LowerLeft.X - Chart.Size.Width / 8;
        double y1 = Chart.LowerLeft.Y - Chart.Size.Height / 8;
        double x2 = Chart.UpperRight.X + Chart.Size.Width / 8;
        double y2 = Chart.UpperRight.Y + Chart.Size.Height / 8;
        Chart.ZoomIn(x1, y1, x2, y2);
      }
    }

    private void pictureBox_MouseEnter(object sender, EventArgs e) {
      pictureBox.Focus();
    }

    protected virtual void ChartControl_KeyDown(object sender, KeyEventArgs e) {
      if (Chart.Mode == ChartMode.Select) {
        if (e.KeyCode == Keys.Delete) {
          IList<IPrimitive> selected = Chart.Group.SelectedPrimitives;
          Chart.UpdateEnabled = false;
          foreach (IPrimitive primitive in selected)
            Chart.Group.Remove(primitive);
          Chart.UpdateEnabled = true;
          Chart.EnforceUpdate();
        }
      }
    }

    private void moveToolStripMenuItem_Click(object sender, EventArgs e) {
      SetMode(ChartMode.Move);
    }
    private void zoomToolStripMenuItem_Click(object sender, EventArgs e) {
      SetMode(ChartMode.Zoom);
    }
    private void selectToolStripMenuItem_Click(object sender, EventArgs e) {
      SetMode(ChartMode.Select);
    }
    private void oneLayerUpToolStripMenuItem_Click(object sender, EventArgs e) {
      if (Chart.Group.SelectedPrimitives.Count == 1) {
        Chart.Group.SelectedPrimitives[0].OneLayerUp();
      }
    }
    private void oneLayerDownToolStripMenuItem_Click(object sender, EventArgs e) {
      if (Chart.Group.SelectedPrimitives.Count == 1) {
        Chart.Group.SelectedPrimitives[0].OneLayerDown();
      }
    }
    private void intoForegroundToolStripMenuItem_Click(object sender, EventArgs e) {
      if (Chart.Group.SelectedPrimitives.Count == 1) {
        Chart.Group.SelectedPrimitives[0].IntoForeground();
      }
    }
    private void intoBackgroundToolStripMenuItem_Click(object sender, EventArgs e) {
      if (Chart.Group.SelectedPrimitives.Count == 1) {
        Chart.Group.SelectedPrimitives[0].IntoBackground();
      }
    }
    private void propertiesToolStripMenuItem_Click(object sender, EventArgs e) {
      if (Chart.Group.SelectedPrimitives.Count == 1) {
        PropertiesDialog dialog = new PropertiesDialog(Chart.Group.SelectedPrimitives[0]);
        dialog.ShowDialog(this);
        dialog.Dispose();
      }
    }

    private void GenerateImage() {
      if (!Visible) {
        renderingRequired = true;
      } else {
        if ((pictureBox.Width == 0) || (pictureBox.Height == 0)) {
          bitmap = null;
        } else {
          bitmap = new Bitmap(pictureBox.Width, pictureBox.Height);
          if (Chart != null) {
            Graphics graphics = Graphics.FromImage(bitmap);
            graphics.SetClip(new System.Drawing.Rectangle(0, 0, pictureBox.Width, pictureBox.Height));
            Chart.Render(graphics, pictureBox.Width, pictureBox.Height);
            graphics.Dispose();
          }
        }
        if (pictureBox.Image != null) pictureBox.Image.Dispose();
        pictureBox.Image = bitmap;
        renderingRequired = false;
      }
    }

    private void SetMode(ChartMode mode) {
      moveToolStripMenuItem.Checked = false;
      zoomToolStripMenuItem.Checked = false;
      selectToolStripMenuItem.Checked = false;
      if (mode == ChartMode.Move) moveToolStripMenuItem.Checked = true;
      else if (mode == ChartMode.Zoom) zoomToolStripMenuItem.Checked = true;
      else if (mode == ChartMode.Select) selectToolStripMenuItem.Checked = true;
      Chart.Mode = mode;
    }

    private void SaveImageAsBitmap(string filename) {
      if (bitmap == null) throw new Exception("Bitmap is null.");
      bitmap.Save(filename);
    }

    private void SaveImageAsEmf(string filename) {
      using (Graphics g = CreateGraphics()) {
        using (Metafile file = new Metafile(filename, g.GetHdc())) {
          using (Graphics emfFile = Graphics.FromImage(file)) {
            Chart.Render(emfFile, pictureBox.Width, pictureBox.Height);
          }
        }
        g.ReleaseHdc();
      }
    }

    private void saveImageToolStripMenuItem_Click(object sender, EventArgs e) {
      if (saveFileDialog.ShowDialog() == DialogResult.OK) {
        string filename = saveFileDialog.FileName.ToLower();
        if (filename.EndsWith("bmp")) SaveImageAsBitmap(filename);
        else if (filename.EndsWith("emf")) SaveImageAsEmf(filename);
        else SaveImageAsBitmap(filename);
      }
    }
  }
}