package com.gllic.workweixin.utils;

import com.gllic.workweixin.mqdto.Renderer;
import com.sun.javafx.charts.Legend;
import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartUtilities;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.CategoryAxis;
import org.jfree.chart.axis.ValueAxis;
import org.jfree.chart.block.BlockBorder;
import org.jfree.chart.labels.ItemLabelAnchor;
import org.jfree.chart.labels.ItemLabelPosition;
import org.jfree.chart.labels.StandardCategoryItemLabelGenerator;
import org.jfree.chart.plot.CategoryPlot;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.renderer.category.BarPainter;
import org.jfree.chart.renderer.category.BarRenderer;
import org.jfree.chart.renderer.category.StandardBarPainter;
import org.jfree.chart.renderer.xy.StandardXYBarPainter;
import org.jfree.data.category.CategoryDataset;
import org.jfree.data.general.DatasetUtilities;
import org.jfree.ui.TextAnchor;

import java.awt.*;
import java.io.*;
import java.util.UUID;

/**
 * @program: workweixin-mobile-api
 * @ClassName JFreeChartUtil
 * @description:
 * @author: Marlo
 * @create: 2021-06-23 17:23
 * @Version 1.0
 **/
public class JFreeChartUtil {
    private static final String CHART_PATH = "C:/myselefzxq/image/img/";


    /**
     * 生成柱状图
     */
    public static String makeBarChart1() {
        double[][] data = new double[][]{{5.00, 4.67, 3.65, 3.11, 2.87, 2.58, 2.58, 2.1, 1.89, 1.65}};
        String[] rowKeys = {"当日预收标保：万元"};
        String[] columnKeys = {"无锡滨湖-董瑞芳", "无锡新城-赵彩霞", "无锡滨湖-张邦伦", "无锡滨湖-李馨", "无锡滨湖-李莺霞", "盐城-丁立桂", "徐州-郑某", "无锡滨湖-朱玉珍", "泰州-王巧凤", "无锡城区-秦志芳"};
        //初始化柱体颜色
        String[] colorValues = {"#FE5F63", "#4AD07C", "#FDCB54", "#FDCB54", "#FDCB54", "#FDCB54", "#FDCB54", "#FDCB54", "#FDCB54", "#FDCB54"};
        CategoryDataset dataset = getBarData(data, rowKeys, columnKeys);
        String imageName = createBarChart(dataset, "", "", "个人业绩十强排名(2021/6/23)", colorValues);
        return imageName;
    }
    public static String makeBarChart2() {
        double[][] data = new double[][]{{9.52, 5.00, 3.21, 3.00, 2.78, 2.60, 2.41, 2.20, 1.89, 1.62}};
        String[] rowKeys = {"当日预收标保：万元"};
        String[] columnKeys = {"无锡滨湖-董瑞芳", "无锡新城-赵彩霞", "无锡滨湖-张邦伦", "无锡滨湖-李馨", "无锡滨湖-李莺霞", "盐城-丁立桂", "徐州-郑某", "无锡滨湖-朱玉珍", "泰州-王巧凤", "无锡城区-秦志芳"};
        //初始化柱体颜色
        String[] colorValues = {"#2197FF", "#7CCC48", "#21C3CA", "#21C3CA", "#21C3CA", "#21C3CA", "#21C3CA", "#21C3CA", "#21C3CA", "#21C3CA"};
        CategoryDataset dataset = getBarData(data, rowKeys, columnKeys);
        String imageName = createBarChart(dataset, "", "", "营业组十强排名(2021/6/23)", colorValues);
        return imageName;
    }

