Class BatchImageLoopClose

flow_control.py:125–335

@VariantSupport()
class BatchImageLoopClose:
    def __init__(self):
        pass

    @classmethod
    def INPUT_TYPES(cls):
        inputs = {
            "required": {
                "flow_control": ("FLOW_CONTROL", {"rawLink": True}),
                "current_image": ("IMAGE",),
                "current_mask": ("MASK",),
                "max_iterations": ("INT", {"forceInput": True}),
            },
            "optional": {
                "pass_back": ("BOOLEAN", {"default": False}),  # 新增：控制是否传回图片
            },
            "hidden": {
                "dynprompt": "DYNPROMPT",
                "unique_id": "UNIQUE_ID",
                "result_images": ("IMAGE",),
                "result_masks": ("MASK",),
                "iteration_count": ("INT", {"default": 0}),
            }
        }
        return inputs

    RETURN_TYPES = tuple(["IMAGE", "MASK"])
    RETURN_NAMES = tuple(["result_images", "result_masks"])
    FUNCTION = "while_loop_close"
    CATEGORY = "CyberEveLoop🐰"

    def explore_dependencies(self, node_id, dynprompt, upstream, parent_ids):
        node_info = dynprompt.get_node(node_id)
        if "inputs" not in node_info:
            return

        for k, v in node_info["inputs"].items():
            if is_link(v):
                parent_id = v[0]
                display_id = dynprompt.get_display_node_id(parent_id)
                display_node = dynprompt.get_node(display_id)
                class_type = display_node["class_type"]
                # 排除循环结束节点
                if class_type not in ['BatchImageLoopClose']:
                    parent_ids.append(display_id)
                if parent_id not in upstream:
                    upstream[parent_id] = []
                    self.explore_dependencies(parent_id, dynprompt, upstream, parent_ids)
                upstream[parent_id].append(node_id)

    def explore_output_nodes(self, dynprompt, upstream, output_nodes, parent_ids):
        """探索并添加输出节点的连接"""
        for parent_id in upstream:
            display_id = dynprompt.get_display_node_id(parent_id)
            for output_id in output_nodes:
                id = output_nodes[output_id][0]
                if id in parent_ids and display_id == id and output_id not in upstream[parent_id]:
                    if '.' in parent_id: