72 lines
3.2 KiB
C++
72 lines
3.2 KiB
C++
#include "LightEnvironment.h"
|
|
#include "Graphics/Graphics.h"
|
|
|
|
using namespace Seele;
|
|
|
|
LightEnvironment::LightEnvironment(Gfx::PGraphics graphics) : graphics(graphics) {
|
|
layout = graphics->createDescriptorLayout("pLightEnv");
|
|
layout->addDescriptorBinding(Gfx::DescriptorBinding{
|
|
.name = "directionalLights",
|
|
.descriptorType = Gfx::SE_DESCRIPTOR_TYPE_STORAGE_BUFFER,
|
|
});
|
|
layout->addDescriptorBinding(Gfx::DescriptorBinding{
|
|
.name = "numDirectionalLights",
|
|
.uniformLength = sizeof(uint)
|
|
});
|
|
layout->addDescriptorBinding(Gfx::DescriptorBinding{
|
|
.name = "pointLights",
|
|
.descriptorType = Gfx::SE_DESCRIPTOR_TYPE_STORAGE_BUFFER,
|
|
});
|
|
layout->addDescriptorBinding(Gfx::DescriptorBinding{
|
|
.name = "numPointLights",
|
|
.uniformLength = sizeof(uint)
|
|
});
|
|
layout->create();
|
|
|
|
directionalLights = graphics->createShaderBuffer(ShaderBufferCreateInfo{
|
|
.name = "DirectionalLights",
|
|
});
|
|
pointLights = graphics->createShaderBuffer(ShaderBufferCreateInfo{
|
|
.name = "PointLights",
|
|
});
|
|
}
|
|
|
|
LightEnvironment::~LightEnvironment() {}
|
|
|
|
void LightEnvironment::reset() {
|
|
layout->reset();
|
|
set = layout->allocateDescriptorSet();
|
|
dirs.clear();
|
|
points.clear();
|
|
}
|
|
|
|
void LightEnvironment::addDirectionalLight(Component::DirectionalLight dirLight) { dirs.add(dirLight); }
|
|
|
|
void LightEnvironment::addPointLight(Component::PointLight pointLight) { points.add(pointLight); }
|
|
|
|
void LightEnvironment::commit() {
|
|
directionalLights->rotateBuffer(sizeof(Component::DirectionalLight) * dirs.size());
|
|
directionalLights->updateContents(0, sizeof(Component::DirectionalLight) * dirs.size(), dirs.data());
|
|
directionalLights->pipelineBarrier(Gfx::SE_ACCESS_TRANSFER_WRITE_BIT, Gfx::SE_PIPELINE_STAGE_TRANSFER_BIT,
|
|
Gfx::SE_ACCESS_SHADER_READ_BIT | Gfx::SE_ACCESS_TRANSFER_WRITE_BIT,
|
|
Gfx::SE_PIPELINE_STAGE_FRAGMENT_SHADER_BIT | Gfx::SE_PIPELINE_STAGE_COMPUTE_SHADER_BIT |
|
|
Gfx::SE_PIPELINE_STAGE_TRANSFER_BIT);
|
|
pointLights->rotateBuffer(sizeof(Component::PointLight) * points.size());
|
|
pointLights->updateContents(0, sizeof(Component::PointLight) * points.size(), points.data());
|
|
pointLights->pipelineBarrier(Gfx::SE_ACCESS_TRANSFER_WRITE_BIT, Gfx::SE_PIPELINE_STAGE_TRANSFER_BIT,
|
|
Gfx::SE_ACCESS_SHADER_READ_BIT | Gfx::SE_ACCESS_TRANSFER_WRITE_BIT,
|
|
Gfx::SE_PIPELINE_STAGE_FRAGMENT_SHADER_BIT | Gfx::SE_PIPELINE_STAGE_COMPUTE_SHADER_BIT |
|
|
Gfx::SE_PIPELINE_STAGE_TRANSFER_BIT);
|
|
uint32 numPointLights = points.size();
|
|
uint32 numDirectionalLights = dirs.size();
|
|
set->updateConstants("numPointLights", 0, &numPointLights);
|
|
set->updateBuffer("pointLights", 0, pointLights);
|
|
set->updateConstants("numDirectionalLights", 0, &numDirectionalLights);
|
|
set->updateBuffer("directionalLights", 0, directionalLights);
|
|
set->writeChanges();
|
|
}
|
|
|
|
const Gfx::PDescriptorLayout LightEnvironment::getDescriptorLayout() const { return layout; }
|
|
|
|
Gfx::PDescriptorSet LightEnvironment::getDescriptorSet() { return set; }
|