#ifndef INCLUDED_GPU_SCHEDULER
#define INCLUDED_GPU_SCHEDULER

#include "Scheduler.h"
#include "FifoDmaManager.h"
#include "GpuThreads.h"
#include "HTask.h"
#include "TaskManager.h"

#ifdef __APPLE__
#include <OpenCL/opencl.h>
#else
#include <CL/cl.h>
#endif

extern TaskObject gpu_task_list[MAX_TASK_OBJECT];

class GpuScheduler : public Scheduler {
 public:
    typedef struct gpubuffer {
        cl_int allocate_size;
        cl_int size;
        cl_mem *buf; // clCreateBuffer
        cl_event *event;
    } GpuBuffer, *GpuBufferPtr;
    GpuScheduler();
    virtual ~GpuScheduler();
    void init_impl(int useRefDma);
    void init_gpu();
    void wait_for_event(cl_event* event,GpuBufferPtr m, memaddr* reply,TaskListPtr* taskList,int cur);
    void run();
    
    void mail_write_from_host(memaddr data) {
        fifoDmaManager->mail_write_from_host(data);
    }

    memaddr mail_read_from_host() {
        return fifoDmaManager->mail_read_from_host();
    }

    int has_mail_from_host() {
        return fifoDmaManager->has_mail_from_host();
    }

    cl_platform_id platform_id;
    cl_device_id device_id;
    cl_uint ret_num_platforms;
    cl_uint ret_num_devices;
    cl_context context;
    cl_command_queue command_queue;
    cl_int ret;
    memaddr reply[2];
    cl_kernel kernel[2];
    cl_event kernel_event[2];
    GpuBuffer memin[2];
    GpuBuffer memout[2];
    HTask::htask_flag flag;
private:
    FifoDmaManager *fifoDmaManager;
    int load_kernel(int cmd);
    cl_mem createBuffer(GpuBuffer m, int i, cl_context context, cl_mem_flags flags, size_t size, cl_int *error);
    void initGpuBuffer(GpuBuffer m);
    void destroyGpuBuffer(GpuBuffer m);
};

#define GpuSchedRegister(str, filename, functionname)   \
    gpu_register_task(str, filename, functionname);
#endif

extern void gpu_register_task(int cmd,const char* filename,const char* functionname);