diff options
Diffstat (limited to 'SwifTools')
-rw-r--r-- | SwifTools/SConscript | 3 | ||||
-rw-r--r-- | SwifTools/SpellChecker.cpp | 52 | ||||
-rw-r--r-- | SwifTools/SpellChecker.h | 25 |
3 files changed, 79 insertions, 1 deletions
diff --git a/SwifTools/SConscript b/SwifTools/SConscript index 3b25269..6f49e7f 100644 --- a/SwifTools/SConscript +++ b/SwifTools/SConscript @@ -7,7 +7,7 @@ Import("env") if env["SCONS_STAGE"] == "flags" : env["SWIFTOOLS_FLAGS"] = { "LIBPATH": [Dir(".")], - "LIBS": ["SwifTools"] + "LIBS": ["SwifTools", "libaspell"] } ################################################################################ @@ -29,6 +29,7 @@ if env["SCONS_STAGE"] == "build" : "Linkify.cpp", "TabComplete.cpp", "LastLineTracker.cpp", + "SpellChecker.cpp", ] if swiftools_env.get("HAVE_SPARKLE", 0) : 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)); + } + } +} + +} diff --git a/SwifTools/SpellChecker.h b/SwifTools/SpellChecker.h new file mode 100644 index 0000000..3a9ff38 --- /dev/null +++ b/SwifTools/SpellChecker.h @@ -0,0 +1,25 @@ +/* + * Copyright (c) 2011 Voicu Vlad + * Licensed under the Simplified BSD license. + * See Documentation/Licenses/GPLv3.txt for more information. + */ + +#include <vector> +#include <boost/algorithm/string.hpp> + +#pragma once + +class AspellSpeller; +class AspellConfig; + +namespace Swift { + class SpellChecker { + public: + SpellChecker(); + bool isCorrect(const std::string& word); + void getSuggestions(const std::string& word, std::vector<std::string>& list); + private: + AspellSpeller* speller_; + AspellConfig* config_; + }; +} |