package com.ruoyi.satisfaction.support;

import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.satisfaction.domain.ServiceSubtaskDetailRatioExport;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFClientAnchor;
import org.apache.poi.xssf.usermodel.XSSFDrawing;
import org.apache.poi.xssf.usermodel.XSSFPicture;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import javax.imageio.ImageIO;
import java.awt.BasicStroke;
import java.awt.Color;
import java.awt.Font;
import java.awt.FontMetrics;
import java.awt.Graphics2D;
import java.awt.RenderingHints;
import java.awt.geom.Arc2D;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

/**
 * 「问题详情占比饼图」Sheet：每道题一个饼图（带百分比标签和图例），
 * 排版参照纸质报告：每题上方是题目，左边饼图、右边图例，一页 2 列 × 3 行。
 * <p>
 * 用 Java2D 画成 PNG 再贴进 Excel，保证「百分比直接标在扇区上」的效果与报告一致
 * （POI 4.1.2 的原生图表没有数据标签 API）。
 *
 * @author smartor
 */
public class MydOptionRatioPieChartBuilder implements ExcelUtil.ChartSheetBuilder {

    /** Sheet 名称 */
    public static final String SHEET_NAME = "问题详情占比饼图";

    /** 单张图片宽高（像素） */
    private static final int IMG_W = 560;
    private static final int IMG_H = 400;

    /** 每张图占用的列数 / 行数（用于单元格锚点定位） */
    private static final int COLS_PER_IMG = 9;
    private static final int ROWS_PER_IMG = 20;

    /** 每行放几张图 */
    private static final int PER_ROW = 2;

    /** 扇形配色 */
    private static final Color[] PALETTE = {
            new Color(0x5B, 0x8F, 0xF9),
            new Color(0x61, 0xDD, 0xAA),
            new Color(0xF6, 0xBD, 0x16),
            new Color(0x72, 0x62, 0xFD),
            new Color(0x78, 0xD3, 0xF8),
            new Color(0x96, 0x61, 0xBC),
            new Color(0xFF, 0x98, 0x4D),
            new Color(0xFF, 0x6D, 0x6D)
    };

    private final List<ServiceSubtaskDetailRatioExport> rows;

    private final String footer;

    public MydOptionRatioPieChartBuilder(List<ServiceSubtaskDetailRatioExport> rows, String footer) {
        this.rows = rows == null ? new ArrayList<>() : rows;
        this.footer = footer;
    }

    @Override
    public void build(XSSFWorkbook workbook) {
        if (rows.isEmpty()) {
            return;
        }
        XSSFSheet sheet = workbook.createSheet(SHEET_NAME);
        sheet.setColumnWidth(0, 12 * 256);
        for (int i = 1; i < PER_ROW * COLS_PER_IMG; i++) {
            sheet.setColumnWidth(i, 10 * 256);
        }

        XSSFDrawing drawing = sheet.createDrawingPatriarch();
        int slot = 0;
        for (Map.Entry<String, List<ServiceSubtaskDetailRatioExport>> taskEntry : groupByTask().entrySet()) {
            // 每个任务一行标题
            int titleRow = (slot / PER_ROW) * ROWS_PER_IMG;
            org.apache.poi.ss.usermodel.Row tr = sheet.getRow(titleRow) == null
                    ? sheet.createRow(titleRow) : sheet.getRow(titleRow);
            tr.createCell(0).setCellValue("任务名称：" + taskEntry.getKey());

            for (Map.Entry<String, LinkedHashMap<String, Double>> q : groupByQuestion(taskEntry.getValue()).entrySet()) {
                int col = (slot % PER_ROW) * COLS_PER_IMG;
                int row = (slot / PER_ROW) * ROWS_PER_IMG + 1;
                try {
                    byte[] png = renderPie(q.getKey(), q.getValue());
                    int picIndex = workbook.addPicture(png, Workbook.PICTURE_TYPE_PNG);
                    XSSFClientAnchor anchor = new XSSFClientAnchor(0, 0, 0, 0,
                            col, row, col + COLS_PER_IMG - 1, row + ROWS_PER_IMG - 2);
                    XSSFPicture pic = drawing.createPicture(anchor, picIndex);
                    // 尺寸完全由锚点决定（不要 resize()，否则会按 POI 默认列宽重算）
                } catch (Exception e) {
                    org.apache.poi.ss.usermodel.Row errRow = sheet.getRow(row) == null
                            ? sheet.createRow(row) : sheet.getRow(row);
                    errRow.createCell(col).setCellValue(q.getKey() + "：饼图生成失败");
                }
                slot++;
            }
            slot = ((slot + PER_ROW - 1) / PER_ROW) * PER_ROW;
        }

        // 页脚：医院--科室
        int footerRow = ((slot + PER_ROW - 1) / PER_ROW) * ROWS_PER_IMG + 1;
        org.apache.poi.ss.usermodel.Row fr = sheet.createRow(footerRow);
        fr.createCell(0).setCellValue(resolveFooter());
    }

