Make features layer grepable.

danieldk · May 3, 2016 · 750306b · 750306b
1 parent 085a6fa
commit 750306b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/layer.rs b/src/layer.rs
@@ -1,4 +1,4 @@
-use conllx::Token;
+use conllx::{Features, Token};
 use std::collections::HashMap;
 
 pub type LayerCallback = fn(&Token) -> Option<&str>;
@@ -8,6 +8,7 @@ lazy_static! {
         {
             let mut m: HashMap<&'static str, LayerCallback> = HashMap::new();
             m.insert("cpos", cpos);
+            m.insert("features", features);
             m.insert("form", form);
             m.insert("headrel", head_rel);
             m.insert("pheadrel", p_head_rel);
@@ -21,6 +22,10 @@ fn cpos(t: &Token) -> Option<&str> {
     t.cpos()
 }
 
+fn features(t: &Token) -> Option<&str> {
+    t.features().as_ref().map(Features::as_str)
+}
+
 fn form(t: &Token) -> Option<&str> {
     t.form()
 }