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

   1 // Copyright 2011 Google Inc. All Rights Reserved.
   2 //
   3 // Licensed under the Apache License, Version 2.0 (the "License");
   4 // you may not use this file except in compliance with the License.
   5 // You may obtain a copy of the License at
   6 //
   7 //     http://www.apache.org/licenses/LICENSE-2.0
   8 //
   9 // Unless required by applicable law or agreed to in writing, software
  10 // distributed under the License is distributed on an "AS IS" BASIS,
  11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 // See the License for the specific language governing permissions and
  13 // limitations under the License.
  14
  15 package com.hughes.android.dictionary.engine;
  16
  17 import java.io.IOException;
  18 import java.io.PrintStream;
  19 import java.io.RandomAccessFile;
  20 import java.util.ArrayList;
  21 import java.util.List;
  22 import java.util.regex.Pattern;
  23
  24 import com.hughes.android.dictionary.engine.HtmlEntry.Row;
  25 import com.hughes.util.raf.RAFSerializable;
  26 import com.hughes.util.raf.RAFSerializer;
  27 import com.ibm.icu.text.Transliterator;
  28
  29 public class PairEntry extends AbstractEntry implements RAFSerializable<PairEntry>, Comparable<PairEntry> {
  30
  31   public final List<Pair> pairs;
  32
  33   public PairEntry(final EntrySource entrySource) {
  34     super(entrySource);
  35     pairs = new ArrayList<Pair>(1);
  36   }
  37
  38   public PairEntry(final EntrySource entrySource, final String lang1, final String lang2) {
  39     this(entrySource);
  40     this.pairs.add(new Pair(lang1, lang2));
  41   }
  42
  43   public PairEntry(final Dictionary dictionary, final RandomAccessFile raf) throws IOException {
  44     super(dictionary, raf);
  45     final int size = raf.readInt();
  46     pairs = new ArrayList<PairEntry.Pair>(size);
  47     for (int i = 0; i < size; ++i) {
  48       pairs.add(new Pair(raf.readUTF(), raf.readUTF()));
  49     }
  50   }
  51   @Override
  52   public void write(RandomAccessFile raf) throws IOException {
  53     super.write(raf);
  54     // TODO: this could be a short.
  55     raf.writeInt(pairs.size());
  56     for (int i = 0; i < pairs.size(); ++i) {
  57       assert pairs.get(i).lang1.length() > 0;
  58       raf.writeUTF(pairs.get(i).lang1);
  59       raf.writeUTF(pairs.get(i).lang2);
  60     }
  61   }
  62
  63   static final class Serializer implements RAFSerializer<PairEntry> {
  64
  65     final Dictionary dictionary;
  66
  67     Serializer(Dictionary dictionary) {
  68       this.dictionary = dictionary;
  69     }
  70
  71     @Override
  72     public PairEntry read(RandomAccessFile raf) throws IOException {
  73       return new PairEntry(dictionary, raf);
  74     }
  75
  76     @Override
  77     public void write(RandomAccessFile raf, PairEntry t) throws IOException {
  78       t.write(raf);
  79     }
  80   };
  81
  82   @Override
  83   public int addToDictionary(final Dictionary dictionary) {
  84     dictionary.pairEntries.add(this);
  85     return dictionary.pairEntries.size() - 1;
  86   }
  87
  88   @Override
  89   public RowBase CreateRow(int entryIndex, int rowIndex, Index dictionaryIndex) {
  90     return new Row(entryIndex, rowIndex, dictionaryIndex);
  91   }
  92
  93
  94   // --------------------------------------------------------------------
  95
  96
  97   public static class Row extends RowBase {
  98
  99     Row(final RandomAccessFile raf, final int thisRowIndex,
 100         final Index index) throws IOException {
 101       super(raf, thisRowIndex, index);
 102     }
 103
 104     Row(final int referenceIndex, final int thisRowIndex,
 105         final Index index) {
 106       super(referenceIndex, thisRowIndex, index);
 107     }
 108
 109     @Override
 110     public String toString() {
 111       return getRawText(false);
 112     }
 113
 114     public PairEntry getEntry() {
 115       return index.dict.pairEntries.get(referenceIndex);
 116     }
 117
 118     @Override
 119     public void print(PrintStream out) {
 120       final PairEntry pairEntry = getEntry();
 121       for (int i = 0; i < pairEntry.pairs.size(); ++i) {
 122         out.print((i == 0 ? "  " : "    ") + pairEntry.pairs.get(i));
 123         out.println();
 124       }
 125     }
 126
 127     @Override
 128     public String getRawText(boolean compact) {
 129       final PairEntry pairEntry = getEntry();
 130       return pairEntry.getRawText(compact);
 131     }
 132
 133     @Override
 134     public RowMatchType matches(final List<String> searchTokens, final Pattern orderedMatchPattern, final Transliterator normalizer, final boolean swapPairEntries) {
 135       final int side = swapPairEntries ? 1 : 0;
 136       final List<Pair> pairs = getEntry().pairs;
 137       final String[] pairSides = new String[pairs.size()];
 138       for (int i = 0; i < pairs.size(); ++i) {
 139         pairSides[i] = normalizer.transform(pairs.get(i).get(side));
 140       }
 141       for (int i = searchTokens.size() - 1; i >= 0; --i) {
 142         final String searchToken = searchTokens.get(i);
 143         boolean found = false;
 144         for (final String pairSide : pairSides) {
 145           found |= pairSide.contains(searchToken);
 146         }
 147         if (!found) {
 148           return RowMatchType.NO_MATCH;
 149         }
 150       }
 151       for (final String pairSide : pairSides) {
 152         if (orderedMatchPattern.matcher(pairSide).find()) {
 153           return RowMatchType.ORDERED_MATCH;
 154         }
 155       }
 156       return RowMatchType.BAG_OF_WORDS_MATCH;
 157     }
 158
 159     @Override
 160     public int getSideLength(boolean swapPairEntries) {
 161       int result = 0;
 162       final int side = swapPairEntries ? 1 : 0;
 163       for (final Pair pair : getEntry().pairs) {
 164         result += pair.get(side).length();
 165       }
 166       return result;
 167     }
 168
 169   }
 170
 171   public String getRawText(final boolean compact) {
 172     if (compact) {
 173       return this.pairs.get(0).toStringTab();
 174     }
 175     final StringBuilder builder = new StringBuilder();
 176     for (int i = 0; i < this.pairs.size(); ++i) {
 177       if (i > 0) {
 178         builder.append(" | ");
 179       }
 180       builder.append(this.pairs.get(i).lang1);
 181     }
 182     builder.append("\t");
 183     for (int i = 0; i < this.pairs.size(); ++i) {
 184       if (i > 0) {
 185         builder.append(" | ");
 186       }
 187       builder.append(this.pairs.get(i).lang2);
 188     }
 189     return builder.toString();
 190   }
 191
 192   @Override
 193   public int compareTo(final PairEntry that) {
 194     return this.getRawText(false).compareTo(that.getRawText(false));
 195   }
 196
 197   @Override
 198   public String toString() {
 199     return getRawText(false);
 200   }
 201
 202   // -----------------------------------------------------------------------
 203
 204   public static final class Pair {
 205
 206     public final String lang1;
 207     public final String lang2;
 208
 209     public Pair(final String lang1, final String lang2) {
 210       this.lang1 = lang1;
 211       this.lang2 = lang2;
 212       if (!(lang1.trim().length() > 0 && lang2.trim().length() > 0)) {
 213         System.err.println("poop");
 214       }
 215       assert lang1.trim().length() > 0 || lang2.trim().length() > 0 : "Empty pair!!!";
 216       assert lang1.trim().length() > 0 && lang2.trim().length() > 0 : "Empty pair!!!";
 217     }
 218
 219     public Pair(final String lang1, final String lang2, final boolean swap) {
 220       this(swap ? lang2 : lang1, swap ? lang1 : lang2);
 221     }
 222
 223     public String toString() {
 224       return lang1 + " :: " + lang2;
 225     }
 226
 227     public String toStringTab() {
 228       return lang1 + "\t" + lang2;
 229     }
 230
 231     public String get(int i) {
 232       if (i == 0) {
 233         return lang1;
 234       } else if (i == 1) {
 235         return lang2;
 236       }
 237       throw new IllegalArgumentException();
 238     }
 239
 240   }
 241 }