1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
#pragma once
#include "base/string_utils.hpp"
#include <cstdint>
#include <map>
namespace search
{
class IdfMap
{
public:
struct Delegate
{
virtual ~Delegate() = default;
virtual uint64_t GetNumDocs(strings::UniString const & token, bool isPrefix) const = 0;
};
IdfMap(Delegate const & delegate, double unknownIdf);
void Set(strings::UniString const & s, bool isPrefix, double idf)
{
SetImpl(isPrefix ? m_prefixIdfs : m_fullIdfs, s, idf);
}
double Get(strings::UniString const & s, bool isPrefix)
{
return GetImpl(isPrefix ? m_prefixIdfs : m_fullIdfs, s, isPrefix);
}
private:
using Map = std::map<strings::UniString, double>;
void SetImpl(Map & idfs, strings::UniString const & s, double idf) { idfs[s] = idf; }
double GetImpl(Map & idfs, strings::UniString const & s, bool isPrefix);
Map m_fullIdfs;
Map m_prefixIdfs;
Delegate const & m_delegate;
double m_unknownIdf;
};
} // namespace search
|