Class BasicShapeProcessor

modules/basic_shape_processor.py:1349–1597

基本图形处理模块 处理流程： 1. 从context.elements中筛选基本图形 2. 对每个图形提取填充色和描边色 3. 生成XML片段 4. 可选：运行CV补充检测遗漏的矩形

# ======================== 基本图形处理器 ========================
class BasicShapeProcessor(BaseProcessor):
    """
    基本图形处理模块
    
    处理流程：
        1. 从context.elements中筛选基本图形
        2. 对每个图形提取填充色和描边色
        3. 生成XML片段
        4. 可选：运行CV补充检测遗漏的矩形
    """
    
    def __init__(self, config=None, enable_cv_detection: bool = True):
        """
        Args:
            config: 处理配置
            enable_cv_detection: 是否启用CV补充检测（检测SAM3遗漏的矩形）
        """
        super().__init__(config)
        self.enable_cv_detection = enable_cv_detection
    
    def process(self, context: ProcessingContext) -> ProcessingResult:
        """
        处理入口
        
        Args:
            context: 处理上下文
            
        Returns:
            ProcessingResult
        """
        self._log("开始处理基本图形")
        
        if not context.image_path or not os.path.exists(context.image_path):
            return ProcessingResult(
                success=False,
                error_message="图片路径无效"
            )
        
        cv2_image = cv2.imread(context.image_path)
        if cv2_image is None:
            return ProcessingResult(
                success=False,
                error_message="无法读取图片"
            )
        
        # 筛选基本图形
        elements_to_process = self._get_elements_to_process(context.elements)
        
        # 计算画布面积，用于判断大面积元素
        canvas_area = context.canvas_width * context.canvas_height
        
        processed_count = 0
        for elem in elements_to_process:
            try:
                self._process_element(elem, cv2_image, canvas_area)
                processed_count += 1
            except Exception as e:
                elem.processing_notes.append(f"处理失败: {str(e)}")