com.centit.support.office.Watermark4Pdf Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of centit-office Show documentation
office(doc、xls、ppt) pdf 等文档处理
There is a newer version: 2.2.1812
package com.centit.support.office;

import java.io.FileOutputStream;
import java.io.IOException;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.centit.support.algorithm.DatetimeOpt;
import com.itextpdf.text.BaseColor;
import com.itextpdf.text.DocumentException;
import com.itextpdf.text.Element;
import com.itextpdf.text.Rectangle;
import com.itextpdf.text.pdf.BaseFont;
import com.itextpdf.text.pdf.PdfContentByte;
import com.itextpdf.text.pdf.PdfGState;
import com.itextpdf.text.pdf.PdfReader;
import com.itextpdf.text.pdf.PdfStamper;

public abstract class Watermark4Pdf {
	public static boolean runFlag = false;

	private static Log logger = LogFactory.getLog(Watermark4Pdf.class);
	
	/**
	 * 为文档添加水印：目前只支持给pdf、word、excel、ppt增加水印，并且输出只能是Pdf文件。
	 * 
	 * @param inputFile       源文件路径及文件  
	 * @param waterMarkStr    水印字符串
	 * @return 是否成功
	 */
	public static boolean addWatermark(String inputFile,  String waterMarkStr) {
		//将源office文件转换为pdf
		//String suffix = inputFile.substring(inputFile.lastIndexOf("."));
		String tmpPdfFile = inputFile.substring(0,inputFile.lastIndexOf("."))+ 
				 DatetimeOpt.convertDateToString(DatetimeOpt.currentUtilDate(), "yyyyMMddHHmmssSSS")+".pdf";
		String wartermarkFile = inputFile.substring(0,inputFile.lastIndexOf("."))+ ".pdf";
		if(! OfficeToPdf.office2Pdf(inputFile,tmpPdfFile))
			return false;
		
		return  addWatermark4Pdf(tmpPdfFile, 
				wartermarkFile,
				waterMarkStr, 
				0.4f, 
				45f, 
				60f);
		
	}
	
	
	/**
	 * 为文档添加水印：目前只支持给pdf、word、excel、ppt增加水印，并且输出只能是Pdf文件。
	 * 
	 * @param inputFile       源文件路径及文件 名称
	 * @param outputFile      输出文件路径及文件名称
	 * @param waterMarkStr    水印字符串
	 * @param opacity         文字透明度(1-10)
	 * @param rotation        旋转度数(1-170)
	 * @param fontSize        字体大小(1-1)
	 * @return boolean        目前 不支持位置自定义：因设置了文字大小、倾斜度后不好计算水印文字的长宽数据。
	 */
	public static boolean addWatermark4Pdf(String inputFile, 
									String outputFile,
									String waterMarkStr, 
									float opacity, 
									float rotation, 
									float fontSize) {
		PdfContentByte content = null;
		BaseFont base = null;
		Rectangle pageRect = null;
		PdfGState gs = new PdfGState();
		PdfReader pdfReader = null;
		PdfStamper pdfStamper = null;
		try{
			pdfReader =new PdfReader(inputFile);
			pdfStamper = new PdfStamper(pdfReader, new FileOutputStream(
						outputFile));
			
			base = BaseFont.createFont("STSongStd-Light", "UniGB-UCS2-H",
					BaseFont.NOT_EMBEDDED);
			if (base == null || pdfStamper == null) {
				return false;
			}
			// 设置透明度为0.4
			gs.setFillOpacity(opacity);
			gs.setStrokeOpacity(opacity);
			int toPage = pdfStamper.getReader().getNumberOfPages();
			for (int i = 1; i <= toPage; i++) {
				pageRect = pdfStamper.getReader().getPageSizeWithRotation(i);
				// 计算水印X,Y坐标
				float x = pageRect.getWidth() / 2;
				float y = pageRect.getHeight() / 2;
				// 获得PDF最顶层
				content = pdfStamper.getOverContent(i);
				content.saveState();
				// set Transparency
				content.setGState(gs);
				content.beginText();
				content.setColorFill(BaseColor.GRAY);
				content.setFontAndSize(base, fontSize);
				// 水印文字成45度角倾斜
				content.showTextAligned(Element.ALIGN_CENTER, waterMarkStr, x,
						y, rotation);
				content.endText();
			}
			
		} catch (IOException | DocumentException e1) {
			e1.printStackTrace();
			return false;
		} finally {
			content = null;
			base = null;
			pageRect = null;
			try {
				if(pdfReader!=null)
					pdfReader.close();
				if(pdfStamper!=null)
					pdfStamper.close();				
			} catch (DocumentException | IOException e) {
				logger.error(e.getMessage(),e);//e.printStackTrace();
			}			
		}
		return true;
	}
		
}