package com.ots.project.tool;
|
import com.ots.common.enums.LangTypeEnum;
|
import com.ots.framework.config.EssConfig;
|
import com.ots.project.exam.domain.TLibraryCode;
|
import com.ots.project.tool.report.MAQ.base.BaseCondition;
|
import lombok.extern.slf4j.Slf4j;
|
import org.jfree.chart.ChartFactory;
|
import org.jfree.chart.ChartUtils;
|
import org.jfree.chart.JFreeChart;
|
import org.jfree.chart.axis.CategoryAxis;
|
import org.jfree.chart.axis.ValueAxis;
|
import org.jfree.chart.labels.StandardCategoryItemLabelGenerator;
|
import org.jfree.chart.plot.CategoryPlot;
|
import org.jfree.chart.plot.PlotOrientation;
|
import org.jfree.chart.renderer.category.BarRenderer;
|
import org.jfree.data.category.CategoryDataset;
|
import org.jfree.data.category.DefaultCategoryDataset;
|
import org.springframework.stereotype.Component;
|
import java.awt.*;
|
import java.io.FileOutputStream;
|
import java.util.Map;
|
import java.util.UUID;
|
|
@Component
|
@Slf4j
|
public class BarChart extends BaseCondition {
|
String percentile_score = "Percentile Score";
|
String rowKey1 = "Task/Results Orientation";
|
String rowKey2 = "People Orientation";
|
JFreeChart chart;
|
private BarChart() {
|
}
|
|
public String generateHistogram(Map<String, String> map, LangTypeEnum langTypeEnum) {
|
TLibraryCode library1 = getMaqReportLibrary(map, "LIBMAQ31003", langTypeEnum);
|
TLibraryCode library2 = getMaqReportLibrary(map, "LIBMAQ31005", langTypeEnum);
|
TLibraryCode library3 = getMaqReportLibrary(map, "LIBMAQ32005", langTypeEnum);
|
percentile_score = library1.getLangTypeContext(langTypeEnum);
|
rowKey1 = library2.getLangTypeContext(langTypeEnum);
|
rowKey2 = library3.getLangTypeContext(langTypeEnum);
|
DefaultCategoryDataset data = getDefaultCategoryDataset(map);
|
chart = getjFreeChart(data);
|
FileOutputStream fos_jpg = null;
|
String CHART_PATH = EssConfig.getUploadPath();
|
String chartName = CHART_PATH + "/" + UUID.randomUUID() + ".png";
|
try {
|
fos_jpg = new FileOutputStream(chartName);
|
ChartUtils.writeChartAsPNG(fos_jpg, chart, 1000, 200, true, 10);
|
} catch (Exception e) {
|
e.printStackTrace();
|
} finally {
|
try {
|
fos_jpg.close();
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
}
|
return chartName;
|
}
|
private JFreeChart getjFreeChart(DefaultCategoryDataset data) {
|
JFreeChart chart = ChartFactory.createBarChart(
|
"",
|
"",
|
percentile_score,
|
data,
|
PlotOrientation.HORIZONTAL,
|
false,
|
false,
|
false);
|
|
|
CategoryPlot plot = chart.getCategoryPlot();
|
|
|
|
CategoryAxis domain = plot.getDomainAxis();
|
|
domain.setTickLabelFont(new Font("黑体", Font.BOLD, 16));
|
|
domain.setLabelFont(new Font("黑体", Font.BOLD, 20));
|
|
ValueAxis rangeAxis = plot.getRangeAxis();
|
rangeAxis.setLabelFont(new Font("黑体", Font.BOLD, 16));
|
|
|
chart.getTitle().setFont(new Font("黑体", Font.BOLD, 16));
|
BarRenderer renderer = new BarRenderer();
|
|
|
|
|
|
renderer.setDefaultOutlinePaint(Color.BLACK);
|
|
renderer.setDrawBarOutline(true);
|
|
renderer.setSeriesPaint(0, new Color(132, 140, 221));
|
renderer.setSeriesPaint(1, new Color(45, 193, 112));
|
|
renderer.setItemMargin(0.0);
|
|
renderer.setIncludeBaseInRange(true);
|
renderer.setDefaultItemLabelGenerator(new StandardCategoryItemLabelGenerator());
|
renderer.setDefaultItemLabelsVisible(true);
|
plot.setRenderer(renderer);
|
|
plot.setForegroundAlpha(1.0f);
|
return chart;
|
}
|
private DefaultCategoryDataset getDefaultCategoryDataset(Map<String, String> map) {
|
int taskResultsOrientation;
|
int peopleOrientation;
|
try {
|
taskResultsOrientation = Integer.valueOf(map.get("P_Task31"));
|
peopleOrientation = Integer.valueOf(map.get("People32"));
|
} catch (Exception ex) {
|
taskResultsOrientation = 0;
|
peopleOrientation = 0;
|
log.error("柱状图参数为空，{}" + ex.getMessage(), ex);
|
}
|
return (DefaultCategoryDataset) getDataSet(taskResultsOrientation, peopleOrientation);
|
}
|
|
public CategoryDataset getDataSet(int taskResultsOrientation, int peopleOrientation) {
|
DefaultCategoryDataset data = new DefaultCategoryDataset();
|
|
data.setValue(0, rowKey1, rowKey1);
|
data.setValue(taskResultsOrientation, rowKey2, rowKey1);
|
data.setValue(peopleOrientation, rowKey1, rowKey2);
|
data.setValue(0, rowKey2, rowKey2);
|
return data;
|
}
|
@Override
|
public String getName() {
|
return null;
|
}
|
@Override
|
public TLibraryCode calculate(Map<String, String> map, LangTypeEnum langType) {
|
return null;
|
}
|
@Override
|
public Map<String, String> getMAQwaterDropsImages() {
|
return null;
|
}
|
}
|
