game_positions.h

#pragma once
 
#include "scid/core/game_cursor.h"
#include "scid/core/notation.h"
#include "scid/core/pgn/traversal.h"
 
#include <cassert>
#include <cstdint>
#include <string>
#include <vector>
 
namespace scid::core::gamepos {
 
struct GamePos {
  uint32_t RAVdepth;
  uint32_t RAVnum;
  std::string FEN;       // "Forsyth-Edwards Notation" describing the position.
  std::vector<int> NAGs; // "Numeric Annotation Glyph"
  std::string comment;   // text annotation of the position.
  std::string lastMoveSAN; // move that was played to reach the position.
};
 
template <typename TCont>
inline void collectPositions(const scid::core::Game &game, TCont &dest) {
  scid::core::GameCursor cursor(game);
  do {
    if (cursor.isAtVariationStart() && !cursor.isAtGameStart())
      continue;
 
    dest.emplace_back();
    auto &gamepos = dest.back();
    char strBuf[256];
    auto position = cursor.currentPosition();
    assert(position);
    position->PrintFEN(strBuf, sizeof(strBuf));
    gamepos.FEN = strBuf;
    gamepos.RAVdepth = cursor.variationDepth();
    gamepos.RAVnum = cursor.variationIndex();
    if (auto move = cursor.previousMove()) {
      for (auto nag : move->metadata.nags)
        gamepos.NAGs.push_back(scid::core::nagCode(nag));
      gamepos.comment = move->metadata.comment;
    } else if (auto variation = cursor.currentVariation()) {
      gamepos.comment = variation->initialComment;
    } else {
      gamepos.comment = game.movetext().initialComment;
    }
    gamepos.lastMoveSAN =
        scid::core::notation::previousSan(game, cursor.location());
 
  } while (scid::core::pgn::nextLocation(cursor));
}
 
inline std::vector<GamePos> collectPositions(const scid::core::Game &game) {
  std::vector<GamePos> res;
  collectPositions(game, res);
  return res;
}
 
} // namespace scid::core::gamepos