Method while_loop_close

flow_control.py:238–335

(self, flow_control, current_image, current_mask, max_iterations, 
                        pass_back=False, iteration_count=0, result_images=None, result_masks=None,
                        dynprompt=None, unique_id=None,)

        return result_images, result_masks

    def while_loop_close(self, flow_control, current_image, current_mask, max_iterations, 
                        pass_back=False, iteration_count=0, result_images=None, result_masks=None,
                        dynprompt=None, unique_id=None,):
        print(f"Iteration {iteration_count} of {max_iterations}")
        
        # 标准化输入，确保格式一致
        current_image, current_mask = self.standardize_input(current_image, current_mask)

        # 验证迭代计数
        if iteration_count >= max_iterations:
            raise ValueError(f"Iteration count {iteration_count} exceeds max iterations {max_iterations}")

        # 结果初始化或验证
        if result_images is None or result_masks is None:
            result_images, result_masks = self.initialize_results(max_iterations, current_image, current_mask)
        else:
            # 验证现有结果的维度和格式
            assert result_images.shape[0] == max_iterations and len(result_images.shape) == 4, \
                f"Result images must be 4D [B,H,W,C] with batch size {max_iterations}"
            assert result_masks.shape[0] == max_iterations and len(result_masks.shape) == 3, \
                f"Result masks must be 3D [B,H,W] with batch size {max_iterations}"
            
        # 存储当前结果
        result_images[iteration_count:iteration_count+1] = current_image
        result_masks[iteration_count:iteration_count+1] = current_mask
        
        # 检查是否继续循环
        if iteration_count == max_iterations - 1:
            print(f"Loop finished with {iteration_count + 1} iterations")
            return (result_images, result_masks)

        # 准备下一次循环
        this_node = dynprompt.get_node(unique_id)
        upstream = {}
        parent_ids = []
        self.explore_dependencies(unique_id, dynprompt, upstream, parent_ids)
        parent_ids = list(set(parent_ids))  # 去重

        # 获取并处理输出节点
        prompts = dynprompt.get_original_prompt()
        output_nodes = {}
        for id in prompts:
            node = prompts[id]
            if "inputs" not in node:
                continue
            class_type = node["class_type"]
            if class_type in ALL_NODE_CLASS_MAPPINGS:
                class_def = ALL_NODE_CLASS_MAPPINGS[class_type]
                if hasattr(class_def, 'OUTPUT_NODE') and class_def.OUTPUT_NODE == True:
                    for k, v in node['inputs'].items():
                        if is_link(v):
                            output_nodes[id] = v

        # 创建新图
        graph = GraphBuilder()
        self.explore_output_nodes(dynprompt, upstream, output_nodes, parent_ids)
        
        contained = {}