118 lines
4.2 KiB
Markdown
118 lines
4.2 KiB
Markdown
|
|
# 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
|
|
```python
|
|
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,)
|
|
|
|
```
|