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

#include "constants.hpp"

static int do_round(const lib94::warrior *w1, const lib94::warrior *w2) {
  const lib94::warrior *ws[2] = {w1, w2};

  lib94::instruction background = {
    .op = lib94::DAT, .mod = lib94::F,
    .amode = lib94::DIRECT, .bmode = lib94::DIRECT,
    .anumber = 0, .bnumber = 0
  };

  lib94::clear_core(background);

  lib94::init_round(ws, 2);

  for (int i = 0; i < STEPS_TO_TIE; ++i) {
    const lib94::warrior *result = lib94::single_step<false>();
    if (result == w1)
      return -1;
    if (result == w2)
      return 1;
  }

  return 0;
}

void worker_main(const lib94::warrior *const *warriors) {
  int buffer[3] = {-1};

  while (1) {
    MPI_Send(buffer, 3, MPI_INT, 0, 1, MPI_COMM_WORLD);
    MPI_Recv(buffer, 3, MPI_INT, 0, 0, MPI_COMM_WORLD, 0);

    if (buffer[0] == -1)
      return;

    buffer[2] = 0;

    for (int i = 0; i < ROUNDS_PER_CHUNK; ++i)
      buffer[2] += do_round(warriors[buffer[0]], warriors[buffer[1]]);
  }
}