o · Ïiy ã@s¼UddlmZddlZddlmZddlmZddlmZm Z Gdd„de eƒZ e j de j d e jd e jd e jd e jd e jde jdiZded<Gdd„de jƒZGdd„deƒZddd„ZdS)é)Ú annotationsN)ÚEnum)Úoverride)ÚComfyExtensionÚioc@s,eZdZdZdZdZdZdZdZdZ dZ d S) Ú AspectRatioz 1:1 (Square)z 3:2 (Photo)z4:3 (Standard)z16:9 (Widescreen)z21:9 (Ultrawide)z2:3 (Portrait Photo)z3:4 (Portrait Standard)z9:16 (Portrait Widescreen)N) Ú__name__Ú __module__Ú __qualname__ÚSQUAREÚPHOTO_HÚ STANDARD_HÚ WIDESCREEN_HÚ ULTRAWIDE_HÚPHOTO_VÚ STANDARD_VÚ WIDESCREEN_V©rrú;/mnt/c/Users/fbmor/ComfyUI/comfy_extras/nodes_resolution.pyrsr)ér)éé)ér)éé )ér)rr)rr)rrz"dict[AspectRatio, tuple[int, int]]Ú ASPECT_RATIOSc@s*eZdZdZedd„ƒZed d d „ƒZd S)ÚResolutionSelectorzBCalculate width and height from aspect ratio and megapixel target.cCs\tjddddtjjdttjddtjjdd d d d d d gtjjdddtjjdddgdS)NrzResolution SelectorÚutilszwCalculate width and height from aspect ratio and megapixel target. Useful for setting up Empty Latent Image dimensions.Ú aspect_ratioz+The aspect ratio for the output dimensions.)ÚoptionsÚdefaultÚtooltipÚ megapixelsgð?gš™™™™™¹?g0@u:Target total megapixels. 1.0 MP ≈ 1024×1024 for square.)r!ÚminÚmaxÚstepr"Úwidthz+Calculated width in pixels (multiple of 8).)r"Úheightz,Calculated height in pixels (multiple of 8).)Únode_idÚ display_nameÚcategoryÚ descriptionÚinputsÚoutputs) rÚSchemaÚComboÚInputrr ÚFloatÚIntÚOutput)ÚclsrrrÚ define_schema"s8üúùÿÿüëz ResolutionSelector.define_schemarÚstrr#ÚfloatÚreturnú io.NodeOutputc Cs^t|\}}|dd}t |||¡}t||dƒd}t||dƒd}t ||¡S)Nié)rÚmathÚsqrtÚroundrÚ NodeOutput) r5rr#Úw_ratioÚh_ratioÚ total_pixelsÚscaler'r(rrrÚexecuteCs   zResolutionSelector.executeN)rr7r#r8r9r:)rr r Ú__doc__Ú classmethodr6rDrrrrrs   rc@seZdZeddd„ƒZdS)ÚResolutionExtensionr9úlist[type[io.ComfyNode]]cÃstgS©N)r)ÚselfrrrÚ get_node_listNs€ÿz!ResolutionExtension.get_node_listN)r9rH)rr r rrKrrrrrGMsrGr9cÃstƒSrI)rGrrrrÚcomfy_entrypointUs€rL)r9rG)Ú __future__rr<ÚenumrÚtyping_extensionsrÚcomfy_api.latestrrr7rr r r rrrrrrÚ__annotations__Ú ComfyNoderrGrLrrrrÚs$    ø .