UltimateZoneBuilder/Source/Core/Controls/ThingBrowserControl.cs

#region ================== Copyright (c) 2007 Pascal vd Heiden

/*
 * Copyright (c) 2007 Pascal vd Heiden, www.codeimp.com
 * This program is released under GNU General Public License
 * 
 * This program 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.
 * 
 */

#endregion

#region ================== Namespaces

using System;
using System.Collections.Generic;
using System.Windows.Forms;
using CodeImp.DoomBuilder.Config;

#endregion

namespace CodeImp.DoomBuilder.Controls
{
	public partial class ThingBrowserControl : UserControl
	{
		#region ================== Events

		public delegate void TypeChangedDeletegate(ThingTypeInfo value);
		public delegate void TypeDoubleClickDeletegate();

		public event TypeChangedDeletegate OnTypeChanged;
		public event TypeDoubleClickDeletegate OnTypeDoubleClicked;

		#endregion

		#region ================== Variables

		private List<TreeNode> nodes;
		private ThingTypeInfo thinginfo;
		private bool doupdatenode;
		private bool doupdatetextbox;
		
		#endregion

		#region ================== Properties

		public string TypeStringValue { get { return typeid.Text; } }
		public bool UseMultiSelection { get { return typelist.UseMultiSelection; } set { typelist.UseMultiSelection = value; } }

		#endregion

		#region ================== Constructor

		// Constructor
		public ThingBrowserControl()
		{
			InitializeComponent();
		}

		// This sets up the control
		public void Setup()
		{
			// Go for all predefined categories
			typelist.Nodes.Clear();
			nodes = new List<TreeNode>();
			foreach(ThingCategory tc in General.Map.Data.ThingCategories)
			{
				// Create category
				TreeNode cn = typelist.Nodes.Add(tc.Name, tc.Title);
				if((tc.Color >= 0) && (tc.Color < thingimages.Images.Count)) cn.ImageIndex = tc.Color;
				cn.SelectedImageIndex = cn.ImageIndex;
				foreach(ThingTypeInfo ti in tc.Things)
				{
					// Create thing
					TreeNode n = cn.Nodes.Add(ti.Title);
					if((ti.Color >= 0) && (ti.Color < thingimages.Images.Count)) n.ImageIndex = ti.Color;
					n.SelectedImageIndex = n.ImageIndex;
					n.Tag = ti;
					nodes.Add(n);
				}
			}

			doupdatenode = true;
			doupdatetextbox = true;
		}

		#endregion
		
		#region ================== Methods

		// Select a type
		public void SelectType(int type)
		{
			// Set type index
			typeid.Text = type.ToString();
			typeid_TextChanged(this, EventArgs.Empty);
		}

		// Return selected type info
		public ThingTypeInfo GetSelectedInfo()
		{
			return thinginfo;
		}

		// This clears the type
		public void ClearSelectedType()
		{
			doupdatenode = false;

			// Clear selection
			typelist.ClearSelectedNodes(); //mxd
			typeid.Text = "";

			// Collapse nodes
			foreach(TreeNode n in nodes)
				if(n.Parent.IsExpanded) n.Parent.Collapse();
			
			doupdatenode = true;
		}

		// Result
		public int GetResult(int original)
		{
			//mxd
			if(typelist.UseMultiSelection && typelist.SelectedNodes.Count > 1) {
				List<TreeNode> validNodes = getValidNodes();

				//get a random ThingTypeInfo from valid nodes
				if(validNodes.Count > 1) {
					ThingTypeInfo ti = validNodes[General.Random(0, validNodes.Count - 1)].Tag as ThingTypeInfo;
					return ti.Index;
				}
			}
			
			return typeid.GetResult(original);
		}

		//mxd
		private List<TreeNode> getValidNodes() 
		{
			List<TreeNode> validNodes = new List<TreeNode>();

			foreach(TreeNode n in typelist.SelectedNodes) {
				if((n.Nodes.Count == 0) && (n.Tag is ThingTypeInfo))
					validNodes.Add(n);
			}

			return validNodes;
		}

		#endregion

		#region ================== Events

		// List double-clicked (mxd)
		private void typelist_MouseDoubleClick(object sender, MouseEventArgs e) 
		{
			var info = typelist.HitTest(e.Location);
			if(info.Node != null && info.Location == TreeViewHitTestLocations.Label) 
			{
				// Node is a child node?
				if((info.Node.Nodes.Count == 0) && (info.Node.Tag is ThingTypeInfo)) 
				{
					if((OnTypeDoubleClicked != null) && (typeid.Text.Length > 0)) OnTypeDoubleClicked();
				}
			}
		}
		
