javaFx: get stage from controller during initialization

FXMLLoader loader = new FXMLLoader(getClass().getResource("MyGui.fxml"));
Parent root = loader.load();
MyController controller = (MyController)loader.getController();
controller.setStageAndSetupListeners(stage);