sample_regions()

std::vector< Kinesis::Vision::FieldSample > MayaFlux::Yantra::VisionExtractor::sample_regions	(	const std::shared_ptr< Core::VKImage > &	source,
		const Kinesis::Vision::VisionAnalysis &	analysis,
		uint32_t	source_w,
		uint32_t	source_h
	)

sample() for every find_elements box in one dispatch, in box order.

Definition at line 504 of file VisionExtractor.cpp.

{
    if (!analysis.find_elements || analysis.find_elements->boxes.empty())
        return {};
 
    const auto& boxes = analysis.find_elements->boxes;
    const auto count = static_cast<uint32_t>(boxes.size());
 
    std::vector<uint32_t> rects;
    rects.reserve(boxes.size() * 4);
    for (const auto& b : boxes) {
        const auto rect = pixel_rect(b, source_w, source_h);
        rects.insert(rects.end(), rect.begin(), rect.end());
    }
 
    auto& ctx = reduce_context();
    use_shader(ctx, "region_sample_batch.comp.spv", k_wg_reduce, sizeof(BatchCountPC));
    ctx.stage_image(source);
    ctx.set_binding_data(3, std::span<const uint32_t>(rects));
    ctx.set_push_constants(BatchCountPC { .count = count });
 
    const auto raw = dispatch_reduce(ctx, 256U * count, static_cast<size_t>(count) * 3 * sizeof(glm::vec4));
    std::vector<Kinesis::Vision::FieldSample> out;
    if (raw.size() < static_cast<size_t>(count) * 12)
        return out;
 
    out.reserve(count);
    for (uint32_t i = 0; i < count; ++i)
        out.push_back(field_sample(raw.data() + static_cast<size_t>(i) * 12));
    return out;
}

References b, count, MayaFlux::Kinesis::Vision::VisionAnalysis::find_elements, out, and reduce_context().

Referenced by run_operation().

Here is the call graph for this function:

Here is the caller graph for this function: