skills/ppt-station-skill/ppt_station/chart_builder/layout.py

"""
图表布局配置模块

控制图表的结构性属性：
- 图例位置、大小
- 横轴配置（日期轴、刻度）
- 纵轴配置（位置、刻度）
"""

from pptx.enum.chart import XL_LEGEND_POSITION, XL_TICK_MARK, XL_TICK_LABEL_POSITION
from pptx.util import Pt
from typing import Optional, Dict, Any

# ============================================================================
# 图例配置
# ============================================================================

class LegendConfig:
    """图例配置"""
    
    # 预设位置
    BOTTOM = XL_LEGEND_POSITION.BOTTOM      # 底部
    TOP = XL_LEGEND_POSITION.TOP            # 顶部
    LEFT = XL_LEGEND_POSITION.LEFT          # 左侧
    RIGHT = XL_LEGEND_POSITION.RIGHT        # 右侧
    CORNER = XL_LEGEND_POSITION.CORNER      # 右上角
    
    def __init__(
        self,
        position = XL_LEGEND_POSITION.BOTTOM,
        font_size_pt: float = 10,
        font_name: str = "黑体",
        include_in_layout: bool = False,
    ):
        """
        初始化图例配置
        
        Args:
            position: 图例位置
                - XL_LEGEND_POSITION.BOTTOM (默认)
                - XL_LEGEND_POSITION.TOP
                - XL_LEGEND_POSITION.LEFT
                - XL_LEGEND_POSITION.RIGHT
                - XL_LEGEND_POSITION.CORNER
            font_size_pt: 字体大小（pt）
            font_name: 字体名称（默认：黑体）
            include_in_layout: 是否包含在布局中（False 防止图例覆盖图表）
        """
        self.position = position
        self.font_size_pt = font_size_pt
        self.font_name = font_name
        self.include_in_layout = include_in_layout
    
    def apply_to_chart(self, chart):
        """应用到图表"""
        chart.has_legend = True
        chart.legend.position = self.position
        chart.legend.include_in_layout = self.include_in_layout
        
        if self.font_size_pt:
            chart.legend.font.size = Pt(self.font_size_pt)
        if self.font_name:
            chart.legend.font.name = self.font_name
        
        print(f"  - 图例配置: 位置={self._position_name()}, 字体={self.font_name} {self.font_size_pt}pt")
    
    def _position_name(self):
        """获取位置名称"""
        position_names = {
            XL_LEGEND_POSITION.BOTTOM: "底部",
            XL_LEGEND_POSITION.TOP: "顶部",
            XL_LEGEND_POSITION.LEFT: "左侧",
            XL_LEGEND_POSITION.RIGHT: "右侧",
            XL_LEGEND_POSITION.CORNER: "右上角",
        }
        return position_names.get(self.position, "未知")


# ============================================================================
# 横轴配置
# ============================================================================

class CategoryAxisConfig:
    """横轴（分类轴）配置"""
    
    def __init__(
        self,
        is_date_axis: bool = False,
        major_unit: Optional[float] = None,
        major_unit_days: Optional[int] = None,
        tick_label_position = XL_TICK_LABEL_POSITION.LOW,
        major_tick_mark = XL_TICK_MARK.NONE,
        minor_tick_mark = XL_TICK_MARK.NONE,
        number_format: Optional[str] = None,
        font_size_pt: float = 10,
        font_name: str = "黑体",
    ):
        """
        初始化横轴配置
        
        Args:
            is_date_axis: 是否为日期轴
            major_unit: 主刻度单位（用于日期轴，单位为天）
            major_unit_days: 主刻度单位（天）- 便捷参数，与 major_unit 相同
            tick_label_position: 刻度标签位置
            major_tick_mark: 主刻度线样式
            minor_tick_mark: 次刻度线样式
            number_format: 数字格式（如 'yyyy-mm-dd'）
            font_size_pt: 字体大小
            font_name: 字体名称（默认：黑体）
        """
        self.is_date_axis = is_date_axis
        self.major_unit = major_unit or major_unit_days
        self.tick_label_position = tick_label_position
        self.major_tick_mark = major_tick_mark
        self.minor_tick_mark = minor_tick_mark
        self.number_format = number_format
        self.font_size_pt = font_size_pt
        self.font_name = font_name
    
    def apply_to_chart(self, chart):
        """应用到图表"""
        try:
            category_axis = chart.category_axis
            
            # 设置日期轴
            if self.is_date_axis:
                from pptx.enum.chart import XL_CATEGORY_TYPE
                category_axis.category_type = XL_CATEGORY_TYPE.TIME_SCALE
                print(f"  - 横轴: 日期轴")
                
                # 设置刻度单位
                if self.major_unit:
                    from pptx.enum.chart import XL_TIME_UNIT
                    category_axis.major_unit = self.major_unit
                    category_axis.major_unit_scale = XL_TIME_UNIT.DAYS
                    print(f"  - 刻度单位: {self.major_unit} 天")
            else:
                print(f"  - 横轴: 分类轴")
            
            # 设置刻度标签位置
            category_axis.tick_label_position = self.tick_label_position
            
            # 设置刻度线
            category_axis.major_tick_mark = self.major_tick_mark
            category_axis.minor_tick_mark = self.minor_tick_mark
            
            # 设置数字格式
            if self.number_format:
                category_axis.tick_labels.number_format = self.number_format
                print(f"  - 日期格式: {self.number_format}")
            
            # 设置字体大小和字体名称
            if self.font_size_pt:
                category_axis.tick_labels.font.size = Pt(self.font_size_pt)
            if self.font_name:
                category_axis.tick_labels.font.name = self.font_name
            print(f"  - 横轴字体: {self.font_name} {self.font_size_pt}pt")
                
        except Exception as e:
            print(f"  ⚠️ 横轴配置失败: {e}")


