图片添加文字水印和图片水印

/**
 * 〈〉
 *
 * @author WatermelonBoy
 */
public class WaterBase {
    // 水印透明度
    private static float alpha = 0.9f;
    // 水印横向位置
    private static int positionWidth = 150;
    // 水印纵向位置
    private static int positionHeight = 100;
    // 水印文字字体
    private static Font font = new Font("宋体", Font.BOLD, 30);
    // 水印文字颜色
    private static Color color = Color.red;

    public static void setImageMarkOptions(float alpha, int positionWidth, int positionHeight, Font font, Color color) {
        if (alpha != 0.0f) WaterBase.alpha = alpha;
        if (positionWidth != 0) WaterBase.positionWidth = positionWidth;
        if (positionHeight != 0) WaterBase.positionHeight = positionHeight;
        if (font != null) WaterBase.font = font;
        if (color != null) WaterBase.color = color;
    }

    public static void main(String[] args) throws IOException {
        String srcImgPath = "F:\GitLocalStorage\xiakesong\services\house-service\src\main\resources\wallhaven.jpg";
        String iconPath = "F:\GitLocalStorage\xiakesong\services\house-service\src\main\resources\icon.jpg";
        String targerIconPath2 = "F:\GitLocalStorage\xiakesong\services\house-service\src\main\resources\targetImg.jpg";
        String logoText = "@侠客颂";

        //可通过此方法调试间距,颜色.不适用则默认初始化数据
        //setImageMarkOptions(0.9f, 250, 100, null, null);

        /**给图片添加水印图片,水印图片旋转0*/
        //markImageByIcon(iconPath, srcImgPath, targerIconPath2, 0);

        /**给图片添加水印文字,水印图片旋转-45*/
        //markImageByText(logoText, srcImgPath, targerIconPath2, -45);

        /**给图片添加水印文字Hutool工具类*/
        FileInputStream fileInputStream = new FileInputStream(new File("gg"));
        FileOutputStream fileOutputStream = new FileOutputStream(new File(targerIconPath2));

        BufferedImage srcImg = ImageIO.read(fileInputStream);
        int height = srcImg.getHeight() - 100;
        int width = srcImg.getWidth() - 200;
        fileInputStream = new FileInputStream(new File(srcImgPath));
        ImgUtil.pressText(fileInputStream, fileOutputStream, logoText, Color.CYAN, new Font("微软雅黑", Font.BOLD, 60), 0, -600, 0.2f);
        fileInputStream.close();
        fileOutputStream.close();

    }

    //读取图片转为Base64字符串
    public static String TuImage(String args) {

        String imgFile = args;//待处理的图片
        InputStream in = null;
        byte[] data = null;
        //读取图片字节数组
        try {
            in = new FileInputStream(imgFile);
            data = new byte[in.available()];
            in.read(data);
            in.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
        //对字节数组Base64编码
        BASE64Encoder encoder = new BASE64Encoder();
        return encoder.encode(data);
    }

    public static String markImageByIcon(String iconPath, String srcImgPath, String targerPath, Integer degree) {

        BASE64Encoder base64en = new BASE64Encoder();
        BASE64Decoder base64de = new BASE64Decoder();
        InputStream is = null;
        OutputStream os = null;
        byte[] b;
        try {
            //针对上送的图片是Base64场景
            b = base64de.decodeBuffer(TuImage(srcImgPath));
            is = new java.io.ByteArrayInputStream(b);
            BufferedImage srcImg = ImageIO.read(is);


            BufferedImage buffImg = new BufferedImage(srcImg.getWidth(null), srcImg.getHeight(null), BufferedImage.TYPE_INT_RGB);

            // 1、得到画笔对象
            Graphics2D g = buffImg.createGraphics();

            // 2、设置对线段的锯齿状边缘处理
            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);
            // 3、设置水印旋转
            if (null != degree) {
                g.rotate(Math.toRadians(degree), (double) buffImg.getWidth() / 2, (double) buffImg.getHeight() / 2);
            }

            // 4、水印图片的路径 水印图片一般为gif或者png的，这样可设置透明度(本地地址可直接填写,服务器地址ImageIcon导包位置一定要正确)
            ImageIcon imgIcon = new ImageIcon(iconPath);

            // 5、得到Image对象
            Image img = imgIcon.getImage();

            g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, alpha));

            // 6、水印图片的位置
            g.drawImage(img, positionWidth, positionHeight, null);
            g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER));
            // 7、释放资源
            g.dispose();

            // 8、生成图片
            os = new FileOutputStream(targerPath);
            ImageIO.write(buffImg, "JPG", os);

            //把图片转换为字节
            ByteArrayOutputStream bot = new ByteArrayOutputStream();
            JPEGImageEncoder encoder = JPEGCodec.createJPEGEncoder(bot);
            encoder.encode(buffImg);
            byte[] rstByte = bot.toByteArray();
            srcImgPath = base64en.encode(rstByte);
            System.out.println("图片完成添加水印图片");
            return srcImgPath;

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if (null != os)
                    os.close();
                if (null != is)
                    is.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return targerPath;
    }

    /**
     * 给图片添加水印文字
     *
     * @param logoText   水印文字
     * @param srcImgPath 源图片路径
     * @param targerPath 目标图片路径
     */
    public static void markImageByText(String logoText, String srcImgPath,
                                       String targerPath) {
        markImageByText(logoText, srcImgPath, targerPath, null);
    }

    /**
     * 给图片添加水印文字、可设置水印文字的旋转角度
     *
     * @param logoText
     * @param srcImgPath
     * @param targerPath
     * @param degree
     */
    public static void markImageByText(String logoText, String srcImgPath,
                                       String targerPath, Integer degree) {

        BASE64Decoder base64de = new BASE64Decoder();
        InputStream is = null;
        OutputStream os = null;
        byte[] b;
        try {
            // 1、源图片
            //针对上送的图片是Base64场景
            b = base64de.decodeBuffer(TuImage(srcImgPath));
            is = new java.io.ByteArrayInputStream(b);
            BufferedImage srcImg = ImageIO.read(is);
            int height = srcImg.getHeight() - 100;
            int width = srcImg.getWidth() - 200;
            System.out.println(height);
            System.out.println(width);
            setImageMarkOptions(0.9f, width, height, null, null);

            //Image srcImg = ImageIO.read(new File(srcImgPath));
            BufferedImage buffImg = new BufferedImage(srcImg.getWidth(null), srcImg.getHeight(null), BufferedImage.TYPE_INT_RGB);

            // 2、得到画笔对象
            Graphics2D g = buffImg.createGraphics();
            // 3、设置对线段的锯齿状边缘处理
            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);
            // 4、设置水印旋转
            if (null != degree) {
                g.rotate(Math.toRadians(degree), (double) buffImg.getWidth() / 2, (double) buffImg.getHeight() / 2);
            }
            // 5、设置水印文字颜色
            g.setColor(color);
            // 6、设置水印文字Font
            g.setFont(font);
            // 7、设置水印文字透明度
            g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, alpha));
            // 8、第一参数->设置的内容，后面两个参数->文字在图片上的坐标位置(x,y)
            g.drawString(logoText, positionWidth, positionHeight);
            // 9、释放资源
            g.dispose();
            // 10、生成图片
            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();
            }
        }
    }

}