Improve BaseTab implementation for PGN files

2 files changed, 40 insertions, 5 deletions

diff --git a/src/base_tab/gamebase/PGNGameBase.cpp b/src/base_tab/gamebase/PGNGameBase.cpp
index 09d87d8..39775d1 100644
--- a/src/base_tab/gamebase/PGNGameBase.cpp
+++ b/src/base_tab/gamebase/PGNGameBase.cpp
@@ -1,5 +1,35 @@
 #include "PGNGameBase.hpp"
 
-PGNGameBase::PGNGameBase(std::string pgn_file) {
-    
-}
\ No newline at end of file
+PGNGameBase::PGNGameBase(std::string pgn_file)
+    : pgn(new pgnp::PGN()), hasNextGame(false) {
+  pgn->FromFile(pgn_file);
+  ParseNextGame();
+}
+
+bool PGNGameBase::HasNextGame() { return (hasNextGame); }
+
+void PGNGameBase::ParseNextGame() {
+  try {
+    pgn->ParseNextGame();
+    hasNextGame = true;
+  } catch (...) {
+    hasNextGame = false;
+  }
+}
+
+Game *PGNGameBase::GetNextGame() {
+  pgnp::HalfMove *pgnp_moves = new pgnp::HalfMove();
+  pgn->GetMoves(pgnp_moves);
+  std::string fen = "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1";
+  if (pgn->HasTag("FEN")) {
+    fen = pgn->GetTagValue("FEN");
+  }
+  HalfMove *m = new HalfMove(pgnp_moves, fen);
+  Game *g = new Game(m, fen);
+  for (std::string &s : pgn->GetTagList()) {
+    g->SetTag(s, pgn->GetTagValue(s));
+  }
+
+  ParseNextGame();
+  return (g);
+}
diff --git a/src/base_tab/gamebase/PGNGameBase.hpp b/src/base_tab/gamebase/PGNGameBase.hpp
index 2aa0840..b58bdca 100644
--- a/src/base_tab/gamebase/PGNGameBase.hpp
+++ b/src/base_tab/gamebase/PGNGameBase.hpp
@@ -1,10 +1,15 @@
 #include "GameBase.hpp"
+#include "pgnp.hpp"
 
 class PGNGameBase : public GameBase {
+    pgnp::PGN *pgn;
+    bool hasNextGame;
+
+    void ParseNextGame();
 public:
   PGNGameBase(std::string pgn_file);
 
-  bool HasNextGame() { return (false); }
+  bool HasNextGame();
   Game *GetGame(std::uint32_t id) { return (new Game()); };
-  Game *GetNextGame() { return (new Game()); };
+  Game *GetNextGame();
 };
\ No newline at end of file