Method LightProbeProcessingPass

src/render/LightProbeProcessingPass.cpp:61–125

using namespace donut::render;

LightProbeProcessingPass::LightProbeProcessingPass(
    nvrhi::IDevice* device,
    std::shared_ptr<ShaderFactory> shaderFactory,
    std::shared_ptr<CommonRenderPasses> commonPasses,
    uint32_t intermediateTextureSize,
    nvrhi::Format intermediateTextureFormat)
    : m_Device(device)
    , m_IntermediateTextureSize(intermediateTextureSize)
    , m_CommonPasses(commonPasses)
{
    m_GeometryShader = shaderFactory->CreateAutoShader("donut/passes/light_probe.hlsl", "cubemap_gs", DONUT_MAKE_PLATFORM_SHADER(g_light_probe_cubemap_gs), nullptr, nvrhi::ShaderType::Geometry);
    m_MipPixelShader = shaderFactory->CreateAutoShader("donut/passes/light_probe.hlsl", "mip_ps", DONUT_MAKE_PLATFORM_SHADER(g_light_probe_mip_ps), nullptr, nvrhi::ShaderType::Pixel);
    m_DiffusePixelShader = shaderFactory->CreateAutoShader("donut/passes/light_probe.hlsl", "diffuse_probe_ps", DONUT_MAKE_PLATFORM_SHADER(g_light_probe_diffuse_probe_ps), nullptr, nvrhi::ShaderType::Pixel);
    m_SpecularPixelShader = shaderFactory->CreateAutoShader("donut/passes/light_probe.hlsl", "specular_probe_ps", DONUT_MAKE_PLATFORM_SHADER(g_light_probe_specular_probe_ps), nullptr, nvrhi::ShaderType::Pixel);
    m_EnvironmentBrdfPixelShader = shaderFactory->CreateAutoShader("donut/passes/light_probe.hlsl", "environment_brdf_ps", DONUT_MAKE_PLATFORM_SHADER(g_light_probe_environment_brdf_ps), nullptr, nvrhi::ShaderType::Pixel);

    nvrhi::BindingLayoutDesc layoutDesc;
    layoutDesc.visibility = nvrhi::ShaderType::Pixel;
    layoutDesc.bindings = {
        nvrhi::BindingLayoutItem::VolatileConstantBuffer(0),
        nvrhi::BindingLayoutItem::Sampler(0),
        nvrhi::BindingLayoutItem::Texture_SRV(0),
    };
    m_BindingLayout = device->createBindingLayout(layoutDesc);

    nvrhi::BufferDesc constantBufferDesc;
    constantBufferDesc.byteSize = sizeof(LightProbeProcessingConstants);
    constantBufferDesc.debugName = "LightProbeProcessingConstants";
    constantBufferDesc.isConstantBuffer = true;
    constantBufferDesc.isVolatile = true;
    constantBufferDesc.maxVersions = 64;
    m_LightProbeCB = device->createBuffer(constantBufferDesc);

    assert(intermediateTextureSize > 0);

    nvrhi::TextureDesc cubemapDesc;
    cubemapDesc.arraySize = 6;
    cubemapDesc.width = intermediateTextureSize;
    cubemapDesc.height = intermediateTextureSize;
    cubemapDesc.mipLevels = static_cast<uint32_t>(floor(dm::log2f(static_cast<float>(intermediateTextureSize)))) + 1;
    cubemapDesc.dimension = nvrhi::TextureDimension::TextureCube;
    cubemapDesc.isRenderTarget = true;
    cubemapDesc.format = intermediateTextureFormat;
    cubemapDesc.initialState = nvrhi::ResourceStates::RenderTarget;
    cubemapDesc.keepInitialState = true;
    cubemapDesc.clearValue = nvrhi::Color(0.f);
    cubemapDesc.useClearValue = true;

    m_IntermediateTexture = m_Device->createTexture(cubemapDesc);

    m_EnvironmentBrdfTextureSize = 64;

    nvrhi::TextureDesc brdfTextureDesc;
    brdfTextureDesc.width = m_EnvironmentBrdfTextureSize;
    brdfTextureDesc.height = m_EnvironmentBrdfTextureSize;
    brdfTextureDesc.format = nvrhi::Format::RG16_FLOAT;
    brdfTextureDesc.initialState = nvrhi::ResourceStates::ShaderResource;
    brdfTextureDesc.keepInitialState = true;