# ============================================================================
# 纵轴配置
# ============================================================================

class ValueAxisConfig:
    """纵轴（数值轴）配置"""
    
    def __init__(
        self,
        tick_label_position = XL_TICK_LABEL_POSITION.LOW,
        major_tick_mark = XL_TICK_MARK.OUTSIDE,
        minor_tick_mark = XL_TICK_MARK.NONE,
        number_format: Optional[str] = None,
        font_size_pt: float = 10,
        font_name: str = "黑体",
        has_major_gridlines: bool = True,
        min_value: Optional[float] = None,
        max_value: Optional[float] = None,
        major_unit: Optional[float] = None,
    ):
        """
        初始化纵轴配置
        
        Args:
            tick_label_position: 刻度标签位置
            major_tick_mark: 主刻度线样式
            minor_tick_mark: 次刻度线样式
            number_format: 数字格式（如 '0.00%'）
            font_size_pt: 字体大小
            font_name: 字体名称（默认：黑体）
            has_major_gridlines: 是否显示主网格线
            min_value: 最小值（可选）
            max_value: 最大值（可选）
            major_unit: 主刻度单位（可选）
        """
        self.tick_label_position = tick_label_position
        self.major_tick_mark = major_tick_mark
        self.minor_tick_mark = minor_tick_mark
        self.number_format = number_format
        self.font_size_pt = font_size_pt
        self.font_name = font_name
        self.has_major_gridlines = has_major_gridlines
        self.min_value = min_value
        self.max_value = max_value
        self.major_unit = major_unit
    
    def apply_to_chart(self, chart):
        """应用到图表（主值轴）"""
        try:
            # ⭐ 通过 XML 直接设置主值轴，确保格式生效
            chart_element = chart._element
            val_ax_elements = chart_element.findall('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}valAx')
            
            # 主值轴是 position='l' 的那个
            primary_ax = None
            for ax in val_ax_elements:
                ax_pos = ax.find('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}axPos')
                if ax_pos is not None and ax_pos.get('val') == 'l':
                    primary_ax = ax
                    break
            
            if primary_ax is None:
                print(f"  ⚠️ 未找到主值轴")
                return
            
            from lxml import etree
            
            # 设置数字格式（通过 XML）
            if self.number_format:
                num_fmt = primary_ax.find('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}numFmt')
                if num_fmt is None:
                    # 创建 numFmt 元素
                    num_fmt = etree.Element('{http://schemas.openxmlformats.org/drawingml/2006/chart}numFmt')
                    # 在 tickLblPos 之后插入
                    tick_lbl_pos = primary_ax.find('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}tickLblPos')
                    if tick_lbl_pos is not None:
                        parent_list = list(primary_ax)
                        pos_index = parent_list.index(tick_lbl_pos)
                        primary_ax.insert(pos_index + 1, num_fmt)
                    else:
                        primary_ax.append(num_fmt)
                
                num_fmt.set('formatCode', self.number_format)
                num_fmt.set('sourceLinked', '0')
                print(f"  - 主值轴格式: {self.number_format}")
            
            # 设置字体（通过 XML）
            if self.font_size_pt or self.font_name:
                self._apply_font_to_axis_xml(primary_ax, self.font_name, self.font_size_pt)
                print(f"  - 主值轴字体: {self.font_name} {self.font_size_pt}pt")
            
            # 设置轴范围和刻度间隔（通过 XML）
            if self.min_value is not None or self.max_value is not None:
                scaling = primary_ax.find('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}scaling')
                if scaling is None:
                    scaling = etree.Element('{http://schemas.openxmlformats.org/drawingml/2006/chart}scaling')
                    # 在 axId 之后插入
                    ax_id = primary_ax.find('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}axId')
                    if ax_id is not None:
                        parent_list = list(primary_ax)
                        ax_id_index = parent_list.index(ax_id)
                        primary_ax.insert(ax_id_index + 1, scaling)
                    else:
                        primary_ax.insert(0, scaling)
                    
                    # 添加 orientation
                    orientation = etree.SubElement(scaling, '{http://schemas.openxmlformats.org/drawingml/2006/chart}orientation')
                    orientation.set('val', 'minMax')
                
                if self.min_value is not None:
                    min_elem = scaling.find('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}min')
                    if min_elem is None:
                        min_elem = etree.SubElement(scaling, '{http://schemas.openxmlformats.org/drawingml/2006/chart}min')
                    min_elem.set('val', str(self.min_value))
                    print(f"  - 主值轴最小值: {self.min_value}")
                
                if self.max_value is not None:
                    max_elem = scaling.find('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}max')
                    if max_elem is None:
                        max_elem = etree.SubElement(scaling, '{http://schemas.openxmlformats.org/drawingml/2006/chart}max')
                    max_elem.set('val', str(self.max_value))
                    print(f"  - 主值轴最大值: {self.max_value}")
            
            if self.major_unit is not None:
                major_unit_elem = primary_ax.find('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}majorUnit')
                if major_unit_elem is None:
                    major_unit_elem = etree.SubElement(primary_ax, '{http://schemas.openxmlformats.org/drawingml/2006/chart}majorUnit')
                major_unit_elem.set('val', str(self.major_unit))
                print(f"  - 主值轴刻度间隔: {self.major_unit}")
            
        except Exception as e:
            print(f"  ⚠️ 主值轴配置失败: {e}")
            import traceback
            traceback.print_exc()
    
    def _apply_font_to_axis_xml(self, axis_element, font_name: str, font_size_pt: float):
        """通过 XML 设置轴字体"""
        from lxml import etree
        
        # 查找或创建 txPr
        txPr = axis_element.find('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}txPr')
        if txPr is None:
            txPr = etree.SubElement(axis_element, '{http://schemas.openxmlformats.org/drawingml/2006/chart}txPr')
            bodyPr = etree.SubElement(txPr, '{http://schemas.openxmlformats.org/drawingml/2006/main}bodyPr')
            lstStyle = etree.SubElement(txPr, '{http://schemas.openxmlformats.org/drawingml/2006/main}lstStyle')
        
        # 设置字体
        p = txPr.find('.//{http://schemas.openxmlformats.org/drawingml/2006/main}p')
        if p is None:
            p = etree.SubElement(txPr, '{http://schemas.openxmlformats.org/drawingml/2006/main}p')
        
        pPr = p.find('.//{http://schemas.openxmlformats.org/drawingml/2006/main}pPr')
        if pPr is None:
            pPr = etree.SubElement(p, '{http://schemas.openxmlformats.org/drawingml/2006/main}pPr')
        
        defRPr = pPr.find('.//{http://schemas.openxmlformats.org/drawingml/2006/main}defRPr')
        if defRPr is None:
            defRPr = etree.SubElement(pPr, '{http://schemas.openxmlformats.org/drawingml/2006/main}defRPr')
        
        if font_size_pt:
            defRPr.set('sz', str(int(font_size_pt * 100)))
        
        if font_name:
            latin = defRPr.find('.//{http://schemas.openxmlformats.org/drawingml/2006/main}latin')
            if latin is None:
                latin = etree.SubElement(defRPr, '{http://schemas.openxmlformats.org/drawingml/2006/main}latin')
            latin.set('typeface', font_name)



