Java 代码添加水印工具类

pom 依赖

    <dependency>
          <groupId>com.itextpdf</groupId>
          <artifactId>itextpdf</artifactId>
          <version>5.3.2</version>
    </dependency>

WatermarkUtil

package cn.tempus.archive.util;

import com.itextpdf.text.BaseColor;
import com.itextpdf.text.Element;
import com.itextpdf.text.Image;
import com.itextpdf.text.pdf.*;

import javax.imageio.ImageIO;
import javax.swing.*;
import java.awt.*; 
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;


public class WatermarkUtil {

    /**水印格式:居中*/
    public static final Integer WATERMARK_STYLE_CENTER = 10;

    /**水印格式:平铺*/
    public static final Integer WATERMARK_STYLE_TILED = 20;

    /**水印间隔距离*/
    public static final Integer WATERMARK_INTERVAL = 20;

   /**
     * 给pdf添加文字水印
     * @param srcFile 原文件位置
     * @param destFile  加水印后文件位置
     * @param text  文字水印内容
     * @param watermarkStyle 水印格式(10: 居中, 20:平铺)
     */
    public static void setTextWaterMarkForPdf(String srcFile, String destFile, String text, Integer watermarkStyle)throws Exception {
        PdfReader reader = new PdfReader(srcFile);
        PdfStamper stamper = new PdfStamper(reader, new FileOutputStream(destFile));

        PdfGState gs = new PdfGState();
        gs.setFillOpacity(0.3f);// 设置透明度
        BaseFont font = BaseFont.createFont("Courier", "Cp1252", false);

        JLabel label = new JLabel();
        label.setText(text);
        FontMetrics metrics = label.getFontMetrics(label.getFont());
        int textH = metrics.getHeight() * 2;
        int textW = metrics.stringWidth(label.getText()) * 2;

        int total = reader.getNumberOfPages() + 1;
        PdfContentByte content;
        for (int i = 1; i < total; i++) {
            content = stamper.getUnderContent(i);
            content.beginText();
            content.setGState(gs);

            content.setColorFill(BaseColor.RED);
            content.setFontAndSize(font, 30);

            // 开始写入水印
            float pageWidth = reader.getPageSize(i).getWidth();
            float pageHigh = reader.getPageSize(i).getHeight();
            if (WATERMARK_STYLE_TILED.equals(watermarkStyle)) {
                for (float heigh = WATERMARK_INTERVAL; heigh < pageHigh; heigh += textH * 5) {
                    for (float width = WATERMARK_INTERVAL; width < pageWidth; width += textW * 2) {
                        content.showTextAligned(Element.ALIGN_LEFT, text, width, heigh, -15);
                    }
                }
            } else {
                float textWidth = (pageWidth - textW) / 2;
                float textHeight = (pageHigh - textH) / 2;
                content.showTextAligned(Element.ALIGN_LEFT, text, textWidth, textHeight, -15);
            }
            content.endText();
          }
          stamper.close();
    }

    /**
       * 给pdf设置图片水印
       * @param sourceFilePath 原文件位置
       * @param fileWaterMarkPath 加水印后文件输出位置
       * @param watermarkPath   水印位置
       * @param watermarkStyle 水印格式(10: 居中, 20:平铺)
       */
     public static void setImageWaterMarkForPdf(String sourceFilePath, String fileWaterMarkPath, String watermarkPath, Integer watermarkStyle) throws Exception {
        PdfReader reader = new PdfReader(sourceFilePath);
        PdfStamper stamp = new PdfStamper(reader, new FileOutputStream(fileWaterMarkPath));

        Image img = Image.getInstance(watermarkPath);
        img.setRotationDegrees(-15);
        float watermarkWidth = img.getWidth();
        float watermarkHigh = img.getHeight();
    
        PdfGState gs = new PdfGState();
        gs.setFillOpacity(0.3f);// 设置透明度
    
        int total = reader.getNumberOfPages() + 1;
        PdfContentByte content;
        for (int i = 1; i < total; i++) {
            content = stamp.getUnderContent(i); // 在内容下方加水印
            content.setGState(gs);
        
            float pageWidth = reader.getPageSize(i).getWidth();
            float pageHigh = reader.getPageSize(i).getHeight();
            if(WATERMARK_STYLE_TILED.equals(watermarkStyle)) {
                for(float heigh = WATERMARK_INTERVAL; heigh < pageHigh; heigh += watermarkHigh * 1.5) {
                    for(float width = WATERMARK_INTERVAL; width < pageWidth; width += watermarkWidth * 1.5) {
                        img.setAbsolutePosition(width, heigh);
                        content.addImage(img);
                    }
                }
            }else {
                img.setAbsolutePosition((pageWidth-watermarkWidth)/2, (pageHigh - watermarkHigh)/2);
                content.addImage(img);
            }
        }
        stamp.close();
        reader.close();
    }


