/**
*
*/
/**
* @author
*
*/
package com.qunar.wireless.ugc.caravan.business.las.record;
import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.net.URL;
import java.util.Calendar;
import java.util.List;
import javax.annotation.Resource;
import javax.imageio.ImageIO;
import org.apache.commons.lang.RandomStringUtils;
import org.apache.log4j.Logger;
import org.springframework.stereotype.Component;
import com.qunar.wireless.ugc.caravan.jsonbean.las.response.record.inner.BaseRecordInfo;
import com.qunar.wireless.ugc.caravan.jsonbean.las.response.record.inner.CityInfo;
import com.qunar.wireless.ugc.caravan.jsonbean.las.response.record.inner.FeelingInfo;
import com.qunar.wireless.ugc.exception.DaoException;
import com.qunar.wireless.ugc.exception.ServiceException;
import com.qunar.wireless.ugc.jsonbean.request.upload.UploadedPic;
import com.qunar.wireless.ugc.jsonbean.response.pic.inner.AlbumInfo;
import com.qunar.wireless.ugc.jsonbean.response.pic.inner.PicInfo;
import com.qunar.wireless.ugc.module.comon.Suite;
import com.qunar.wireless.ugc.module.pic.Album;
import com.qunar.wireless.ugc.util.AlbumUtils;
import com.qunar.wireless.ugc.util.PicUploadUtil;
import com.qunar.wireless.ugc.util.PicUtil;
import com.qunar.wireless.ugc.util.StringUtils;
import com.qunar.wireless.ugc.util.WebConstants;
@Component(value="mergePic")
public class MergePic{
@Resource
private AlbumUtils albumUtils;
private int headHeight =0;//头像高度
private int headWidth = 0;//头像宽度
private int innerMargin = 7;//头像内部边距
private int leftMargin = 18;//图片距离左边画布距离
private int margin = 30;//图片之间间隔
private int titleHeight = 150;//标题高度
private int lineHeight = 14;//行高
private int mainCanvasHeight =0;//图片和描述的总高度
private int mainCanvasWidth = 450;//主画布宽度
private static final String backgroundUrlString = "http://img1.qunarzz.com/wugc/p170/201209/03/1531e6d21fa15a6a93835fbb.png";//背景图片
private static final String anchorIconString = "http://img1.qunarzz.com/wugc/p245/201209/03/0a22fedb3d530d1393835fbb.png";
private static final String timeIconString = "http://img1.qunarzz.com/wugc/p182/201209/03/24536348de2d6c1a93835fbb.png";//日期图片
private static final String userHeadUrl = "http://source.qunar.com/site/images/wap/lvtu/lvtu_userhead.gif";
private static final Logger log = Logger.getLogger(MergePic.class);
/**
* @param album
* @param visitUserId
* @param injectAllRecordList 是否注入游记中所有记录列表（会过滤删除和黄反状态）
* @param pureText 照片描述/心情描述是否转换为纯文本，true表示@标签转换为@XXX（被@的用户昵称），</br> 客户端仅显示不可点击；false表示在@标签内加入昵称，仍然是标签的形式，客户端可以点击
* @return
* @throws DaoException
* @throws ServiceException
* @return 图片服务器返回的url
*/
public String createMergedPic(Album album, Long userId, boolean injectAllRecordList,boolean pureText) throws DaoException,
ServiceException {
String result = "";
try {
AlbumInfo albumInfo = albumUtils.getAlbumDetail(album, userId, injectAllRecordList,pureText);
//开始分析头像
String headUrl=albumInfo.getUserInfo().getHeadUrl();
if(headUrl==null){
headUrl = userHeadUrl;
}
URL url = new URL(headUrl);
BufferedImage headBufferedImage = ImageIO.read(url);
headHeight = headBufferedImage.getHeight();//得到头像高度
headWidth = headBufferedImage.getWidth();//得到头像宽度
//start------开始读取游记题目-------------------------------------------------------------------------------------
OutputStream outputStream = getTitle(albumInfo);
ByteArrayInputStream byteArrayInputStream = null;
if(outputStream instanceof ByteArrayOutputStream){
byteArrayInputStream = new ByteArrayInputStream( ((ByteArrayOutputStream) outputStream).toByteArray());
}
BufferedImage titleImage = ImageIO.read(byteArrayInputStream);
//开始分析游记
List<BaseRecordInfo> recordInfoList = albumInfo.getRecordList();
log.info("共生成了"+albumInfo.getRecordTotal()+"个图片或文字!");
for(BaseRecordInfo baseRecordInfo:recordInfoList){
if(baseRecordInfo.getType() == Suite.RecordType.PIC.getValue()){//如果是图片
PicInfo pic = (PicInfo)baseRecordInfo;
String picUrl = PicUtil.getMiniUrl(pic.getUrl(), WebConstants.PIC_SIZE_450_5000);
System.out.println(picUrl);
URL pUrl = new URL(picUrl);
BufferedImage bufferedImage = ImageIO.read(pUrl);
int height= bufferedImage.getHeight();//得到图片高度
mainCanvasHeight +=height;
int textheight = StringUtils.getStringLineNum(pic.getDesc(),48);//假设每行最多显示48个字节
mainCanvasHeight +=textheight;//加上图片描述的高度
mainCanvasHeight += lineHeight;//加上最下面表示时间的一行高度
}else if(baseRecordInfo.getType() == Suite.RecordType.FEELING.getValue()){//如果是心情
FeelingInfo feelingInfo = (FeelingInfo)baseRecordInfo;
String desc = feelingInfo.getDesc();
int textheight = StringUtils.getStringLineNum(desc,48);//假设每行最多显示48个字节
mainCanvasHeight +=textheight;//加上图片描述的高度
mainCanvasHeight += lineHeight;//加上最下面表示时间的一行高度
}
}
//开始生成主画布，在上面生成拼图
BufferedImage mainBufferedImage = new BufferedImage(mainCanvasWidth, mainCanvasHeight, BufferedImage.TYPE_INT_RGB);//创建一个画布
Graphics2D graphics2d = (Graphics2D)mainBufferedImage.getGraphics(); //获取画布的画笔
//生成头像栏
graphics2d.setBackground(new Color(236,230,211));//创建背景色
graphics2d.clearRect(0, 0, mainCanvasWidth, mainCanvasHeight);
graphics2d.setPaint(new Color(129,183,83));
graphics2d.fillRect(0, 0, mainCanvasWidth, headHeight+innerMargin*2);
graphics2d.drawImage(headBufferedImage, leftMargin, innerMargin, null);
graphics2d.setPaint(new Color(246,250,242));
graphics2d.setFont(new Font("宋体",Font.BOLD,20));
graphics2d.drawString("", headWidth+innerMargin*2, headHeight/2+innerMargin);//头像右边的字
//生成标题图片
graphics2d.drawImage(titleImage, leftMargin, headHeight+innerMargin*2+margin, null);
int baseHight = headHeight+innerMargin*2+margin+titleImage.getHeight();//标题图片的高度
//开始生成游记的每个图片和心情图片
for(BaseRecordInfo baseRecordInfo:recordInfoList){
OutputStream oStream = getPicText(baseRecordInfo);
if(oStream instanceof ByteArrayOutputStream){
byteArrayInputStream = new ByteArrayInputStream( ((ByteArrayOutputStream) oStream).toByteArray());
}
BufferedImage picTextmage = ImageIO.read(byteArrayInputStream);
baseHight +=margin;
graphics2d.drawImage(picTextmage, leftMargin, baseHight, null);
baseHight +=picTextmage.getHeight();
}
//将生成后的拼图写到服务器share目录下
String random = RandomStringUtils.random(5, true, true);
String fileName = Calendar.getInstance().getTimeInMillis() + random + ".jpg";
String filePath = PicUtil.getShareFileUploadDir() + File.separator + fileName;
File file = new File(filePath);
boolean f = ImageIO.write(mainBufferedImage, "jpg", file);
if(f){
log.info("生成拼图成功!");
}else{
log.info("生成拼图出错!");
}
//将生成的拼图上传到图片服务器中,并返回图片的CDN地址
List<UploadedPic> uploadedPicList = PicUploadUtil.uploadImg(file);
if (uploadedPicList != null && uploadedPicList.size() > 0) {
UploadedPic uploadedPic = uploadedPicList.get(0);
result = uploadedPic != null && org.apache.commons.lang.StringUtils.isNotBlank(uploadedPic.getImg()) ? PicUtil
.getPicCDNUrl(uploadedPic.getImg()) : result;
}
} catch (DaoException e) {
e.printStackTrace();
throw new DaoException();
} catch (ServiceException e) {
e.printStackTrace();
throw new ServiceException();
} catch (IOException e) {
e.printStackTrace();
}catch (Exception e) {
e.printStackTrace();
}
return result;
}
private OutputStream getPicText(BaseRecordInfo baseRecordInfo) {
OutputStream outputStream = new ByteArrayOutputStream();
try {
int canvasHeight =0;//主画布高度
int height = 0;
int textheight =0;//描述或心情的高度
BufferedImage bufferedImage = null;
String desc ="";
if(baseRecordInfo.getType() == Suite.RecordType.PIC.getValue()){//如果是图片
PicInfo pic = (PicInfo)baseRecordInfo;
desc = pic.getDesc();
String picUrl = PicUtil.getMiniUrl(pic.getUrl(), WebConstants.PIC_SIZE_450_5000);
URL pUrl = new URL(picUrl);
bufferedImage = ImageIO.read(pUrl);
height= bufferedImage.getHeight();//得到图片高度
canvasHeight +=height;
textheight = StringUtils.getStringLineNum(desc,48);//假设每行最多显示48个字节
canvasHeight +=textheight;//加上图片描述的高度
canvasHeight += lineHeight;//加上最下面表示时间的一行高度
}else if(baseRecordInfo.getType() == Suite.RecordType.FEELING.getValue()){//如果是心情
FeelingInfo feelingInfo = (FeelingInfo)baseRecordInfo;
desc = feelingInfo.getDesc();
textheight = StringUtils.getStringLineNum(desc,48);//假设每行最多显示48个字节
canvasHeight +=textheight;//加上图片描述的高度
canvasHeight += lineHeight;//加上最下面表示时间的一行高度
}
BufferedImage mianBufferedImage = new BufferedImage(mainCanvasWidth, canvasHeight, BufferedImage.TYPE_INT_RGB);//创建一个画布
Graphics2D graphics2d = (Graphics2D)mianBufferedImage.getGraphics(); //获取画布的画笔
graphics2d.setBackground(new Color(255,255,255));//创建背景色
graphics2d.clearRect(0, 0, mainCanvasWidth, canvasHeight);
if(baseRecordInfo.getType() == Suite.RecordType.PIC.getValue()){//如果是图片
PicInfo pic = (PicInfo)baseRecordInfo;
String time = pic.getCreateTime();
graphics2d.drawImage(bufferedImage, 0, 0, null);//图片
if(desc != null && !desc.equals("")){//描述
graphics2d.setPaint(new Color(60,60,60));
graphics2d.setFont(new Font("宋体",Font.PLAIN,14));
graphics2d.drawString(desc, leftMargin+4, height);
}
if(pic.getAddressOpenType() ==Suite.WhetherType.YES.getValue()){//POI
URL anchorUrl = new URL(anchorIconString);
BufferedImage anchorBufferedImage = ImageIO.read(anchorUrl);
String addressNameString = pic.getAddressName() +" " +pic.getParentAddressName();
graphics2d.drawImage(anchorBufferedImage, leftMargin, canvasHeight-lineHeight, null);
graphics2d.setPaint(new Color(65,152,167));//地址颜色，TODO
graphics2d.setFont(new Font("宋体",Font.PLAIN,14));
graphics2d.drawString(addressNameString, leftMargin+4, canvasHeight-lineHeight);
}
if(time != null && !time.equals("")){//时间
URL timerUrl = new URL(timeIconString);
BufferedImage timeBufferedImage = ImageIO.read(timerUrl);
graphics2d.drawImage(timeBufferedImage, 250, canvasHeight-lineHeight, null);//时间位置相对固定的
graphics2d.setPaint(new Color(65,152,167));
graphics2d.setFont(new Font("宋体",Font.PLAIN,14));
graphics2d.drawString(time, 270, canvasHeight-lineHeight);
}
}else if(baseRecordInfo.getType() == Suite.RecordType.FEELING.getValue()){//如果是心情
FeelingInfo feelingInfo = (FeelingInfo)baseRecordInfo;
String time = feelingInfo.getCreateTime();
graphics2d.setPaint(new Color(60,60,60));
graphics2d.setFont(new Font("宋体",Font.PLAIN,14));
graphics2d.drawString(desc, 0, 0); //画描述
if(feelingInfo.getAddressOpenType() ==Suite.WhetherType.YES.getValue()){//POI
URL anchorUrl = new URL(anchorIconString);
BufferedImage anchorBufferedImage = ImageIO.read(anchorUrl);
String addressNameString = feelingInfo.getAddressName() +" " +feelingInfo.getParentAddressName();
graphics2d.drawImage(anchorBufferedImage, leftMargin, canvasHeight-lineHeight, null);
graphics2d.setPaint(new Color(65,152,167));//地址颜色，TODO
graphics2d.setFont(new Font("宋体",Font.PLAIN,14));
graphics2d.drawString(addressNameString, leftMargin+4, canvasHeight-lineHeight);
}
if(time != null && !time.equals("")){//时间
URL timerUrl = new URL(timeIconString);
BufferedImage timeBufferedImage = ImageIO.read(timerUrl);
graphics2d.drawImage(timeBufferedImage, 250, canvasHeight-lineHeight, null);//时间位置相对固定的
graphics2d.setPaint(new Color(65,152,167));
graphics2d.setFont(new Font("宋体",Font.PLAIN,14));
graphics2d.drawString(time, 270, canvasHeight-lineHeight);
}
}
boolean f = ImageIO.write(mianBufferedImage, "jpg", outputStream);
if(f){
log.info("生成图片成功!");
}else{
log.info("生成图片出错!");
}
} catch (IOException e) {
log.info("生成图片出错!");
e.printStackTrace();
}
return outputStream;
}
public OutputStream getTitle(AlbumInfo albumInfo){
OutputStream outputStream = new ByteArrayOutputStream();
try {
int width = 430;//标题图片的宽度
int height = 150;//标题图片的高度
String title = albumInfo.getName();
title = StringUtils.truncateStr(title,20);
String dateString = albumInfo.getStartDate();
String tripString = ""+albumInfo.getTravelDayNum()+"天旅程";
String cityString = "";
List<CityInfo> cList = albumInfo.getCityList();
for(CityInfo cityInfo:cList){
cityString +=cityInfo.getName()+" ";
}
URL url = new URL(backgroundUrlString);
BufferedImage headBufferedImage = ImageIO.read(url);
BufferedImage bufferedImage = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);//创建一个画布
Graphics2D graphics2d = (Graphics2D)bufferedImage.getGraphics(); //获取画布的画笔
graphics2d.setBackground(new Color(236,230,211));//创建背景色
graphics2d.clearRect(0, 0, width, height);
graphics2d.drawImage(headBufferedImage, 0, 0, null);
graphics2d.setPaint(new Color(60,60,60));
graphics2d.setFont(new Font("宋体",Font.BOLD,20));
graphics2d.drawString(title, 20, 30);
graphics2d.setFont(new Font("宋体",Font.PLAIN,14));
graphics2d.drawString(dateString, 45, 70);
graphics2d.drawString(tripString, 200, 70);
graphics2d.drawString(cityString, 30, 100);
boolean f = ImageIO.write(bufferedImage, "jpg", outputStream);
if(f){
log.info("生成图片成功!");
}else{
log.info("生成图片出错!");
}
} catch (IOException e) {
log.info("生成图片出错!");
e.printStackTrace();
}
return outputStream;
}
}
java拼图
猜你喜欢
转载自sunway.iteye.com/blog/1671317
今日推荐
周排行