Function unify_element_styles

modules/basic_shape_processor.py:660–704

统一相似大小和类型的基本图形的边框厚度。 注意：参考 sam3_extractor.py 的简化逻辑，默认边框宽度为1， 这里主要用于确保同类元素风格一致。

(elements: list)

# ======================== 样式统一 ========================
def unify_element_styles(elements: list) -> list:
    """
    统一相似大小和类型的基本图形的边框厚度。
    
    注意：参考 sam3_extractor.py 的简化逻辑，默认边框宽度为1，
    这里主要用于确保同类元素风格一致。
    """
    if not elements:
        return elements

    groups = {}
    
    for i, elem in enumerate(elements):
        shape_type = elem.get("_type", "rectangle")
        bbox = elem["bbox"]
        w = bbox[2] - bbox[0]
        h = bbox[3] - bbox[1]
        diag = math.sqrt(w**2 + h**2)
        size_key = int(round(diag / 20))
        
        key = (shape_type, size_key)
        if key not in groups:
            groups[key] = []
        groups[key].append(i)
        
    for key, indices in groups.items():
        if len(indices) < 2:
            continue
        
        # 获取边框宽度，如果不存在则默认为1
        widths = []
        for i in indices:
            style = elements[i].get("_style", {})
            widths.append(style.get("stroke_width", 1))
        
        if not widths:
            continue
        median_width = int(np.median(widths))
        
        for i in indices:
            if "_style" not in elements[i]:
                elements[i]["_style"] = {}
            elements[i]["_style"]["stroke_width"] = median_width
            
    return elements


# ======================== CV矩形检测优化辅助函数 ========================