From 36ac2f0cf54740dc7055e08496385625aa28f128 Mon Sep 17 00:00:00 2001
From: NeeEoo <NeeEoo@users.noreply.github.com>
Date: Tue, 16 Apr 2024 13:02:02 +0200
Subject: [PATCH] Added selection helpers

---
 source/funkin/editors/stage/StageEditor.hx | 120 +++++++++++++++++----
 1 file changed, 98 insertions(+), 22 deletions(-)

diff --git a/source/funkin/editors/stage/StageEditor.hx b/source/funkin/editors/stage/StageEditor.hx
index 463accdb..b552d415 100644
--- a/source/funkin/editors/stage/StageEditor.hx
+++ b/source/funkin/editors/stage/StageEditor.hx
@@ -100,6 +100,42 @@ class StageEditor extends UIState {
 					}
 				]
 			},
+			/*{
+				label: "Edit",
+				childs: [
+					{
+						label: "Undo",
+						keybind: [CONTROL, Z],
+						onSelect: _edit_undo,
+					},
+					{
+						label: "Redo",
+						keybind: [CONTROL, SHIFT, Z],
+						onSelect: _edit_redo,
+					},
+					null,
+				]
+			},*/
+			{
+				label: "Select",
+				childs: [
+					{
+						label: "All",
+						keybind: [CONTROL, A],
+						onSelect: (_) -> _select_all(_),
+					},
+					{
+						label: "Deselect",
+						keybind: [CONTROL, D],
+						onSelect: (_) -> _select_deselect(_),
+					},
+					{
+						label: "Inverse",
+						keybind: [CONTROL, SHIFT, I],
+						onSelect: (_) -> _select_inverse(_),
+					},
+				]
+			},
 			{
 				label: "View",
 				childs: [
@@ -381,9 +417,12 @@ class StageEditor extends UIState {
 				}
 			}*/
 
-			for(sprite in selection) {
-				if(sprite is FunkinSprite) {
-					handleSelection(cast sprite);
+			// TODO: make this work with multiple selections
+			if(selection.length == 1) {
+				for(sprite in selection) {
+					if(sprite is FunkinSprite) {
+						handleSelection(cast sprite);
+					}
 				}
 			}
 
@@ -410,25 +449,6 @@ class StageEditor extends UIState {
 		SaveWarning.showWarning = undos.unsaved;
 	}
 
-	public function selectSprite(_sprite:FunkinSprite) {
-		selection = new Selection([]);
-		var sprites = stageSpritesWindow.buttons.members.map((o) -> o.getSprite()).filter((o) -> o != null);// && o.animateAtlas == null);
-		for(sprite in sprites) {
-			if(sprite is FunkinSprite) {
-				var sprite:FunkinSprite = cast sprite;
-				sprite.extra.set(exID("selected"), false);
-				sprite.extra.get(exID("button")).selected = false;
-			}
-		}
-		if(_sprite is FunkinSprite) {
-			var sprite:FunkinSprite = cast _sprite;
-			sprite.extra.set(exID("selected"), true);
-			sprite.extra.get(exID("button")).selected = true;
-			selection = new Selection([sprite]);
-			Logs.trace("Selected " + sprite.name);
-		}
-	}
-
 	// TOP MENU OPTIONS
 	#if REGION
 	function _file_exit(_) {
@@ -542,6 +562,62 @@ class StageEditor extends UIState {
 			undos.addToUndo(CEditInfo(oldInfo, newInfo));*/
 	}
 
+	public function selectSprite(_sprite:FunkinSprite) {
+		_select_deselect(null, false);
+
+		if(_sprite is FunkinSprite) {
+			selection = new Selection([cast _sprite]);
+		}
+		updateSelection();
+	}
+
+	function getSprites() {
+		return stageSpritesWindow.buttons.members.map((o) -> o.getSprite()).filter((o) -> o != null);
+	}
+
+	function updateSelection() {
+		var sprites = getSprites();
+		// Unselect all
+		for(sprite in sprites) {
+			if(sprite is FunkinSprite) {
+				var sprite:FunkinSprite = cast sprite;
+				sprite.extra.set(exID("selected"), false);
+				sprite.extra.get(exID("button")).selected = false;
+			}
+		}
+		// Mark selected as selected
+		for(sprite in selection) {
+			if(sprite is FunkinSprite) {
+				var sprite:FunkinSprite = cast sprite;
+				sprite.extra.set(exID("selected"), true);
+				sprite.extra.get(exID("button")).selected = true;
+				Logs.trace("Selected " + sprite.name);
+			}
+		}
+	}
+
+	function _select_all(_, checkSelection:Bool = true) {
+		_select_deselect(null, false);
+		var sprites = getSprites();
+		selection = new Selection(sprites);
+		if(checkSelection) updateSelection();
+	}
+
+	function _select_deselect(_, checkSelection:Bool = true) {
+		var sprites = getSprites();// && o.animateAtlas == null);
+		selection = new Selection();
+		if(checkSelection) updateSelection();
+	}
+
+	function _select_inverse(_, checkSelection:Bool = true) {
+		var oldSelection = selection;
+		_select_all(null, false);
+		for(sprite in oldSelection) {
+			selection.remove(sprite);
+		}
+		if(checkSelection) updateSelection();
+	}
+
 	var zoom(default, set):Float = 0;
 	var __camZoom(default, set):Float = 1;
 	function set_zoom(val:Float) {