Class SingleImageLoopOpen

flow_control.py:339–381

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

    @classmethod
    def INPUT_TYPES(cls):
        inputs = {
            "required": {
                "image": ("IMAGE",),
                "max_iterations": ("INT", {"default": 5, "min": 1, "max": 100}),
            },
            "optional": {
                "mask": ("MASK",),
            },
            "hidden": {
                "unique_id": "UNIQUE_ID",
                "iteration_count": ("INT", {"default": 0}),
                "previous_image": ("IMAGE",),
                "previous_mask": ("MASK",),
            }
        }
        return inputs

    RETURN_TYPES = tuple(["FLOW_CONTROL", "IMAGE", "MASK", "INT", "INT"])
    RETURN_NAMES = tuple(["FLOW_CONTROL", "current_image", "current_mask", "max_iterations", "iteration_count"])
    FUNCTION = "loop_open"
    CATEGORY = "CyberEveLoop🐰"

    def loop_open(self, image, max_iterations, mask=None, unique_id=None, 
                 iteration_count=0, previous_image=None, previous_mask=None):
        print(f"SingleImageLoopOpen Processing iteration {iteration_count}")
        
        # 确保维度正确
        if len(image.shape) == 3:
            image = image.unsqueeze(0)
        if mask is not None and len(mask.shape) == 2:
            mask = mask.unsqueeze(0)
            
        # 使用上一次循环的结果（如果有）
        current_image = previous_image if previous_image is not None and iteration_count > 0 else image
        current_mask = previous_mask if previous_mask is not None and iteration_count > 0 else mask
            
        return tuple(["stub", current_image, current_mask, max_iterations, iteration_count])

@VariantSupport()
class SingleImageLoopClose: