src/com/hughes/android/dictionary/engine/IndexBuilder.java

   1 package com.hughes.android.dictionary.engine;
   2
   3 import java.util.ArrayList;
   4 import java.util.EnumMap;
   5 import java.util.HashSet;
   6 import java.util.List;
   7 import java.util.Map;
   8 import java.util.Set;
   9 import java.util.SortedMap;
  10 import java.util.TreeMap;
  11
  12 import com.hughes.android.dictionary.Language;
  13
  14 public class IndexBuilder {
  15
  16   final DictionaryBuilder dictionaryBuilder;
  17   final Index index;
  18
  19   final SortedMap<String, TokenData> tokenToData;
  20
  21   @SuppressWarnings("unchecked")
  22   IndexBuilder(final DictionaryBuilder dictionaryBuilder, final String shortName, final String longName, final Language language) {
  23     this.dictionaryBuilder = dictionaryBuilder;
  24     index = new Index(dictionaryBuilder.dictionary, shortName, longName, language);
  25     tokenToData = new TreeMap<String, TokenData>(language.getSortCollator());
  26   }
  27
  28   public void build() {
  29     final Set<EntryData> tokenEntryDatas = new HashSet<EntryData>();
  30     final List<RowBase> rows = index.rows;
  31     for (final TokenData tokenData : tokenToData.values()) {
  32       tokenEntryDatas.clear();
  33       final int indexRow = index.sortedIndexEntries.size();
  34       index.sortedIndexEntries.add(new Index.IndexEntry(tokenData.token, rows.size()));
  35       rows.add(new TokenRow(indexRow, rows.size(), index));
  36       System.out.println("Added TokenRow: " + rows.get(rows.size() - 1));
  37       int count = 0;
  38       System.out.println("TOKEN: " + tokenData.token);
  39       for (final Map.Entry<EntryTypeName, List<EntryData>> typeToEntry : tokenData.typeToEntries.entrySet()) {
  40         for (final EntryData entryData : typeToEntry.getValue()) {
  41           if (tokenEntryDatas.add(entryData)) {
  42             rows.add(new PairEntry.Row(entryData.index(), rows.size(), index));
  43             ++count;
  44
  45             System.out.print("  " + typeToEntry.getKey() + ": ");
  46             rows.get(rows.size() - 1).print(System.out);
  47             System.out.println();
  48           }
  49         }
  50       }
  51     }
  52   }
  53
  54   static class TokenData {
  55     final String token;
  56
  57     final Map<EntryTypeName, List<EntryData>> typeToEntries = new EnumMap<EntryTypeName, List<EntryData>>(EntryTypeName.class);
  58
  59     TokenData(final String token) {
  60       assert token.equals(token.trim());
  61       assert token.length() > 0;
  62       this.token = token;
  63     }
  64   }
  65
  66   public TokenData getOrCreateTokenData(final String token) {
  67     TokenData tokenData = tokenToData.get(token);
  68     if (tokenData == null) {
  69       tokenData = new TokenData(token);
  70       tokenToData.put(token, tokenData);
  71     }
  72     return tokenData;
  73   }
  74
  75   public List<EntryData> getOrCreateEntries(final String token, final EntryTypeName entryTypeName) {
  76     final TokenData tokenData = getOrCreateTokenData(token);
  77     List<EntryData> entries = tokenData.typeToEntries.get(entryTypeName);
  78     if (entries == null) {
  79       entries = new ArrayList<EntryData>();
  80       tokenData.typeToEntries.put(entryTypeName, entries);
  81     }
  82     return entries;
  83   }
  84
  85
  86 }