Class ProcessingContext

modules/base.py:24–66

处理上下文 - 在整个处理流程中共享的数据 每个模块可以从context中读取需要的数据，也可以往里写入处理结果。 这样不同模块之间可以通过context传递数据。

@dataclass
class ProcessingContext:
    """
    处理上下文 - 在整个处理流程中共享的数据
    
    每个模块可以从context中读取需要的数据，也可以往里写入处理结果。
    这样不同模块之间可以通过context传递数据。
    """
    # === 输入信息 ===
    image_path: str                              # 原始图片路径
    canvas_width: int = 0                        # 画布宽度
    canvas_height: int = 0                       # 画布高度
    
    # === 元素数据（各模块共享） ===
    elements: List['ElementInfo'] = field(default_factory=list)
    
    # === XML片段（各子模块生成，XMLMerger合并） ===
    xml_fragments: List['XMLFragment'] = field(default_factory=list)
    
    # === 配置 ===
    config: Optional['ProcessingConfig'] = None
    output_dir: str = "./output"
    
    # === 模型实例（可共享，避免重复加载） ===
    shared_models: Dict[str, Any] = field(default_factory=dict)
    
    # === 中间结果（用于debug或二次处理） ===
    intermediate_results: Dict[str, Any] = field(default_factory=dict)
    
    def get_elements_by_type(self, element_type: str) -> List['ElementInfo']:
        """按类型获取元素"""
        return [e for e in self.elements if e.element_type == element_type]
    
    def get_elements_without_xml(self) -> List['ElementInfo']:
        """获取还没有生成XML的元素"""
        return [e for e in self.elements if not e.has_xml()]
    
    def get_elements_with_xml(self) -> List['ElementInfo']:
        """获取已经生成XML的元素"""
        return [e for e in self.elements if e.has_xml()]
    
    def add_xml_fragment(self, fragment: 'XMLFragment'):
        """添加XML片段"""
        self.xml_fragments.append(fragment)


class BaseProcessor(ABC):