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

Add last location memoization in Lua context

Browse Source
This commit is contained in:
Michael Krasnyk 2017-10-03 14:17:45 +02:00
parent 545097cf06
commit 11e7b6e911
5 changed files with 103 additions and 65 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
include/extractor/location_dependent_data.hpp
View File

@ -36,7 +36,9 @@ struct LocationDependentData
bool empty() const { return rtree.empty(); }
property_t operator()(const point_t &point, const char *key) const;
std::vector<std::size_t> GetPropertyIndexes(const point_t &point) const;
property_t FindByKey(const std::vector<std::size_t> &property_indexes, const char *key) const;
private:
void loadLocationDependentData(const boost::filesystem::path &file_path,

7
include/extractor/scripting_environment_lua.hpp
View File

@ -22,7 +22,8 @@ namespace extractor
struct LuaScriptingContext final
{
LuaScriptingContext(const LocationDependentData &location_dependent_data)
: location_dependent_data(location_dependent_data)
: location_dependent_data(location_dependent_data),
last_location_point(0., 180.) // assume (0,180) is invalid coordinate
{
}
@ -52,7 +53,11 @@ struct LuaScriptingContext final
int api_version;
sol::table profile_table;
// Reference to immutable location dependent data and locations memo
const LocationDependentData &location_dependent_data;
LocationDependentData::point_t last_location_point;
std::vector<std::size_t> last_location_indexes;
};
/**

31
src/extractor/location_dependent_data.cpp
View File

@ -191,25 +191,32 @@ void LocationDependentData::loadLocationDependentData(
}
}
LocationDependentData::property_t LocationDependentData::operator()(const point_t &point,
const char *key) const
LocationDependentData::property_t
LocationDependentData::FindByKey(const std::vector<std::size_t> &property_indexes,
const char *key) const
{
property_t result;
auto setter = [this, &result, &key](const rtree_t::value_type &rtree_entry) {
const auto properties_index = polygons[rtree_entry.second].second;
const auto &polygon_properties = properties[properties_index];
for (auto index : property_indexes)
{
const auto &polygon_properties = properties[index];
const auto it = polygon_properties.find(key);
if (it != polygon_properties.end())
{
result = it->second;
return it->second;
}
}
return property_t{};
}
std::vector<std::size_t> LocationDependentData::GetPropertyIndexes(const point_t &point) const
{
std::vector<std::size_t> result;
auto inserter = [this, &result](const rtree_t::value_type &rtree_entry) {
const auto properties_index = polygons[rtree_entry.second].second;
result.push_back(properties_index);
};
// Search the R-tree and collect a Lua table of tags that correspond to the location
rtree.query(boost::geometry::index::satisfies(
[&result](const rtree_t::value_type &) { return result.which() == 0; }) &&
boost::geometry::index::intersects(point) &&
rtree.query(boost::geometry::index::intersects(point) &&
boost::geometry::index::satisfies([this, &point](const rtree_t::value_type &v) {
// Simple point-in-polygon algorithm adapted from
@ -265,7 +272,7 @@ LocationDependentData::property_t LocationDependentData::operator()(const point_
return inside;
}),
boost::make_function_output_iterator(std::ref(setter)));
boost::make_function_output_iterator(std::ref(inserter)));
return result;
}

12
src/extractor/scripting_environment_lua.cpp
View File

@ -309,7 +309,17 @@ void Sol2ScriptingEnvironment::InitContext(LuaScriptingContext &context)
// at one or many locations must be provided
const auto &nodes = way.nodes();
const auto &location = nodes.back().location();
auto value = context.location_dependent_data({location.lon(), location.lat()}, key);
const LocationDependentData::point_t point{location.lon(), location.lat()};
if (!boost::geometry::equals(context.last_location_point, point))
{
context.last_location_point = point;
context.last_location_indexes =
context.location_dependent_data.GetPropertyIndexes(point);
}
auto value =
context.location_dependent_data.FindByKey(context.last_location_indexes, key);
return boost::apply_visitor(to_lua_object(context.state), value);
});

114
unit_tests/extractor/location_dependent_data_tests.cpp
View File

@ -1,5 +1,7 @@
#include "extractor/location_dependent_data.hpp"
#include "../common/range_tools.hpp"
#include <boost/filesystem.hpp>
#include <boost/test/test_case_template.hpp>
#include <boost/test/unit_test.hpp>
@ -51,15 +53,20 @@ BOOST_AUTO_TEST_CASE(polygon_tests)
LocationDependentData data({fixture.temporary_file});
BOOST_CHECK_EQUAL(data(point_t(0, 0), "answer").which(), 0);
BOOST_CHECK_EQUAL(boost::get<double>(data(point_t(1, 1), "answer")), 42);
BOOST_CHECK_EQUAL(boost::get<double>(data(point_t(0, 1), "answer")), 42);
BOOST_CHECK_EQUAL(boost::get<double>(data(point_t(0.5, -0.5), "answer")), 42);
BOOST_CHECK_EQUAL(boost::get<double>(data(point_t(0, -3), "answer")), 42);
BOOST_CHECK_EQUAL(boost::get<double>(data(point_t(-0.75, 0.75), "answer")), 42);
BOOST_CHECK_EQUAL(boost::get<double>(data(point_t(2, 0), "answer")), 42.);
BOOST_CHECK_EQUAL(boost::get<bool>(data(point_t(1, 7), "answer")), true);
BOOST_CHECK_EQUAL(boost::get<bool>(data(point_t(-2, 6), "answer")), true);
BOOST_CHECK(data.GetPropertyIndexes(point_t(0, 0)).empty());
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(1, 1)), 0);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(0, 1)), 0);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(0.5, -0.5)), 0);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(0, -3)), 0);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(-0.75, 0.75)), 0);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(2, 0)), 0);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(1, 7)), 1);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(-2, 6)), 1);
BOOST_CHECK_EQUAL(data.FindByKey({}, "answer").which(), 0);
BOOST_CHECK_EQUAL(data.FindByKey({0}, "foo").which(), 0);
BOOST_CHECK_EQUAL(boost::get<double>(data.FindByKey({0}, "answer")), 42);
BOOST_CHECK_EQUAL(boost::get<bool>(data.FindByKey({1}, "answer")), true);
}
BOOST_AUTO_TEST_CASE(multy_polygon_tests)
@ -82,11 +89,11 @@ BOOST_AUTO_TEST_CASE(multy_polygon_tests)
LocationDependentData data({fixture.temporary_file});
BOOST_CHECK_EQUAL(data(point_t(0, 2), "answer").which(), 0);
BOOST_CHECK_EQUAL(data(point_t(0, -3), "answer").which(), 0);
BOOST_CHECK_EQUAL(boost::get<double>(data(point_t(0, 0), "answer")), 42.);
BOOST_CHECK_EQUAL(boost::get<double>(data(point_t(5, 0), "answer")), 42.);
BOOST_CHECK_EQUAL(boost::get<double>(data(point_t(-5, 0), "answer")), 42.);
BOOST_CHECK(data.GetPropertyIndexes(point_t(0, 2)).empty());
BOOST_CHECK(data.GetPropertyIndexes(point_t(0, -3)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(0, 0)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(5, 0)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(-5, 0)).empty());
}
BOOST_AUTO_TEST_CASE(polygon_merging_tests)
@ -113,19 +120,26 @@ BOOST_AUTO_TEST_CASE(polygon_merging_tests)
LocationDependentData data({fixture.temporary_file});
BOOST_CHECK_EQUAL(boost::get<std::string>(data(point_t(-3, 3), "answer")), "a");
BOOST_CHECK_EQUAL(boost::get<std::string>(data(point_t(-3, 1), "answer")), "a");
BOOST_CHECK_EQUAL(boost::get<std::string>(data(point_t(-3, -3), "answer")), "a");
BOOST_CHECK_EQUAL(boost::get<std::string>(data(point_t(0, 3), "answer")), "a");
BOOST_CHECK_EQUAL(boost::get<std::string>(data(point_t(1, 0), "answer")), "a");
BOOST_CHECK_EQUAL(boost::get<std::string>(data(point_t(2, -3), "answer")), "a");
BOOST_CHECK_EQUAL(boost::get<std::string>(data(point_t(3, 0), "answer")), "a");
BOOST_CHECK_EQUAL(boost::get<std::string>(data(point_t(4, 3), "answer")), "b");
BOOST_CHECK_EQUAL(boost::get<std::string>(data(point_t(6, 1), "answer")), "b");
BOOST_CHECK_EQUAL(boost::get<std::string>(data(point_t(7, 0), "answer")), "b");
BOOST_CHECK_EQUAL(boost::get<std::string>(data(point_t(8, 3), "answer")), "c");
BOOST_CHECK_EQUAL(boost::get<std::string>(data(point_t(8, -1), "answer")), "c");
BOOST_CHECK_EQUAL(boost::get<std::string>(data(point_t(8, -3), "answer")), "c");
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(-3, 3)), 0);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(-3, 1)), 0);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(-3, -3)), 0);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(0, 3)), 0);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(1, 0)), 0, 1);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(2, -3)), 0, 1, 2);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(3, 0)), 0, 1, 2);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(4, 3)), 1, 2);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(6, 1)), 1, 2);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(7, 0)), 1, 2);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(8, 3)), 2);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(8, -1)), 2);
CHECK_EQUAL_RANGE(data.GetPropertyIndexes(point_t(8, -3)), 2);
BOOST_CHECK_EQUAL(boost::get<std::string>(data.FindByKey({0}, "answer")), "a");
BOOST_CHECK_EQUAL(boost::get<std::string>(data.FindByKey({1}, "answer")), "b");
BOOST_CHECK_EQUAL(boost::get<std::string>(data.FindByKey({2}, "answer")), "c");
BOOST_CHECK_EQUAL(boost::get<std::string>(data.FindByKey({0, 1, 2}, "answer")), "a");
BOOST_CHECK_EQUAL(boost::get<std::string>(data.FindByKey({1, 2}, "answer")), "b");
BOOST_CHECK_EQUAL(boost::get<std::string>(data.FindByKey({2, 1, 0}, "answer")), "c");
}
BOOST_AUTO_TEST_CASE(staircase_polygon)
@ -143,31 +157,31 @@ BOOST_AUTO_TEST_CASE(staircase_polygon)
LocationDependentData data({fixture.temporary_file});
// all corners
BOOST_CHECK_NE(data(point_t(0, 0), "answer").which(), 0);
BOOST_CHECK_NE(data(point_t(0, 1), "answer").which(), 0);
BOOST_CHECK_NE(data(point_t(1, 1), "answer").which(), 0);
BOOST_CHECK_NE(data(point_t(1, 2), "answer").which(), 0);
BOOST_CHECK_NE(data(point_t(2, 2), "answer").which(), 0);
BOOST_CHECK_NE(data(point_t(2, 3), "answer").which(), 0);
BOOST_CHECK_NE(data(point_t(3, 3), "answer").which(), 0);
BOOST_CHECK_NE(data(point_t(3, 0), "answer").which(), 0);
BOOST_CHECK(!data.GetPropertyIndexes(point_t(0, 0)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(0, 1)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(1, 1)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(1, 2)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(2, 2)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(2, 3)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(3, 3)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(3, 0)).empty());
// // at x = 1
BOOST_CHECK_EQUAL(data(point_t(1, -0.5), "answer").which(), 0);
BOOST_CHECK_NE(data(point_t(1, 0), "answer").which(), 0);
BOOST_CHECK_NE(data(point_t(1, 0.5), "answer").which(), 0);
BOOST_CHECK_NE(data(point_t(1, 1.5), "answer").which(), 0);
BOOST_CHECK_EQUAL(data(point_t(1, 2.5), "answer").which(), 0);
BOOST_CHECK_EQUAL(data(point_t(3.5, 2), "answer").which(), 0);
// at x = 1
BOOST_CHECK(data.GetPropertyIndexes(point_t(1, -0.5)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(1, 0)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(1, 0.5)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(1, 1.5)).empty());
BOOST_CHECK(data.GetPropertyIndexes(point_t(1, 2.5)).empty());
BOOST_CHECK(data.GetPropertyIndexes(point_t(3.5, 2)).empty());
// // at y = 2
BOOST_CHECK_EQUAL(data(point_t(0.5, 2), "answer").which(), 0);
BOOST_CHECK_NE(data(point_t(1, 2), "answer").which(), 0);
BOOST_CHECK_NE(data(point_t(1.5, 2), "answer").which(), 0);
BOOST_CHECK_NE(data(point_t(2, 2), "answer").which(), 0);
BOOST_CHECK_NE(data(point_t(2.5, 2), "answer").which(), 0);
BOOST_CHECK_NE(data(point_t(3, 2), "answer").which(), 0);
BOOST_CHECK_EQUAL(data(point_t(3.5, 2), "answer").which(), 0);
// at y = 2
BOOST_CHECK(data.GetPropertyIndexes(point_t(0.5, 2)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(1, 2)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(1.5, 2)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(2, 2)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(2.5, 2)).empty());
BOOST_CHECK(!data.GetPropertyIndexes(point_t(3, 2)).empty());
BOOST_CHECK(data.GetPropertyIndexes(point_t(3.5, 2)).empty());
}
BOOST_AUTO_TEST_SUITE_END()