tutorial.Parse template and transform.Flatten

martinpopel · martinpopel · commit 7a91696d7831 · 2017-05-09T13:01:10.000+02:00
the latter is a bit faster and shorter to write alternative for
util.Eval node='node.parent=node.root'
diff --git a/udapi/block/transform/flatten.py b/udapi/block/transform/flatten.py
@@ -0,0 +1,8 @@
+"""transform.Flatten block for flattening trees."""
+from udapi.core.block import Block
+
+class Flatten(Block):
+    """Apply `node.parent = node.root` on all nodes."""
+
+    def process_node(self, node):
+        node.parent = node.root
diff --git a/udapi/block/tutorial/parse.py b/udapi/block/tutorial/parse.py
@@ -0,0 +1,21 @@
+"""tutorial.Parse block template.
+
+Usage:
+udapy read.Conllu zone=gold files=sample.conllu \
+      read.Conllu zone=pred files=sample.conllu \
+      transform.Flatten zones=pred \
+      tutorial.Parse zones=pred \
+      eval.Parsing gold_zone=gold \
+      util.MarkDiff gold_zone=gold \
+      write.TextModeTreesHtml marked_only=1 files=parse-diff.html
+"""
+from udapi.core.block import Block
+
+class Parse(Block):
+    """Dependency parsing."""
+
+    def process_tree(self, root):
+        # TODO: Your task: implement a better heuristics than "right chain"
+        for node in root.descendants:
+            if node.next_node:
+                node.parent = node.next_node
