2e9a7d9c1a
Makes turn restrictions into dedicated structures and diferentiates between them via a variant. Ensures that we do not accidentally mess up ID types within our application. In addition this improves the restriction performance by only parsing all edges once at the cost of (at the time of writing) 22MB in terms of main memory usage.
72 lines
2.3 KiB
C++
72 lines
2.3 KiB
C++
#ifndef EXTRACTION_CONTAINERS_HPP
|
|
#define EXTRACTION_CONTAINERS_HPP
|
|
|
|
#include "extractor/first_and_last_segment_of_way.hpp"
|
|
#include "extractor/internal_extractor_edge.hpp"
|
|
#include "extractor/query_node.hpp"
|
|
#include "extractor/restriction.hpp"
|
|
#include "extractor/scripting_environment.hpp"
|
|
|
|
#include "storage/io.hpp"
|
|
|
|
namespace osrm
|
|
{
|
|
namespace extractor
|
|
{
|
|
|
|
/**
|
|
* Uses memory containers to store all the data that
|
|
* is collected by the extractor callbacks.
|
|
*
|
|
* The data is the filtered, aggregated and finally written to disk.
|
|
*/
|
|
class ExtractionContainers
|
|
{
|
|
void PrepareNodes();
|
|
void PrepareRestrictions();
|
|
void PrepareEdges(ScriptingEnvironment &scripting_environment);
|
|
|
|
void WriteNodes(storage::io::FileWriter &file_out) const;
|
|
void WriteConditionalRestrictions(const std::string &restrictions_file_name);
|
|
void WriteEdges(storage::io::FileWriter &file_out) const;
|
|
void WriteCharData(const std::string &file_name);
|
|
|
|
public:
|
|
using NodeIDVector = std::vector<OSMNodeID>;
|
|
using NodeVector = std::vector<QueryNode>;
|
|
using EdgeVector = std::vector<InternalExtractorEdge>;
|
|
using WayIDStartEndVector = std::vector<FirstAndLastSegmentOfWay>;
|
|
using NameCharData = std::vector<unsigned char>;
|
|
using NameOffsets = std::vector<unsigned>;
|
|
|
|
std::vector<OSMNodeID> barrier_nodes;
|
|
std::vector<OSMNodeID> traffic_lights;
|
|
NodeIDVector used_node_id_list;
|
|
NodeVector all_nodes_list;
|
|
EdgeVector all_edges_list;
|
|
NameCharData name_char_data;
|
|
NameOffsets name_offsets;
|
|
// an adjacency array containing all turn lane masks
|
|
WayIDStartEndVector way_start_end_id_list;
|
|
|
|
unsigned max_internal_node_id;
|
|
|
|
// list of restrictions before we transform them into the output types
|
|
std::vector<InputConditionalTurnRestriction> restrictions_list;
|
|
|
|
// turn restrictions split into conditional and unconditional turn restrictions
|
|
std::vector<ConditionalTurnRestriction> conditional_turn_restrictions;
|
|
std::vector<TurnRestriction> unconditional_turn_restrictions;
|
|
|
|
ExtractionContainers();
|
|
|
|
void PrepareData(ScriptingEnvironment &scripting_environment,
|
|
const std::string &osrm_path,
|
|
const std::string &restrictions_file_name,
|
|
const std::string &names_data_path);
|
|
};
|
|
}
|
|
}
|
|
|
|
#endif /* EXTRACTION_CONTAINERS_HPP */
|