reduce_tracks()

Kinesis::Vision::TrackObjectsAnalysis MayaFlux::Yantra::VisionAnalyzer::reduce_tracks ( const std::vector< Kinesis::Vision::TrackResult > &  tracks, uint32_t  w, uint32_t  h  )

private

Dispatch track_reduce.comp over tracks and fill centroid/velocity/bounds/mean_speed_px/age stats on the result.

No-op (default fields) when tracks is empty.

Definition at line 91 of file VisionAnalyzer.cpp.

{
    Kinesis::Vision::TrackObjectsAnalysis result;
    result.tracks = tracks;
 
    if (tracks.empty())
        return result;
 
    std::vector<float> track_data;
    track_data.reserve(tracks.size() * 4);
    std::vector<float> extra_data;
    extra_data.reserve(tracks.size() * 2);
    for (const auto& t : tracks) {
        track_data.push_back(t.position.x);
        track_data.push_back(t.position.y);
        track_data.push_back(t.previous.x);
        track_data.push_back(t.previous.y);
        extra_data.push_back(static_cast<float>(t.age));
        extra_data.push_back(t.tracked ? 1.0F : 0.0F);
    }
 
    if (!m_track_reducer) {
        m_track_reducer = std::make_shared<ShaderExecutionContext<>>(
            GpuComputeConfig {
                .shader_path = "track_reduce.comp.spv",
                .workgroup_size = { 256, 1, 1 },
                .push_constant_size = sizeof(TrackReducePC) });
    }
 
    m_track_reducer->input(0, track_data)
        .output(1, 3 * sizeof(glm::vec4))
        .input(2, extra_data)
        .push(TrackReducePC { .count = static_cast<uint32_t>(tracks.size()) });
 
    const auto output = m_track_reducer->execute(Datum<std::vector<Kakshya::DataVariant>> {}, ExecutionContext {});
    const auto raw = ShaderExecutionContext<>::read_output<float>(output, 1);
 
    if (raw.size() >= 12) {
        const auto n = static_cast<float>(tracks.size());
        result.centroid = { raw[0] / n, raw[1] / n };
        result.velocity = { raw[2] / n, raw[3] / n };
        result.bounds = {
            .x = raw[4],
            .y = raw[5],
            .w = raw[6] - raw[4],
            .h = raw[7] - raw[5],
        };
        result.mean_speed_px = glm::length(glm::vec2 {
            result.velocity.x * static_cast<float>(w),
            result.velocity.y * static_cast<float>(h) });
        result.mean_age = raw[8] / n;
        result.max_age = static_cast<uint32_t>(std::lround(raw[9]));
        result.tracked_count = static_cast<uint32_t>(std::lround(raw[10]));
        result.lost_count = static_cast<uint32_t>(tracks.size()) - result.tracked_count;
    }
 
    return result;
}

References MayaFlux::Kinesis::Vision::TrackObjectsAnalysis::bounds, MayaFlux::Kinesis::Vision::TrackObjectsAnalysis::centroid, h, input, MayaFlux::Kinesis::Vision::TrackObjectsAnalysis::lost_count, m_track_reducer, MayaFlux::Kinesis::Vision::TrackObjectsAnalysis::max_age, MayaFlux::Kinesis::Vision::TrackObjectsAnalysis::mean_age, MayaFlux::Kinesis::Vision::TrackObjectsAnalysis::mean_speed_px, MayaFlux::Portal::Graphics::GpuComputeConfig::shader_path, MayaFlux::Kinesis::Vision::TrackObjectsAnalysis::tracked_count, MayaFlux::Kinesis::Vision::TrackObjectsAnalysis::tracks, MayaFlux::Kinesis::Vision::TrackObjectsAnalysis::velocity, and MayaFlux::Kinesis::Vision::BoundingBox::x.

Referenced by assemble().

Here is the caller graph for this function: