corefud.MarkNested

martinpopel · martinpopel · commit 81a65bf8a192 · 2022-02-10T12:24:15.000+01:00
in future, I would like to merge all the corefud.Mark* blocks
into one universal block, but for now, let's archive this one
diff --git a/udapi/block/corefud/marknested.py b/udapi/block/corefud/marknested.py
@@ -0,0 +1,44 @@
+from udapi.core.block import Block
+import udapi.core.coref
+import itertools
+
+class MarkNested(Block):
+    """Find nested mentions."""
+
+    def __init__(self, same_cluster_only=True, both_discontinuous=False, multiword_only=False,
+                 print_form=False, log=True, mark=True, **kwargs):
+        super().__init__(**kwargs)
+        self.same_cluster_only = same_cluster_only
+        self.both_discontinuous = both_discontinuous
+        self.multiword_only = multiword_only
+        self.print_form = print_form
+        self.log = log
+        self.mark = mark
+
+    def _print(self, mention):
+        if self.print_form:
+            return mention.cluster.cluster_id + ':' + ' '.join([w.form for w in mention.words])
+        else:
+            return mention.cluster.cluster_id + ':' + mention.span
+
+    def process_tree(self, tree):
+        mentions = set()
+        for node in tree.descendants_and_empty:
+            for m in node.coref_mentions:
+                mentions.add(m)
+        for mA, mB in itertools.combinations(mentions, 2):
+            if self.same_cluster_only and mA.cluster != mB.cluster:
+                continue
+            if self.both_discontinuous and (',' not in mA.span or ',' not in mB.span):
+                continue
+            sA, sB = set(mA.words), set(mB.words)
+            if not (sA <= sB) and not (sB <= sA):
+                continue
+            if self.multiword_only and (len(sA) == 1 or len(sB) == 1):
+                continue
+            if self.mark:
+                for w in mA.words + mB.words:
+                    w.misc['Mark'] = 1
+                mA.words[0].misc['Mark'] = f"{self._print(mA)}+{self._print(mB)}"
+            if self.log:
+                print(f"nested mentions at {tree.sent_id}: {self._print(mA)} + {self._print(mB)}")
