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
119
120
121
122
123
124
125
126
127
128
129
130
|
// Copyright (c) 2006-2008 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
// This tool converts Hunspell .aff/.dic pairs to a combined binary dictionary
// format (.bdic). This format is more compact, and can be more efficiently
// read by the client application.
//
// We do this conversion manually before publishing dictionary files. It is not
// part of any build process.
//
// See PrintHelp() below for usage.
#include <stdio.h>
#include "base/file_util.h"
#include "base/icu_util.h"
#include "base/logging.h"
#include "base/process_util.h"
#include "base/string_util.h"
#include "chrome/third_party/hunspell/google/bdict_reader.h"
#include "chrome/third_party/hunspell/google/bdict_writer.h"
#include "chrome/tools/convert_dict/aff_reader.h"
#include "chrome/tools/convert_dict/dic_reader.h"
namespace {
// Compares the given word list with the serialized trie to make sure they
// are the same.
bool VerifyWords(const convert_dict::DicReader::WordList& org_words,
const std::string& serialized) {
hunspell::BDictReader reader;
if (!reader.Init(reinterpret_cast<const unsigned char*>(serialized.data()),
serialized.size())) {
printf("BDict is invalid\n");
return false;
}
hunspell::WordIterator iter = reader.GetAllWordIterator();
int affix_ids[hunspell::BDict::MAX_AFFIXES_PER_WORD];
static const int buf_size = 128;
char buf[buf_size];
for (size_t i = 0; i < org_words.size(); i++) {
int affix_matches = iter.Advance(buf, buf_size, affix_ids);
if (affix_matches == 0)
return false; // Found the end before we expectd.
if (org_words[i].first != buf)
return false; // Word doesn't match.
if (affix_matches != static_cast<int>(org_words[i].second.size()))
return false; // Different number of affix indices.
// Check the individual affix indices.
for (size_t affix_index = 0; affix_index < org_words[i].second.size();
affix_index++) {
if (affix_ids[affix_index] != org_words[i].second[affix_index])
return false; // Index doesn't match.
}
}
return true;
}
int PrintHelp() {
printf("Usage: convert_dict <dicfile base name>\n\n");
printf("Example:\n");
printf(" convert_dict en-US\nwill read en-US.dic / en-US.aff and\n");
printf("generate en-US.bdic\n\n");
return 1;
}
} // namespace
int main(int argc, char* argv[]) {
base::EnableTerminationOnHeapCorruption();
if (argc != 2)
return PrintHelp();
icu_util::Initialize();
std::string file_base = argv[1];
std::string aff_name = file_base + ".aff";
printf("Reading %s ...\n", aff_name.c_str());
convert_dict::AffReader aff_reader(aff_name.c_str());
if (!aff_reader.Read()) {
printf("Unable to read the aff file.\n");
return 1;
}
std::string dic_name = file_base + ".dic";
printf("Reading %s ...\n", dic_name.c_str());
convert_dict::DicReader dic_reader(dic_name.c_str());
if (!dic_reader.Read(&aff_reader)) {
printf("Unable to read the dic file.\n");
return 1;
}
hunspell::BDictWriter writer;
writer.SetComment(aff_reader.comments());
writer.SetAffixRules(aff_reader.affix_rules());
writer.SetAffixGroups(aff_reader.GetAffixGroups());
writer.SetReplacements(aff_reader.replacements());
writer.SetOtherCommands(aff_reader.other_commands());
writer.SetWords(dic_reader.words());
printf("Serializing...\n");
std::string serialized = writer.GetBDict();
printf("Verifying...\n");
if (!VerifyWords(dic_reader.words(), serialized)) {
printf("ERROR converting, the dictionary does not check out OK.");
return 1;
}
std::string out_name = file_base + ".bdic";
printf("Writing %s ...\n", out_name.c_str());
FILE* out_file = file_util::OpenFile(out_name, "wb");
if (!out_file) {
printf("ERROR writing file\n");
return 1;
}
size_t written = fwrite(&serialized[0], 1, serialized.size(), out_file);
CHECK(written == serialized.size());
file_util::CloseFile(out_file);
return 0;
}
|
