collect_gpu_outputs()

output_type MayaFlux::Yantra::TextureExecutionContext::collect_gpu_outputs ( const GpuChannelResult &  , const std::vector< std::vector< double > > &  , const DataStructureInfo &    )

inlineoverrideprotectedvirtual

In CONTAINER mode: downloads the storage image at the declared output binding into a new TextureContainer placed in output.container.

In SCALAR mode: returns an empty DataIO (use collect_result() instead).

Reimplemented from MayaFlux::Yantra::GpuExecutionContext< std::vector< Kakshya::DataVariant >, std::vector< Kakshya::DataVariant > >.

Definition at line 479 of file TextureExecutionContext.hpp.

    {
        if (m_output_mode == OutputMode::SCALAR)
            return output_type {};
 
        auto img = get_output_image(output_slot().binding.binding);
        if (!img) {
            error<std::runtime_error>(
                Journal::Component::Yantra,
                Journal::Context::BufferProcessing,
                std::source_location::current(),
                "TextureExecutionContext: no output image at declared output binding after dispatch");
        }
 
        Portal::Graphics::TextureLoom::instance().transition_layout(
            img, vk::ImageLayout::eGeneral, vk::ImageLayout::eShaderReadOnlyOptimal);
 
        if (m_output_mode == OutputMode::IMAGE)
            return output_type {};
 
        const uint32_t w = m_output_dim_override
            ? m_output_dim_override->first
            : (m_pending_container ? m_pending_container->get_width() : img->get_width());
        const uint32_t h = m_output_dim_override
            ? m_output_dim_override->second
            : (m_pending_container ? m_pending_container->get_height() : img->get_height());
 
        if (!m_output_container) {
            m_output_container = std::make_shared<Kakshya::TextureContainer>(
                w, h, m_output_format);
        }
 
        if (!m_download_staging) {
            m_download_staging = Buffers::create_image_staging_buffer(
                m_output_container->byte_size());
        }
 
        m_output_container->from_image(img, m_download_staging, 0);
 
        output_type result;
        result.container = std::static_pointer_cast<Kakshya::SignalSourceContainer>(
            m_output_container);
        return result;
    }

References MayaFlux::Yantra::Datum< T >::container, and h.