/*

    ......... 2018 Ivan Mahonin

    This program is free software: you can redistribute it and/or modify

    it under the terms of the GNU General Public License as published by

    the Free Software Foundation, either version 3 of the License, or

    (at your option) any later version.

    This program is distributed in the hope that it will be useful,

    but WITHOUT ANY WARRANTY; without even the implied warranty of

    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License

    along with this program.  If not, see <http://www.gnu.org/licenses/>.

*/

#ifndef _CUDACONTEXT_H_

#define _CUDACONTEXT_H_

#include <vector>

#include <cuda.h>

class CudaParams {

private:

	std::vector<char> params_buffer;

	std::vector<int> params_offsets;

	std::vector<void*> params_pointers;

	std::vector<void*> params_extra;

	size_t params_buffer_size;

public:

	CudaParams();

	void reset();

	CudaParams& add(const void* data, int size, int align);

	template<typename T>

	CudaParams& add(const T &data, int align)

		{ return add(&data, sizeof(data), align); }

	template<typename T>

	CudaParams& add(const T &data)

		{ return add(data, __alignof(T)); }

	void** get_params() const

		{ return params_pointers.empty() ? NULL : &(const_cast<CudaParams*>(this)->params_pointers.front()); }

	void** get_extra() const

		{ return params_extra.empty() ? NULL : &(const_cast<CudaParams*>(this)->params_extra.front()); }

};

class CudaContext {

public:

	CUdevice device;

	CUcontext context;

	CUresult err;

	CudaContext();

	~CudaContext();

	void hello();

};

#endif