diff --git a/src/main/java/inf112/fiasko/roborally/game_wrapper/ScreenManager.java b/src/main/java/inf112/fiasko/roborally/game_wrapper/ScreenManager.java index 50e0fc0..0091314 100644 --- a/src/main/java/inf112/fiasko/roborally/game_wrapper/ScreenManager.java +++ b/src/main/java/inf112/fiasko/roborally/game_wrapper/ScreenManager.java @@ -12,6 +12,7 @@ public class ScreenManager { private StartMenuScreen startMenuScreen; private UsernameScreen usernameScreen; private IPAddressScreen ipAddressScreen; + private LobbyScreen lobbyScreen; /** * Gets an instance of the main menu screen @@ -24,6 +25,12 @@ public class ScreenManager { } return powerDownScreen; } + public synchronized LobbyScreen getLobbyScreen(RoboRallyWrapper roboRallyWrapper) { + if (this.lobbyScreen == null) { + this.lobbyScreen = new LobbyScreen(roboRallyWrapper); + } + return lobbyScreen; + } public synchronized IPAddressScreen getIPAddressScreen(RoboRallyWrapper roboRallyWrapper) { if (this.ipAddressScreen == null) {