DetectionSequence.h

//
// Created by wrede on 19.04.16.
//

#ifndef GBMOT_DETECTIONSEQUENCE_H
#define GBMOT_DETECTIONSEQUENCE_H


#include <string>
#include "ObjectData.h"

namespace core
{
    class DetectionSequence
    {
    private:
        std::string name_;

        std::unordered_map<size_t, std::vector<ObjectDataPtr>> objects_;
        //TODO ORIGINAL
//        std::vector<std::vector<ObjectDataPtr>> objects_;

        size_t frame_offset_;

        size_t frame_count_;
    public:
        DetectionSequence(std::string const & name = "DetectionSequence");

        void AddObject(ObjectDataPtr object_data);

        void Clear();

        std::string GetName() const;

        ObjectDataPtr GetObject(size_t frame_index, size_t object_index);

        size_t GetFrameCount() const;

        size_t GetFrameOffset() const;

        size_t GetObjectCount(size_t frame_index);

        friend std::ostream & operator<<(std::ostream & os, DetectionSequence const & obj);
    };
}


#endif //GBMOT_DETECTIONSEQUENCE_H