on_after_execute()

void MayaFlux::Buffers::RelaxationStepProcessor::on_after_execute ( Portal::Graphics::CommandBufferID  cmd_id, const std::shared_ptr< VKBuffer > &  buffer  )

overrideprotectedvirtual

Swap front/back generation index and, if requested, stage a snapshot download of the buffer just written.

Parameters

cmd_id	Command buffer the completed dispatch was recorded into.
buffer	The attached RelaxationGridBuffer, received as VKBuffer.

Reimplemented from MayaFlux::Buffers::ShaderProcessor.

Definition at line 109 of file RelaxationStepProcessor.cpp.

{
    auto grid = std::dynamic_pointer_cast<RelaxationGridBuffer>(buffer);
    if (!grid) {
        return;
    }
 
    if (!grid->consume_snapshot_request()) {
        return;
    }
 
    auto& resources = grid->get_buffer_resources();
    const auto& front = resources.back_buffers[grid->front_index()];
 
    std::vector<uint8_t> bytes(grid->get_state_bytes());
 
    auto buffer_service = Registry::BackendRegistry::instance()
                              .get_service<Registry::Service::BufferService>();
 
    if (!m_snapshot_staging || m_snapshot_staging->get_size_bytes() < bytes.size()) {
        m_snapshot_staging = create_staging_buffer(bytes.size());
    }
 
    auto handle = buffer_service->copy_buffer_fenced(
        static_cast<void*>(front.buffer),
        static_cast<void*>(m_snapshot_staging->get_buffer()),
        bytes.size(), 0, 0);
 
    buffer_service->wait_fenced(handle);
 
    auto& staging_resources = m_snapshot_staging->get_buffer_resources();
    buffer_service->invalidate_range(staging_resources.memory, 0, bytes.size());
 
    std::memcpy(bytes.data(), staging_resources.mapped_ptr, bytes.size());
    buffer_service->release_fenced(handle);
 
    grid->snapshot_source()->signal(bytes);
}

References MayaFlux::Buffers::create_staging_buffer(), MayaFlux::Registry::BackendRegistry::get_service(), MayaFlux::Registry::BackendRegistry::instance(), and m_snapshot_staging.

Here is the call graph for this function: