1 files changed, 42 insertions, 0 deletions
diff --git a/decoder/ff_charset.cc b/decoder/ff_charset.cc
new file mode 100644
index 00000000..33afc1a7
--- /dev/null
+++ b/decoder/ff_charset.cc
@@ -0,0 +1,42 @@
+#include "ff_charset.h"
+
+#include "fdict.h"
+#include "stringlib.h"
+
+using namespace std;
+
+NonLatinCount::NonLatinCount(const string& param) : FeatureFunction(), fid_(FD::Convert("NonLatinCount")) {}
+
+bool ContainsNonLatin(const char* word) {
+  int cur = 0;
+  while(word[cur]) {
+    const int size = UTF8Len(word[cur]);
+    if (size > 1) return true;
+    cur += size;  
+  }
+  return false;
+}
+
+void NonLatinCount::TraversalFeaturesImpl(const SentenceMetadata& smeta,
+                                          const Hypergraph::Edge& edge,
+                                          const std::vector<const void*>& ant_contexts,
+                                          FeatureVector* features,
+                                          FeatureVector* estimated_features,
+                                          void* context) const {
+  const vector<WordID>& e = edge.rule_->e();
+  int count = 0;
+  for (int i = 0; i < e.size(); ++i) {
+    if (e[i] > 0) {
+      map<WordID, bool>::iterator it = is_non_latin_.find(e[i]);
+      if (it == is_non_latin_.end()) {
+        if ((is_non_latin_[e[i]] = ContainsNonLatin(TD::Convert(e[i]))))
+          ++count;
+      } else {
+        if (it->second)
+          ++count;
+      }
+    }
+  }
+  if (count) features->set_value(fid_, count);
+}
+