kirylkaveryn/organicmaps
1
Fork 0
forked from organicmaps/organicmaps-tmp
Code Pull requests Activity

[search] Add helper for fast DiscoveryButton search

Browse source
This commit is contained in:
tatiana-yan 2018-07-16 21:52:36 +03:00 • committed by mpimenov
parent 6e824691f7
commit be9d13f61b
2 changed files with 48 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
search/utils.cpp
View file

@ -1,8 +1,14 @@
#include "search/utils.hpp"
#include "search/categories_cache.hpp"
#include "search/features_filter.hpp"
#include "search/geometry_cache.hpp"
#include "search/mwm_context.hpp"
#include "indexer/data_source.hpp"
#include <cctype>
#include <utility>
using namespace std;
@ -63,4 +69,37 @@ MwmSet::MwmHandle FindWorld(DataSource const & dataSource)
dataSource.GetMwmsInfo(infos);
return FindWorld(dataSource, infos);
}
void ForEachOfTypesInRect(DataSource const & dataSource, set<uint32_t> const & types,
m2::RectD const & pivot, FeatureIndexCallback const & fn)
{
vector<shared_ptr<MwmInfo>> infos;
dataSource.GetMwmsInfo(infos);
CategoriesCache cache(types, {} /* cancellable */);
auto pivotRectsCache = PivotRectsCache(1 /* maxNumEntries */, {} /* cancellable */,
max(pivot.SizeX(), pivot.SizeY()) /* maxRadiusMeters */);
for (auto const & info : infos)
{
if (!pivot.IsIntersect(info->m_bordersRect))
continue;
auto handle = dataSource.GetMwmHandleById(MwmSet::MwmId(info));
auto & value = *handle.GetValue<MwmValue>();
if (!value.HasSearchIndex())
continue;
MwmContext const mwmContext(move(handle));
auto features = cache.Get(mwmContext);
auto const pivotFeatures = pivotRectsCache.Get(mwmContext, pivot, scales::GetUpperScale());
ViewportFilter const filter(pivotFeatures, 0 /* threshold */);
features = filter.Filter(features);
MwmSet::MwmId mwmId(info);
features.ForEach([&fn, &mwmId](uint64_t bit) {
fn(FeatureID(mwmId, ::base::asserted_cast<uint32_t>(bit)));
});
}
}
} // namespace search

9
search/utils.hpp
View file

@ -5,11 +5,14 @@
#include "search/token_slice.hpp"
#include "indexer/categories_holder.hpp"
#include "indexer/feature_decl.hpp"
#include "indexer/mwm_set.hpp"
#include "indexer/search_delimiters.hpp"
#include "indexer/search_string_utils.hpp"
#include "indexer/trie.hpp"
#include "geometry/rect2d.hpp"
#include "base/levenshtein_dfa.hpp"
#include "base/stl_helpers.hpp"
#include "base/string_utils.hpp"
@ -18,6 +21,7 @@
#include <cstdint>
#include <functional>
#include <memory>
#include <set>
#include <vector>
class DataSource;
@ -117,4 +121,9 @@ bool FillCategories(QuerySliceOnRawStrings<T> const & slice, Locales const & loc
MwmSet::MwmHandle FindWorld(DataSource const & dataSource,
std::vector<std::shared_ptr<MwmInfo>> const & infos);
MwmSet::MwmHandle FindWorld(DataSource const & dataSource);
using FeatureIndexCallback = std::function<void(FeatureID const &)>;
// Applies |fn| to each feature index of type from |types| in |rect|.
void ForEachOfTypesInRect(DataSource const & dataSource, std::set<uint32_t> const & types,
m2::RectD const & rect, FeatureIndexCallback const & fn);
} // namespace search