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       int count = 0;
  37       for (final List<EntryData> entryDatas : tokenData.typeToEntries.values()) {
  38         for (final EntryData entryData : entryDatas) {
  39           if (tokenEntryDatas.add(entryData)) {
  40             rows.add(new PairEntry.Row(entryData.index(), rows.size(), index));
  41             ++count;
  42           }
  43         }
  44       }
  45       System.out.println(count + " ENTRIES FOR TOKEN " + tokenData.token);
  46     }
  47   }
  48
  49   static class TokenData {
  50     final String token;
  51
  52     final Map<EntryTypeName, List<EntryData>> typeToEntries = new EnumMap<EntryTypeName, List<EntryData>>(EntryTypeName.class);
  53
  54     TokenData(final String token) {
  55       assert token.equals(token.trim());
  56       assert token.length() > 0;
  57       this.token = token;
  58     }
  59   }
  60
  61   public TokenData getOrCreateTokenData(final String token) {
  62     TokenData tokenData = tokenToData.get(token);
  63     if (tokenData == null) {
  64       tokenData = new TokenData(token);
  65       tokenToData.put(token, tokenData);
  66     }
  67     return tokenData;
  68   }
  69
  70   public List<EntryData> getOrCreateEntries(final String token, final EntryTypeName entryTypeName) {
  71     final TokenData tokenData = getOrCreateTokenData(token);
  72     List<EntryData> entries = tokenData.typeToEntries.get(entryTypeName);
  73     if (entries == null) {
  74       entries = new ArrayList<EntryData>();
  75       tokenData.typeToEntries.put(entryTypeName, entries);
  76     }
  77     return entries;
  78   }
  79
  80
  81 }