給圖片添加水印,並支持瀏覽器下載輸出

業務需求:給圖片新增水印,並支持瀏覽器下載導出成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();

 

 

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章