VulkanMaterial.hpp

Go to the documentation of this file.

#pragma once
#include "Engine/Utility/Defines.hpp"
#ifdef USING_VULKAN

#include "Engine/Material/Material.hpp"

#include <ThirdParty/Vulkan/Include/vulkan/vulkan.h>

#include "Engine/Renderer/Vulkan/VulkanLogicalDevice.hpp"
#include "Engine/Renderer/Vulkan/VulkanDescriptorPool.hpp"
#include "Engine/Renderer/Vulkan/VulkanBuffer.hpp"
#include "Engine/Texture/VulkanTexture.hpp"

#include <ThirdParty/EASTL-master/include/EASTL/vector.h>

namespace Engine {

    class VulkanRenderer;

    class VulkanMaterial : public Material
    {
    public:

        static VkDescriptorSetLayout CreateMaterialDescriptorSetLayout(VulkanLogicalDevice* device);

        VulkanMaterial(const aiScene* scene, uint32_t materialIndex, eastl::string modelName);
        ~VulkanMaterial();

        VkDescriptorSet CreateMaterialDescriptorSet(size_t threadID, size_t pipelineID, size_t set, VkDescriptorSetLayout layout);

        VkDescriptorSet GetMaterialDescriptorSet(size_t threadID, size_t pipelineID, size_t set);

        virtual void SetDiffuseTexture(eastl::shared_ptr<Texture> diffuseTexture);

        virtual void SetBumpMapTexture(eastl::shared_ptr<Texture> bumpMapTexture);
        
        virtual void SetSpecularTexture(eastl::shared_ptr<Texture> specularTexture);

    protected:
        eastl::unique_ptr<VulkanBuffer> materialDataBuffer_;

        eastl::vector<eastl::vector<eastl::vector<VkDescriptorSet>>> materialDescriptorSets_;

        virtual void UpdateMaterialData();

        friend VulkanRenderer;

        static VulkanRenderer* renderer_;
        static VulkanLogicalDevice* device_;
        static VkCommandPool commandPool_;
        static VmaAllocator allocator_;
        static VulkanDescriptorPool* descriptorPool_;

        static void InitMaterialData(VulkanRenderer* renderer, VulkanLogicalDevice* device, VulkanDescriptorPool* descriptorPool, VmaAllocator allocator);
    };

} // namespace Engine
#endif // USING_VULKAN