		// Thing type selection changed
		private void typelist_AfterSelect(object sender, TreeViewEventArgs e)
		{
			if(doupdatetextbox)
			{
				//mxd. Got a valid multiselection? Well, can't show any useful info about that...
				if(typelist.UseMultiSelection && typelist.SelectedNodes.Count > 1 && getValidNodes().Count > 1) {
					doupdatenode = false;
					if(!string.IsNullOrEmpty(typeid.Text)) // Event will be raised in typeid_OnTextChanged
						typeid.Text = "";
					else if(OnTypeChanged != null) // Or raise event here
						OnTypeChanged(thinginfo); 
					doupdatenode = true;
				}
				else if(typelist.SelectedNode != null) //Anything selected?
				{
					TreeNode n = typelist.SelectedNode;

					// Node is a child node?
					if((n.Nodes.Count == 0) && (n.Tag is ThingTypeInfo))
					{
						ThingTypeInfo ti = (n.Tag as ThingTypeInfo);

						// Show info
						doupdatenode = false;
						typeid.Text = ti.Index.ToString();
						doupdatenode = true;
					}
				}
			}
		}

		// Thing type index changed
		private void typeid_TextChanged(object sender, EventArgs e)
		{
			bool knownthing = false;

			// Any text?
			if(typeid.Text.Length > 0)
			{
				// Get the info
				int typeindex = typeid.GetResult(0);
				thinginfo = General.Map.Data.GetThingInfoEx(typeindex);
				if(thinginfo != null)
				{
					knownthing = true;

					// Size
					sizelabel.Text = (thinginfo.Radius * 2) + " x " + thinginfo.Height;

					// Hangs from ceiling
					if(thinginfo.Hangs) positionlabel.Text = "Ceiling"; else positionlabel.Text = "Floor";

					// Blocking
					switch(thinginfo.Blocking)
					{
						case ThingTypeInfo.THING_BLOCKING_NONE: blockinglabel.Text = "No"; break;
						case ThingTypeInfo.THING_BLOCKING_FULL: blockinglabel.Text = "Completely"; break;
						case ThingTypeInfo.THING_BLOCKING_HEIGHT: blockinglabel.Text = "True-Height"; break;
						default: blockinglabel.Text = "Unknown"; break;
					}
				}

				if(doupdatenode)
				{
					doupdatetextbox = false;
					typelist.ClearSelectedNodes();
					foreach(TreeNode n in nodes)
					{
						// Matching node?
						if((n.Tag as ThingTypeInfo).Index == typeindex)
						{
							// Select this
							if(typelist.Nodes.Contains(n.Parent)) { //mxd. Tree node may've been removed during filtering
								n.Parent.Expand();
								typelist.SelectedNode = n;
								n.EnsureVisible();
							}
						}
					}
					doupdatetextbox = true;
				}
			}
			else
			{
				thinginfo = null;
				if(doupdatenode) typelist.ClearSelectedNodes();
			}

			// No known thing?
			if(!knownthing)
			{
				sizelabel.Text = "-";
				positionlabel.Text = "-";
				blockinglabel.Text = "-";
			}

			// Raise event
			if(OnTypeChanged != null) OnTypeChanged(thinginfo);
		}

		// Layout update!
		private void ThingBrowserControl_Layout(object sender, LayoutEventArgs e)
		{
			ThingBrowserControl_SizeChanged(sender, EventArgs.Empty);
		}

		private void ThingBrowserControl_Resize(object sender, EventArgs e)
		{
			ThingBrowserControl_SizeChanged(sender, EventArgs.Empty);
		}

		private void ThingBrowserControl_SizeChanged(object sender, EventArgs e)
		{
			blockingcaption.Left = infopanel.Width / 2;
			blockinglabel.Left = blockingcaption.Right + blockingcaption.Margin.Right;
			sizecaption.Left = blockingcaption.Right - sizecaption.Width;
			sizelabel.Left = sizecaption.Right + sizecaption.Margin.Right;
		}

		//mxd
		private void typelist_MouseEnter(object sender, EventArgs e) {
			typelist.Focus();
		}

		//mxd
		private void bClear_Click(object sender, EventArgs e) {
			tbFilter.Clear();
		}

		//mxd
		private void tbFilter_TextChanged(object sender, EventArgs e) {
			typelist.SuspendLayout();

			if(string.IsNullOrEmpty(tbFilter.Text)) {
				Setup();
				typeid_TextChanged(this, EventArgs.Empty);
			} else {
				// Go for all predefined categories
				typelist.ClearSelectedNodes();
				typelist.Nodes.Clear();

				string match = tbFilter.Text.ToUpperInvariant();
				foreach(TreeNode node in nodes){
					if(node.Text.ToUpperInvariant().Contains(match)) {
						typelist.Nodes.Add(node);
					}
				}

				doupdatenode = true;
				doupdatetextbox = true;
			}

			typelist.ResumeLayout();
		}
		
		#endregion
	}
}