Java 使用iText7生成带页码的PDF文件（同时生成目录，但是不会合并两个PDF）

一、效果图

1.带页码效果

2.目录效果

前言：Java 使用iText7生成带页码的PDF文件，同时生成目录PDF，但限于水平，暂时还在摸索合并两个PDF。不过看了一下，iText好像有生成目录的代码，可以查看我另一篇博客我copy了一份说明QAQ。

点击这里可下载源码

二、实现代码

1.PDFUtil.java（添加标题、锻炼辅助类）

package test.hui;

import java.io.IOException;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.Map;

import com.itextpdf.io.font.PdfEncodings;
import com.itextpdf.io.image.ImageData;
import com.itextpdf.io.image.ImageDataFactory;
import com.itextpdf.kernel.events.IEventHandler;
import com.itextpdf.kernel.events.PdfDocumentEvent;
import com.itextpdf.kernel.font.PdfFont;
import com.itextpdf.kernel.font.PdfFontFactory;
import com.itextpdf.kernel.pdf.PdfDocument;
import com.itextpdf.kernel.pdf.PdfWriter;
import com.itextpdf.kernel.pdf.xobject.PdfImageXObject;
import com.itextpdf.layout.Document;
import com.itextpdf.layout.element.AreaBreak;
import com.itextpdf.layout.element.Image;
import com.itextpdf.layout.element.Paragraph;
import com.itextpdf.layout.property.TextAlignment;

/**
*
*
*
*
* @author Hui
*
*/
public class PDFUtil {

public static String DEST2 = "";//文件路径
public static PdfFont sysFont =null;
private Map<String, Integer> catalogMap = new LinkedHashMap<String, Integer>();//<标题，页数>

public PDFUtil(String url){
try {
//sysFont = PdfFontFactory.createFont("STSongStd-Light", "UniGB-UCS2-H", false);//中文设置
sysFont = PdfFontFactory.createFont("c://windows//fonts//simsun.ttc,1",PdfEncodings.IDENTITY_H, false);//中文设置,解决特殊字符错误
DEST2=url;
} catch (IOException e) {
e.printStackTrace();
}
}

/**
* 创建pdf doc
* @return doc
* */
public Document createPdfDoc() throws Exception{
PdfDocument pdfDoc = new PdfDocument(new PdfWriter(DEST2));
Document doc = new Document(pdfDoc);//构建文档对象
//设置页码
TextFooterEventHandler eh= new TextFooterEventHandler(doc);
pdfDoc.addEventHandler(PdfDocumentEvent.END_PAGE,eh);

return doc;
}

/**
* 添加大标题
* @param
*
* */
public void addTitle(Document doc,String title) throws Exception{

Paragraph paragraph = new Paragraph();
paragraph.add(title).setFont(sysFont).setBold().setFontSize(20).setTextAlignment(TextAlignment.CENTER);
doc.add(paragraph);
}

/**
* 添加一级标题
* @param
*
* */
public void addHeading1(Document doc,String pg) throws Exception{
Paragraph paragraph = new Paragraph();
paragraph.add(pg).setFont(sysFont).setBold().setFontSize(16);
doc.add(paragraph);
catalogMap.put(pg, doc.getPdfDocument().getNumberOfPages());
}

/**
* 添加二级标题
* @param
*
* */
public  void addHeading2(Document doc,String pg) throws Exception{
Paragraph paragraph = new Paragraph();
paragraph.add(pg).setFont(sysFont).setBold().setFontSize(14).setFirstLineIndent(14);
doc.add(paragraph);
catalogMap.put(pg, doc.getPdfDocument().getNumberOfPages());
}

/**
* 添加三级标题
* @param
*
* */
public  void addHeading3(Document doc,String pg) throws Exception{
Paragraph paragraph = new Paragraph();
paragraph.add(pg).setFont(sysFont).setBold().setFontSize(12);
doc.add(paragraph);
catalogMap.put(pg, doc.getPdfDocument().getNumberOfPages());
}

/**
* 添加段落
* @param
* */
public void addParagraph(Document doc,String pg) throws Exception{
Paragraph paragraph = new Paragraph();
paragraph.add(pg).setFont(sysFont).setFirstLineIndent(20);//中文字体，首行缩进
doc.add(paragraph);
}

/**
* 添加图片
* @param
* */
public void addImg(Document doc,String url) throws Exception{
ImageData imgData=ImageDataFactory.create(url);
Image pic=new Image(imgData);
doc.add(pic);
}

public Map<String, Integer> getCatalogMap() {
return catalogMap;
}

public void setCatalogMap(Map<String, Integer> catalogMap) {
this.catalogMap = catalogMap;
}

}

