Improve UI

1 files changed, 16 insertions, 0 deletions

diff --git a/src/game_tab/right_panel/GameTabRightPanel.cpp b/src/game_tab/right_panel/GameTabRightPanel.cpp
index 99ab4fd..013bc13 100644
--- a/src/game_tab/right_panel/GameTabRightPanel.cpp
+++ b/src/game_tab/right_panel/GameTabRightPanel.cpp
@@ -40,6 +40,13 @@ GameTabRightPanel::GameTabRightPanel(wxFrame *parent, std::shared_ptr<Game> game
   this->Bind(wxEVT_BUTTON, &GameTabRightPanel::OnDelete, this, DELETE_BTN);
   this->Bind(wxEVT_BUTTON, &GameTabRightPanel::OnLiveAnalysis, this,
              LIVE_ANALYSIS_GAME_BUTTON);
+  nag_panel->Bind(wxEVT_BUTTON, [p=this](wxCommandEvent &e){
+    HalfMove *m = p->game->GetCurrentMove();
+    if (m != NULL) {
+      m->nag=p->GetNagFromStr(((wxButton*)e.GetEventObject())->GetLabel().ToStdString());
+      p->editor_canvas->Refresh();
+    }
+  });
 
   ApplyPreferences();
 }
@@ -204,4 +211,13 @@ void GameTabRightPanel::OnNextMove(wxCommandEvent &event) {
   game->Next();
   Notify();
   NotifyBoard();
+}
+
+std::string GameTabRightPanel::GetNagFromStr(std::string str){
+  // TODO: Bind more NAG!
+  if(str=="!")
+    return "$1";
+  if(str=="?")
+    return "$2";
+  return "";
 }
\ No newline at end of file