    /**
     * 按任务名称分组，保持原有顺序
     */
    private Map<String, List<ServiceSubtaskDetailRatioExport>> groupByTask() {
        Map<String, List<ServiceSubtaskDetailRatioExport>> grouped = new LinkedHashMap<>();
        for (ServiceSubtaskDetailRatioExport row : rows) {
            String taskName = row.getTaskname() == null ? "" : row.getTaskname();
            grouped.computeIfAbsent(taskName, k -> new ArrayList<>()).add(row);
        }
        return grouped;
    }

    /**
     * 题目 -> (选项 -> 人数)，保持出现顺序
     */
    private Map<String, LinkedHashMap<String, Double>> groupByQuestion(List<ServiceSubtaskDetailRatioExport> taskRows) {
        Map<String, LinkedHashMap<String, Double>> result = new LinkedHashMap<>();
        for (ServiceSubtaskDetailRatioExport row : taskRows) {
            String question = row.getQuestiontext() == null ? "" : row.getQuestiontext();
            String option = row.getOptionresult() == null ? "" : row.getOptionresult();
            result.computeIfAbsent(question, k -> new LinkedHashMap<>())
                    .put(option, parseCount(row.getCount()));
        }
        return result;
    }

    /**
     * 页脚：优先用传入值，否则取第一行数据里的科室名称
     */
    private String resolveFooter() {
        if (footer != null && !footer.trim().isEmpty()) {
            return footer;
        }
        for (ServiceSubtaskDetailRatioExport row : rows) {
            if (row.getDeptname() != null && !row.getDeptname().trim().isEmpty()) {
                return row.getDeptname();
            }
            if (row.getLeavehospitaldistrictname() != null && !row.getLeavehospitaldistrictname().trim().isEmpty()) {
                return row.getLeavehospitaldistrictname();
            }
        }
        return "";
    }

    /**
     * 画一张饼图：标题 + 饼（扇区内标百分比） + 右侧图例（选项 + 百分比）
     */
    private byte[] renderPie(String question, LinkedHashMap<String, Double> options) throws Exception {
        BufferedImage img = new BufferedImage(IMG_W, IMG_H, BufferedImage.TYPE_INT_RGB);
        Graphics2D g = img.createGraphics();
        g.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
        g.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON);
        g.setColor(Color.WHITE);
        g.fillRect(0, 0, IMG_W, IMG_H);

        // 标题（最多两行）
        g.setFont(new Font("Microsoft YaHei", Font.BOLD, 18));
        g.setColor(new Color(0x33, 0x33, 0x33));
        drawWrapped(g, question, 16, 28, IMG_W - 32);

        double total = 0D;
        for (Double v : options.values()) {
            total += v == null ? 0D : v;
        }

        int cx = 165;
        int cy = 235;
        int r = 120;

        if (total <= 0) {
            g.setFont(new Font("Microsoft YaHei", Font.PLAIN, 16));
            g.setColor(new Color(0x99, 0x99, 0x99));
            g.drawString("暂无作答数据", cx - 45, cy);
        } else {
            double startAngle = -90D;
            int i = 0;
            for (Map.Entry<String, Double> entry : options.entrySet()) {
                double count = entry.getValue() == null ? 0D : entry.getValue();
                double angle = count / total * 360D;
                g.setColor(PALETTE[i % PALETTE.length]);
                g.fill(new Arc2D.Double(cx - r, cy - r, r * 2D, r * 2D, startAngle, angle, Arc2D.PIE));
                g.setColor(Color.WHITE);
                g.setStroke(new BasicStroke(2f));
                g.draw(new Arc2D.Double(cx - r, cy - r, r * 2D, r * 2D, startAngle, angle, Arc2D.PIE));

                // 扇区中间标百分比
                String pct = formatPercent(count, total);
                double mid = Math.toRadians(startAngle + angle / 2D);
                int tx = (int) (cx + Math.cos(mid) * r * 0.58);
                int ty = (int) (cy + Math.sin(mid) * r * 0.58);
                g.setFont(new Font("Microsoft YaHei", Font.BOLD, 15));
                g.setColor(Color.WHITE);
                drawCentered(g, pct, tx, ty);

                startAngle += angle;
                i++;
            }
        }

        // 右侧图例
        g.setFont(new Font("Microsoft YaHei", Font.PLAIN, 14));
        FontMetrics fm = g.getFontMetrics();
        int lx = cx + r + 42;
        int ly = cy - r + 30;
        int i = 0;
        for (Map.Entry<String, Double> entry : options.entrySet()) {
            g.setColor(PALETTE[i % PALETTE.length]);
            g.fillRect(lx, ly - 10, 12, 12);
            g.setColor(new Color(0x33, 0x33, 0x33));
            double count = entry.getValue() == null ? 0D : entry.getValue();
            String text = entry.getKey() + " " + formatPercent(count, total);
            g.drawString(text, lx + 18, ly);
            ly += Math.max(22, fm.getHeight() + 4);
            i++;
        }

        g.dispose();
        ByteArrayOutputStream bos = new ByteArrayOutputStream();
        ImageIO.write(img, "png", bos);
        return bos.toByteArray();
    }

    /**
     * 百分比：总量为 0 时显示 0.00%
     */
    private String formatPercent(double count, double total) {
        if (total <= 0) {
            return "0.00%";
        }
        return String.format("%.2f%%", count * 100D / total);
    }

    /**
     * 单行居中绘制文本
     */
    private void drawCentered(Graphics2D g, String text, int cx, int cy) {
        FontMetrics fm = g.getFontMetrics();
        g.drawString(text, cx - fm.stringWidth(text) / 2, cy + fm.getAscent() / 2 - 2);
    }

    /**
     * 按宽度自动换行绘制标题
     */
    private void drawWrapped(Graphics2D g, String text, int x, int y, int maxWidth) {
        if (text == null) {
            return;
        }
        FontMetrics fm = g.getFontMetrics();
        StringBuilder line = new StringBuilder();
        int lineY = y;
        for (int i = 0; i < text.length(); i++) {
            char ch = text.charAt(i);
            if (fm.stringWidth(line.toString() + ch) > maxWidth) {
                g.drawString(line.toString(), x, lineY);
                lineY += fm.getHeight();
                line.setLength(0);
            }
            line.append(ch);
        }
        if (line.length() > 0) {
            g.drawString(line.toString(), x, lineY);
        }
    }

    /**
     * 数量字符串 -> 数字
     */
    private double parseCount(String count) {
        if (count == null || count.trim().isEmpty()) {
            return 0D;
        }
        try {
            return Double.parseDouble(count.trim());
        } catch (NumberFormatException e) {
            return 0D;
        }
    }
}