#ifndef MMFLOORGROUNDTRUTHPOINTS_H
#define MMFLOORGROUNDTRUTHPOINTS_H

#include "MapLayer.h"
#include "MMFloorGroundTruthPoint.h"

#include <Indoor/floorplan/v2/Floorplan.h>

/**
 * layer that contains all of one floor's GroundTruthPoints
 */
class MMFloorGroundTruthPoints : public MapLayer {

    Floorplan::Floor* floor;

public:

    /** ctor with the floor */
    MMFloorGroundTruthPoints(MapLayer* parent, Floorplan::Floor* floor) : MapLayer(parent, MapLayerType::FLOOR_GROUND_TRUTH_POINTS), floor(floor) {

        // the POIs
        for (Floorplan::GroundTruthPoint* gtp : floor->gtpoints) {
            addElement(new MMFloorGroundTruthPoint(this, floor, gtp));
        }

    }

    /** get the corresponding floor from the underlying model */
    Floorplan::Floor* getFloor() {return floor;}

    //TODO: check
	MMFloorGroundTruthPoint* createGroundTruthPoint(Floorplan::GroundTruthPoint* gtp) {

        // add to underlying model
        floor->gtpoints.push_back(gtp);

        // add to myself as element
		//addElement(new MMFloorGroundTruthPoint(this, floor, gtp));

		// add to myself as element
		MMFloorGroundTruthPoint* mm = new MMFloorGroundTruthPoint(this, floor, gtp);
		addElement(mm);
		return mm;

    }

    std::string getLayerName() const override {return "GroundTruthPoints";}

};

#endif // MMFLOORGROUNDTRUTHPOINTS_H