summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorVlad Voicu <vladvoic@gmail.com>2011-11-28 16:37:32 (GMT)
committerKevin Smith <git@kismith.co.uk>2012-03-09 15:03:59 (GMT)
commitb8f10af29258705b4610e99ba69f5ecab42ef6ae (patch)
tree9f182e47e07b5cfda70079a8e1ac9a80eaa981f1 /SwifTools/SpellChecker.cpp
parent0833f7da453db9cd0fc3a78c793e7532663ab86b (diff)
downloadswift-contrib-b8f10af29258705b4610e99ba69f5ecab42ef6ae.zip
swift-contrib-b8f10af29258705b4610e99ba69f5ecab42ef6ae.tar.bz2
Initial, buggy version of a minimal spell-checker
Diffstat (limited to 'SwifTools/SpellChecker.cpp')
-rw-r--r--SwifTools/SpellChecker.cpp52
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));
+ }
+ }
+}
+
+}