Zangzhipeng
2 years ago
parent
f896fca1be
commit
d8a2d8271d
16 changed files with 571 additions and 97 deletions
@ -0,0 +1,176 @@ |
||||
package com.hisense.hiatmp.server_api.utils; |
||||
|
||||
import javax.imageio.ImageIO; |
||||
import java.awt.*; |
||||
import java.awt.image.BufferedImage; |
||||
import java.io.File; |
||||
import java.io.FileOutputStream; |
||||
import java.io.IOException; |
||||
import java.net.URL; |
||||
|
||||
/** |
||||
* 添加水印util |
||||
*/ |
||||
public class WatermarkUtil { |
||||
|
||||
/** |
||||
* 添加水印 |
||||
* |
||||
* @param pictureSourceType 图片来源类型 |
||||
* @param watermarkType 水印类型 |
||||
* @param path 图片路径 |
||||
* @param watermark 水印内容（文字水印内容/图片水印的存放路径） |
||||
* @param formatName 图片格式 |
||||
* @param targetPath 输出图片存放路径 |
||||
* @param color 水印颜色 |
||||
*/ |
||||
public static void addWatermark(PictureSourceType pictureSourceType, WatermarkType watermarkType, String path, String watermark, String formatName, String targetPath, Color color) { |
||||
if (null == pictureSourceType) { |
||||
throw new RuntimeException("图片来源类型不能为空"); |
||||
} |
||||
if (null == watermarkType) { |
||||
throw new RuntimeException("水印类型不能为空"); |
||||
} |
||||
Image image; |
||||
if (pictureSourceType == PictureSourceType.LOCAL_PICTURE) { |
||||
// 读取本地图片
|
||||
image = readLocalPicture(path); |
||||
} else { |
||||
// 读取网络图片
|
||||
image = readNetworkPicture(path); |
||||
} |
||||
// 水印处理
|
||||
manageWatermark(image, watermarkType, watermark, formatName, targetPath, color); |
||||
} |
||||
|
||||
|
||||
//图片来源类型枚举
|
||||
public enum PictureSourceType { |
||||
//本地图片
|
||||
LOCAL_PICTURE, |
||||
//网络图片
|
||||
NETWORK_PICTURE; |
||||
} |
||||
|
||||
//水印类型枚举
|
||||
public enum WatermarkType { |
||||
//文字水印
|
||||
TEXT_WATERMARK, |
||||
//图片水印
|
||||
IMAGE_WATERMARK; |
||||
} |
||||
|
||||
/** |
||||
* 读取本地图片 |
||||
* |
||||
* @param path 本地图片存放路径 |
||||
*/ |
||||
public static Image readLocalPicture(String path) { |
||||
if (null == path) { |
||||
throw new RuntimeException("本地图片路径不能为空"); |
||||
} |
||||
// 读取原图片信息 得到文件
|
||||
File srcImgFile = new File(path); |
||||
try { |
||||
// 将文件对象转化为图片对象
|
||||
return ImageIO.read(srcImgFile); |
||||
} catch (IOException e) { |
||||
throw new RuntimeException(e); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 读取网络图片 |
||||
* |
||||
* @param path 网络图片地址 |
||||
*/ |
||||
public static Image readNetworkPicture(String path) { |
||||
if (null == path) { |
||||
throw new RuntimeException("网络图片路径不能为空"); |
||||
} |
||||
try { |
||||
// 创建一个URL对象,获取网络图片的地址信息
|
||||
URL url = new URL(path); |
||||
// 将URL对象输入流转化为图片对象 (url.openStream()方法，获得一个输入流)
|
||||
BufferedImage bugImg = ImageIO.read(url.openStream()); |
||||
if (null == bugImg) { |
||||
throw new RuntimeException("网络图片地址不正确"); |
||||
} |
||||
return bugImg; |
||||
} catch (IOException e) { |
||||
throw new RuntimeException(e); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 水印处理 |
||||
* |
||||
* @param image 图片对象 |
||||
* @param watermarkType 水印类型（1-文字水印 2-图片水印） |
||||
* @param watermark 水印内容（文字水印内容/图片水印的存放路径） |
||||
* @param formatName 图片格式 |
||||
* @param tarImgPath 输出图片存放路径 |
||||
* @param color 水印颜色 |
||||
*/ |
||||
public static void manageWatermark(Image image, WatermarkType watermarkType, String watermark, String formatName, String tarImgPath, Color color) { |
||||
int imgWidth = image.getWidth(null); |
||||
int imgHeight = image.getHeight(null); |
||||
BufferedImage bufImg = new BufferedImage(imgWidth, imgHeight, BufferedImage.TYPE_INT_RGB); |
||||
// 加水印,创建画笔
|
||||
Graphics2D graphics = bufImg.createGraphics(); |
||||
// 绘制原始图片
|
||||
graphics.drawImage(image, 0, 0, imgWidth, imgHeight, null); |
||||
|
||||
// 校验水印的类型
|
||||
if (watermarkType == WatermarkType.TEXT_WATERMARK) { |
||||
if (watermark.isEmpty()) { |
||||
throw new RuntimeException("文字水印内容不能为空"); |
||||
} |
||||
// 添加文字水印：
|
||||
// 根据图片的背景设置水印颜色
|
||||
graphics.setColor(color == null ? Color.RED : color); |
||||
// 设置字体 画笔字体样式为微软雅黑，加粗，文字大小为45pt
|
||||
graphics.setFont(new Font("微软雅黑", Font.BOLD, 45)); |
||||
// 设置水印的坐标(为原图片中间位置)
|
||||
int x = (imgWidth - getWatermarkLength(watermark, graphics)) / 2; |
||||
int y = imgHeight / 2; |
||||
// 画出水印 第一个参数是水印内容，第二个参数是x轴坐标，第三个参数是y轴坐标
|
||||
graphics.drawString(watermark, x, y); |
||||
graphics.dispose(); |
||||
} else { |
||||
// 添加图片水印：
|
||||
if (watermark.isEmpty()) { |
||||
throw new RuntimeException("图片水印存放路径不能为空"); |
||||
} |
||||
Image srcWatermark = readLocalPicture(watermark); |
||||
int watermarkWidth = srcWatermark.getWidth(null); |
||||
int watermarkHeight = srcWatermark.getHeight(null); |
||||
// 设置 alpha 透明度：alpha 必须是范围 [0.0, 1.0] 之内（包含边界值）的一个浮点数字
|
||||
graphics.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, 0.9f)); |
||||
// 绘制水印图片 坐标为中间位置
|
||||
graphics.drawImage(srcWatermark, (imgWidth - watermarkWidth) / 2, (imgHeight - watermarkHeight) / 2, watermarkWidth, watermarkHeight, null); |
||||
graphics.dispose(); |
||||
} |
||||
// 输出图片
|
||||
try { |
||||
FileOutputStream outImgStream = new FileOutputStream(tarImgPath); |
||||
ImageIO.write(bufImg, formatName, outImgStream); |
||||
outImgStream.flush(); |
||||
outImgStream.close(); |
||||
} catch (IOException e) { |
||||
throw new RuntimeException(e); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 获取水印文字的长度 |
||||
* |
||||
* @param watermarkContent 文字水印内容 |
||||
* @param graphics 图像类 |
||||
*/ |
||||
private static int getWatermarkLength(String watermarkContent, Graphics2D graphics) { |
||||
return graphics.getFontMetrics(graphics.getFont()).charsWidth(watermarkContent.toCharArray(), 0, watermarkContent.length()); |
||||
} |
||||
|
||||
} |
||||
|
||||
Loading…
Reference in new issue