itext7 给pdf添加水印

package com.fh.util.pdf;

import com.itextpdf.io.source.ByteArrayOutputStream;
import com.itextpdf.kernel.color.Color;
import com.itextpdf.kernel.color.DeviceRgb;
import com.itextpdf.kernel.font.PdfFont;
import com.itextpdf.kernel.font.PdfFontFactory;
import com.itextpdf.kernel.geom.Rectangle;
import com.itextpdf.kernel.pdf.PdfDocument;
import com.itextpdf.kernel.pdf.PdfPage;
import com.itextpdf.kernel.pdf.PdfReader;
import com.itextpdf.kernel.pdf.PdfWriter;
import com.itextpdf.kernel.pdf.canvas.PdfCanvas;
import com.itextpdf.kernel.pdf.extgstate.PdfExtGState;

public class ItextPdfUnit {

    //给pdf添加水印
    public static byte[] addWatermarkToPdf(String path,String watermarkStr)throws Exception{
        // 设置水印相关参数
        watermarkStr += '\t' + watermarkStr;
        watermarkStr += '\t' + watermarkStr;
        Color color = new DeviceRgb(80, 80, 80); // 水印颜色（黑色）
        float opacity = 0.15f; // 不透明度
        float fontSize = 50f; // 字体大小
        float angle = 45f; // 角度
        float xOffset = -180f; // X 偏移量
        float yOffset = 100f; // Y 偏移量
        float lineSpacing = 120f; // 字符串行间距

        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
        PdfReader pdfReader = new PdfReader(path).setUnethicalReading(true);// 以非标准方法打开文件
        PdfWriter pdfWriter = new PdfWriter(outputStream);
        PdfDocument pdfDoc = new PdfDocument(pdfReader, pdfWriter);


        int pageCount = pdfDoc.getNumberOfPages();
        PdfFont font = PdfFontFactory.createFont("STSongStd-Light", "UniGB-UCS2-H", true);

        for (int i = 1; i <= pageCount; i++) {
            PdfPage page = pdfDoc.getPage(i);
            PdfCanvas canvas = new PdfCanvas(page);

            Rectangle pageSize = page.getPageSize();
            float x = pageSize.getLeft() + xOffset;
            float y = pageSize.getTop() - yOffset;

            PdfExtGState gs = new PdfExtGState();
            gs.setFillOpacity(opacity); // 设置不透明度
            canvas.setExtGState(gs);

            canvas.saveState()
                    .setFillColor(color)
                    .setExtGState(new PdfExtGState().setFillOpacity(opacity))
                    .concatMatrix(Math.cos(angle), Math.sin(angle), -Math.sin(angle), Math.cos(angle), x, y);

            // 计算每条水印的高度
            float watermarkHeight = font.getFontProgram().getFontMetrics().getAscender() * fontSize;

            // 按行铺满水印
            float startY = 0;
            while (startY < pageSize.getHeight() + 200) {
                canvas.beginText()
                        .setFontAndSize(font, fontSize)
                        .setTextMatrix(1, 0, 0, 1, 0, -startY) // 加负号以保持斜着
                        .showText(watermarkStr)
                        .endText();
                startY += lineSpacing;
                canvas.concatMatrix(1, 0, 0, 1, -120, 0); // 每条水印相对上一条向左偏移 200
            }

            canvas.restoreState();
            canvas.release();
        }
        pdfDoc.close();
        return outputStream.toByteArray();
    }


}