根据关键字位置为PDF添加文本和图片

Myth丶恋晨 2023-03-12 09:12 29阅读 0赞

*  首先引入ITEXT5的jar包

### 1、存放根据关键字查找位置后的信息： ###

public class TextLocal {
    
        /**
         * 关键字在PDF中的X坐标
         */
        private float x;
    
        /**
         * 关键字在PDF中的Y坐标
         */
        private float y;
    
        /**
         * 关键字在PDF中的页码
         */
        private int pageNum;
    
        /**
         * 关键字在PDF中的显示出来的长度
         */
        private float length;
    
        /**
         * 具体内容
         */
        private String content;
    
        public TextLocal(){}
    
        public TextLocal(float x, float y, int pageNum) {
            this.x = x;
            this.y = y;
            this.pageNum = pageNum;
        }
    
        public float getX() {
            return x;
        }
    
        public void setX(float x) {
            this.x = x;
        }
    
        public float getY() {
            return y;
        }
    
        public void setY(float y) {
            this.y = y;
        }
    
        public int getPageNum() {
            return pageNum;
        }
    
        public void setPageNum(int pageNum) {
            this.pageNum = pageNum;
        }
    
        public float getLength() {
            return length;
        }
    
        public void setLength(float length) {
            this.length = length;
        }
    
        public String getContent() {
            return content;
        }
    
        public void setContent(String content) {
            this.content = content;
        }
    }

### 2、存放要添加的信息 ###

public class TargetWordItem extends TextLocal {
    
        //需要替换的关键字，替换后的内容和位置
    
        //文档中搜索的关键字
        private String key;
    
        //写入的内容
        private String value;
    
        //偏移的位置 1:上  2：右  3：下面   4：左边
        private int site = 2;
    
        //偏移的量
        private float size = 1;
    
        public TargetWordItem(String key, String value, float size) {
            super();
            this.key = key;
            this.value = value;
            this.size = size;
        }
    
        public TargetWordItem() {
            super();
        }
    
        public TargetWordItem(String key, String value, int site, float size) {
            super();
            this.key = key;
            this.value = value;
            this.site = site;
            this.size = size;
        }
    
        public String getKey() {
            return key;
        }
    
        public void setKey(String key) {
            this.key = key;
        }
    
        public String getValue() {
            return value;
        }
    
        public void setValue(String value) {
            this.value = value;
        }
    
        public int getSite() {
            return site;
        }
    
        public void setSite(int site) {
            this.site = site;
        }
    
        public float getSize() {
            return size;
        }
    
        public void setSize(float size) {
            this.size = size;
        }
    }

### 3、最终的调用方法 ###

import cn.hutool.core.util.StrUtil;
    import com.itextpdf.awt.geom.Rectangle2D;
    import com.itextpdf.text.Image;
    import com.itextpdf.text.pdf.BaseFont;
    import com.itextpdf.text.pdf.PdfContentByte;
    import com.itextpdf.text.pdf.PdfReader;
    import com.itextpdf.text.pdf.PdfStamper;
    import com.itextpdf.text.pdf.parser.ImageRenderInfo;
    import com.itextpdf.text.pdf.parser.PdfReaderContentParser;
    import com.itextpdf.text.pdf.parser.RenderListener;
    import com.itextpdf.text.pdf.parser.TextRenderInfo;
    
    import java.io.FileOutputStream;
    import java.io.IOException;
    import java.util.ArrayList;
    import java.util.List;
    import java.util.Objects;
    
    public class ChangePdfText {
    
        /**
         * 从PDF中读取内容
         * 内容与关键字比对，如果满足条件，则在匹配内容的指定位置，增加需要显示的内容（替换关键字，修改PDF）
         */
        public static void main(String[] args) throws Exception {
    
            String path = "E:\\workspace\\pdfSource\\cs2.pdf";
            String outPath = path.replace(".pdf", "_temp.pdf");
    
            String imgPath = "C:\\Users\\hp\\Desktop\\测试印章图片.png";
    
            PdfReader reader = new PdfReader(path);
            PdfStamper stamper = new PdfStamper(reader, new FileOutputStream(outPath));
    
            //关键字，以及替换后的内容及位置
            List<TargetWordItem> keyList = new ArrayList<>();
            keyList.add(new TargetWordItem("受理日期", "20200514", 1));
            keyList.add(new TargetWordItem("申报日期", "20200514", 1));
    
            //找到的位置，匹配到的关键字
            List<TargetWordItem> keyItemList = matchPage(reader, keyList);
    
            //修改PDF
            for(int i = 0; i < keyItemList.size(); i++) {
                TargetWordItem keyItem = keyItemList.get(i);
                PdfContentByte overContent = stamper.getOverContent(keyItem.getPageNum());
    
                /*
                    1.添加文字
                 */
                overContent.beginText();
                //字体和大小
                BaseFont bf = BaseFont.createFont("C:/Windows/Fonts/simsun.ttc,1", BaseFont.IDENTITY_H, BaseFont.NOT_EMBEDDED);
                overContent.setFontAndSize(bf, 10F);
                //位置
                overContent.setTextMatrix(keyItem.getX() + keyItem.getSize() + 30, keyItem.getY());
                //内容
                overContent.showText(keyItem.getValue());
                overContent.endText();
    
    
                /*
                    2.添加图片
                 */
                Image image = Image.getInstance(imgPath);
                //图片位置
                image.setAbsolutePosition(keyItem.getX() + keyItem.getSize() + 30, keyItem.getY());
                //图片大小
                image.scaleToFit(113, 113);
                overContent.addImage(image);
                overContent.stroke();
            }
    
            stamper.close();
            reader.close();
        }
    
    
    
        /**
         * 满足关键字的位置
         */
        public static List<TargetWordItem> matchPage(PdfReader reader, List<TargetWordItem> keywordList) throws Exception {
    
            List<TargetWordItem> keyItemList = new ArrayList<>();
            //那些满足关键字
            for (TargetWordItem key : keywordList) {
                List<TextLocal> allItemList = getKeyWord(key.getKey(), reader);
                for (TextLocal pageItem : allItemList) {
                    if(Objects.equals(key.getKey(), pageItem.getContent())) {
    
                        key.setPageNum(pageItem.getPageNum());
                        key.setX(pageItem.getX());
                        key.setY(pageItem.getY());
    
                        keyItemList.add(key);
    
                        //找到第一个就结束
                        break;
                    }
                }
            }
    
            return keyItemList;
        }
    
    
        /**
         * 获取关键字坐标信息
         * @param keyWord 关键字
         * @return 坐标信息
         * @throws IOException IOException
         */
        private static List<TextLocal> getKeyWord(String keyWord, PdfReader pdfReader) throws IOException {
    
            List<TextLocal> textLocals = new ArrayList<>();
            //通过指定的RenderListener处理来自PDFReader页面的内容
            PdfReaderContentParser contentParser = new PdfReaderContentParser(pdfReader);
            int pageNum = pdfReader.getNumberOfPages();
            for (int i = 1; i <= pageNum; i++) {
                //用于存放文本信息
                StringBuffer sb = new StringBuffer();
                //使用指定的监听器处理来自指定页码的内容
                int finalI = i;
                contentParser.processContent(i, new RenderListener() {
    
                    /**
                     * 当一个新的文本块打开时调用
                     */
                    @Override
                    public void beginTextBlock() {
    
                    }
    
                    /**
                     * 当文本应该呈现时调用
                     * @param textRenderInfo 指定要呈现什么信息
                     */
                    @Override
                    public void renderText(TextRenderInfo textRenderInfo) {
                        //获取整页内容
                        String text = sb.append(textRenderInfo.getText()).toString();
                        if (!StrUtil.isEmpty(text) && text.contains(keyWord)) {
                            //获取坐标
                            Rectangle2D.Float boundingRectangle = textRenderInfo.getBaseline().getBoundingRectange();
                            float x = boundingRectangle.x;
                            float y = boundingRectangle.y;
                            //将关键字添加到关键字列表中
                            TextLocal textLocal = new TextLocal(x, y, finalI);
                            textLocal.setLength(keyWord.length());
                            textLocal.setContent(keyWord);
                            textLocals.add(textLocal);
                            sb.setLength(0);
                        }
                    }
    
                    /**
                     * 当一个文本块结束时调用
                     */
                    @Override
                    public void endTextBlock() {
    
                    }
    
                    /**
                     * 当图像应该被渲染时调用
                     * @param imageRenderInfo 指定要呈现什么信息
                     */
                    @Override
                    public void renderImage(ImageRenderInfo imageRenderInfo) {
    
                    }
                });
            }
    
            return textLocals;
        }
    }

## ##