- separates node-based graph creation and compression from edge-based graph creation - moves usage of edge-based node data-container to pre-processing as well, unifying access to node-based data - single struct instead of separate vectors for annotation data in engine (single place of modification)
76 lines
2.6 KiB
C++
76 lines
2.6 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 WriteEdges(storage::io::FileWriter &file_out) const;
|
|
void WriteMetadata(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 AnnotationDataVector = std::vector<NodeBasedEdgeAnnotation>;
|
|
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_signals;
|
|
NodeIDVector used_node_id_list;
|
|
NodeVector all_nodes_list;
|
|
EdgeVector all_edges_list;
|
|
AnnotationDataVector all_edges_annotation_data_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. Input containers
|
|
// reference OSMNodeIDs. We can only transform them to the correct internal IDs after we've read
|
|
// everything. Without a multi-parse approach, we have to remember the output restrictions
|
|
// before converting them to the internal formats
|
|
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 &names_data_path);
|
|
};
|
|
}
|
|
}
|
|
|
|
#endif /* EXTRACTION_CONTAINERS_HPP */
|