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

#include <hc.hpp>

#include <iostream>

// added for checking HSA profile
#include <hc.hpp>

// test C++AMP with fine-grained SVM
// requires HSA Full Profile to operate successfully

#define SIZE (16)

using namespace hc;

// test supply a class with operator() to parallel_for_each
// the class will call a separate functor with a customized ctor
class user_functor {
public:
  int (&input)[SIZE];

  user_functor(int (&t)[SIZE]) [[cpu, hc]] : input(t) {}

  void operator() (hc::index<1>& idx) [[hc]] {
    input[idx[0]] = idx[0];
  }
};

class prog {
  user_functor& kernel;

public:
  prog(user_functor& f) [[cpu, hc]] : kernel(f) {
  }

  void operator() (hc::index<1>& idx) [[hc]] {
    kernel(idx);
  }

  void run() {
    parallel_for_each(extent<1>(SIZE), *this);
  }

  // verify output
  bool test() {
    bool ret = true;
    for (int i = 0; i < SIZE; ++i) {
      if (kernel.input[i] != i) {
        ret = false;
        break;
      }
    }
    return true;
  }
};

int main() {
  bool ret = true;
 
  // only conduct the test in case we are running on a HSA full profile stack
  hc::accelerator acc;
  if (acc.is_hsa_accelerator() &&
      acc.get_profile() == hc::hcAgentProfileFull) {

    // prepare test data
    int input[SIZE] { 0 };
  
    // launch kernel
    user_functor kernel(input);
    prog p(kernel);
    p.run();
  
    // check result
    ret &= p.test();
  }

  return !(ret == true);
}