1 package com.hughes.android.dictionary.parser;
3 import java.io.BufferedReader;
5 import java.io.FileInputStream;
6 import java.io.IOException;
7 import java.io.InputStreamReader;
8 import java.nio.charset.Charset;
9 import java.util.Arrays;
10 import java.util.LinkedHashSet;
11 import java.util.List;
13 import java.util.logging.Logger;
14 import java.util.regex.Matcher;
15 import java.util.regex.Pattern;
17 import com.hughes.android.dictionary.engine.DictionaryBuilder;
18 import com.hughes.android.dictionary.engine.EntryData;
19 import com.hughes.android.dictionary.engine.EntryTypeName;
20 import com.hughes.android.dictionary.engine.IndexBuilder;
21 import com.hughes.android.dictionary.engine.Language;
22 import com.hughes.android.dictionary.engine.PairEntry;
23 import com.hughes.android.dictionary.engine.PairEntry.Pair;
25 public class DictFileParser {
27 static final Logger logger = Logger.getLogger(DictFileParser.class.getName());
30 public static final Pattern TAB = Pattern.compile("\\t");
33 public static final Pattern DOUBLE_COLON = Pattern.compile(" :: ");
34 public static final Pattern PIPE = Pattern.compile("\\|");
36 static final Pattern SPACES = Pattern.compile("\\s+");
37 // static final Pattern DE_NOUN = Pattern.compile("([^ ]+) *\\{(m|f|n|pl)\\}");
38 // static final Pattern EN_VERB = Pattern.compile("^to ([^ ]+)");
40 static final Pattern BRACKETED = Pattern.compile("\\[([^]]+)\\]");
41 static final Pattern PARENTHESIZED = Pattern.compile("\\(([^)]+)\\)");
42 static final Pattern CURLY_BRACED = Pattern.compile("\\{([^}]+)\\}");
44 static final Pattern NON_CHAR_DASH = Pattern.compile("[^-'\\p{L}0-9]+");
45 static final Pattern NON_CHAR = Pattern.compile("[^\\p{L}0-9]+");
47 static final Pattern TRIM_PUNC = Pattern.compile("^[^\\p{L}0-9]+|[^\\p{L}0-9]+$");
49 final Charset charset;
50 final boolean flipCols;
52 final Pattern fieldSplit;
53 final Pattern subfieldSplit;
55 final DictionaryBuilder dictBuilder;
56 final IndexBuilder[] langIndexBuilders;
57 final IndexBuilder bothIndexBuilder;
59 // final Set<String> alreadyDone = new HashSet<String>();
61 public DictFileParser(final Charset charset, boolean flipCols,
62 final Pattern fieldSplit, final Pattern subfieldSplit,
63 final DictionaryBuilder dictBuilder, final IndexBuilder[] langIndexBuilders,
64 final IndexBuilder bothIndexBuilder) {
65 this.charset = charset;
66 this.flipCols = flipCols;
67 this.fieldSplit = fieldSplit;
68 this.subfieldSplit = subfieldSplit;
69 this.dictBuilder = dictBuilder;
70 this.langIndexBuilders = langIndexBuilders;
71 this.bothIndexBuilder = bothIndexBuilder;
74 public void parseFile(final File file) throws IOException {
75 final BufferedReader reader = new BufferedReader(new InputStreamReader(new FileInputStream(file), charset));
78 while ((line = reader.readLine()) != null) {
79 if (count % 10000 == 0) {
80 logger.info("count=" + count + ", line=" + line);
87 private void parseLine(final String line) {
88 if (line.startsWith("#") || line.length() == 0) {
89 logger.info("Skipping comment line: " + line);
92 final String[] fields = fieldSplit.split(line);
93 if (fields.length != 2) {
94 logger.warning("Malformed line: " + line);
98 fields[0] = SPACES.matcher(fields[0]).replaceAll(" ").trim();
99 fields[1] = SPACES.matcher(fields[1]).replaceAll(" ").trim();
101 final String temp = fields[0];
102 fields[0] = fields[1];
106 final String[][] subfields = new String[2][];
107 if (subfieldSplit != null) {
108 subfields[0] = subfieldSplit.split(fields[0]);
109 subfields[1] = subfieldSplit.split(fields[1]);
110 if (subfields[0].length != subfields[1].length) {
111 logger.warning("Number of subfields doesn't match: " + line);
115 subfields[0] = new String[] { fields[0] };
116 subfields[1] = new String[] { fields[1] };
119 final PairEntry pairEntry = new PairEntry();
120 for (int i = 0; i < subfields[0].length; ++i) {
121 subfields[0][i] = subfields[0][i].trim();
122 subfields[1][i] = subfields[1][i].trim();
123 pairEntry.pairs.add(new Pair(subfields[0][i], subfields[1][i]));
125 final EntryData entryData = new EntryData(dictBuilder.dictionary.pairEntries.size(), pairEntry);
126 dictBuilder.dictionary.pairEntries.add(pairEntry);
128 for (int l = 0; l < 2; ++l) {
129 // alreadyDone.clear();
131 for (int j = 0; j < subfields[l].length; ++j) {
132 String subfield = subfields[l][j];
133 final IndexBuilder indexBuilder = langIndexBuilders[l];
134 if (indexBuilder.index.sortLanguage == Language.de) {
135 subfield = parseField_DE(indexBuilder, subfield, entryData, j);
136 } else if (indexBuilder.index.sortLanguage == Language.en) {
137 subfield = parseField_EN(indexBuilder, subfield, entryData, j);
139 parseFieldGeneric(indexBuilder, subfield, entryData, j, subfields[l].length);
144 private void parseFieldGeneric(final IndexBuilder indexBuilder, String field,
145 final EntryData entryData, final int subfieldIdx, final int numSubFields) {
146 // remove bracketed and parenthesized stuff.
147 final StringBuilder bracketed = new StringBuilder();
148 final StringBuilder parenthesized = new StringBuilder();
151 while ((matcher = BRACKETED.matcher(field)).find()) {
152 bracketed.append(matcher.group(1)).append(" ");
153 field = matcher.replaceFirst(" ");
156 while ((matcher = PARENTHESIZED.matcher(field)).find()) {
157 parenthesized.append(matcher.group(1)).append(" ");
158 field = matcher.replaceFirst(" ");
161 field = SPACES.matcher(field).replaceAll(" ").trim();
163 // split words on non -A-z0-9, do them.
164 final String[] tokens = NON_CHAR_DASH.split(field);
166 final EntryTypeName entryTypeName;
167 if (numSubFields == 1) {
168 assert subfieldIdx == 0;
169 if (tokens.length == 1) {
170 entryTypeName = EntryTypeName.ONE_WORD;
171 } else if (tokens.length == 2) {
172 entryTypeName = EntryTypeName.TWO_WORDS;
173 } else if (tokens.length == 3) {
174 entryTypeName = EntryTypeName.THREE_WORDS;
175 } else if (tokens.length == 4) {
176 entryTypeName = EntryTypeName.FOUR_WORDS;
178 entryTypeName = EntryTypeName.FIVE_OR_MORE_WORDS;
181 assert numSubFields > 1;
182 if (subfieldIdx == 0) {
183 if (tokens.length == 1) {
184 entryTypeName = EntryTypeName.MULTIROW_HEAD_ONE_WORD;
186 entryTypeName = EntryTypeName.MULTIROW_HEAD_MANY_WORDS;
189 assert subfieldIdx > 0;
190 if (tokens.length == 1) {
191 entryTypeName = EntryTypeName.MULTIROW_TAIL_ONE_WORD;
193 entryTypeName = EntryTypeName.MULTIROW_TAIL_MANY_WORDS;
198 for (String token : tokens) {
199 token = TRIM_PUNC.matcher(token).replaceAll("");
200 if (/*!alreadyDone.contains(token) && */token.length() > 0) {
201 final List<EntryData> entries = indexBuilder.getOrCreateEntries(token, entryTypeName);
202 entries.add(entryData);
203 // alreadyDone.add(token);
205 // also split words on dashes, do them, too.
206 if (token.contains("-")) {
207 final String[] dashed = token.split("-");
208 for (final String dashedToken : dashed) {
209 if (/*!alreadyDone.contains(dashedToken) && */dashedToken.length() > 0) {
210 final List<EntryData> dashEntries = indexBuilder.getOrCreateEntries(dashedToken, EntryTypeName.PART_OF_HYPHENATED);
211 dashEntries.add(entryData);
216 } // if (!alreadyDone.contains(token)) {
217 } // for (final String token : tokens) {
219 // process bracketed stuff (split on spaces and dashes always)
220 final String[] bracketedTokens = NON_CHAR.split(bracketed.toString());
221 for (final String token : bracketedTokens) {
222 assert !token.contains("-");
223 if (/*!alreadyDone.contains(token) && */token.length() > 0) {
224 final List<EntryData> entries = indexBuilder.getOrCreateEntries(token, EntryTypeName.BRACKETED);
225 entries.add(entryData);
229 // process paren stuff
230 final String[] parenTokens = NON_CHAR.split(parenthesized.toString());
231 for (final String token : parenTokens) {
232 assert !token.contains("-");
233 if (/*!alreadyDone.contains(token) && */token.length() > 0) {
234 final List<EntryData> entries = indexBuilder.getOrCreateEntries(token, EntryTypeName.PARENTHESIZED);
235 entries.add(entryData);
241 private String parseField_DE(final IndexBuilder indexBuilder, String field,
242 final EntryData entryData, final int subfieldIdx) {
244 // final Matcher matcher = DE_NOUN.matcher(field);
245 // while (matcher.find()) {
246 // final String noun = matcher.group(1);
247 //final String gender = matcher.group(2);
248 // if (alreadyDone.add(noun)) {
249 // System.out.println("Found DE noun " + noun + ", " + gender);
250 // final List<EntryData> entries = indexBuilder.getOrCreateEntries(noun, EntryTypeName.NOUN);
251 // entries.add(entryData);
255 // In English, curly braces are used for different tenses.
256 field = CURLY_BRACED.matcher(field).replaceAll(" ");
261 private String parseField_EN(final IndexBuilder indexBuilder, String field,
262 final EntryData entryData, final int subfieldIdx) {
263 if (field.startsWith("to ")) {
264 field = field.substring(3);
269 public static final Set<String> tokenize(final String text, final Pattern pattern) {
270 final String[] split = pattern.split(text);
271 final Set<String> result = new LinkedHashSet<String>(Arrays.asList(split));