Danbias/Code/OysterGraphics/Core/Buffer.h

#pragma once
#ifndef CoreBuffer
#define CoreBuffer

#include "CoreIncludes.h"

namespace Oyster
{
	class Buffer
	{
	public:
		enum BUFFER_TYPE
		{
			VERTEX_BUFFER,
			INDEX_BUFFER,
			CONSTANT_BUFFER_VS,
			CONSTANT_BUFFER_GS,
			CONSTANT_BUFFER_PS,
			CONSTANT_BUFFER_CS,
			STRUCTURED_BUFFER,
			BUFFER_TYPE_COUNT
		};

		enum BUFFER_USAGE
		{
			BUFFER_DEFAULT,
			BUFFER_STREAM_OUT_TARGET,
			BUFFER_CPU_WRITE,
			BUFFER_CPU_WRITE_DISCARD,
			BUFFER_CPU_READ,
			BUFFER_USAGE_COUNT,
			BUFFER_USAGE_IMMUTABLE
		};

		struct BUFFER_INIT_DESC
		{
			BUFFER_TYPE		Type;
			UINT32			NumElements;
			UINT32			ElementSize;
			BUFFER_USAGE	Usage;
			void*			InitData;

			BUFFER_INIT_DESC()
			{
				InitData = NULL;
				Usage = BUFFER_DEFAULT;
			}
		};
	protected:
		ID3D11Buffer*	mBuffer;
		BUFFER_TYPE		mType;
		BUFFER_USAGE	mUsage;

		UINT32			mElementSize;
		UINT32			mElementCount;
	public:
		Buffer();
		virtual ~Buffer();

		HRESULT Init(const BUFFER_INIT_DESC& initDesc);

		void* Map();
		void Unmap();

		operator ID3D11Buffer*();
		operator const ID3D11Buffer*() const;

		HRESULT Apply(UINT32 misc = 0) const;

		ID3D11Buffer* GetBufferPointer();
		UINT32 GetVertexSize();
		UINT32 GetElementCount();
	};
	
}

#endif