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

// Parallel STL headers
#include <coordinate>
#include <experimental/algorithm>
#include <experimental/execution_policy>

#define _DEBUG (0)
#include "test_base.h"


// negative test
// no for_each_n shall commence
template<typename T, size_t SIZE, int FIRST_OFFSET, int LAST_OFFSET>
bool test_negative(void) {

  auto f = [](T& v) [[hc,cpu]] { v = 1; };
  using std::experimental::parallel::par;

  bool ret = true;
  ret &= run_and_compare<T, SIZE>([f](T (&input1)[SIZE],
                                      T (&input2)[SIZE]) {
    // There's no for_each_n in STL
    std::experimental::parallel::
    for_each_n(par, std::begin(input2) + FIRST_OFFSET, (LAST_OFFSET - FIRST_OFFSET), f);
  });

  return ret;
}

// positive test
// for_each_n shall commence
template<typename T, size_t SIZE, size_t FIRST_OFFSET, size_t TEST_LENGTH>
bool test(void) {

  // test kernel
  auto f = [&](T& v) [[hc,cpu]]
  {
    v *= 8;
    v += 3;
  };

  using std::experimental::parallel::par;

  bool ret = true;
  ret &= run_and_compare<T, SIZE>([f](T (&input1)[SIZE],
                                      T (&input2)[SIZE]) {
    std::for_each(std::begin(input1)+ FIRST_OFFSET, std::begin(input1) + TEST_LENGTH, f);
    std::experimental::parallel::
    for_each_n(par, std::begin(input2) + FIRST_OFFSET, TEST_LENGTH, f);
  });


  return ret;
}

int main() {
  bool ret = true;

  // positive tests
  ret &= test<int, TEST_SIZE, 0, 2>();
  ret &= test<unsigned, TEST_SIZE, 0, 2>();
  ret &= test<float, TEST_SIZE, 0, 2>();
  ret &= test<double, TEST_SIZE, 0, 2>();

  // negative tests
  ret &= test_negative<int, TEST_SIZE, 2, 0>();
  ret &= test_negative<unsigned, TEST_SIZE, 2, 0>();
  ret &= test_negative<float, TEST_SIZE, 2, 0>();
  ret &= test_negative<double, TEST_SIZE, 2, 0>();

  ret &= test_negative<int, TEST_SIZE, COL * 2, COL>();
  ret &= test_negative<unsigned, TEST_SIZE, COL * 2, COL>();
  ret &= test_negative<float, TEST_SIZE, COL * 2, COL>();
  ret &= test_negative<double, TEST_SIZE, COL * 2, COL>();

  ret &= test_negative<int, ROW * COL, COL * 2, COL * 2 - COL / 2>();
  ret &= test_negative<unsigned, ROW * COL, COL * 2, COL * 2 - COL / 2>();
  ret &= test_negative<float, ROW * COL, COL * 2, COL * 2 - COL / 2>();
  ret &= test_negative<double, ROW * COL, COL * 2, COL * 2 - COL / 2>();

  return !(ret == true);
}