# ============================================================================
# 默认配置
# ============================================================================

# 默认图例配置：底部、9pt 黑体
DEFAULT_LEGEND_CONFIG = LegendConfig(
    position=XL_LEGEND_POSITION.BOTTOM,
    font_size_pt=9,
    font_name="黑体",
    include_in_layout=False,
)

# 默认横轴配置：普通分类轴、外侧刻度线、9pt 黑体
DEFAULT_CATEGORY_AXIS_CONFIG = CategoryAxisConfig(
    is_date_axis=False,
    major_tick_mark=XL_TICK_MARK.OUTSIDE,  # ⭐ 改为 OUTSIDE（与原图一致）
    minor_tick_mark=XL_TICK_MARK.NONE,
    font_size_pt=9,
    font_name="黑体",
)

# 默认纵轴配置：无网格线、9pt 黑体
DEFAULT_VALUE_AXIS_CONFIG = ValueAxisConfig(
    tick_label_position=XL_TICK_LABEL_POSITION.LOW,
    major_tick_mark=XL_TICK_MARK.OUTSIDE,
    minor_tick_mark=XL_TICK_MARK.NONE,
    font_size_pt=9,
    font_name="黑体",
    has_major_gridlines=False,  # ⭐ 改为 False（原图无网格线）
)


