PDF文件添加二維碼水印教程

時間  2019-12-04
標籤 pdf 文件 添加 二維 水印 教程 简体版
原文   原文鏈接

maven配置iText的jar，主要不是全部私服都有iText的jar，maven倉庫沒有的，能夠去https://mvnrepository.com/artifact/com.itextpdf/itextpdf/5.5.12 這裏下載java

<!-- itextpdf -->
        <dependency>
            <groupId>com.itextpdf</groupId>
            <artifactId>itextpdf</artifactId>
            <version>5.5.12</version>
        </dependency>

一樣先寫個工具類,這裏是加文字水印和圖片水印的app

import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Calendar;

import com.itextpdf.text.BaseColor;
import com.itextpdf.text.DocumentException;
import com.itextpdf.text.Element;
import com.itextpdf.text.Image;
import com.itextpdf.text.pdf.BaseFont;
import com.itextpdf.text.pdf.PdfContentByte;
import com.itextpdf.text.pdf.PdfGState;
import com.itextpdf.text.pdf.PdfReader;
import com.itextpdf.text.pdf.PdfStamper;



public class WaterMarkPDFUtils {
    
    public static void main(String[] args)throws IOException, DocumentException {
        // 要輸出的pdf文件
        BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(new File("C:/Users/admin/Desktop/target.pdf")));
        Calendar cal = Calendar.getInstance();
        SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd hh:mm:ss");
        // 將pdf文件先加水印而後輸出
        setWatermark(bos, "C:/Users/admin/Desktop/t1.pdf", format.format(cal.getTime()) + "  下載使用人：" + "測試user", 16);
    }

    /**
     * 
     * @param bos輸出文件的位置
     * @param filePath
     *            原PDF位置
     * @param waterMarkName
     *            頁腳添加水印
     * @param permission
     *            權限碼
     * @throws DocumentException
     * @throws IOException
     */
    public static void setWatermark(BufferedOutputStream bos, String filePath, String waterMarkName, int permission)
            throws  IOException, DocumentException {
        PdfReader reader = new PdfReader(filePath);
        PdfStamper stamper = new PdfStamper(reader, bos);
        int total = reader.getNumberOfPages() + 1;
        PdfContentByte waterMar;
        
        PdfGState gs = new PdfGState();
        long startTime = System.currentTimeMillis();
        System.out.println("PDF加圖片水印>> start");
        for (int i = 1; i < total; i++) {
            //content = stamper.getOverContent(i);// 在內容上方加水印
            waterMar = stamper.getUnderContent(1);//在內容下方加水印
            // 設置圖片透明度爲0.2f
            gs.setFillOpacity(0.2f);
            // 設置筆觸字體不透明度爲0.4f
            gs.setStrokeOpacity(0.4f);
            // 開始水印處理
            waterMar.beginText();
            // 設置透明度
            waterMar.setGState(gs);
            // 設置水印字體參數及大小
            waterMar.setFontAndSize(BaseFont.createFont(BaseFont.HELVETICA, BaseFont.WINANSI, BaseFont.NOT_EMBEDDED), 60);
            // 設置水印對齊方式 水印內容 X座標 Y座標 旋轉角度
            waterMar.showTextAligned(Element.ALIGN_CENTER, "公司內部文件，請注意保密！",  500, 430, 45);
            // 設置水印顏色
            waterMar.setColorFill(BaseColor.GRAY);
            // 建立水印圖片
            Image image = Image.getInstance("C:/Users/admin/Desktop/icon.jpg");
            // 水印圖片位置
            image.setAbsolutePosition(380, 720); 
            // 邊框固定
            image.scaleToFit(200, 200);
            // 設置旋轉弧度
            //image.setRotation(30);// 旋轉 弧度
            // 設置旋轉角度
            //image.setRotationDegrees(45);// 旋轉 角度
            // 設置等比縮放
            image.scalePercent(90);
            // 自定義大小
            image.scaleAbsolute(200,100);
            // 附件加上水印圖片
            waterMar.addImage(image);
            // 完成水印添加
            waterMar.endText();
            // stroke
            waterMar.stroke();
        }
        long endTime = System.currentTimeMillis();
        System.out.println("PDF加圖片水印>> ok 所用時間:[{}]"+(endTime-startTime)+"s");
        stamper.close();
        reader.close();
    }
   
}

PDF加上水印
在這裏插入圖片描述maven

【拓展功能】
ok，這只是基本功能，而後要對其進行拓展工具

業務場景：要在上傳的pdf文件自動加上二維碼水印，用戶能夠掃描二維碼獲取對應數據測試

首先二維碼裏面其實也就是一些數據，好比一個連接，或者一堆文字等等，這裏能夠經過Google開源的zxing庫來事項生成二維碼圖片，而後附加到圖片，造成水印字體

maven配置zxing對應jar：google

