◆ execute_shader()

void MayaFlux::Buffers::ComputeProcessor::execute_shader ( const std::shared_ptr< VKBuffer > & buffer )
overrideprivatevirtual
Implements MayaFlux::Buffers::ShaderProcessor.
Definition at line 210 of file ComputeProcessor.cpp.
{
    if (m_pipeline_id == Portal::Graphics::INVALID_COMPUTE_PIPELINE || m_descriptor_set_ids.empty()) {
        MF_ERROR(Journal::Component::Buffers, Journal::Context::BufferProcessing,
            "Cannot dispatch without pipeline and descriptors");
        return;
    }
 
    if (m_descriptor_set_ids.empty()) {
        MF_RT_ERROR(Journal::Component::Buffers, Journal::Context::BufferProcessing,
            "Descriptor sets not initialized");
        return;
    }
 
    auto& foundry = Portal::Graphics::get_shader_foundry();
    auto& compute_press = Portal::Graphics::get_compute_press();
 
    auto cmd_id = foundry.begin_commands(Portal::Graphics::ShaderFoundry::CommandBufferType::COMPUTE);
 
    m_last_command_buffer = cmd_id;
    m_last_processed_buffer = buffer;
 
    compute_press.bind_pipeline(cmd_id, m_pipeline_id);
 
    auto& descriptor_bindings = buffer->get_pipeline_context().descriptor_buffer_bindings;
    if (!descriptor_bindings.empty()) {
        for (const auto& binding : descriptor_bindings) {
            if (binding.set >= m_descriptor_set_ids.size()) {
                MF_RT_ERROR(Journal::Component::Buffers, Journal::Context::BufferProcessing,
                    "Descriptor set index {} out of range", binding.set);
                continue;
            }
 
            foundry.update_descriptor_buffer(
                m_descriptor_set_ids[binding.set],
                binding.binding,
                binding.type,
                binding.buffer_info.buffer,
                binding.buffer_info.offset,
                binding.buffer_info.range);
        }
    }
 
    if (!m_descriptor_set_ids.empty()) {
        compute_press.bind_descriptor_sets(cmd_id, m_pipeline_id, m_descriptor_set_ids);
    }
 
    on_before_execute(cmd_id, buffer);
 
    const uint32_t iterations = std::max(1U, m_dispatch_config.iteration_count);
    const auto dispatch_size = calculate_dispatch_size(buffer);
 
    for (uint32_t i = 0; i < iterations; ++i) {
        if (!on_iteration(cmd_id, buffer, i)) {
            continue;
        }
 
        const auto& pc_bindings = buffer->get_pipeline_context().push_constant_bindings;
 
        if (!pc_bindings.empty()) {
            size_t required = 0;
            for (const auto& pc : pc_bindings) {
                required = std::max(required, static_cast<size_t>(pc.offset) + pc.data.size());
            }
 
            m_push_constant_scratch.assign(required, 0);
 
            for (const auto& pc : pc_bindings) {
                std::memcpy(m_push_constant_scratch.data() + pc.offset, pc.data.data(), pc.data.size());
            }
 
            compute_press.push_constants(
                cmd_id, m_pipeline_id,
                m_push_constant_scratch.data(),
                m_push_constant_scratch.size());
        } else if (!m_push_constant_data.empty()) {
            compute_press.push_constants(
                cmd_id, m_pipeline_id,
                m_push_constant_data.data(),
                m_push_constant_data.size());
        }
 
        compute_press.dispatch(cmd_id, dispatch_size[0], dispatch_size[1], dispatch_size[2]);
 
        if (i + 1 < iterations) {
            on_iteration_barrier(cmd_id, buffer, i);
        }
    }
 
    on_after_execute(cmd_id, buffer);
 
    foundry.buffer_barrier(
        cmd_id,
        buffer->get_buffer(),
        vk::AccessFlagBits::eShaderWrite,
        vk::AccessFlagBits::eShaderRead | vk::AccessFlagBits::eTransferRead,
        vk::PipelineStageFlagBits::eComputeShader,
        vk::PipelineStageFlagBits::eComputeShader | vk::PipelineStageFlagBits::eTransfer);
 
    submit_recorded(cmd_id, buffer);
}
Here is the call graph for this function: