Method OmmBuildQueue

Rtxpt/OpacityMicroMap/OmmBuildQueue.cpp:145–179

}

OmmBuildQueue::OmmBuildQueue(
    nvrhi::DeviceHandle& device, 
    std::shared_ptr<donut::engine::DescriptorTableManager> descriptorTable, 
    std::shared_ptr<donut::engine::ShaderFactory> shaderFactory)
    : m_device(device)
    , m_descriptorTable(std::move(descriptorTable))
    , m_shaderFactory(std::move(shaderFactory))
{
    omm::GpuBakeNvrhi::ShaderProvider provider;

    provider.bindingOffsets = nvrhi::VulkanBindingOffsets { .shaderResource = 200, .sampler = 100, .constantBuffer = 300, .unorderedAccess = 400 };
    //provider.bindingOffsets = nvrhi::VulkanBindingOffsets { .shaderResource = UINT_MAX, .sampler = UINT_MAX, .constantBuffer = UINT_MAX, .unorderedAccess = UINT_MAX }; <- if MR gets accepted

    provider.shaders = [this](nvrhi::ShaderType type, const char* shaderName, const char* shaderEntryName)->nvrhi::ShaderHandle
    {
        std::vector<donut::engine::ShaderMacro> defines = { donut::engine::ShaderMacro("COMPILER_DXC", "1") };
        std::string shaderNameStr = std::string("omm/Omm/libraries/omm-lib/shaders/") + shaderName;
        return m_shaderFactory->CreateShader(shaderNameStr.c_str(), shaderEntryName, &defines, type);
    };

    omm::GpuBakeNvrhi::MessageCallback messageCb = [](omm::MessageSeverity severity, const char* message) {
        donut::log::info("[OMM SDK]: %d %s", severity, message);
    };

    // Intialize the the internal baker, which records some buffer updates into a command list
    nvrhi::CommandListHandle initCommandList = m_device->createCommandList();
    initCommandList->open();
    m_baker = std::make_unique<omm::GpuBakeNvrhi>(m_device, initCommandList, false /*debug*/, &provider, messageCb);

    // Submit baker init command list
    initCommandList->close();
    m_device->executeCommandList(initCommandList);
    // TODO: Do we really need to wait for device iddle here, or can we just subscribe a query to ensure resources are up by the time we use them?
    m_device->waitForIdle();
}

OmmBuildQueue::~OmmBuildQueue()
{