ComfyUI-CustomNodeKit/integer_setting_node.py at master · user2318/ComfyUI-CustomNodeKit · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
import math


class IntegerSettingNode:
    """自定义整数设置节点

    功能：
    - 设置起始值 start 和步进值 step
    - 输入数值自动对齐到 start + step*n
    - 支持负整数
    """

    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {
                "start": ("INT", {
                    "default": 0,
                    "min": -99999999,
                    "max": 99999999,
                    "step": 1,
                    "display": "number",
                }),
                "step": ("INT", {
                    "default": 1,
                    "min": 1,
                    "max": 99999999,
                    "step": 1,
                    "display": "number",
                }),
                "value": ("INT", {
                    "default": 0,
                    "min": -99999999,
                    "max": 99999999,
                    "step": 1,
                    "display": "number",
                }),
            }
        }

    RETURN_TYPES = ("INT",)
    RETURN_NAMES = ("aligned_value",)
    OUTPUT_NODE = False
    FUNCTION = "apply"
    CATEGORY = "CustomNodes/Utils"

    def apply(self, start, step, value):
        aligned = self._align_to_step(start, step, value)
        return (aligned,)

    def _align_to_step(self, start, step, value):
        """将数值对齐到最接近的 start + step*n

        对齐规则：
        1. 计算 value 在等差数列 start + step*n 中的两个最近候选值
        2. 选择距离更近的候选值
        3. 如果距离相等，选择更接近 start 的值
        """
        if step <= 0:
            return start

        # 计算 n = (value - start) / step
        n = (value - start) / step

        # 取 floor 和 ceil
        n_floor = int(math.floor(n))
        n_ceil = n_floor + 1

        val_floor = start + n_floor * step
        val_ceil = start + n_ceil * step

        diff_floor = abs(value - val_floor)
        diff_ceil = abs(value - val_ceil)

        if diff_floor < diff_ceil:
            return val_floor
        elif diff_ceil < diff_floor:
            return val_ceil
        else:
            # 距离相等时，选择更接近 start 的值
            if abs(val_floor - start) <= abs(val_ceil - start):
                return val_floor
            else:
                return val_ceil


# 节点注册
NODE_CLASS_MAPPINGS = {
    "IntegerSettingNode": IntegerSettingNode,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "IntegerSettingNode": "Integer Setting (整数设置)",
}