Tracklet.h

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

#ifndef GBMOT_TRACKLET_H
#define GBMOT_TRACKLET_H

#include <cstdlib>
#include <vector>
#include "ObjectData.h"

namespace core
{
    class Tracklet;
    typedef std::shared_ptr<Tracklet> TrackletPtr;

    class Tracklet : public ObjectData
    {
    private:
        std::vector<ObjectDataPtr> path_objects_;

        size_t last_frame_index_;

        virtual void Print(std::ostream& os) const;
    public:
        Tracklet();

        void AddPathObject(ObjectDataPtr obj, bool overwrite = false);

        size_t GetFirstFrameIndex() const;

        size_t GetLastFrameIndex() const;

        ObjectDataPtr GetPathObject(size_t i);

        size_t GetPathObjectCount() const;

        void InterpolateMissingFrames();

        virtual double CompareTo(ObjectDataPtr obj) const override;

        virtual ObjectDataPtr Interpolate(ObjectDataPtr obj,
                                          double fraction) const override;

        virtual void Visualize(cv::Mat& image, cv::Scalar& color) const override;

        void Visualize(cv::Mat& image, cv::Scalar& color, size_t frame,
                       size_t predecessor_count, size_t successor_count) const;

        void Flatten();
    };
}


#endif //GBMOT_TRACKLET_H