package com.nova.sankuai.infra.utils;
|
|
import lombok.extern.slf4j.Slf4j;
|
import net.coobird.thumbnailator.Thumbnails;
|
import org.springframework.mock.web.MockMultipartFile;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.imageio.ImageIO;
|
import java.awt.*;
|
import java.awt.geom.AffineTransform;
|
import java.awt.image.BufferedImage;
|
import java.awt.image.ColorModel;
|
import java.awt.image.WritableRaster;
|
import java.io.*;
|
import java.util.UUID;
|
|
@Slf4j
|
public class CompressImageUtil {
|
|
/**
|
* 按尺寸原比例缩放图片
|
*
|
* @throws IOException
|
*/
|
public static MultipartFile imgThumb(MultipartFile file) throws Exception {
|
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
Thumbnails.of(file.getInputStream()).size(100, 100).keepAspectRatio(false).toOutputStream(outputStream);
|
byte[] bytes = outputStream.toByteArray();
|
InputStream inputStream = new ByteArrayInputStream(bytes);
|
MultipartFile multipartFile = new MockMultipartFile(file.getOriginalFilename(), file.getOriginalFilename(), file.getContentType(), inputStream);
|
return multipartFile;
|
}
|
|
/**
|
* 按照比例进行缩放
|
*
|
* @param source 输入源
|
* @param output 输出源
|
* @param scale 比例
|
* @throws IOException
|
*/
|
public static void imgScale(String source, String output, double scale) throws IOException {
|
Thumbnails.of(source).scale(scale).toFile(output);
|
}
|
|
public static File transferToFile(MultipartFile multipartFile) {
|
// 选择用缓冲区来实现这个转换即使用java 创建的临时文件 使用 MultipartFile.transferto()方法 。
|
File file = null;
|
try {
|
String originalFilename = multipartFile.getOriginalFilename();
|
String[] filename = originalFilename.split("\\.");
|
file = File.createTempFile(filename[0], filename[1]);
|
multipartFile.transferTo(file);
|
file.deleteOnExit();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
return file;
|
}
|
|
|
/**
|
* 等比例缩放
|
*
|
* @param srcImg 原图
|
* @param destImg 目标保存位置
|
* @param proportion 压缩比例
|
*/
|
public static String zoomImage(File srcImg, String destImg, Double proportion) {
|
File exist = exist(srcImg, destImg);
|
reSize(srcImg, exist, null, null, proportion);
|
return exist.getName();
|
}
|
|
/**
|
* 固定长度缩放
|
*
|
* @param srcImg 原图
|
* @param destImg 目标保存位置
|
* @param width 期望宽带
|
* @param height 期望长度
|
*/
|
public static String zoomImage(File srcImg, String destImg, Integer width, Integer height) {
|
File exist = exist(srcImg, destImg);
|
reSize(srcImg, exist, width, height, null);
|
return exist.getName();
|
}
|
|
public static File exist(File srcImg, String destImg) {
|
if (srcImg == null) {
|
return null;
|
}
|
String fix = null;
|
String name = srcImg.getName();
|
int i = name.indexOf(".");
|
if (i > 0) {
|
fix = name.substring(i);
|
}
|
name = UUID.randomUUID().toString() + fix;
|
File file = new File(destImg + name);
|
if (!file.getParentFile().exists()) {
|
file.getParentFile().mkdirs();
|
}
|
return file;
|
}
|
|
/**
|
* @param srcImg 原图片
|
* @param destImg 目标位置
|
* @param width 期望宽
|
* @param height 期望高
|
* @param proportion 等比例(0.01-1)(等比例不为空是等比例优先)
|
*/
|
public static void reSize(File srcImg, File destImg, Integer width,
|
Integer height, Double proportion) {
|
String type = getImageType(srcImg);
|
if (type == null) {
|
return;
|
}
|
// if (width < 0 || height < 0) {
|
// return;
|
// }
|
|
BufferedImage srcImage = null;
|
try {
|
srcImage = ImageIO.read(srcImg);
|
System.out.println("srcImg size=" + srcImage.getWidth() + "X" + srcImage.getHeight());
|
} catch (IOException e) {
|
e.printStackTrace();
|
return;
|
}
|
if (srcImage != null) {
|
// targetW，targetH分别表示目标长和宽
|
BufferedImage target = null;
|
Double sx = null;
|
Double sy = null;
|
// 等比缩放
|
if (proportion != null) {
|
sx = sy = proportion;
|
width = (int) (sx * srcImage.getWidth());
|
height = (int) (sy * srcImage.getHeight());
|
} else {
|
sx = (double) width / srcImage.getWidth();
|
sy = (double) height / srcImage.getHeight();
|
}
|
System.out.println("destImg size=" + width + "X" + height);
|
ColorModel cm = srcImage.getColorModel();
|
WritableRaster raster = cm.createCompatibleWritableRaster(width, height);
|
boolean alphaPremultiplied = cm.isAlphaPremultiplied();
|
|
target = new BufferedImage(cm, raster, alphaPremultiplied, null);
|
Graphics2D g = target.createGraphics();
|
// smoother than exlax:
|
g.setRenderingHint(RenderingHints.KEY_RENDERING, RenderingHints.VALUE_RENDER_QUALITY);
|
g.drawRenderedImage(srcImage, AffineTransform.getScaleInstance(sx, sy));
|
g.dispose();
|
// 将转换后的图片保存
|
try {
|
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
ImageIO.write(target, type, baos);
|
FileOutputStream fos = new FileOutputStream(destImg);
|
fos.write(baos.toByteArray());
|
fos.flush();
|
fos.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
|
/**
|
* 获取文件后缀不带.
|
*
|
* @param file 文件后缀名
|
* @return
|
*/
|
public static String getImageType(File file) {
|
if (file != null && file.exists() && file.isFile()) {
|
String fileName = file.getName();
|
int index = fileName.lastIndexOf(".");
|
if (index != -1 && index < fileName.length() - 1) {
|
return fileName.substring(index + 1);
|
}
|
}
|
return null;
|
}
|
}
|
