Seele/src/Engine/Graphics/Descriptor.cpp

#include "Descriptor.h"

using namespace Seele;
using namespace Seele::Gfx;

DescriptorLayout::DescriptorLayout(const std::string& name) : name(name) {}

DescriptorLayout::DescriptorLayout(const DescriptorLayout& other) {
  descriptorBindings.resize(other.descriptorBindings.size());
  for (uint32 i = 0; i < descriptorBindings.size(); ++i) {
    descriptorBindings[i] = other.descriptorBindings[i];
  }
}

DescriptorLayout& DescriptorLayout::operator=(const DescriptorLayout& other) {
  if (this != &other) {
    descriptorBindings.resize(other.descriptorBindings.size());
    for (uint32 i = 0; i < descriptorBindings.size(); ++i) {
      descriptorBindings[i] = other.descriptorBindings[i];
    }
  }
  return *this;
}

DescriptorLayout::~DescriptorLayout() {}

void DescriptorLayout::addDescriptorBinding(uint32 bindingIndex, SeDescriptorType type, SeImageViewType textureType, uint32 arrayCount,
                                            SeDescriptorBindingFlags bindingFlags, SeShaderStageFlags shaderStages) {
  if (descriptorBindings.size() <= bindingIndex) {
    descriptorBindings.resize(bindingIndex + 1);
  }
  descriptorBindings[bindingIndex] = DescriptorBinding{
      .binding = bindingIndex,
      .descriptorType = type,
      .textureType = textureType,
      .descriptorCount = arrayCount,
      .bindingFlags = bindingFlags,
      .shaderStages = shaderStages,
  };
}

PDescriptorSet DescriptorLayout::allocateDescriptorSet() { return pool->allocateDescriptorSet(); }

void DescriptorLayout::reset() { pool->reset(); }

DescriptorPool::DescriptorPool() {}

DescriptorPool::~DescriptorPool() {}

DescriptorSet::DescriptorSet(PDescriptorLayout layout) : layout(layout) {}

DescriptorSet::~DescriptorSet() {}

PipelineLayout::PipelineLayout() {}

PipelineLayout::PipelineLayout(PPipelineLayout baseLayout) {
  if (baseLayout != nullptr) {
    descriptorSetLayouts = baseLayout->descriptorSetLayouts;
    pushConstants = baseLayout->pushConstants;
  }
}

PipelineLayout::~PipelineLayout() {}

void PipelineLayout::addDescriptorLayout(PDescriptorLayout layout) {
  descriptorSetLayouts[layout->getName()] = layout;
}

void PipelineLayout::addPushConstants(const SePushConstantRange& pushConstant) { pushConstants.add(pushConstant); }

void PipelineLayout::addMapping(Map<std::string, uint32> mapping)
{
    for(const auto& [name, index] : mapping)
    {
        if(parameterMapping.contains(name))
        {
            assert(parameterMapping[name] == index);
        }
        parameterMapping[name] = index;
    }
}