package com.example.demo;
 
import java.awt.AlphaComposite;
import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.RenderingHints;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
 
import javax.imageio.ImageIO;
 
/**
 * @author XuYangWei
 * @Date 2021/1/28 09:10
 */
public class WatermarkUtils {
 // 水印透明度
 private static float alpha = 0.5f;
 // 水印文字大小
 public static final int FONT_SIZE = 18;
 // 水印文字字體
 private static Font font = new Font("宋體", Font.PLAIN, FONT_SIZE);
 // 水印文字顏色
 private static Color color = Color.gray;
 // 水印之間的間隔
 private static final int XMOVE = 80;
 // 水印之間的間隔
 private static final int YMOVE = 80;
 
 /**
  * 給圖片添加水印文字
  *
  * @param logoText 水印文字
  * @param srcImgPath 源圖片路徑
  * @param targerPath 目標(biāo)圖片路徑
  */
 public static void ImageByText(String logoText, String srcImgPath, String targerPath) {
  ImageByText(logoText, srcImgPath, targerPath, null);
 }
 
 
 /**
  * 獲取文本長(zhǎng)度。漢字為1:1，英文和數(shù)字為2:1
  */
 private static int getTextLength(String text) {
  int length = text.length();
  for (int i = 0; i < text.length(); i++) {
   String s = String.valueOf(text.charAt(i));
   if (s.getBytes().length > 1) {
    length++;
   }
  }
  length = length % 2 == 0 ? length / 2 : length / 2 + 1;
  return length;
 }
 
 
 /**
  * 給圖片添加水印文字、可設(shè)置水印文字的旋轉(zhuǎn)角度
  *
  * @param logoText
  * @param srcImgPath
  * @param targerPath
  * @param degree
  */
 public static void ImageByText(String logoText, String srcImgPath, String targerPath, Integer degree) {
 
  InputStream is = null;
  OutputStream os = null;
  try {
   // 源圖片
   Image srcImg = ImageIO.read(new File(srcImgPath));
   int width = srcImg.getWidth(null);// 原圖寬度
   int height = srcImg.getHeight(null);// 原圖高度
   BufferedImage buffImg = new BufferedImage(srcImg.getWidth(null), srcImg.getHeight(null),
     BufferedImage.TYPE_INT_RGB);
   // 得到畫筆對(duì)象
   Graphics2D g = buffImg.createGraphics();
   // 設(shè)置對(duì)線段的鋸齒狀邊緣處理
   g.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BILINEAR);
   g.drawImage(srcImg.getScaledInstance(srcImg.getWidth(null), srcImg.getHeight(null), Image.SCALE_SMOOTH),
     0, 0, null);
   // 設(shè)置水印旋轉(zhuǎn)
   if (null != degree) {
    g.rotate(Math.toRadians(degree), (double) buffImg.getWidth() / 2, (double) buffImg.getHeight() / 2);
   }
   // 設(shè)置水印文字顏色
   g.setColor(color);
   // 設(shè)置水印文字Font
   g.setFont(font);
   // 設(shè)置水印文字透明度
   g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, alpha));
 
   int x = -width / 2;
   int y = -height / 2;
   int markWidth = FONT_SIZE * getTextLength(logoText);// 字體長(zhǎng)度
   int markHeight = FONT_SIZE;// 字體高度
 
   // 循環(huán)添加水印
   while (x < width * 1.5) {
    y = -height / 2;
    while (y < height * 1.5) {
     g.drawString(logoText, x, y);
 
     y += markHeight + YMOVE;
    }
    x += markWidth + XMOVE;
   }
   // 釋放資源
   g.dispose();
   // 生成圖片
   os = new FileOutputStream(targerPath);
   ImageIO.write(buffImg, "JPG", os);
   System.out.println("添加水印文字成功!");
  } catch (Exception e) {
   e.printStackTrace();
  } finally {
   try {
    if (null != is)
     is.close();
   } catch (Exception e) {
    e.printStackTrace();
   }
   try {
    if (null != os)
     os.close();
   } catch (Exception e) {
    e.printStackTrace();
   }
  }
 }
 
 public static void main(String[] args) {
  String srcImgPath = "D:/1.jpg";
  // 水印文字
  String logoText = "打印無效";
  String targerTextPath2 = "D:/2.jpg";
  System.out.println("給圖片添加水印文字開始...");
  // 給圖片添加斜水印文字
  WatermarkUtils.ImageByText(logoText, srcImgPath, targerTextPath2, -40);
  System.out.println("給圖片添加水印文字結(jié)束...");
 }
}