feat(ui): add thumbnail caching

TagStudioDev · Jan 11, 2025 · f126036 · f126036
1 parent fce9785
commit f126036
Show file tree

Hide file tree

Showing 6 changed files with 204 additions and 98 deletions.
diff --git a/tagstudio/src/core/constants.py b/tagstudio/src/core/constants.py
@@ -9,6 +9,7 @@
 TS_FOLDER_NAME: str = ".TagStudio"
 BACKUP_FOLDER_NAME: str = "backups"
 COLLAGE_FOLDER_NAME: str = "collages"
+THUMB_CACHE_NAME: str = "thumbs"
 
 FONT_SAMPLE_TEXT: str = (
     """ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789!?@$%(){}[]"""

diff --git a/tagstudio/src/core/library/alchemy/library.py b/tagstudio/src/core/library/alchemy/library.py
@@ -166,7 +166,7 @@ class Library:
     def close(self):
         if self.engine:
             self.engine.dispose()
-        self.library_dir = None
+        self.library_dir: Path | None = None
         self.storage_path = None
         self.folder = None
         self.included_files = set()

diff --git a/tagstudio/src/qt/helpers/gradient.py b/tagstudio/src/qt/helpers/gradient.py
@@ -6,7 +6,7 @@
 
 
 def four_corner_gradient(
-    image: Image.Image, size: tuple[int, int], mask: Image.Image
+    image: Image.Image, size: tuple[int, int], mask: Image.Image | None = None
 ) -> Image.Image:
     if image.size != size:
         # Four-Corner Gradient Background.
@@ -29,11 +29,17 @@ def four_corner_gradient(
         )
 
         final = Image.new("RGBA", bg.size, (0, 0, 0, 0))
-        final.paste(bg, mask=mask.getchannel(0))
+        if mask:
+            final.paste(bg, mask=mask.getchannel(0))
+        else:
+            final = bg
 
     else:
         final = Image.new("RGBA", size, (0, 0, 0, 0))
-        final.paste(image, mask=mask.getchannel(0))
+        if mask:
+            final.paste(image, mask=mask.getchannel(0))
+        else:
+            final = image
 
     if final.mode != "RGBA":
         final = final.convert("RGBA")

diff --git a/tagstudio/src/qt/widgets/item_thumb.py b/tagstudio/src/qt/widgets/item_thumb.py
@@ -202,7 +202,7 @@ def __init__(
         self.thumb_layout.addWidget(self.bottom_container)
 
         self.thumb_button = ThumbButton(self.thumb_container, thumb_size)
-        self.renderer = ThumbRenderer()
+        self.renderer = ThumbRenderer(self.lib)
         self.renderer.updated.connect(
             lambda timestamp, image, size, filename, ext: (
                 self.update_thumb(timestamp, image=image),

diff --git a/tagstudio/src/qt/widgets/preview/preview_thumb.py b/tagstudio/src/qt/widgets/preview/preview_thumb.py
@@ -74,7 +74,7 @@ def __init__(self, library: Library, driver: "QtDriver"):
 
         self.preview_vid = VideoPlayer(driver)
         self.preview_vid.hide()
-        self.thumb_renderer = ThumbRenderer()
+        self.thumb_renderer = ThumbRenderer(self.lib)
         self.thumb_renderer.updated.connect(lambda ts, i, s: (self.preview_img.setIcon(i)))
         self.thumb_renderer.updated_ratio.connect(
             lambda ratio: (