on_before_gpu_dispatch()

void MayaFlux::Yantra::TextureExecutionContext::on_before_gpu_dispatch ( const std::vector< std::vector< double > > &  , const DataStructureInfo &    )

inlineoverrideprotectedvirtual

Stages the pending TextureContainer at the declared input image binding and, in CONTAINER or IMAGE mode, allocates the output storage image at the declared output binding.

Reimplemented from MayaFlux::Yantra::GpuDispatchCore.

Definition at line 402 of file TextureExecutionContext.hpp.

    {
        if (m_pending_container) {
            const auto sampler = Portal::Graphics::SamplerForge::instance().get_default_linear();
            std::shared_ptr<Core::VKImage> img;
            if (m_pending_container->get_layer_count() > 1) {
                if (!m_upload_staging) {
                    m_upload_staging = Buffers::create_image_staging_buffer(
                        m_pending_container->byte_size() * m_pending_container->get_layer_count());
                }
                img = m_pending_container->to_image_array(m_upload_staging);
            } else {
                if (!m_upload_staging) {
                    m_upload_staging = Buffers::create_image_staging_buffer(
                        m_pending_container->byte_size());
                }
                img = m_pending_container->to_image(m_pending_layer, m_upload_staging);
            }
 
            auto& in_slot = input_slot();
            if (in_slot.binding.element_type == GpuBufferBinding::ElementType::IMAGE_STORAGE) {
                stage_image_at(in_slot.binding.binding, img, GpuBufferBinding::ElementType::IMAGE_STORAGE);
            } else {
                stage_image_at(in_slot.binding.binding, img, GpuBufferBinding::ElementType::IMAGE_SAMPLED, sampler);
            }
            in_slot.image = img;
 
            if (m_output_mode == OutputMode::CONTAINER
                || m_output_mode == OutputMode::IMAGE) {
                const uint32_t ow = m_output_dim_override
                    ? m_output_dim_override->first
                    : m_pending_container->get_width();
                const uint32_t oh = m_output_dim_override
                    ? m_output_dim_override->second
                    : m_pending_container->get_height();
                prepare_output_image(ow, oh);
            }
        }
    }