業務需求:給圖片新增水印,並支持瀏覽器下載導出成zip包。
1、第一反應,通過畫筆給圖片新增加水印,圖片新增水印成功,但是通過IO流導出到瀏覽器下載的時候中文變成方格,檢查了一遍文字格式都是UTF-8的也不知道問題再哪裏。
所以這種方法,暫時只支持單一的對圖片新增水印功能,不支持IO流的讀寫輸出中文亂碼,遇見想通過問題的哪位大佬可以幫答一下。系統是Windows的。
public class waterMarkTest {
public static void main(String[] args) {
System.out.println("開始水印:");
new waterMarkTest().addWatermark("D:/aaa.jpg", "D:/bbb.jpg", "helloword!", "jpg");
System.out.println("水印完成。");
}
/**
* @description
* @param sourceImgPath 源圖片路徑
* @param tarImgPath 保存的圖片路徑
* @param waterMarkContent 水印內容
* @param fileExt 圖片格式
* @return void
*/
public void addWatermark(String sourceImgPath, String tarImgPath, String waterMarkContent,String fileExt){
Font font = new Font("宋體", Font.BOLD, 24);//水印字體,大小
Color markContentColor = Color.white;//水印顏色
Integer degree = -45;//設置水印文字的旋轉角度
float alpha = 1.0f;//設置水印透明度 默認爲1.0 值越小顏色越淺
OutputStream outImgStream = null;
try {
File srcImgFile = new File(sourceImgPath);//得到文件
Image srcImg = ImageIO.read(srcImgFile);//文件轉化爲圖片
int srcImgWidth = srcImg.getWidth(null);//獲取圖片的寬
int srcImgHeight = srcImg.getHeight(null);//獲取圖片的高
// 加水印
BufferedImage bufImg = new BufferedImage(srcImgWidth, srcImgHeight, BufferedImage.TYPE_INT_RGB);
Graphics2D g = bufImg.createGraphics();//得到畫筆
g.drawImage(srcImg, 0, 0, srcImgWidth, srcImgHeight, null);
g.setColor(markContentColor); //設置水印顏色
g.setFont(font); //設置字體
g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, alpha));//設置水印文字透明度
if (null != degree) {
g.rotate(Math.toRadians(degree),(double)bufImg.getWidth(),(double)bufImg.getHeight());//設置水印旋轉
}
JLabel label = new JLabel(waterMarkContent);
FontMetrics metrics = label.getFontMetrics(font);
int width = metrics.stringWidth(label.getText());//文字水印的寬
int rowsNumber = srcImgHeight/width+srcImgHeight%width;// 圖片的高 除以 文字水印的寬 打印的行數(以文字水印的寬爲間隔)
int columnsNumber = srcImgWidth/width+srcImgWidth%width;//圖片的寬 除以 文字水印的寬 每行打印的列數(以文字水印的寬爲間隔)
//防止圖片太小而文字水印太長,所以至少打印一次
if(rowsNumber < 1){
rowsNumber = 1;
}
if(columnsNumber < 1){
columnsNumber = 1;
}
for(int j=0;j<rowsNumber;j++){
for(int i=0;i<columnsNumber;i++){
g.drawString(waterMarkContent, i*width + j*width, -i*width + j*width);//畫出水印,並設置水印位置
}
}
g.dispose();// 釋放資源
// 輸出圖片
outImgStream = new FileOutputStream(tarImgPath);
ImageIO.write(bufImg, fileExt, outImgStream);
} catch (Exception e) {
e.printStackTrace();
e.getMessage();
} finally{
try {
if(outImgStream != null){
outImgStream.flush();
outImgStream.close();
}
} catch (Exception e) {
e.printStackTrace();
e.getMessage();
}
}
}
}
這種方法給圖片新增水印沒問題,但是通過IO流讀取圖片文件數據時水印中的中文文字會亂碼。估計是再IO讀取通過畫筆添加中文水印的方式不識別,所以中文變成方框。
其實圖片已經加成功了,但是IO下載中文變成方框。所以用了第二種方法通過IO先讀圖片存入二進制流,然後把水印信息加進去,再通過IO下載的時候就不會有中文成方框的問題。
2、方法二,通過io給圖片新增水印,得到返回的二進制數據流,可以替換調原文件。
/**
* 圖片加水印
* @param bytes 圖片二進制流
* @param text 文本
* @param extension 擴展名, 如“JPEG”
* @return 加水印後的byte數組
*/
public static byte[] pdfWatermark(byte[] bytes, String text, String extension) {
PDFWatermarkProperties pdfWatermarkProperties=new PDFWatermarkProperties();
pdfWatermarkProperties.setColor( "144,144,144");
pdfWatermarkProperties.setDegree(-45);
pdfWatermarkProperties.setFont("宋體");
pdfWatermarkProperties.setFontBold(true);
pdfWatermarkProperties.setTransparency(0.1f);
ByteArrayInputStream inputStream = new ByteArrayInputStream(bytes);
String ext = extension;
// if (null == extension) {
// // 拓展名
// ext = ImageUtils.getImageFormat(inputStream);
// inputStream = new ByteArrayInputStream(bytes);
// if (null == ext) {
// ext ="jpg";
// }
// }
int fontSize = 110;
int interval = 400;
// String[] rgb = pdfWatermarkProperties.getColor().split(",");
String[] rgb = new String[]{"0","0","0"};
// 水印顏色
Color color = new Color(Integer.valueOf(rgb[0]),Integer.valueOf(rgb[1]),Integer.valueOf(rgb[2]));
try {
Image srcImg = ImageIO.read(inputStream);
// 原始圖寬
int srcImgWidth = srcImg.getWidth(null);
// 原始圖高
int srcImgHeight = srcImg.getHeight(null);
Font font = null;
fontSize = (int)Math.ceil(srcImgWidth*fontSize/(double)2000);
if (pdfWatermarkProperties.getFontBold()) {
font = new Font(pdfWatermarkProperties.getFont(), Font.BOLD, fontSize);
} else {
font = new Font(pdfWatermarkProperties.getFont(), Font.PLAIN, fontSize);
}
// 加水印
BufferedImage bufImg = new BufferedImage(srcImgWidth, srcImgHeight, BufferedImage.TYPE_INT_RGB);
// 畫筆
Graphics2D g = bufImg.createGraphics();
g.drawImage(srcImg, 0, 0, srcImgWidth, srcImgHeight, null);
// 顏色
g.setColor(color);
// 字體
g.setFont(font);
// 設置水印文字透明度
g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, pdfWatermarkProperties.getTransparency()));
g.rotate(Math.toRadians(pdfWatermarkProperties.getDegree()), (double) bufImg.getWidth() / 2, (double) bufImg.getHeight() / 2);
int width = text.length() * fontSize;
int radio = (int) (width * 1.5);
// 繪製圖片位置
for (int y = -1 * interval; y <= bufImg.getHeight() + 2 * interval; y = y + interval + fontSize) {
for (int x = -5 * width; x <= bufImg.getWidth() + 3 * width; x = x + interval + radio) {
g.drawString(text, x, y);
}
}
g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER));
g.dispose();// 釋放資源
// 輸出圖片
ByteArrayOutputStream os = new ByteArrayOutputStream();
ImageIO.write(bufImg, ext, os);
// 輸出圖片
return os.toByteArray();
} catch (Exception e) {
logger.error("", e);
return new byte[0];
}
}
得到圖片新增水印之後的二進制流數據之後,可以替換掉原來的圖片輸出。
byte[] watermark = WatermarkImgUtils.pdfWatermark(bytes, photoWatermark + user.getAccountName(), "jpg");
OutputStream out = new FileOutputStream(pathname.substring(0, pathname.indexOf(".")));
out.write(watermark);
out.flush();
System.out.println("download success");
out.close();