package com.fk.util;
import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.geom.AffineTransform;
import java.awt.image.AffineTransformOp;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;
/**
* 组合头像生成
*/
public abstract class UpdateGroupLogoUtil {
/**
* 图片格式:JPG
*/
private static final String PICTRUE_FORMATE = "jpg";
/**
* 图片宽高
*/
private static final Integer PICTURE_SIZE = 30;
/**
* 图片间隔
*/
private static final Integer PICTURE_INTERNAL = 2;
/**
* 生成组合头像
*
* @param paths 用户头像路径列表
* @throws IOException
*/
public static void generate(List<String> paths, String outPath) throws IOException {
generate(paths, PICTURE_SIZE, PICTURE_INTERNAL, outPath);
}
/**
* 生成组合头像
*
* @param paths 用户头像路径列表
* @param length 画板的宽高和高度
* @param interval 画板中的图片间距
* @param outPath 生成后的头像保存路径
* @throws IOException
*/
public static void generate(List<String> paths, int length, int interval, String outPath) throws IOException {
int wh = (length - interval * 4) / 3; // 每个图片的宽高和高度:图片数>6
if (paths.size() == 1) {
wh = length - interval * 2; // 每个图片的宽高和高度:图片数=1
}
if (paths.size() > 1 && paths.size() < 5) {
wh = (length - interval * 3) / 2; // 每个图片的宽高和高度:图片数>0并且<5
}
List<BufferedImage> bufferedImages = new ArrayList<BufferedImage>();
for (int i = 0; i < paths.size(); i++) {
bufferedImages.add(UpdateGroupLogoUtil.resize(paths.get(i), wh, wh, false));
}
// BufferedImage.TYPE_INT_RGB可以自己定义可查看API
BufferedImage outImage = new BufferedImage(length, length, BufferedImage.TYPE_INT_RGB);
// 生成画布
Graphics g = outImage.getGraphics();
Graphics2D g2d = (Graphics2D) g;
// 设置背景色
g2d.setBackground(new Color(221, 222, 224));
// 通过使用当前绘图表面的背景色进行填充来清除指定的矩形。
g2d.clearRect(0, 0, length, length);
// 开始拼凑 根据图片的数量判断该生成那种样式的组合头像
for (int i = 1; i <= bufferedImages.size(); i++) {
int j = i % 3 + 1;
if (bufferedImages.size() < 5) {
j = i % 2 + 1;
}
int x = interval * j + wh * (j - 1);
int split = (wh + interval) / 2;
if (bufferedImages.size() == 9) {
if (i <= 3) {//第一层
g2d.drawImage(bufferedImages.get(i - 1), x + 1, wh * 2 + interval * 3 + 1, null);
} else if (i <= 6) {//第二层
g2d.drawImage(bufferedImages.get(i - 1), x + 1, wh + interval * 2 + 1, null);
} else {//第三层
g2d.drawImage(bufferedImages.get(i - 1), x + 1, interval + 1, null);
}
} else if (bufferedImages.size() == 8) {
if (i <= 3) {
g2d.drawImage(bufferedImages.get(i - 1), x + 1, wh * 2 + interval * 3 + 1, null);
} else if (i <= 6) {
g2d.drawImage(bufferedImages.get(i - 1), x + 1, wh + interval * 2 + 1, null);
} else {
g2d.drawImage(bufferedImages.get(i - 1), x - split + 1, interval + 1, null);
}
} else if (bufferedImages.size() == 7) {
if (i <= 3) {
g2d.drawImage(bufferedImages.get(i - 1), x + 1, wh * 2 + interval * 3 + 1, null);
} else if (i <= 6) {
g2d.drawImage(bufferedImages.get(i - 1), x + 1, wh + interval * 2 + 1, null);
} else {
g2d.drawImage(bufferedImages.get(i - 1), (length - wh) / 2 + 1, interval + 1, null);
}
} else if (bufferedImages.size() == 6) {
if (i <= 3) {
g2d.drawImage(bufferedImages.get(i - 1), x + 1, wh * 2 + interval * 3 - split + 1, null);
} else if (i <= 6) {
g2d.drawImage(bufferedImages.get(i - 1), x + 1, wh + interval * 2 - split + 1, null);
}
} else if (bufferedImages.size() == 5) {
if (i <= 3) {
g2d.drawImage(bufferedImages.get(i - 1), x + 1, wh * 2 + interval * 3 - split, null);
} else {
g2d.drawImage(bufferedImages.get(i - 1), x - split + 1, wh + interval * 2 - split, null);
}
} else if (bufferedImages.size() == 4) {
if (i <= 2) {
g2d.drawImage(bufferedImages.get(i - 1), x, wh + interval * 2, null);
} else {
g2d.drawImage(bufferedImages.get(i - 1), x, interval, null);
}
} else if (bufferedImages.size() == 3) {//三个头像分两层
if (i <= 2) {
g2d.drawImage(bufferedImages.get(i - 1), x, wh + interval * 2, null);
} else {
g2d.drawImage(bufferedImages.get(i - 1), x - split, interval, null);
}
} else if (bufferedImages.size() == 2) {
g2d.drawImage(bufferedImages.get(i - 1), x, wh + interval * 2 - split, null);
} else if (bufferedImages.size() == 1) {
g2d.drawImage(bufferedImages.get(i - 1), interval, interval, null);
}
// 需要改变颜色的话在这里绘上颜色。可能会用到AlphaComposite类
}
ImageIO.write(outImage, PICTRUE_FORMATE, new File(outPath));
}
public static BufferedImage resize(String filePath, int width, int height) {
try {
// BufferedImage bi = ImageIO.read(new File(filePath));
BufferedImage bi = getBufferedImageDestUrl(filePath);
bi.getSubimage(0, 0, width, height);
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
/**
* 图片缩放
*
* @param filePath 图片路径
* @param height 高度
* @param width 宽度
* @param bb 比例不对时是否需要补白
*/
public static BufferedImage resize(String filePath, int height, int width, boolean bb) {
try {
double ratio = 0; // 缩放比例
//File f = new File(filePath);
// BufferedImage bi = ImageIO.read(f);
BufferedImage bi = getBufferedImageDestUrl(filePath);
Image itemp = bi.getScaledInstance(width, height, Image.SCALE_SMOOTH);
// 计算比例
if ((bi.getHeight() > height) || (bi.getWidth() > width)) {
if (bi.getHeight() > bi.getWidth()) {
ratio = (new Integer(height)).doubleValue() / bi.getHeight();
} else {
ratio = (new Integer(width)).doubleValue() / bi.getWidth();
}
AffineTransformOp op = new AffineTransformOp(AffineTransform.getScaleInstance(ratio, ratio), null);
itemp = op.filter(bi, null);
}
if (bb) {
BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
Graphics2D g = image.createGraphics();
g.setColor(Color.white);
g.fillRect(0, 0, width, height);
if (width == itemp.getWidth(null)) {
g.drawImage(itemp, 0, (height - itemp.getHeight(null)) / 2, itemp.getWidth(null), itemp.getHeight(null), Color.white, null);
} else {
g.drawImage(itemp, (width - itemp.getWidth(null)) / 2, 0, itemp.getWidth(null), itemp.getHeight(null), Color.white, null);
}
g.dispose();
itemp = image;
}
return (BufferedImage) itemp;
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
/**
* 远程图片转BufferedImage
* @param destUrl 远程图片地址
* @return
*/
public static BufferedImage getBufferedImageDestUrl(String destUrl) {
HttpURLConnection conn = null;
BufferedImage image = null;
try {
URL url = new URL(destUrl);
conn = (HttpURLConnection) url.openConnection();
if (conn.getResponseCode() == 200) {
image = ImageIO.read(conn.getInputStream());
return image;
}
} catch (Exception e) {
e.printStackTrace();
} finally {
conn.disconnect();
}
return image;
}
public static void main(String[] args) throws Exception {
ArrayList<String> picUrls = new ArrayList<>();
String pic1 = "http://wework.qpic.cn/bizmail/zpmh9YKvHtKoguJibdtpCba98H2s9hzqzGG9Uayd9oQialFj0ZKsV6rg/0";
String pic2 = "http://wework.qpic.cn/bizmail/WYfNrO0wy2xO4q1NZFOOF7FDzicnngSEowPWus74t7BIowf2d56ibMkQ/0";
String pic3 = "http://wework.qpic.cn/bizmail/KRick1k6R9Wx41N73QxLiaXCPtIKQVEl00lu5vz8gOctukiaTAuk0kocw/0";
picUrls.add(pic1);
picUrls.add(pic2);
picUrls.add(pic3);
//2 调用工具类 生成九宫格 并保存在已有路径
UpdateGroupLogoUtil.generate(picUrls, "/Users/name/Downloads/dest.jpg");
}
}