POI对word模板内容替换及图片插入
POI对word模板内容替换及图片插入
项目中需要将一些数据及图片填入word模板。看了一些前辈的博客，借鉴了一些东西，在此总结一下。如有不对的地方请指正，侵权请联系删除。本人使用的poi版本为3.9。
一、pom.xml
org.apache.poi
poi
3.9
org.apache.poi
poi-scratchpad
3.9
org.apache.poi
poi-ooxml
3.9
二、WordUtil工具类
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.xwpf.usermodel.*;
import org.apache.xmlbeans.XmlException;
import org.apache.xmlbeans.XmlToken;
import org.openxmlformats.schemas.drawingml.x2006.main.CTNonVisualDrawingProps;
import org.openxmlformats.schemas.drawingml.x2006.main.CTPositiveSize2D;
import org.openxmlformats.schemas.drawingml.x2006.wordprocessingDrawing.CTInline;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
public class WordUtil {
/**
*适用于word 2007
* @param param 需要替换的参数
* @param template 模板
*/
public static XWPFDocument generateWord(Map<String, Object> param, String template) {
XWPFDocument doc = null;
try {
OPCPackage pack = POIXMLDocument.openPackage(template);//通过路径获取word模板
doc = new CustomXWPFDocument(pack);
//通过InputStream 获取模板，此方法适用于jar包部署
// doc = new CustomXWPFDocument(template);
if (param != null && param.size() > 0) {
//处理段落
List<XWPFParagraph> paragraphList = doc.getParagraphs();
processParagraphs(paragraphList, param, doc);
//处理表格
Iterator<XWPFTable> it = doc.getTablesIterator();
while (it.hasNext()) {
XWPFTable table = it.next();
List<XWPFTableRow> rows = table.getRows();
for (XWPFTableRow row : rows) {
List<XWPFTableCell> cells = row.getTableCells();
for (XWPFTableCell cell : cells) {
List<XWPFParagraph> paragraphListTable = cell.getParagraphs();
processParagraphs(paragraphListTable, param, doc);
}
}
}
}
} catch (Exception e) {
e.printStackTrace();
}
return doc;
}
/**
* 处理段落
* @param paragraphList
* @throws FileNotFoundException
* @throws InvalidFormatException
*/
public static void processParagraphs(List<XWPFParagraph> paragraphList, Map<String, Object> param, XWPFDocument doc) throws InvalidFormatException, FileNotFoundException {
if (paragraphList != null && paragraphList.size() > 0) {
for (XWPFParagraph paragraph : paragraphList) {
List<XWPFRun> runs = paragraph.getRuns();
for (XWPFRun run : runs) {
String text = run.getText(0);
if (text != null) {
boolean isSetText = false;
for (Entry<String, Object> entry : param.entrySet()) {
String key = "${" + entry.getKey() + "}";
if (text.indexOf(key) != -1) {
isSetText = true;
Object value = entry.getValue();
if (value instanceof String) {//文本替换
text = text.replace(key, value.toString());
} else if (value instanceof Map) { //图片替换
text = text.replace(key, "");
Map pic = (Map) value;
int width = Integer.parseInt(pic.get("width").toString());
int height = Integer.parseInt(pic.get("height").toString());
int picType = getPictureType(pic.get("type").toString());
//获取图片流，因本人项目中适用流
//InputStream is = (InputStream) pic.get("content");
String byteArray = (String) pic.get("content");
CTInline inline = run.getCTR().addNewDrawing().addNewInline();
insertPicture(doc, byteArray, inline, width, height,picType);
}
}
}
if (isSetText) {
run.setText(text, 0);
}
}
}
}
}
}
/**
* 获取图片对应类型代码
* @param picType
* @return int
*/
private static int getPictureType(String picType) {
int res = CustomXWPFDocument.PICTURE_TYPE_PICT;
if (picType != null) {
if (picType.equalsIgnoreCase("png")) {
res = CustomXWPFDocument.PICTURE_TYPE_PNG;
} else if (picType.equalsIgnoreCase("dib")) {
res = CustomXWPFDocument.PICTURE_TYPE_DIB;
} else if (picType.equalsIgnoreCase("emf")) {
res = CustomXWPFDocument.PICTURE_TYPE_EMF;
} else if (picType.equalsIgnoreCase("jpg") || picType.equalsIgnoreCase("jpeg")) {
res = CustomXWPFDocument.PICTURE_TYPE_JPEG;
} else if (picType.equalsIgnoreCase("wmf")) {
res = CustomXWPFDocument.PICTURE_TYPE_WMF;
}
}
return res;
}
/**
* insert Picture
* @param document
* @param filePath
* @param inline
* @param width
* @param height
* @throws InvalidFormatException
* @throws FileNotFoundException
*/
private static void insertPicture(XWPFDocument document, String filePath,
CTInline inline, int width,
int height,int imgType) throws InvalidFormatException,
FileNotFoundException {
//通过流获取图片，因本人项目中，是通过流获取
//document.addPictureData(imgFile,imgType);
document.addPictureData(new FileInputStream(filePath),imgType);
int id = document.getAllPictures().size() - 1;
final int EMU = 9525;
width *= EMU;
height *= EMU;
String blipId =
document.getAllPictures().get(id).getPackageRelationship().getId();
String picXml = getPicXml(blipId, width, height);
XmlToken xmlToken = null;
try {
xmlToken = XmlToken.Factory.parse(picXml);
} catch (XmlException xe) {
xe.printStackTrace();
}
inline.set(xmlToken);
inline.setDistT(0);
inline.setDistB(0);
inline.setDistL(0);
inline.setDistR(0);
CTPositiveSize2D extent = inline.addNewExtent();
extent.setCx(width);
extent.setCy(height);
CTNonVisualDrawingProps docPr = inline.addNewDocPr();
docPr.setId(id);
docPr.setName("IMG_" + id);
docPr.setDescr("IMG_" + id);
}
/**
* get the xml of the picture
*
* @param blipId
* @param width
* @param height
* @return
*/
private static String getPicXml(String blipId, int width, int height) {
String picXml =
"" + "" +
" " +
" " +
" " + " + 0 +
"\" name=\"Generated\"/>" + " " +
" + blipId +
"\" xmlns:r=\"http://schemas.openxmlformats.org/officeDocument/2006/relationships\"/>" +
" " + " " + " " +
" " +
" + width + "\" cy=\"" + height +
"\"/>" + " " +
" " +
" 三、CustomXWPFDocument类
import org.apache.poi.openxml4j.opc.OPCPackage;
import org.apache.poi.xwpf.usermodel.XWPFDocument;
import java.io.IOException;
import java.io.InputStream;
public class CustomXWPFDocument extends XWPFDocument {
public CustomXWPFDocument(InputStream in) throws IOException {
super(in);
}
public CustomXWPFDocument() {
super();
}
public CustomXWPFDocument(OPCPackage pkg) throws IOException {
super(pkg);
}
}
四、word模板
五、测试类
public static void main(String[] args) throws Exception {
Map<String,Object> param = new HashMap<String, Object>();
map.put("proName","合规性分析测试");
map.put("proType","总体规划");
map.put("nonConstrArea","25438834.17");
map.put("constrArea","0.00");
Map<String,Object> header = new HashMap<String, Object>();
header.put("width", 556);
header.put("height", 230);
header.put("type", "png");
header.put("content", "F:\\picture\\test.png");//图片路径
param.put("{picture}",header);
XWPFDocument doc = WordUtil.generateWord(param, "F:\\wordTemplate.docx");
FileOutputStream fopts = new FileOutputStream("F:\\test.docx");
doc.write(fopts);
fopts.close();
}
六、本人项目中模板展示
结语：WordUtil 工具类对word模板内容替换，图片插入，个人觉得够用。望能帮助到您！