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