70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
# encoding: utf-8
|
|
"""
|
|
"""
|
|
|
|
from inspect import cleandoc as _cleandoc
|
|
from math import sqrt as _sqrt
|
|
|
|
from comfy_api.latest import io as _io
|
|
|
|
from . import _io_simple as _i
|
|
from ._io_simple import _io_override
|
|
from . import _meta
|
|
from .__docstring_formatter import format_docstring as _format_docstring
|
|
from .funcs import simple_result_from_approx_wh as _simple_result_from_approx_wh
|
|
|
|
|
|
class BestResolutionScale(_io.ComfyNode):
|
|
"""
|
|
Scale width and height by the same value + round them.
|
|
"""
|
|
_schema = _io.Schema(
|
|
'BestResolutionScale',
|
|
display_name="Scale (Best-Res)",
|
|
category=_meta.category,
|
|
description=_format_docstring(_cleandoc(__doc__)),
|
|
inputs=[
|
|
_io_override(_i.res, 'width'),
|
|
_io_override(_i.res, 'height'),
|
|
_io_override(_i.step_default, 'step'),
|
|
_i.scale,
|
|
_io.Boolean.Input('direction', default=True, label_on='res * scale', label_off='res / scale', tooltip=(
|
|
"Sometimes it's easier to specify scale by it's inverse: how much to scale DOWN by. This toggle controls that."
|
|
)),
|
|
],
|
|
|
|
hidden=[_io.Hidden.unique_id],
|
|
|
|
outputs=[
|
|
_io.Int.Output('WIDTH', display_name='width'),
|
|
_io.Int.Output('HEIGHT', display_name='height'),
|
|
_io.Float.Output('SCALE', display_name='scale'),
|
|
],
|
|
|
|
is_output_node=True,
|
|
)
|
|
|
|
@classmethod
|
|
def define_schema(cls) -> _io.Schema:
|
|
return cls._schema
|
|
|
|
@classmethod
|
|
def execute(
|
|
cls,
|
|
width: int, height: int, step: int, scale: float, direction: bool,
|
|
# show: bool,
|
|
) -> _io.NodeOutput:
|
|
scale = float(scale)
|
|
if not direction:
|
|
scale = 1.0 / scale
|
|
width_f = scale * width
|
|
height_f = scale * height
|
|
out_width, out_height = _simple_result_from_approx_wh(
|
|
width_f, height_f, step,
|
|
# show,
|
|
unique_id=cls.hidden.unique_id,
|
|
target_square_size=_sqrt(width_f * height_f),
|
|
status_suffix=f"\n({width}/{height}) * {scale:.3f} scale"
|
|
)
|
|
return _io.NodeOutput(out_width, out_height, scale)
|