select_element()

std::optional< size_t > MayaFlux::Yantra::VisionExtractor::select_element	(	const Kinesis::Vision::VisionAnalysis &	analysis,
		const ElementFocus &	focus
	)

Index into find_elements boxes of the element ranked first by focus, chosen on the GPU from the analyzer's own area, shape and nearest-neighbour results.

Empty without find_elements boxes.

Definition at line 299 of file VisionExtractor.cpp.

{
    if (!analysis.find_elements || analysis.find_elements->boxes.empty())
        return std::nullopt;
 
    const auto& elements = *analysis.find_elements;
    const auto count = elements.boxes.size();
 
    std::vector<float> metrics;
    metrics.reserve(count * 6);
    for (size_t i = 0; i < count; ++i) {
        const auto& box = elements.boxes[i];
        const uint32_t label = box.label_id != 0 ? box.label_id : static_cast<uint32_t>(i + 1U);
        const auto contour = std::ranges::find_if(elements.contours,
            [label](const auto& c) { return c.label_id == label; });
 
        metrics.push_back(contour != elements.contours.end() ? contour->area : box.w * box.h);
        metrics.push_back(i < elements.shapes.size() ? elements.shapes[i].compactness : 0.0F);
        metrics.push_back(i < elements.shapes.size() ? elements.shapes[i].aspect_ratio : 0.0F);
        metrics.push_back(i < elements.nearest_neighbor_distance_px.size() ? elements.nearest_neighbor_distance_px[i] : 0.0F);
        metrics.push_back(box.x + box.w * 0.5F);
        metrics.push_back(box.y + box.h * 0.5F);
    }
 
    auto& ctx = reduce_context();
    use_shader(ctx, "element_select.comp.spv", k_wg_reduce, sizeof(SelectPC));
    ctx.set_binding_data(3, std::span<const float>(metrics));
    ctx.set_push_constants(SelectPC {
        .count = static_cast<uint32_t>(count),
        .criterion = static_cast<uint32_t>(focus.criterion),
        .largest = focus.largest ? 1U : 0U,
        .px = focus.point.x,
        .py = focus.point.y,
    });
 
    const auto raw = dispatch_reduce(ctx, 256, 2 * sizeof(float));
    if (raw.empty())
        return std::nullopt;
 
    const auto index = static_cast<size_t>(std::lround(raw[0]));
    return index < count ? std::optional<size_t> { index } : std::nullopt;
}

References count, criterion, MayaFlux::Yantra::ElementFocus::criterion, MayaFlux::Kinesis::Vision::VisionAnalysis::find_elements, index, label, MayaFlux::Yantra::ElementFocus::largest, MayaFlux::Yantra::ElementFocus::point, and reduce_context().

Referenced by focus_box_index(), label_for_analysis(), and run_operation().

Here is the call graph for this function:

Here is the caller graph for this function: