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 com.hughes.util.StringUtil;
  18 import com.hughes.util.raf.RAFListSerializer;
  19 import com.hughes.util.raf.RAFListSerializerSkippable;
  20 import com.hughes.util.raf.RAFSerializable;
  21 import com.ibm.icu.text.Transliterator;
  22
  23 import java.io.DataInput;
  24 import java.io.DataOutput;
  25 import java.io.IOException;
  26 import java.io.PrintStream;
  27 import java.util.ArrayList;
  28 import java.util.Collections;
  29 import java.util.List;
  30 import java.util.regex.Pattern;
  31
  32 public class PairEntry extends AbstractEntry implements RAFSerializable<PairEntry>,
  33     Comparable<PairEntry> {
  34
  35     public final List<Pair> pairs;
  36
  37     public PairEntry(final EntrySource entrySource) {
  38         super(entrySource);
  39         pairs = new ArrayList<Pair>(1);
  40     }
  41
  42     public PairEntry(final EntrySource entrySource, final String lang1, final String lang2) {
  43         this(entrySource);
  44         this.pairs.add(new Pair(lang1, lang2));
  45     }
  46
  47     public PairEntry(final Dictionary dictionary, final DataInput raf, final int index)
  48     throws IOException {
  49         super(dictionary, raf, index);
  50         final int size = dictionary.dictFileVersion >= 7 ? StringUtil.readVarInt(raf) : raf.readInt();
  51         // Use singletonList for better performance in common case
  52         if (size == 1) pairs = Collections.singletonList(new Pair(raf.readUTF(), raf.readUTF()));
  53         else
  54         {
  55             pairs = new ArrayList<PairEntry.Pair>(size);
  56             for (int i = 0; i < size; ++i) {
  57                 pairs.add(new Pair(raf.readUTF(), raf.readUTF()));
  58             }
  59         }
  60     }
  61
  62     @Override
  63     public void write(DataOutput raf) throws IOException {
  64         super.write(raf);
  65         StringUtil.writeVarInt(raf, pairs.size());
  66         for (int i = 0; i < pairs.size(); ++i) {
  67             assert pairs.get(i).lang1.length() > 0;
  68             raf.writeUTF(pairs.get(i).lang1);
  69             raf.writeUTF(pairs.get(i).lang2);
  70         }
  71     }
  72
  73     static final class Serializer implements RAFListSerializerSkippable<PairEntry> {
  74
  75         final Dictionary dictionary;
  76
  77         Serializer(Dictionary dictionary) {
  78             this.dictionary = dictionary;
  79         }
  80
  81         @Override
  82         public PairEntry read(DataInput raf, int index) throws IOException {
  83             return new PairEntry(dictionary, raf, index);
  84         }
  85
  86         @Override
  87         public void skip(DataInput raf, int index) throws IOException {
  88             final int size;
  89             if (dictionary.dictFileVersion >= 7)
  90             {
  91                 StringUtil.readVarInt(raf);
  92                 size = StringUtil.readVarInt(raf);
  93             }
  94             else
  95             {
  96                 raf.skipBytes(2);
  97                 size = raf.readInt();
  98             }
  99             for (int i = 0; i < 2*size; ++i) {
 100                 int l = raf.readUnsignedShort();
 101                 raf.skipBytes(l);
 102             }
 103         }
 104
 105         @Override
 106         public void write(DataOutput raf, PairEntry t) throws IOException {
 107             t.write(raf);
 108         }
 109     }
 110
 111     @Override
 112     public void addToDictionary(final Dictionary dictionary) {
 113         assert index == -1;
 114         dictionary.pairEntries.add(this);
 115         index = dictionary.pairEntries.size() - 1;
 116     }
 117
 118     @Override
 119     public RowBase CreateRow(int rowIndex, Index dictionaryIndex) {
 120         return new Row(this.index, rowIndex, dictionaryIndex);
 121     }
 122
 123     // --------------------------------------------------------------------
 124
 125     public static class Row extends RowBase {
 126
 127         Row(final DataInput raf, final int thisRowIndex,
 128             final Index index, int extra) throws IOException {
 129             super(raf, thisRowIndex, index, extra);
 130         }
 131
 132         Row(final int referenceIndex, final int thisRowIndex,
 133             final Index index) {
 134             super(referenceIndex, thisRowIndex, index);
 135         }
 136
 137         @Override
 138         public String toString() {
 139             return getRawText(false);
 140         }
 141
 142         public PairEntry getEntry() {
 143             return index.dict.pairEntries.get(referenceIndex);
 144         }
 145
 146         @Override
 147         public void print(PrintStream out) {
 148             final PairEntry pairEntry = getEntry();
 149             for (int i = 0; i < pairEntry.pairs.size(); ++i) {
 150                 out.print((i == 0 ? "  " : "    ") + pairEntry.pairs.get(i));
 151                 out.println();
 152             }
 153         }
 154
 155         @Override
 156         public String getRawText(boolean compact) {
 157             final PairEntry pairEntry = getEntry();
 158             return pairEntry.getRawText(compact);
 159         }
 160
 161         @Override
 162         public RowMatchType matches(final List<String> searchTokens,
 163                                     final Pattern orderedMatchPattern, final Transliterator normalizer,
 164                                     final boolean swapPairEntries) {
 165             final int side = swapPairEntries ? 1 : 0;
 166             final List<Pair> pairs = getEntry().pairs;
 167             final String[] pairSides = new String[pairs.size()];
 168             for (int i = 0; i < pairs.size(); ++i) {
 169                 pairSides[i] = normalizer.transform(pairs.get(i).get(side));
 170             }
 171             for (int i = searchTokens.size() - 1; i >= 0; --i) {
 172                 final String searchToken = searchTokens.get(i);
 173                 boolean found = false;
 174                 for (final String pairSide : pairSides) {
 175                     found |= pairSide.contains(searchToken);
 176                 }
 177                 if (!found) {
 178                     return RowMatchType.NO_MATCH;
 179                 }
 180             }
 181             for (final String pairSide : pairSides) {
 182                 if (orderedMatchPattern.matcher(pairSide).find()) {
 183                     return RowMatchType.ORDERED_MATCH;
 184                 }
 185             }
 186             return RowMatchType.BAG_OF_WORDS_MATCH;
 187         }
 188
 189         @Override
 190         public int getSideLength(boolean swapPairEntries) {
 191             int result = 0;
 192             final int side = swapPairEntries ? 1 : 0;
 193             for (final Pair pair : getEntry().pairs) {
 194                 result += pair.get(side).length();
 195             }
 196             return result;
 197         }
 198
 199     }
 200
 201     public String getRawText(final boolean compact) {
 202         if (compact) {
 203             return this.pairs.get(0).toStringTab();
 204         }
 205         final StringBuilder builder = new StringBuilder();
 206         for (int i = 0; i < this.pairs.size(); ++i) {
 207             if (i > 0) {
 208                 builder.append(" | ");
 209             }
 210             builder.append(this.pairs.get(i).lang1);
 211         }
 212         builder.append("\t");
 213         for (int i = 0; i < this.pairs.size(); ++i) {
 214             if (i > 0) {
 215                 builder.append(" | ");
 216             }
 217             builder.append(this.pairs.get(i).lang2);
 218         }
 219         return builder.toString();
 220     }
 221
 222     @Override
 223     public int compareTo(final PairEntry that) {
 224         return this.getRawText(false).compareTo(that.getRawText(false));
 225     }
 226
 227     @Override
 228     public String toString() {
 229         return getRawText(false);
 230     }
 231
 232     // -----------------------------------------------------------------------
 233
 234     public static final class Pair {
 235
 236         public final String lang1;
 237         public final String lang2;
 238
 239         public Pair(final String lang1, final String lang2) {
 240             this.lang1 = lang1;
 241             this.lang2 = lang2;
 242             assert lang1.trim().length() > 0 && lang2.trim().length() > 0 : "Empty pair!!!";
 243         }
 244
 245         public Pair(final String lang1, final String lang2, final boolean swap) {
 246             this(swap ? lang2 : lang1, swap ? lang1 : lang2);
 247         }
 248
 249         public String toString() {
 250             return lang1 + " :: " + lang2;
 251         }
 252
 253         public String toStringTab() {
 254             return lang1 + "\t" + lang2;
 255         }
 256
 257         public String get(int i) {
 258             if (i == 0) {
 259                 return lang1;
 260             } else if (i == 1) {
 261                 return lang2;
 262             }
 263             throw new IllegalArgumentException();
 264         }
 265
 266         @Override
 267         public boolean equals(Object o)
 268         {
 269             if (o == this) return true;
 270             if (!(o instanceof Pair)) return false;
 271             Pair p = (Pair)o;
 272             return p.lang1.equals(lang1) && p.lang2.equals(lang2);
 273         }
 274
 275         @Override
 276         public int hashCode()
 277         {
 278             return (lang1 + "|" + lang2).hashCode();
 279         }
 280     }
 281 }