diff options
Diffstat (limited to 'generator')
-rw-r--r-- | generator/generator_tests_support/test_feature.cpp | 8 | ||||
-rw-r--r-- | generator/generator_tests_support/test_feature.hpp | 2 |
2 files changed, 10 insertions, 0 deletions
diff --git a/generator/generator_tests_support/test_feature.cpp b/generator/generator_tests_support/test_feature.cpp index 5b6f4831a7..1ab2e297d8 100644 --- a/generator/generator_tests_support/test_feature.cpp +++ b/generator/generator_tests_support/test_feature.cpp @@ -313,6 +313,14 @@ string TestPOI::ToDebugString() const return os.str(); } +feature::TypesHolder TestPOI::GetTypes() const +{ + feature::TypesHolder types; + for (auto const path : m_types) + types.Add(classif().GetTypeByPath(path)); + return types; +} + // TestMultilingualPOI ----------------------------------------------------------------------------- TestMultilingualPOI::TestMultilingualPOI(m2::PointD const & center, string const & defaultName, map<string, string> const & multilingualNames) diff --git a/generator/generator_tests_support/test_feature.hpp b/generator/generator_tests_support/test_feature.hpp index 7591cc44a5..e9fbc7b01c 100644 --- a/generator/generator_tests_support/test_feature.hpp +++ b/generator/generator_tests_support/test_feature.hpp @@ -1,5 +1,6 @@ #pragma once +#include "indexer/feature_data.hpp" #include "indexer/feature_decl.hpp" #include "indexer/feature_meta.hpp" #include "indexer/mwm_set.hpp" @@ -156,6 +157,7 @@ public: void Serialize(FeatureBuilder1 & fb) const override; std::string ToDebugString() const override; + feature::TypesHolder GetTypes() const; void SetHouseNumber(std::string const & houseNumber) { m_houseNumber = houseNumber; } void SetStreetName(std::string const & name) { m_streetName = name; } void SetTypes(std::vector<std::vector<std::string>> const & types) { m_types = types; } |