import matplotlib.pyplot as plt
import pandas as pd

# 设置matplotlib支持中文显示
plt.rcParams["font.family"] = ["SimHei"]# 设置matplotlib支持中文显示


class BasicVisualizer:
    @staticmethod
    def save_to_excel(counter, filename="弹幕统计.xlsx"):
        """保存弹幕统计结果到Excel文件"""
        # 将Counter中的数据转换为DataFrame，取出现次数最多的弹幕（默认全部）
        # 列名分别为"弹幕内容"和"出现次数"
        df = pd.DataFrame(counter.most_common(), columns=["弹幕内容", "出现次数"])
        # 保存DataFrame到Excel文件，不包含索引列
        df.to_excel(filename, index=False)
        print(f"Excel保存完成: {filename}")
        return df

    @staticmethod
    def plot_top8(ai_counter, filename="前8弹幕.png"):
        """生成出现次数排名前8的AI相关弹幕柱状图并保存"""
        # 获取出现次数最多的前8条AI相关弹幕
        top8 = ai_counter.most_common(8)
        # 若没有数据，则提示并返回
        if not top8:
            print("无数据生成前8弹幕图")
            return

        # 提取弹幕内容和对应的出现次数
        texts = [item[0] for item in top8]
        counts = [item[1] for item in top8]

        # 创建画布，设置大小为12x6
        plt.figure(figsize=(12, 6))
        # 绘制水平柱状图，使用自定义颜色
        bars = plt.barh(texts, counts, color=["#D16721","#E40DEB", '#2196F3', '#f44336', '#FFC107', '#5E06EB',"#52F321", "#36f4d4"])
        # 设置x轴标签
        plt.xlabel("出现次数")
        # 设置图表标题
        plt.title("AI相关弹幕Top8")
        # 逆序排列y轴（使出现次数最多的弹幕在最上方）
        plt.gca().invert_yaxis()

        # 为每个柱子添加数值标签
        for bar in bars:
            width = bar.get_width()  # 获取柱子宽度（即出现次数）
            # 在柱子右侧添加数值标签，垂直居中对齐
            plt.text(width + 0, bar.get_y() + bar.get_height()/2, f"{width}", va="center")

        # 自动调整布局，避免元素重叠
        plt.tight_layout()
        # 保存图表到文件，设置dpi为300以保证清晰度
        plt.savefig(filename, dpi=300)
        # 显示图表
        plt.show()
        print(f"前8弹幕图保存: {filename}")