CommonSelectorLogic for ActorSelectorLogic and TileSelectionLogic

OpenRA.Mods.Common/Widgets/Logic/Editor/CommonSelectorLogic.cs

@@ -0,0 +1,91 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2017 The OpenRA Developers (see AUTHORS)
+ * This file is part of OpenRA, which is free software. It is made
+ * available to you 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. For more
+ * information, see COPYING.
+ */
+#endregion
+
+using System;
+using System.Collections.Generic;
+using OpenRA.Graphics;
+using OpenRA.Widgets;
+
+namespace OpenRA.Mods.Common.Widgets.Logic
+{
+	public abstract class CommonSelectorLogic : ChromeLogic
+	{
+		protected readonly Widget Widget;
+		protected readonly World World;
+		protected readonly WorldRenderer WorldRenderer;
+		protected readonly EditorViewportControllerWidget Editor;
+
+		protected readonly HashSet<string> SelectedCategories = new HashSet<string>();
+		protected readonly List<string> FilteredCategories = new List<string>();
+
+		protected string[] allCategories;
+		protected string searchFilter;
+
+		public CommonSelectorLogic(Widget widget, World world, WorldRenderer worldRenderer)
+		{
+			this.Widget = widget;
+			this.World = world;
+			this.WorldRenderer = worldRenderer;
+			Editor = widget.Parent.Get<EditorViewportControllerWidget>("MAP_EDITOR");
+		}
+
+		protected Widget CreateCategoriesPanel(ScrollPanelWidget panel)
+		{
+			var categoriesPanel = Ui.LoadWidget("ACTOR_CATEGORY_FILTER_PANEL", null, new WidgetArgs());
+			var categoryTemplate = categoriesPanel.Get<CheckboxWidget>("CATEGORY_TEMPLATE");
+
+			var selectButtons = categoriesPanel.Get<ContainerWidget>("SELECT_CATEGORIES_BUTTONS");
+			categoriesPanel.AddChild(selectButtons);
+
+			var selectAll = selectButtons.Get<ButtonWidget>("SELECT_ALL");
+			selectAll.OnClick = () =>
+			{
+				SelectedCategories.Clear();
+				foreach (var c in allCategories)
+					SelectedCategories.Add(c);
+
+				InitializePreviews();
+			};
+
+			var selectNone = selectButtons.Get<ButtonWidget>("SELECT_NONE");
+			selectNone.OnClick = () =>
+			{
+				SelectedCategories.Clear();
+				InitializePreviews();
+			};
+
+			var categoryHeight = 5 + selectButtons.Bounds.Height;
+			foreach (var cat in FilteredCategories)
+			{
+				var category = (CheckboxWidget)categoryTemplate.Clone();
+				category.GetText = () => cat;
+				category.IsChecked = () => SelectedCategories.Contains(cat);
+				category.IsVisible = () => true;
+				category.OnClick = () =>
+				{
+					if (!SelectedCategories.Remove(cat))
+						SelectedCategories.Add(cat);
+
+					InitializePreviews();
+				};
+
+				categoriesPanel.AddChild(category);
+				categoryHeight += categoryTemplate.Bounds.Height;
+			}
+
+			categoriesPanel.Bounds.Height = Math.Min(categoryHeight, panel.Bounds.Height);
+
+			return categoriesPanel;
+		}
+
+		protected abstract void InitializePreviews();
+	}
+}