#include <engine/platform/graphics/directx/render_command.hpp>
#include <engine/platform/graphics/directx/shared_context.hpp>

namespace Light {

dxRenderCommand::dxRenderCommand(Ref<dxSharedContext> sharedContext): m_context(sharedContext)
{
}

void dxRenderCommand::SwapBuffers()
{
#ifdef LIGHT_DEBUG
	HRESULT hr;
	if (FAILED(hr = m_context->GetSwapChain()->Present(1u, 0u)))
	{
		if (hr == DXGI_ERROR_DEVICE_REMOVED)
		{
			LOG(critical, "dxRenderCommand::SwapBuffers: DeviceRemoved:");
			LOG(critical, "        {}", m_context->GetDevice()->GetDeviceRemovedReason());
			throw dxException(hr, __FILE__, __LINE__);
		}
	}
#else
	m_context->GetSwapChain()->Present(0u, 0u);
#endif
}

void dxRenderCommand::ClearBackBuffer(const glm::vec4 &clearColor)
{
	m_context->GetDeviceContext()->ClearRenderTargetView(
	    m_context->GetRenderTargetView().Get(),
	    &clearColor[0]
	);
}

void dxRenderCommand::Draw(unsigned int count)
{
	m_context->GetDeviceContext()->Draw(count, 0u);
}

void dxRenderCommand::DrawIndexed(unsigned int count)
{
	m_context->GetDeviceContext()->DrawIndexed(count, 0u, 0u);
}

void dxRenderCommand::DefaultTargetFramebuffer()
{
	m_context->GetDeviceContext()
	    ->OMSetRenderTargets(1, m_context->GetRenderTargetView().GetAddressOf(), nullptr);
}

void dxRenderCommand::SetViewport(
    unsigned int x,
    unsigned int y,
    unsigned int width,
    unsigned int height
)
{
	// #todo: maybe call this somewhere else??
	SetResolution(width, height);

	// create viewport
	D3D11_VIEWPORT viewport;

	viewport.TopLeftX = x;
	viewport.TopLeftY = y;

	viewport.Width = width;
	viewport.Height = height;

	viewport.MinDepth = 0.0f;
	viewport.MaxDepth = 1.0f;

	// set viewport
	m_context->GetDeviceContext()->RSSetViewports(1u, &viewport);
}

void dxRenderCommand::SetResolution(unsigned int width, unsigned int height)
{
	HRESULT hr;

	// remove render target
	ID3D11RenderTargetView *nullViews[] = { nullptr };
	m_context->GetDeviceContext()->OMSetRenderTargets(1u, nullViews, nullptr);
	m_context->GetRenderTargetViewRef().Reset();

	// resize buffer
	DXC(m_context->GetSwapChain()
	        ->ResizeBuffers(0u, width, height, DXGI_FORMAT_R8G8B8A8_UNORM, NULL));

	// create render target
	Microsoft::WRL::ComPtr<ID3D11Resource> backBuffer = nullptr;
	DXC(m_context->GetSwapChain()->GetBuffer(0u, __uuidof(ID3D11Resource), &backBuffer));
	DXC(m_context->GetDevice()->CreateRenderTargetView(
	    backBuffer.Get(),
	    nullptr,
	    &m_context->GetRenderTargetViewRef()
	));

	// set render target
	m_context->GetDeviceContext()
	    ->OMSetRenderTargets(1u, m_context->GetRenderTargetView().GetAddressOf(), nullptr);
}

} // namespace Light