IDEOGRAM_RESOLUTIONS = [
    (512, 1536), (576, 1408), (576, 1472), (576, 1536),
    (640, 1024), (640, 1344), (640, 1408), (640, 1472), (640, 1536),
    (704, 1152), (704, 1216), (704, 1280), (704, 1344), (704, 1408), (704, 1472),
    (720, 1280), (736, 1312),
    (768, 1024), (768, 1088), (768, 1152), (768, 1216), (768, 1232), (768, 1280), (768, 1344),
    (832, 960), (832, 1024), (832, 1088), (832, 1152), (832, 1216), (832, 1248),
    (864, 1152),
    (896, 960), (896, 1024), (896, 1088), (896, 1120), (896, 1152),
    (960, 832), (960, 896), (960, 1024), (960, 1088),
    (1024, 640), (1024, 768), (1024, 832), (1024, 896), (1024, 960), (1024, 1024),
    (1088, 768), (1088, 832), (1088, 896), (1088, 960),
    (1120, 896),
    (1152, 704), (1152, 768), (1152, 832), (1152, 864), (1152, 896),
    (1216, 704), (1216, 768), (1216, 832),
    (1232, 768),
    (1248, 832),
    (1280, 704), (1280, 720), (1280, 768), (1280, 800),
    (1312, 736),
    (1344, 640), (1344, 704), (1344, 768),
    (1408, 576), (1408, 640), (1408, 704),
    (1472, 576), (1472, 640), (1472, 704),
    (1536, 512), (1536, 576), (1536, 640),
]

SDXL_SD3_FLUX_RESOLUTIONS = [
    (640, 1536),
    (768, 1344),
    (832, 1216),
    (896, 1152),
    (1024, 1024),
    (1152, 896),
    (1216, 832),
    (1344, 768),
    (1536, 640),
]

LTVX_RESOLUTIONS = [
    (768, 512)
]

SD_RESOLUTIONS = [
    (512, 512),
]

COSMOS_RESOLUTIONS = [
    (960, 960),
    (960, 704),
    (704, 960),
    (1280, 704),
    # cosmos does not support portrait resolutions
]

HUNYUAN_VIDEO_RESOLUTIONS = [
    (1280, 720),
    (720, 1280),
    (1104, 832),
    (832, 1104),
    (960, 960),
    (720, 720),
    (960, 544),
    (544, 960),
    (832, 624),
    (624, 832),
    (720, 720),
]

WAN_VIDEO_14B_RESOLUTIONS = [
    (720, 1280),
    (1280, 720),
    (480, 832),
    (832, 480)
]

WAN_VIDEO_1_3B_RESOLUTIONS = [
    (480, 832),
    (832, 480)
]

WAN_VIDEO_14B_EXTENDED_RESOLUTIONS = [
    (720, 1280),
    (1280, 720),
    (960, 960),
    (1088, 832),
    (832, 1088),
    (480, 832),
    (832, 480),
    (624, 624),
    (704, 544),
    (544, 704)
]

HIDREAM_1_EDIT_RESOLUTIONS = [
    (768, 768),
]

KONTEXT_RESOLUTIONS = [
    (672, 1568),
    (688, 1504),
    (720, 1456),
    (752, 1392),
    (800, 1328),
    (832, 1248),
    (880, 1184),
    (944, 1104),
    (1024, 1024),
    (1104, 944),
    (1184, 880),
    (1248, 832),
    (1328, 800),
    (1392, 752),
    (1456, 720),
    (1504, 688),
    (1568, 672),
]

QWEN_IMAGE_RESOLUTIONS = [
    (1328, 1328),
    (1664, 928),
    (928, 1664),
    (1472, 1104),
    (1104, 1472),
    (1584, 1056),
    (1056, 1584),
]

RESOLUTION_MAP = {
    "SDXL/SD3/Flux": SDXL_SD3_FLUX_RESOLUTIONS,
    "SD1.5": SD_RESOLUTIONS,
    "LTXV": LTVX_RESOLUTIONS,
    "Ideogram": IDEOGRAM_RESOLUTIONS,
    "Cosmos": COSMOS_RESOLUTIONS,
    "HunyuanVideo": HUNYUAN_VIDEO_RESOLUTIONS,
    "WAN 14b": WAN_VIDEO_14B_RESOLUTIONS,
    "WAN 1.3b": WAN_VIDEO_1_3B_RESOLUTIONS,
    "WAN 14b with extras": WAN_VIDEO_14B_EXTENDED_RESOLUTIONS,
    "HiDream 1 Edit": HIDREAM_1_EDIT_RESOLUTIONS,
    "Kontext": KONTEXT_RESOLUTIONS,
    "Unknown": [],
    "Qwen Image": QWEN_IMAGE_RESOLUTIONS,
}

RESOLUTION_NAMES = list(RESOLUTION_MAP.keys())