Moved CategoriesHolder from search into indexer library. Need to access it from Editor.

1 files changed, 106 insertions, 0 deletions

diff --git a/indexer/categories_holder.hpp b/indexer/categories_holder.hpp
new file mode 100644
index 0000000000..b6717020ad
--- /dev/null
+++ b/indexer/categories_holder.hpp
@@ -0,0 +1,106 @@
+#pragma once
+#include "base/string_utils.hpp"
+
+#include "std/vector.hpp"
+#include "std/map.hpp"
+#include "std/string.hpp"
+#include "std/iostream.hpp"
+#include "std/shared_ptr.hpp"
+
+
+class Reader;
+
+class CategoriesHolder
+{
+public:
+  struct Category
+  {
+    static const uint8_t EMPTY_PREFIX_LENGTH = 10;
+
+    struct Name
+    {
+      string m_name;
+      /// This language/locale code is completely different from our built-in langs in multilang_utf8_string.cpp
+      /// and is only used for mapping user's input language to our values in categories.txt file
+      int8_t m_locale;
+      uint8_t m_prefixLengthToSuggest;
+    };
+
+    vector<Name> m_synonyms;
+
+    inline void Swap(Category & r)
+    {
+      m_synonyms.swap(r.m_synonyms);
+    }
+  };
+
+private:
+  typedef strings::UniString StringT;
+  typedef multimap<uint32_t, shared_ptr<Category> > Type2CategoryContT;
+  typedef multimap<pair<int8_t, StringT>, uint32_t> Name2CatContT;
+  typedef Type2CategoryContT::const_iterator IteratorT;
+
+  Type2CategoryContT m_type2cat;
+  Name2CatContT m_name2type;
+
+public:
+  CategoriesHolder() {}
+  /// Takes ownership of reader.
+  explicit CategoriesHolder(Reader * reader);
+
+  void LoadFromStream(istream & s);
+
+  template <class ToDo>
+  void ForEachCategory(ToDo toDo) const
+  {
+    for (IteratorT i = m_type2cat.begin(); i != m_type2cat.end(); ++i)
+      toDo(*i->second);
+  }
+
+  template <class ToDo>
+  void ForEachName(ToDo toDo) const
+  {
+    for (IteratorT i = m_type2cat.begin(); i != m_type2cat.end(); ++i)
+      for (size_t j = 0; j < i->second->m_synonyms.size(); ++j)
+        toDo(i->second->m_synonyms[j]);
+  }
+
+  template <class ToDo>
+  void ForEachTypeByName(int8_t locale, StringT const & name, ToDo toDo) const
+  {
+    typedef typename Name2CatContT::const_iterator IterT;
+
+    pair<IterT, IterT> range = m_name2type.equal_range(make_pair(locale, name));
+    while (range.first != range.second)
+    {
+      toDo(range.first->second);
+      ++range.first;
+    }
+  }
+
+  /// Search name for type with preffered locale language.
+  /// If no name for this language, return first (en) name.
+  /// @return false if no categories for type.
+  bool GetNameByType(uint32_t type, int8_t locale, string & name) const;
+
+  bool IsTypeExist(uint32_t type) const;
+
+  inline void Swap(CategoriesHolder & r)
+  {
+    m_type2cat.swap(r.m_type2cat);
+    m_name2type.swap(r.m_name2type);
+  }
+
+  /// Converts any language locale from UI to internal integer code
+  static int8_t MapLocaleToInteger(string const & locale);
+  static int8_t const UNSUPPORTED_LOCALE_CODE = -1;
+
+private:
+  void AddCategory(Category & cat, vector<uint32_t> & types);
+  static bool ValidKeyToken(StringT const & s);
+};
+
+inline void swap(CategoriesHolder & a, CategoriesHolder & b)
+{
+  return a.Swap(b);
+}