sdxl.py

from .base import BaseNode, GLOBAL_CATEGORY

MODULE_CATEGORY = f"{GLOBAL_CATEGORY}/sdxl"


class HelperNodes_SDXLCommonResolutions(BaseNode):
    @classmethod
    def INPUT_TYPES(cls) -> dict:
        return {
            "required": {
                "dimensions": ([
                    "640 x 1536 (5:12 Portrait)",
                    "768 x 1344 (4:7 Portrait)",
                    "832 x 1216 (13:19 Portrait)",
                    "896 x 1152 (7:9 Portrait)",
                    "1024 x 1024 (1:1 Square)",
                    "1152 x 896 (9:7 Landscape)",
                    "1216 x 832 (19:13 Landscape)",
                    "1344 x 768 (7:4 Landscape)",
                    "1536 x 640 (12:5 Landscape)"
                ],)
            }
        }

    RETURN_TYPES = ("INT", "INT", "STRING", "STRING",)
    RETURN_NAMES = ("width", "height", "aspect ratio", "orientation",)

    CATEGORY = MODULE_CATEGORY

    def process(self, dimensions: str) -> tuple:
        dim, asp = dimensions.split(' (', 1)
        sasp: str = asp.strip('()')
        aspect = sasp.split(' ', 1)[0]
        orient = sasp.split(' ', 1)[1]
        dims: str = dim.lower().split(' x ')
        fwidth = float(dims[0])
        fheight = float(dims[1])
        width = int(fwidth) if fwidth.is_integer() else round(fwidth, 2)
        height = int(fheight) if fheight.is_integer() else round(fheight, 2)

        return width, height, aspect, orient.lower()