src/MSAOpenCLKernel.h - master - redgpu/ofxMSAOpenCL

src/MSAOpenCLKernel.h
82 строки3 KB
Elie Zananiri
fixed spacing!
07 апр 2016, 05:21
07 апр 2016, 05:210598683
/***********************************************************************

OpenCL Kernel
Do not instanciate this directly.
Instead use OpenCL::loadKernel or OpenCLProgram::loadKernel

************************************************************************/ 


#pragma once

#include "ofMain.h"
#ifdef __APPLE__
#include <OpenCL/Opencl.h>
#else
#include <CL/opencl.h>
#endif
#include "MSAOpenCLMemoryObject.h"
#include "MSAOpenCLBufferManagedT.h"

namespace msa { 
	// ff decl
	class OpenCL;
	class OpenCLBuffer;

	// ----------

	class OpenCLKernel {
		friend class OpenCLProgram;

	public:

		~OpenCLKernel();


		// assign buffer to arguments
		bool setArg(int argNumber, void* argp_, size_t size_);
		bool setArg(int argNumber, float v) { return setArg(argNumber, &v, sizeof(v)); }
		bool setArg(int argNumber, int v) { return setArg(argNumber, &v, sizeof(v)); }
		bool setArg(int argNumber, ofVec2f v) { return setArg(argNumber, &v, sizeof(v)); }
		bool setArg(int argNumber, ofVec3f v) { return setArg(argNumber, &v, sizeof(v)); }
		bool setArg(int argNumber, ofVec4f v) { return setArg(argNumber, &v, sizeof(v)); }
		bool setArg(int argNumber, ofFloatColor c) { return setArg(argNumber, &c, sizeof(c)); }
		bool setArg(int argNumber, cl_mem v) { return setArg(argNumber, &v, sizeof(v)); }
		bool setArg(int argNumber, OpenCLMemoryObject& memObject);

		template<typename T>
		bool setArg(int argNumber, OpenCLBufferManagedT<T> &managedBuf){ return setArg(argNumber, managedBuf.getCLBuffer()); };
        
		// doesn't work on windows. templates get confused :(
        // template<class T>
        // bool setArg(int argNumber, T &arg) { return setArg(argNumber, &arg, sizeof(arg)); }
        

		// run the kernel
		// globalSize and localSize should be int arrays with same number of dimensions as numDimensions
		// leave localSize blank to let OpenCL determine optimum
		cl_int  bindOpenGLInterOp();
		void	run(int numDimensions, size_t *globalSize, size_t *localSize = NULL, cl_uint eventsInWaitList_ = 0, const cl_event* eventWaitList_ = NULL, cl_event* runEvent_ = NULL);
		cl_int  unbindOpenGLInterOp();

		// some wrappers for above to create the size arrays on the run
		void	run1D(size_t globalSize, size_t localSize = 0);
		void	run2D(size_t globalSizeX, size_t globalSizeY, size_t localSizeX = 0, size_t localSizeY = 0, cl_uint eventsInWaitList_ = 0, const cl_event* eventWaitList_ = NULL, cl_event* runEvent_ = NULL);
		void	run3D(size_t globalSizeX, size_t globalSizeY, size_t globalSizeZ, size_t localSizeX = 0, size_t localSizeY = 0, size_t localSizeZ = 0);


		cl_kernel& getCLKernel();
		string getName();

	protected:
		string			name;
		OpenCL*		pOpenCL;
		cl_kernel		clKernel;

		OpenCLKernel(OpenCL *pOpenCL, cl_kernel clKernel, string name);
	private:
		/// holds a list of all objects we need to flag up to openGL through 
		/// clEnqueueAcquireGLObjects before we run.
		vector<cl_mem>	mOpenGLInteropArguments;
	};
}