#include "CmD3D11RenderTexture.h"
#include "CmD3D11DepthStencilBuffer.h"
#include "CmD3D11RenderSystem.h"
#include "CmD3D11Device.h"
#include "CmD3D11Texture.h"
#include "CmD3D11Mappings.h"
#include "CmTextureManager.h"
#include "CmException.h"

namespace CamelotEngine
{
	D3D11RenderTexture::D3D11RenderTexture()
		:RenderTexture(), mRenderTargetView(nullptr)
	{

	}

	D3D11RenderTexture::~D3D11RenderTexture()
	{
		SAFE_RELEASE(mRenderTargetView);
	}

	ID3D11RenderTargetView* D3D11RenderTexture::createRenderTargetView(const SurfaceDesc& surfaceDesc)
	{
		D3D11_RENDER_TARGET_VIEW_DESC RTVDesc;
		ZeroMemory(&RTVDesc, sizeof(RTVDesc));

		D3D11Texture* d3d11Texture = static_cast<D3D11Texture*>(surfaceDesc.texture.get());

		D3D11_SHADER_RESOURCE_VIEW_DESC SRVDesc = d3d11Texture->getSRVDesc();
		RTVDesc.Format = SRVDesc.Format;

		switch(SRVDesc.ViewDimension)
		{
		case D3D11_SRV_DIMENSION_BUFFER:
			RTVDesc.ViewDimension = D3D11_RTV_DIMENSION_BUFFER;
			break;
		case D3D11_SRV_DIMENSION_TEXTURE1D:
			RTVDesc.ViewDimension = D3D11_RTV_DIMENSION_TEXTURE1D;
			RTVDesc.Texture1D.MipSlice = surfaceDesc.mipLevel;
			break;
		case D3D11_SRV_DIMENSION_TEXTURE1DARRAY:
			RTVDesc.ViewDimension = D3D11_RTV_DIMENSION_TEXTURE1DARRAY;
			RTVDesc.Texture1DArray.FirstArraySlice = surfaceDesc.face;
			RTVDesc.Texture1DArray.ArraySize = surfaceDesc.numFaces;
			RTVDesc.Texture1DArray.MipSlice = surfaceDesc.mipLevel;
			break;
		case D3D11_SRV_DIMENSION_TEXTURECUBE:
			RTVDesc.ViewDimension = D3D11_RTV_DIMENSION_TEXTURE2DARRAY;
			RTVDesc.Texture2DArray.FirstArraySlice = surfaceDesc.face;
			RTVDesc.Texture2DArray.ArraySize = surfaceDesc.numFaces;
			RTVDesc.Texture2DArray.MipSlice = surfaceDesc.mipLevel;
			break;
		case D3D11_SRV_DIMENSION_TEXTURE2D:
			RTVDesc.ViewDimension = D3D11_RTV_DIMENSION_TEXTURE2D;
			RTVDesc.Texture2D.MipSlice = surfaceDesc.mipLevel;
			break;
		case D3D11_SRV_DIMENSION_TEXTURE2DARRAY:
			RTVDesc.ViewDimension = D3D11_RTV_DIMENSION_TEXTURE2DARRAY;
			RTVDesc.Texture2DArray.FirstArraySlice = surfaceDesc.face;
			RTVDesc.Texture2DArray.ArraySize = surfaceDesc.numFaces;
			RTVDesc.Texture2DArray.MipSlice = surfaceDesc.mipLevel;
			break;
		case D3D11_SRV_DIMENSION_TEXTURE2DMS:
			RTVDesc.ViewDimension = D3D11_RTV_DIMENSION_TEXTURE2DMS;
			break;
		case D3D11_SRV_DIMENSION_TEXTURE2DMSARRAY:
			RTVDesc.ViewDimension = D3D11_RTV_DIMENSION_TEXTURE2DMSARRAY;
			RTVDesc.Texture2DMSArray.FirstArraySlice = surfaceDesc.face;
			RTVDesc.Texture2DMSArray.ArraySize = surfaceDesc.numFaces;
			break;
		case D3D11_SRV_DIMENSION_TEXTURE3D:
			RTVDesc.ViewDimension = D3D11_RTV_DIMENSION_TEXTURE3D;
			RTVDesc.Texture3D.MipSlice = surfaceDesc.mipLevel;
			RTVDesc.Texture3D.FirstWSlice = surfaceDesc.face;
			RTVDesc.Texture3D.WSize = surfaceDesc.numFaces;
			break;
		default:
			assert(false);
		}

		ID3D11RenderTargetView* rtv;
		D3D11RenderSystem* rs = static_cast<D3D11RenderSystem*>(RenderSystem::instancePtr());
		D3D11Device& device = rs->getPrimaryDevice();
		HRESULT hr = device.getD3D11Device()->CreateRenderTargetView(d3d11Texture->getDX11Resource(), &RTVDesc, &rtv);

		if (FAILED(hr) || device.hasError())
		{
			String errorDescription = device.getErrorDescription();
			CM_EXCEPT(RenderingAPIException, "Error creating Render Target View\nError Description:" + errorDescription);
		}

		return rtv;
	}

	void D3D11RenderTexture::createInternalResourcesImpl()
	{
		SAFE_RELEASE(mRenderTargetView);

		mRenderTargetView = createRenderTargetView(mSurface);
	}

	void D3D11RenderTexture::getCustomAttribute(const String& name, void* pData)
	{
		if(name == "RTV")
		{
			ID3D11RenderTargetView** pRTVs = (ID3D11RenderTargetView **)pData;						
			*pRTVs = mRenderTargetView;		

			return;
		}
		else if(name == "DSV")
		{
			ID3D11DepthStencilView** pDSV = (ID3D11DepthStencilView **)pData;
			D3D11DepthStencilBuffer* d3d11depthStencilBuffer = static_cast<D3D11DepthStencilBuffer*>(mDepthStencilBuffer.get());

			*pDSV = d3d11depthStencilBuffer->getDepthStencilView();
			return;
		}
	}
}