package com.kexin.locatorService.utils;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileOutputStream;

/**
 * @ProjectName: locatorService
 * @Package: com.lqm.utils
 * @ClassName: MyTest
 * @Author: luqiming
 * @Description:
 * @Date: 2020/10/29 11:48
 * @Version: 1.0
 
 
*/
public class AddWatermarkUtil {
    public static void waterPress(String srcImgPath, String outImgPath,
                           Color markContentColor, int fontSize, String waterMarkContent) {
        try {
            String[] waterMarkContents = waterMarkContent.split("\\|\\|");
            // 讀取原圖片資訊
 

            File srcImgFile = new File(srcImgPath);
            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);
            Font font = new Font("宋體", Font.PLAIN, fontSize);
            // 根據圖片的背景設定水印顏色
            g.setColor(markContentColor);
            // 設定水印文字字型
            g.setFont(font);
            // 陣列長度
            int contentLength = waterMarkContents.length;
            // 獲取水印文字中最長的
            int maxLength = 0;
            for (int i = 0; i < contentLength; i++) {
                int fontlen = getWatermarkLength(waterMarkContents[i], g);
                if (maxLength < fontlen) {
                    maxLength = fontlen;
                }
            }

            for (int j = 0; j < contentLength; j++) {
                waterMarkContent = waterMarkContents[j];
                int tempX = 10;
                int tempY = fontSize;
                // 單字元長度
                int tempCharLen = 0;
                // 單行字元總長度臨時計算
                int tempLineLen = 0;
                StringBuffer sb = new StringBuffer();
                for (int i = 0; i < waterMarkContent.length(); i++) {
                    char tempChar = waterMarkContent.charAt(i);
                    tempCharLen = getCharLen(tempChar, g);
                    tempLineLen += tempCharLen;
                    if (tempLineLen >= srcImgWidth) {
                        // 長度已經滿一行,進行文字疊加
                        g.drawString(sb.toString(), tempX, tempY);
                        // 清空內容,重新追加
                        sb.delete(0, sb.length());
                        tempLineLen = 0;
                    }
                    // 追加字元
                    sb.append(tempChar);
                }
                // 通過設定後兩個輸入引數給水印定位
                g.drawString(sb.toString(), 20, srcImgHeight - (contentLength - j - 1) * tempY-50);
            }
            g.dispose();

            // 輸出圖片
            FileOutputStream outImgStream = new FileOutputStream(outImgPath);
            ImageIO.write(bufImg, "jpg", outImgStream);
            outImgStream.flush();
            outImgStream.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static int getCharLen(char c, Graphics2D g) {
        return g.getFontMetrics(g.getFont()).charWidth(c);
    }

    /**
     * 獲取水印文字總長度
     *
     * @paramwaterMarkContent水印的文字
     * @paramg
     * @return水印文字總長度
     */
    public static int getWatermarkLength(String waterMarkContent, Graphics2D g) {
        return g.getFontMetrics(g.getFont()).charsWidth(
                waterMarkContent.toCharArray(), 0, waterMarkContent.length());
    }


    public static void main(String[] args) {
        // 原圖位置, 輸出圖片位置, 水印文字顏色, 水印文字
        String font = "張天愛||就很完美||2020-05-27 17:00:00";
        String inputAddress = "F:/UpupooWallpaper/1.jpg";
        String outputAddress = "F:/UpupooWallpaper/1.jpg";
        Color color = Color.GREEN;
        waterPress(inputAddress, outputAddress, color, 50, font);
    }
}

水印前效果：

水印後效果：

關於水印位置，需要修改：

//左下角
 g.drawString(sb.toString(), 0, srcImgHeight - (contentLength - j - 1) * tempY);

//右下角
 g.drawString(sb.toString(), srcImgWidth - maxLength, srcImgHeight - (contentLength - j - 1) * tempY);
            

關於在Windows環境不會出現中文亂碼，部署到linux環境出現中文亂碼問題解決：

參考：https://blog.csdn.net/baidu_33320352/article/details/80002718

文章來源：

https://blog.csdn.net/qq_33704186/article/details/107634885