Open-Harbor/osrm-backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 3 Wiki Activity

Small ProcessElements refactoring

Browse Source
This commit is contained in:
Michael Krasnyk 2017-09-14 13:22:31 +02:00
parent 94169a20de
commit 64ad308e9d
1 changed files with 41 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
src/extractor/scripting_environment_lua.cpp
View File

@ -288,14 +288,19 @@ void Sol2ScriptingEnvironment::InitContext(LuaScriptingContext &context)
"version",
&osmium::Way::version);
context.state.new_usertype<osmium::RelationMember>("RelationMember",
"role", &osmium::RelationMember::role,
"type", &osmium::RelationMember::type,
"id", [](const osmium::RelationMember &member) -> osmium::object_id_type {
context.state.new_usertype<osmium::RelationMember>(
"RelationMember",
"role",
&osmium::RelationMember::role,
"type",
&osmium::RelationMember::type,
"id",
[](const osmium::RelationMember &member) -> osmium::object_id_type {
return member.ref();
});
context.state.new_usertype<osmium::Relation>("Relation",
context.state.new_usertype<osmium::Relation>(
"Relation",
"get_value_by_key",
&get_value_by_key<osmium::Relation>,
"id",
@ -303,7 +308,7 @@ void Sol2ScriptingEnvironment::InitContext(LuaScriptingContext &context)
"version",
&osmium::Relation::version,
"members",
[](const osmium::Relation &rel) -> const osmium::RelationMemberList& {
[](const osmium::Relation &rel) -> const osmium::RelationMemberList & {
return rel.members();
});
@ -400,13 +405,11 @@ void Sol2ScriptingEnvironment::InitContext(LuaScriptingContext &context)
context.state.new_usertype<ExtractionRelation>(
"ExtractionRelation",
sol::meta_function::new_index,
[](ExtractionRelation &rel, const osmium::RelationMember &member) -> ExtractionRelation::AttributesMap& {
return rel.GetMember(member);
},
[](ExtractionRelation &rel, const osmium::RelationMember &member)
-> ExtractionRelation::AttributesMap & { return rel.GetMember(member); },
sol::meta_function::index,
[](ExtractionRelation &rel, const osmium::RelationMember &member) -> ExtractionRelation::AttributesMap& {
return rel.GetMember(member);
},
[](ExtractionRelation &rel, const osmium::RelationMember &member)
-> ExtractionRelation::AttributesMap & { return rel.GetMember(member); },
"restriction",
sol::property([](const ExtractionRelation &rel) { return rel.is_restriction; },
[](ExtractionRelation &rel, bool flag) { rel.is_restriction = flag; }));
@ -689,34 +692,34 @@ void Sol2ScriptingEnvironment::ProcessElements(
switch (entity->type())
{
case osmium::item_type::node:
{
const auto &node = static_cast<const osmium::Node &>(*entity);
result_node.clear();
if (local_context.has_node_function &&
(!static_cast<const osmium::Node &>(*entity).tags().empty() ||
local_context.properties.call_tagless_node_function))
(!node.tags().empty() || local_context.properties.call_tagless_node_function))
{
const osmium::Node &node = static_cast<const osmium::Node &>(*entity);
const auto &id = ExtractionRelation::OsmIDTyped(node.id(), osmium::item_type::node);
local_context.ProcessNode(node, result_node, relations.Get(id));
}
resulting_nodes.push_back(std::pair<const osmium::Node &, ExtractionNode>(
static_cast<const osmium::Node &>(*entity), std::move(result_node)));
resulting_nodes.push_back({node, std::move(result_node)});
}
break;
case osmium::item_type::way:
{
const osmium::Way &way = static_cast<const osmium::Way &>(*entity);
result_way.clear();
if (local_context.has_way_function)
{
const osmium::Way &way = static_cast<const osmium::Way &>(*entity);
const auto &id = ExtractionRelation::OsmIDTyped(way.id(), osmium::item_type::way);
local_context.ProcessWay(way, result_way, relations.Get(id));
}
resulting_ways.push_back(std::pair<const osmium::Way &, ExtractionWay>(
static_cast<const osmium::Way &>(*entity), std::move(result_way)));
resulting_ways.push_back({way, std::move(result_way)});
}
break;
case osmium::item_type::relation:
{
auto result_res =
restriction_parser.TryParse(static_cast<const osmium::Relation &>(*entity));
if (result_res)
const auto &relation = static_cast<const osmium::Relation &>(*entity);
if (auto result_res = restriction_parser.TryParse(relation))
{
resulting_restrictions.push_back(*result_res);
}
@ -726,13 +729,9 @@ void Sol2ScriptingEnvironment::ProcessElements(
result_relation.clear();
if (local_context.has_relation_function)
{
local_context.ProcessRelation(static_cast<const osmium::Relation &>(*entity),
result_relation);
local_context.ProcessRelation(relation, result_relation);
}
resulting_relations.push_back(
std::pair<const osmium::Relation &, ExtractionRelation>(
static_cast<const osmium::Relation &>(*entity),
std::move(result_relation)));
resulting_relations.push_back({relation, std::move(result_relation)});
}
}
break;