Shader.h

// Copyright (c) 2024 Michael Carlie. All Rights Reserved.
//
// This software is released under the MIT License.
// https://opensource.org/licenses/MIT
 
#ifndef INCLUDE_VULKANENGINE_SHADER_H_
#define INCLUDE_VULKANENGINE_SHADER_H_
 
#include <VulkanEngine/Descriptor.h>
#include <VulkanEngine/ShaderModule.h>
 
#include <vector>
 
namespace VulkanEngine {
 
class Shader {
 public:
  explicit Shader(
      const std::vector<std::shared_ptr<ShaderModule>>& _shader_modules);
 
  ~Shader();
 
  void setDescriptors(
      const std::vector<std::vector<std::shared_ptr<Descriptor>>>&
          _descriptors);
 
  void bindDescriptorSet(const vk::CommandBuffer& command_buffer,
                         uint32_t descriptor_set_index);
 
  const std::vector<vk::PipelineShaderStageCreateInfo>& getVkShaderStages()
      const;
 
  const vk::PipelineLayout createVkPipelineLayout();
 
 private:
  std::vector<vk::PipelineShaderStageCreateInfo> shader_stages;
 
  std::vector<std::shared_ptr<ShaderModule>> shader_modules;
 
  std::vector<std::vector<std::shared_ptr<Descriptor>>> descriptors;
 
  vk::DescriptorPool vk_descriptor_pool;
 
  std::vector<vk::DescriptorSetLayout> vk_descriptor_set_layouts;
 
  std::vector<vk::DescriptorSet> vk_descriptor_sets;
 
  vk::PipelineLayout vk_pipeline_layout;
};
 
}  // namespace VulkanEngine
 
#endif  // INCLUDE_VULKANENGINE_SHADER_H_