Show filename in window title.

pixelocr/gui/ocrengine.py

@@ -44,7 +44,7 @@ class OCREngine(QThread):
         self.glyphdb.save()
 
     def load_page(self, filename):
-        return Page(Image.fromfile(filename).unframe(10))
+        return Page(Image.fromfile(filename).unframe(10), filename)
 
     def run(self):
         for page_text in self.recognize():

pixelocr/gui/window.py

@@ -38,13 +38,13 @@ from .glypheditor import GlyphDBEdit
 
 
 class MainWindow(QMainWindow):
+    BASE_TITLE = 'PixelOCR'
     dbedit = None
-
     glyphEntered = signal([str])
 
     def __init__(self, ocr):
         super().__init__()
-        self.setWindowTitle('PixelOCR')
+        self.setDocumentTitle(None)
 
         self.ocr = ocr
 
@@ -56,6 +56,7 @@ class MainWindow(QMainWindow):
         self.glyphEdit.glyphEntered.connect(self.glyphEntered)
         self.glyphEdit.glyphEntered.connect(self.pageScene.clearHighlight)
         ocr.pageChanged.connect(self.pageScene.setPage)
+        ocr.pageChanged.connect(self.showPageTitle)
         ocr.unknownGlyph.connect(self.unknownGlyph)
 
         layout = QVBoxLayout(centralWidget)
@@ -107,3 +108,14 @@ class MainWindow(QMainWindow):
         if self.dbedit is None:
             self.dbedit = GlyphDBEdit(self.ocr.glyphdb)
         self.dbedit.show()
+
+    def setDocumentTitle(self, title):
+        if title:
+            full_title = '{} – {}'.format(title, self.BASE_TITLE)
+        else:
+            full_title = self.BASE_TITLE
+        self.setWindowTitle(full_title)
+
+    def showPageTitle(self, page):
+        self.setDocumentTitle(page.filename)
+

pixelocr/page.py

@@ -29,8 +29,9 @@ CONNECTIVITY8 = ndimage.generate_binary_structure(2, 2)
 
 
 class PageObject(object):
-    def __init__(self, image):
+    def __init__(self, image, filename=None):
         self.image = image
+        self.filename = filename
 
     def _repr_png_(self):
         return self.image._repr_png_()