redgpu/bullet: Bullet Physics SDK: real-time collision detection and multi-physics simulation for VR, games, visual effects, robotics, machine learning etc.

src/Bullet3OpenCL/ParallelPrimitives/b3PrefixScanCL.h

36 строк818 B

erwin coumans

share more code between OpenCL kernels and C++ by moving kernel data structures/code to the 'shared' folders

07 ноя 2013, 07:57

07 ноя 2013, 07:57a9a758d


#ifndef B3_PREFIX_SCAN_CL_H
#define B3_PREFIX_SCAN_CL_H

#include "b3OpenCLArray.h"
#include "b3BufferInfoCL.h"
#include "Bullet3Common/b3AlignedObjectArray.h"

class b3PrefixScanCL
{
	enum
	{
		BLOCK_SIZE = 128
	};

//	Option m_option;

	cl_command_queue	m_commandQueue;

	cl_kernel m_localScanKernel;
	cl_kernel m_blockSumKernel;
	cl_kernel m_propagationKernel;

	b3OpenCLArray<unsigned int>* m_workBuffer;


	public:
		
	b3PrefixScanCL(cl_context ctx, cl_device_id device, cl_command_queue queue,int size=0);

	virtual ~b3PrefixScanCL();

	void execute(b3OpenCLArray<unsigned int>& src, b3OpenCLArray<unsigned int>& dst, int n, unsigned int* sum = 0);
	void executeHost(b3AlignedObjectArray<unsigned int>& src, b3AlignedObjectArray<unsigned int>& dst, int n, unsigned int* sum=0);
};

#endif //B3_PREFIX_SCAN_CL_H