Add .edges property to Container

pdfplumber/pdf.py

@@ -8,9 +8,44 @@ from pdfminer.pdfinterp import PDFResourceManager, PDFPageInterpreter
 from pdfminer.layout import LAParams, LTChar, LTImage, LTPage
 from pdfminer.converter import PDFPageAggregator
 
+from itertools import chain
 import re
 lt_pat = re.compile(r"^LT")
 
+def rect_to_edges(rect):
+    top, bottom, left, right = [ dict(rect) for x in range(4) ]
+    top.update({
+        "object_type": "edge_top",
+        "height": 0,
+        "y0": rect["y1"],
+        "orientation": "h"
+    })
+    bottom.update({
+        "object_type": "edge_bottom",
+        "height": 0,
+        "doctop": rect["doctop"] + rect["height"],
+        "y1": rect["y0"],
+        "orientation": "h"
+    })
+    left.update({
+        "object_type": "edge_left",
+        "width": 0,
+        "x1": rect["x0"],
+        "orientation": "v"
+    })
+    right.update({
+        "object_type": "edge_right",
+        "width": 0,
+        "x0": rect["x1"],
+        "orientation": "v"
+    })
+    return [ top, bottom, left, right ]
+
+def line_to_edge(line):
+    edge = dict(line)
+    edge["is_horizontal"] = line["y0"] == line["y1"]
+    return edge
+
 class Container(object):
     @property
     def rects(self):
@@ -36,6 +71,13 @@ class Container(object):
     def annos(self):
         return self.objects.get("anno", [])
 
+    @property
+    def edges(self):
+        rect_edges_gen = (rect_to_edges(r) for r in self.rects)
+        rect_edges = list(chain(*rect_edges_gen))
+        line_edges = list(map(line_to_edge, self.lines))
+        return rect_edges + line_edges
+
 class Page(Container):
     def __init__(self, layout, initial_doctop=0):
         self.layout = layout