src/com/hughes/android/dictionary/engine/NormalizeComparator.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.ibm.icu.text.Transliterator;
  18
  19 import java.util.Comparator;
  20
  21 public class NormalizeComparator implements Comparator<String> {
  22
  23     private final Transliterator normalizer;
  24     private final Comparator<Object> comparator;
  25     private final int version;
  26
  27     public NormalizeComparator(final Transliterator normalizer,
  28                                final Comparator<Object> comparator, int version) {
  29         this.normalizer = normalizer;
  30         this.comparator = comparator;
  31         this.version = version;
  32     }
  33
  34     // Handles comparison between items containing "-".
  35     // Also replaces other problematic cases like "thorn".
  36     public static int compareWithoutDash(final String a, final String b, final Comparator<Object> c, int version) {
  37         if (version < 7) return 0;
  38         String s1 = a.replace("-", "").replace("þ", "th").replace("Þ", "Th");
  39         String s2 = b.replace("-", "").replace("þ", "th").replace("Þ", "Th");
  40         return c.compare(s1, s2);
  41     }
  42
  43     @Override
  44     public int compare(final String s1, final String s2) {
  45         final String n1 = normalizer == null ? s1.toLowerCase() : normalizer.transform(s1);
  46         final String n2 = normalizer == null ? s2.toLowerCase() : normalizer.transform(s2);
  47         int cn = compareWithoutDash(n1, n2, comparator, version);
  48         if (cn != 0) {
  49             return cn;
  50         }
  51         cn = comparator.compare(n1, n2);
  52         if (cn != 0) {
  53             return cn;
  54         }
  55         return comparator.compare(s1, s2);
  56     }
  57
  58 }