PDF操作

Posted 2022-03-12 chenligeng

package com.pdf.util;

import com.itextpdf.io.source.ByteArrayOutputStream;
import com.itextpdf.kernel.font.PdfFont;
import com.itextpdf.kernel.font.PdfFontFactory;
import com.itextpdf.kernel.pdf.PdfDocument;
import com.itextpdf.layout.element.Cell;
import com.itextpdf.layout.element.Paragraph;
import com.itextpdf.layout.element.Table;
import com.itextpdf.layout.property.HorizontalAlignment;
import com.itextpdf.layout.property.TextAlignment;
import com.itextpdf.layout.property.UnitValue;
import com.itextpdf.text.DocumentException;
import com.itextpdf.text.Rectangle;
import com.itextpdf.text.pdf.*;
import com.pdf.entity.pdf;
import org.apache.ibatis.io.Resources;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;

import java.io.*;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class PdfUtils 
    public static final String DEST2 = "E:\\test4.pdf";//输出文件路径
    public static void OutPutTableFile(String dest) throws Exception
        PdfDocument pdfDoc = new PdfDocument(new com.itextpdf.kernel.pdf.PdfWriter(dest));
        com.itextpdf.layout.Document doc = new com.itextpdf.layout.Document(pdfDoc);

        PdfFont sysFont = PdfFontFactory.createFont("STSongStd-Light", "UniGB-UCS2-H", false);
        Table table = new Table(new float[]1,1,1,7).setWidth(UnitValue.createPercentValue(100));
//        Cell cell1=new Cell().add(new Paragraph("考生序号")).setFont(sysFont).setBackgroundColor(new DeviceRgb(221,234,238));
        Cell cell1=new Cell().add(new Paragraph("考生序号")).setFont(sysFont).setFontSize(10).setTextAlignment(TextAlignment.CENTER);
        cell1.setWidth(Float.parseFloat("50"));
        Cell cell2=new Cell().add(new Paragraph("姓名")).setFont(sysFont).setFontSize(10).setTextAlignment(TextAlignment.CENTER);
//        Cell cell3=new Cell().add(new Paragraph("座位号")).setFont(sysFont).setBorder(new SolidBorder(new DeviceRgb(221,234,238), 1));
        cell2.setWidth(Float.parseFloat("50"));
        Cell cell3=new Cell().add(new Paragraph("座位号")).setFont(sysFont).setFontSize(10).setTextAlignment(TextAlignment.CENTER);
