COSC-3P93-Project/Step 3/include/opencl/cl.h

74 lines
2.2 KiB
C++

/*
* 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