    /**
      * 给图片添加文字水印
      * @param srcImgPath 原图片位置
      * @param tarImgPath 加水印后图片输出位置
      * @param waterMarkContent 文字水印内容
      * @param watermarkStyle 水印格式(10: 居中, 20:平铺)
      */
    public static void setTextWaterMarkForImage(String srcImgPath, String tarImgPath, String waterMarkContent, Integer watermarkStyle) throws Exception {
        File srcImgFile = new File(srcImgPath);
        java.awt.Image srcImg = ImageIO.read(srcImgFile);
        int srcImgWidth = srcImg.getWidth(null);
        int srcImgHeight = srcImg.getHeight(null);
        // 加水印
        BufferedImage bufImg = new BufferedImage(srcImgWidth, srcImgHeight, BufferedImage.TYPE_INT_RGB);
        Graphics2D g = bufImg.createGraphics();
        g.drawImage(srcImg, 0, 0, srcImgWidth, srcImgHeight, null);
        g.setColor(Color.RED); 
        g.setFont(new Font("Calibri Light", Font.PLAIN, 30)); 
        g.rotate(Math.toRadians(15)); //水印旋转
        g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, 0.3f)); //设置水印透明度
    
        //设置水印的坐标
        int textWidth = g.getFontMetrics(g.getFont()).charsWidth(waterMarkContent.toCharArray(), 0, waterMarkContent.length());
        int textHight = g.getFontMetrics(g.getFont()).getHeight();
        if(WatermarkUtil.WATERMARK_STYLE_TILED.equals(watermarkStyle)) {
            for (float y = 0; y < srcImgHeight; y += textHight * 2) {
                for (float x = 0; x < srcImgWidth; x += textWidth) {
                    g.drawString(waterMarkContent, x + y, y-x);  
                }
            }
        }else {
            int x = (srcImgWidth - textWidth/2)/2;  
            int y = (srcImgHeight - textHight*4)/2;  
            g.drawString(waterMarkContent, x, y);  
        }
        g.dispose();  
        // 输出图片  
        FileOutputStream outImgStream = new FileOutputStream(tarImgPath);  
        ImageIO.write(bufImg, "jpg", outImgStream);
        outImgStream.flush();  
        outImgStream.close();  
    }


    /**
      * 给图片添加图片水印
      * @param srcImgPath 原图片位置
      * @param markImgPath 图片水印的位置
      * @param tarImgPath 加水印后图片位置
      * @param watermarkStyle 水印格式(10: 居中, 20:平铺)
      */
    public static void setImageWaterMarkForImage(String srcImgPath, String markImgPath, String tarImgPath, Integer watermarkStyle) throws Exception{
        java.awt.Image img = ImageIO.read(new File(srcImgPath));
        java.awt.Image mark = ImageIO.read(new File(markImgPath));
    
        int imgWidth = img.getWidth(null);
        int imgHeight = img.getHeight(null);
        BufferedImage bufImg = new BufferedImage(imgWidth, imgHeight, BufferedImage.TYPE_INT_RGB);
    
        Graphics2D g = bufImg.createGraphics();
        g.drawImage(img, 0, 0, bufImg.getWidth(), bufImg.getHeight(), null);
        g.rotate(Math.toRadians(15)); //水印旋转
        g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, 0.1f)); //设置水印透明度
    
        int waterWidth = mark.getWidth(null);
        int waterHight = mark.getHeight(null);
        if(WatermarkUtil.WATERMARK_STYLE_TILED.equals(watermarkStyle)) {
            for (int y = 0; y < imgHeight; y += waterHight * 2) {
                for (int x = 0; x < imgWidth; x += waterWidth) {
                    g.drawImage(mark, x + y, y-x, null);  
                }
             }
        }else {
            int x = (imgWidth - waterWidth/2)/2;
            int y = (imgHeight - waterHight*3)/2;
            g.drawImage(mark, x, y, null);
        }
        g.dispose();
        FileOutputStream outImgStream = new FileOutputStream(tarImgPath);
        ImageIO.write(bufImg, "jpg", outImgStream);
        outImgStream.flush();
        outImgStream.close();
    }


    /**获取文件类型*/
    public static String getFileType(String uri) throws IOException {
        Path path = Paths.get(uri);
        return Files.probeContentType(path);
    }
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 202,802评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,109评论 2 379
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,683评论 0 335
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,458评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,452评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,505评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,901评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,550评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,763评论 1 296
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,556评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,629评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,330评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,898评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,897评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,140评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,807评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,339评论 2 342

推荐阅读更多精彩内容