CtlInheritance.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Drawing;
using System.Data;
using System.Text;
using System.Windows.Forms;

using IfcDoc.Schema.DOC;
using IfcDoc.Format.PNG;

namespace IfcDoc
{
	public partial class CtlInheritance : ScrollableControl
	{
		private DocProject m_project;
		private DocModelView m_modelview;
		private DocEntity m_entity;
		private DocEntity m_highlight;
		private DocEntity m_selection;
		private ToolMode m_mode;
		private Image m_image;
		private Dictionary<Rectangle, DocEntity> m_hitmap;
		private Rectangle m_rcHighlight;
		private Rectangle m_rcSelection;

		public CtlInheritance()
		{
			InitializeComponent();

			this.DoubleBuffered = true;
			this.ResizeRedraw = true;
		}

		public DocProject Project
		{
			get
			{
				return this.m_project;
			}
			set
			{
				this.m_project = value;
			}
		}

		public DocModelView ModelView
		{
			get
			{
				return this.m_modelview;
			}
			set
			{
				this.m_modelview = value;
			}
		}

		public DocEntity Entity
		{
			get
			{
				return this.m_entity;
			}
			set
			{
				this.m_entity = value;
				this.m_selection = null;
				this.m_highlight = null;
				this.Render();
			}
		}

		public ToolMode Mode
		{
			get
			{
				return this.m_mode;
			}
			set
			{
				this.m_mode = value;
			}
		}

		public DocEntity Selection
		{
			get
			{
				return this.m_selection;
			}
		}

		public event EventHandler SelectionChanged;

		public void Render()
		{
			if (this.m_image != null)
			{
				this.m_image.Dispose();
				this.m_image = null;
			}

			Dictionary<DocObject, bool> included = null;
			if (this.m_modelview != null)
			{
				included = new Dictionary<DocObject, bool>();
				this.m_project.RegisterObjectsInScope(this.m_modelview, included);
			}

			this.m_hitmap = new Dictionary<Rectangle, DocEntity>();
			this.m_image = FormatPNG.CreateInheritanceDiagram(this.m_project, included, this.m_entity, null, this.Font, this.m_hitmap);
			if (this.m_image != null)
			{
				this.AutoScrollMinSize = new Size(this.m_image.Width, this.m_image.Height);
			}
			this.Invalidate();
		}

		protected override void OnPaint(PaintEventArgs pe)
		{
			if (this.m_image == null)
				return;

			Graphics g = pe.Graphics;
			g.TranslateTransform(this.AutoScrollPosition.X, this.AutoScrollPosition.Y);
			g.DrawImage(this.m_image, Point.Empty);

			if (this.m_highlight != null)
			{
				g.DrawRectangle(Pens.Red, this.m_rcHighlight);
			}
		}

		private DocEntity Pick(Point pt, out Rectangle rectangle)
		{
			rectangle = Rectangle.Empty;
			pt.X -= this.AutoScrollPosition.X;
			pt.Y -= this.AutoScrollPosition.Y;

			foreach (Rectangle rc in this.m_hitmap.Keys)
			{
				if (rc.Contains(pt))
				{
					rectangle = rc;
					DocEntity docEnt = this.m_hitmap[rc];
					return docEnt;
				}
			}

			return null;
		}

		private void CtlInheritance_MouseDown(object sender, MouseEventArgs e)
		{
			this.m_selection = this.Pick(e.Location, out this.m_rcSelection);
			switch (this.m_mode)
			{
				case ToolMode.Select:
					if (this.SelectionChanged != null)
					{
						this.SelectionChanged(this, EventArgs.Empty);
					}
					break;

				case ToolMode.Move:
					if (this.m_selection != null)
					{
						this.m_selection._InheritanceDiagramFlag = !this.m_selection._InheritanceDiagramFlag;
						this.m_selection.Status = (this.m_selection._InheritanceDiagramFlag ? "H" : String.Empty);
						this.Render();
					}
					break;
			}
		}

		private void CtlInheritance_MouseMove(object sender, MouseEventArgs e)
		{
			this.m_highlight = this.Pick(e.Location, out this.m_rcHighlight);
			switch (this.m_mode)
			{
				case ToolMode.Select:
					if (m_highlight != null)
					{
						this.Cursor = Cursors.Hand;
					}
					else
					{
						this.Cursor = Cursors.Default;
					}
					break;

				case ToolMode.Move:
					if (m_highlight != null)
					{
						this.Cursor = Cursors.UpArrow;
					}
					else
					{
						this.Cursor = Cursors.Default;
					}
					break;
			}

			this.Invalidate();
		}

		private void CtlInheritance_MouseUp(object sender, MouseEventArgs e)
		{

		}
	}
}