<!-- 條形碼、二維碼生成 -->
        <dependency>
            <groupId>com.google.zxing</groupId>
            <artifactId>core</artifactId>
            <version>2.2</version>
        </dependency>
        <dependency>
            <groupId>com.google.zxing</groupId>
            <artifactId>javase</artifactId>
            <version>2.2</version>
        </dependency>

寫個工具類用於生成二維碼圖片：url

import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.Hashtable;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.WriterException;
import com.google.zxing.client.j2se.MatrixToImageWriter;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;

/**
 * 二維碼生成工具類
 */
public class QrCodeUtils {
    
    /**
     * 生成二維碼
     * @author nicky.ma
     * @date   2019年6月11日下午4:39:16
     * @param contents 二維碼的內容
     * @param width 二維碼圖片寬度
     * @param height 二維碼圖片高度
     */
    public static BufferedImage createQrCodeBufferdImage(String contents, int width, int height){
        Hashtable hints= new Hashtable();   
        hints.put(EncodeHintType.CHARACTER_SET, "utf-8");  
        BufferedImage image = null;
        try {
            BitMatrix bitMatrix = new MultiFormatWriter().encode(
                    contents,BarcodeFormat.QR_CODE, width, height, hints);
            image = toBufferedImage(bitMatrix);
        } catch (WriterException e) {
            e.printStackTrace();
        } 
        return image;
    }

    public static BufferedImage toBufferedImage(BitMatrix matrix) {
        int width = matrix.getWidth();
        int height = matrix.getHeight();
        BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
        for (int x = 0; x < width; x++) {
          for (int y = 0; y < height; y++) {
            image.setRGB(x, y, matrix.get(x, y) ? 0xFF000000 : 0xFFFFFFFF);
          }
        }
        return image;
    }
}
import java.awt.AlphaComposite;
import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.RenderingHints;
import java.awt.image.BufferedImage;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.text.SimpleDateFormat;
import java.util.Date;

import javax.imageio.ImageIO;
import javax.swing.ImageIcon;

import com.itextpdf.text.BadElementException;
import com.itextpdf.text.DocumentException;
import com.itextpdf.text.Rectangle;
import com.itextpdf.text.pdf.PdfContentByte;
import com.itextpdf.text.pdf.PdfGState;
import com.itextpdf.text.pdf.PdfReader;
import com.itextpdf.text.pdf.PdfStamper;
import com.itextpdf.text.pdf.PdfStructTreeController.returnType;
import com.itextpdf.text.pdf.parser.PdfImageObject.ImageBytesType;
import com.stuff.stuffmanage.model.CommonStuffModel;
/**
 * 
 * <pre>
 *  進行水印處理. <br>
 * </pre>
 *
 * @author mazq
 * @date 2019/06/11
 */
public class WaterMarkUtils {
    
    //Logger LOG = LoggerFactory.getLogger(WaterMarkUtils.class);
    
    
    /**
     *  生成二維碼
     * @author nicky.ma
     * @date   2019年6月12日下午2:15:51
     * @param commonStuffModel
     * @return
     */
    private static BufferedImage createQrCodeImg(CommonStuffModel commonStuffModel){
        StringBuffer strBuf = new StringBuffer();
        strBuf.append("材料入庫時間：").append(new SimpleDateFormat("yyyy-MM-dd").format(new Date())).append("\n");
        strBuf.append("材料有效期：").append(commonStuffModel.getValidEndDateStr()).append("\n");
        strBuf.append("材料名稱：").append(commonStuffModel.getStuffName()).append("\n");
        strBuf.append("材料目錄：").append(commonStuffModel.getDirName()).append("\n");
        strBuf.append("材料版本：").append(commonStuffModel.getVersion()).append("\n");
        strBuf.append("出具單位：").append(commonStuffModel.getIssueUnit()).append("\n");
        // 生成二維碼
        BufferedImage bufferedImage = QrCodeUtils.createQrCodeBufferdImage(strBuf.toString(), 175, 175);
        return bufferedImage;
    }
    
    /**
     * PDF附件添加二維碼
     * @author nicky.ma
     * @date   2019年6月11日下午3:42:15
     * @param bos 輸出文件的位置
     * @param input 輸入文件流
     */
    public static void setQrCodeForPDF(BufferedOutputStream bos, InputStream input, 
            CommonStuffModel commonStuffModel){
        BufferedImage bufferedImage = createQrCodeImg(apprCommonStuffModel);
        try {
            //PDF附件加上二維碼水印
            setWatermarkForPDF(bos, input, bufferedImage);
        } catch (IOException e) {
            e.printStackTrace();
        } catch (DocumentException e) {
            e.printStackTrace();
        }
    }
    
    /**
     * 爲PDF附件添加圖片水印
     * @author nicky.ma
     * @date   2019/6/11 12:00:32
     * @param bos 輸出文件的位置
     * @param input 輸入文件流
     * @param image 水印圖片
     */
    public static void setWatermarkForPDF(BufferedOutputStream bos, InputStream input, BufferedImage image)
            throws  IOException, DocumentException {
        PdfReader reader = new PdfReader(input);
        PdfStamper stamper = new PdfStamper(reader, bos);
        int total = reader.getNumberOfPages() + 1;
        PdfContentByte waterMar;
        
        PdfGState gs = new PdfGState();
        long startTime = System.currentTimeMillis();
        System.out.println("PDF加圖片水印 start");
        for (int i = 1; i < total; i++) {
            //content = stamper.getOverContent(i);// 在內容上方加水印
            waterMar = stamper.getUnderContent(1);//在內容下方加水印
            // 設置圖片透明度爲0.2f
            //gs.setFillOpacity(0.2f);
            // 設置筆觸字體不透明度爲0.4f
            //gs.setStrokeOpacity(0.4f);
            // 開始水印處理
            waterMar.beginText();
            // 設置透明度
            waterMar.setGState(gs);
            // 設置水印字體參數及大小
            //waterMar.setFontAndSize(BaseFont.createFont(BaseFont.HELVETICA, BaseFont.WINANSI, BaseFont.NOT_EMBEDDED), 60);
            // 設置水印對齊方式 水印內容 X座標 Y座標 旋轉角度
            //waterMar.showTextAligned(Element.ALIGN_CENTER, "公司內部文件，請注意保密！",  500, 430, 45);
            // 設置水印顏色
            //waterMar.setColorFill(BaseColor.GRAY);
            // 建立水印圖片
            com.itextpdf.text.Image itextimage = getImage(image,99);
            // 水印圖片位置
            itextimage.setAbsolutePosition(415, 745); 
            // 邊框固定
            itextimage.scaleToFit(200, 200);
            // 設置旋轉弧度
            //image.setRotation(30);// 旋轉 弧度
            // 設置旋轉角度
            //image.setRotationDegrees(45);// 旋轉 角度
            // 設置等比縮放
            //itextimage.scalePercent(90);
            // 自定義大小
            itextimage.scaleAbsolute(100,100);
            // 附件加上水印圖片
            waterMar.addImage(itextimage);
            // 完成水印添加
            waterMar.endText();
            // stroke
            waterMar.stroke();
        }
        long endTime = System.currentTimeMillis();
        System.out.println("PDF加圖片水印 ok 所用時間:"+(endTime-startTime)+"s");
        stamper.close();
        reader.close();
    }
}

對於上傳的文件，咱們怎麼知道類型？若是用Spring提供的MultipartFile，這裏能夠獲取ContentType來判斷，這裏只提供思路excel

/**文件類型集合*/
    private static Map<String,String> FILE_TYPES =new HashMap<String,String>();


    static{
        FILE_TYPES.put("xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");//xlsx
        FILE_TYPES.put("xls", "application/vnd.ms-excel");//xls
        FILE_TYPES.put("docx", "application/vnd.openxmlformats-officedocument.wordprocessingml.document");//docx
        FILE_TYPES.put("doc", "application/msword");//doc
        FILE_TYPES.put("jpg", "image/jpeg");//jpg
        FILE_TYPES.put("png", "image/png");//png
        FILE_TYPES.put("gif", "image/gif");//gif
        FILE_TYPES.put("bmp", "image/bmp");//bmp
        FILE_TYPES.put("txt", "text/plain");//txt
        FILE_TYPES.put("pdf", "application/pdf");//pdf
        FILE_TYPES.put("zip", "application/x-zip-compressed");//zip
        FILE_TYPES.put("rar", "application/octet-stream");//rar
    }

有了工具類以後，咱們須要獲取文件上傳的inputStreamcode

public void upload(MultipartFile myfiles,String url,String rootPath,CommonStuffModel commonStuffModel)throws Exception{
    
        if(!myfiles.isEmpty()){   
            File localFile = new File(rootPath+url);  
            File parentFile = localFile.getParentFile();
            if(!parentFile.exists()){
                parentFile.mkdirs();
            }
           
            String contentType = myfiles.getContentType();
         if (FILE_TYPES.get("pdf").equals(contentType)) {//上傳了PDF附件
                    InputStream inputStream = myfiles.getInputStream();
                    BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(localFile));
                    WaterMarkUtils.setQrCodeForPDF(bos, inputStream, commonStuffModel);
                }else{
                    myfiles.transferTo(localFile);
                }
        }
}

ok，而後對上傳的PDF文件就能夠加上二維碼
在這裏插入圖片描述

相關文章
相關標籤/搜索
每日一句
    每一个你不满意的现在，都有一个你没有努力的曾经。
本站公眾號
   歡迎關注本站公眾號,獲取更多信息
聯繫我們 最近搜索 最新文章 沪ICP备13005482号-10 MyBatis教程 SQL 教程 MySQL教程 Java 教程 Thymeleaf 教程 Hibernate教程 Spring教程 Redis教程