    /**
     * 柱状图
     *
     * @param dataset    数据集
     * @param xName      x轴的说明（如种类，时间等）
     * @param yName      y轴的说明（如速度，时间等）
     * @param chartTitle 图标题
     * @return
     */
    public static String createBarChart(CategoryDataset dataset, String xName,
                                        String yName, String chartTitle, String[] colorValues) {
        String chartName = UUID.randomUUID().toString();
        JFreeChart chart = ChartFactory.createBarChart(chartTitle, // 图表标题
                xName, // 目录轴的显示标签
                yName, // 数值轴的显示标签
                dataset, // 数据集
                PlotOrientation.HORIZONTAL, // 图表方向：水平、垂直
                true, // 是否显示图例(对于简单的柱状图必须是false),(true时，x轴legend下柱子的标题)
                false, // 是否生成工具
                false // 是否生成URL链接
        );
        //y轴数据样式
        Font yfont = new Font("宋体", Font.PLAIN, 24);
        //柱体数值的样式
        Font itemLabelFont = new Font("宋体", Font.PLAIN, 20);
        //图例数据样式
        Font legendFont = new Font("宋体", Font.PLAIN, 24);
        //Font labelFont = new Font("SansSerif", Font.TRUETYPE_FONT, 12);
        Font labelFont = new Font("宋体", Font.BOLD, 28);
        //设置整个图片的标题字体
        chart.getTitle().setFont(labelFont);
        //设置x轴上代表柱子的标题,与上面legend：true 配合使用
        chart.getLegend().setItemFont(legendFont);
        //设置图例边框颜色
        chart.getLegend().setFrame(new BlockBorder(Color.white));
        /*
         * VALUE_TEXT_ANTIALIAS_OFF表示将文字的抗锯齿关闭,
         * 使用的关闭抗锯齿后，字体尽量选择12到14号的宋体字,这样文字最清晰好看
         */
        //chart.getRenderingHints().put(RenderingHints.KEY_TEXT_ANTIALIASING,RenderingHints.VALUE_TEXT_ANTIALIAS_OFF);
        chart.setTextAntiAlias(false);
        //设置边框颜色
        chart.setBackgroundPaint(Color.white);
        // create plot
        CategoryPlot plot = chart.getCategoryPlot();
        plot.setOutlinePaint(null);
        // 设置横虚线可见
        plot.setRangeGridlinesVisible(false);
        // 虚线色彩
        plot.setRangeGridlinePaint(Color.gray);

        // 数据轴精度
/*        NumberAxis vn = (NumberAxis) plot.getRangeAxis();
        // vn.setAutoRangeIncludesZero(true);
        DecimalFormat df = new DecimalFormat("#0.00");
        vn.setNumberFormatOverride(df); // 数据轴数据标签的显示格式*/
        // y轴设置
        CategoryAxis domainAxis = plot.getDomainAxis();
        domainAxis.setLabelFont(labelFont);// 轴标题
        domainAxis.setTickLabelFont(labelFont);// 轴数值
        //设置Y轴数据样式
        domainAxis.setTickLabelFont(yfont);
        // Lable（Math.PI/3.0）度倾斜
        // domainAxis.setCategoryLabelPositions(CategoryLabelPositions
        // .createUpRotationLabelPositions(Math.PI / 3.0));

        domainAxis.setMaximumCategoryLabelWidthRatio(0.6f);// 横轴上的 Lable 是否完整显示

        // 设置距离图片上端距离
        domainAxis.setLowerMargin(0.01);
        // 设置距离图片下端距离
        domainAxis.setUpperMargin(0.01);
        // 设置 columnKey 是否间隔显示
        // domainAxis.setSkipCategoryLabelsToFit(true);

        plot.setDomainAxis(domainAxis);
        // 设置柱图背景色（注意，系统取色的时候要使用16位的模式来查看颜色编码，这样比较准确）
        //plot.setBackgroundPaint(new Color(255, 255, 204));
        plot.setBackgroundPaint(Color.white);

        // x轴设置
        ValueAxis rangeAxis = plot.getRangeAxis();
        rangeAxis.setLabelFont(labelFont);
        rangeAxis.setTickLabelFont(labelFont);
        // 设置最高的一个 Item 与图片顶端的距离
        rangeAxis.setUpperMargin(0.15);
        // 设置最低的一个 Item 与图片底端的距离
        rangeAxis.setLowerMargin(0.15);
        //隐藏X轴
        rangeAxis.setVisible(false);

        plot.setRangeAxis(rangeAxis);

        BarRenderer renderer = new Renderer(colorValues);
        // 设置柱子宽度
        renderer.setMaximumBarWidth(0.05);
        // 设置柱子高度
        renderer.setMinimumBarLength(0.2);
        // 设置柱子边框颜色
        renderer.setBaseOutlinePaint(Color.BLACK);
        // 设置柱子边框可见
        renderer.setDrawBarOutline(false);

        // // 设置柱的颜色
        renderer.setSeriesPaint(0, new Color(255, 255, 255));

        // 设置每个地区所包含的平行柱的之间距离
        renderer.setItemMargin(0.0);

        // 显示每个柱的数值，并修改该数值的字体属性
        renderer.setIncludeBaseInRange(true);

        renderer
                .setBaseItemLabelGenerator(new StandardCategoryItemLabelGenerator());
        renderer.setBaseItemLabelsVisible(true);
        //设置柱体数值的样式
        renderer.setBaseItemLabelFont(itemLabelFont);
        //设置柱体数值的位置
        renderer.setBasePositiveItemLabelPosition(new ItemLabelPosition(
                ItemLabelAnchor.OUTSIDE3, TextAnchor.CENTER_RIGHT));
        //去除柱体阴影效果
        renderer.setShadowVisible(false);
        //设置柱体的颜色为纯色
        renderer.setBarPainter(new StandardBarPainter());
        renderer.setItemLabelAnchorOffset(-5D);

        plot.setRenderer(renderer);
        // 设置柱的透明度
        plot.setForegroundAlpha(1.0f);

        FileOutputStream fos_jpg = null;
        try {
            isChartPathExist(CHART_PATH);
            String chartPath = CHART_PATH + chartName + ".png";
            fos_jpg = new FileOutputStream(chartPath);
            //ChartUtilities.writeChartAsJPEG(fos_jpg,1.0f,chart,600, 1200,null);
            ChartUtilities.writeChartAsPNG(fos_jpg, chart, 600, 900, true, 10);
            return chartName;
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        } finally {
            try {
                fos_jpg.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * 判断文件夹是否存在，如果不存在则新建
     *
     * @param chartPath
     */
    private static void isChartPathExist(String chartPath) {
        File file = new File(chartPath);
        if (!file.exists()) {
            file.mkdirs();
            // log.info("CHART_PATH="+CHART_PATH+"create.");
        }
    }

    // 柱状图,折线图 数据集
    public static CategoryDataset getBarData(double[][] data, String[] rowKeys,
                                             String[] columnKeys) {
        return DatasetUtilities
                .createCategoryDataset(rowKeys, columnKeys, data);

    }
}

JFreeChart 中各属性值设置