geodata.hpp

Go to the documentation of this file.

#pragma once
#ifndef GEODATA_HPP
#define GEODATA_HPP


#include "settings.hpp"
#include <fstream>
#include <boost/archive/text_oarchive.hpp>
#include <boost/archive/text_iarchive.hpp>
#include <boost/serialization/vector.hpp>
#include <boost/serialization/shared_ptr.hpp>
#include <boost/serialization/shared_ptr_132.hpp>

class Node;
class Way;
class Relation;
class NodeKdTree;
template<class id_t, class data_t>
class RTree;

class Geodata
{
public:
    Geodata() = default;
    virtual ~Geodata() = default;

    TESTABLE void insertNodes(const shared_ptr<std::vector<Node> >& nodes);
    TESTABLE void insertWays(const shared_ptr<std::vector<Way> >& ways);
    TESTABLE void insertRelations(const shared_ptr<std::vector<Relation> >& relations);

    bool containsData(const FixedRect &rect) const;
    TESTABLE shared_ptr<std::vector<NodeId> > getNodeIDs(const FixedRect& rect) const;

    TESTABLE shared_ptr<std::vector<WayId> > getWayIDs(const FixedRect& rect) const;
    TESTABLE shared_ptr<std::vector<RelId> > getRelationIDs(const FixedRect& rect) const;

    TESTABLE Node* getNode(NodeId id) const;
    TESTABLE Way* getWay(WayId id) const ;
    TESTABLE Relation* getRelation(RelId id) const;

    TESTABLE void load(const string& path);
    TESTABLE void save(const string& path);

protected:
    shared_ptr<std::vector<Way> > ways;
    shared_ptr<std::vector<Node> > nodes;
    shared_ptr<std::vector<Relation> > relations;

    shared_ptr<RTree<NodeId, FixedPoint>> nodesTree;
    shared_ptr<RTree<WayId, FixedRect>> waysTree;
    shared_ptr<RTree<RelId, FixedRect>> relTree;

private:
    void buildTrees(const string& nodePath, const string& wayPath, const string& relationPath);
    void serialize(const string& serPath) const;
    TESTABLE FixedRect calculateBoundingBox(const Way& way) const;
    TESTABLE FixedRect calculateBoundingBox(const Relation& relation) const;
    FixedRect calculateBoundingBox(const std::vector<NodeId>& nodeIDs) const;
    FixedRect calculateBoundingBox(const std::vector<WayId>& nodeIDs) const;

    friend class boost::serialization::access;
    template<typename Archive>
    void serialize(Archive &ar, const unsigned int version){
        ar & nodes;
        ar & ways;
        ar & relations;
        ar & nodesTree;
        ar & waysTree;
        ar & relTree;
    }

};

#endif