package com.estate.expert.admin.stat.report.render;
|
|
import com.alibaba.fastjson.JSONArray;
|
import com.alibaba.fastjson.JSONObject;
|
import com.deepoove.poi.util.TableTools;
|
import com.estate.expert.admin.domain.exam.record.error.ErrorLevel;
|
import lombok.extern.log4j.Log4j2;
|
import org.apache.poi.xwpf.usermodel.ParagraphAlignment;
|
import org.apache.poi.xwpf.usermodel.XWPFTable;
|
import org.apache.poi.xwpf.usermodel.XWPFTableCell;
|
import org.apache.poi.xwpf.usermodel.XWPFTableRow;
|
import org.openxmlformats.schemas.wordprocessingml.x2006.main.STBorder;
|
|
import java.math.BigInteger;
|
|
/**
|
* 渲染帮助类
|
*/
|
@Log4j2
|
public class RenderHelper {
|
|
// 颜色常量
|
private static final String COLOR_COMMON = "000000";
|
private static final String COLOR_P1 = "FF0000";
|
private static final String COLOR_P2 = "FF8000";
|
|
/**
|
* 渲染单元格
|
* @param row 行
|
* @param text 文本
|
*/
|
public static void renderCell(XWPFTableRow row, String text) {
|
var cell = row.createCell();
|
cell.getParagraphs().get(0).setAlignment(ParagraphAlignment.CENTER);
|
renderContentCell(cell, text, false, null);
|
}
|
|
/**
|
* 渲染单元格
|
* @param row 行
|
* @param text 文本
|
* @param textColor 文本颜色
|
*/
|
public static void renderCell(XWPFTableRow row, String text, String textColor) {
|
var cell = row.createCell();
|
cell.getParagraphs().get(0).setAlignment(ParagraphAlignment.CENTER);
|
renderContentCell(cell, text, false, textColor);
|
}
|
|
/**
|
* 渲染内容单元格
|
* @param cell 单元格
|
* @param text 文本
|
* @param isTitleCell 是否是标题单元格
|
*/
|
public static void renderContentCell(XWPFTableCell cell, String text, boolean isTitleCell) {
|
renderContentCell(cell, text, isTitleCell, null);
|
}
|
|
/**
|
* 渲染内容单元格
|
* @param cell 单元格
|
* @param text 文本
|
* @param isTitleCell 是否是标题单元格
|
* @param textColor 文本颜色
|
*/
|
public static void renderContentCell(XWPFTableCell cell, String text, boolean isTitleCell, String textColor){
|
var borders = cell.getCTTc().addNewTcPr().addNewTcBorders();
|
borders.addNewTop().setVal(STBorder.SINGLE);
|
borders.addNewBottom().setVal(STBorder.SINGLE);
|
borders.addNewLeft().setVal(STBorder.SINGLE);
|
borders.addNewRight().setVal(STBorder.SINGLE);
|
|
// 设置边框的颜色和宽度
|
borders.getTop().setColor("000000"); // 设置为黑色
|
borders.getTop().setSz(BigInteger.valueOf(4)); // 设置宽度为4
|
borders.getBottom().setColor("000000");
|
borders.getBottom().setSz(BigInteger.valueOf(4));
|
borders.getLeft().setColor("000000");
|
borders.getLeft().setSz(BigInteger.valueOf(4));
|
borders.getRight().setColor("000000");
|
borders.getRight().setSz(BigInteger.valueOf(4));
|
|
// 设置字体样式
|
var run = cell.getParagraphs().get(0).createRun();
|
run.setFontFamily("宋体"); // 设置字体为宋体
|
run.setFontSize(12); // 设置字体大小为小四（12磅）
|
run.setText(text);
|
|
if(textColor != null) {
|
run.setColor(textColor);
|
} else {
|
run.setColor("000000");
|
}
|
|
// 设置单元格的水平和垂直对齐方式
|
cell.getParagraphs().get(0).setAlignment(ParagraphAlignment.CENTER);
|
cell.setVerticalAlignment(XWPFTableCell.XWPFVertAlign.CENTER);
|
|
// 设置单元格的背景颜色
|
if(isTitleCell)
|
cell.setColor("D9D9D9");
|
}
|
|
/**
|
* 合并垂直单元格
|
* @param table 表格
|
* @param startRow 开始行
|
* @param col 列
|
*/
|
public static void mergeCellsVertically(XWPFTable table, int startRow, int col) {
|
int start = 0;
|
int size = table.getRows().size();
|
String content = table.getRow(startRow).getCell(col).getText();
|
for(int i = 1; i < size; i++){
|
var cell = table.getRow(i).getCell(col);
|
String cellText = cell.getText();
|
if(cellText.equals(content)){
|
continue;
|
}
|
if(i - start > 1) {
|
// log.info("Merge cells from {} to {}", start, i - 1);
|
TableTools.mergeCellsVertically(table, col, start, i - 1);
|
}
|
start = i;
|
content = cellText;
|
}
|
|
// Merge the last group of cells
|
if(size - start > 1){
|
// log.info("Merge cells from {} to {}", start, size - 1);
|
TableTools.mergeCellsVertically(table, col, start, size - 1);
|
}
|
}
|
|
/**
|
* 获取最高隐患等级颜色
|
* @param errorRecords 错误记录
|
* @return 最高隐患等级颜色
|
*/
|
public static String highLevelColor(JSONArray errorRecords) {
|
var minLevel = ErrorLevel.P4;
|
|
for (var errorRecord : errorRecords) {
|
var e = (JSONObject) errorRecord;
|
var levelStr = e.getString("隐患等级");
|
var level = ErrorLevel.getByDesc(levelStr);
|
if (level != null && level.ordinal() < minLevel.ordinal()) {
|
minLevel = level;
|
}
|
}
|
|
switch (minLevel) {
|
case P1:
|
return COLOR_P1;
|
case P2:
|
return COLOR_P2;
|
default:
|
return COLOR_COMMON;
|
}
|
}
|
|
/**
|
* 根据隐患等级获取颜色
|
* @param errorLevel 隐患等级字符串
|
* @return 对应的颜色
|
*/
|
public static String getErrorLevelColor(String errorLevel) {
|
if (errorLevel == null) {
|
return COLOR_COMMON;
|
}
|
if (errorLevel.equals("一级隐患")) {
|
return COLOR_P1;
|
} else if (errorLevel.equals("二级隐患")) {
|
return COLOR_P2;
|
}
|
return COLOR_COMMON;
|
}
|
}
|
