Documentation

Class name: AdjustRect
Category: Bmad
Output node: False

AdjustRect节点旨在根据指定的修改参数和舍入模式调整矩形的尺寸和位置。它抽象了几何变换的复杂性，为各种应用场景提供了一种简化的方式来重新校准矩形坐标。

Input types

Required

a
- 表示原始矩形的一个坐标(x1或y1)，在确定新调整后矩形位置时起关键作用。
- Comfy dtype: INT
- Python dtype: int
b
- 表示原始矩形的另一个坐标(x2或y2)，有助于计算新矩形的尺寸和位置。
- Comfy dtype: INT
- Python dtype: int
c
- 作为调整矩形宽度的修改参数，直接影响调整后矩形的最终尺寸。
- Comfy dtype: INT
- Python dtype: int
d
- 作为调整矩形高度的修改参数，直接影响调整后矩形的最终尺寸。
- Comfy dtype: INT
- Python dtype: int
xm
- 指定矩形宽度调整的修改参数，影响新宽度的计算。
- Comfy dtype: INT
- Python dtype: int
ym
- 指定矩形高度调整的修改参数，影响新高度的计算。
- Comfy dtype: INT
- Python dtype: int
round_mode
- 决定在调整过程中应用的舍入模式，影响最终尺寸的计算方式。
- Comfy dtype: COMBO[STRING]
- Python dtype: str
input_format
- 定义输入坐标的格式，影响调整过程中坐标的解释和转换方式。
- Comfy dtype: COMBO[STRING]
- Python dtype: str
output_format
- 指定输出坐标的格式，决定调整后矩形的尺寸和位置如何表示。
- Comfy dtype: COMBO[STRING]
- Python dtype: str

Output types

int
- 调整后矩形的坐标，反映了根据输入参数应用的修改。它表示为四个整数的元组。
- Comfy dtype: INT
- Python dtype: Tuple[int, int, int, int]

Usage tips

Infra type: CPU
Common nodes: unknown

Source code

class AdjustRect:
    round_mode_map = {
        'Floor': math.floor,  # may be close to the image's edge, keep rect tight
        'Ceil': math.ceil,  # need the margin and image's edges aren't near
        'Round': round,  # whatever fits closest to the original rect
        'Exact': lambda v: 1  # force exact measurement
    }
    round_modes = list(round_mode_map.keys())

    @classmethod
    def INPUT_TYPES(s):
        return {"required": {
            "a": ("INT", {"min": 0, "max": np.iinfo(np.int32).max, "step": 1}),
            "b": ("INT", {"min": 0, "max": np.iinfo(np.int32).max, "step": 1}),
            "c": ("INT", {"min": 0, "max": np.iinfo(np.int32).max, "step": 1}),
            "d": ("INT", {"min": 0, "max": np.iinfo(np.int32).max, "step": 1}),
            "xm": ("INT", {"default": 64, "min": 2, "max": 1280, "step": 2}),
            "ym": ("INT", {"default": 64, "min": 2, "max": 1280, "step": 2}),
            "round_mode": (s.round_modes, {"default": s.round_modes[2]}),
            "input_format": (rect_modes, {"default": rect_modes[1]}),
            "output_format": (rect_modes, {"default": rect_modes[1]}),
        }}

    RETURN_TYPES = tuple(["INT" for x in range(4)])
    FUNCTION = "adjust"
    CATEGORY = "Bmad"

    def adjust(self, a, b, c, d, xm, ym, round_mode, input_format, output_format):
        x1, y1, x2, y2 = rect_modes_map[input_format]["toBounds"](a, b, c, d)
        center_x = (x1 + x2) // 2 + 1
        center_y = (y1 + y2) // 2 + 1
        # reasoning:
        # 00 01 02 03 04 05
        # __ -- -- -- -- __ ( original len of 4 )
        # __ x1 __ cx __ x2 ( target len of 4   )
        # most crop implementations include x1 but exclude x2;
        # thus is closer to original input
        # note that xm and ym are always even

        half_new_len_x = self.round_mode_map[round_mode]((x2 - x1) / xm) * xm // 2
        half_new_len_y = self.round_mode_map[round_mode]((y2 - y1) / ym) * ym // 2

        # note: these points can fall outside the image space
        x2 = x1 = center_x
        x2 += half_new_len_x
        x1 -= half_new_len_x
        y2 = y1 = center_y
        y2 += half_new_len_y
        y1 -= half_new_len_y

        # convert to desired output format
        x1, y1, x2, y2 = rect_modes_map[output_format]["fromBounds"](x1, y1, x2, y2)

        return (x1, y1, x2, y2,)

4.2 KiB Raw Permalink Blame History

Documentation

Input types

Required

Output types

Usage tips

Source code

4.2 KiB

Raw Permalink Blame History