sk: spolocnost & stuffs

Author: starenka

src/main/java/cz/monitora/elasticsearch/analyzer/slovak/SlovakStemmer.java

@@ -98,6 +98,23 @@ private int removePrefixes(char[] s, int len) {
 
 	@SuppressWarnings({ "NPathComplexity", "CyclomaticComplexity" })
 	private int removeCase(char[] s, int len) {
+		if (len >= 9) {
+			if (endsWith(s, len, "osti") || endsWith(s, len, "ostí")) {
+				s[len - 2] = 'ť';
+				return len - 1;
+			}
+			if (endsWith(s, len, "osťou")) {
+				return len - 2;
+			}
+			if (endsWith(s, len, "osťami")) {
+				return len - 3;
+			}
+			if (endsWith(s, len, "ostiach")) {
+				s[len - 5] = 'ť';
+				return len - 4;
+			}
+		}
+
 		if (len > 7 && endsWith(s, len, "atoch")) {
 			return len - 5;
 		}

src/main/java/cz/monitora/elasticsearch/analyzer/slovak/SlovakStemmerASCIIFold.java

@@ -98,6 +98,21 @@ private int removePrefixes(char[] s, int len) {
 
 	@SuppressWarnings({ "NPathComplexity", "CyclomaticComplexity" })
 	private int removeCase(char[] s, int len) {
+		if (len >= 9) {
+			if (endsWith(s, len, "osti")) {
+				return len - 1;
+			}
+			if (endsWith(s, len, "ostou")) {
+				return len - 2;
+			}
+			if (endsWith(s, len, "ostami")) {
+				return len - 3;
+			}
+			if (endsWith(s, len, "ostiach")) {
+				return len - 4;
+			}
+		}
+
 		if (len > 7 && endsWith(s, len, "atoch")) {
 			return len - 5;
 		}

src/test/java/cz/monitora/elasticsearch/analyzer/slovak/SlovakStemmerASCIIFoldTest.java

@@ -50,6 +50,14 @@ public class SlovakStemmerASCIIFoldTest {
 			"lomnica, lomnic",
 			"lomnice, lomnic",
 			"lomniciam, lomnic",
+
+			// st stuff
+			"spolocnost, spolocnost",
+			"spolocnosti, spolocnost",
+			"spolocnostou, spolocnost",
+			"spolocnostami, spolocnost",
+			"spolocnosti, spolocnost",
+			"spolocnostiach, spolocnost",				
 	})
 	public void test_stem(String val, String exp) {
 		final SlovakStemmerASCIIFold stemmer = new SlovakStemmerASCIIFold();

src/test/java/cz/monitora/elasticsearch/analyzer/slovak/SlovakStemmerTest.java

@@ -50,6 +50,14 @@ public class SlovakStemmerTest {
 			"lomnica, lomnic",
 			"lomnice, lomnic",
 			"lomniciam, lomnic",
+
+			// sť stuff
+			"spoločnosť, spoločnosť",
+			"spoločnosti, spoločnosť",
+			"spoločnosťou, spoločnosť",
+			"spoločnosťami, spoločnosť",
+			"spoločností, spoločnosť",
+			"spoločnostiach, spoločnosť",
 	})
 	public void test_stem(String val, String exp) {
 		final SlovakStemmer stemmer = new SlovakStemmer();