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
|
// 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/icu_util.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 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[]) {
process_util::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;
fopen_s(&out_file, out_name.c_str(), "wb");
if (!out_file) {
printf("ERROR writing file\n");
return 1;
}
fwrite(&serialized[0], 1, serialized.size(), out_file);
fclose(out_file);
return 0;
}
|