Fix multiword search with incomplete words.

[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 b0efdee3b9a04d83e74c19683ed69c23a59fc299..c744334b150ea5073aa9081e71915a40de606656 100644 (file)
--- a/src/com/hughes/android/dictionary/engine/NormalizeComparator.java
+++ b/src/com/hughes/android/dictionary/engine/NormalizeComparator.java
@@ -14,30 +14,49 @@
 
 package com.hughes.android.dictionary.engine;
 
-import java.util.Comparator;
-
 import com.ibm.icu.text.Transliterator;
 
+import java.util.Comparator;
+
 public class NormalizeComparator implements Comparator<String> {
-  
-  final Transliterator normalizer;
-  final Comparator<Object> comparator;
-
-  public NormalizeComparator(final Transliterator normalizer,
-      final Comparator<Object> comparator) {
-    this.normalizer = normalizer;
-    this.comparator = comparator;
-  }
-
-  @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);
-    if (cn != 0) {
-      return cn;
+
+    private final Transliterator normalizer;
+    private final Comparator<Object> comparator;
+    private final int version;
+
+    public NormalizeComparator(final Transliterator normalizer,
+                               final Comparator<Object> comparator, int version) {
+        this.normalizer = normalizer;
+        this.comparator = comparator;
+        this.version = version;
+    }
+
+    public static String withoutDash(final String a) {
+        return a.replace("-", "").replace("þ", "th").replace("Þ", "Th");
+    }
+
+    // Handles comparison between items containing "-".
+    // Also replaces other problematic cases like "thorn".
+    public static int compareWithoutDash(final String a, final String b, final Comparator<Object> c, int version) {
+        if (version < 7) return 0;
+        String s1 = withoutDash(a);
+        String s2 = withoutDash(b);
+        return c.compare(s1, s2);
+    }
+
+    @Override
+    public int compare(final String s1, final String s2) {
+        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;
+        }
+        return comparator.compare(s1, s2);
     }
-    return comparator.compare(s1, s2);
-  }
 
 }