Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
io.github.dengchen2020.core.utils.ImageUtils Maven / Gradle / Ivy
package io.github.dengchen2020.core.utils;
import org.springframework.util.FastByteArrayOutputStream;
import javax.imageio.IIOImage;
import javax.imageio.ImageIO;
import javax.imageio.ImageWriteParam;
import javax.imageio.ImageWriter;
import javax.imageio.stream.ImageOutputStream;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.Base64;
/**
* 图片工具类
*
* @author dengchen
* @since 2023/6/8
*/
public abstract class ImageUtils {
/**
* 将图片压缩
*
* @param inputStream
* @param quality
* @return
*/
public static void compression(InputStream inputStream, float quality, boolean lossless, OutputStream outputStream) {
try (inputStream; outputStream) {
BufferedImage img = ImageIO.read(inputStream);
compression(img, quality, lossless, outputStream);
} catch (IOException e) {
throw new RuntimeException("读取图片文件失败");
}
}
/**
* 压缩图片
*
* @param file
* @param quality
* @return
*/
public static void compression(File file, float quality, boolean lossless, OutputStream outputStream) {
try (outputStream) {
BufferedImage img = ImageIO.read(file);
compression(img, quality, lossless, outputStream);
} catch (IOException e) {
throw new RuntimeException("读取图片文件失败");
}
}
/**
* 将图片流转Base64，会进行压缩
*
* @param inputStream {@link InputStream}
* @return Base64字符串
*/
public static String imgToBase64(InputStream inputStream, float quality, boolean lossless) {
try (inputStream) {
BufferedImage img = ImageIO.read(inputStream);
return imgToBase64(img, quality, lossless);
} catch (IOException e) {
throw new RuntimeException("读取图片文件失败");
}
}
/**
* 将图片文件转Base64，会进行压缩
*
* @param imageFile {@link File}
* @return Base64字符串
*/
public static String imgToBase64(File imageFile, float quality, boolean lossless) {
try {
BufferedImage img = ImageIO.read(imageFile);
return imgToBase64(img, quality, lossless);
} catch (IOException e) {
throw new RuntimeException("读取图片文件失败");
}
}
/**
* 将压缩质量设置为0和1之间的值。默认情况下仅支持单个压缩质量设置; 编写器可以提供ImageWriteParam扩展版本，提供更多控制。对于有损压缩方案，压缩质量应控制文件大小和图像质量之间的权衡 (例如，通过在编写JPEG图像时选择量化表)。对于无损方案，可以使用压缩质量来控制文件大小和执行压缩所花费的时间之间的权衡 (例如，通过优化行过滤器和在写入PNG图像时设置ZLIB压缩级别)。
* 0.0的压缩质量设置最一般地解释为 “高压缩是重要的”，而1.0的设置最一般地解释为 “高图像质量是重要的”。
* 如果有多个压缩类型但没有设置，则会引发IllegalStateException。
* 默认实现检查是否支持压缩，且压缩模式为mode _explicit。如果是这样，如果getCompressionTypes() 返回null或compressionType为非null ，则设置compressionQuality实例变量。
*
* @param image 图片
* @param quality 压缩质量
* @param lossless 是否无损
* @return Base64字符串
*/
private static String imgToBase64(BufferedImage image, float quality, boolean lossless) {
try (FastByteArrayOutputStream baos = new FastByteArrayOutputStream()) {
compression(image, quality, lossless, baos);
return Base64.getEncoder().encodeToString(baos.toByteArray());
}
}
/**
* 将压缩质量设置为0和1之间的值。默认情况下仅支持单个压缩质量设置; 编写器可以提供ImageWriteParam扩展版本，提供更多控制。对于有损压缩方案，压缩质量应控制文件大小和图像质量之间的权衡 (例如，通过在编写JPEG图像时选择量化表)。对于无损方案，可以使用压缩质量来控制文件大小和执行压缩所花费的时间之间的权衡 (例如，通过优化行过滤器和在写入PNG图像时设置ZLIB压缩级别)。
* 0.0的压缩质量设置最一般地解释为 “高压缩是重要的”，而1.0的设置最一般地解释为 “高图像质量是重要的”。
* 如果有多个压缩类型但没有设置，则会引发IllegalStateException。
* 默认实现检查是否支持压缩，且压缩模式为mode _explicit。如果是这样，如果getCompressionTypes() 返回null或compressionType为非null ，则设置compressionQuality实例变量。
*
* @param image 图片
* @param quality 压缩质量
* @param lossless 是否无损
*/
private static void compression(BufferedImage image, float quality, boolean lossless, OutputStream outputStream) {
if (quality >= 1.0) {
quality = 0.9f;
}
if (quality < 0.0) {
quality = 0.0f;
}
ImageWriter writer;
if (lossless) {
writer = ImageIO.getImageWritersByFormatName("png").next();
} else {
writer = ImageIO.getImageWritersByFormatName("jpg").next();
}
ImageWriteParam param = writer.getDefaultWriteParam();
param.setCompressionMode(ImageWriteParam.MODE_EXPLICIT);
if (lossless) {
param.setCompressionType("Deflate");
}
param.setCompressionQuality(quality);
try (ImageOutputStream ios = ImageIO.createImageOutputStream(outputStream)) {
writer.setOutput(ios);
writer.write(null, new IIOImage(image, null, null), param);
writer.dispose();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
/**
* 计算图片sha256
*
* @param data
* @return
*/
public static String calculateSHA256(byte[] data) {
MessageDigest md;
try {
md = MessageDigest.getInstance("SHA-256");
} catch (NoSuchAlgorithmException e) {
throw new RuntimeException("SHA-256算法在当前环境不可用");
}
md.update(data);
byte[] digest = md.digest();
StringBuilder sb = new StringBuilder();
for (byte b : digest) {
sb.append(String.format("%02x", b));
}
return sb.toString();
}
/**
* 添加文本水印
*
* @param imgInputStream 图片输入流
* @param text 文本水印
*/
public static void addTextWatermark(InputStream imgInputStream, String text, OutputStream outputStream) {
addTextWatermark(imgInputStream, text, 10, 30, outputStream);
}
/**
* 添加文本水印
*
* @param inputStream 图片字节数组
* @param text 文本水印
* @param x 水印X轴位置
* @param y 水印Y轴位置
*/
public static void addTextWatermark(InputStream inputStream, String text, int x, int y, OutputStream outputStream) {
try (inputStream; outputStream) {
BufferedImage inputImage = ImageIO.read(inputStream);
int width = inputImage.getWidth();
int height = inputImage.getHeight();
BufferedImage watermarkedImage = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
Graphics2D g2d = (Graphics2D) watermarkedImage.getGraphics();
g2d.drawImage(inputImage, 0, 0, null);
g2d.setColor(Color.BLUE);
g2d.setFont(new Font("SimHei", Font.BOLD, 20));
// 设置支持中文字符
g2d.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON);
g2d.drawString(text, width / 2 - x, height / 2 - y + g2d.getFont().getSize() * 2);
ImageIO.write(watermarkedImage, "jpg", outputStream);
} catch (Exception e) {
throw new RuntimeException("添加水印失败：", e);
}
}
/**
* 添加图片水印
*
* @param imgInputStream 图片输入流
* @param watermarkInputStream 图片水印输入流
*/
public static void addImageWatermark(InputStream imgInputStream, InputStream watermarkInputStream, OutputStream outputStream) {
addImageWatermark(imgInputStream, watermarkInputStream, 10, 15, outputStream);
}
/**
* 添加图片水印
*
* @param imgInputStream 图片字节数组
* @param watermarkInputStream 图片水印
* @param x 水印X轴位置
* @param y 水印Y轴位置
*/
public static void addImageWatermark(InputStream imgInputStream, InputStream watermarkInputStream, int x, int y, OutputStream outputStream) {
try (imgInputStream; watermarkInputStream; outputStream) {
BufferedImage inputImage = ImageIO.read(imgInputStream);
int width = inputImage.getWidth();
int height = inputImage.getHeight();
BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
Graphics2D g2d = (Graphics2D) image.getGraphics();
g2d.drawImage(inputImage, 0, 0, null);
BufferedImage watermarkImage = ImageIO.read(watermarkInputStream);
int watermarkWidth = watermarkImage.getWidth();
int watermarkHeight = watermarkImage.getHeight();
g2d.drawImage(watermarkImage, width / 2 - x, height / 2 - y, watermarkWidth, watermarkHeight, null);
ImageIO.write(image, "jpg", outputStream);
} catch (IOException e) {
throw new RuntimeException("添加水印失败：", e);
}
}
/**
* 调整图片尺寸
*
* @param imageInputStream 图片输入流
* @param targetWidth 宽度
* @param targetHeight 高度
*/
public static void resizeImage(InputStream imageInputStream, int targetWidth, int targetHeight, OutputStream outputStream) throws IOException {
try (imageInputStream; outputStream) {
// 读取原始图片
BufferedImage image = ImageIO.read(imageInputStream);
// 创建一个新的BufferedImage对象，尺寸为新的宽高
BufferedImage resizedImage = new BufferedImage(targetWidth, targetHeight, BufferedImage.TYPE_INT_RGB);
// 使用Graphics2D对象将原始图片绘制到新尺寸的BufferedImage上
Graphics2D g = resizedImage.createGraphics();
g.drawImage(image, 0, 0, targetWidth, targetHeight, null);
g.dispose();
// 将调整尺寸后的图片保存为字节数组
ImageIO.write(resizedImage, "jpg", outputStream);
} catch (IOException e) {
throw new IOException("更改图片尺寸失败", e);
}
}
}
