import java.awt.AlphaComposite;
import java.awt.Color;
import java.awt.Font;
import java.awt.FontMetrics;
import java.awt.Graphics2D;
import java.awt.RenderingHints;
import java.awt.geom.AffineTransform;
import java.awt.image.BufferedImage;
import java.io.File;
import java.util.Iterator;

import javax.imageio.IIOImage;
import javax.imageio.ImageIO;
import javax.imageio.ImageReader;
import javax.imageio.ImageWriteParam;
import javax.imageio.ImageWriter;
import javax.imageio.stream.FileImageInputStream;
import javax.imageio.stream.FileImageOutputStream;

import org.junit.Test;

public class Wjt {

 @Test
 public void t3() throws Exception {
  File rootDir = new File("C:/Users/Administrator/Desktop/tem1");
  String[] fa = { "1.png", "2.jpg" };
  for (String s : fa) {
   String f = s;
   {
    FileImageInputStream fileImageInputStream = new FileImageInputStream(new File(rootDir, f));
    Iterator<ImageReader> imageReaders = ImageIO.getImageReaders(fileImageInputStream); //根據(jù)圖片文件獲取讀取器
    ImageReader imageReader = imageReaders.hasNext() ? imageReaders.next() : null; //檢測讀取器是否存在
    imageReader.setInput(fileImageInputStream); //設(shè)置讀取器輸入流
    BufferedImage bufferedImage = imageReader.read(0); //從讀取器取BufferedImage，這里只有一個，所以可以直接寫0
    { //開始加水印
     Graphics2D graphics2d = bufferedImage.createGraphics(); //取畫筆
     graphics2d.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON); //設(shè)置畫筆文字抗鋸齒
     graphics2d.transform(AffineTransform.getTranslateInstance(bufferedImage.getWidth() / 2, bufferedImage.getHeight() / 2)); //將畫筆移動到圖片中央
     graphics2d.transform(AffineTransform.getRotateInstance(Math.toRadians(45))); //將畫筆旋轉(zhuǎn)一定弧度
     float radius = Double.valueOf(Math.sqrt(bufferedImage.getWidth() * bufferedImage.getWidth() + bufferedImage.getHeight() * bufferedImage.getHeight()) / 2).floatValue(); //計算圖片外切圓半徑
     Font waterMarkerFont = new Font("微軟雅黑", Font.PLAIN, bufferedImage.getHeight() / 25); //設(shè)置畫筆要用的字體，這里可以設(shè)置字體樣式和大小
     graphics2d.setFont(waterMarkerFont); //設(shè)置畫筆字體
     graphics2d.setColor(Color.WHITE); //設(shè)置畫筆顏色
     graphics2d.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, 0.3F)); //設(shè)置畫筆操作透明度
     FontMetrics fontMetrics = graphics2d.getFontMetrics(); //取畫筆字體信息
     String waterMarker = "THIS IS WATERMARK."; //水印文字
     int stringWidth = fontMetrics.stringWidth(waterMarker); //計算水印文字寬度，很多人卡在了這一步
     for (float i = -radius; i <= radius; i += (stringWidth + bufferedImage.getHeight() / 25)) { //循環(huán)添加水印，這里可以設(shè)置水印水平距離，單位：px
      for (float j = -radius; j <= radius; j += (fontMetrics.getHeight() + bufferedImage.getHeight() / 25)) { //這里可以設(shè)置水印垂直距離，單位：px
       graphics2d.drawString(waterMarker, i, j); //畫水印
      }
     }
     graphics2d.dispose(); //釋放畫筆資源
    }
    ImageWriter imageWriter = ImageIO.getImageWriter(imageReader); //根據(jù)圖片文件讀取器獲取圖片文件寫入器
    String[] split = f.split("\\.");
    FileImageOutputStream fileOutputStream = new FileImageOutputStream(new File(rootDir, split[0] + "r." + split[1])); //生成目標圖片文件路徑
    imageWriter.setOutput(fileOutputStream); //設(shè)置寫入器輸出流
    IIOImage iioImage = new IIOImage(bufferedImage, null, null);
    ImageWriteParam imageWriteParam = imageWriter.getDefaultWriteParam();
    if (imageReader.getFormatName().equals("JPEG")) { //jpg格式圖片可設(shè)置輸出圖片文件的質(zhì)量
     imageWriteParam.setCompressionMode(ImageWriteParam.MODE_EXPLICIT); //這一行固定這樣寫，跟下一行搭配
     imageWriteParam.setCompressionQuality(0.947F); //跟上一行搭配，設(shè)置輸出圖片文件的質(zhì)量     
    }
    imageWriter.write(null, iioImage, imageWriteParam); //寫文件
    imageWriter.dispose(); //關(guān)閉寫入器
    imageReader.dispose(); //關(guān)閉讀取器
   }
  }
 }
}