Update tab management system

1 files changed, 5 insertions, 4 deletions

diff --git a/src/MainWindow.cpp b/src/MainWindow.cpp
index 2243f86..8b6e0fe 100644
--- a/src/MainWindow.cpp
+++ b/src/MainWindow.cpp
@@ -191,9 +191,9 @@ void MainWindow::NewGame(bool useFen) {
 }
 
 void MainWindow::OnNewGame(wxCommandEvent &event) {
-  std::shared_ptr<Game> *g = (std::shared_ptr<Game>*)event.GetClientData();
-  NewGame(*g);
-  delete g;
+  TabInfos *tab = (TabInfos*)event.GetClientData();
+  TabInfos *i=NewGame(tab->GetGame());
+  i->Link(tab);
 }
 
 void MainWindow::OnPageChange(wxAuiNotebookEvent &event) {
@@ -215,8 +215,9 @@ void MainWindow::OnRefreshTabTitle(wxCommandEvent &event) {
   }
 }
 
-void MainWindow::NewGame(std::shared_ptr<Game> game) {
+TabInfos* MainWindow::NewGame(std::shared_ptr<Game> game) {
   GameTab *gt = new GameTab((wxFrame *)notebook, game);
   notebook->AddPage(gt, gt->GetLabel());
   notebook->SetSelection(notebook->GetPageIndex(gt));
+  return(gt);
 }