PDF MERGE JAVA

package cn.com.tdl.base.pdf;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import com.lowagie.text.DocumentException;
import com.lowagie.text.pdf.PdfCopyFields;
import com.lowagie.text.pdf.PdfReader;

/**
 * 两个pdf文件合并. 多文件合并;
 * 
 * @author liuhualiang
 * 
 */
public class MyMergePDF {

	/**
	 * @param args
	 * @throws IOException
	 * @throws DocumentException
	 */
	public static void main(String[] args) {

		// creating list of readers of pdf files
		try {
			PdfReader reader1 = new PdfReader("D://A.pdf");
			PdfReader reader2 = new PdfReader("D://B.pdf");

			List<PdfReader> pdfReaderList = new ArrayList<PdfReader>();

			// Adding readers to the files
			pdfReaderList.add(reader1);
			pdfReaderList.add(reader2);

			// Creating the PdfCopyFields
			// This copy field object will take the reader object and copy the
			// pages to its object(copy)
			// So just by adding a document we can create a PDF with your merged
			// pdf

			PdfCopyFields copy = new PdfCopyFields(new FileOutputStream(
					"D://Output.pdf"));

			copy.open();

			if (null != pdfReaderList && !pdfReaderList.isEmpty()) {
				Iterator<PdfReader> iter = pdfReaderList.iterator();
				while (iter.hasNext()) {
					String pageNOs = "";
					PdfReader pdfReader = (PdfReader) iter.next();
					int noOfPages = pdfReader.getNumberOfPages();
					if (noOfPages > 0) {
						pageNOs = getNumderOfPages(noOfPages);
					}
					copy.addDocument(pdfReader, pageNOs);
				}
			}
			copy.close();
		} catch (DocumentException e) {
			e.printStackTrace();
		} catch (FileNotFoundException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
	}

	/**
	 * PDF文件合并
	 * 
	 * @param file1
	 * @param file2
	 * @param outFile
	 * @throws Throwable 
	 */
	public static void mergePdfs(String file1, String file2, String outFile) throws Throwable {
		// creating list of readers of pdf files
		try {
			// 文件不存在处理
			if (!new File(file1).exists()) {
				System.out.println("file:" + file1 + "不存在");
				outFile = file2;
				return;
			}
			if (!new File(file2).exists()) {
				System.out.println("file:" + file2 + "不存在");
				outFile = file1;
				return;
			}

			PdfReader reader1 = new PdfReader(file1);
			PdfReader reader2 = new PdfReader(file2);

			List<PdfReader> pdfReaderList = new ArrayList<PdfReader>();

			// Adding readers to the files
			pdfReaderList.add(reader1);
			pdfReaderList.add(reader2);

			// Creating the PdfCopyFields
			// This copy field object will take the reader object and copy the
			// pages to its object(copy)
			// So just by adding a document we can create a PDF with your merged
			// pdf

			PdfCopyFields copy = new PdfCopyFields(
					new FileOutputStream(outFile));

			copy.open();

			if (null != pdfReaderList && !pdfReaderList.isEmpty()) {
				Iterator<PdfReader> iter = pdfReaderList.iterator();
				while (iter.hasNext()) {
					String pageNOs = "";
					PdfReader pdfReader = (PdfReader) iter.next();
					int noOfPages = pdfReader.getNumberOfPages();
					if (noOfPages > 0) {
						pageNOs = getNumderOfPages(noOfPages);
					}
					copy.addDocument(pdfReader, pageNOs);
				}
			}
			copy.close();
		} catch (DocumentException e) {
			throw new Throwable(e);
		} catch (FileNotFoundException e) {
			throw new Throwable(e);
		} catch (IOException e) {
			throw new Throwable(e);
		} finally {
			if (new File(file1).exists()) {
				new File(file1).delete();
			}
			if (new File(file2).exists()) {
				new File(file2).delete();
			}
		}
	}

	/**
	 * 多文件合并.
	 * 
	 * @param arrPdfFile
	 * @param outFile
	 * @throws Throwable 
	 */
	public static void mergePdfs(String[] arrPdfFile, String outFile) throws Throwable {
		// creating list of readers of pdf files
		try {
			// 文件不存在处理
			List<PdfReader> pdfReaderList = new ArrayList<PdfReader>();

			for (int i = 0; i < arrPdfFile.length; i++) {
				PdfReader reader2 = new PdfReader(arrPdfFile[i]);

				// Adding readers to the files
				pdfReaderList.add(reader2);

			}

			// Creating the PdfCopyFields
			// This copy field object will take the reader object and copy the
			// pages to its object(copy)
			// So just by adding a document we can create a PDF with your merged
			// pdf

			PdfCopyFields copy = new PdfCopyFields(
					new FileOutputStream(outFile));

			copy.open();

			if (null != pdfReaderList && !pdfReaderList.isEmpty()) {
				Iterator<PdfReader> iter = pdfReaderList.iterator();
				while (iter.hasNext()) {
					String pageNOs = "";
					PdfReader pdfReader = (PdfReader) iter.next();
					int noOfPages = pdfReader.getNumberOfPages();
					if (noOfPages > 0) {
						pageNOs = getNumderOfPages(noOfPages);
					}
					copy.addDocument(pdfReader, pageNOs);
				}
			}
			copy.close();
		} catch (DocumentException e) {
			throw new Throwable(e);
		} catch (FileNotFoundException e) {
			throw new Throwable(e);
		} catch (IOException e) {
			throw new Throwable(e);
		} finally {

			for (int i = 0; i < arrPdfFile.length; i++) {

				if (new File(arrPdfFile[i]).exists()) {
					new File(arrPdfFile[i]).delete();
				}
			}

		}
	}

	/**
	 * Function to get page numbers in string with comma separated
	 * 
	 * @param noOfPages
	 * @return
	 */
	private static String getNumderOfPages(int noOfPages) {
		String pageNOs = "";
		boolean flag = false;
		for (int i = 0; i < noOfPages; i++) {

			if (flag == true) {
				Integer c = (Integer) i;
				pageNOs = pageNOs.concat("," + c.toString());
			}
			if (flag == false) {
				Integer c = (Integer) i;
				pageNOs = c.toString();
				flag = true;
			}
		}
		return pageNOs;
	}
}