package com.chinaztt.mes.technology.util; import com.chinaztt.mes.technology.dto.DocumentMaterialCostDTO; import com.chinaztt.mes.technology.entity.RoutingOperationParam; import lombok.SneakyThrows; import org.apache.poi.xwpf.usermodel.*; import org.apache.xmlbeans.XmlCursor; import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTTblWidth; import org.openxmlformats.schemas.wordprocessingml.x2006.main.STTblWidth; import java.io.ByteArrayOutputStream; import java.io.InputStream; import java.math.BigInteger; import java.util.List; import java.util.Map; import java.util.regex.Matcher; import java.util.regex.Pattern; /** * 工艺文件导出工具类 * @author zhangxy */ public class DocumentWordExportUtils { private static final int BYTE_LEN = 100; /** * @param document 被写入的文件 * @param style word 模板的文字样式 从左往右 对应 从0开始 * @param context 要写入的文字 * @param bold 是否加粗 * @param font 字体 */ public static void createText(XWPFDocument document, String style, String context, boolean bold, String font) { XWPFParagraph firstParagraph = document.createParagraph(); firstParagraph.setAlignment(ParagraphAlignment.LEFT); if (style != null) { firstParagraph.setStyle(style); } XWPFRun run = firstParagraph.createRun(); run.setText(context); run.setBold(bold); run.setFontFamily(font); // run.addBreak(); } /** * @param document * @param stringListMap map <规格 ， 当前工序的参数的list> */ public static void createParaTable(XWPFDocument document, Map> stringListMap) { XWPFParagraph paragraph1 = document.createParagraph(); XWPFRun paragraphRun1 = paragraph1.createRun(); paragraphRun1.setText("\r"); XWPFTable comTable = document.createTable(); CTTblWidth comTableWidth = comTable.getCTTbl().addNewTblPr().addNewTblW(); comTableWidth.setType(STTblWidth.DXA); comTableWidth.setW(BigInteger.valueOf(9072)); //获取总列的数量 int colTotal = stringListMap.keySet().size() * 2; XWPFTableRow infoTableRowOne = comTable.getRow(0); //获取总行数 int rowTotal = 0; for (String key : stringListMap.keySet()) { if (stringListMap.get(key).size() > rowTotal) { rowTotal = stringListMap.get(key).size(); } } rowTotal += 1; //画表格 先画第一行 infoTableRowOne.getCell(0).setText(""); for (int i = 1; i < colTotal; i++) { infoTableRowOne.addNewTableCell().setText(""); } //再画第一列 for (int i = 1; i < rowTotal; i++) { XWPFTableRow infoTableRow = comTable.createRow(); infoTableRow.getCell(0).setText(""); } //别的列和行 for (int i = 1; i < colTotal; i++) { for (int j = 1; j < rowTotal; j++) { XWPFTableRow infoTableRow = comTable.getRow(j); infoTableRow.getCell(i).setText(""); } } //当前行 int col = 0; for (String key : stringListMap.keySet()) { infoTableRowOne.getCell(col).setText(key); infoTableRowOne.getCell(col + 1).setText(""); List processDocMaterialPageList = stringListMap.get(key); for (int j = 0; j < processDocMaterialPageList.size(); j++) { XWPFTableRow infoTableRow = comTable.getRow(j + 1); infoTableRow.getCell(col).setText(processDocMaterialPageList.get(j).getParameterItem()); infoTableRow.getCell(col + 1).setText(processDocMaterialPageList.get(j).getParamValue()); } col += 2; } } /** * @param document 被写入数据的文件 * @param map 需要写入的数据 MAP形式 * @param rows 每一行多少个 */ public static void createMeterialTable(XWPFDocument document, Map> map, int rows) { //这边为固定格式 不要动 XWPFParagraph paragraph1 = document.createParagraph(); XWPFRun paragraphRun1 = paragraph1.createRun(); paragraphRun1.setText("\r"); XWPFTable comTable = document.createTable(); CTTblWidth comTableWidth = comTable.getCTTbl().addNewTblPr().addNewTblW(); comTableWidth.setType(STTblWidth.DXA); comTableWidth.setW(BigInteger.valueOf(9072)); //只有一张表的时候 if (map.keySet().size() == 1) { for (String key : map.keySet()) { //这边开始处理数据 List list = map.get(key); //总数量 除以 行数 去除余数 先把整行数的 数据加进去 ， int total = (list.size() / rows); //获取第一行 XWPFTableRow infoTableRowOne = comTable.getRow(0); //第一行第一格里面写入物料的规格 infoTableRowOne.getCell(0).setText(key); //for 循环 卡 空格位置 相当与设置列 for (int x = 1; x < rows; x++) { infoTableRowOne.addNewTableCell().setText(""); } //第二行往后 for (int i = 0; i < total; i++) { //处理物料名称行 XWPFTableRow infoTableRowTwo = comTable.createRow(); for (int y = 0; y < rows; y++) { infoTableRowTwo.getCell(y).setText(list.get(i * rows + y).getPartName()); } //处理物料数量行 XWPFTableRow infoTableRowThree = comTable.createRow(); for (int z = 0; z < rows; z++) { infoTableRowThree.getCell(z).setText(list.get(i * rows + z).getQuantity() + ""); } } //这边开始处理取整后余下来的几条数据，最后会增加两行 if (list.size() - total * rows != 0) { XWPFTableRow infoTableRowTwo = comTable.createRow(); XWPFTableRow infoTableRowThree = comTable.createRow(); for (int i = 0; i < list.size() - total * rows; i++) { infoTableRowTwo.getCell(i).setText(list.get(total * rows + i).getPartName()); infoTableRowThree.getCell(i).setText(list.get(total * rows + i).getQuantity() + ""); } } } } else { //获取总列的数量 int colTotal = map.keySet().size() * 2; XWPFTableRow infoTableRowOne = comTable.getRow(0); //获取总行数 int rowTotal = 0; for (String key : map.keySet()) { if (map.get(key).size() > rowTotal) { rowTotal = map.get(key).size(); } } rowTotal += 1; //画表格 先画第一行 infoTableRowOne.getCell(0).setText(""); for (int i = 1; i < colTotal; i++) { infoTableRowOne.addNewTableCell().setText(""); } //再画第一列 for (int i = 1; i < rowTotal; i++) { XWPFTableRow infoTableRow = comTable.createRow(); infoTableRow.getCell(0).setText(""); } //别的列和行 for (int i = 1; i < colTotal; i++) { for (int j = 1; j < rowTotal; j++) { XWPFTableRow infoTableRow = comTable.getRow(j); infoTableRow.getCell(i).setText(""); } } //当前行 int col = 0; for (String key : map.keySet()) { infoTableRowOne.getCell(col).setText(key); infoTableRowOne.getCell(col + 1).setText(""); List processDocMaterialPageList = map.get(key); for (int j = 0; j < processDocMaterialPageList.size(); j++) { XWPFTableRow infoTableRow = comTable.getRow(j + 1); infoTableRow.getCell(col).setText(processDocMaterialPageList.get(j).getPartName()); infoTableRow.getCell(col + 1).setText(processDocMaterialPageList.get(j).getQuantity() + ""); } col += 2; } } } public static XWPFParagraph newParagraph(XWPFParagraph paragraph) { XmlCursor xmlCursor = paragraph.getCTP().newCursor(); xmlCursor.toNextSibling(); paragraph = paragraph.getDocument().insertNewParagraph(xmlCursor); return paragraph; } /** * @param str 被处理的字符串 * @param firstChar 识别的第一个字符 * @param secondChar 识别的第二个字符 * @param times 出现的次数 * @return */ public static String returnStation(String str, String firstChar, String secondChar, int times) { String result = ""; try { Pattern pattern1 = Pattern.compile(firstChar); Matcher findMatcher1 = pattern1.matcher(str); Pattern pattern2 = Pattern.compile(secondChar); Matcher findMatcher2 = pattern2.matcher(str); int number = 0; int station1 = 0; int station2 = 0; while (findMatcher1.find()) { number++; if (number == times) { break; } } station1 = findMatcher1.start(); number = 0; while (findMatcher2.find()) { number++; if (number == times) { break; } } station2 = findMatcher2.start(); result = str.substring(station1 + 1, station2); } catch (Exception e) { e.printStackTrace(); } finally { } return result; } public static String[] returnStation(String str, String firstChar) { String[] sourceStrArray = str.split(firstChar); return sourceStrArray; } @SneakyThrows public static final byte[] input2byte(InputStream inStream) { ByteArrayOutputStream swapStream = new ByteArrayOutputStream(); byte[] buff = new byte[BYTE_LEN]; int rc = 0; while ((rc = inStream.read(buff, 0, BYTE_LEN)) > 0) { swapStream.write(buff, 0, rc); } byte[] in2b = swapStream.toByteArray(); return in2b; } }