commit

pom.xml

@@ -61,6 +61,7 @@
         <docker-java.version>3.2.13</docker-java.version>
         <itextpdf.version>5.5.13.3</itextpdf.version>
         <itext-asian.version>5.2.0</itext-asian.version>
+        <jfreechart.version>1.5.0</jfreechart.version>
     </properties>
 
 
@@ -87,6 +88,14 @@
             </dependency>
             <!-- pdf - 结束 -->
 
+            <!-- 饼图 - 开始 -->
+            <dependency>
+                <groupId>org.jfree</groupId>
+                <artifactId>jfreechart</artifactId>
+                <version>${jfreechart.version}</version>
+            </dependency>
+            <!-- 饼图 - 结束 -->
+
             <!-- docker 客户端 -->
             <dependency>
                 <groupId>com.github.docker-java</groupId>

simulation-resource-server/pom.xml

@@ -151,6 +151,10 @@
             <version>5.0.3</version>
             <scope>compile</scope>
         </dependency>
+        <dependency>
+            <groupId>org.jfree</groupId>
+            <artifactId>jfreechart</artifactId>
+        </dependency>
 
     </dependencies>
     <build>

simulation-resource-server/src/main/java/com/css/simulation/resource/server/app/impl/SimulationProjectServiceImpl.java

