NormalizeComparator: improve type, support null normalizer.

[Dictionary.git] / src / com / hughes / android / dictionary / engine / NormalizeComparator.java

diff --git a/src/com/hughes/android/dictionary/engine/NormalizeComparator.java b/src/com/hughes/android/dictionary/engine/NormalizeComparator.java
index c0fef4f282431a52d7280040c5cb4f9edb27d6d8..7803bf0413d3508486099ec20268f41f7131addc 100644 (file)
--- a/src/com/hughes/android/dictionary/engine/NormalizeComparator.java
+++ b/src/com/hughes/android/dictionary/engine/NormalizeComparator.java
@@ -21,19 +21,34 @@ import java.util.Comparator;
 public class NormalizeComparator implements Comparator<String> {
 
     final Transliterator normalizer;
-    final Comparator<Object> comparator;
+    final Comparator<String> comparator;
+    int version;
 
     public NormalizeComparator(final Transliterator normalizer,
-            final Comparator<Object> comparator) {
+                               final Comparator<String> comparator, int version) {
         this.normalizer = normalizer;
         this.comparator = comparator;
+        this.version = version;
+    }
+
+    // Handles comparison between items containing "-".
+    // Also replaces other problematic cases like "thorn".
+    public static int compareWithoutDash(final String a, final String b, final Comparator c, int version) {
+        if (version < 7) return 0;
+        String s1 = a.replace("-", "").replace("þ", "th").replace("Þ", "Th");
+        String s2 = b.replace("-", "").replace("þ", "th").replace("Þ", "Th");
+        return c.compare(s1, s2);
     }
 
     @Override
     public int compare(final String s1, final String s2) {
-        final String n1 = normalizer.transform(s1);
-        final String n2 = normalizer.transform(s2);
-        final int cn = comparator.compare(n1, n2);
+        final String n1 = normalizer == null ? s1.toLowerCase() : normalizer.transform(s1);
+        final String n2 = normalizer == null ? s2.toLowerCase() : normalizer.transform(s2);
+        int cn = compareWithoutDash(n1, n2, comparator, version);
+        if (cn != 0) {
+            return cn;
+        }
+        cn = comparator.compare(n1, n2);
         if (cn != 0) {
             return cn;
         }