NStage.h

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

#ifndef GBMOT_KOERNERTRACKING_H
#define GBMOT_KOERNERTRACKING_H

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

namespace algo
{
    class NStage
    {
    private:
        const size_t max_frame_skip_;

        const double penalty_values_;

        const size_t max_tracklet_counts_;
    public:
        NStage(size_t max_frame_skip, double penalty_value,
                 size_t max_tracklet_count);

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

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

        void ExtractTracks(DirectedGraph& tlt_graph,
                           size_t depth,
                           std::vector<core::TrackletPtr>& tracks);
    };
}


#endif //GBMOT_KOERNERTRACKING_H