sample_full_frame()

Kinesis::Vision::FieldSample MayaFlux::Yantra::VisionAnalyzer::sample_full_frame ( const std::shared_ptr< Core::VKImage > &  image, uint32_t  w, uint32_t  h  )

private

Dispatch region_sample.comp over the whole image (no region concept for MeasureAppearance).

Definition at line 290 of file VisionAnalyzer.cpp.

{
    if (!m_gradient_sample_ctx) {
        m_gradient_sample_ctx = std::make_shared<TextureExecutionContext>(
            GpuComputeConfig {
                .shader_path = "region_sample.comp.spv",
                .workgroup_size = { 256, 1, 1 },
                .push_constant_size = sizeof(RegionSamplePC) },
            Portal::Graphics::ImageFormat::RGBA8,
            TextureExecutionContext::OutputMode::SCALAR,
            1,
            std::vector<GpuBufferBinding> {
                { .set = 0, .binding = 2, .direction = GpuBufferBinding::Direction::OUTPUT, .element_type = GpuBufferBinding::ElementType::FLOAT32 } });
    }
 
    auto& foundry = Portal::Graphics::get_shader_foundry();
 
    m_gradient_sample_ctx->stage_image(image);
    m_gradient_sample_ctx->set_push_constants(RegionSamplePC { .px = 0, .py = 0, .pw = w, .ph = h });
    m_gradient_sample_ctx->set_output_size(2, 3 * sizeof(glm::vec4));
    m_gradient_sample_ctx->set_output_dimensions(1, 1);
 
    const auto fence = m_gradient_sample_ctx->dispatch_async({});
    m_gradient_sample_ctx->clear_output_dimensions();
    foundry.wait_for_fence(fence);
    foundry.release_fence(fence);
 
    const auto gpu_result = m_gradient_sample_ctx->collect_result();
 
    Kinesis::Vision::FieldSample out;
    if (auto it = gpu_result.aux.find(2); it != gpu_result.aux.end() && it->second.size() >= 3 * sizeof(glm::vec4)) {
        std::array<glm::vec4, 3> raw {};
        std::memcpy(raw.data(), it->second.data(), sizeof(raw));
        out.mean = raw[0];
        out.min_val = raw[1];
        out.max_val = raw[2];
    }
    return out;
}

References fence, MayaFlux::Portal::Graphics::GpuBufferBinding::FLOAT32, MayaFlux::Portal::Graphics::get_shader_foundry(), h, image, m_gradient_sample_ctx, out, MayaFlux::Portal::Graphics::GpuBufferBinding::OUTPUT, MayaFlux::Yantra::TextureExecutionContext::SCALAR, and MayaFlux::Portal::Graphics::GpuComputeConfig::shader_path.

Referenced by measure_appearance().

Here is the call graph for this function:

Here is the caller graph for this function: