Method BakeOmmArrayData

Rtxpt/OpacityMicroMap/OmmBuildQueue.cpp:302–343

}

void OmmBuildQueue::BakeOmmArrayData(nvrhi::ICommandList& commandList, BuildTask& task)
{
    commandList.beginTrackingBufferState(task.buffers.ommIndexBuffer, nvrhi::ResourceStates::Common);
    commandList.beginTrackingBufferState(task.buffers.ommDescBuffer, nvrhi::ResourceStates::Common);
    commandList.beginTrackingBufferState(task.buffers.ommDescArrayHistogramBuffer, nvrhi::ResourceStates::Common);
    commandList.beginTrackingBufferState(task.buffers.ommIndexArrayHistogramBuffer, nvrhi::ResourceStates::Common);
    commandList.beginTrackingBufferState(task.buffers.ommPostDispatchInfoBuffer, nvrhi::ResourceStates::Common);
    commandList.beginTrackingBufferState(task.buffers.ommArrayDataBuffer, nvrhi::ResourceStates::Common);

    commandList.clearBufferUInt(task.buffers.ommArrayDataBuffer, 0);

    std::shared_ptr < donut::engine::MeshInfo > mesh = task.input.mesh;

    for (uint32_t i = 0; i < task.input.geometries.size(); ++i)
    {
        const BufferInfo& bufferInfo = task.bufferInfos[i];
        const OmmBuildQueue::BuildInput::Geometry& geom = task.input.geometries[i];

        omm::GpuBakeNvrhi::Input input = GetBakeInput(omm::GpuBakeNvrhi::Operation::Bake, *mesh.get(), geom);

        omm::GpuBakeNvrhi::Buffers output;
        output.ommArrayBuffer = task.buffers.ommArrayDataBuffer;
        output.ommArrayBufferOffset = (uint32_t)bufferInfo.ommArrayDataOffset;
        output.ommDescBuffer = task.buffers.ommDescBuffer;
        output.ommDescBufferOffset = (uint32_t)bufferInfo.ommDescArrayOffset;
        output.ommIndexBuffer = task.buffers.ommIndexBuffer;
        output.ommIndexBufferOffset = (uint32_t)bufferInfo.ommIndexOffset;
        output.ommDescArrayHistogramBuffer = task.buffers.ommDescArrayHistogramBuffer;
        output.ommDescArrayHistogramBufferOffset = (uint32_t)bufferInfo.ommDescArrayHistogramOffset;
        output.ommIndexHistogramBuffer = task.buffers.ommIndexArrayHistogramBuffer;
        output.ommIndexHistogramBufferOffset = (uint32_t)bufferInfo.ommIndexHistogramOffset;
        output.ommPostDispatchInfoBuffer = task.buffers.ommPostDispatchInfoBuffer;
        output.ommPostDispatchInfoBufferOffset = (uint32_t)bufferInfo.ommPostDispatchInfoOffset;

        m_baker->Dispatch(&commandList, input, output);

        omm::GpuBakeNvrhi::PreDispatchInfo setupInfo;
        m_baker->GetPreDispatchInfo(input, setupInfo);

        commandList.copyBuffer(task.buffers.ommReadbackBuffer, bufferInfo.ommPostDispatchInfoReadbackOffset, task.buffers.ommPostDispatchInfoBuffer, bufferInfo.ommPostDispatchInfoOffset, setupInfo.ommPostDispatchInfoBufferSize);
    }
}

std::vector<bvh::OmmAttachment> OmmBuildQueue::BuildOMMAttachments(nvrhi::ICommandList& commandList, BuildTask& task)
{