readback_primary()

std::vector< float > MayaFlux::Yantra::GpuDispatchCore::readback_primary ( size_t  float_count )

protected

Read back the primary output buffer into a float vector.

Selects the first OUTPUT or INPUT_OUTPUT binding. Caps readback to the lesser of the requested float count and the allocated buffer size.

Parameters

float_count

Number of float elements to attempt to read.

Returns

Float vector of length min(float_count, allocated / sizeof(float)).

Definition at line 448 of file GpuDispatchCore.cpp.

{
    const size_t idx = find_first_output_index();
 
    for (const auto& b : m_bindings) {
        if (b.binding != idx)
            continue;
        if (b.element_type == GpuBufferBinding::ElementType::IMAGE_STORAGE
            || b.element_type == GpuBufferBinding::ElementType::IMAGE_SAMPLED)
            return {};
        break;
    }
 
    const size_t allocated = m_resources.buffer_allocated_bytes(dispatch_key(), idx);
    const size_t byte_size = std::min(float_count * sizeof(float), allocated);
    std::vector<float> out(byte_size / sizeof(float));
    m_resources.download(dispatch_key(), idx, out.data(), byte_size);
    return out;
}

References b, MayaFlux::Yantra::GpuResourceManager::buffer_allocated_bytes(), dispatch_key(), MayaFlux::Yantra::GpuResourceManager::download(), find_first_output_index(), MayaFlux::Portal::Graphics::GpuBufferBinding::IMAGE_SAMPLED, MayaFlux::Portal::Graphics::GpuBufferBinding::IMAGE_STORAGE, m_bindings, m_resources, and out.

Referenced by MayaFlux::Yantra::ShaderExecutionContext< InputType, OutputType >::collect_result(), dispatch_core(), dispatch_core_chained(), and dispatch_core_chained_indirect().

Here is the call graph for this function:

Here is the caller graph for this function: