NStage.h

//
// Created by wrede on 25.04.16.
//

#ifndef GBMOT_NSTAGE_H
#define GBMOT_NSTAGE_H

#include "../core/DetectionSequence.h"
#include "../core/Tracklet.h"
#include "../graph/Definitions.h"

namespace algo
{
    class NStage
    {
    private:
        std::vector<size_t> max_frame_skips_;

        std::vector<double> penalty_values_;

        std::vector<size_t> max_tracklet_counts_;

        size_t iterations_;

        double edge_weight_threshold_;

        std::vector<std::unordered_map<std::string, double>> constraints_;

        void CreateObjectGraph(DirectedGraph & graph, core::DetectionSequence & detections);

        void CreateTrackletGraph(DirectedGraph & obj_graph, DirectedGraph & tlt_graph,
                                 size_t frame_count, size_t iteration);

        void ExtractTracks(DirectedGraph& tlt_graph, size_t depth,
                           std::vector<core::TrackletPtr> & tracks);
    public:
        NStage(std::vector<size_t> max_frame_skip,
               std::vector<double> penalty_value,
               std::vector<size_t> max_tracklet_count,
               double edge_weight_threshold,
               std::vector<std::unordered_map<std::string, double>> constraints);

        void Run(core::DetectionSequence & sequence,
                 std::vector<core::TrackletPtr> & tracks);
    };
}


#endif //GBMOT_NSTAGE_H