Skip to content

M
midjourney-proxy
Commit 8b4e8aee by 田超
Options

moidfy pic

parent 8118fd05
Showing with 228 additions and 114 deletions
pcloud-common/src/main/java/com/pcloud/common/utils/PictureUtil.java
...@@ -13,14 +13,12 @@ import org.slf4j.LoggerFactory; ...@@ -13,14 +13,12 @@ import org.slf4j.LoggerFactory;
import javax.imageio.ImageIO; import javax.imageio.ImageIO;
import java.awt.*; import java.awt.*;
import java.awt.geom.RoundRectangle2D;
import java.awt.image.BufferedImage; import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream; import java.io.*;
import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStream;
import java.net.URL; import java.net.URL;
import java.util.ArrayList;
import java.util.HashMap; import java.util.HashMap;
import java.util.List;
import java.util.Map; import java.util.Map;
/** /**
...@@ -36,19 +34,9 @@ public class PictureUtil { ...@@ -36,19 +34,9 @@ public class PictureUtil {
private static String playPic = "https://file.5rs.me/oss/uploadfe/png/3c687bffac4d885ecfff8517d6c3f9c2.png"; private static String playPic = "https://file.5rs.me/oss/uploadfe/png/3c687bffac4d885ecfff8517d6c3f9c2.png";
private static String defaultBackgroundPic = "https://oss.5rs.me/oss/uploadfe/png/3003b8978e85052f96ababdf7d46f70a.png"; private static String defaultBackgroundPic = "https://oss.5rs.me/oss/uploadfe/png/3003b8978e85052f96ababdf7d46f70a.png";
private static String borderPic = "C:\\Users\\LiHao\\Desktop\\封面图\\书本背景.png";
public static Color getImagePixel(String imgFile) {
BufferedImage bi = null; public static Color getImagePixel(BufferedImage bi) {
try {
if(imgFile.contains("http")){
bi = ImageIO.read(new URL(imgFile));
}else{
bi = ImageIO.read(new File(imgFile));
}
} catch (Exception e) {
e.printStackTrace();
}
int width = bi.getWidth(); int width = bi.getWidth();
int height = bi.getHeight(); int height = bi.getHeight();
int sum=width*height; int sum=width*height;
...@@ -57,18 +45,119 @@ public class PictureUtil { ...@@ -57,18 +45,119 @@ public class PictureUtil {
int R = 0; int R = 0;
int G = 0; int G = 0;
int B = 0; int B = 0;
List<String> list=new ArrayList<String>();
for (int i = minx; i < width; i++) { for (int i = minx; i < width; i++) {
for (int j = miny; j < height; j++) { for (int j = miny; j < height; j++) {
int pixel = bi.getRGB(i, j); int pixel = bi.getRGB(i, j);
R +=(pixel & 0xff0000) >> 16; if(pixel==Color.white.getRGB() || pixel==Color.black.getRGB()){
G +=(pixel & 0xff00) >> 8; continue;
B +=(pixel & 0xff); }
R =(pixel & 0xff0000) >> 16;
G =(pixel & 0xff00) >> 8;
B =(pixel & 0xff);
if((R>230&&G>230&&B>230) ||(R<30&&G<30&&B<30)){
continue;
}
list.add(R+"-"+G+"-"+B);
}
}
// R = R/sum;
// G = G/sum;
// B = B/sum;
// return new Color(R,G,B);
return getMaxCount(list);
}
public static Color getMaxCount(List<String> s) {
List<String> list = s;
Map<String, Integer> map = new HashMap<String, Integer>();
for (String c : list) {
Integer l = map.get(c);
if (l == null)
l = 1;
else
l++;
map.put(c, l);
}
String max = null;
long num = 0;
for (Map.Entry<String, Integer> entry : map.entrySet()) {
String key = entry.getKey();
Integer temp = entry.getValue();
if (max == null || temp > num) {
max = key;
num = temp;
} }
} }
R = R/sum; String str[] = max.split("\\-");
G = G/sum; if (str.length == 3) {
B = B/sum; return new Color(Integer.parseInt(str[0]), Integer.parseInt(str[1]),
return new Color(R,G,B); Integer.parseInt(str[2]));
}
return Color.gray;
}
public static String getB16(int R,int G,int B){
String[] h=new String[256];
h[0]="00";h[1]="01";h[2]="02";h[3]="03";h[4]="04";h[5]="05";
h[6]="06";h[7]="07";h[8]="08";h[9]="09";h[10]="0A";h[11]="0B";
h[12]="0C";h[13]="0D";h[14]="0E";h[15]="0F";h[16]="10";h[17]="11";
h[18]="12";h[19]="13";h[20]="14";h[21]="15";h[22]="16";h[23]="17";
h[24]="18";h[25]="19";h[26]="1A";h[27]="1B";h[28]="1C";h[29]="1D";
h[30]="1E";h[31]="1F";h[32]="20";h[33]="21";h[34]="22";h[35]="23";
h[36]="24";h[37]="25";h[38]="26";h[39]="27";h[40]="28";h[41]="29";
h[42]="2A";h[43]="2B";h[44]="2C";h[45]="2D";h[46]="2E";h[47]="2F";
h[48]="30";h[49]="31";h[50]="32";h[51]="33";h[52]="34";h[53]="35";
h[54]="36";h[55]="37";h[56]="38";h[57]="39";h[58]="3A";h[59]="3B";
h[60]="3C";h[61]="3D";h[62]="3E";h[63]="3F";h[64]="40";h[65]="41";
h[66]="42";h[67]="43";h[68]="44";h[69]="45";h[70]="46";h[71]="47";
h[72]="48";h[73]="49";h[74]="4A";h[75]="4B";h[76]="4C";h[77]="4D";
h[78]="4E";h[79]="4F";h[80]="50";h[81]="51";h[82]="52";h[83]="53";
h[84]="54";h[85]="55";h[86]="56";h[87]="57";h[88]="58";h[89]="59";
h[90]="5A";h[91]="5B";h[92]="5C";h[93]="5D";h[94]="5E";h[95]="6F";
h[96]="60";h[97]="61";h[98]="62";h[99]="63";h[100]="64";h[101]="65";
h[102]="66";h[103]="67";h[104]="68";h[105]="69";h[106]="6A";h[107]="6B";
h[108]="6C";h[109]="6D";h[110]="6E";h[111]="6F";h[112]="70";h[113]="71";
h[114]="72";h[115]="73";h[116]="74";h[117]="75";h[118]="76";h[119]="77";
h[120]="78";h[121]="79";h[122]="7A";h[123]="7B";h[124]="7C";h[125]="7D";
h[126]="7E";h[127]="7F";h[128]="80";h[129]="81";h[130]="82";h[131]="83";
h[132]="84";h[133]="85";h[134]="86";h[135]="87";h[136]="88";h[137]="89";
h[138]="8A";h[139]="8B";h[140]="8C";h[141]="8D";h[142]="8E";h[143]="8F";
h[144]="90";h[145]="91";h[146]="92";h[147]="93";h[148]="94";h[149]="95";
h[150]="96";h[151]="97";h[152]="98";h[153]="99";h[154]="9A";h[155]="9B";
h[156]="9C";h[157]="9D";h[158]="9E";h[159]="9F";h[160]="A0";h[161]="A1";
h[162]="A2";h[163]="A3";h[164]="A4";h[165]="A5";h[166]="A6";h[167]="A7";
h[168]="A8";h[169]="A9";h[170]="AA";h[171]="AB";h[172]="AC";h[173]="AD";
h[174]="AE";h[175]="AF";h[176]="B0";h[177]="B1";h[178]="B2";h[179]="B3";
h[180]="B4";h[181]="B5";h[182]="B6";h[183]="B7";h[184]="B8";h[185]="B9";
h[186]="BA";h[187]="BB";h[188]="BC";h[189]="BD";h[190]="BE";h[191]="BF";
h[192]="C0";h[193]="C1";h[194]="C2";h[195]="C3";h[196]="C4";h[197]="C5";
h[198]="C6";h[199]="C7";h[200]="C8";h[201]="C9";h[202]="CA";h[203]="CB";
h[204]="CC";h[205]="CD";h[206]="CE";h[207]="CF";h[208]="D0";h[209]="D1";
h[210]="D2";h[211]="D3";h[212]="D4";h[213]="D5";h[214]="D6";h[215]="D7";
h[216]="D8";h[217]="D9";h[218]="DA";h[219]="DB";h[220]="DC";h[221]="DD";
h[222]="DE";h[223]="DF";h[224]="E0";h[225]="E1";h[226]="E2";h[227]="E3";
h[228]="E4";h[229]="E5";h[230]="E6";h[231]="E7";h[232]="E8";h[233]="E9";
h[234]="EA";h[235]="EB";h[236]="EC";h[237]="ED";h[238]="EE";h[239]="EF";
h[240]="F0";h[241]="F1";h[242]="F2";h[243]="F3";h[244]="F4";h[245]="F5";
h[246]="F6";h[247]="F7";h[248]="F8";h[249]="F9";h[250]="FA";h[251]="FB";
h[252]="FC";h[253]="FD";h[254]="FE";h[255]="FF";
return "#"+h[R] + h[G] + h[B];
}
public static Color getImagePixel(String imgFile) {
BufferedImage bi = null;
try {
if(imgFile.contains("http")){
bi = ImageIO.read(new URL(imgFile));
}else{
bi = ImageIO.read(new File(imgFile));
}
} catch (Exception e) {
e.printStackTrace();
}
return getImagePixel(bi);
} }
public static String toBrowserHexValue(int number) { public static String toBrowserHexValue(int number) {
...@@ -80,47 +169,26 @@ public class PictureUtil { ...@@ -80,47 +169,26 @@ public class PictureUtil {
return builder.toString().toUpperCase(); return builder.toString().toUpperCase();
} }
private static BufferedImage subPicByWhite(BufferedImage picImage){
return picImage.getSubimage(9, 9, 699, 300);
}
public static Color toColorFromString(String colorStr){ public static Color toColorFromString(String colorStr){
colorStr = colorStr.substring(2); colorStr = colorStr.substring(2);
Color color = new Color(Integer.parseInt(colorStr, 16)) ; Color color = new Color(Integer.parseInt(colorStr, 16)) ;
return color; return color;
} }
private static BufferedImage subPic(BufferedImage picImage){ private static BufferedImage subPicByWhite(BufferedImage picImage){
int w=picImage.getWidth(); return picImage.getSubimage(9, 9, 699, 300);
int h=picImage.getHeight();
int x1=w,y1=h,x2=0,y2=0;
for(int i=0;i<w;i++){
for(int j=0;j<h;j++){
int rgb=picImage.getRGB(i, j);
if(rgb!=0){//不透明
if(i<x1) x1=i;
if(j<y1) y1=j;
if(i>x2) x2=i;
if(j>y2) y2=j;
}
}
}
return picImage.getSubimage(x1, y1, x2-x1+1, y2-y1+1);
} }
public static String getQrCover(String bookPic,boolean isDefault){ public static String getQrCover(String bookPic,boolean isDefault){
String resultPath = ""; String resultPath = "";
try { try {
Color color; Color color = null;
if(bookPic.contains("/oss/")){ try {
String s = SimpleHttpUtils.httpGet(bookPic + "?x-oss-process=image/average-hue", null); color = getImagePixel(bookPic);
System.out.println(s); }catch (Exception e){
Map map = JSONUtils.jsonToMap(s, String.class, String.class); color = Color.gray;
String rgb = (String)map.get("RGB");
color = toColorFromString(rgb);
}else{
color = getImagePixel(bookPic);
} }
String lightIcon = "https://oss.5rs.me/oss/uploadfe/png/12dcbc38528d926912c444765d786a9a.png"; String lightIcon = "https://oss.5rs.me/oss/uploadfe/png/12dcbc38528d926912c444765d786a9a.png";
String deepIcon = "https://oss.5rs.me/oss/uploadfe/png/60612a7415bdac8533568ae0c21a94b6.png"; String deepIcon = "https://oss.5rs.me/oss/uploadfe/png/60612a7415bdac8533568ae0c21a94b6.png";
...@@ -150,85 +218,131 @@ public class PictureUtil { ...@@ -150,85 +218,131 @@ public class PictureUtil {
} }
return resultPath; return resultPath;
} }
public static void main(String[] args) { public static void main(String[] args) {
// String bookPic = "https://file.5rs.me/oss/upload/image/jpg/c2c7f0b85f9f4eb683cd86a2945bc5e5.jpg";
String s = "";
String bookPic = "C:\\Users\\LiHao\\Desktop\\封面图\\yy.png";
String playPic = "C:\\Users\\LiHao\\Desktop\\封面图\\播放按钮.png";
String defaultBackgroundPic = "C:\\Users\\LiHao\\Desktop\\封面图\\默认图背景.png";
String resultPath = "C:\\Users\\LiHao\\Desktop\\封面图\\result.jpg";
String oath = "C:\\Users\\LiHao\\Desktop\\封面图\\书本背景.png";
Boolean isDefalut = false; // String bookPic = "https://file.5rs.me/oss/upload/image/jpg/c2c7f0b85f9f4eb683cd86a2945bc5e5.jpg";
try { String bookPic = "https://oss.5rs.me/oss/uploadfe/jpg/e88c2f608fe4365f5c0efd7dda0eb59e.jpg";
//取书封面和播放按钮图 Color color;
BufferedImage bookPicImage; if(bookPic.contains("/oss/")){
BufferedImage playPicImage; String s = SimpleHttpUtils.httpGet(bookPic + "?x-oss-process=image/average-hue", null);
BufferedImage defaultBackgroudImage; System.out.println(s);
BufferedImage oathI; Map map = JSONUtils.jsonToMap(s, String.class, String.class);
if(bookPic.contains("https:")){ String rgb = (String)map.get("RGB");
bookPicImage = ImageIO.read(new URL(bookPic)); color = toColorFromString(rgb);
playPicImage = ImageIO.read(new URL(playPic)); }else{
defaultBackgroudImage = ImageIO.read(new URL(defaultBackgroundPic)); color = getImagePixel(bookPic);
oathI = ImageIO.read(new URL(oath)); }
}else{
bookPicImage = ImageIO.read(new File(bookPic));
playPicImage = ImageIO.read(new File(playPic));
defaultBackgroudImage = ImageIO.read(new File(defaultBackgroundPic));
oathI = ImageIO.read(new File(oath));
}
//画一个空的背景 String lightIcon = "https://oss.5rs.me/oss/uploadfe/png/12dcbc38528d926912c444765d786a9a.png";
BufferedImage bg = new BufferedImage(702, 299, BufferedImage.TYPE_3BYTE_BGR); String deepIcon = "https://oss.5rs.me/oss/uploadfe/png/60612a7415bdac8533568ae0c21a94b6.png";
Graphics2D g = bg.createGraphics(); String icon;
//获取主题色 if(color!=null && (color.getRed()<180||color.getBlue()<180||color.getGreen()<180)){
Color c = getImagePixel(bookPic); icon = lightIcon;
bookPicImage = subPic(bookPicImage); }else{
//画纯色的背景或者默认背景图 icon = deepIcon;
g.setColor(c); }
if(isDefalut){
g.drawImage(defaultBackgroudImage.getScaledInstance(702,196, Image.SCALE_DEFAULT), 0, 0, null);
}else {
g.fillRect(0, 0, 702, 196);
}
g.setColor(new Color(252, 252, 252));
g.fillRect(0, 196, 702, 102);
g.setColor(new Color(221, 221, 221)); String html = "<div style = \"position: relative;width: 700px;height: 300px;border: 0px solid green;background-color: #fcfcfc;text-align: center;\"> <div style = \"position: absolute;width: 100%;height: 66%;background-color: rgb("+color.getRed()+","+color.getGreen()+","+color.getBlue()+");overflow: hidden;\"> <div style = \"position: absolute;top: 20px;right: -20px;background-image: url('"+icon+"');background-size: 100%;width: 219px;height: 149px;\"> </div> </div><div style=\"position: relative;display: inline-block;width: 158px;height: 225px;box-shadow: 0px 4px 14px 2px rgba(14, 47, 70, 0.2);border-radius: 4px;border: solid 1px #ddd; margin-top: 35px;background: url('"+bookPic+"') no-repeat;background-size: cover;background-position: center;\"></div></div>";
//插入书籍图和播放按钮 Map<String,String>header = new HashMap<>();
header.put("Content-Type", "application/json");
Map<String,String>param = new HashMap<>();
param.put("htmlCode",html);
// g.drawImage(oathI.getScaledInstance(185,252, Image.SCALE_DEFAULT), 258, 33, null); String postResult = HttpKit.post("http://192.168.92.111:8214/transcode/htm2img/jpg", JSON.toJSONString(param));
g.drawImage(bookPicImage.getScaledInstance(157,224, Image.SCALE_DEFAULT), 265, 30, null);
g.drawImage(playPicImage.getScaledInstance(80,80, Image.SCALE_DEFAULT), 306, 105, null);
g.drawRect(265,30,172,238);
//保存图片 // String s = SimpleHttpUtils.httpRequest("http://localhost:8214/transcode/htm2img/jpg", param,"POST","utf-8",header);
if(bookPic.contains("https:")) { Map map = JSONUtils.jsonToMap(postResult, String.class, String.class);
ByteArrayOutputStream os = new ByteArrayOutputStream(); String pic = (String)map.get("data");
ImageIO.write(bg, ImageTypeEnum.JPG.value, os); System.out.println(pic);
UploadResultInfo uploadResultInfo = OssUtils.uploadFileByte(os.toByteArray(), UUIDUitl.taskName(), ImageTypeEnum.JPG.value);
os.close();
s = uploadResultInfo == null ? null : uploadResultInfo.getUrl(); try {
}else{ BufferedImage bookPicImage = ImageIO.read(new URL(pic));
OutputStream os = new FileOutputStream(resultPath); BufferedImage bufferedImage = subPicByWhite(bookPicImage);
ImageIO.write(bg, ImageTypeEnum.JPG.value, os); OutputStream os = new FileOutputStream("C:\\Users\\LiHao\\Desktop\\封面图\\result.jpg");
os.close(); ImageIO.write(bufferedImage, ImageTypeEnum.JPG.value, os);
} os.close();
System.out.println(s); } catch (IOException e) {
} catch (Exception e) { e.printStackTrace();
System.out.println(e.getMessage());
} }
// String bookPic = "C:\\Users\\LiHao\\Desktop\\封面图\\aa.jpg";
// String playPic = "C:\\Users\\LiHao\\Desktop\\封面图\\播放按钮.png";
// String defaultBackgroundPic = "C:\\Users\\LiHao\\Desktop\\封面图\\默认图背景.png";
// String resultPath = "C:\\Users\\LiHao\\Desktop\\封面图\\result.jpg";
// Boolean isDefalut = false;
// try {
// //取书封面和播放按钮图
// BufferedImage bookPicImage;
// BufferedImage playPicImage;
// BufferedImage defaultBackgroudImage;
// if(bookPic.contains("https:")){
// bookPicImage = ImageIO.read(new URL(bookPic));
// playPicImage = ImageIO.read(new URL(playPic));
// defaultBackgroudImage = ImageIO.read(new URL(defaultBackgroundPic));
// }else{
// bookPicImage = ImageIO.read(new File(bookPic));
// playPicImage = ImageIO.read(new File(playPic));
// defaultBackgroudImage = ImageIO.read(new File(defaultBackgroundPic));
// }
//
// //画一个空的背景
// BufferedImage bg = new BufferedImage(702, 299, BufferedImage.TYPE_3BYTE_BGR);
// Graphics2D g = bg.createGraphics();
// //获取主题色
// Color c = getImagePixel(bookPic);
//
//
// //画纯色的背景或者默认背景图
// g.setColor(c);
// if(isDefalut){
// g.drawImage(defaultBackgroudImage.getScaledInstance(702,196, Image.SCALE_DEFAULT), 0, 0, null);
// }else {
// g.fillRect(0, 0, 702, 196);
// }
// g.setColor(new Color(252, 252, 252));
// g.fillRect(0, 196, 702, 102);
//
// g.setColor(new Color(221, 221, 221));
// //插入书籍图和播放按钮
// g.drawImage(bookPicImage.getScaledInstance(172,238, Image.SCALE_DEFAULT), 265, 30, null);
// g.drawImage(playPicImage.getScaledInstance(80,80, Image.SCALE_DEFAULT), 306, 105, null);
// g.drawRect(265,30,172,238);
//
// for(int i=0;i<3;i++){
//
// }
//
// //保存图片
// if(bookPic.contains("https:")) {
// ByteArrayOutputStream os = new ByteArrayOutputStream();
// ImageIO.write(bg, ImageTypeEnum.JPG.value, os);
// UploadResultInfo uploadResultInfo = OssUtils.uploadFileByte(os.toByteArray(), UUIDUitl.taskName(), ImageTypeEnum.JPG.value);
// os.close();
// s = uploadResultInfo == null ? null : uploadResultInfo.getUrl();
// }else{
// OutputStream os = new FileOutputStream(resultPath);
// ImageIO.write(bg, ImageTypeEnum.JPG.value, os);
// os.close();
// }
// System.out.println(s);
// } catch (Exception e) {
// System.out.println(e.getMessage());
// }
} }
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment