package com.prlife.ecore.mobile.groupClaims.api.util;
import com.itextpdf.text.Font;
import com.itextpdf.text.pdf.AcroFields;
import com.itextpdf.text.pdf.BaseFont;
import com.itextpdf.text.pdf.PdfReader;
import com.itextpdf.text.pdf.PdfStamper;
import com.prlife.ecore.common.api.util.DateUtil;
import com.prlife.ecore.mobile.groupClaims.api.constant.ENUM_GROUPCLAIMS_CLAIMS_TYPE;
import com.prlife.ecore.mobile.groupClaims.api.constant.ENUM_GROUPCLAIMS_ID_NAME;
import com.prlife.ecore.mobile.groupClaims.api.dto.GroupClaimsDTO;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.rendering.PDFRenderer;
import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.Date;
/**
* Created with Intellij IDEA.
* Author: HeZhao
* Date: 2018/8/26 13:42
* Description:
*/
public class PDFUtil {
private PDFUtil(){}
private static final Log logger = LogFactory.getLog(PDFUtil.class);
/**
* 根据pdf模板填充相应的值生成PDF文件
*
* @param templateDir PDF模板路径
* @param dto 填充的内容对象
* @param outputDir 生成的PDF文件路径及名称
*/
public static boolean generatePDF(String templateDir, GroupClaimsDTO dto, String outputDir){
boolean flag = false;
try {
// 准备对象
PdfReader pdfReader = new PdfReader(templateDir);
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
PdfStamper pdfStamper = new PdfStamper(pdfReader, outputStream);
// 设置中文字体
BaseFont bf = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);
Font font = new Font(bf, 10, Font.NORMAL);
// 填充文本域内容
AcroFields fields = pdfStamper.getAcroFields();
fields.addSubstitutionFont(font.getBaseFont());
String policyCode = dto.getPolicyCode().replaceAll("_", "&");
fields.setField("policyCode", policyCode);
fields.setField("companyName", dto.getCompanyName());
fields.setField("applyId", dto.getApplyId());
fields.setField("insuredName", dto.getInsuredName());
fields.setField("insuredIdType", ENUM_GROUPCLAIMS_ID_NAME.getNameByValue(dto.getInsuredIdType()));
fields.setField("insuredIdNo", dto.getInsuredIdNo());
fields.setField("insuredMobile", dto.getInsuredMobile());
String accidentDateStr = DateUtil.getDateStr(dto.getAccidentDate(), "yyyy-MM-dd");
fields.setField("accidentDate", accidentDateStr);
fields.setField("visitHospital", dto.getVisitHospital());
fields.setField("accidentAfter", dto.getAccidentAfter());
fields.setField("applicantName", dto.getApplicantName());
fields.setField("applicantIdType", ENUM_GROUPCLAIMS_ID_NAME.getNameByValue(dto.getApplicantIdType()));
fields.setField("applicantIdNo", dto.getApplicantIdNo());
fields.setField("applicantMobile", dto.getApplicantMobile());
fields.setField("claimsType", ENUM_GROUPCLAIMS_CLAIMS_TYPE.getNameByValue(dto.getClaimsType()));
fields.setField("applyAmount", dto.getApplyAmount());
fields.setField("bankName", dto.getBankName());
fields.setField("accountNo", dto.getAccountNo());
fields.setField("accountName", dto.getAccountName());
fields.setField("address", dto.getAddress());
String createdDateStr = DateUtil.getDateStr(dto.getCreatedDate(), "yyyy-MM-dd");
fields.setField("createdDate", createdDateStr);
// 设置为true/false在点击生成的pdf文档的填充域时有区别,
pdfStamper.setFormFlattening(true);
pdfStamper.close();
FileOutputStream fileOutputStream = new FileOutputStream(outputDir);
fileOutputStream.write(outputStream.toByteArray());
logger.info("生成PDF文件成功");
flag = true;
} catch (Exception e) {
logger.error("生成PDF文件异常");
e.printStackTrace();
}
return flag;
}
/***
* PDF文件转图片(PNG)
*
* @param pdfFilePath PDF文件完整路径
* @param destImgPath 图片存放的文件夹
* @param dpi dpi越大转换后越清晰,相对转换速度越慢
*/
public static boolean pdf2Image(String pdfFilePath, String destImgPath, int dpi) {
File file = new File(pdfFilePath);
PDDocument pdDocument;
boolean flag = false;
try {
String imgPDFPath = file.getParent();
int dot = file.getName().lastIndexOf('.');
String imagePDFName = file.getName().substring(0, dot); // 获取图片文件名
String imgFolderPath = null;
if (destImgPath.equals("")) {
imgFolderPath = imgPDFPath + File.separator + imagePDFName;// 获取图片存放的文件夹路径
} else {
imgFolderPath = destImgPath;
}
if (createDirectory(imgFolderPath)) {
pdDocument = PDDocument.load(file);
PDFRenderer renderer = new PDFRenderer(pdDocument);
/* dpi越大转换后越清晰,相对转换速度越慢 */
PdfReader reader = new PdfReader(pdfFilePath);
int pages = reader.getNumberOfPages();
StringBuffer imgFilePath = null;
for (int i = 0; i < pages; i++) {
String imgFilePathPrefix = imgFolderPath + File.separator + imagePDFName;
imgFilePath = new StringBuffer();
imgFilePath.append(imgFilePathPrefix);
imgFilePath.append("_");
imgFilePath.append(String.valueOf(i + 1));
imgFilePath.append(".png");
File dstFile = new File(imgFilePath.toString());
BufferedImage image = renderer.renderImageWithDPI(i, dpi);
ImageIO.write(image, "png", dstFile);
}
logger.info("PDF文件转PNG图片成功");
pdDocument.close();
flag = true;
}
} catch (IOException e) {
logger.error("PDF文件转PNG图片异常");
e.printStackTrace();
}
return flag;
}
private static boolean createDirectory(String folder) {
File dir = new File(folder);
if (dir.exists()) {
return true;
} else {
return dir.mkdirs();
}
}
public static void main(String[] args) {
String templateDir = "/Users/ivan/ideaWorkSpace/policyWorkspace/trunk/UAT/e-core/WebRoot/gclaims_apply_temp.pdf";
String outputDir = "/Users/ivan/ideaWorkSpace/policyWorkspace/trunk/UAT/e-core/dist/gclaims_output2.pdf";
GroupClaimsDTO groupClaimsDTO = new GroupClaimsDTO();
groupClaimsDTO.setApplicantName("迪昂强姬");
groupClaimsDTO.setInsuredIdType("1");
groupClaimsDTO.setApplicantIdType("1");
groupClaimsDTO.setClaimsType("1");
groupClaimsDTO.setAccidentDate(new Date());
groupClaimsDTO.setCreatedDate(new Date());
generatePDF(templateDir, groupClaimsDTO, outputDir);
String pdfFilePath = "/Users/ivan/ideaWorkSpace/policyWorkspace/trunk/UAT/e-core/dist/gclaims_output2.pdf";
String destImgPath = "/Users/ivan/ideaWorkSpace/policyWorkspace/trunk/UAT/e-core/dist/";
pdf2Image(pdfFilePath, destImgPath, 300);
}
}