diff options
Diffstat (limited to 'SwifTools/SpellChecker.cpp')
-rw-r--r-- | SwifTools/SpellChecker.cpp | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/SwifTools/SpellChecker.cpp b/SwifTools/SpellChecker.cpp new file mode 100644 index 0000000..8c9aeda --- /dev/null +++ b/SwifTools/SpellChecker.cpp @@ -0,0 +1,52 @@ +/* + * Copyright (c) 2011 Voicu Vlad + * Licensed under the Simplified BSD license. + * See Documentation/Licenses/GPLv3.txt for more information. + */ + +#include <SwifTools/SpellChecker.h> + +#include <algorithm> +#include <aspell.h> +#include <boost/algorithm/string.hpp> + + +namespace Swift { + +SpellChecker::SpellChecker() { + AspellCanHaveError* ret; + config_ = new_aspell_config(); + ret = new_aspell_speller(config_); + if (aspell_error(ret) != 0) { + //TODO(vladv): Proper log the error + //printf("Error: %s\n",aspell_error_message(ret)); + delete_aspell_can_have_error(ret); + } + speller_ = to_aspell_speller(ret); +} + +bool SpellChecker::isCorrect(const std::string& word) { + if (!word.empty()) { + int aspell_error = aspell_speller_check(speller_, word.c_str(), -1); + if (aspell_error == 1) { + return true; + } else { + //TODO(vladv): Proper log the error + } + } + return false; +} + +void SpellChecker::getSuggestions(const std::string& word, std::vector<std::string>& list) { + const AspellWordList *alist; + if (!word.empty()) { + alist = aspell_speller_suggest(speller_, word.c_str(), -1); + AspellStringEnumeration *els = aspell_word_list_elements(alist); + const char* aword; + while (((aword = aspell_string_enumeration_next(els)) != 0) && list.size() <= 5) { + list.push_back(std::string(aword)); + } + } +} + +} |