frame_detection_data.h

// Projectname: amos-ss16-proj5
//
// Created on 02.06.2016.
// Copyright (c) 2016 de.fau.cs.osr.amos2016.gruppe5
//
// This file is part of the AMOS Project 2016 @ FAU
// (Friedrich-Alexander University Erlangen-Nürnberg)
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public
// License along with this program. If not, see
// <http://www.gnu.org/licenses/>.
//

#ifndef FRAME_DETECTION_DATA_H
#define FRAME_DETECTION_DATA_H


//std
#include <vector>
#include <map>
#include <string>

//local
#include "element.h"

// enum for the types of detected objects
enum ObjectType { OBJECT_HUMAN = 0, OBJECT_VEHICLE = 1 };


class FrameDetectionData {
    
public:
    
    std::vector<Element> GetElementsOfType(ObjectType type);

    void AddElementsOfType(ObjectType type, std::vector<Element> elements);


private:
    std::map<ObjectType, std::vector<Element> > all_elements_;   
};


#endif // FRAME_DETECTION_DATA_H