# ============================================================================
# 完整布局配置
# ============================================================================

class ChartLayoutConfig:
    """完整的图表布局配置"""
    
    def __init__(
        self,
        title: Optional[str] = None,
        legend_config: Optional[LegendConfig] = None,
        category_axis_config: Optional[CategoryAxisConfig] = None,
        value_axis_config: Optional[ValueAxisConfig] = None,
        secondary_value_axis_config: Optional[ValueAxisConfig] = None,  # ⭐ 新增次值轴配置
        date_axis_config = None,  # DateAxisConfig（避免循环导入）
    ):
        """
        初始化布局配置
        
        Args:
            title: 图表标题
            legend_config: 图例配置
            category_axis_config: 横轴配置
            value_axis_config: 主值轴（左轴）配置
            secondary_value_axis_config: 次值轴（右轴）配置
            date_axis_config: 日期轴配置（DateAxisConfig，用于精确控制日期轴）
        """
        self.title = title
        self.legend_config = legend_config or DEFAULT_LEGEND_CONFIG
        self.category_axis_config = category_axis_config or DEFAULT_CATEGORY_AXIS_CONFIG
        self.value_axis_config = value_axis_config or DEFAULT_VALUE_AXIS_CONFIG
        self.secondary_value_axis_config = secondary_value_axis_config  # ⭐ 新增
        self.date_axis_config = date_axis_config  # ⭐ 新增
    
    def apply_to_chart(self, chart):
        """应用所有配置到图表"""
        print(f"\n⚙️ 应用布局配置:")
        
        # 应用标题
        if self.title:
            try:
                chart.has_title = True
                chart.chart_title.text_frame.text = self.title
                print(f"  - 标题: {self.title}")
            except Exception as e:
                print(f"  ⚠️ 标题设置失败: {e}")
        
        # 应用图例配置
        if self.legend_config:
            self.legend_config.apply_to_chart(chart)
        
        # ⭐ 应用日期轴配置（优先于普通横轴配置）
        if self.date_axis_config:
            self.date_axis_config.apply_to_chart(chart)
        # 应用横轴配置（如果没有日期轴配置）
        elif self.category_axis_config:
            self.category_axis_config.apply_to_chart(chart)
        
        # 应用主值轴配置（左轴）
        if self.value_axis_config:
            self.value_axis_config.apply_to_chart(chart)
        
        # ⭐ 应用次值轴配置（右轴）
        if self.secondary_value_axis_config:
            self._apply_secondary_axis_config(chart)
    
    def _apply_secondary_axis_config(self, chart):
        """应用次值轴配置（通过 XML）"""
        try:
            # 通过 XML 查找次值轴
            chart_element = chart._element
            val_ax_elements = chart_element.findall('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}valAx')
            
            # 次值轴通常是第二个 valAx（position='r'）
            secondary_ax = None
            for ax in val_ax_elements:
                ax_pos = ax.find('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}axPos')
                if ax_pos is not None and ax_pos.get('val') == 'r':
                    secondary_ax = ax
                    break
            
            if secondary_ax is None:
                print(f"  ⚠️ 未找到次值轴")
                return
            
            config = self.secondary_value_axis_config
            
            # 设置数字格式
            if config.number_format:
                num_fmt = secondary_ax.find('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}numFmt')
                if num_fmt is not None:
                    num_fmt.set('formatCode', config.number_format)
                    num_fmt.set('sourceLinked', '0')
                print(f"  - 次值轴格式: {config.number_format}")
            
            # 设置字体
            if config.font_size_pt or config.font_name:
                from pptx.util import Pt
                from lxml import etree
                
                # 查找或创建 txPr (文本属性)
                txPr = secondary_ax.find('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}txPr')
                if txPr is None:
                    txPr = etree.SubElement(secondary_ax, '{http://schemas.openxmlformats.org/drawingml/2006/chart}txPr')
                    bodyPr = etree.SubElement(txPr, '{http://schemas.openxmlformats.org/drawingml/2006/main}bodyPr')
                    lstStyle = etree.SubElement(txPr, '{http://schemas.openxmlformats.org/drawingml/2006/main}lstStyle')
                
                # 设置字体
                p = txPr.find('.//{http://schemas.openxmlformats.org/drawingml/2006/main}p')
                if p is None:
                    p = etree.SubElement(txPr, '{http://schemas.openxmlformats.org/drawingml/2006/main}p')
                
                pPr = p.find('.//{http://schemas.openxmlformats.org/drawingml/2006/main}pPr')
                if pPr is None:
                    pPr = etree.SubElement(p, '{http://schemas.openxmlformats.org/drawingml/2006/main}pPr')
                
                defRPr = pPr.find('.//{http://schemas.openxmlformats.org/drawingml/2006/main}defRPr')
                if defRPr is None:
                    defRPr = etree.SubElement(pPr, '{http://schemas.openxmlformats.org/drawingml/2006/main}defRPr')
                
                if config.font_size_pt:
                    defRPr.set('sz', str(int(config.font_size_pt * 100)))
                
                if config.font_name:
                    latin = defRPr.find('.//{http://schemas.openxmlformats.org/drawingml/2006/main}latin')
                    if latin is None:
                        latin = etree.SubElement(defRPr, '{http://schemas.openxmlformats.org/drawingml/2006/main}latin')
                    latin.set('typeface', config.font_name)
                
                print(f"  - 次值轴字体: {config.font_name} {config.font_size_pt}pt")
            
            # ⭐ 设置轴范围和刻度间隔（通过 XML）
            from lxml import etree
            
            if config.min_value is not None or config.max_value is not None:
                # 查找或创建 scaling 元素
                scaling = secondary_ax.find('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}scaling')
                if scaling is None:
                    scaling = etree.SubElement(secondary_ax, '{http://schemas.openxmlformats.org/drawingml/2006/chart}scaling')
                    # 添加 orientation
                    orientation = etree.SubElement(scaling, '{http://schemas.openxmlformats.org/drawingml/2006/chart}orientation')
                    orientation.set('val', 'minMax')
                
                if config.min_value is not None:
                    min_elem = scaling.find('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}min')
                    if min_elem is None:
                        min_elem = etree.SubElement(scaling, '{http://schemas.openxmlformats.org/drawingml/2006/chart}min')
                    min_elem.set('val', str(config.min_value))
                    print(f"  - 次值轴最小值: {config.min_value}")
                
                if config.max_value is not None:
                    max_elem = scaling.find('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}max')
                    if max_elem is None:
                        max_elem = etree.SubElement(scaling, '{http://schemas.openxmlformats.org/drawingml/2006/chart}max')
                    max_elem.set('val', str(config.max_value))
                    print(f"  - 次值轴最大值: {config.max_value}")
            
            if config.major_unit is not None:
                major_unit_elem = secondary_ax.find('.//{http://schemas.openxmlformats.org/drawingml/2006/chart}majorUnit')
                if major_unit_elem is None:
                    major_unit_elem = etree.SubElement(secondary_ax, '{http://schemas.openxmlformats.org/drawingml/2006/chart}majorUnit')
                major_unit_elem.set('val', str(config.major_unit))
                print(f"  - 次值轴刻度间隔: {config.major_unit}")
            
        except Exception as e:
            print(f"  ⚠️ 次值轴配置失败: {e}")
            import traceback
            traceback.print_exc()