package com.ots.project.tool.report.LAQ.chart;
|
|
import com.ots.framework.config.EssConfig;
|
import com.ots.project.tool.report.LAQ.LAQTemplate;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.poi.util.Units;
|
import org.apache.poi.xwpf.usermodel.Document;
|
import org.apache.poi.xwpf.usermodel.XWPFDocument;
|
import org.apache.poi.xwpf.usermodel.XWPFParagraph;
|
import org.apache.poi.xwpf.usermodel.XWPFRun;
|
|
import java.io.*;
|
import java.nio.file.Files;
|
import java.nio.file.Path;
|
import java.nio.file.Paths;
|
import java.nio.file.StandardCopyOption;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Random;
|
import java.util.UUID;
|
|
/**
|
* 采用python生成图片，再插入
|
*/
|
@Slf4j
|
public class LAQAllPng {
|
|
public static void changeChart(XWPFDocument document, Map<String, Object> textMap) {
|
log.info("【进入全景图渲染模式】");
|
try{
|
List<LAQTemplate> laqTemplateList = (List<LAQTemplate>)textMap.get("allList");
|
//生成图片
|
|
String laqExcelPath = (String) textMap.get("laqExcelPath");
|
log.info("laqExcelPath文件路径："+laqExcelPath);
|
|
// 指定要检查或创建的目录路径
|
String baseDirectoryPath = "/home/data/laq/";
|
|
String bgPath = baseDirectoryPath+"bgPng/";
|
String reportPath = baseDirectoryPath+"reportCharts/"+ UUID.randomUUID()+"/";
|
|
// 创建 File 对象
|
File reportDirectory = new File(reportPath);
|
// 检查目录是否存在
|
if (!reportDirectory.exists()) {
|
// 目录不存在，创建目录
|
boolean created = reportDirectory.mkdirs(); // 使用 mkdirs() 方法可以创建多级目录
|
}
|
|
// 构建命令和参数
|
ProcessBuilder processBuilder = new ProcessBuilder(
|
"/root/miniconda3/envs/generate-charts-env/bin/python", // Python 执行程序
|
"/root/python-program/generate_charts.py", // Python 脚本路径
|
laqExcelPath, bgPath, reportPath // 传递给 Python 脚本的参数
|
);
|
|
try {
|
// 启动进程
|
Process process = processBuilder.start();
|
|
// 获取标准输出
|
BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
|
String line;
|
while ((line = reader.readLine()) != null) {
|
log.info("【执行py生成图表结果】"+line);
|
}
|
|
// 获取错误输出（用于调试）
|
BufferedReader errorReader = new BufferedReader(new InputStreamReader(process.getErrorStream()));
|
while ((line = errorReader.readLine()) != null) {
|
log.info("【执行py生成图表结果222】错误："+line);
|
}
|
|
// 等待进程完成
|
int exitCode = process.waitFor();
|
log.info("执行python完成，图表已经生成到路径："+reportPath);
|
} catch (IOException | InterruptedException e) {
|
e.printStackTrace();
|
}
|
|
|
updateImage(document,reportPath);
|
System.out.println("全景图渲染完成");
|
|
//要删除生成图表路径。
|
deleteDirectory(reportDirectory);
|
log.info("目录："+reportPath+",删除成功");
|
|
}
|
catch (Exception ex)
|
{
|
log.error("LAQ全景图异常："+ex.getMessage());
|
}
|
}
|
|
|
/**
|
* 删除目录及其内容
|
* @param dir 要删除的目录
|
* @return 删除是否成功
|
*/
|
public static boolean deleteDirectory(File dir) {
|
// 如果目录不存在，直接返回 false
|
if (!dir.exists()) {
|
return false;
|
}
|
|
// 如果是目录，则删除目录中的所有内容
|
if (dir.isDirectory()) {
|
File[] files = dir.listFiles();
|
if (files != null) {
|
for (File file : files) {
|
// 递归删除子文件或子目录
|
if (!deleteDirectory(file)) {
|
return false; // 如果删除子文件/子目录失败，返回 false
|
}
|
}
|
}
|
}
|
|
// 删除空目录或文件
|
return dir.delete();
|
}
|
|
public static void updateImage(XWPFDocument document,String reportImgPath){
|
//替换图片
|
try{
|
// 获取所有段落
|
List<XWPFParagraph> paragraphs = document.getParagraphs();
|
|
//测试用的
|
// String img1 = "D:\\桌面文件\\LAQ全景图报告\\111.png";
|
// String img2 ="D:\\桌面文件\\LAQ全景图报告\\222.png";
|
// String img3 ="D:\\桌面文件\\LAQ全景图报告\\333.png";
|
|
// /home/data
|
String img1 = reportImgPath+"charts_top.png";
|
String img2 =reportImgPath+"charts_middle.png";
|
String img3 =reportImgPath+"charts_bottom.png";
|
|
// 遍历段落进行替换操作
|
synchronized (paragraphs) {
|
for (XWPFParagraph paragraph : paragraphs) {
|
String text = paragraph.getText();
|
// 将图片插入到段落中
|
int width = Units.toEMU(580); // 图片宽度
|
int height = Units.toEMU(700); // 图片高度
|
|
//用%%号在前面就被替换了
|
if (text.contains("{{box0}}")) { // 检查是否包含要替换的文字变量
|
// 获取所有运行的列表
|
List<XWPFRun> runs = paragraph.getRuns();
|
// 移除所有运行
|
for (int i = runs.size() - 1; i >= 0 ; i--) {
|
paragraph.removeRun(i);
|
}
|
|
//创建Random类对象
|
Random random = new Random();
|
//产生随机数
|
int number = random.nextInt(999) + 1;
|
|
// 创建图片对象
|
XWPFRun run = paragraph.createRun();
|
int pictureType = Document.PICTURE_TYPE_PNG; // 图片类型
|
run.addPicture(new FileInputStream(img1), pictureType, "Seal"+number, width, height);
|
|
}
|
if (text.contains("{{box1}}")) { // 检查是否包含要替换的文字变量
|
|
// 获取所有运行的列表
|
List<XWPFRun> runs = paragraph.getRuns();
|
// 移除所有运行
|
for (int i = runs.size() - 1; i >= 0 ; i--) {
|
paragraph.removeRun(i);
|
}
|
// 创建图片对象
|
XWPFRun run = paragraph.createRun();
|
int pictureType = Document.PICTURE_TYPE_PNG; // 图片类型
|
|
run.addPicture(new FileInputStream(img2), pictureType, "image.png", width, height);
|
|
}
|
if (text.contains("{{box2}}")) { // 检查是否包含要替换的文字变量
|
|
// 获取所有运行的列表
|
List<XWPFRun> runs = paragraph.getRuns();
|
// 移除所有运行
|
for (int i = runs.size() - 1; i >= 0 ; i--) {
|
paragraph.removeRun(i);
|
}
|
// 创建图片对象
|
XWPFRun run = paragraph.createRun();
|
int pictureType = Document.PICTURE_TYPE_PNG; // 图片类型
|
|
run.addPicture(new FileInputStream(img3), pictureType, "image.png", width, height);
|
|
}
|
}
|
}
|
|
|
|
}
|
catch (Exception ex)
|
{
|
log.info("【异常】"+ex.getMessage());
|
}
|
}
|
|
}
|
