Initial multiselect checkpoint, still long way to go

2024-06-14 14:08:08 -07:00
parent 4c7e465098
commit 0bf508df1f
6 changed files with 595 additions and 0 deletions
--- a/MultiSelect.cs
+++ b/MultiSelect.cs
@@ -0,0 +1,172 @@
+using EFT.InventoryLogic;
+using EFT.UI.DragAndDrop;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using TMPro;
+using UnityEngine;
+
+namespace UIFixes
+{
+    public class MultiSelect
+    {
+        private static GameObject SelectedMarkTemplate;
+        private static GameObject SelectedBackgroundTemplate;
+
+        private static readonly HashSet<GridItemView> SelectedItemViews = [];
+        private static readonly List<ItemContextClass> SelectedItemContexts = [];
+
+        public static void Initialize()
+        {
+            // Grab the selection objects from ragfair as templates
+            RagfairNewOfferItemView ragfairNewOfferItemView = ItemViewFactory.CreateFromPool<RagfairNewOfferItemView>("ragfair_layout");
+            SelectedMarkTemplate = UnityEngine.Object.Instantiate(ragfairNewOfferItemView.R().SelectedMark, null, false);
+            SelectedBackgroundTemplate = UnityEngine.Object.Instantiate(ragfairNewOfferItemView.R().SelectedBackground, null, false);
+            ragfairNewOfferItemView.ReturnToPool();
+        }
+
+        public static void Toggle(GridItemView itemView)
+        {
+            if (!itemView.IsInteractable)
+            {
+                return;
+            }
+
+            if (SelectedItemViews.Contains(itemView))
+            {
+                Deselect(itemView);
+            }
+            else
+            {
+                Select(itemView);
+            }
+        }
+
+        public static void Clear()
+        {
+            // ToList() because we'll be modifying the collection
+            foreach (GridItemView itemView in SelectedItemViews.ToList())
+            {
+                Deselect(itemView);
+            }
+        }
+
+        public static void Select(GridItemView itemView)
+        {
+            if (itemView.IsInteractable && SelectedItemViews.Add(itemView))
+            {
+                ShowSelection(itemView);
+            }
+        }
+
+        public static void Deselect(GridItemView itemView)
+        {
+            if (SelectedItemViews.Remove(itemView))
+            {
+                HideSelection(itemView);
+            }
+        }
+
+        public static IEnumerable<ItemView> ItemViews
+        {
+            get { return SelectedItemViews; }
+        }
+
+        public static IEnumerable<ItemContextClass> ItemContexts
+        {
+            get { return SelectedItemContexts; }
+        }
+
+        public static int Count
+        {
+            get { return SelectedItemViews.Count; }
+        }
+
+        public static bool Contains(GridItemView itemView)
+        {
+            return SelectedItemViews.Contains(itemView);
+        }
+
+        public static bool Active
+        {
+            get { return SelectedItemViews.Any(); }
+        }
+
+        public static bool IsSelected(GridItemView itemView)
+        {
+            return SelectedItemViews.Contains(itemView);
+        }
+
+        public static void BeginDrag()
+        {
+            foreach (ItemView itemView in SelectedItemViews)
+            {
+                SelectedItemContexts.Add(new ItemContextClass(itemView.ItemContext, itemView.ItemRotation));
+            }
+        }
+
+        public static void EndDrag()
+        {
+            foreach(ItemContextClass itemContext in SelectedItemContexts)
+            {
+                itemContext.Dispose();
+            }
+
+            SelectedItemContexts.Clear();
+        }
+
+        public static void ShowDragCount(DraggedItemView draggedItemView)
+        {
+            if (Count > 1)
+            {
+                GameObject textOverlay = new("MultiSelectText", [typeof(RectTransform), typeof(TextMeshProUGUI)]);
+                textOverlay.transform.parent = draggedItemView.transform;
+                textOverlay.transform.SetAsLastSibling();
+                textOverlay.SetActive(true);
+
+                RectTransform overlayRect = textOverlay.GetComponent<RectTransform>();
+                overlayRect.anchorMin = Vector2.zero;
+                overlayRect.anchorMax = Vector2.one;
+                overlayRect.anchoredPosition = new Vector2(0.5f, 0.5f);
+
+                TextMeshProUGUI text = textOverlay.GetComponent<TextMeshProUGUI>();
+                text.text = MultiSelect.Count.ToString();
+                text.fontSize = 36;
+                text.alignment = TextAlignmentOptions.Baseline;
+            }
+        }
+
+        private static void ShowSelection(GridItemView itemView)
+        {
+            GameObject selectedMark = itemView.transform.Find("SelectedMark")?.gameObject;
+            if (selectedMark == null)
+            {
+                selectedMark = UnityEngine.Object.Instantiate(SelectedMarkTemplate, itemView.transform, false);
+                selectedMark.name = "SelectedMark";
+            }
+
+            selectedMark.SetActive(true);
+
+            GameObject selectedBackground = itemView.transform.Find("SelectedBackground")?.gameObject;
+            if (selectedBackground == null)
+            {
+                selectedBackground = UnityEngine.Object.Instantiate(SelectedBackgroundTemplate, itemView.transform, false);
+                selectedBackground.transform.SetAsFirstSibling();
+                selectedBackground.name = "SelectedBackground";
+            }
+
+            selectedBackground.SetActive(true);
+        }
+
+        private static void HideSelection(GridItemView itemView)
+        {
+            GameObject selectedMark = itemView.transform.Find("SelectedMark")?.gameObject;
+            GameObject selectedBackground = itemView.transform.Find("SelectedBackground")?.gameObject;
+
+            selectedMark?.SetActive(false);
+            selectedBackground?.SetActive(false);
+        }
+    }
+}