BigUpdate

md2gost/layout_tracker.py

@@ -0,0 +1,53 @@
+from copy import copy
+
+from docx.shared import Length
+
+
+class LayoutState:
+    def __init__(self, max_height: Length, max_width: Length):
+        self.max_height: Length = max_height
+        self.max_width: Length = max_width
+        self._current_height: Length = Length(0)
+
+    def new_page(self):
+        self._current_height += self.remaining_page_height
+
+    @property
+    def current_page_height(self):
+        return self._current_height % self.max_height
+
+    @property
+    def remaining_page_height(self) -> Length:
+        return self.max_height - self.current_page_height
+
+    @property
+    def page(self):
+        return self._current_height // self.max_height + 1
+
+    def add_height(self, height: Length):
+        self._current_height += height
+
+
+class LayoutTracker:
+    def __init__(self, max_height: Length, max_width: Length):
+        self._state = LayoutState(max_height, max_width)
+        self._is_new_page = False
+
+    @property
+    def current_state(self):
+        return copy(self._state)
+
+    @property
+    def is_new_page(self):
+        return self._is_new_page
+
+    def add_height(self, height: Length):
+        page = self._state.page
+        self._state.add_height(height)
+        self._is_new_page = self._state.page > page
+
+    def can_fit_to_page(self, height: Length):
+        return height <= self._state.remaining_page_height
+
+    def new_page(self):
+        self._state.new_page()