109 lines
2.9 KiB
C++
109 lines
2.9 KiB
C++
#ifndef _LIVOX_GEN1_OPENCL_COLLATING_AND_MESHING_ENGINE_H
|
|
#define _LIVOX_GEN1_OPENCL_COLLATING_AND_MESHING_ENGINE_H
|
|
|
|
#include <boostAsioLinkageFix.h>
|
|
#include <cstdint>
|
|
#include <cstddef>
|
|
#include <memory>
|
|
#include <functional>
|
|
#define CL_TARGET_OPENCL_VERSION 300
|
|
#include <CL/cl.h>
|
|
#include "stagingBuffer.h"
|
|
#include "frameAssemblyDesc.h"
|
|
|
|
namespace smo {
|
|
namespace stim_buff {
|
|
|
|
class PcloudStimulusBuffer;
|
|
|
|
class OpenClCollatingAndMeshingEngine
|
|
{
|
|
public:
|
|
explicit OpenClCollatingAndMeshingEngine(PcloudStimulusBuffer& parent);
|
|
~OpenClCollatingAndMeshingEngine();
|
|
|
|
// Non-copyable, movable
|
|
OpenClCollatingAndMeshingEngine(
|
|
const OpenClCollatingAndMeshingEngine&) = delete;
|
|
OpenClCollatingAndMeshingEngine& operator=(
|
|
const OpenClCollatingAndMeshingEngine&) = delete;
|
|
OpenClCollatingAndMeshingEngine(
|
|
OpenClCollatingAndMeshingEngine&&) = default;
|
|
OpenClCollatingAndMeshingEngine& operator=(
|
|
OpenClCollatingAndMeshingEngine&&) = default;
|
|
|
|
bool setup();
|
|
void finalize();
|
|
|
|
// Callback function types
|
|
typedef std::function<void()> compactKernelCbFn;
|
|
typedef std::function<void()> collateKernelCbFn;
|
|
|
|
bool startCompactKernel(
|
|
StagingBuffer& assemblyBuff, uint32_t nSucceeded,
|
|
compactKernelCbFn callback);
|
|
bool startCollateKernel(
|
|
StagingBuffer& assemblyBuff, StagingBuffer& collationBuff,
|
|
collateKernelCbFn callback);
|
|
|
|
void stopCompactKernel();
|
|
void stopCollateKernel();
|
|
void stop();
|
|
|
|
private:
|
|
PcloudStimulusBuffer& parent;
|
|
|
|
// OpenCL infrastructure
|
|
cl_platform_id platform;
|
|
cl_device_id device;
|
|
cl_context context;
|
|
cl_command_queue commandQueue;
|
|
cl_program slotCompactorProgram;
|
|
cl_program collateProgram;
|
|
cl_kernel slotCompactorKernel;
|
|
cl_kernel collateKernel;
|
|
bool isSetup;
|
|
|
|
// OpenCL buffers
|
|
cl_mem clAssemblyBuffer;
|
|
cl_mem clCollationBuffer;
|
|
|
|
// State tracking
|
|
bool isRunning;
|
|
cl_event currentKernelEvent;
|
|
|
|
// Memory tracking
|
|
void* assemblyBufferPtr;
|
|
size_t assemblyBufferSize;
|
|
void* collationBufferPtr;
|
|
size_t collationBufferSize;
|
|
|
|
// Frame descriptor (cached from setup)
|
|
std::shared_ptr<FrameAssemblyDesc> frameAssemblyDesc;
|
|
|
|
// Callback storage
|
|
compactKernelCbFn compactKernelCb;
|
|
collateKernelCbFn collateKernelCb;
|
|
|
|
// Static callbacks for OpenCL events
|
|
static void CL_CALLBACK compactKernelEventCallback(
|
|
cl_event event, cl_int event_command_exec_status, void* user_data);
|
|
static void CL_CALLBACK collateKernelEventCallback(
|
|
cl_event event, cl_int event_command_exec_status, void* user_data);
|
|
|
|
// Private helper methods
|
|
bool compileAndPrepareKernel(
|
|
const char* kernelSource, size_t kernelSourceLen,
|
|
const char* kernelName, cl_program& program, cl_kernel& kernel);
|
|
bool compileAndPrepareKernels();
|
|
bool setupSlotCompactorsArgs(
|
|
StagingBuffer& assemblyBuff, uint32_t nSucceeded);
|
|
bool setupCollateDgramsArgs(StagingBuffer& assemblyBuff);
|
|
};
|
|
|
|
} // namespace stim_buff
|
|
} // namespace smo
|
|
|
|
#endif // _LIVOX_GEN1_OPENCL_COLLATING_AND_MESHING_ENGINE_H
|
|
|