2.TextFooterEventHandler.java（生成页码辅助类）

package test.hui;

import java.io.IOException;

import com.itextpdf.io.font.FontConstants;
import com.itextpdf.kernel.events.Event;
import com.itextpdf.kernel.events.IEventHandler;
import com.itextpdf.kernel.events.PdfDocumentEvent;
import com.itextpdf.kernel.font.PdfFontFactory;
import com.itextpdf.kernel.geom.Rectangle;
import com.itextpdf.kernel.pdf.canvas.PdfCanvas;
import com.itextpdf.layout.Document;

/**
*
*
*
*
* @author Hui
*
*
*/
public class TextFooterEventHandler implements IEventHandler {

protected Document doc;
protected int page;
public TextFooterEventHandler(Document doc) {
this.doc = doc;

}
@Override
public void handleEvent(Event event) {
PdfDocumentEvent docEvent = (PdfDocumentEvent) event;
PdfCanvas canvas = new PdfCanvas(docEvent.getPage());
Rectangle pageSize = docEvent.getPage().getPageSize();
page++;
canvas.beginText();
try {
canvas.setFontAndSize(PdfFontFactory.createFont(FontConstants.HELVETICA_OBLIQUE), 15);
} catch (IOException e) {
e.printStackTrace();
}
canvas.moveText((pageSize.getRight() - doc.getRightMargin() + (pageSize.getLeft() + doc.getLeftMargin())) / 2,pageSize.getBottom() + doc.getBottomMargin())
.showText(""+page)
.endText()
.release();
}

}

3.Testp.java

package test.hui;

import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;

import com.itextpdf.kernel.events.PdfDocumentEvent;
import com.itextpdf.kernel.pdf.PdfDocument;
import com.itextpdf.layout.Document;

public class Testp{

public static String url="G:\\test4.pdf";//这里两个路径应改成你自己的路径
public static String urlOrig="G:\\test1.pdf";

static Map<String, Integer> catalogMap = new HashMap<String, Integer>();

public static void main(String[] args) throws Exception {

long old = System.currentTimeMillis();

//生成内容pdf
PDFUtil pdfUtil=new PDFUtil(url);
Document doc=pdfUtil.createPdfDoc();
PdfDocument pdfDoc=doc.getPdfDocument();

//这里生成页码，引用TextFooterEventHandler的方法
TextFooterEventHandler eh= new TextFooterEventHandler(doc);
pdfDoc.addEventHandler(PdfDocumentEvent.END_PAGE,eh);

pdfUtil.addTitle(doc, "预案");
for(int i=0;i<15;i++){
pdfUtil.addHeading1(doc,i+" 标题");
if(i%2==0){
pdfUtil.addHeading2(doc,i+".1 小标题标题");
}
pdfUtil.addParagraph(doc, "内容QAQAQZAAAWQSXFDSFSD内容QAQAQZAAAWQSXFDSFSD内容QAQAQZAAAWQSXFDSFSD内容QAQAQZAAAWQSXFDSFSD内容QAQAQZAAAWQSXFDSFSD");
}

pdfDoc.close();
doc.close();

//------------------分割线--------------

//生成目录方法，暂时只能生产目录的pdf，还需努力合并
PDFUtil pdfUtil2=new PDFUtil(urlOrig);
Document doc2=pdfUtil2.createPdfDoc();
System.out.println(doc2.getPdfDocument().getCatalog().getPdfObject());
for(Entry<String, Integer> entry:pdfUtil.getCatalogMap().entrySet()){

c650
String title=entry.getKey();
String page=entry.getValue()+"";
if(countInString(title, ".")==0){
pdfUtil2.addParagraph(doc2, title+"......."+page);
}else if(countInString(title, ".")==1){
pdfUtil2.addParagraph(doc2, title+"......."+page);
}else{

}
}
doc2.close();

long now = System.currentTimeMillis();
System.out.println("共耗时：" + ((now - old) / 1000.0) + "秒\n\n" + "文件保存在:"+url+","+urlOrig);

}

//用来控制生成目录
public static int countInString(String str1, String str2) {
int total = 0;
for (String tmp = str1; tmp != null && tmp.length() >= str2.length();){
if(tmp.indexOf(str2) == 0){
total++;
tmp = tmp.substring(str2.length());
}else{
tmp = tmp.substring(1);
}
}
return total;
}

}