127f3157ac24f56f4746eb315f6f048ea5eb4a16.svn-base 4.67 KB
/**
 * Project Name:html2pdf File Name:PdfReportHeaderFooter.java Package Name:com.kirin.common.util.pdf Date:2015年11月18日下午4:36:17 Copyright (c) 2015, PHXL All Rights Reserved.
 */

package com.phxl.common.util;

import java.io.IOException;

import com.itextpdf.text.Document;
import com.itextpdf.text.DocumentException;
import com.itextpdf.text.Element;
import com.itextpdf.text.Font;
import com.itextpdf.text.Image;
import com.itextpdf.text.PageSize;
import com.itextpdf.text.Phrase;
import com.itextpdf.text.Rectangle;
import com.itextpdf.text.pdf.BaseFont;
import com.itextpdf.text.pdf.ColumnText;
import com.itextpdf.text.pdf.PdfContentByte;
import com.itextpdf.text.pdf.PdfPageEventHelper;
import com.itextpdf.text.pdf.PdfTemplate;
import com.itextpdf.text.pdf.PdfWriter;

/**
 * ClassName:PdfReportHeaderFooter <br/>
 * Function: TODO ADD FUNCTION. <br/>
 * Reason: TODO ADD REASON. <br/>
 * Date: 2015年11月18日 下午4:36:17 <br/>
 * 
 * @author
 * @version
 * @since JDK 1.6
 * @see
 */
public class PdfReportHeaderFooter extends PdfPageEventHelper {

	public String header = "";
	public int presentFontSize = 12;
	public Rectangle pageSize = PageSize.A4;
	public PdfTemplate total;
	public BaseFont bf = null;
	public Font fontDetail = null;
	
	public String waterMarkImgSrc = "";

	public PdfReportHeaderFooter() {
	}

	public PdfReportHeaderFooter(String header, int presentFontSize, Rectangle pageSize) {
		this.header = header;
		this.presentFontSize = presentFontSize;
		this.pageSize = pageSize;
	}

	public PdfReportHeaderFooter(String waterMarkImgSrc) {
		this.waterMarkImgSrc = waterMarkImgSrc;
	}
	
	public void setHeader(String header) {
		this.header = header;
	}

	public void setPresentFontSize(int presentFontSize) {
		this.presentFontSize = presentFontSize;
	}

	
	public String getWaterMarkImgSrc() {
		return waterMarkImgSrc;
	}

	
	public void setWaterMarkImgSrc(String waterMarkImgSrc) {
		this.waterMarkImgSrc = waterMarkImgSrc;
	}

	public void onOpenDocument(PdfWriter writer, Document document) {
		total = writer.getDirectContent().createTemplate(50, 50);
	}

	public void onEndPage(PdfWriter writer, Document document) {
		try {
			if (bf == null) {
				bf = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", false);
			}
			if (fontDetail == null) {
				fontDetail = new Font(bf, presentFontSize, Font.NORMAL);// 数据体字体
			}
		} catch (DocumentException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
		// 1.写入页眉
		ColumnText.showTextAligned(writer.getDirectContent(), Element.ALIGN_LEFT, new Phrase(header, fontDetail),
				document.left(), document.top() + 20, 0);
		// 2.写入前半部分的 第 X页/共
		int pageS = writer.getPageNumber();
		String foot1 = "第 " + pageS + " 页 /共";
		Phrase footer = new Phrase(foot1, fontDetail);
		// 3.计算前半部分的foot1的长度,后面好定位最后一部分的'Y页'这俩字的x轴坐标,字体长度也要计算进去 = len
		float len = bf.getWidthPoint(foot1, presentFontSize);
		// 4.拿到当前的PdfContentByte
		PdfContentByte cb = writer.getDirectContent();
		// 5.写入页脚1,x轴就是(右margin+左margin + right() -left()- len)/2.0F
		// 再给偏移20F适合人类视觉感受,否则肉眼看上去就太偏左了
		// ,y轴就是底边界-20,否则就贴边重叠到数据体里了就不是页脚了;注意Y轴是从下往上累加的,最上方的Top值是大于Bottom好几百开外的。
		ColumnText.showTextAligned(cb, Element.ALIGN_CENTER, footer,
				(document.rightMargin() + document.right() + document.leftMargin() - document.left() - len) / 2.0F
						+ 20F,
				document.bottom() - 20, 0);
		// 6.写入页脚2的模板(就是页脚的Y页这俩字)添加到文档中,计算模板的和Y轴,X=(右边界-左边界 - 前半部分的len值)/2.0F +
		// len , y 轴和之前的保持一致,底边界-20
		cb.addTemplate(total,
				(document.rightMargin() + document.right() + document.leftMargin() - document.left()) / 2.0F + 20F,
				document.bottom() - 20);
		
		if(!waterMarkImgSrc.equals("")){
			//增加水印图片
			PdfContentByte canvas = writer.getDirectContentUnder();
			Image image;
			try {
				image = Image.getInstance(waterMarkImgSrc);
				image.scaleAbsolute(PageSize.A4.rotate());
				image.setAbsolutePosition(0, 0);
				canvas.addImage(image);
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
	}

	public void onCloseDocument(PdfWriter writer, Document document) {
		total.beginText();
		total.setFontAndSize(bf, presentFontSize);// 生成的模版的字体、颜色
		String foot2 = " " + (writer.getPageNumber() - 1) + " 页";
		total.showText(foot2);// 模版显示的内容
		total.endText();
		total.closePath();
	}
}