package com.ruoyi.worker.utils;
|
|
|
import com.ruoyi.worker.entity.MultipartFileDto;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.imageio.ImageIO;
|
|
import java.awt.AlphaComposite;
|
|
import java.awt.Color;
|
|
import java.awt.Font;
|
|
import java.awt.Graphics2D;
|
|
import java.awt.RenderingHints;
|
|
import java.awt.image.BufferedImage;
|
|
import java.io.*;
|
|
/**
|
|
* @author inspur
|
|
* 定义文字加水印
|
|
*/
|
|
public class FileImageWatermarkUtils {
|
|
/**
|
|
* 水印字体
|
|
*/
|
|
private static final Font FONT = new Font("宋体", Font.PLAIN, 40);
|
|
/**
|
|
* 透明度
|
|
*/
|
|
private static final AlphaComposite COMPOSITE = AlphaComposite
|
|
.getInstance(AlphaComposite.SRC_OVER, 0.9f);
|
|
/**
|
|
* 水印之间的间隔
|
|
*/
|
|
private static final int X_MOVE = 150;
|
|
/**
|
|
* 水印之间的间隔
|
|
*/
|
|
private static final int Y_MOVE = 200;
|
|
/**
|
|
* 打水印(文字)
|
|
*
|
|
* @param file MultipartFile
|
|
* @return MultipartFile
|
|
*/
|
|
public static MultipartFile markWithContent(MultipartFile file,String keyword,String address) {
|
|
FileOutputStream fos = null;
|
|
try {
|
|
BufferedImage srcImg = ImageIO.read(file.getInputStream());
|
|
// 图片宽、高
|
|
int imgWidth = srcImg.getWidth();
|
|
int imgHeight = srcImg.getHeight();
|
|
// 图片缓存
|
|
BufferedImage bufImg = new BufferedImage(imgWidth, imgHeight, BufferedImage.TYPE_INT_RGB);
|
|
// 创建绘图工具
|
|
Graphics2D graphics = bufImg.createGraphics();
|
|
// 画入原始图像
|
|
graphics.drawImage(srcImg, 0, 0, imgWidth, imgHeight, null);
|
|
// 设置水印颜色
|
|
graphics.setColor(Color.RED);
|
|
// 设置水印透明度
|
|
graphics.setComposite(COMPOSITE);
|
|
// 设置倾斜角度
|
|
/* graphics.rotate(Math.toRadians(-35), (double) bufImg.getWidth() / 2,
|
|
(double) bufImg.getHeight() / 2);*/
|
|
// 设置水印字体
|
|
graphics.setFont(FONT);
|
|
// 消除java.awt.Font字体的锯齿
|
|
// graphics.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
|
|
int xCoordinate , yCoordinate;
|
int xCoordinate1 , yCoordinate1;
|
int xCoordinate2 , yCoordinate2;
|
|
String mark = "ls65535";
|
|
// 字体长度
|
|
int markWidth = FONT.getSize() * getTextLength(keyword);
|
|
// 字体高度
|
|
int markHeight = FONT.getSize();
|
|
//位置控制 水印位于图片 下方右下角
|
if (address.length()<=23){
|
|
|
xCoordinate = (int) ((imgWidth - getWatermarkLength(keyword, graphics)) * 0.05);
|
|
yCoordinate = (int) (imgHeight * 0.91);
|
|
xCoordinate1 = (int) ((imgWidth - getWatermarkLength(keyword, graphics)) * 0.05);
|
|
yCoordinate1 = (int) (imgHeight * 0.96);
|
|
|
graphics.drawString("地址：" + address, xCoordinate, yCoordinate);
|
graphics.drawString("时间：" + keyword, xCoordinate1, yCoordinate1);
|
|
InputStream inputStream = buffToInputStream(bufImg);
|
|
|
// 释放画图工具
|
|
graphics.dispose();
|
|
return new MultipartFileDto(file.getName(),file.getOriginalFilename(),file.getContentType(),inputStream);
|
|
}else {
|
xCoordinate2 = (int) ((imgWidth - getWatermarkLength(keyword, graphics)) * 0.05);
|
|
yCoordinate2 = (int) (imgHeight * 0.91);
|
|
xCoordinate = (int) ((imgWidth - getWatermarkLength(keyword, graphics)) * 0.05);
|
|
yCoordinate = (int) (imgHeight * 0.88);
|
|
xCoordinate1 = (int) ((imgWidth - getWatermarkLength(keyword, graphics)) * 0.05);
|
|
yCoordinate1 = (int) (imgHeight * 0.96);
|
|
|
graphics.drawString("地址：" + address.substring(0,20), xCoordinate, yCoordinate);
|
graphics.drawString(address.substring(20), xCoordinate2, yCoordinate2);
|
|
graphics.drawString("时间：" + keyword, xCoordinate1, yCoordinate1);
|
|
InputStream inputStream = buffToInputStream(bufImg);
|
|
|
// 释放画图工具
|
|
graphics.dispose();
|
|
return new MultipartFileDto(file.getName(),file.getOriginalFilename(),file.getContentType(),inputStream);
|
}
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
} finally {
|
|
if (fos != null) {
|
|
try {
|
|
fos.flush();
|
|
fos.close();
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
/**
|
|
* 计算水印文本长度
|
|
* 1、中文长度即文本长度 2、英文长度为文本长度二分之一
|
|
* @param text
|
|
* @return
|
|
*/
|
|
public 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;
|
|
}
|
|
public static InputStream buffToInputStream(BufferedImage buffer) throws IOException {
|
|
ByteArrayOutputStream os = new ByteArrayOutputStream();
|
|
ImageIO.write(buffer, "png", os);
|
|
return new ByteArrayInputStream(os.toByteArray());
|
|
}
|
|
public static int getWatermarkLength(String waterMarkContent, Graphics2D g) {
|
|
return g.getFontMetrics(g.getFont()).charsWidth(waterMarkContent.toCharArray(), 0, waterMarkContent.length());
|
|
}
|
|
}
|
