aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/codecs/greedy_dict
diff options
context:
space:
mode:
authorDevtools Arcadia <arcadia-devtools@yandex-team.ru>2022-02-07 18:08:42 +0300
committerDevtools Arcadia <arcadia-devtools@mous.vla.yp-c.yandex.net>2022-02-07 18:08:42 +0300
commit1110808a9d39d4b808aef724c861a2e1a38d2a69 (patch)
treee26c9fed0de5d9873cce7e00bc214573dc2195b7 /library/cpp/codecs/greedy_dict
downloadydb-1110808a9d39d4b808aef724c861a2e1a38d2a69.tar.gz
intermediate changes
ref:cde9a383711a11544ce7e107a78147fb96cc4029
Diffstat (limited to 'library/cpp/codecs/greedy_dict')
-rw-r--r--library/cpp/codecs/greedy_dict/gd_builder.cpp142
-rw-r--r--library/cpp/codecs/greedy_dict/gd_builder.h94
-rw-r--r--library/cpp/codecs/greedy_dict/gd_entry.cpp98
-rw-r--r--library/cpp/codecs/greedy_dict/gd_entry.h103
-rw-r--r--library/cpp/codecs/greedy_dict/gd_stats.h79
-rw-r--r--library/cpp/codecs/greedy_dict/ut/greedy_dict_ut.cpp282
-rw-r--r--library/cpp/codecs/greedy_dict/ut/ya.make9
-rw-r--r--library/cpp/codecs/greedy_dict/ya.make15
8 files changed, 822 insertions, 0 deletions
diff --git a/library/cpp/codecs/greedy_dict/gd_builder.cpp b/library/cpp/codecs/greedy_dict/gd_builder.cpp
new file mode 100644
index 00000000000..561bfbca015
--- /dev/null
+++ b/library/cpp/codecs/greedy_dict/gd_builder.cpp
@@ -0,0 +1,142 @@
+#include "gd_builder.h"
+
+#include <library/cpp/string_utils/relaxed_escaper/relaxed_escaper.h>
+#include <util/generic/algorithm.h>
+
+#include <util/random/shuffle.h>
+#include <util/stream/output.h>
+#include <util/string/printf.h>
+#include <util/system/rusage.h>
+
+namespace NGreedyDict {
+ void TDictBuilder::RebuildCounts(ui32 maxcand, bool final) {
+ if (!Current) {
+ Current = MakeHolder<TEntrySet>();
+ Current->InitWithAlpha();
+ }
+
+ TEntrySet& set = *Current;
+
+ for (auto& it : set)
+ it.Count = 0;
+
+ CompoundCounts = nullptr;
+ CompoundCountsPool.Clear();
+
+ if (!final) {
+ CompoundCounts = MakeHolder<TCompoundCounts>(&CompoundCountsPool);
+ CompoundCounts->reserve(maxcand);
+ }
+
+ Shuffle(Input.begin(), Input.end(), Rng);
+
+ for (auto str : Input) {
+ if (!final && CompoundCounts->size() > maxcand)
+ break;
+
+ i32 prev = -1;
+
+ while (!!str) {
+ TEntry* e = set.FindPrefix(str);
+ ui32 num = e->Number;
+
+ e->Count += 1;
+ if (!final && prev >= 0) {
+ (*CompoundCounts)[Compose(prev, num)] += 1;
+ }
+
+ prev = num;
+ ++set.TotalCount;
+ }
+ }
+
+ Current->SetModelP();
+ }
+
+ ui32 TDictBuilder::BuildNextGeneration(ui32 maxent) {
+ TAutoPtr<TEntrySet> newset = new TEntrySet;
+ newset->InitWithAlpha();
+ maxent -= newset->size();
+
+ ui32 additions = 0;
+ ui32 deletions = 0;
+
+ {
+ const TEntrySet& set = *Current;
+
+ Candidates.clear();
+ const ui32 total = set.TotalCount;
+ const float minpval = Settings.MinPValue;
+ const EEntryStatTest test = Settings.StatTest;
+ const EEntryScore score = Settings.Score;
+ const ui32 mincnt = Settings.MinAbsCount;
+
+ for (const auto& it : set) {
+ const TEntry& e = it;
+ float modelp = e.ModelP;
+ ui32 cnt = e.Count;
+
+ if (e.HasPrefix() && e.Count > mincnt && StatTest(test, modelp, cnt, total) > minpval)
+ Candidates.push_back(TCandidate(-Score(score, e.Len(), modelp, cnt, total), it.Number));
+ }
+
+ if (!!CompoundCounts) {
+ for (TCompoundCounts::const_iterator it = CompoundCounts->begin(); it != CompoundCounts->end(); ++it) {
+ const TEntry& prev = set.Get(Prev(it->first));
+ const TEntry& next = set.Get(Next(it->first));
+ float modelp = ModelP(prev.Count, next.Count, total);
+ ui32 cnt = it->second;
+ if (cnt > mincnt && StatTest(test, modelp, cnt, total) > minpval)
+ Candidates.push_back(TCandidate(-Score(score, prev.Len() + next.Len(), modelp, cnt, total), it->first));
+ }
+ }
+
+ Sort(Candidates.begin(), Candidates.end());
+
+ if (Candidates.size() > maxent)
+ Candidates.resize(maxent);
+
+ for (const auto& candidate : Candidates) {
+ if (IsCompound(candidate.second)) {
+ additions++;
+ newset->Add(set.Get(Prev(candidate.second)).Str, set.Get(Next(candidate.second)).Str);
+ } else {
+ newset->Add(set.Get(candidate.second).Str);
+ }
+ }
+
+ deletions = set.size() - (newset->size() - additions);
+ }
+
+ Current = newset;
+ Current->BuildHierarchy();
+ return deletions + additions;
+ }
+
+ ui32 TDictBuilder::Build(ui32 maxentries, ui32 maxiters, ui32 mindiff) {
+ size_t totalsz = 0;
+ for (auto it : Input)
+ totalsz += it.size();
+
+ while (maxiters) {
+ maxiters--;
+
+ RebuildCounts(maxentries * Settings.GrowLimit, false);
+
+ if (Settings.Verbose) {
+ TString mess = Sprintf("iter:%" PRIu32 " sz:%" PRIu32 " pend:%" PRIu32, maxiters, (ui32)Current->size(), (ui32)CompoundCounts->size());
+ Clog << Sprintf("%-110s RSS=%" PRIu32 "M", mess.data(), (ui32)(TRusage::Get().MaxRss >> 20)) << Endl;
+ }
+
+ ui32 diff = BuildNextGeneration(maxentries);
+
+ if (Current->size() == maxentries && diff < mindiff)
+ break;
+ }
+
+ RebuildCounts(0, true);
+ Current->SetScores(Settings.Score);
+ return maxiters;
+ }
+
+}
diff --git a/library/cpp/codecs/greedy_dict/gd_builder.h b/library/cpp/codecs/greedy_dict/gd_builder.h
new file mode 100644
index 00000000000..b8e9a5e37be
--- /dev/null
+++ b/library/cpp/codecs/greedy_dict/gd_builder.h
@@ -0,0 +1,94 @@
+#pragma once
+
+#include "gd_entry.h"
+
+#include <util/generic/hash.h>
+#include <util/random/fast.h>
+
+namespace NGreedyDict {
+ struct TBuildSettings {
+ EEntryStatTest StatTest = EST_SIMPLE_NORM;
+ EEntryScore Score = ES_LEN_SIMPLE;
+
+ float MinPValue = 0.75;
+ ui32 MinAbsCount = 10;
+ ui32 GrowLimit = 10; // times of maxentries
+ bool Verbose = false;
+ };
+
+ class TDictBuilder {
+ using TCompoundCounts = THashMap<ui64, ui32, THash<ui64>, TEqualTo<ui64>, TPoolAllocator>;
+ using TCandidate = std::pair<float, ui64>;
+ using TCandidates = TVector<TCandidate>;
+
+ private:
+ TFastRng64 Rng{0x1a5d0ac170565c1c, 0x0be7bc27, 0x6235f6f57820aa0d, 0xafdc7fb};
+ TStringBufs Input;
+
+ THolder<TEntrySet> Current;
+
+ TMemoryPool CompoundCountsPool;
+ THolder<TCompoundCounts> CompoundCounts;
+
+ TCandidates Candidates;
+
+ TBuildSettings Settings;
+
+ public:
+ TDictBuilder(const TBuildSettings& s = TBuildSettings())
+ : CompoundCountsPool(8112, TMemoryPool::TLinearGrow::Instance())
+ , Settings(s)
+ {
+ }
+
+ void SetInput(const TStringBufs& in) {
+ Input = in;
+ }
+
+ const TBuildSettings& GetSettings() const {
+ return Settings;
+ }
+
+ TBuildSettings& GetSettings() {
+ return Settings;
+ }
+
+ void SetSettings(const TBuildSettings& s) {
+ Settings = s;
+ }
+
+ TEntrySet& EntrySet() {
+ return *Current;
+ }
+
+ const TEntrySet& EntrySet() const {
+ return *Current;
+ }
+
+ THolder<TEntrySet> ReleaseEntrySet() {
+ return std::move(Current);
+ }
+
+ ui32 /*iters*/ Build(ui32 maxentries, ui32 maxiters = 16, ui32 mindiff = 10);
+
+ public:
+ void RebuildCounts(ui32 maxcand, bool final);
+ ui32 /*diff size*/ BuildNextGeneration(ui32 maxent);
+
+ static bool IsCompound(ui64 ent) {
+ return ent & 0xFFFFFFFF00000000ULL;
+ }
+
+ static ui32 Next(ui64 ent) {
+ return ent;
+ }
+ static ui32 Prev(ui64 ent) {
+ return (ent >> 32) - 1;
+ }
+
+ static ui64 Compose(ui32 prev, ui32 next) {
+ return ((prev + 1ULL) << 32) | next;
+ }
+ };
+
+}
diff --git a/library/cpp/codecs/greedy_dict/gd_entry.cpp b/library/cpp/codecs/greedy_dict/gd_entry.cpp
new file mode 100644
index 00000000000..2c315c7f7cf
--- /dev/null
+++ b/library/cpp/codecs/greedy_dict/gd_entry.cpp
@@ -0,0 +1,98 @@
+#include "gd_entry.h"
+#include "gd_stats.h"
+
+#include <util/generic/algorithm.h>
+#include <util/generic/singleton.h>
+
+namespace NGreedyDict {
+ class TAlphas {
+ char Memory[512];
+
+ public:
+ TStringBufs Alphas;
+
+ TAlphas() {
+ for (ui32 i = 0; i < 256; ++i) {
+ Memory[2 * i] = (char)i;
+ Memory[2 * i + 1] = 0;
+
+ Alphas.push_back(TStringBuf(&Memory[2 * i], 1));
+ }
+ }
+ };
+
+ void TEntrySet::InitWithAlpha() {
+ Pool.ClearKeepFirstChunk();
+ const TStringBufs& a = Singleton<TAlphas>()->Alphas;
+ for (auto it : a) {
+ Add(it);
+ }
+ BuildHierarchy();
+ }
+
+ void TEntrySet::BuildHierarchy() {
+ Sort(begin(), end(), TEntry::StrLess);
+
+ TCompactTrieBuilder<char, ui32, TAsIsPacker<ui32>> builder(CTBF_PREFIX_GROUPED);
+
+ for (iterator it = begin(); it != end(); ++it) {
+ it->Number = (it - begin());
+ TStringBuf suff = it->Str;
+ size_t len = 0;
+ ui32 val = 0;
+
+ if (builder.FindLongestPrefix(suff.data(), suff.size(), &len, &val) && len) {
+ it->NearestPrefix = val;
+ }
+
+ builder.Add(suff.data(), suff.size(), it->Number);
+ }
+
+ TBufferOutput bout;
+ builder.Save(bout);
+ Trie.Init(TBlob::FromBuffer(bout.Buffer()));
+ }
+
+ TEntry* TEntrySet::FindPrefix(TStringBuf& str) {
+ size_t len = 0;
+ ui32 off = 0;
+
+ if (!Trie.FindLongestPrefix(str, &len, &off)) {
+ return nullptr;
+ }
+
+ str.Skip(len);
+ return &Get(off);
+ }
+
+ void TEntrySet::SetModelP() {
+ for (iterator it = begin(); it != end(); ++it) {
+ TEntry& e = *it;
+
+ if (!e.HasPrefix()) {
+ e.ModelP = 0;
+ continue;
+ }
+
+ TStringBuf suff = e.Str;
+ const TEntry& p = Get(e.NearestPrefix);
+ suff.Skip(p.Len());
+
+ float modelp = float(p.Count + e.Count) / TotalCount;
+
+ while (!!suff) {
+ TEntry* pp = FindPrefix(suff);
+ modelp *= float(pp->Count + e.Count) / TotalCount;
+ }
+
+ e.ModelP = modelp;
+ }
+ }
+
+ void TEntrySet::SetScores(EEntryScore s) {
+ for (auto& it : *this) {
+ it.Score = Score(s, it.Len(), it.ModelP, it.Count, TotalCount);
+ }
+ }
+
+}
diff --git a/library/cpp/codecs/greedy_dict/gd_entry.h b/library/cpp/codecs/greedy_dict/gd_entry.h
new file mode 100644
index 00000000000..18b5be0e156
--- /dev/null
+++ b/library/cpp/codecs/greedy_dict/gd_entry.h
@@ -0,0 +1,103 @@
+#pragma once
+
+#include "gd_stats.h"
+
+#include <library/cpp/containers/comptrie/comptrie.h>
+
+#include <util/generic/ptr.h>
+#include <util/generic/strbuf.h>
+#include <util/generic/vector.h>
+
+#include <util/memory/pool.h>
+
+namespace NGreedyDict {
+ using TStringBufs = TVector<TStringBuf>;
+
+ struct TEntry {
+ static const i32 NoPrefix = -1;
+
+ TStringBuf Str;
+
+ i32 NearestPrefix = NoPrefix;
+ ui32 Count = 0;
+ ui32 Number = 0;
+ float ModelP = 0;
+ float Score = 0;
+
+ TEntry(TStringBuf b = TStringBuf(), ui32 cnt = 0)
+ : Str(b)
+ , Count(cnt)
+ {
+ }
+
+ bool HasPrefix() const {
+ return NearestPrefix != NoPrefix;
+ }
+ ui32 Len() const {
+ return Str.size();
+ }
+
+ static bool StrLess(const TEntry& a, const TEntry& b) {
+ return a.Str < b.Str;
+ }
+ static bool NumberLess(const TEntry& a, const TEntry& b) {
+ return a.Number < b.Number;
+ }
+ static bool ScoreMore(const TEntry& a, const TEntry& b) {
+ return a.Score > b.Score;
+ }
+ };
+
+ class TEntrySet: public TVector<TEntry>, TNonCopyable {
+ TMemoryPool Pool{8112};
+ TCompactTrie<char, ui32, TAsIsPacker<ui32>> Trie;
+
+ public:
+ ui32 TotalCount = 0;
+
+ void InitWithAlpha();
+
+ void Add(TStringBuf a) {
+ push_back(TStringBuf(Pool.Append(a.data(), a.size()), a.size()));
+ }
+
+ void Add(TStringBuf a, TStringBuf b) {
+ size_t sz = a.size() + b.size();
+ char* p = (char*)Pool.Allocate(sz);
+ memcpy(p, a.data(), a.size());
+ memcpy(p + a.size(), b.data(), b.size());
+ push_back(TStringBuf(p, sz));
+ }
+
+ TEntry& Get(ui32 idx) {
+ return (*this)[idx];
+ }
+
+ const TEntry& Get(ui32 idx) const {
+ return (*this)[idx];
+ }
+
+ void BuildHierarchy();
+
+ // longest prefix
+ TEntry* FindPrefix(TStringBuf& str);
+
+ const TEntry* FindPrefix(TStringBuf& str) const {
+ return ((TEntrySet*)this)->FindPrefix(str);
+ }
+
+ const TEntry* FirstPrefix(const TEntry& e, TStringBuf& suff) {
+ if (!e.HasPrefix())
+ return nullptr;
+
+ const TEntry& p = Get(e.NearestPrefix);
+ suff = e.Str;
+ suff.Skip(p.Str.size());
+ return &p;
+ }
+
+ void SetModelP();
+ void SetScores(EEntryScore);
+ };
+
+}
diff --git a/library/cpp/codecs/greedy_dict/gd_stats.h b/library/cpp/codecs/greedy_dict/gd_stats.h
new file mode 100644
index 00000000000..b63c4c38d23
--- /dev/null
+++ b/library/cpp/codecs/greedy_dict/gd_stats.h
@@ -0,0 +1,79 @@
+#pragma once
+
+#include <util/generic/ymath.h>
+#include <util/generic/algorithm.h>
+#include <util/generic/yexception.h>
+
+namespace NGreedyDict {
+ enum EEntryScore {
+ ES_COUNT,
+ ES_LEN_COUNT,
+ ES_SIMPLE,
+ ES_LEN_SIMPLE,
+ ES_SOLAR
+ };
+
+ enum EEntryStatTest {
+ EST_NONE = 0,
+ EST_SIMPLE_NORM = 2
+ };
+
+ inline float ModelP(ui32 countA, ui32 countB, ui32 total) {
+ return float(countA) * countB / total / total;
+ }
+
+ // P (ab | dependent)
+ inline float SimpleTest(float modelp, ui32 countAB, ui32 total) {
+ float realp = float(countAB) / total;
+ return modelp >= realp ? 0 : (realp - modelp);
+ }
+
+ inline float SolarTest(float modelp, ui32 countAB, ui32 total) {
+ float realp = float(countAB) / total;
+ return modelp >= realp ? 0 : (modelp + realp * (log(realp / modelp) - 1));
+ }
+
+ // P (ab | dependent) / P (ab)
+ inline float SimpleTestNorm(float modelp, ui32 countAB, ui32 total) {
+ float realp = float(countAB) / total;
+ return modelp >= realp ? 0 : (realp - modelp) / realp;
+ }
+
+ inline float StatTest(EEntryStatTest test, float modelp, ui32 countAB, ui32 total) {
+ if (!total) {
+ return 0;
+ }
+ switch (test) {
+ case EST_NONE:
+ return 1;
+ case EST_SIMPLE_NORM:
+ return SimpleTestNorm(modelp, countAB, total);
+ }
+ Y_FAIL("no way!");
+ return 0;
+ }
+
+ inline float Score(EEntryScore score, ui32 len, float modelp, ui32 count, ui32 total) {
+ if (!total) {
+ return 0;
+ }
+ ui32 m = 1;
+ switch (score) {
+ case ES_LEN_COUNT:
+ m = len;
+ [[fallthrough]];
+ case ES_COUNT:
+ return m * count;
+ case ES_LEN_SIMPLE:
+ m = len;
+ [[fallthrough]];
+ case ES_SIMPLE:
+ return m * SimpleTest(modelp, count, total);
+ case ES_SOLAR:
+ return SolarTest(modelp, count, total);
+ }
+ Y_FAIL("no way!");
+ return 0;
+ }
+
+}
diff --git a/library/cpp/codecs/greedy_dict/ut/greedy_dict_ut.cpp b/library/cpp/codecs/greedy_dict/ut/greedy_dict_ut.cpp
new file mode 100644
index 00000000000..679089a11be
--- /dev/null
+++ b/library/cpp/codecs/greedy_dict/ut/greedy_dict_ut.cpp
@@ -0,0 +1,282 @@
+#include "gd_builder.h"
+
+#include <library/cpp/testing/unittest/registar.h>
+#include <library/cpp/string_utils/relaxed_escaper/relaxed_escaper.h>
+#include <util/string/printf.h>
+#include <util/generic/ymath.h>
+
+class TGreedyDictTest: public TTestBase {
+ UNIT_TEST_SUITE(TGreedyDictTest);
+ UNIT_TEST(TestEntrySet)
+ UNIT_TEST(TestBuilder0)
+ UNIT_TEST(TestBuilder)
+ UNIT_TEST_SUITE_END();
+
+ void TestEntrySet() {
+ using namespace NGreedyDict;
+
+ {
+ TEntrySet d;
+
+ d.InitWithAlpha();
+
+ for (TEntrySet::const_iterator it = d.begin(); it != d.end(); ++it) {
+ UNIT_ASSERT_C(!it->HasPrefix(), Sprintf("%u -> %u", it->Number, it->NearestPrefix));
+ UNIT_ASSERT_VALUES_EQUAL(it->Number, (ui32)(it - d.begin()));
+ }
+
+ UNIT_ASSERT_VALUES_EQUAL(d.size(), 256u);
+ TStringBuf s = "aaabbb";
+ UNIT_ASSERT_VALUES_EQUAL(d.FindPrefix(s)->Str, "a");
+ UNIT_ASSERT_VALUES_EQUAL(s, "aabbb");
+ UNIT_ASSERT_VALUES_EQUAL(d.FindPrefix(s)->Str, "a");
+ UNIT_ASSERT_VALUES_EQUAL(s, "abbb");
+ UNIT_ASSERT_VALUES_EQUAL(d.FindPrefix(s)->Str, "a");
+ UNIT_ASSERT_VALUES_EQUAL(s, "bbb");
+ UNIT_ASSERT_VALUES_EQUAL(d.FindPrefix(s)->Str, "b");
+ UNIT_ASSERT_VALUES_EQUAL(s, "bb");
+ UNIT_ASSERT_VALUES_EQUAL(d.FindPrefix(s)->Str, "b");
+ UNIT_ASSERT_VALUES_EQUAL(s, "b");
+ UNIT_ASSERT_VALUES_EQUAL(d.FindPrefix(s)->Str, "b");
+ UNIT_ASSERT_VALUES_EQUAL(s, "");
+ s = TStringBuf("", 1);
+ UNIT_ASSERT_VALUES_EQUAL(d.FindPrefix(s)->Str, TStringBuf("", 1));
+ UNIT_ASSERT_VALUES_EQUAL(s, "");
+ s = "\xFF";
+ UNIT_ASSERT_VALUES_EQUAL(d.FindPrefix(s)->Str, "\xFF");
+ UNIT_ASSERT_VALUES_EQUAL(s, "");
+ }
+ {
+ TEntrySet d;
+ d.Add("a");
+ d.Add("b");
+ d.Add("b", "a");
+ d.BuildHierarchy();
+
+ UNIT_ASSERT_VALUES_EQUAL(d.size(), 3u);
+
+ TStringBuf s = "bab";
+ UNIT_ASSERT_VALUES_EQUAL(d.FindPrefix(s)->Str, "ba");
+ UNIT_ASSERT_VALUES_EQUAL(s, "b");
+ }
+ {
+ TEntrySet d;
+
+ d.Add("a");
+ d.Add("aa");
+ d.Add("aaa");
+ d.Add("aab");
+ d.Add("b");
+ d.Add("ba");
+
+ d.BuildHierarchy();
+
+ UNIT_ASSERT_VALUES_EQUAL(d.size(), 6u);
+ {
+ TStringBuf s = "aaaaa";
+ const TEntry* e = d.FindPrefix(s);
+ UNIT_ASSERT_VALUES_EQUAL(e->Str, "aaa");
+ UNIT_ASSERT_VALUES_EQUAL(e->Number, 2u);
+ UNIT_ASSERT_VALUES_EQUAL(e->NearestPrefix, 1);
+ UNIT_ASSERT_VALUES_EQUAL(s, "aa");
+ }
+
+ {
+ TStringBuf s = "a";
+ const TEntry* e = d.FindPrefix(s);
+ UNIT_ASSERT_VALUES_EQUAL(e->Str, "a");
+ UNIT_ASSERT_VALUES_EQUAL(e->Number, 0u);
+ UNIT_ASSERT_VALUES_EQUAL(e->NearestPrefix, -1);
+ UNIT_ASSERT_VALUES_EQUAL(s, "");
+ }
+
+ {
+ TStringBuf s = "bab";
+ const TEntry* e = d.FindPrefix(s);
+ UNIT_ASSERT_VALUES_EQUAL(e->Str, "ba");
+ UNIT_ASSERT_VALUES_EQUAL(e->Number, 5u);
+ UNIT_ASSERT_VALUES_EQUAL(e->NearestPrefix, 4);
+ UNIT_ASSERT_VALUES_EQUAL(s, "b");
+ }
+
+ {
+ TStringBuf s = "bba";
+ const TEntry* e = d.FindPrefix(s);
+ UNIT_ASSERT_VALUES_EQUAL(e->Str, "b");
+ UNIT_ASSERT_VALUES_EQUAL(e->Number, 4u);
+ UNIT_ASSERT_VALUES_EQUAL(e->NearestPrefix, -1);
+ UNIT_ASSERT_VALUES_EQUAL(s, "ba");
+ }
+ }
+ }
+
+ void TestBuilder0() {
+ using namespace NGreedyDict;
+ ui32 a = 1, b = 11;
+ ui64 ab = TDictBuilder::Compose(a, b);
+ UNIT_ASSERT(TDictBuilder::IsCompound(ab));
+ UNIT_ASSERT_VALUES_EQUAL(TDictBuilder::Prev(ab), a);
+ UNIT_ASSERT_VALUES_EQUAL(TDictBuilder::Next(ab), b);
+ }
+
+ void FillData(NGreedyDict::TStringBufs& data) {
+ static const char* urls[] = {"http://53.ru/car/motors/foreign/opel/tigra/", "http://abakan.24au.ru/tender/85904/", "http://anm15.gulaig.com/", "http://avto-parts.com/mercedes-benz/mercedes-benz-w220-1998-2005/category-442/category-443/", "http://ballooncousin.co.uk/", "http://benzol.ru/equipment/?id=1211&parent=514", "http://blazingseorank.com/blazing-seo-rank-free-website-analysis-to-increase-rank-and-traffic-450.html", "http://blogblaugrana.contadorwebmasters.com/", "http://bristolhash.org.uk/bh3cntct.php", "http://broker.borovichi.ru/category/item/3/1/0/8/28/257", "http://canoncompactcamerax.blogspot.com/", "http://classifieds.smashits.com/p,107881,email-to-friend.htm", "http://conferences.ksde.org/Portals/132/FallAssessment/SAVETHEDAY-FA09.pdf", "http://eway.vn/raovat/325-dien-tu-gia-dung/337-dieu-hoa/98041-b1-sua-may-lanh-quan-binh-tan-sua-may-lanh-quan-binh-chanh-hh-979676119-toan-quoc.html", "http://gallery.e2bn.org/asset73204_8-.html", "http://goplay.nsw.gov.au/activities-for-kids/by/historic-houses-trust/?startdate=2012-07-10", "http://grichards19067.multiply.com/", "http://hotkovo.egent.ru/user/89262269084/", "http://howimetyourself.com/?redirect_to=http://gomiso.com/m/suits/seasons/2/episodes/2", "http://islamqa.com/hi/ref/9014/DEAD%20PEOPLE%20GOOD%20DEEDS", "http://lapras.rutube.ru/", "http://nceluiko.ya.ru/", "http://nyanyanyanyaa.beon.ru/", "http://ozbo.com/Leaf-River-DV-7SS-7-0-MP-Game-Camera-K1-32541.html", "http://sbantom.ru/catalog/chasy/632753.html", "http://shopingoff.com/index.php?option=com_virtuemart&Itemid=65&category_id=&page=shop.browse&manufacturer_id=122&limit=32&limitstart=96", "http://shopingoff.com/katalog-odezhdy/manufacturer/62-christian-audigier.html?limit=32&start=448", "https://webwinkel.ah.nl/process?fh_location=//ecommerce/nl_NL/categories%3C%7Becommerce_shoc1%7D/it_show_product_code_1384%3E%7B10%3B20%7D/pr_startdate%3C20120519/pr_enddate%3E20120519/pr_ltc_allowed%3E%7Bbowi%7D/categories%3C%7Becommerce_shoc1_1al%7D/categories%3C%7Becommerce_shoc1_1al_1ahal%7D&&action=albert_noscript.modules.build", "http://top100.rambler.ru/navi/?theme=208/210/371&rgn=17", "http://volgogradskaya-oblast.extra-m.ru/classifieds/rabota/vakansii/banki-investicii/901467/", "http://wikien4.appspot.com/wiki/Warburg_hypothesis", "http://wola_baranowska.kamerzysta24.com.pl/", "http://www.10dot0dot0dot1.com/", "http://www.anima-redux.ru/index.php?key=gifts+teenage+girls", "http://www.aquaticabyseaworld.com/Calendar.aspx/CP/CP/CP/sp-us/CP/CP/ParkMap/Tickets/Weather.aspx", "http://www.autousa.com/360spin/2012_cadillac_ctssportwagon_3.6awdpremiumcollection.htm", "http://www.booking.com/city/gb/paignton-aireborough.html?inac=0&lang=pl", "http://www.booking.com/city/it/vodo-cadore.en.html", "http://www.booking.com/district/us/new-york/rockefeller-center.html&lang=no", "http://www.booking.com/hotel/bg/crown-fort-club.lv.html", "http://www.booking.com/hotel/ca/gouverneur-rimouski.ar.html", "http://www.booking.com/hotel/ch/l-auberge-du-chalet-a-gobet.fi.html", "http://www.booking.com/hotel/de/mark-garni.ru.html?aid=337384;label=yandex-hotel-mark-garni-68157-%7Bparam1%7D", "http://www.booking.com/hotel/de/mercure-goldschmieding-castrop-rauxel.ro.html", "http://www.booking.com/hotel/de/zollenspieker-fahrhaus.fr.html", "http://www.booking.com/hotel/es/jardin-metropolitano.ca.html", "http://www.booking.com/hotel/fr/clim.fr.html", "http://www.booking.com/hotel/fr/radisson-sas-toulouse-airport.et.html", "http://www.booking.com/hotel/gb/stgileshotel.ro.html?srfid=68c7fe42a03653a8796c84435c5299e4X16?tab=4", "http://www.booking.com/hotel/gr/rodos-park-suites.ru.html", "http://www.booking.com/hotel/id/le-grande-suites-bali.ru.html", "http://www.booking.com/hotel/it/mozart.it.html?aid=321655", "http://www.booking.com/hotel/ni/bahia-del-sol-villas.ru.html?dcid=1;dva=0", "http://www.booking.com/hotel/nl/cpschiphol.ro.html.ro.html?tab=4", "http://www.booking.com/hotel/th/laem-din.en-gb.html", "http://www.booking.com/hotel/th/tinidee-ranong.en.html", "http://www.booking.com/hotel/us/best-western-plus-merrimack-valley.hu.html", "http://www.booking.com/hotel/vn/tan-hai-long.km.html", "http://www.booking.com/landmark/au/royal-brisbane-women-s-hospital.vi.html", "http://www.booking.com/landmark/hk/nam-cheong-station.html&lang=id", "http://www.booking.com/landmark/it/spanish-steps.ca.html", "http://www.booking.com/landmark/sg/asian-civilisations-museum.html&lang=fi", "http://www.booking.com/place/fi-1376029.pt.html", "http://www.booking.com/place/tn257337.pl.html", "http://www.booking.com/region/ca/niagarafalls.ar.html&selected_currency=PLN", "http://www.booking.com/region/mx/queretaro.pt-pt.html&selected_currency=AUD", "http://www.booking.com/searchresults.en.html?city=20063074", "http://www.booking.com/searchresults.et.html?checkin=;checkout=;city=-394632", "http://www.booking.com/searchresults.lv.html?region=3936", "http://www.cevredanismanlari.com/index.php/component/k2/index.php/mevzuat/genel-yazlar/item/dosyalar/index.php?option=com_k2&view=item&id=16:iso-14001-%C3%A7evre-y%C3%B6netim-sistemi&Itemid=132&limitstart=107120", "http://www.dh-wholesaler.com/MENS-POLO-RACING-TEE-RL-p-417.html", "http://www.employabilityonline.net/", "http://www.esso.inc.ru/board/tools.php?event=profile&pname=Invinerrq", "http://www.filesurgery.ru/searchfw/kids_clothes-3.html", "http://www.furnitureandcarpetsource.com/Item.aspx?ItemID=-2107311899&ItemNum=53-T3048", "http://www.gets.cn/product/Gold-Sand-Lampwork-Glass-Beads--Flat-round--28x28x13mm_p260717.html", "http://www.gets.cn/wholesale-Sterling-Silver-Pendant-Findings-3577_S--L-Star-P-1.html?view=1&by=1", "http://www.homeandgardenadvice.com/diy/Mortgages_Loans_and_Financing/9221.html", "http://www.hongkongairport.com/eng/index.html/passenger/passenger/transport/to-from-airport/business/about-the-airport/transport/shopping/entertainment/t2/passenger/interactive-map.html", "http://www.hongkongairport.com/eng/index.html/shopping/insideshopping/all/passenger/transfer-transit/all/airline-information/shopping/entertainment/t2/business/about-the-airport/welcome.html", "http://www.hongkongairport.com/eng/index.html/transport/business/about-the-airport/transport/business/airport-authority/passenger/shopping/dining/all/dining.html", "http://www.idedge.com/index.cfm/fuseaction/category.display/category_id/298/index.cfm", "http://www.istanbulburda.com/aramalar.php", "http://www.jewelryinthenet.com/ads/AdDetail.aspx?AdID=1-0311002490689&stid=22-0111001020877", "http://www.johnnydepp.ru/forum/index.php?showtopic=1629&mode=linearplus&view=findpost&p=186977", "http://www.johnnydepp.ru/forum/index.php?showtopic=476&st=60&p=87379&", "http://www.joseleano.com/joomla/index.php/audio", "http://www.kaplicarehberi.com/tag/sakar-ilicali-kaplicalari/feed", "http://www.khaber.com.tr/arama.html?key=%C3%A7avdar", "http://www.kiz-oyunlari1.com/1783/4437/4363/1056/4170/Bump-Copter2-.html", "http://www.kiz-oyunlari1.com/3752/2612/4175/1166/3649/1047/Angelina-Oyunu.html", "http://www.kiz-oyunlari1.com/4266/3630/3665/3286/4121/301/3274/Sinir-Sinekler-.html", "http://www.kuldiga.lv/index.php?f=8&cat=371", "http://www.kuldiga.lv/index.php/img/index.php?l=lv&art_id=1836&show_c=&cat=85", "http://www.patronessa.ru/remontiruemsya/kuzovnie30raboti.html", "http://www.rapdict.org/Nu_Money?title=Talk:Nu_Money&action=edit", "http://www.serafin-phu.tabor24.com/?page=8", "http://www.shoes-store.org/brand1/Kids/Minnetonka.html", "http://www.shoes-store.org/shoes-store.xml", "http://www.way2allah.com/khotab-download-34695.htm"};
+ data.clear();
+ data.insert(data.begin(), urls, urls + Y_ARRAY_SIZE(urls));
+ }
+
+ typedef THashMap<TStringBuf, NGreedyDict::TEntry> TDict;
+
+ TAutoPtr<NGreedyDict::TEntrySet> DoTestBuilder(const NGreedyDict::TBuildSettings& s,
+ TDict& res) {
+ using namespace NGreedyDict;
+
+ TStringBufs data;
+ FillData(data);
+
+ TDictBuilder b(s);
+ b.SetInput(data);
+ b.Build(256 + 128);
+
+ TEntrySet& set = b.EntrySet();
+
+ for (const auto& it : set) {
+ if (it.Score) {
+ res[it.Str] = it;
+ }
+ }
+
+ return b.ReleaseEntrySet();
+ }
+
+ void DoAssertEntry(TStringBuf entry, ui32 number, i32 parent, float score, const TDict& dict) {
+ TDict::const_iterator it = dict.find(entry);
+ UNIT_ASSERT_C(it != dict.end(), entry);
+ UNIT_ASSERT_VALUES_EQUAL_C(it->second.Number, number, entry);
+ UNIT_ASSERT_VALUES_EQUAL_C(it->second.NearestPrefix, parent, entry);
+ UNIT_ASSERT_VALUES_EQUAL_C(round(it->second.Score * 10000), round(score * 10000), entry);
+ }
+
+ void TestBuilder() {
+ TAutoPtr<NGreedyDict::TEntrySet> set;
+ THashMap<TStringBuf, NGreedyDict::TEntry> res;
+ NGreedyDict::TBuildSettings s;
+ set = DoTestBuilder(s, res);
+
+ UNIT_ASSERT_VALUES_EQUAL(set->size(), 295u);
+ UNIT_ASSERT_VALUES_EQUAL(res.size(), 110u);
+
+ DoAssertEntry("%", 37, -1, 0.00375193, res);
+ DoAssertEntry("%7", 38, 37, 0.00513299, res);
+ DoAssertEntry("&", 39, -1, 0.00794527, res);
+ DoAssertEntry("+", 44, -1, 0.000441404, res);
+ DoAssertEntry(",", 45, -1, 0.000441404, res);
+ DoAssertEntry("-", 46, -1, 0.0417126, res);
+ DoAssertEntry(".", 47, -1, 0.0196425, res);
+ DoAssertEntry(".com/", 48, 47, 0.0374482, res);
+ DoAssertEntry(".html", 49, 47, 0.0496577, res);
+ DoAssertEntry(".html?", 50, 49, 0.0153908, res);
+ DoAssertEntry(".php", 51, 47, 0.0123585, res);
+ DoAssertEntry(".ru/", 52, 47, 0.0150027, res);
+ DoAssertEntry("/", 53, -1, 0.0452439, res);
+ DoAssertEntry("/index", 54, 53, 0.0158905, res);
+ DoAssertEntry("0", 55, -1, 0.00816597, res);
+ DoAssertEntry("1", 56, -1, 0.0167733, res);
+ DoAssertEntry("10", 57, 56, 0.00530474, res);
+ DoAssertEntry("2", 58, -1, 0.0101523, res);
+ DoAssertEntry("20", 59, 58, 0.00674234, res);
+ DoAssertEntry("3", 60, -1, 0.01258, res);
+ DoAssertEntry("32", 61, 60, 0.00490697, res);
+ DoAssertEntry("4", 62, -1, 0.00993158, res);
+ DoAssertEntry("5", 63, -1, 0.00617965, res);
+ DoAssertEntry("6", 64, -1, 0.00971088, res);
+ DoAssertEntry("7", 65, -1, 0.0101523, res);
+ DoAssertEntry("8", 66, -1, 0.00728316, res);
+ DoAssertEntry("9", 67, -1, 0.00728316, res);
+ DoAssertEntry(":", 68, -1, 0.000662106, res);
+ DoAssertEntry(";", 69, -1, 0.000882807, res);
+ DoAssertEntry("=", 71, -1, 0.01258, res);
+ DoAssertEntry("?", 73, -1, 0.00397263, res);
+ DoAssertEntry("A", 75, -1, 0.00264842, res);
+ DoAssertEntry("B", 76, -1, 0.00220702, res);
+ DoAssertEntry("C", 77, -1, 0.00353123, res);
+ DoAssertEntry("D", 78, -1, 0.00375193, res);
+ DoAssertEntry("E", 79, -1, 0.00286912, res);
+ DoAssertEntry("F", 80, -1, 0.00110351, res);
+ DoAssertEntry("G", 81, -1, 0.00110351, res);
+ DoAssertEntry("H", 82, -1, 0.000220702, res);
+ DoAssertEntry("I", 83, -1, 0.00198632, res);
+ DoAssertEntry("K", 85, -1, 0.000441404, res);
+ DoAssertEntry("L", 86, -1, 0.00198632, res);
+ DoAssertEntry("M", 87, -1, 0.00154491, res);
+ DoAssertEntry("N", 88, -1, 0.00154491, res);
+ DoAssertEntry("O", 89, -1, 0.00132421, res);
+ DoAssertEntry("P", 90, -1, 0.00308983, res);
+ DoAssertEntry("R", 92, -1, 0.000662106, res);
+ DoAssertEntry("S", 93, -1, 0.00264842, res);
+ DoAssertEntry("T", 94, -1, 0.00110351, res);
+ DoAssertEntry("U", 95, -1, 0.000220702, res);
+ DoAssertEntry("V", 96, -1, 0.000441404, res);
+ DoAssertEntry("W", 97, -1, 0.000441404, res);
+ DoAssertEntry("X", 98, -1, 0.000220702, res);
+ DoAssertEntry("Y", 99, -1, 0.000220702, res);
+ DoAssertEntry("_", 105, -1, 0.00904877, res);
+ DoAssertEntry("a", 107, -1, 0.0505407, res);
+ DoAssertEntry("an", 108, 107, 0.018273, res);
+ DoAssertEntry("ar", 109, 107, 0.0169385, res);
+ DoAssertEntry("b", 110, -1, 0.0156698, res);
+ DoAssertEntry("c", 111, -1, 0.018539, res);
+ DoAssertEntry("cat", 112, 111, 0.00846732, res);
+ DoAssertEntry("ch", 113, 111, 0.00644872, res);
+ DoAssertEntry("com", 114, 111, 0.00724235, res);
+ DoAssertEntry("ct", 115, 111, 0.00605729, res);
+ DoAssertEntry("d", 116, -1, 0.020746, res);
+ DoAssertEntry("di", 117, 116, 0.00730659, res);
+ DoAssertEntry("e", 118, -1, 0.0624586, res);
+ DoAssertEntry("en", 119, 118, 0.0108999, res);
+ DoAssertEntry("ent", 120, 119, 0.00616002, res);
+ DoAssertEntry("f", 121, -1, 0.00860737, res);
+ DoAssertEntry("fi", 122, 121, 0.00423196, res);
+ DoAssertEntry("g", 123, -1, 0.0180975, res);
+ DoAssertEntry("go", 124, 123, 0.00601862, res);
+ DoAssertEntry("h", 125, -1, 0.010373, res);
+ DoAssertEntry("ho", 126, 125, 0.00570298, res);
+ DoAssertEntry("http://", 127, 125, 0.0494372, res);
+ DoAssertEntry("http://www.", 128, 127, 0.0849702, res);
+ DoAssertEntry("http://www.booking.com/", 129, 128, 0.071066, res);
+ DoAssertEntry("http://www.booking.com/hotel/", 130, 129, 0.121607, res);
+ DoAssertEntry("i", 131, -1, 0.0258221, res);
+ DoAssertEntry("id=", 132, 131, 0.00725369, res);
+ DoAssertEntry("im", 133, 131, 0.00373318, res);
+ DoAssertEntry("in", 134, 131, 0.013625, res);
+ DoAssertEntry("ing", 135, 134, 0.00795491, res);
+ DoAssertEntry("ion", 136, 131, 0.00796149, res);
+ DoAssertEntry("it", 137, 131, 0.00953416, res);
+ DoAssertEntry("j", 138, -1, 0.00132421, res);
+ DoAssertEntry("k", 139, -1, 0.0134628, res);
+ DoAssertEntry("l", 140, -1, 0.0381814, res);
+ DoAssertEntry("m", 141, -1, 0.0174354, res);
+ DoAssertEntry("mer", 142, 141, 0.00711846, res);
+ DoAssertEntry("n", 143, -1, 0.0132421, res);
+ DoAssertEntry("o", 144, -1, 0.0302362, res);
+ DoAssertEntry("on", 145, 144, 0.00802271, res);
+ DoAssertEntry("ou", 146, 144, 0.00414545, res);
+ DoAssertEntry("p", 147, -1, 0.0225116, res);
+ DoAssertEntry("port", 148, 147, 0.0123532, res);
+ DoAssertEntry("q", 149, -1, 0.00176561, res);
+ DoAssertEntry("r", 150, -1, 0.0401677, res);
+ DoAssertEntry("ran", 151, 150, 0.00686918, res);
+ DoAssertEntry("s", 152, -1, 0.0487751, res);
+ DoAssertEntry("sho", 153, 152, 0.0113876, res);
+ DoAssertEntry("t", 154, -1, 0.0379607, res);
+ DoAssertEntry("u", 155, -1, 0.0211874, res);
+ DoAssertEntry("v", 156, -1, 0.00595895, res);
+ DoAssertEntry("vi", 157, 156, 0.00480673, res);
+ DoAssertEntry("w", 158, -1, 0.00816597, res);
+ DoAssertEntry("x", 159, -1, 0.00375193, res);
+ DoAssertEntry("y", 160, -1, 0.0130214, res);
+ DoAssertEntry("z", 161, -1, 0.00353123, res);
+ }
+};
+
+UNIT_TEST_SUITE_REGISTRATION(TGreedyDictTest);
diff --git a/library/cpp/codecs/greedy_dict/ut/ya.make b/library/cpp/codecs/greedy_dict/ut/ya.make
new file mode 100644
index 00000000000..bd67d1a4522
--- /dev/null
+++ b/library/cpp/codecs/greedy_dict/ut/ya.make
@@ -0,0 +1,9 @@
+UNITTEST_FOR(library/cpp/codecs/greedy_dict)
+
+OWNER(velavokr)
+
+SRCS(
+ greedy_dict_ut.cpp
+)
+
+END()
diff --git a/library/cpp/codecs/greedy_dict/ya.make b/library/cpp/codecs/greedy_dict/ya.make
new file mode 100644
index 00000000000..2a57224f7e1
--- /dev/null
+++ b/library/cpp/codecs/greedy_dict/ya.make
@@ -0,0 +1,15 @@
+OWNER(velavokr)
+
+LIBRARY()
+
+SRCS(
+ gd_builder.cpp
+ gd_entry.cpp
+)
+
+PEERDIR(
+ library/cpp/containers/comptrie
+ library/cpp/string_utils/relaxed_escaper
+)
+
+END()