aniani/toot
1
0
Fork 0
mirror of https://github.com/ihabunek/toot.git synced 2024-06-30 06:35:24 +00:00
Code

refac: tui.images: Optimize image class selection

Browse Source 
- Change: Execute image class selection logic once and cache the result.
This commit is contained in:
AnonymouX47 2024-06-04 08:46:35 +01:00
parent efab6cf556
commit b264927da9
1 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
toot/tui/images.py
View File

@ -11,6 +11,7 @@ try:
from PIL import Image, ImageDraw from PIL import Image, ImageDraw
_IMAGE_PIXEL_FORMATS = frozenset({'kitty', 'iterm'}) _IMAGE_PIXEL_FORMATS = frozenset({'kitty', 'iterm'})
_ImageCls = None
TuiScreen = UrwidImageScreen TuiScreen = UrwidImageScreen
disable_queries() disable_queries()
@ -23,13 +24,20 @@ try:
def get_base_image(image, image_format) -> BaseImage: def get_base_image(image, image_format) -> BaseImage:
# we don't autodetect kitty, iterm; we choose based on option switches # we don't autodetect kitty, iterm; we choose based on option switches
global _ImageCls
if not _ImageCls:
BaseImage.forced_support = True BaseImage.forced_support = True
if image_format == 'kitty': _ImageCls = (
return KittyImage(image) KittyImage
elif image_format == 'iterm': if image_format == 'kitty'
return ITerm2Image(image) else ITerm2Image
else: if image_format == 'iterm'
return BlockImage(image) else BlockImage
)
return _ImageCls(image)
def resize_image(basewidth: int, baseheight: int, img: Image.Image) -> Image.Image: def resize_image(basewidth: int, baseheight: int, img: Image.Image) -> Image.Image:
if baseheight and not basewidth: if baseheight and not basewidth: