/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
// Original file Copyright Crytek GMBH or its affiliates, used under license.

#pragma once


template <typename T, int constantBufferSlot>
class CTypedConstantBuffer
{
    T m_hostBuffer;

    CConstantBuffer* m_constantBuffer;

public:
    CTypedConstantBuffer()
        : m_constantBuffer(nullptr) {}

    CConstantBuffer* GetDeviceConstantBuffer()
    {
        if (!m_constantBuffer)
        {
            CreateDeviceBuffer();
        }
        return m_constantBuffer;
    }

    void CreateDeviceBuffer()
    {
        int size = sizeof(T);
        m_constantBuffer = gcpRendD3D->m_DevBufMan.CreateConstantBuffer(size);
    }
    void CopyToDevice()
    {
        D3D11_MAPPED_SUBRESOURCE mapped;
        HRESULT hr =
            gcpRendD3D.GetDeviceContext().Map(m_constantBuffer, 0, D3D11_MAP_WRITE_DISCARD, 0, &mapped);

        if (hr != S_OK)
        {
            return;
        }

        memcpy(mapped.pData, &m_hostBuffer, sizeof(T));

        gcpRendD3D.GetDeviceContext().Unmap(m_constantBuffer, 0);
#if defined(AZ_RESTRICTED_PLATFORM)
    #if defined(AZ_PLATFORM_XENIA)
        #include "Xenia/GraphicsHelpers_h_xenia.inl"
    #elif defined(AZ_PLATFORM_PROVO)
        #include "Provo/GraphicsHelpers_h_provo.inl"
    #elif defined(AZ_PLATFORM_SALEM)
        #include "Salem/GraphicsHelpers_h_salem.inl"
    #endif
#endif
    }
    void Bind()
    {
        ID3D11Buffer* buf[] = { m_constantBuffer };
        gcpRendD3D.GetDeviceContext().CSSetConstantBuffers(constantBufferSlot, 1u, buf);
    }
    void BindPixelShader()
    {
        ID3D11Buffer* buf[] = { m_constantBuffer };
        gcpRendD3D.GetDeviceContext().PSSetConstantBuffers(constantBufferSlot, 1u, buf);
    }
    void BindGeometryShader()
    {
        ID3D11Buffer* buf[] = { m_constantBuffer };
        gcpRendD3D.GetDeviceContext().GSSetConstantBuffers(constantBufferSlot, 1u, buf);
    }
    T* operator->() { return &m_hostBuffer; }
    bool IsDeviceBufferAllocated() { return m_constantBuffer != nullptr; }
    T& operator=(const T& hostData)
    {
        return m_hostBuffer = hostData;
    }
};