// RUN: %hc %s -o %t.out && %t.out

#include <hc.hpp>

#include <iostream>
#include <random>

#define LOOP_COUNT (1024)

#define TEST_DEBUG (0)

/// test which checks the behavior of:
/// completion_future::wait() with differnt hcWaitMode
template<size_t grid_size, size_t tile_size>
hc::completion_future execute(hc::array_view<const int, 1>& av1,
                              hc::array_view<const int, 1>& av2,
                              hc::array_view<int, 1>& av3) {
  // run HC parallel_for_each
  return hc::parallel_for_each(hc::tiled_extent<1>(grid_size, tile_size), [=](hc::tiled_index<1>& idx) [[hc]] {
    for (int i = 0; i < LOOP_COUNT; ++i) {
      av3(idx) = av1(idx) + av2(idx);
    }
  });
}

template<size_t grid_size>
bool verify(hc::array_view<const int, 1>& av1,
            hc::array_view<const int, 1>& av2,
            hc::array_view<int, 1>& av3) {
  for (int i = 0; i < grid_size; ++i) {
    if (av3[i] != av1[i] + av2[i]) {
      return false;
    }
  }
  return true;
}

bool test(bool useWaitMode, hc::hcWaitMode mode = hc::hcWaitModeBlocked) {
  bool ret = true;

  std::random_device rd;
  std::uniform_int_distribution<int32_t> int_dist;

  // initialize test data
  std::vector<int> table1(1024);
  std::vector<int> table2(1024);
  std::vector<int> table3(1024);
  for (int i = 0; i < 1024; ++i) {
    table1[i] = int_dist(rd);
    table2[i] = int_dist(rd);
  }
  hc::array_view<const int, 1> av1(1024, table1);
  hc::array_view<const int, 1> av2(1024, table2);
  hc::array_view<int, 1> av3(1024, table3);

  // launch kernel
  hc::completion_future fut = execute<1024, 16>(av1, av2, av3);

  // wait on the future
  if (!useWaitMode) {
    fut.wait();
  } else {
    fut.wait(mode);
  }

  // wait on the future again
  if (!useWaitMode) {
    fut.wait();
  } else {
    fut.wait(mode);
  }

  // verify computation result
  ret &= verify<1024>(av1, av2, av3);

  return ret;
}

int main() {
  bool ret = true;

  ret &= test(false);
  ret &= test(true, hc::hcWaitModeBlocked);
  ret &= test(true, hc::hcWaitModeActive);

  return !(ret == true);
}