#include <stdio.h>
#include <string.h>
#include <assert.h>

#define CL_TARGET_OPENCL_VERSION 300
#include <CL/cl.h>
#include <CL/cl_ext.h>

// Định nghĩa các biến toàn cục trong file
static cl_device_id     device_id;
static cl_context       context = NULL;
static cl_command_queue queue = NULL;
static cl_program       program = NULL;
static cl_kernel        kernel = NULL;

// Khai báo nguyên mẫu (Prototype) của hàm cục bộ trong file
static void set_work_size(size_t *ws, cl_int x, cl_int y);

#define MAKE_STRING(x) #x
#define MAX_PLATFORM_NUM 4
#define MAX_GPU_UNIT     4
#define KERNEL_FUNC      "kernel_hello"
                         // Phải khớp với mã hạt nhân (Kernel code)

// Mã nguồn (Source code) của hàm hạt nhân
const char *kernel_code = MAKE_STRING(
  __kernel void kernel_hello(void)
  {
    printf("[Kernel] Hello, World!\n");
  }
);

int main(int argc, char *argv[])
{
  char  device_vendor[40], device_name[80];
  size_t  size, gws[2], lws[2];
  cl_uint  num;
  cl_platform_id  platform_ids[MAX_PLATFORM_NUM];
  cl_device_id  device_ids[MAX_GPU_UNIT];
  cl_queue_properties queue_properties[] = {
    CL_QUEUE_PROPERTIES, CL_QUEUE_PROFILING_ENABLE,
    0 };

  // ① Lấy OpenCL Device ID ---------------------
  // ・Lấy OpenCL Platform ID
  clGetPlatformIDs(MAX_PLATFORM_NUM, platform_ids, &num);

  // ・Lấy OpenCL Device ID
  clGetDeviceIDs(platform_ids[0], CL_DEVICE_TYPE_DEFAULT, MAX_GPU_UNIT, device_ids, &num);

  // ② Chọn OpenCL Device sẽ sử dụng -----------
  // ・Chọn thiết bị OpenCL đầu tiên
  device_id = device_ids[0];

  // ③ Hiển thị thông tin thiết bị OpenCL -----------------
  clGetDeviceInfo(device_id, CL_DEVICE_VENDOR, sizeof(device_vendor), device_vendor, &size);
  device_vendor[size] = 0;
  printf("[Host] device_vendor=%s\n", device_vendor);
  clGetDeviceInfo(device_ids[0], CL_DEVICE_NAME, sizeof(device_name), device_name, &size);
  device_name[size] = 0;
  printf("[Host] device_name=%s\n", device_name);

  // ④ Chuẩn bị thực thi thiết bị OpenCL -----------------
  // ・Tạo OpenCL Context
  context = clCreateContext(NULL, 1, &device_id, NULL, NULL, NULL);

  // ・Tạo OpenCL Command Queue
  queue = clCreateCommandQueueWithProperties(context, device_id, queue_properties, NULL);

  // ⑤ Build (Biên dịch) chương trình hạt nhân OpenCL --------
  // ・Tạo chương trình hạt nhân từ mã nguồn
  size = strlen(kernel_code);
  program = clCreateProgramWithSource(context, 1, &kernel_code, &size, NULL);

  // ・Build chương trình hạt nhân
  clBuildProgram(program, 1, &device_id, NULL, NULL, NULL);

  // ・Tạo OpenCL Kernel (Hạt nhân)
  kernel = clCreateKernel(program, KERNEL_FUNC, NULL);

  // ⑥ Thực thi OpenCL Kernel ---------------------
  // ・Thực thi Kernel
  set_work_size(gws, 1, 0);
  set_work_size(lws, 1, 0);
  clEnqueueNDRangeKernel(queue, kernel, 1, NULL, gws, lws, 0, NULL, NULL);

  // Đợi cho đến khi toàn bộ task đưa vào Command Queue kết thúc
  clFinish(queue);

  // Giải phóng các tài nguyên (Resource)
  clReleaseKernel(kernel);
  clReleaseProgram(program);
  clReleaseCommandQueue(queue);
  clReleaseContext(context);

  return 0;
}

static void set_work_size(size_t *ws, cl_int x, cl_int y)
{
  ws[0] = x;
  ws[1] = y;
}