/*
 * Created by Brett Terpstra 6920201 on 20/11/22.
 * Copyright (c) 2022 Brett Terpstra. All Rights Reserved.
 */

#ifndef STEP_3_CL_H
#define STEP_3_CL_H

// OpenCL includes
#include <CL/cl.h>

#include <engine/util/std.h>

namespace Raytracing {

    class CLProgram {
        private:
            cl_int m_CL_ERR{};
            std::string m_source;
            
            cl_device_id m_deviceID{};
            cl_context m_context{};
            cl_command_queue m_commandQueue{};
            cl_program m_program{};
            
            std::unordered_map<std::string, cl_mem> buffers;
            std::unordered_map<std::string, cl_kernel> kernels;
            
        public:
            explicit CLProgram(const std::string& file);
            void loadCLShader(cl_context context, cl_device_id deviceID);
            
            void createKernel(const std::string& kernelName);
            void createBuffer(const std::string& bufferName, cl_mem_flags flags, size_t bytes);
            void createBuffer(const std::string& bufferName, cl_mem_flags flags, size_t bytes, void* ptr);
            
            void setKernelArgument(const std::string& kernel, const std::string& buffer, int argIndex);
            
            void writeBuffer(const std::string& buffer, size_t bytes, void* ptr, cl_bool blocking = CL_TRUE, size_t offset = 0);
            void readBuffer(const std::string& buffer, size_t bytes, void* ptr, cl_bool blocking = CL_TRUE, size_t offset = 0);
            
            void flushCommands();
            void finishCommands();
            
            ~CLProgram();
    };
    
    class OpenCL {
        private:
            cl_int m_CL_ERR;
            cl_uint m_numPlatforms;
            int m_activePlatform;
            
            cl_platform_id* m_platformIDs;
            cl_uint m_numOfPlatformIDs{};
            
            cl_device_id m_deviceID{};
            cl_uint m_numOfDevices{};
            
            cl_context m_context;
            
            static void printDeviceInfo(cl_device_id device);
        public:
            explicit OpenCL(int platformID = 0, int deviceID = 0);
            static void init();
            static void createCLProgram(CLProgram& program);
        
            ~OpenCL();
    };

}

#endif //STEP_3_CL_H