morfologik.stemming.DictionaryAttribute Maven / Gradle / Ivy
package morfologik.stemming;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;
/**
* Attributes applying to {@link Dictionary} and {@link DictionaryMetadata}.
*/
public enum DictionaryAttribute {
/**
* Logical fields separator inside the FSA.
*/
SEPARATOR("fsa.dict.separator") {
@Override
public Character fromString(String separator) {
if (separator == null || separator.length() != 1) {
throw new IllegalArgumentException("Attribute " + propertyName
+ " must be a single character.");
}
char charValue = separator.charAt(0);
if (Character.isHighSurrogate(charValue) ||
Character.isLowSurrogate(charValue)) {
throw new IllegalArgumentException(
"Field separator character cannot be part of a surrogate pair: " + separator);
}
return charValue;
}
},
/**
* Character to byte encoding used for strings inside the FSA.
*/
ENCODING("fsa.dict.encoding") {
@Override
public Charset fromString(String charsetName) {
return Charset.forName(charsetName);
}
},
/**
* If the dictionary was compiled with prefix compression.
*/
USES_PREFIXES("fsa.dict.uses-prefixes") {
@Override
public Boolean fromString(String value) {
return booleanValue(value);
}
},
/**
* If the dictionary was compiled with infix compression.
*/
USES_INFIXES("fsa.dict.uses-infixes") {
@Override
public Boolean fromString(String value) {
return booleanValue(value);
}
},
/**
* If the spelling dictionary is supposed to ignore words containing digits
*/
IGNORE_NUMBERS("fsa.dict.speller.ignore-numbers") {
@Override
public Boolean fromString(String value) {
return booleanValue(value);
}
},
/**
* If the spelling dictionary is supposed to ignore punctuation.
*/
IGNORE_PUNCTUATION("fsa.dict.speller.ignore-punctuation") {
@Override
public Boolean fromString(String value) {
return booleanValue(value);
}
},
/**
* If the spelling dictionary is supposed to ignore CamelCase words.
*/
IGNORE_CAMEL_CASE("fsa.dict.speller.ignore-camel-case") {
@Override
public Boolean fromString(String value) {
return booleanValue(value);
}
},
/**
* If the spelling dictionary is supposed to ignore ALL UPPERCASE words.
*/
IGNORE_ALL_UPPERCASE("fsa.dict.speller.ignore-all-uppercase") {
@Override
public Boolean fromString(String value) {
return booleanValue(value);
}
},
/**
* If the spelling dictionary is supposed to ignore diacritics, so that
* 'a' would be treated as equivalent to 'ą'.
*/
IGNORE_DIACRITICS("fsa.dict.speller.ignore-diacritics") {
@Override
public Boolean fromString(String value) {
return booleanValue(value);
}
},
/**
* if the spelling dictionary is supposed to treat upper and lower case
* as equivalent.
*/
CONVERT_CASE("fsa.dict.speller.convert-case") {
@Override
public Boolean fromString(String value) {
return booleanValue(value);
}
},
/**
* If the spelling dictionary is supposed to split runOnWords.
*/
RUN_ON_WORDS("fsa.dict.speller.runon-words") {
@Override
public Boolean fromString(String value) {
return booleanValue(value);
}
},
/** Locale associated with the dictionary. */
LOCALE("fsa.dict.speller.locale") {
@Override
public Locale fromString(String value) {
return new Locale(value);
}
},
/**
* Replacement pairs for non-obvious candidate search in a speller dictionary.
* For example, Polish rz is phonetically equivalent to ż,
* and this may be specified here to allow looking for replacements of rz with ż
* and vice versa.
*/
REPLACEMENT_PAIRS("fsa.dict.speller.replacement-pairs") {
@Override
public Map> fromString(String value) throws IllegalArgumentException {
Map> replacementPairs = new HashMap>();
final String[] replacements = value.split(",\\s*");
for (final String stringPair : replacements) {
final String[] twoStrings = stringPair.trim().split(" ");
if (twoStrings.length == 2) {
if (!replacementPairs.containsKey(twoStrings[0])) {
List strList = new ArrayList();
strList.add(twoStrings[1]);
replacementPairs.put(twoStrings[0], strList);
} else {
replacementPairs.get(twoStrings[0]).add(twoStrings[1]);
}
} else {
throw new IllegalArgumentException("Attribute " + propertyName
+ " is not in the proper format: " + value);
}
}
return replacementPairs;
}
},
/**
* Equivalent characters (treated similarly as equivalent chars with and without
* diacritics). For example, Polish ł can be specified as equivalent to l.
*
* This implements a feature similar to hunspell MAP in the affix file.
*/
EQUIVALENT_CHARS("fsa.dict.speller.equivalent-chars") {
@Override
public Map> fromString(String value) throws IllegalArgumentException {
Map> equivalentCharacters =
new HashMap>();
final String[] eqChars = value.split(",\\s*");
for (final String characterPair : eqChars) {
final String[] twoChars = characterPair.trim().split(" ");
if (twoChars.length == 2
&& twoChars[0].length() == 1
&& twoChars[1].length() == 1) {
char fromChar = twoChars[0].charAt(0);
char toChar = twoChars[1].charAt(0);
if (!equivalentCharacters.containsKey(fromChar)) {
List chList = new ArrayList();
equivalentCharacters.put(fromChar, chList);
}
equivalentCharacters.get(fromChar).add(toChar);
} else {
throw new IllegalArgumentException("Attribute " + propertyName
+ " is not in the proper format: " + value);
}
}
return equivalentCharacters;
}
},
/**
* Dictionary license attribute.
*/
LICENSE("fsa.dict.license"),
/**
* Dictionary author.
*/
AUTHOR("fsa.dict.author"),
/**
* Dictionary creation date.
*/
CREATION_DATE("fsa.dict.created");
/**
* Property name for this attribute.
*/
public final String propertyName;
/**
* Converts a string to the given attribute's value (covariants used).
*
* @throws IllegalArgumentException
* If the input string cannot be converted to the attribute's
* value.
*/
public Object fromString(String value) throws IllegalArgumentException {
return value;
}
/**
* Return an {@link DictionaryAttribute} by its {@link #propertyName}.
*/
public static DictionaryAttribute fromPropertyName(String propertyName) {
DictionaryAttribute value = attrsByPropertyName.get(propertyName);
if (value == null) {
throw new IllegalArgumentException("No attribute for property: " + propertyName);
}
return value;
}
private static final Map attrsByPropertyName;
static {
attrsByPropertyName = new HashMap();
for (DictionaryAttribute attr : DictionaryAttribute.values()) {
if (attrsByPropertyName.put(attr.propertyName, attr) != null) {
throw new RuntimeException("Duplicate property key for: " + attr);
}
}
}
/**
* Private enum instance constructor.
*/
private DictionaryAttribute(String propertyName) {
this.propertyName = propertyName;
}
private static Boolean booleanValue(String value) {
value = value.toLowerCase();
if ("true".equals(value) || "yes".equals(value) || "on".equals(value)) {
return Boolean.TRUE;
}
if ("false".equals(value) || "no".equals(value) || "off".equals(value)) {
return Boolean.FALSE;
}
throw new IllegalArgumentException("Not a boolean value: " + value);
}
}