diff --git a/src/main/java/seng302/visualiser/GameClient.java b/src/main/java/seng302/visualiser/GameClient.java index 028ea1e3..2a2ce3d4 100644 --- a/src/main/java/seng302/visualiser/GameClient.java +++ b/src/main/java/seng302/visualiser/GameClient.java @@ -72,7 +72,6 @@ public class GameClient { socketThread.addStreamObserver(this::parsePackets); LobbyController lobbyController = loadLobby(); - lobbyController.setPlayerListSource(clientLobbyList); lobbyController.disableReadyButton(); lobbyController.setTitle("Connected to host - IP : " + ipAddress + " Port : " + portNumber); lobbyController.addCloseListener((exitCause) -> this.loadStartScreen()); @@ -93,7 +92,6 @@ public class GameClient { } socketThread.addStreamObserver(this::parsePackets); LobbyController lobbyController = loadLobby(); - lobbyController.setPlayerListSource(clientLobbyList); lobbyController.setTitle("Hosting Lobby - IP : " + ipAddress + " Port : " + portNumber); lobbyController.addCloseListener(exitCause -> { if (exitCause == CloseStatus.READY) { @@ -133,7 +131,11 @@ public class GameClient { } catch (IOException e) { e.printStackTrace(); } - return fxmlLoader.getController(); + LobbyController lobbyController = fxmlLoader.getController(); + lobbyController.setPlayerListSource(clientLobbyList); + lobbyController.setPlayerID(socketThread.getClientId()); + + return lobbyController; } private void loadRaceView() { diff --git a/src/main/java/seng302/visualiser/controllers/LobbyController.java b/src/main/java/seng302/visualiser/controllers/LobbyController.java index 809bd3b5..113ae70e 100644 --- a/src/main/java/seng302/visualiser/controllers/LobbyController.java +++ b/src/main/java/seng302/visualiser/controllers/LobbyController.java @@ -14,6 +14,7 @@ import javafx.scene.image.ImageView; import javafx.scene.text.Text; import seng302.gameServer.GameStages; import seng302.gameServer.GameState; +import seng302.visualiser.GameView; /** * A class describing the actions of the lobby screen @@ -72,6 +73,7 @@ public class LobbyController { private List