java 给图片加水印

 java给图片添加水印 

下面是工具类

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.io.ClassPathResource;

import javax.imageio.ImageIO;
import javax.imageio.stream.ImageOutputStream;
import javax.swing.*;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;

/**
 * 
 * @description 添加图片水印
 */
public class ImageWatermarkUtil {
    private static Logger logger = LoggerFactory.getLogger(ImageWatermarkUtil.class);

    private static String imgType[] = {".png", ".jpg", ".jpeg", ".bmp"};

    /**
     * 给图片增加水印
     *
     * @param inputImg 图片文件流
     * @param suffix   文件结尾 例如.png/.jpg
     * @return
     * @throws IOException
     */
    public static InputStream addPicMark(InputStream inputImg, String suffix) {
        try {
            boolean addPicMark = false;
            for (String type : imgType) {
                if (type.equalsIgnoreCase(suffix)) {
                    addPicMark = true;
                }
            }
            if (!addPicMark) {
                return inputImg;
            }

            ClassPathResource classPathResource = new ClassPathResource("static/assets/img/watermark.png");
            InputStream inputStream = classPathResource.getInputStream();
            Image mark = ImageIO.read(inputStream);
            Image img = ImageIO.read(inputImg);
            // 加图片水印
            int imgWidth = img.getWidth(null);
            int imgHeight = img.getHeight(null);
            int markWidth = mark.getWidth(null);
            int markHeight = mark.getHeight(null);
            if (imgWidth <= markWidth || imgHeight <= markHeight) {
                return inputImg;
            }
            BufferedImage bufImg = new BufferedImage(imgWidth, imgHeight,
                    BufferedImage.TYPE_INT_RGB);
            //创建一个指定 BufferedImage 的 Graphics2D 对象
            Graphics2D g = bufImg.createGraphics();
            //设置对线段的锯齿状边缘处理
            g.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BILINEAR);
            //呈现一个图像，在绘制前进行从图像空间到用户空间的转换
            g.drawImage(img.getScaledInstance(imgWidth, imgHeight, Image.SCALE_SMOOTH), 0, 0, null);
            //水印图象的路径 水印一般为gif或者png的，这样可设置透明度
            ImageIcon imgIcon = new ImageIcon(mark);
            //得到Image对象。
            Image con = imgIcon.getImage();
            //透明度，最小值为0，最大值为1
            float clarity = 0.9f;
            g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, clarity));
            //表示水印图片的坐标位置(x,y)
            g.drawImage(con, imgWidth - markWidth - 10, imgHeight - markHeight - 10, null);
            ByteArrayOutputStream bs = new ByteArrayOutputStream();
            ImageOutputStream imOut = ImageIO.createImageOutputStream(bs);
            ImageIO.write(bufImg, suffix.substring(1,suffix.length()), imOut);
            InputStream is = new ByteArrayInputStream(bs.toByteArray());
            return is;
        } catch (Exception e) {
            logger.error("图片水印失败！", e);
        }
        return inputImg;
    }
}