rebase branch onto most recent changes from develop branch
This commit is contained in:
parent
344bdbb707
commit
3c563f7073
@ -45,7 +45,7 @@ EdgeBasedGraphFactory::EdgeBasedGraphFactory(
|
||||
std::unique_ptr<RestrictionMap> restriction_map,
|
||||
std::vector<NodeID> &barrier_node_list,
|
||||
std::vector<NodeID> &traffic_light_node_list,
|
||||
std::vector<NodeInfo> &node_info_list,
|
||||
std::vector<QueryNode> &node_info_list,
|
||||
SpeedProfileProperties &speed_profile)
|
||||
: speed_profile(speed_profile),
|
||||
m_number_of_edge_based_nodes(std::numeric_limits<unsigned>::max()),
|
||||
|
@ -63,7 +63,7 @@ class EdgeBasedGraphFactory
|
||||
std::unique_ptr<RestrictionMap> restricion_map,
|
||||
std::vector<NodeID> &barrier_node_list,
|
||||
std::vector<NodeID> &traffic_light_node_list,
|
||||
std::vector<QueryNode> &m_node_info_list,
|
||||
std::vector<QueryNode> &node_info_list,
|
||||
SpeedProfileProperties &speed_profile);
|
||||
|
||||
void Run(const std::string &original_edge_data_filename,
|
||||
|
@ -178,100 +178,35 @@ int Extractor::Run(int argc, char *argv[])
|
||||
|
||||
RestrictionParser restriction_parser(scripting_environment);
|
||||
|
||||
// move to header
|
||||
std::atomic_bool parsing_done {false};
|
||||
std::atomic_bool loading_done {false};
|
||||
// // move to header
|
||||
// std::atomic_bool parsing_done {false};
|
||||
// std::atomic_bool loading_done {false};
|
||||
|
||||
ConcurrentQueue<std::shared_ptr<osmium::memory::Buffer>> parse_queue(128);
|
||||
ConcurrentQueue<std::shared_ptr<ResultBuffer>> result_queue(128);
|
||||
// ConcurrentQueue<std::shared_ptr<osmium::memory::Buffer>> parse_queue(128);
|
||||
// ConcurrentQueue<std::shared_ptr<ResultBuffer>> result_queue(128);
|
||||
|
||||
std::thread loading_thread([&]{
|
||||
while (osmium::memory::Buffer buffer = reader.read())
|
||||
{
|
||||
parse_queue.push(std::make_shared<osmium::memory::Buffer>(std::move(buffer)));
|
||||
}
|
||||
loading_done = true;
|
||||
});
|
||||
// std::thread loading_thread([&]{
|
||||
// while (osmium::memory::Buffer buffer = reader.read())
|
||||
// {
|
||||
// parse_queue.push(std::make_shared<osmium::memory::Buffer>(std::move(buffer)));
|
||||
// }
|
||||
// loading_done = true;
|
||||
// });
|
||||
|
||||
// parsing threads
|
||||
while (!loading_done || !parse_queue.empty())
|
||||
{
|
||||
std::shared_ptr<osmium::memory::Buffer> current_buffer;
|
||||
if (!parse_queue.try_pop(current_buffer))
|
||||
{
|
||||
continue;
|
||||
}
|
||||
// // parsing threads
|
||||
// while (!loading_done || !parse_queue.empty())
|
||||
// {
|
||||
// std::shared_ptr<osmium::memory::Buffer> current_buffer;
|
||||
// if (!parse_queue.try_pop(current_buffer))
|
||||
// {
|
||||
// continue;
|
||||
// }
|
||||
|
||||
ExtractionNode result_node;
|
||||
ExtractionWay result_way;
|
||||
|
||||
std::shared_ptr<ResultBuffer> result_buffer = std::make_shared<ResultBuffer>();;
|
||||
for (osmium::OSMEntity &entity : *current_buffer)
|
||||
{
|
||||
switch (entity.type())
|
||||
{
|
||||
case osmium::item_type::node:
|
||||
++number_of_nodes;
|
||||
result_node.Clear();
|
||||
luabind::call_function<void>(lua_state,
|
||||
"node_function",
|
||||
boost::cref(static_cast<osmium::Node &>(entity)),
|
||||
boost::ref(result_node));
|
||||
result_buffer->nodes.emplace_back(osmium::NodeRef{static_cast<osmium::Node &>(entity).id(), static_cast<osmium::Node &>(entity).location()}, result_node);
|
||||
break;
|
||||
case osmium::item_type::way:
|
||||
++number_of_ways;
|
||||
result_way.Clear();
|
||||
luabind::call_function<void>(lua_state,
|
||||
"way_function",
|
||||
boost::cref(static_cast<osmium::Way &>(entity)),
|
||||
boost::ref(result_way));
|
||||
result_way.id = static_cast<osmium::Way &>(entity).id();
|
||||
result_buffer->ways.emplace_back(osmium::WayNodeList{static_cast<osmium::Way &>(entity).nodes()}, result_way);
|
||||
break;
|
||||
case osmium::item_type::relation:
|
||||
++number_of_relations;
|
||||
result_buffer->restrictions.emplace_back(restriction_parser.TryParse(static_cast<osmium::Relation &>(entity)));
|
||||
break;
|
||||
default:
|
||||
++number_of_others;
|
||||
break;
|
||||
}
|
||||
}
|
||||
result_queue.push(result_buffer);
|
||||
parsing_done = true;
|
||||
}
|
||||
|
||||
while (!parsing_done || !result_queue.empty())
|
||||
{
|
||||
std::shared_ptr<ResultBuffer> current_buffer;
|
||||
if (!result_queue.try_pop(current_buffer))
|
||||
{
|
||||
for (const auto &node : current_buffer->nodes)
|
||||
{
|
||||
extractor_callbacks->ProcessNode(node.first,
|
||||
node.second);
|
||||
}
|
||||
for (auto &way : current_buffer->ways)
|
||||
{
|
||||
extractor_callbacks->ProcessWay(way.first,
|
||||
way.second);
|
||||
}
|
||||
for (const auto &restriction : current_buffer->restrictions)
|
||||
{
|
||||
extractor_callbacks->ProcessRestriction(restriction);
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
loading_thread.join();
|
||||
|
||||
// TODO: join parser threads
|
||||
|
||||
// while (osmium::memory::Buffer buffer = reader.read())
|
||||
// {
|
||||
// for (osmium::OSMEntity &entity : buffer)
|
||||
// std::shared_ptr<ResultBuffer> result_buffer = std::make_shared<ResultBuffer>();;
|
||||
// for (osmium::OSMEntity &entity : *current_buffer)
|
||||
// {
|
||||
// switch (entity.type())
|
||||
// {
|
||||
@ -282,8 +217,7 @@ int Extractor::Run(int argc, char *argv[])
|
||||
// "node_function",
|
||||
// boost::cref(static_cast<osmium::Node &>(entity)),
|
||||
// boost::ref(result_node));
|
||||
// extractor_callbacks->ProcessNode(static_cast<osmium::Node &>(entity),
|
||||
// result_node);
|
||||
// result_buffer->nodes.emplace_back(osmium::NodeRef{static_cast<osmium::Node &>(entity).id(), static_cast<osmium::Node &>(entity).location()}, result_node);
|
||||
// break;
|
||||
// case osmium::item_type::way:
|
||||
// ++number_of_ways;
|
||||
@ -292,19 +226,85 @@ int Extractor::Run(int argc, char *argv[])
|
||||
// "way_function",
|
||||
// boost::cref(static_cast<osmium::Way &>(entity)),
|
||||
// boost::ref(result_way));
|
||||
// extractor_callbacks->ProcessWay(static_cast<osmium::Way &>(entity), result_way);
|
||||
// result_way.id = static_cast<osmium::Way &>(entity).id();
|
||||
// result_buffer->ways.emplace_back(std::move(static_cast<osmium::Way &>(entity).nodes()), result_way);
|
||||
// break;
|
||||
// case osmium::item_type::relation:
|
||||
// ++number_of_relations;
|
||||
// extractor_callbacks->ProcessRestriction(
|
||||
// restriction_parser.TryParse(static_cast<osmium::Relation &>(entity)));
|
||||
// result_buffer->restrictions.emplace_back(restriction_parser.TryParse(static_cast<osmium::Relation &>(entity)));
|
||||
// break;
|
||||
// default:
|
||||
// ++number_of_others;
|
||||
// break;
|
||||
// }
|
||||
// }
|
||||
// result_queue.push(result_buffer);
|
||||
// parsing_done = true;
|
||||
// }
|
||||
|
||||
// while (!parsing_done || !result_queue.empty())
|
||||
// {
|
||||
// std::shared_ptr<ResultBuffer> current_buffer;
|
||||
// if (!result_queue.try_pop(current_buffer))
|
||||
// {
|
||||
// for (const auto &node : current_buffer->nodes)
|
||||
// {
|
||||
// extractor_callbacks->ProcessNode(node.first,
|
||||
// node.second);
|
||||
// }
|
||||
// for (auto &way : current_buffer->ways)
|
||||
// {
|
||||
// extractor_callbacks->ProcessWay(way.first,
|
||||
// way.second);
|
||||
// }
|
||||
// for (const auto &restriction : current_buffer->restrictions)
|
||||
// {
|
||||
// extractor_callbacks->ProcessRestriction(restriction);
|
||||
// }
|
||||
|
||||
// }
|
||||
// }
|
||||
|
||||
// loading_thread.join();
|
||||
|
||||
// // TODO: join parser threads
|
||||
|
||||
while (osmium::memory::Buffer buffer = reader.read())
|
||||
{
|
||||
for (osmium::OSMEntity &entity : buffer)
|
||||
{
|
||||
switch (entity.type())
|
||||
{
|
||||
case osmium::item_type::node:
|
||||
++number_of_nodes;
|
||||
result_node.Clear();
|
||||
luabind::call_function<void>(lua_state,
|
||||
"node_function",
|
||||
boost::cref(static_cast<osmium::Node &>(entity)),
|
||||
boost::ref(result_node));
|
||||
extractor_callbacks->ProcessNode(static_cast<osmium::Node &>(entity),
|
||||
result_node);
|
||||
break;
|
||||
case osmium::item_type::way:
|
||||
++number_of_ways;
|
||||
result_way.Clear();
|
||||
luabind::call_function<void>(lua_state,
|
||||
"way_function",
|
||||
boost::cref(static_cast<osmium::Way &>(entity)),
|
||||
boost::ref(result_way));
|
||||
extractor_callbacks->ProcessWay(static_cast<osmium::Way &>(entity), result_way);
|
||||
break;
|
||||
case osmium::item_type::relation:
|
||||
++number_of_relations;
|
||||
extractor_callbacks->ProcessRestriction(
|
||||
restriction_parser.TryParse(static_cast<osmium::Relation &>(entity)));
|
||||
break;
|
||||
default:
|
||||
++number_of_others;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
TIMER_STOP(parsing);
|
||||
SimpleLogger().Write() << "Parsing finished after " << TIMER_SEC(parsing) << " seconds";
|
||||
SimpleLogger().Write() << "Raw input contains " << number_of_nodes << " nodes, "
|
||||
@ -328,11 +328,6 @@ int Extractor::Run(int argc, char *argv[])
|
||||
<< "./osrm-prepare " << extractor_config.output_file_name
|
||||
<< std::endl;
|
||||
}
|
||||
catch (boost::program_options::too_many_positional_options_error &)
|
||||
{
|
||||
SimpleLogger().Write(logWARNING) << "Only one input file can be specified";
|
||||
return 1;
|
||||
}
|
||||
catch (std::exception &e)
|
||||
{
|
||||
SimpleLogger().Write(logWARNING) << e.what();
|
||||
|
@ -48,13 +48,13 @@ ExtractorCallbacks::ExtractorCallbacks(ExtractionContainers &extraction_containe
|
||||
}
|
||||
|
||||
/** warning: caller needs to take care of synchronization! */
|
||||
void ExtractorCallbacks::ProcessNode(const osmium::NodeRef &osm_input_node,
|
||||
void ExtractorCallbacks::ProcessNode(const osmium::Node &input_node,
|
||||
const ExtractionNode &result_node)
|
||||
{
|
||||
external_memory.all_nodes_list.push_back({
|
||||
static_cast<int>(osm_input_node.location().lat() * COORDINATE_PRECISION),
|
||||
static_cast<int>(osm_input_node.location().lon() * COORDINATE_PRECISION),
|
||||
static_cast<NodeID>(osm_input_node.ref()),
|
||||
static_cast<int>(input_node.location().lat() * COORDINATE_PRECISION),
|
||||
static_cast<int>(input_node.location().lon() * COORDINATE_PRECISION),
|
||||
static_cast<NodeID>(input_node.id()),
|
||||
result_node.barrier,
|
||||
result_node.traffic_lights
|
||||
});
|
||||
@ -69,7 +69,7 @@ void ExtractorCallbacks::ProcessRestriction(
|
||||
}
|
||||
}
|
||||
/** warning: caller needs to take care of synchronization! */
|
||||
void ExtractorCallbacks::ProcessWay(const osmium::WayNodeList &input_way, const ExtractionWay &parsed_way)
|
||||
void ExtractorCallbacks::ProcessWay(const osmium::Way &input_way, const ExtractionWay &parsed_way)
|
||||
{
|
||||
if (((0 >= parsed_way.forward_speed) ||
|
||||
(TRAVEL_MODE_INACCESSIBLE == parsed_way.forward_travel_mode)) &&
|
||||
@ -80,7 +80,7 @@ void ExtractorCallbacks::ProcessWay(const osmium::WayNodeList &input_way, const
|
||||
return;
|
||||
}
|
||||
|
||||
if (input_way.size() <= 1)
|
||||
if (input_way.nodes().size() <= 1)
|
||||
{ // safe-guard against broken data
|
||||
return;
|
||||
}
|
||||
@ -88,16 +88,15 @@ void ExtractorCallbacks::ProcessWay(const osmium::WayNodeList &input_way, const
|
||||
if (std::numeric_limits<decltype(parsed_way.id)>::max() == parsed_way.id)
|
||||
{
|
||||
SimpleLogger().Write(logDEBUG) << "found bogus way with id: " << parsed_way.id
|
||||
<< " of size " << input_way.size();
|
||||
<< " of size " << input_way.nodes().size();
|
||||
return;
|
||||
}
|
||||
|
||||
if (0 < parsed_way.duration)
|
||||
{
|
||||
// TODO: iterate all way segments and set duration corresponding to the length of each
|
||||
// segment
|
||||
const_cast<ExtractionWay&>(parsed_way).forward_speed = parsed_way.duration / (input_way.size() - 1);
|
||||
const_cast<ExtractionWay&>(parsed_way).backward_speed = parsed_way.duration / (input_way.size() - 1);
|
||||
const_cast<ExtractionWay&>(parsed_way).forward_speed = parsed_way.duration / (input_way.nodes().size() - 1);
|
||||
const_cast<ExtractionWay&>(parsed_way).backward_speed = parsed_way.duration / (input_way.nodes().size() - 1);
|
||||
}
|
||||
|
||||
if (std::numeric_limits<double>::epsilon() >= std::abs(-1. - parsed_way.forward_speed))
|
||||
@ -154,24 +153,23 @@ void ExtractorCallbacks::ProcessWay(const osmium::WayNodeList &input_way, const
|
||||
const_cast<ExtractionWay&>(parsed_way).forward_travel_mode = parsed_way.backward_travel_mode;
|
||||
const_cast<ExtractionWay&>(parsed_way).backward_travel_mode = TRAVEL_MODE_INACCESSIBLE;
|
||||
osrm::for_each_pair(
|
||||
input_way.crbegin(), input_way.crend(), pair_wise_segment_split);
|
||||
external_memory.used_node_id_list.push_back(input_way.front().ref());
|
||||
input_way.nodes().crbegin(), input_way.nodes().crend(), pair_wise_segment_split);
|
||||
external_memory.used_node_id_list.push_back(input_way.nodes().front().ref());
|
||||
}
|
||||
else
|
||||
{
|
||||
osrm::for_each_pair(
|
||||
input_way.cbegin(), input_way.cend(), pair_wise_segment_split);
|
||||
external_memory.used_node_id_list.push_back(input_way.back().ref());
|
||||
input_way.nodes().cbegin(), input_way.nodes().cend(), pair_wise_segment_split);
|
||||
external_memory.used_node_id_list.push_back(input_way.nodes().back().ref());
|
||||
}
|
||||
|
||||
// The following information is needed to identify start and end segments of restrictions
|
||||
// The following information is needed to identify start and end segments of restrictions
|
||||
external_memory.way_start_end_id_list.push_back(
|
||||
{(EdgeID)parsed_way.id,
|
||||
(NodeID)input_way[0].ref(),
|
||||
(NodeID)input_way[1].ref(),
|
||||
(NodeID)input_way[input_way.size() - 2].ref(),
|
||||
(NodeID)input_way.back().ref()});
|
||||
(NodeID)input_way.nodes()[0].ref(),
|
||||
(NodeID)input_way.nodes()[1].ref(),
|
||||
(NodeID)input_way.nodes()[input_way.nodes().size() - 2].ref(),
|
||||
(NodeID)input_way.nodes().back().ref()});
|
||||
|
||||
if (split_edge)
|
||||
{ // Only true if the way should be split
|
||||
@ -198,24 +196,24 @@ void ExtractorCallbacks::ProcessWay(const osmium::WayNodeList &input_way, const
|
||||
if (is_opposite_way)
|
||||
{
|
||||
// SimpleLogger().Write() << "opposite2";
|
||||
osrm::for_each_pair(input_way.crbegin(),
|
||||
input_way.crend(),
|
||||
osrm::for_each_pair(input_way.nodes().crbegin(),
|
||||
input_way.nodes().crend(),
|
||||
pair_wise_segment_split_2);
|
||||
external_memory.used_node_id_list.push_back(input_way.front().ref());
|
||||
external_memory.used_node_id_list.push_back(input_way.nodes().front().ref());
|
||||
}
|
||||
else
|
||||
{
|
||||
osrm::for_each_pair(input_way.cbegin(),
|
||||
input_way.cend(),
|
||||
osrm::for_each_pair(input_way.nodes().cbegin(),
|
||||
input_way.nodes().cend(),
|
||||
pair_wise_segment_split_2);
|
||||
external_memory.used_node_id_list.push_back(input_way.back().ref());
|
||||
external_memory.used_node_id_list.push_back(input_way.nodes().back().ref());
|
||||
}
|
||||
|
||||
external_memory.way_start_end_id_list.push_back(
|
||||
{(EdgeID)parsed_way.id,
|
||||
(NodeID)input_way[1].ref(),
|
||||
(NodeID)input_way[0].ref(),
|
||||
(NodeID)input_way.back().ref(),
|
||||
(NodeID)input_way[input_way.size() - 2].ref()});
|
||||
(NodeID)input_way.nodes()[1].ref(),
|
||||
(NodeID)input_way.nodes()[0].ref(),
|
||||
(NodeID)input_way.nodes().back().ref(),
|
||||
(NodeID)input_way.nodes()[input_way.nodes().size() - 2].ref()});
|
||||
}
|
||||
}
|
||||
|
@ -56,13 +56,13 @@ class ExtractorCallbacks
|
||||
std::unordered_map<std::string, NodeID> &string_map);
|
||||
|
||||
// warning: caller needs to take care of synchronization!
|
||||
void ProcessNode(const osmium::NodeRef ¤t_node, const ExtractionNode &result_node);
|
||||
void ProcessNode(const osmium::Node ¤t_node, const ExtractionNode &result_node);
|
||||
|
||||
// warning: caller needs to take care of synchronization!
|
||||
void ProcessRestriction(const mapbox::util::optional<InputRestrictionContainer> &restriction);
|
||||
|
||||
// warning: caller needs to take care of synchronization!
|
||||
void ProcessWay(const osmium::WayNodeList ¤t_way, const ExtractionWay &result_way);
|
||||
void ProcessWay(const osmium::Way ¤t_way, const ExtractionWay &result_way);
|
||||
};
|
||||
|
||||
#endif /* EXTRACTOR_CALLBACKS_H */
|
||||
|
@ -29,7 +29,7 @@ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#include "../Util/GitDescription.h"
|
||||
#include "../Util/IniFileUtil.h"
|
||||
#include "../Util/SimpleLogger.h"
|
||||
#include "../Util/simple_logger.hpp"
|
||||
|
||||
#include <boost/filesystem.hpp>
|
||||
#include <boost/program_options.hpp>
|
||||
|
@ -32,7 +32,7 @@ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
#include "../DataStructures/ExternalMemoryNode.h"
|
||||
#include "../Util/LuaUtil.h"
|
||||
#include "../Util/OSRMException.h"
|
||||
#include "../Util/SimpleLogger.h"
|
||||
#include "../Util/simple_logger.hpp"
|
||||
|
||||
#include <boost/algorithm/string.hpp>
|
||||
#include <boost/algorithm/string/regex.hpp>
|
||||
|
Loading…
Reference in New Issue
Block a user