ObjectDataMap.h

//
// Created by wrede on 22.04.16.
//

#ifndef GBMOT_OBJECTDATAMAP_H
#define GBMOT_OBJECTDATAMAP_H

#include <string>
#include <unordered_map>
#include <vector>
#include <cmath>
#include "ObjectData.h"

namespace core
{
    //TODO RENAME
    class ObjectDataMap : public ObjectData
    {
    private:
        std::unordered_map<std::string, std::pair<double, double>> value_weight_map_;

        virtual void Print(std::ostream& os) const;
    public:
        ObjectDataMap(size_t frame_index);

        ObjectDataMap(
                size_t frame_index,
                std::vector<std::string> keys,
                std::vector<double> value_list);

        ObjectDataMap(
                size_t frame_index,
                std::vector<std::string> keys,
                std::vector<double> value_list,
                std::vector<double> weight_list);

        ObjectDataMap(
                size_t frame_index,
                std::vector<std::string> keys,
                std::vector<std::pair<double, double>> value_weight_list);

        double GetValue(std::string key);

        double GetWeight(std::string key);

        //TODO RENAME
        void Put(std::string key, double value, double weight);

        void Put(std::string key, std::pair<double, double> value_weight);

        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;
    };
}


#endif //GBMOT_OBJECTDATAMAP_H