itextpdf html轉pdf 添加頁碼和水印

itextpdf html轉pdf 添加頁碼和水印，需要在Document打開前給PdfWriter添加一個pageEvent

調用代碼

import java.io.ByteArrayInputStream;
import java.io.FileOutputStream;
import java.io.StringWriter;
import java.nio.charset.Charset;

import com.itextpdf.text.Document;
import com.itextpdf.text.pdf.PdfWriter;
import com.itextpdf.tool.xml.XMLWorkerHelper;
import com.jfinal.log.Logger;
import com.jfinal.render.FreeMarkerRender;

import freemarker.template.Template;

public class PdfKit {
    private static final Logger log = Logger.getLogger(PdfKit.class);
    /**
     * 導出pdf
     * 
     * @param templateView-freemarker模板相對路徑
     * @param os-輸出流
     * @param model-模板所需要的數據
     * @throws Exception
     */
    public static void export(String templateView,Object model,String temp) throws Exception {
        log.info("生成的合同文件地址："+temp);
        FileOutputStream tempOs = new FileOutputStream(temp);  
        Template template = FreeMarkerRender.getConfiguration().getTemplate(templateView);
        StringWriter result = new StringWriter();
        template.process(model, result);
        String htmlData = result.toString();//模版變量
        Document document = new Document();
        PdfWriter writer = PdfWriter.getInstance(document, tempOs);
        
        //添加水印和頁碼
        PDFBuilder builder = new PDFBuilder();
        writer.setPageEvent(builder);
        
        document.open();
        XMLWorkerHelper.getInstance().parseXHtml(writer, document,
            new ByteArrayInputStream(htmlData.getBytes()), Charset.forName("UTF-8"));
        document.close();
    }

}

PDFBuilder代碼

此部分代碼來自於http://my.oschina.net/u/203582/blog/121838?fromerr=S6TtPCeH

package com.sign.system.kit;

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;
import com.sign.system.PathCons;
import com.sign.util.PathUtil;

/**
 * 設置頁面附加屬性
 *
 */
public class PDFBuilder extends PdfPageEventHelper {

    /**
     * 頁眉
     */
    public String header = "";

    /**
     * 文檔字體大小，頁腳頁眉最好和文本大小一致
     */
    public int presentFontSize = 12;

    /**
     * 文檔頁面大小，最好前面傳入，否則默認為A4紙張
     */
    public Rectangle pageSize = PageSize.A4;

    // 模板
    public PdfTemplate total;

    // 基礎字體對象
    public BaseFont bf = null;

    // 利用基礎字體生成的字體對象，一般用於生成中文文字
    public Font fontDetail = null;

    /**
     * 
     * Creates a new instance of PdfReportM1HeaderFooter 無參構造方法.
     * 
     */
    public PDFBuilder() {

    }

    /**
     * 
     * Creates a new instance of PdfReportM1HeaderFooter 構造方法.
     * 
     * @param yeMei
     *            頁眉字符串
     * @param presentFontSize
     *            數據體字體大小
     * @param pageSize
     *            頁面文檔大小，A4，A5，A6橫轉翻轉等Rectangle對象
     */
    public PDFBuilder(String yeMei, int presentFontSize, Rectangle pageSize) {
        this.header = yeMei;
        this.presentFontSize = presentFontSize;
        this.pageSize = pageSize;
    }

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

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

    /**
     * 
     * TODO 文檔打開時創建模板
     * 
     * @see com.itextpdf.text.pdf.PdfPageEventHelper#onOpenDocument(com.itextpdf.text.pdf.PdfWriter,
     *      com.itextpdf.text.Document)
     */
    public void onOpenDocument(PdfWriter writer, Document document) {
        total = writer.getDirectContent().createTemplate(50, 50);// 共 頁 的矩形的長寬高
    }

    /**
     * 
     * TODO 關閉每頁的時候，寫入頁眉，寫入'第幾頁共'這幾個字。
     * 
     * @see com.itextpdf.text.pdf.PdfPageEventHelper#onEndPage(com.itextpdf.text.pdf.PdfWriter,
     *      com.itextpdf.text.Document)
     */
    public void onEndPage(PdfWriter writer, Document document) {
        this.addPage(writer, document);
        this.addWatermark(writer);
    }
    
    //加分頁
    public void addPage(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); // 調節模版顯示的位置

    }
    
    //加水印
    public void addWatermark(PdfWriter writer){
        // 水印圖片
        Image image;
        try {
            image = Image.getInstance(PathUtil.getRootClassPath(PathCons.FILE_PDF_IMG));
            PdfContentByte content = writer.getDirectContentUnder();
            content.beginText();
            // 開始寫入水印
            for(int k=0;k<5;k++){
                for (int j = 0; j <4; j++) {
                    image.setAbsolutePosition(150*j,170*k);
                    content.addImage(image);
                }
            }
            content.endText();
        } catch (IOException | DocumentException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }

    /**
     * 
     * TODO 關閉文檔時，替換模板，完成整個頁眉頁腳組件
     * 
     * @see com.itextpdf.text.pdf.PdfPageEventHelper#onCloseDocument(com.itextpdf.text.pdf.PdfWriter,
     *      com.itextpdf.text.Document)
     */
    public void onCloseDocument(PdfWriter writer, Document document) {
        // 7.最后一步了，就是關閉文檔的時候，將模板替換成實際的 Y 值,至此，page x of y 制作完畢，完美兼容各種文檔size。
        total.beginText();
        total.setFontAndSize(bf, presentFontSize);// 生成的模版的字體、顏色
        String foot2 = " " + (writer.getPageNumber()-1) + " 頁";
        total.showText(foot2);// 模版顯示的內容
        total.endText();
        total.closePath();
    }
}