#pragma once

#include <mbgl/annotation/annotation.hpp>
#include <mbgl/annotation/symbol_annotation_impl.hpp>
#include <mbgl/style/image.hpp>
#include <mbgl/util/noncopyable.hpp>

#include <mutex>
#include <string>
#include <vector>
#include <unordered_set>
#include <unordered_map>

#include <mapbox/std/weak.hpp>

namespace mbgl {

class LatLngBounds;
class AnnotationTile;
class AnnotationTileData;
class SymbolAnnotationImpl;
class ShapeAnnotationImpl;

namespace style {
class Style;
} // namespace style

class AnnotationManager : private util::noncopyable {
public:
    AnnotationManager(style::Style&);
    ~AnnotationManager();

    AnnotationID addAnnotation(const Annotation&);
    bool updateAnnotation(const AnnotationID&, const Annotation&);
    void removeAnnotation(const AnnotationID&);

    void addImage(std::unique_ptr<style::Image>);
    void removeImage(const std::string&);
    double getTopOffsetPixelsForImage(const std::string&);

    void setStyle(style::Style&);
    void onStyleLoaded();

    void updateData();

    void addTile(AnnotationTile&);
    void removeTile(AnnotationTile&);

    static const std::string SourceID;
    static const std::string PointLayerID;
    static const std::string ShapeLayerID;

    mapbox::base::WeakPtr<AnnotationManager> makeWeakPtr() { return weakFactory.makeWeakPtr(); }

private:
    void add(const AnnotationID&, const SymbolAnnotation&);
    void add(const AnnotationID&, const LineAnnotation&);
    void add(const AnnotationID&, const FillAnnotation&);

    void update(const AnnotationID&, const SymbolAnnotation&);
    void update(const AnnotationID&, const LineAnnotation&);
    void update(const AnnotationID&, const FillAnnotation&);

    void remove(const AnnotationID&);

    void updateStyle();

    std::unique_ptr<AnnotationTileData> getTileData(const CanonicalTileID&);

    std::reference_wrapper<style::Style> style;

    std::mutex mutex;

    bool dirty = false;
    
    AnnotationID nextID = 0;

    using SymbolAnnotationTree = boost::geometry::index::rtree<std::shared_ptr<const SymbolAnnotationImpl>, boost::geometry::index::rstar<16, 4>>;
    // Unlike std::unordered_map, std::map is guaranteed to sort by AnnotationID, ensuring that older annotations are below newer annotations.
    // <https://github.com/mapbox/mapbox-gl-native/issues/5691>
    using SymbolAnnotationMap = std::map<AnnotationID, std::shared_ptr<SymbolAnnotationImpl>>;
    using ShapeAnnotationMap = std::map<AnnotationID, std::unique_ptr<ShapeAnnotationImpl>>;
    using ImageMap = std::unordered_map<std::string, style::Image>;

    SymbolAnnotationTree symbolTree;
    SymbolAnnotationMap symbolAnnotations;
    ShapeAnnotationMap shapeAnnotations;
    ImageMap images;

    std::unordered_set<AnnotationTile*> tiles;
    mapbox::base::WeakPtrFactory<AnnotationManager> weakFactory{this};
};

} // namespace mbgl