69 lines
2 KiB
C++
69 lines
2 KiB
C++
|
#include "search/highlighting.hpp"
|
||
|
|
#include "std/target_os.hpp"
|
||
|
|
|
||
|
|
namespace search
|
||
|
|
{
|
||
|
|
namespace
|
||
|
|
{
|
||
|
|
// Makes continuous range for tokens and prefix.
|
||
|
|
template <class Iter, class Value>
|
||
|
|
class CombinedIterator
|
||
|
|
{
|
||
|
|
Iter m_cur;
|
||
|
|
Iter m_end;
|
||
|
|
Value const * m_val;
|
||
|
|
|
||
|
|
public:
|
||
|
|
CombinedIterator(Iter cur, Iter end, Value const * val) : m_cur(cur), m_end(end), m_val(val) {}
|
||
|
|
|
||
|
|
Value const & operator*() const
|
||
|
|
{
|
||
|
|
ASSERT(m_val != nullptr || m_cur != m_end, ("dereferencing of an empty iterator"));
|
||
|
|
if (m_cur != m_end)
|
||
|
|
return *m_cur;
|
||
|
|
|
||
|
|
return *m_val;
|
||
|
|
}
|
||
|
|
|
||
|
|
CombinedIterator & operator++()
|
||
|
|
{
|
||
|
|
if (m_cur != m_end)
|
||
|
|
++m_cur;
|
||
|
|
else
|
||
|
|
m_val = nullptr;
|
||
|
|
return *this;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool operator==(CombinedIterator const & other) const { return m_val == other.m_val && m_cur == other.m_cur; }
|
||
|
|
|
||
|
|
bool operator!=(CombinedIterator const & other) const { return !(*this == other); }
|
||
|
|
};
|
||
|
|
} // namespace
|
||
|
|
|
||
|
|
void HighlightResult(QueryTokens const & tokens, strings::UniString const & prefix, Result & res)
|
||
|
|
{
|
||
|
|
using Iter = QueryTokens::const_iterator;
|
||
|
|
using CombinedIter = CombinedIterator<Iter, strings::UniString>;
|
||
|
|
|
||
|
|
CombinedIter beg(tokens.begin(), tokens.end(), prefix.empty() ? nullptr : &prefix);
|
||
|
|
CombinedIter end(tokens.end() /* cur */, tokens.end() /* end */, nullptr);
|
||
|
|
|
||
|
|
// Highlight Title (potentially including branch)
|
||
|
|
std::string titleForHighlighting = res.GetString();
|
||
|
|
#if defined(OMIM_OS_IPHONE)
|
||
|
|
std::string const & branch = res.GetBranch();
|
||
|
|
|
||
|
|
// On iOS we append branch text to the title for highlighting if it's not already present.
|
||
|
|
if (!branch.empty() && titleForHighlighting.find(branch) == std::string::npos)
|
||
|
|
titleForHighlighting += " " + branch;
|
||
|
|
#endif
|
||
|
|
|
||
|
|
SearchStringTokensIntersectionRanges(
|
||
|
|
titleForHighlighting, beg, end, [&](std::pair<uint16_t, uint16_t> const & range) { res.AddHighlightRange(range); });
|
||
|
|
|
||
|
|
// Highlight description.
|
||
|
|
SearchStringTokensIntersectionRanges(res.GetAddress(), beg, end, [&](std::pair<uint16_t, uint16_t> const & range)
|
||
|
|
{ res.AddDescHighlightRange(range); });
|
||
|
|
}
|
||
|
|
} // namespace search
|