src/com/hughes/android/dictionary/parser/DictFileParser.java

   1 package com.hughes.android.dictionary.parser;
   2
   3 import java.io.BufferedReader;
   4 import java.io.File;
   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;
  12 import java.util.Set;
  13 import java.util.logging.Logger;
  14 import java.util.regex.Matcher;
  15 import java.util.regex.Pattern;
  16
  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;
  24
  25 public class DictFileParser {
  26
  27   static final Logger logger = Logger.getLogger(DictFileParser.class.getName());
  28
  29   // Dictcc
  30   public static final Pattern TAB = Pattern.compile("\\t");
  31
  32   // Chemnitz
  33   public static final Pattern DOUBLE_COLON = Pattern.compile(" :: ");
  34   public static final Pattern PIPE = Pattern.compile("\\|");
  35
  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 ([^ ]+)");
  39
  40   static final Pattern BRACKETED = Pattern.compile("\\[([^]]+)\\]");
  41   static final Pattern PARENTHESIZED = Pattern.compile("\\(([^)]+)\\)");
  42   static final Pattern CURLY_BRACED = Pattern.compile("\\{([^}]+)\\}");
  43
  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]+");
  46
  47   static final Pattern TRIM_PUNC = Pattern.compile("^[^\\p{L}0-9]+|[^\\p{L}0-9]+$");
  48
  49   final Charset charset;
  50   final boolean flipCols;
  51
  52   final Pattern fieldSplit;
  53   final Pattern subfieldSplit;
  54
  55   final DictionaryBuilder dictBuilder;
  56   final IndexBuilder[] langIndexBuilders;
  57   final IndexBuilder bothIndexBuilder;
  58
  59   // final Set<String> alreadyDone = new HashSet<String>();
  60
  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;
  72   }
  73
  74   public void parseFile(final File file) throws IOException {
  75     final BufferedReader reader = new BufferedReader(new InputStreamReader(new FileInputStream(file), charset));
  76     String line;
  77     int count = 0;
  78     while ((line = reader.readLine()) != null) {
  79       if (count % 10000 == 0) {
  80         logger.info("count=" + count + ", line=" + line);
  81       }
  82       parseLine(line);
  83       ++count;
  84     }
  85   }
  86
  87   private void parseLine(final String line) {
  88     if (line.startsWith("#") || line.length() == 0) {
  89       logger.info("Skipping comment line: " + line);
  90       return;
  91     }
  92     final String[] fields = fieldSplit.split(line);
  93     if (fields.length != 2) {
  94       logger.warning("Malformed line: " + line);
  95       return;
  96     }
  97
  98     fields[0] = SPACES.matcher(fields[0]).replaceAll(" ").trim();
  99     fields[1] = SPACES.matcher(fields[1]).replaceAll(" ").trim();
 100     if (flipCols) {
 101       final String temp = fields[0];
 102       fields[0] = fields[1];
 103       fields[1] = temp;
 104     }
 105
 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);
 112         return;
 113       }
 114     } else {
 115       subfields[0] = new String[] { fields[0] };
 116       subfields[1] = new String[] { fields[1] };
 117     }
 118
 119     final Pair[] pairs = new Pair[subfields[0].length];
 120     for (int i = 0; i < pairs.length; ++i) {
 121       subfields[0][i] = subfields[0][i].trim();
 122       subfields[1][i] = subfields[1][i].trim();
 123       pairs[i] = new Pair(subfields[0][i], subfields[1][i]);
 124     }
 125     final PairEntry pairEntry = new PairEntry(pairs);
 126     final EntryData entryData = new EntryData(dictBuilder.dictionary.pairEntries.size(), pairEntry);
 127     dictBuilder.dictionary.pairEntries.add(pairEntry);
 128
 129     for (int l = 0; l < 2; ++l) {
 130       // alreadyDone.clear();
 131
 132       for (int j = 0; j < subfields[l].length; ++j) {
 133         String subfield = subfields[l][j];
 134         final IndexBuilder indexBuilder = langIndexBuilders[l];
 135         if (indexBuilder.index.sortLanguage == Language.de) {
 136           subfield = parseField_DE(indexBuilder, subfield, entryData, j);
 137         } else if (indexBuilder.index.sortLanguage == Language.en) {
 138           subfield = parseField_EN(indexBuilder, subfield, entryData, j);
 139         }
 140         parseFieldGeneric(indexBuilder, subfield, entryData, j, subfields[l].length);
 141       }
 142     }
 143   }
 144
 145   private void parseFieldGeneric(final IndexBuilder indexBuilder, String field,
 146       final EntryData entryData, final int subfieldIdx, final int numSubFields) {
 147     // remove bracketed and parenthesized stuff.
 148     final StringBuilder bracketed = new StringBuilder();
 149     final StringBuilder parenthesized = new StringBuilder();
 150
 151     Matcher matcher;
 152     while ((matcher = BRACKETED.matcher(field)).find()) {
 153       bracketed.append(matcher.group(1)).append(" ");
 154       field = matcher.replaceFirst(" ");
 155     }
 156
 157     while ((matcher = PARENTHESIZED.matcher(field)).find()) {
 158       parenthesized.append(matcher.group(1)).append(" ");
 159       field = matcher.replaceFirst(" ");
 160     }
 161
 162     field = SPACES.matcher(field).replaceAll(" ").trim();
 163
 164     // split words on non -A-z0-9, do them.
 165     final String[] tokens = NON_CHAR_DASH.split(field);
 166
 167     final EntryTypeName entryTypeName;
 168     if (numSubFields == 1) {
 169       assert subfieldIdx == 0;
 170       if (tokens.length == 1) {
 171         entryTypeName = EntryTypeName.ONE_WORD;
 172       } else if (tokens.length == 2) {
 173         entryTypeName = EntryTypeName.TWO_WORDS;
 174       } else if (tokens.length == 3) {
 175         entryTypeName = EntryTypeName.THREE_WORDS;
 176       } else if (tokens.length == 4) {
 177         entryTypeName = EntryTypeName.FOUR_WORDS;
 178       } else {
 179         entryTypeName = EntryTypeName.FIVE_OR_MORE_WORDS;
 180       }
 181     } else {
 182       assert numSubFields > 1;
 183       if (subfieldIdx == 0) {
 184         if (tokens.length == 1) {
 185           entryTypeName = EntryTypeName.MULTIROW_HEAD_ONE_WORD;
 186         } else {
 187           entryTypeName = EntryTypeName.MULTIROW_HEAD_MANY_WORDS;
 188         }
 189       } else {
 190         assert subfieldIdx > 0;
 191         if (tokens.length == 1) {
 192           entryTypeName = EntryTypeName.MULTIROW_TAIL_ONE_WORD;
 193         } else {
 194           entryTypeName = EntryTypeName.MULTIROW_TAIL_MANY_WORDS;
 195         }
 196       }
 197     }
 198
 199     for (String token : tokens) {
 200       token = TRIM_PUNC.matcher(token).replaceAll("");
 201       if (/*!alreadyDone.contains(token) && */token.length() > 0) {
 202         final List<EntryData> entries = indexBuilder.getOrCreateEntries(token, entryTypeName);
 203         entries.add(entryData);
 204         // alreadyDone.add(token);
 205
 206         // also split words on dashes, do them, too.
 207         if (token.contains("-")) {
 208           final String[] dashed = token.split("-");
 209           for (final String dashedToken : dashed) {
 210             if (/*!alreadyDone.contains(dashedToken) && */dashedToken.length() > 0) {
 211               final List<EntryData> dashEntries = indexBuilder.getOrCreateEntries(dashedToken, EntryTypeName.PART_OF_HYPHENATED);
 212               dashEntries.add(entryData);
 213             }
 214           }
 215         }
 216
 217       }  // if (!alreadyDone.contains(token)) {
 218     }  // for (final String token : tokens) {
 219
 220     // process bracketed stuff (split on spaces and dashes always)
 221     final String[] bracketedTokens = NON_CHAR.split(bracketed.toString());
 222     for (final String token : bracketedTokens) {
 223       assert !token.contains("-");
 224       if (/*!alreadyDone.contains(token) && */token.length() > 0) {
 225         final List<EntryData> entries = indexBuilder.getOrCreateEntries(token, EntryTypeName.BRACKETED);
 226         entries.add(entryData);
 227       }
 228     }
 229
 230     // process paren stuff
 231     final String[] parenTokens = NON_CHAR.split(parenthesized.toString());
 232     for (final String token : parenTokens) {
 233       assert !token.contains("-");
 234       if (/*!alreadyDone.contains(token) && */token.length() > 0) {
 235         final List<EntryData> entries = indexBuilder.getOrCreateEntries(token, EntryTypeName.PARENTHESIZED);
 236         entries.add(entryData);
 237       }
 238     }
 239
 240   }
 241
 242   private String parseField_DE(final IndexBuilder indexBuilder, String field,
 243       final EntryData entryData, final int subfieldIdx) {
 244
 245 //    final Matcher matcher = DE_NOUN.matcher(field);
 246 //    while (matcher.find()) {
 247 //      final String noun = matcher.group(1);
 248       //final String gender = matcher.group(2);
 249 //      if (alreadyDone.add(noun)) {
 250         // System.out.println("Found DE noun " + noun + ", " + gender);
 251 //        final List<EntryData> entries = indexBuilder.getOrCreateEntries(noun, EntryTypeName.NOUN);
 252 //        entries.add(entryData);
 253 //      }
 254 //    }
 255
 256     // In English, curly braces are used for different tenses.
 257     field = CURLY_BRACED.matcher(field).replaceAll(" ");
 258
 259     return field;
 260   }
 261
 262   private String parseField_EN(final IndexBuilder indexBuilder, String field,
 263       final EntryData entryData, final int subfieldIdx) {
 264     if (field.startsWith("to ")) {
 265       field = field.substring(3);
 266     }
 267     return field;
 268   }
 269
 270   public static final Set<String> tokenize(final String text, final Pattern pattern) {
 271     final String[] split = pattern.split(text);
 272     final Set<String> result = new LinkedHashSet<String>(Arrays.asList(split));
 273     result.remove("");
 274     return result;
 275   }
 276
 277
 278 }