//        Cell cell4=new Cell().add(new Paragraph("更正栏")).setFont(sysFont).setBorder(Border.NO_BORDER);
        cell3.setWidth(Float.parseFloat("50"));
        Cell cell4=new Cell().add(new Paragraph("更正栏")).setFont(sysFont).setFontSize(10).setTextAlignment(TextAlignment.CENTER);
        cell4.setWidth(Float.parseFloat("100"));
        table.addCell(cell1);
        table.addCell(cell2);
        table.addCell(cell3);
        table.addCell(cell4);
        //从数据库里面查询出数据
        String str = "mybatis-config.xml";
        InputStream in = Resources.getResourceAsStream(str);
        SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(in);
        SqlSession session = sqlSessionFactory.openSession();
        List<pdf> pdfList = session.selectList("SelectPdf");
        for (pdf p :
                pdfList) 
            for (int i=0;i<1;i++)
                Cell cell5=new Cell().add(new Paragraph(String.valueOf(p.getExamNo()))).setFont(sysFont).setFontSize(10).setTextAlignment(TextAlignment.CENTER);
                table.addCell(cell5);
                Cell cell6=new Cell().add(new Paragraph(p.getUsername())).setFont(sysFont).setFontSize(10).setTextAlignment(TextAlignment.CENTER);
                table.addCell(cell6);
                Cell cell7=new Cell().add(new Paragraph(String.valueOf(p.getSitRoom()))).setFont(sysFont).setFontSize(10).setTextAlignment(TextAlignment.CENTER);
                table.addCell(cell7);
                Cell cell8=new Cell().add(new Paragraph(p.getChangTiele())).setFont(sysFont).setFontSize(10).setTextAlignment(TextAlignment.CENTER);
                table.addCell(cell8);
            
        
        doc.add(table.setHorizontalAlignment(HorizontalAlignment.CENTER));
        doc.close();
    
     //填充表单字段
    //    public static final String DEST2 = "E:\\test4.pdf";//文件路径
    public static void OutPutForm(String fileName,String finalPath) throws IOException, DocumentException 
        PdfReader reader = new PdfReader(fileName);
        ByteArrayOutputStream bos = new ByteArrayOutputStream();
        /* 将要生成的目标PDF文件名称 */
        PdfStamper ps = new PdfStamper(reader, bos);
        PdfContentByte under = ps.getUnderContent(1);
        /* 使用中文字体 */
        BaseFont bf = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);
        ArrayList<BaseFont> fontList = new ArrayList<BaseFont>();
        fontList.add(bf);
        /* 取出报表模板中的所有字段 */
        AcroFields fields = ps.getAcroFields();
        fields.setSubstitutionFonts(fontList);
        fillData(fields, data());
        /* 必须要调用这个，否则文档不会生成的 */
        ps.setFormFlattening(true);
        ps.close();
        OutputStream fos = new FileOutputStream(finalPath);
        fos.write(bos.toByteArray());
        fos.flush();
        fos.close();
        bos.close();
    

    public static void fillData(AcroFields fields, Map<String, String> data)
            throws IOException, DocumentException 
        for (String key : data.keySet()) 
            String value = data.get(key);
            fields.setField(key, value); // 为字段赋值,注意字段名称是区分大小写的
        
    

    public static Map<String, String> data() 
        Map<String, String> data = new HashMap<String, String>();
        data.put("examTime", "2019年05月29日");
        data.put("examName", "国家通用语言文字综合能力测试(GYC)");
        data.put("examLvl", "四级");
        data.put("positionName","北京大学");
        data.put("roomNo","301");
        return data;
    

    public static void addTail(String TempPdf,String TempPdf2,String finalPath) throws IOException, DocumentException 
        FileOutputStream outputStream = new FileOutputStream(finalPath,true);
        PdfReader reader = new PdfReader(TempPdf); //读取pdf模板
        PdfReader reader1 = new PdfReader(TempPdf2);  //读取pdf2模板
        Rectangle pagesize = reader.getPageSize(1);
        com.itextpdf.text.Document document = new com.itextpdf.text.Document(pagesize);
        PdfWriter writer = PdfWriter.getInstance(document,outputStream);
        document.open();
        PdfContentByte pdfContentByte = writer.getDirectContentUnder();
        PdfImportedPage pageTemplate = writer.getImportedPage(reader, 1);
        PdfImportedPage pageTemplate1 = writer.getImportedPage(reader1, 1);
        pdfContentByte.addTemplate(pageTemplate,0,-50);
        pdfContentByte.addTemplate(pageTemplate1,0,0);
//        document.add(new Paragraph("Hello World"));
        document.close();
        reader.close();
        File file1 = new File(TempPdf2);
        if (file1.exists())
            file1.delete();
        

        File file = new File(TempPdf);
        if (file.exists())
            file.delete();
        
        System.out.println("执行完成");
    



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

        String fileNames = "C://Users//Administrator//Desktop//座位一览表.pdf"; // pdf模板
        String finalPath = "E://test005.pdf";

        String TempPdf = "E://test4.pdf";
        String TempPdf2 = "E://test005.pdf";
        String  finaPath = "E://chenligeng.pdf";

        OutPutTableFile(TempPdf);
        OutPutForm(fileNames,finalPath);
        addTail(TempPdf,TempPdf2,finaPath);