#include "ltpch.h"
#include "dxBuffers.h"

#include "dxSharedContext.h"

namespace Light {

	dxVertexBuffer::dxVertexBuffer(float* vertices, unsigned int stride, unsigned int count, void* sharedContext)
		: m_Stride(stride)
	{
		HRESULT hr;

		dxSharedContext* dxContext = static_cast<dxSharedContext*>(sharedContext);
		LT_ENGINE_ASSERT(dxContext, "dxShader::dxShader: invalid dxContext");

		m_Device = dxContext->device;
		m_DeviceContext = dxContext->deviceContext;

		D3D11_BUFFER_DESC desc = { 0 };

		desc.BindFlags = D3D11_BIND_VERTEX_BUFFER;
		desc.Usage = D3D11_USAGE_DYNAMIC;
		desc.CPUAccessFlags = D3D11_CPU_ACCESS_WRITE;

		desc.ByteWidth = count * stride;
		desc.StructureByteStride = stride;

		DXC(m_Device->CreateBuffer(&desc, nullptr, &m_Buffer));
	}

	dxVertexBuffer::~dxVertexBuffer()
	{
	}

	void* dxVertexBuffer::Map()
	{
		m_DeviceContext->Map(m_Buffer.Get(), NULL, D3D11_MAP_WRITE_DISCARD, NULL, &m_Map);
		return m_Map.pData;
	}

	void dxVertexBuffer::UnMap()
	{
		m_DeviceContext->Unmap(m_Buffer.Get(), NULL);
	}

	void dxVertexBuffer::Bind()
	{
		static const unsigned int offset = 0u;
		m_DeviceContext->IASetVertexBuffers(0u, 1u, m_Buffer.GetAddressOf(), &m_Stride, &offset);
	}

	void dxVertexBuffer::UnBind()
	{
	}

	dxIndexBuffer::dxIndexBuffer(unsigned int* indices, unsigned int count, void* sharedContext)
	{
		HRESULT hr;

		dxSharedContext* dxContext = static_cast<dxSharedContext*>(sharedContext);
		LT_ENGINE_ASSERT(dxContext, "dxShader::dxShader: invalid dxContext");

		m_Device = dxContext->device;
		m_DeviceContext = dxContext->deviceContext;

		bool hasIndices = !!indices;
		if (!hasIndices)
		{
			if (count % 6 != 0)
			{
				LT_ENGINE_WARN("dxIndexBuffer::dxIndexBuffer: indices can only be null if count is multiple of 6");
				LT_ENGINE_WARN("dxIndexBuffer::dxIndexBuffer: adding {} to count -> {}", (6 - (count % 6)), count + (6 - (count % 6)));
				count = count + (6 - (count % 6));
			}

			indices = new unsigned int[count];
			unsigned int offset = 0;
			for (unsigned int i = 0; i < count; i += 6)
			{
				indices[i + 0] = offset + 0;
				indices[i + 1] = offset + 3;
				indices[i + 2] = offset + 2;

				indices[i + 3] = offset + 2;
				indices[i + 4] = offset + 1;
				indices[i + 5] = offset + 0;

				offset += 4;
			}
		}

		D3D11_BUFFER_DESC bufferDesc = { 0 };
		D3D11_SUBRESOURCE_DATA sd = { 0 };
		bufferDesc.BindFlags = D3D11_BIND_INDEX_BUFFER;
		bufferDesc.Usage = D3D11_USAGE_DEFAULT;

		bufferDesc.ByteWidth = count * sizeof(unsigned int);
		bufferDesc.StructureByteStride = sizeof(unsigned int);

		sd.pSysMem = indices;

		DXC(m_Device->CreateBuffer(&bufferDesc, &sd, &m_Buffer));

		if (!hasIndices)
		{
			delete[] indices;
			indices = nullptr;
		}
	}

	dxIndexBuffer::~dxIndexBuffer()
	{
	}

	void dxIndexBuffer::Bind()
	{
		m_DeviceContext->IASetIndexBuffer(m_Buffer.Get(), DXGI_FORMAT_R32_UINT, 0u);
	}

	void dxIndexBuffer::UnBind()
	{
	}

}