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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
#pragma once
#ifndef MARISA_GRIMOIRE_TRIE_STATE_H_
#define MARISA_GRIMOIRE_TRIE_STATE_H_
#include "../vector.h"
#include "history.h"
namespace marisa {
namespace grimoire {
namespace trie {
// A search agent has its internal state and the status codes are defined
// below.
typedef enum StatusCode {
MARISA_READY_TO_ALL,
MARISA_READY_TO_COMMON_PREFIX_SEARCH,
MARISA_READY_TO_PREDICTIVE_SEARCH,
MARISA_END_OF_COMMON_PREFIX_SEARCH,
MARISA_END_OF_PREDICTIVE_SEARCH,
} StatusCode;
class State {
public:
State()
: key_buf_(), history_(), node_id_(0), query_pos_(0),
history_pos_(0), status_code_(MARISA_READY_TO_ALL) {}
void set_node_id(std::size_t node_id) {
MARISA_DEBUG_IF(node_id > MARISA_UINT32_MAX, MARISA_SIZE_ERROR);
node_id_ = (UInt32)node_id;
}
void set_query_pos(std::size_t query_pos) {
MARISA_DEBUG_IF(query_pos > MARISA_UINT32_MAX, MARISA_SIZE_ERROR);
query_pos_ = (UInt32)query_pos;
}
void set_history_pos(std::size_t history_pos) {
MARISA_DEBUG_IF(history_pos > MARISA_UINT32_MAX, MARISA_SIZE_ERROR);
history_pos_ = (UInt32)history_pos;
}
void set_status_code(StatusCode status_code) {
status_code_ = status_code;
}
std::size_t node_id() const {
return node_id_;
}
std::size_t query_pos() const {
return query_pos_;
}
std::size_t history_pos() const {
return history_pos_;
}
StatusCode status_code() const {
return status_code_;
}
const Vector<char> &key_buf() const {
return key_buf_;
}
const Vector<History> &history() const {
return history_;
}
Vector<char> &key_buf() {
return key_buf_;
}
Vector<History> &history() {
return history_;
}
void reset() {
status_code_ = MARISA_READY_TO_ALL;
}
void lookup_init() {
node_id_ = 0;
query_pos_ = 0;
status_code_ = MARISA_READY_TO_ALL;
}
void reverse_lookup_init() {
key_buf_.resize(0);
key_buf_.reserve(32);
status_code_ = MARISA_READY_TO_ALL;
}
void common_prefix_search_init() {
node_id_ = 0;
query_pos_ = 0;
status_code_ = MARISA_READY_TO_COMMON_PREFIX_SEARCH;
}
void predictive_search_init() {
key_buf_.resize(0);
key_buf_.reserve(64);
history_.resize(0);
history_.reserve(4);
node_id_ = 0;
query_pos_ = 0;
history_pos_ = 0;
status_code_ = MARISA_READY_TO_PREDICTIVE_SEARCH;
}
private:
Vector<char> key_buf_;
Vector<History> history_;
UInt32 node_id_;
UInt32 query_pos_;
UInt32 history_pos_;
StatusCode status_code_;
// Disallows copy and assignment.
State(const State &);
State &operator=(const State &);
};
} // namespace trie
} // namespace grimoire
} // namespace marisa
#endif // MARISA_GRIMOIRE_TRIE_STATE_H_
|
