sort_implementation()

template<ComputeData InputType = std::shared_ptr<Kakshya::SignalSourceContainer>, ComputeData OutputType = InputType>

output_type MayaFlux::Yantra::VisionSorter< InputType, OutputType >::sort_implementation ( const input_type &  input )

inlineoverrideprotectedvirtual

Pure virtual sorting implementation - derived classes implement this.

Parameters

input

Input data with metadata

Returns

Raw sorting output before granularity processing

Implements MayaFlux::Yantra::UniversalSorter< InputType, OutputType >.

Definition at line 213 of file VisionSorter.hpp.

    {
        const auto analysis_it = input.metadata.find("vision_analysis");
        if (analysis_it == input.metadata.end() || !analysis_it->second.has_value()) {
            error<std::runtime_error>(
                Journal::Component::Yantra,
                Journal::Context::ComputeMatrix,
                std::source_location::current(),
                "VisionSorter: no vision_analysis in input metadata");
        }
 
        auto analysis = safe_any_cast_or_throw<VisionAnalysis>(analysis_it->second);
 
        const bool descending = (this->get_direction() == SortingDirection::DESCENDING);
 
        auto sort_collection = [&]<typename T>(std::vector<T>& coll) {
            const size_t n = coll.size();
            std::vector<size_t> idx(n);
            std::iota(idx.begin(), idx.end(), 0);
            std::stable_sort(idx.begin(), idx.end(),
                [&](size_t a, size_t b) {
                    const float ka = m_key_fn
                        ? m_key_fn(a, analysis)
                        : key_for(a, coll, analysis);
                    const float kb = m_key_fn
                        ? m_key_fn(b, analysis)
                        : key_for(b, coll, analysis);
                    return descending ? ka > kb : ka < kb;
                });
            std::vector<T> sorted(n);
            for (size_t i = 0; i < n; ++i)
                sorted[i] = coll[idx[i]];
            coll = std::move(sorted);
        };
 
        if (!analysis.frame.boxes.empty())
            sort_collection(analysis.frame.boxes);
        if (!analysis.frame.contours.empty())
            sort_collection(analysis.frame.contours);
 
        output_type out { input.data };
        out.metadata = input.metadata;
        out.metadata["vision_analysis"] = std::move(analysis);
        return out;
    }

References a, b, MayaFlux::Journal::ComputeMatrix, MayaFlux::Yantra::Datum< T >::data, MayaFlux::Yantra::UniversalSorter< InputType, OutputType >::get_direction(), input, MayaFlux::Yantra::Datum< T >::metadata, and MayaFlux::Journal::Yantra.

Here is the call graph for this function: