//
// Created by maximilian on 7/13/23.
//

#include <boost/rational.hpp>
#include <cstdint>
#include <iostream>
#include <optional>
#include <vector>

#include "game_state.h"
#include "download.h"
#include "myassert.h"


namespace Hanabi {

void test_game() {
  HanabiState<2, 2, 5> state;
  state._stacks[0] = 2;
  state._stacks[1] = 3;
  Card r41 = {0, 4, 1};
  state._draw_pile.push_back({r41, 1});
  state._hands[0] = {y0, y1, y2, r0, r1};
  state._hands[1] = {r1, r1, y1, r3, r2};
//  state._card_positions[r1] = 0;
  state._weighted_draw_pile_size = 1;

  auto state2 = state;

  auto a = state.play(4);
  std::cout << state;
  state.revert_play(a);

  std::cout << state << std::endl;
  std::cout << state2 << std::endl;
  ASSERT(state._hands == state2._hands);
  ASSERT(state._draw_pile == state2._draw_pile);
//  ASSERT(state._card_positions == state2._card_positions);
  ASSERT(state == state2);
}

void download(int turn) {
    auto game = Download::get_game<6,5,4>("996518", turn);
    std::cout << "Analysing state: " << game << std::endl;
    auto res = game.backtrack(1);
    std::cout << "Probability with optimal play: " << res << std::endl;
    std::cout << "Enumerated " << game._enumerated_states << " states" << std::endl;
}

void print_sizes() {
  std::cout << "size of card -> hand map: " << sizeof(HanabiState<5, 3, 4>)
            << std::endl;

  unsigned exp = 32;
  std::cout << "Pair size: " << sizeof(std::pair<std::uint32_t, float>)
            << std::endl;
  std::cout << sizeof(boost::rational<int>) << std::endl;
  std::cout << (1ul << exp) << std::endl;
}

}

int main(int argc, char *argv[]) {
//  Hanabi::test_game();
  if(argc == 2) {
      std::string turn (argv[1]);
      Hanabi::download(std::stoi(turn));
  }

  return 0;
}