d6afe91d8f
* print tracebacks and line numbers for Lua runtime errors * revert format changes * update changelog with lua traceback, #6564 * revert using protected_function for old GetStringListFromFunction and source_function #6564 * add unit test for line numbers in tracebacks, #6564 * apply clang-format (#6564) * remove unused test helper function, #6564 * suppress leaksanitizer warnings in extract-tests, #6564 When the extractor encounters a lua runtime error, some osmium objects are not freed. In production this doesn't matter because these errors bring down OSRM. In the tests we catch them to ensure they occur, and the leaksanitizer flags them.
114 lines
4.2 KiB
C++
114 lines
4.2 KiB
C++
#ifndef SCRIPTING_ENVIRONMENT_LUA_HPP
|
|
#define SCRIPTING_ENVIRONMENT_LUA_HPP
|
|
|
|
#include "extractor/extraction_relation.hpp"
|
|
#include "extractor/location_dependent_data.hpp"
|
|
#include "extractor/raster_source.hpp"
|
|
#include "extractor/scripting_environment.hpp"
|
|
|
|
#include <tbb/enumerable_thread_specific.h>
|
|
|
|
#include <memory>
|
|
#include <mutex>
|
|
#include <string>
|
|
|
|
#include <sol/sol.hpp>
|
|
|
|
namespace osrm::extractor
|
|
{
|
|
|
|
struct LuaScriptingContext final
|
|
{
|
|
LuaScriptingContext(const LocationDependentData &location_dependent_data)
|
|
: location_dependent_data(location_dependent_data),
|
|
last_location_point(0., 180.) // assume (0,180) is invalid coordinate
|
|
{
|
|
}
|
|
|
|
void ProcessNode(const osmium::Node &,
|
|
ExtractionNode &result,
|
|
const ExtractionRelationContainer &relations);
|
|
void ProcessWay(const osmium::Way &,
|
|
ExtractionWay &result,
|
|
const ExtractionRelationContainer &relations);
|
|
|
|
ProfileProperties properties;
|
|
RasterContainer raster_sources;
|
|
sol::state state;
|
|
|
|
bool has_turn_penalty_function = false;
|
|
bool has_node_function = false;
|
|
bool has_way_function = false;
|
|
bool has_segment_function = false;
|
|
|
|
sol::protected_function turn_function;
|
|
sol::protected_function way_function;
|
|
sol::protected_function node_function;
|
|
sol::protected_function segment_function;
|
|
|
|
int api_version = 4;
|
|
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;
|
|
};
|
|
|
|
/**
|
|
* Creates a lua context and binds osmium way, node and relation objects and
|
|
* ExtractionWay and ExtractionNode to lua objects.
|
|
*
|
|
* Each thread has its own lua state which is implemented with thread specific
|
|
* storage from TBB.
|
|
*/
|
|
class Sol2ScriptingEnvironment final : public ScriptingEnvironment
|
|
{
|
|
public:
|
|
static const constexpr int SUPPORTED_MIN_API_VERSION = 0;
|
|
static const constexpr int SUPPORTED_MAX_API_VERSION = 4;
|
|
|
|
explicit Sol2ScriptingEnvironment(
|
|
const std::string &file_name,
|
|
const std::vector<boost::filesystem::path> &location_dependent_data_paths);
|
|
~Sol2ScriptingEnvironment() override = default;
|
|
|
|
const ProfileProperties &GetProfileProperties() override;
|
|
|
|
std::vector<std::vector<std::string>> GetExcludableClasses() override;
|
|
std::vector<std::string> GetNameSuffixList() override;
|
|
std::vector<std::string> GetClassNames() override;
|
|
std::vector<std::string> GetRestrictions() override;
|
|
std::vector<std::string> GetRelations() override;
|
|
void ProcessTurn(ExtractionTurn &turn) override;
|
|
void ProcessSegment(ExtractionSegment &segment) override;
|
|
|
|
void
|
|
ProcessElements(const osmium::memory::Buffer &buffer,
|
|
const RestrictionParser &restriction_parser,
|
|
const ManeuverOverrideRelationParser &maneuver_override_parser,
|
|
const ExtractionRelationContainer &relations,
|
|
std::vector<std::pair<const osmium::Node &, ExtractionNode>> &resulting_nodes,
|
|
std::vector<std::pair<const osmium::Way &, ExtractionWay>> &resulting_ways,
|
|
std::vector<InputTurnRestriction> &resulting_restrictions,
|
|
std::vector<InputManeuverOverride> &resulting_maneuver_overrides) override;
|
|
|
|
bool HasLocationDependentData() const override { return !location_dependent_data.empty(); }
|
|
|
|
private:
|
|
LuaScriptingContext &GetSol2Context();
|
|
|
|
std::vector<std::string> GetStringListFromTable(const std::string &table_name);
|
|
std::vector<std::vector<std::string>> GetStringListsFromTable(const std::string &table_name);
|
|
std::vector<std::string> GetStringListFromFunction(const std::string &function_name);
|
|
|
|
void InitContext(LuaScriptingContext &context);
|
|
std::mutex init_mutex;
|
|
std::string file_name;
|
|
tbb::enumerable_thread_specific<std::unique_ptr<LuaScriptingContext>> script_contexts;
|
|
const LocationDependentData location_dependent_data;
|
|
};
|
|
} // namespace osrm::extractor
|
|
|
|
#endif /* SCRIPTING_ENVIRONMENT_LUA_HPP */
|