ht_d3d11shader.h

#pragma once

#include <ht_platform.h>
#include <ht_directx.h>
#include <ht_d3d11types.h>
#include <ht_shader.h>
#include <unordered_map>

namespace Hatchit {

    namespace Graphics {

        namespace DirectX {

            class HT_API D3D11Shader : public IShader
            {
                typedef std::unordered_map<std::string, ConstantBuffer*>        ConstantBufferTable;
                typedef std::unordered_map<std::string, uint32_t>               TextureTable;
                typedef std::unordered_map<std::string, uint32_t>               SampleTable;
                typedef std::unordered_map<std::string, ConstantBufferVariable> ConstantBufferVariableTable;
            public:
                D3D11Shader(ID3D11Device* device, ID3D11DeviceContext* context);

                virtual ~D3D11Shader();

                bool VInitFromFile(Core::File* file) override;

                void VOnLoaded() override;

                bool VSetData(std::string name, const void* data, size_t size) override;
                bool VSetInt(std::string name, int data) override;
                bool VSetFloat(std::string name, float data) override;
                bool VSetFloat2(std::string name, Math::Vector2 data) override;
                bool VSetFloat3(std::string name, Math::Vector3 data) override;
                bool VSetFloat4(std::string name, Math::Vector4 data) override;
                bool VSetMatrix3(std::string name, Math::Matrix3 data) override;
                bool VSetMatrix4(std::string name, Math::Matrix4 data) override;

                void Activate();

                void DeActivate();

            protected:
                uint32_t                    m_constantBufferCount;
                ConstantBuffer*             m_constantBufferArray;
                ConstantBufferVariableTable m_constantBufferVarTable;
                SampleTable                 m_sampleTable;
                TextureTable                m_textureTable;

                ID3D11Device*           m_device;
                ID3D11DeviceContext*    m_context;
                ID3DBlob*               m_blob;
                ID3D11ShaderReflection* m_reflection;



            protected:
                ConstantBufferVariable* FindVariable(std::string name, size_t size);
                ConstantBuffer*         FindBuffer(std::string name);
                int32_t                 FindTextureBindIndex(std::string name);
                int32_t                 FindSampleBindIndex(std::string name);

                void                    UpdateAllBuffers();


                virtual void            VBind() = 0;
                virtual void            VUnbind() = 0;
                virtual bool            VInitShader() = 0;

                virtual bool            VSetShaderResourceView(std::string name, ID3D11ShaderResourceView* rv) = 0;
                virtual bool            VSetSamplerState(std::string name, ID3D11SamplerState* ss) = 0;
            };
        }
    }
}