Add text mark/span grouper method

2 files changed, 37 insertions, 0 deletions

diff --git a/epy.py b/epy.py
index cbe172d..e1ed035 100755
--- a/epy.py
+++ b/epy.py
@@ -914,6 +914,19 @@ class HTMLtoLines(HTMLParser):
 
         return spans
 
+    @staticmethod
+    def _group_span_by_row(
+        blocks: Sequence[Union[TextMark, TextSpan]]
+    ) -> Mapping[int, List[Union[TextMark, TextSpan]]]:
+        groups: Dict[int, List[Union[TextMark, TextSpan]]] = {}
+        for block in blocks:
+            row = block.start.row
+            if row in groups:
+                groups[row].append(block)
+            else:
+                groups[row] = [block]
+        return groups
+
     def __init__(self, sects={""}):
         HTMLParser.__init__(self)
         self.text = [""]
diff --git a/tests.py b/tests.py
index c958a36..d960e01 100644
--- a/tests.py
+++ b/tests.py
@@ -102,3 +102,27 @@ def test_span_adjustment():
     # assert HTMLtoLines._adjust_wrapped_spans(
     #     text, TextSpan(start=CharPos(row=1, col=7), n_letters=20)
     # ) == [TextSpan(start=CharPos(row=0, col=14), n_letters=3), TextSpan(start=CharPos(row=1, col=0), n_letters=4)]
+
+
+def test_group_blocks():
+    block_list = [
+        TextSpan(start=CharPos(row=0, col=0), n_letters=4),
+        TextSpan(start=CharPos(row=1, col=0), n_letters=4),
+        TextSpan(start=CharPos(row=3, col=0), n_letters=4),
+        TextSpan(start=CharPos(row=3, col=0), n_letters=4),
+        TextSpan(start=CharPos(row=15, col=0), n_letters=4),
+        TextSpan(start=CharPos(row=15, col=0), n_letters=4),
+    ]
+
+    assert HTMLtoLines._group_span_by_row(block_list) == {
+        0: [TextSpan(start=CharPos(row=0, col=0), n_letters=4)],
+        1: [TextSpan(start=CharPos(row=1, col=0), n_letters=4)],
+        3: [
+            TextSpan(start=CharPos(row=3, col=0), n_letters=4),
+            TextSpan(start=CharPos(row=3, col=0), n_letters=4),
+        ],
+        15: [
+            TextSpan(start=CharPos(row=15, col=0), n_letters=4),
+            TextSpan(start=CharPos(row=15, col=0), n_letters=4),
+        ],
+    }