Make menu to state machine

src/MenuManager.h

@@ -1,4 +1,4 @@
-#pragma once
+#pragma once
 #include "render/Renderer.h"
 #include "Environment.h"
 #include "render/TextureManager.h"
@@ -7,37 +7,58 @@
 namespace ZL {
 
 	extern const char* CONST_ZIP_FILE;
-	//extern bool g_exitBgAnimating;
 
-	class MenuManager
-	{
+	enum class GameState {
+		MainMenu,
+		ShipSelectionSingle,
+		ShipSelectionMulti,
+		Gameplay,
+		GameOver,
+		ConnectionLost
+	};
+
+	class MenuManager {
 	protected:
 		Renderer& renderer;
-		std::shared_ptr<UiNode> uiSavedRoot;
-		std::shared_ptr<UiNode> gameOverSavedRoot;
-		std::shared_ptr<UiNode> settingsSavedRoot;
-		std::shared_ptr<UiNode> multiplayerSavedRoot;
+
+		// Pre-loaded UI roots (loaded once in setupMenu)
+		std::shared_ptr<UiNode> mainMenuRoot;
+		std::shared_ptr<UiNode> shipSelectionRoot;
+		std::shared_ptr<UiNode> gameplayRoot;
+		std::shared_ptr<UiNode> gameOverRoot;
+		std::shared_ptr<UiNode> connectionLostRoot;
+
+		GameState state = GameState::MainMenu;
+
+		// State transition methods
+		void enterMainMenu();
+		void enterShipSelectionSingle();
+		void enterShipSelectionMulti();
+		void enterGameplay();
+		void enterGameOver(int score);
+		void enterConnectionLost();
 
 	public:
-		bool uiGameOverShown = false;
-		bool g_exitBgAnimating = false;
-
 		UiManager uiManager;
 
 		MenuManager(Renderer& iRenderer);
 
 		void setupMenu();
 
-		//void showGameOver();
-		void showGameOver(int score);
+		// Returns true for states where Space should render and run (Gameplay, GameOver, ConnectionLost)
+		bool shouldRenderSpace() const;
+		GameState getState() const { return state; }
 
+		// Called by game events
+		void showGameOver(int score);
+		void showConnectionLost();
+
+		// Callbacks set by Game/Space
 		std::function<void()> onRestartPressed;
 		std::function<void(float)> onVelocityChanged;
 		std::function<void()> onFirePressed;
-
 		std::function<void(const std::string&, int)> onSingleplayerPressed;
 		std::function<void(const std::string&, int)> onMultiplayerPressed;
-
 	};
-};
 
+} // namespace ZL