mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-11-19 03:25:41 +00:00
88 lines
2.5 KiB
Python
88 lines
2.5 KiB
Python
"""Functions for normalizing image resolutions"""
|
|
|
|
import contextlib
|
|
|
|
_NAMED_RESOLUTIONS = {
|
|
"HD": (1280, 720),
|
|
"FHD": (1920, 1080),
|
|
"HALF-FHD": (960, 540),
|
|
"2K": (2048, 1080),
|
|
"4K": (3840, 2160),
|
|
"8K": (7680, 4320),
|
|
"16K": (15360, 8640),
|
|
"VGA": (640, 480),
|
|
"SVGA": (800, 600),
|
|
"XGA": (1024, 768),
|
|
"SXGA": (1280, 1024),
|
|
"WXGA+": (1440, 900),
|
|
"HD+": (1600, 900),
|
|
"UXGA": (1600, 1200),
|
|
"WSXGA+": (1680, 1050),
|
|
"WUXGA": (1920, 1200),
|
|
"QWXGA": (2048, 1152),
|
|
"QXGA": (2048, 1536),
|
|
"UWFHD": (2560, 1080),
|
|
"QHD": (2560, 1440),
|
|
"WQXGA": (2560, 1600),
|
|
"UWQHD": (3440, 1440),
|
|
"240p": (426, 240),
|
|
"360p": (640, 360),
|
|
"480p": (854, 480),
|
|
"720p": (1280, 720),
|
|
"1080p": (1920, 1080),
|
|
"1440p": (2560, 1440),
|
|
"2160p": (3840, 2160),
|
|
"NTSC": (720, 480),
|
|
"PAL": (720, 576),
|
|
"QVGA": (320, 240),
|
|
"WVGA": (800, 480),
|
|
"FWVGA": (854, 480),
|
|
"WSVGA": (1024, 600),
|
|
"HDV": (1440, 1080),
|
|
"WQHD": (2560, 1440),
|
|
"UW-UXGA": (2560, 1080),
|
|
"UHD": (3840, 2160),
|
|
"UHD+": (5120, 2880),
|
|
"SVD": (1024, 576), # stable video diffusion
|
|
}
|
|
|
|
_NAMED_RESOLUTIONS = {k.upper(): v for k, v in _NAMED_RESOLUTIONS.items()}
|
|
|
|
|
|
def normalize_image_size(resolution: str | int | tuple[int, int]) -> tuple[int, int]:
|
|
size = _normalize_image_size(resolution)
|
|
if any(s <= 0 for s in size):
|
|
msg = f"Invalid resolution: {resolution!r}"
|
|
raise ValueError(msg)
|
|
return size
|
|
|
|
|
|
def _normalize_image_size(resolution: str | int | tuple[int, int]) -> tuple[int, int]:
|
|
match resolution:
|
|
case (int(), int()):
|
|
return resolution # type: ignore
|
|
case int():
|
|
return resolution, resolution
|
|
case str():
|
|
resolution = resolution.strip().upper()
|
|
resolution = resolution.replace(" ", "").replace("X", ",").replace("*", ",")
|
|
if resolution.upper() in _NAMED_RESOLUTIONS:
|
|
return _NAMED_RESOLUTIONS[resolution.upper()]
|
|
|
|
# is it WIDTH,HEIGHT format?
|
|
try:
|
|
width, height = resolution.split(",")
|
|
return int(width), int(height)
|
|
except ValueError:
|
|
pass
|
|
|
|
# is it just a single number?
|
|
with contextlib.suppress(ValueError):
|
|
return int(resolution), int(resolution)
|
|
|
|
msg = f"Invalid resolution: '{resolution}'"
|
|
raise ValueError(msg)
|
|
case _:
|
|
msg = f"Invalid resolution: {resolution!r}"
|
|
raise ValueError(msg)
|