git-svn-id: svn://db.shs.com.ru/pip@559 12ceb7fc-bf1f-11e4-8940-5bc7170c53b5

2017-11-21 12:25:04 +00:00
parent 11f5ebb23d
commit eb2c99cf02
8 changed files with 995 additions and 20 deletions
--- a/src_main/opencl/piopencl.h
+++ b/src_main/opencl/piopencl.h
@@ -0,0 +1,175 @@
+#ifndef PIOPENCL_H
+#define PIOPENCL_H
+
+#include "pivariant.h"
+
+
+class PIOpenCL {
+public:
+
+	struct KernelArg;
+	struct Device;
+	struct Platform;
+	class Context;
+	class Program;
+	class Kernel;
+
+	typedef PIVector<Device> DeviceList;
+
+	enum AddressQualifier {
+		AddressGlobal,
+		AddressLocal,
+		AddressConstant,
+		AddressPrivate,
+	};
+
+	enum AccessQualifier {
+		AccessReadOnly,
+		AccessWriteOnly,
+		AccessReadWrite,
+		AccessNone,
+	};
+
+	enum TypeQualifier {
+		TypeConst,
+		TypeRestrict,
+		TypeVolatile,
+		TypeNone,
+	};
+
+	enum ArgType {
+		Char = 1,
+		UChar,
+		Short,
+		UShort,
+		Int,
+		UInt,
+		Long,
+		ULong,
+		Float,
+		Double,
+	};
+
+	struct KernelArg {
+		KernelArg();
+		AddressQualifier address_qualifier;
+		AccessQualifier access_qualifier;
+		TypeQualifier type_qualifier;
+		PIString arg_name;
+		PIString type_name;
+		PIString base_type_name;
+		bool is_pointer;
+		ArgType arg_type;
+		int dims;
+	private:
+		friend class Kernel;
+		void init(void * _k, uint index);
+	};
+
+	struct Device {
+		Device() {id = platform_id = 0; max_compute_units = max_clock_frequency = 0; max_memory_size = 0;}
+		bool isValid() const {return id != 0;}
+		PIString displayText() const {return name.trimmed() + " (" + device_version.trimmed() + ")";}
+		void * id;
+		void * platform_id;
+		PIString name;
+		PIString vendor;
+		PIString device_version;
+		PIString driver_version;
+		int max_compute_units;
+		int max_clock_frequency;
+		ullong max_memory_size;
+	};
+
+	struct Platform {
+		Platform() {id = 0;}
+		bool isValid() const {return id != 0;}
+		PIString displayText() const {return name.trimmed() + " (" + version.trimmed() + ", " + profile.trimmed() + ")";}
+		void * id;
+		PIString name;
+		PIString vendor;
+		PIString profile;
+		PIString version;
+		PIStringList extensions;
+		PIVector<Device> devices;
+	};
+
+	class Context {
+		friend class Program;
+	public:
+		~Context();
+		static Context * create(const DeviceList & dl);
+		static Context * create(const Device & d) {return create(DeviceList() << d);}
+		Program * createProgram(const PIString & source, PIString * error = 0);
+	private:
+		Context();
+		void zero();
+		void deletePrograms();
+		PIVector<Program * > programs_;
+		PRIVATE_DECLARATION
+	};
+
+	class Program {
+		friend class Context;
+		friend class Kernel;
+	public:
+		~Program();
+		const PIString & sourceCode() const {return source_;}
+		const Kernel * kernel(int index = 0) const {return kernels_[index];}
+		const PIVector<Kernel * > & kernels() const {return kernels_;}
+	private:
+		Program();
+		void zero();
+		bool initKernels(PIVector<void*> kerns);
+		Context * context_;
+		PIString source_;
+		PIVector<Kernel * > kernels_;
+		PRIVATE_DECLARATION
+	};
+
+	class Kernel {
+		friend class Program;
+	public:
+		const PIString & name() const {return name_;}
+		const PIVector<KernelArg> & args() const {return args_;}
+		template <typename T> bool setArgValue(int index, const T & value) {return setArgValueV(index, PIVariant::fromValue(value));}
+		template <typename T> bool setArgValue(const PIString & arg, const T & value) {return setArgValue(argIndex(arg), value);}
+	private:
+		Kernel();
+		~Kernel();
+		void zero();
+		bool init();
+		bool setArgValueV(int index, const PIVariant & value);
+		int argIndex(const PIString & an) const;
+		KernelArg argByName(const PIString & an) const;
+		Context * context_;
+		Program * program_;
+		PIString name_;
+		PIVector<KernelArg> args_;
+		PRIVATE_DECLARATION
+	};
+
+	static void init();
+	static const PIVector<Platform> & platforms();
+	static const PIVector<Device> devices();
+	static Device deviceByID(void * id);
+	static PIString prepareProgram(const PIString & prog);
+
+private:
+	static PIString prog_header;
+	PIOpenCL() {;}
+	class Initializer {
+	public:
+		Initializer();
+		static Initializer * instance();
+		void init();
+		PIVector<Platform> platforms_;
+		bool inited_;
+	};
+};
+
+
+PICout operator <<(PICout s, const PIOpenCL::KernelArg & v);
+
+
+#endif // PIOPENCL_H