#pragma once

#include "dxvk_buffer.h"
#include "dxvk_descriptor.h"
#include "dxvk_image.h"
#include "dxvk_sampler.h"

namespace dxvk {
  
  /**
   * \brief Bound shader resource
   */
  struct DxvkShaderResourceSlot {
    Rc<DxvkSampler>    sampler;
    Rc<DxvkImageView>  imageView;
    Rc<DxvkBufferView> bufferView;
    DxvkBufferBinding  bufferSlice;
  };
  
  
  /**
   * \brief Shader resource slots
   */
  class DxvkShaderResourceSlots {
    
  public:
    
    DxvkShaderResourceSlots() { }
    DxvkShaderResourceSlots(size_t n) {
      m_resources  .resize(n);
      m_descriptors.resize(n);
    }
    
    uint32_t descriptorCount() const {
      return m_descriptors.size();
    }
    
    const DxvkDescriptorInfo* descriptors() const {
      return m_descriptors.data();
    }
    
    const DxvkShaderResourceSlot& getShaderResource(uint32_t slot) const {
      return m_resources.at(slot);
    }
    
    void bindShaderResource(
            uint32_t                slot,
      const DxvkShaderResourceSlot& resource,
      const DxvkDescriptorInfo&     descriptor) {
      m_resources   .at(slot) = resource;
      m_descriptors .at(slot) = descriptor;
    }
    
  private:
    
    std::vector<DxvkShaderResourceSlot> m_resources;
    std::vector<DxvkDescriptorInfo>     m_descriptors;
    
  };
  
}