From 6d2d870711d54d93aa946d30cce42d08252f5f41 Mon Sep 17 00:00:00 2001
From: Ali MJ Al-Nasrawy <alimjalnasrawy@gmail.com>
Date: Wed, 4 Dec 2024 18:45:46 +0300
Subject: [PATCH] Fix "Scanning contents" progress bar with --redo-ocr

---
 src/ocrmypdf/pdfinfo/layout.py | 31 +++++++++++++------------------
 1 file changed, 13 insertions(+), 18 deletions(-)

diff --git a/src/ocrmypdf/pdfinfo/layout.py b/src/ocrmypdf/pdfinfo/layout.py
index 7660ab52e..6c78f0f4f 100644
--- a/src/ocrmypdf/pdfinfo/layout.py
+++ b/src/ocrmypdf/pdfinfo/layout.py
@@ -338,6 +338,7 @@ def __init__(self, infile: Path, pscript5_mode: bool) -> None:
         self.infile = infile
         self.rman = pdfminer.pdfinterp.PDFResourceManager(caching=True)
         self.disable_boxes_flow = None
+        self.page_iter = None
         self.page_cache: list[PDFPage] = []
         self.pscript5_mode = pscript5_mode
         self.file = None
@@ -345,6 +346,7 @@ def __init__(self, infile: Path, pscript5_mode: bool) -> None:
     def __enter__(self):
         """Enter the context manager."""
         self.file = Path(self.infile).open('rb')
+        self.page_iter = PDFPage.get_pages(self.file)
         return self
 
     def __exit__(self, exc_type, exc_value, traceback):
@@ -353,27 +355,20 @@ def __exit__(self, exc_type, exc_value, traceback):
             self.file.close()
         return True
 
-    def _load_page_cache(self):
-        """Load the page cache."""
-        try:
-            self.page_cache = list(PDFPage.get_pages(self.file))
-            if not self.page_cache:
-                raise InputFileError(
-                    "pdfminer did not find any pages in the input file."
-                )
-            for n, page in enumerate(self.page_cache):
-                if page is None:
-                    raise InputFileError(
-                        f"pdfminer could not process page {n} (counting from 0)."
-                    )
-        except PDFTextExtractionNotAllowed as e:
-            raise EncryptedPdfError() from e
-
     def get_page_analysis(self, pageno: int):
         """Get the page analysis for a given page."""
-        if not self.page_cache:
-            self._load_page_cache()
+        while len(self.page_cache) <= pageno:
+            try:
+                self.page_cache.append(next(self.page_iter))
+            except StopIteration:
+                raise InputFileError(
+                    f"pdfminer did not find page {pageno} in the input file."
+                )
         page = self.page_cache[pageno]
+        if not page:
+            raise InputFileError(
+                f"pdfminer could not process page {pageno} (counting from 0)."
+            )
         dev = TextPositionTracker(
             self.rman,
             laparams=LAParams(