Undo shortcut

kaczy93 · Nov 30, 2023 · 87d675d · 87d675d
1 parent 0a16ed1
commit 87d675d
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 4 deletions.
diff --git a/CentrED/Map/MapManager.cs b/CentrED/Map/MapManager.cs
@@ -363,6 +363,7 @@ public void RemoveTile(StaticTile staticTile)
     }
 
     private MouseState _prevMouseState = Mouse.GetState();
+    private KeyboardState _prevKeyState = Keyboard.GetState();
     public Rectangle ViewRange { get; private set; }
 
     public void Update(GameTime gameTime, bool isActive, bool processMouse, bool processKeyboard)
@@ -450,9 +451,9 @@ public void Update(GameTime gameTime, bool isActive, bool processMouse, bool pro
 
         if (isActive && processKeyboard)
         {
-            var keyboard = Keyboard.GetState();
+            var keyState = Keyboard.GetState();
 
-            foreach (var key in keyboard.GetPressedKeys())
+            foreach (var key in keyState.GetPressedKeys())
             {
                 switch (key)
                 {
@@ -474,6 +475,11 @@ public void Update(GameTime gameTime, bool isActive, bool processMouse, bool pro
                         break;
                 }
             }
+            if(keyState.IsKeyDown(Keys.LeftControl) && keyState.IsKeyDown(Keys.Z) && _prevKeyState.IsKeyUp(Keys.Z))
+            {
+                Client.Undo();
+            }
+            _prevKeyState = keyState;
         }
 
         Camera.Update();

diff --git a/CentrED/UI/UIManager.cs b/CentrED/UI/UIManager.cs
@@ -264,7 +264,14 @@ private void DrawMainMenu()
                     CEDGame.Exit();
                 ImGui.EndMenu();
             }
-
+            if (ImGui.BeginMenu("Edit"))
+            {
+                if (ImGui.MenuItem("Undo", "Ctrl+Z"))
+                {
+                    CEDClient.Undo();
+                }
+                ImGui.EndMenu();
+            }
             if (ImGui.BeginMenu("Tools"))
             {
                 ToolsWindows.ForEach(w => w.DrawMenuItem());

diff --git a/Client/CentrEDClient.cs b/Client/CentrEDClient.cs
@@ -20,7 +20,7 @@ public sealed class CentrEDClient : BaseCentrED, IDisposable
     public AccessLevel AccessLevel { get; internal set; }
     public ushort X { get; private set; }
     public ushort Y { get; private set; }
-    public Stack<Packet> UndoStack = new();
+    internal Stack<Packet> UndoStack = new();
     public List<String> Clients { get; } = new();
     public bool Running;
     private string? _status;