@@ -3023,8 +3023,10 @@ public class SimulationProjectServiceImpl implements SimulationProjectService {
             List<AlgorithmScoreVO> algorithmScoreList = vo.getAlgorithmScoreList();
 
             StringBuilder stringBuilder = new StringBuilder("       " + vo.getAlgorithmName() + "在");
+            Integer totalSceneNum = null;
             for (AlgorithmScoreVO v : algorithmScoreList) {
                 if ("汇总".equals(v.getProjectName())) {
+                    totalSceneNum = v.getSceneNum();
                     continue;
                 }
                 stringBuilder.append(v.getProjectName()).append("项目的").append(v.getSceneNum()).append("个场景中，测试得分为").append(v.getScore()).append("，得分率为").append(v.getScoreRatio()).append("%；");
@@ -3156,6 +3158,12 @@ public class SimulationProjectServiceImpl implements SimulationProjectService {
             }
             document.add(pdfPTable2);
 
+
+            // 场景评价
+            addElement(document, "场景评价", null, bf3, 15, false, 30, true);
+            addElement(document, "       场景测试包共包含" + totalSceneNum + "个场景.", null, bf3, 15, false, 30, false);
+
+
             // ------- 4.详细得分说明 -------
 
             /*

simulation-resource-server/src/main/java/com/css/simulation/resource/server/infra/util/ChartUtil.java

@@ -0,0 +1,215 @@
+package com.css.simulation.resource.server.infra.util;
+
+import lombok.SneakyThrows;
+import org.jfree.chart.ChartFactory;
+import org.jfree.chart.JFreeChart;
+import org.jfree.chart.StandardChartTheme;
+import org.jfree.chart.labels.StandardCategoryItemLabelGenerator;
+import org.jfree.chart.labels.StandardPieSectionLabelGenerator;
+import org.jfree.chart.plot.CategoryPlot;
+import org.jfree.chart.plot.PiePlot;
+import org.jfree.chart.plot.PlotOrientation;
+import org.jfree.chart.renderer.category.BarRenderer;
+import org.jfree.chart.renderer.category.LineAndShapeRenderer;
+import org.jfree.chart.renderer.category.StandardBarPainter;
+import org.jfree.data.category.DefaultCategoryDataset;
+import org.jfree.data.general.DefaultPieDataset;
+
+import java.awt.*;
+import java.text.DecimalFormat;
+import java.text.NumberFormat;
+
+import com.itextpdf.text.Document;
+import com.itextpdf.text.Image;
+import com.itextpdf.text.PageSize;
+import com.itextpdf.text.pdf.PdfWriter;
+import org.apache.commons.io.FileUtils;
+import org.jfree.chart.ChartUtils;
+
+import java.io.*;
+
+
+/**
+ * @author 15765
+ * @date 2024/3/26 12:47
+ */
+public class ChartUtil {
+    private static final Color[] BAR_COLORS = new Color[]{new Color(79, 129, 189), new Color(192, 80, 77), new Color(155, 187, 89),};
+
+    public static final Color[] LINE_COLORS = new Color[]{new Color(237, 123, 46), new Color(90, 154, 213), new Color(164, 164, 164),};
+
+    private static final Color[] PIE_COLORS = new Color[]{new Color(75, 172, 198), new Color(128, 100, 162), new Color(155, 187, 89), new Color(192, 80, 77), new Color(79, 129, 189), new Color(44, 77, 117), new Color(247, 150, 70), new Color(165, 165, 165),};
+
+
+    static StandardChartTheme initChartTheme() {
+        StandardChartTheme currentTheme = new StandardChartTheme("JFree");
+        // 横轴纵轴标题文字大小
+        currentTheme.setLargeFont(new java.awt.Font("宋体", java.awt.Font.BOLD, 15));
+        // 横轴纵轴数值文字大小
+        currentTheme.setRegularFont(new java.awt.Font("宋体", java.awt.Font.PLAIN, 13));
+        currentTheme.setExtraLargeFont(new java.awt.Font("宋体", java.awt.Font.BOLD, 20));
+        // 背景颜色
+        currentTheme.setPlotBackgroundPaint(new Color(255, 255, 204, 0));
+        // 边框线条
+        currentTheme.setPlotOutlinePaint(new Color(0, 0, 0, 0));
+        // 网格线条
+        currentTheme.setRangeGridlinePaint(new Color(78, 74, 74));
+        return currentTheme;
+    }
+
+    /**
+     * 线图
+     *
+     * @param title             标题
+     * @param categoryAxisLabel 分类标签
+     * @param valueAxisLabel    数值标签
+     * @param dataset           数据集
+     * @return org.jfree.chart.JFreeChart
+     * @author Hou_fx
+     * @date 2021.8.4 10:39
+     */
+    public static JFreeChart lineChart(String title, String categoryAxisLabel, String valueAxisLabel, DefaultCategoryDataset dataset) {
+        ChartFactory.setChartTheme(initChartTheme());
+
+        JFreeChart chart = ChartFactory.createLineChart(title,// 图标题
+                categoryAxisLabel,// x轴标题
+                valueAxisLabel,// y轴标题
+                dataset,//数据集
+                PlotOrientation.VERTICAL,//图表方向
+                true, true, false);
+
+        CategoryPlot plot = chart.getCategoryPlot();
+        LineAndShapeRenderer renderer = (LineAndShapeRenderer) plot.getRenderer();
+        // 折现点显示数值
+        renderer.setDefaultItemLabelsVisible(true);
+        renderer.setDefaultItemLabelGenerator(new StandardCategoryItemLabelGenerator());
+        // 更改线条颜色
+        for (int i = 0; i < dataset.getRowKeys().size(); i++) {
+            renderer.setSeriesPaint(i, LINE_COLORS[i]);
+        }
+        return chart;
+    }
+
+
+    /**
+     * 柱状图
+     *
+     * @param title
+     * @param categoryAxisLabel
+     * @param valueAxisLabel
+     * @param dataset           数据集
+     * @return org.jfree.chart.JFreeChart
+     * @author Hou_fx
+     * @date 2021.8.4 14:03
+     */
+    public static JFreeChart barChart(String title, String categoryAxisLabel, String valueAxisLabel, DefaultCategoryDataset dataset) {
+        ChartFactory.setChartTheme(initChartTheme());
+
+        JFreeChart chart = ChartFactory.createBarChart(title, categoryAxisLabel, valueAxisLabel, dataset, PlotOrientation.VERTICAL, true, true, false);
+
+        CategoryPlot plot = chart.getCategoryPlot();
+        BarRenderer renderer = (BarRenderer) plot.getRenderer();
+        // 纯色显示
+        renderer.setBarPainter(new StandardBarPainter());
+        // 柱子上显示小数字
+        renderer.setDefaultItemLabelsVisible(true);
+        renderer.setDefaultItemLabelGenerator(new StandardCategoryItemLabelGenerator());
+        // 设置柱子间隔
+        renderer.setItemMargin(0.0);
+        // 设置柱子颜色
+        for (int i = 0; i < dataset.getRowKeys().size(); i++) {
+            renderer.setSeriesPaint(i, BAR_COLORS[i]);
+        }
+        return chart;
+    }
+
+    /**
+     * 饼图
+     *
+     * @param title
+     * @param dataset
+     * @return org.jfree.chart.JFreeChart
+     * @author Hou_fx
+     * @date 2021.8.4 14:04
+     */
+    public static JFreeChart pieChart(String title, DefaultPieDataset dataset) {
+        ChartFactory.setChartTheme(initChartTheme());
+
+        JFreeChart chart = ChartFactory.createPieChart(title, dataset, true, true, false);
+        PiePlot plot = (PiePlot) chart.getPlot();
+        // 设置扇区颜色
+        for (int i = 0; i < dataset.getKeys().size(); i++) {
+            plot.setSectionPaint(dataset.getKey(i), PIE_COLORS[i]);
+        }
+        // 设置扇区的线条颜色
+        plot.setDefaultSectionOutlinePaint(new Color(255, 255, 255));
+        // 设置扇区的线条大小
+        plot.setDefaultSectionOutlineStroke(new BasicStroke(3));
+        // 设置标签颜色
+        plot.setLabelLinkPaint(new Color(255, 255, 255, 0));
+        // 设置标签背景色
+        plot.setLabelBackgroundPaint(new Color(255, 255, 255, 0));
+        // 设置标签线条颜色
+        plot.setLabelOutlinePaint(new Color(255, 255, 255, 0));
+        // 设置标签阴影颜色
+        plot.setLabelShadowPaint(new Color(255, 255, 255, 0));
+        // 设置饼图阴影颜色
+        plot.setShadowPaint(new Color(255, 255, 255, 0));
+        // 添加标签数字百分比显示
+        plot.setLabelGenerator(new StandardPieSectionLabelGenerator(("{0}{2}"), NumberFormat.getNumberInstance(), new DecimalFormat("0.00%")));
+        return chart;
+    }
+
+    @SneakyThrows
+    public static void main(String[] args) {
+        PdfWriter writer = null;
+        OutputStream outputStream = null;
+        Document doc = null;
+        try {
+            doc = new Document(PageSize.A4);
+            doc.setMargins(55, 46, 27, 20);
+
+            File f = new File("C:\\Users\\Administrator\\Desktop\\新建文件夹\\9.pdf");
+            if (!f.getParentFile().exists()) {
+                FileUtils.forceMkdir(f.getParentFile());
+            }
+            outputStream = new FileOutputStream(f);
+            writer = PdfWriter.getInstance(doc, outputStream);
+
+            doc.open();
+
+
+            // 饼图
+            ByteArrayOutputStream pieBos = new ByteArrayOutputStream();
+            ChartUtils.writeChartAsJPEG(pieBos, ChartUtil.pieChart("饼图", getPieDataset()), 650, 240);
+            Image pieImage = Image.getInstance(pieBos.toByteArray());
+            pieImage.scalePercent(60);
+            doc.add(pieImage);
+            // 饼图
+            ByteArrayOutputStream pieBos1 = new ByteArrayOutputStream();
+            ChartUtils.writeChartAsJPEG(pieBos1, ChartUtil.pieChart("饼图2", getPieDataset()), 650, 240);
+            Image pieImage1 = Image.getInstance(pieBos1.toByteArray());
+            pieImage1.scalePercent(60);
+            doc.add(pieImage1);
+
+        } catch (Exception e) {
+            e.printStackTrace();
+        } finally {
+            doc.close();
+            writer.close();
+            outputStream.close();
+
+        }
+
+    }
+
+    public static DefaultPieDataset getPieDataset() {
+        DefaultPieDataset dataset = new DefaultPieDataset();
+        dataset.setValue("S ", 2);
+        dataset.setValue("A ", 3);
+        dataset.setValue("B ", 1);
+        dataset.setValue("C ", 5);
+        return dataset;
+    }
+
+}