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 305 of file GpuDispatchCore.cpp.

{
    const size_t idx = find_first_output_index();
 
    if (idx < m_bindings.size()) {
        const auto et = m_bindings[idx].element_type;
        if (et == GpuBufferBinding::ElementType::IMAGE_STORAGE
            || et == GpuBufferBinding::ElementType::IMAGE_SAMPLED)
            return {};
    }
 
    const size_t allocated = m_resources.buffer_allocated_bytes(idx);
    const size_t byte_size = std::min(float_count * sizeof(float), allocated);
    std::vector<float> out(byte_size / sizeof(float));
    m_resources.download(idx, out.data(), byte_size);
    return out;
}

References MayaFlux::Yantra::GpuResourceManager::buffer_allocated_bytes(), MayaFlux::Yantra::GpuResourceManager::download(), find_first_output_index(), MayaFlux::Yantra::GpuBufferBinding::IMAGE_SAMPLED, MayaFlux::Yantra::GpuBufferBinding::IMAGE_STORAGE, m_bindings, and m_resources.

Referenced by dispatch_core(), and dispatch_core_chained().

Here is the call graph for this function:

Here is the caller graph for this function: