Java实现PDF水印添加的完整解决方案

<dependency>
    <groupId>com.itextpdf</groupId>
    <artifactId>itext7-core</artifactId>
    <version>7.2.5</version>
</dependency>

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.colors.ColorConstants;
import com.itextpdf.layout.Document;
import com.itextpdf.layout.element.Paragraph;
import com.itextpdf.layout.properties.TextAlignment;
import com.itextpdf.layout.properties.VerticalAlignment;
import com.itextpdf.kernel.font.PdfFont;
import com.itextpdf.kernel.font.PdfFontFactory;
import com.itextpdf.io.font.constants.StandardFonts;
import com.itextpdf.kernel.geom.Rectangle;
import com.itextpdf.kernel.pdf.extgstate.PdfExtGState;

import java.io.File;
import java.io.IOException;

public class PdfWatermarker {

    public static void addTextWatermark(String src, String dest, String watermarkText, float fontSize, float opacity, float rotationDegrees) throws IOException {
        PdfDocument pdfDoc = new PdfDocument(new PdfReader(src), new PdfWriter(dest));
        PdfFont font = PdfFontFactory.createFont(StandardFonts.HELVETICA_BOLD);

        PdfExtGState gs1 = new PdfExtGState().setFillOpacity(opacity); // 设置透明度

        for (int i = 1; i <= pdfDoc.getNumberOfPages(); i++) {
            PdfPage page = pdfDoc.getPage(i);
            Rectangle pageSize = page.getPageSize(); // 获取页面大小

            PdfCanvas canvas = new PdfCanvas(page); // 获取页面画布
            canvas.saveState(); // 保存当前状态
            canvas.setExtGState(gs1); // 应用透明度

            // 设置字体和颜色
            canvas.setFontAndSize(font, fontSize);
            canvas.setFillColor(ColorConstants.LIGHT_GRAY); // 水印颜色

            // 计算水印位置（这里以页面中心为例）
            float x = pageSize.getWidth() / 2;
            float y = pageSize.getHeight() / 2;

            canvas.beginText();
            canvas.setTextMatrix(1, 0, 0, 1, x, y); // 设置文本的起始位置
            canvas.showText(watermarkText);
            canvas.endText();

            // 如果需要旋转水印，可以在这里添加旋转变换
            // canvas.concatMatrix(Math.cos(Math.toRadians(rotationDegrees)), Math.sin(Math.toRadians(rotationDegrees)),
            //                     -Math.sin(Math.toRadians(rotationDegrees)), Math.cos(Math.toRadians(rotationDegrees)),
            //                     x * (1 - Math.cos(Math.toRadians(rotationDegrees))) + y * Math.sin(Math.toRadians(rotationDegrees)),
            //                     y * (1 - Math.cos(Math.toRadians(rotationDegrees))) - x * Math.sin(Math.toRadians(rotationDegrees)));

            // 重新定位并旋转文本，这是一种更直接的旋转方式，适用于整个文本块
            // 注意：iText 7中直接在PdfCanvas上旋转文本可能需要更复杂的矩阵变换，
            // 简单场景下，使用Layout模块的Paragraph更方便。
            // 鉴于这里是Canvas操作，我们可以通过调整文本矩阵来实现旋转，但这会影响文本基线。
            // 为了简化，这里暂时不直接在Canvas上进行复杂旋转，而是聚焦于基础定位和透明度。
            // 如果需要精确旋转，可以考虑将文本封装在PdfFormXObject中再旋转，或使用Layout模块。

            canvas.restoreState(); // 恢复之前保存的状态
        }

        pdfDoc.close();
    }

    public static void main(String[] args) {
        String src = "input.pdf"; // 你的输入PDF文件路径
        String dest = "output_watermarked.pdf"; // 输出PDF文件路径
        String watermarkText = "内部文件 严禁外传";
        float fontSize = 60;
        float opacity = 0.2f; // 20%透明度
        float rotationDegrees = 45; // 旋转45度

        // 确保输入文件存在
        File inputFile = new File(src);
        if (!inputFile.exists()) {
            System.err.println("错误：输入文件不存在！请将 'input.pdf' 放置在程序运行目录下。");
            return;
        }

        try {
            addTextWatermark(src, dest, watermarkText, fontSize, opacity, rotationDegrees);
            System.out.println("水印添加成功！文件已保存至：" + dest);
        } catch (IOException e) {
            System.err.println("添加水印时发生错误：" + e.getMessage());
            e.printStackTrace();
        }
    }
}