util.Mark mark_attr

Author: martinpopel

udapi/block/util/mark.py

@@ -15,7 +15,7 @@ class Mark(Block):
     udapy -TM util.Mark node='node.is_nonprojective()' < in | less -R
     """
 
-    def __init__(self, node, mark=1, add=True, **kwargs):
+    def __init__(self, node, mark=1, mark_attr="Mark", add=True, print_stats=False, **kwargs):
         """Create the Mark block object.
 
         Args:
@@ -24,17 +24,27 @@ def __init__(self, node, mark=1, add=True, **kwargs):
 
         `mark`: the node will be marked with `Mark=<mark>` in `node.misc`. Default=1.
 
+        `mark_attr`: use this MISC attribute name instead of "Mark".
+
         `add`: should we keep existing Mark|ToDo|Bug? Default=True.
         """
         super().__init__(**kwargs)
         self.mark = mark
+        self.mark_attr = mark_attr
         self.node = node
         self.add = add
+        self.print_stats = print_stats
+        self._marked = 0
 
     def process_node(self, node):
         if eval(self.node):
-            node.misc['Mark'] = self.mark
+            node.misc[self.mark_attr] = self.mark
+            self._marked += 1
         elif not self.add:
-            del node.misc['Mark']
+            del node.misc[self.mark_attr]
             del node.misc['ToDo']
             del node.misc['Bug']
+
+    def process_end(self):
+        if self.print_stats:
+            print(f'util.Mark marked {self._marked} nodes')

udapi/block/util/markdiff.py

@@ -17,6 +17,7 @@ def __init__(self, gold_zone, attributes='form,lemma,upos,xpos,deprel,feats,misc
         attributes: Which node attributes should be considered when searching for diffs?
             The tree topology, i.e. node parent is always considered.
         mark: What value should be used in `node.misc['Mark']` of the differing nodes?
+        mark_attr: use this MISC attribute name instead of "Mark".
         add: If False, node.misc attributes Mark, ToDo and Bug will be deleted before running this block,
             so that the marked_only option (e.g. via `udapy -TM`) prints only nodes marked by this block.
         print_stats: How many lines of statistics should be printed? -1 means all.