ComfyUI开发指南 -- 插件开发（上）

class Example:
    def __init__(self):
        pass
    
    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "image": ("IMAGE",),
                "int_field": ("INT", {
                    "default": 0, #默认值
                    "min": 0, #最小值
                    "max": 4096, #最大值
                    "step": 64, #滑动步长
                    "display": "number" # 组件外观: 数值  "number" 或  滑动条 "slider"
                }),
                "float_field": ("FLOAT", {
                    "default": 1.0,
                    "min": 0.0,
                    "max": 10.0,
                    "step": 0.01,
                    "round": 0.001, #表示舍入精度的值将默认设置为步长值。可以设置为 False 以禁用舍入。
                    "display": "number"}),
                "print_to_screen": (["enable", "disable"],),
                "string_field": ("STRING", {
                    "multiline": False, #如果您希望该字段看起来像 ClipTextEncode 节点上的字段显示（即只显示一行字符串，多余部分会在 UI上隐藏 ，只有在用户编辑时才会全部显示） 则为 True
                    "default": "Hello World!"
                }),
            },
        }

    RETURN_TYPES = ("IMAGE",)
    #RETURN_NAMES = ("image_output_name",)

    FUNCTION = "test"

    #OUTPUT_NODE = False

    CATEGORY = "Example"

    def test(self, image, string_field, int_field, float_field, print_to_screen):
        if print_to_screen == "enable":
            print(f"""Your input contains:
                string_field aka input text: {string_field}
                int_field: {int_field}
                float_field: {float_field}
            """)
        #对图像进行一些处理，在本例中只是将其反转
        image = 1.0 - image
        return (image,)

   
    #@classmethod
    #def IS_CHANGED(s, image, string_field, int_field, float_field, print_to_screen):
    #    return ""

# Set the web directory, any .js file in that directory will be loaded by the frontend as a frontend extension
# WEB_DIRECTORY = "./somejs"

# A dictionary that contains all nodes you want to export with their names
# 一个包含所有你想要导出的节点和它们对应的名称的字典， 名称需要全局唯一
# NOTE: names should be globally unique 
NODE_CLASS_MAPPINGS = {
    "Example": Example
}

# A dictionary that contains the friendly/humanly readable titles for the nodes
#  一个包含所有节点最终呈现给用户的名称和系统存储的名称对应关系的字典
NODE_DISPLAY_NAME_MAPPINGS = {
    "Example": "Example Node"
}