#include <lib94/lib94.hpp>
#include <csignal>
#include <cstdio>
#include <mpi.h>

#include "constants.hpp"

static int **score_results;

static int get_result() {
  int result[3];
  MPI_Status status;
  MPI_Recv(result, 3, MPI_INT, MPI_ANY_SOURCE, 1, MPI_COMM_WORLD, &status);

  if (result[0] != -1)
    score_results[result[0]][result[1]] += result[2];

  return status.MPI_SOURCE;
}

void head_main(int comm_size, int warrior_count, const lib94::warrior *const *warriors) {
  score_results = new int *[warrior_count];
  for (int i = 0; i < warrior_count; ++i) {
    score_results[i] = new int[warrior_count];
    for (int j = 0; j < warrior_count; ++j)
      score_results[i][j] = 0;
  }

  int chunks = warrior_count * (warrior_count - 1) / 2 * CHUNKS_PER_PAIR;
  int on_chunk = 0;

  int right_name_width = 0;
  for (int i = 0; i < warrior_count; ++i)
    if ((int)warriors[i]->name.size() > right_name_width)
      right_name_width = warriors[i]->name.size();

  int rank_width = std::max(std::string("rank").size(), std::to_string(comm_size - 1).size());
  int right_round_width = std::to_string(CHUNKS_PER_PAIR * ROUNDS_PER_CHUNK).size();
  int right_chunk_width = std::to_string(chunks).size();

  int left_name_width = std::max(right_name_width, (int)std::string("match").size() - right_name_width - 4);
  int left_round_width = std::max((int)std::to_string((CHUNKS_PER_PAIR - 1) * ROUNDS_PER_CHUNK + 1).size(), (int)std::string("rounds").size() - right_round_width - 3);
  int left_chunk_width = std::max(right_chunk_width, (int)std::string("chunk").size() - right_chunk_width - 3);

  fprintf(stderr, "\x1b""7\x1b[?47h\x1b[?25l\x1b[2J\x1b[0H");
  fprintf(stderr, "%*s | %*s | %*s | %*s", rank_width, "rank",
    left_name_width + 4 + right_name_width, "match",
    left_round_width + 3 + right_round_width, "rounds",
    left_chunk_width + 3 + right_chunk_width, "chunk");

  for (int i = 0; i < warrior_count; ++i)
    for (int j = i + 1; j < warrior_count; ++j)
      for (int x = 0; x < CHUNKS_PER_PAIR; ++x) {
        ++on_chunk;
        int rank = get_result();
        int message[3] = {i, j, x * ROUNDS_PER_CHUNK};
        MPI_Send(message, 2, MPI_INT, rank, 0, MPI_COMM_WORLD);
        fprintf(stderr, "\x1b[%d;0H%*d | %*s vs %*s | %*d - %*d | %*d / %*d\x1b[0K", rank + 1, rank_width, rank,
          left_name_width, warriors[i]->name.c_str(), right_name_width, warriors[j]->name.c_str(),
          left_round_width, x * ROUNDS_PER_CHUNK + 1, right_round_width, x * ROUNDS_PER_CHUNK + ROUNDS_PER_CHUNK,
          left_chunk_width, on_chunk, right_chunk_width, chunks);
      }

  for (int i = 0; i < comm_size - 1; ++i) {
    int rank = get_result();
    int message[3] = {-1};
    MPI_Send(message, 3, MPI_INT, rank, 0, MPI_COMM_WORLD);
    fprintf(stderr, "\x1b[%d;0H%*d | %*s | %*s | %*s\x1b[0K",
      rank + 1, rank_width, rank,
      left_name_width + 4 + right_name_width, "",
      left_round_width + 3 + right_round_width, "",
      left_chunk_width + 3 + right_chunk_width, "done");
  }

  fprintf(stderr, "\x1b[?25h\x1b[?47l\x1b""8");

  for (int i = 0; i < warrior_count; ++i)
    for (int j = i + 1; j < warrior_count; ++j)
      if (score_results[i][j] < 0) {
        score_results[j][i] = -score_results[i][j];
        score_results[i][j] = 0;
      }

  int *tab_widths = new int[warrior_count + 1];
  tab_widths[0] = 0;

  for (int i = 0; i < warrior_count; ++i) {
    int len = warriors[i]->name.size();
    if (len > tab_widths[0])
      tab_widths[0] = len;
    tab_widths[i + 1] = len > 5 ? len : 5;
  }

  printf(" %*s", tab_widths[0], "");
  for (int j = 0; j < warrior_count; ++j)
    printf(" | %*s", tab_widths[j + 1], warriors[j]->name.c_str());
  putchar('\n');

  putchar('-');
  for (int x = 0; x < tab_widths[0]; ++x)
    putchar('-');
  for (int j = 0; j < warrior_count; ++j) {
    printf("-+-");
    for (int x = 0; x < tab_widths[j + 1]; ++x)
      putchar('-');
  }
  printf("-\n");

  for (int i = 0; i < warrior_count; ++i) {
    printf(" %*s", tab_widths[0], warriors[i]->name.c_str());
    for (int j = 0; j < warrior_count; ++j)
      printf(" | %*d", tab_widths[j + 1], score_results[i][j]);
    putchar('\n');
  }
}