LM-Middleware/HexcalMC/Base/Scatter/ScatterPlot.cs

using System;
using System.Collections.Generic;
using System.Drawing;
using System.Windows.Forms;

namespace Scatter
{
	public partial class ScatterPlot : UserControl
	{
		private List<List<double[]>> Points = new List<List<double[]>>();
		private List<PointF[]> ProjPoints = new List<PointF[]>();
		private double f = 1000;
		private double d = 5;
		private double[] d_w = new double[3];
		private double last_azimuth, azimuth = 0, last_elevation, elevation = 0;
		private bool leftMousePressed = false;
		private PointF ptMouseClick;

		public double Distance
		{
			get { return d; }
			set { d = (value >= 0.1) ? d = value : d; UpdateProjection(); }
		}

		public double F
		{
			get { return f; }
			set { f = value; UpdateProjection(); }
		}

		public double[] CameraPos
		{
			get { return d_w; }
			set { d_w = value; UpdateProjection(); }
		}

		public double Azimuth
		{
			get { return azimuth; }
			set { azimuth = value; UpdateProjection(); }
		}

		public double Elevation
		{
			get { return elevation; }
			set { elevation = value; UpdateProjection(); }
		}

		public ScatterPlot()
		{
			InitializeComponent();
			MouseWheelHandler.Add(this, MyOnMouseWheel);
		}

		protected override CreateParams CreateParams
		{
			get
			{
				var cp = base.CreateParams;
				cp.ExStyle |= 0x02000000;    // Turn on WS_EX_COMPOSITED
				return cp;
			}
		}

		private Color[] colorIdx = new Color[] { Color.Blue, Color.Red, Color.Green, Color.Orange, Color.Fuchsia, Color.Black };

		protected override void OnPaint(PaintEventArgs e)
		{
			base.OnPaint(e);

			Graphics g = this.CreateGraphics();
			g.FillRectangle(Brushes.White, new Rectangle(0, 0, this.Width, this.Height));
			if (ProjPoints != null)
			{
				for (int i = 0; i < ProjPoints.Count; i++)
				{
					foreach (PointF p in ProjPoints[i])
					{
						g.FillEllipse(new SolidBrush(colorIdx[i % colorIdx.Length]), new RectangleF(p.X, p.Y, 4, 4));
					}
				}
			}
		}

		public void AddPoint(double x, double y, double z, int series)
		{
			if (Points.Count - 1 < series)
			{
				Points.Add(new List<double[]>());
			}

			Points[series].Add(new double[] { x, y, z });

			foreach (List<double[]> ser in Points)
			{
				if (ProjPoints.Count - 1 < series)
					ProjPoints.Add(Projection.ProjectVector(ser, this.Width, this.Height, f, d_w, azimuth, elevation));
				else
					ProjPoints[series] = Projection.ProjectVector(ser, this.Width, this.Height, f, d_w, azimuth, elevation);
			}
			this.Invalidate();
		}

		public void AddPoints(List<double[]> points)
		{
			List<double[]> _tmp = new List<double[]>(points);
			Points.Add(_tmp);
			ProjPoints.Add(Projection.ProjectVector(Points[Points.Count - 1], this.Width, this.Height, f, d_w, azimuth, elevation));
			UpdateProjection();
		}

		public void Clear()
		{
			ProjPoints.Clear();
			Points.Clear();
			Azimuth = 0;
			Elevation = 0;
		}

		private void ScatterPlot_MouseMove(object sender, MouseEventArgs e)
		{
			if (leftMousePressed)
			{
				azimuth = last_azimuth - (ptMouseClick.X - e.X) / 100;
				elevation = last_elevation + (ptMouseClick.Y - e.Y) / 100;
				UpdateProjection();
			}
		}

		private void ScatterPlot_SizeChanged(object sender, EventArgs e)
		{
			if (ProjPoints != null)
				UpdateProjection();
		}

		private void ScatterPlot_MouseDown(object sender, MouseEventArgs e)
		{
			if (e.Button == System.Windows.Forms.MouseButtons.Left)
			{
				leftMousePressed = true;
				ptMouseClick = new PointF(e.X, e.Y);
				last_azimuth = azimuth;
				last_elevation = elevation;
			}
		}

		private void ScatterPlot_MouseUp(object sender, MouseEventArgs e)
		{
			if (e.Button == System.Windows.Forms.MouseButtons.Left)
				leftMousePressed = false;
		}

		private void MyOnMouseWheel(MouseEventArgs e)
		{
			Distance += -e.Delta / 500D;
		}

		private void UpdateProjection()
		{
			if (ProjPoints == null)
				return;
			double x = d * Math.Cos(elevation) * Math.Cos(azimuth);
			double y = d * Math.Cos(elevation) * Math.Sin(azimuth);
			double z = d * Math.Sin(elevation);
			d_w = new double[3] { -y, z, -x };
			for (int i = 0; i < ProjPoints.Count; i++)
				ProjPoints[i] = Projection.ProjectVector(Points[i], this.Width, this.Height, f, d_w, azimuth, elevation);
			this.Invalidate();
		}
	}
}