blob: e619bfe5a6729046c17faeab3320de7d1bdcfe50 (
plain)
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
|
#include "phrase.h"
namespace extractor {
int Phrase::Arity() const {
return var_pos.size();
}
int Phrase::GetChunkLen(int index) const {
if (var_pos.size() == 0) {
return symbols.size();
} else if (index == 0) {
return var_pos[0];
} else if (index == var_pos.size()) {
return symbols.size() - var_pos.back() - 1;
} else {
return var_pos[index] - var_pos[index - 1] - 1;
}
}
vector<int> Phrase::Get() const {
return symbols;
}
int Phrase::GetSymbol(int position) const {
return symbols[position];
}
int Phrase::GetNumSymbols() const {
return symbols.size();
}
vector<string> Phrase::GetWords() const {
return words;
}
bool Phrase::operator<(const Phrase& other) const {
return symbols < other.symbols;
}
ostream& operator<<(ostream& os, const Phrase& phrase) {
int current_word = 0;
for (size_t i = 0; i < phrase.symbols.size(); ++i) {
if (phrase.symbols[i] < 0) {
os << "[X," << -phrase.symbols[i] << "]";
} else {
os << phrase.words[current_word];
++current_word;
}
if (i + 1 < phrase.symbols.size()) {
os << " ";
}
}
return os;
}
} // namspace extractor
|