Java水印工具类

前一段时间在工作中遇到了一个水印功能,即对JPEG图片添加水印(其它文件格式有待测试)。另外,这个工具只是记录下实现过程,并没有做复杂的封装。

 

希望对看到的人有所帮助,也希望大家多提建议。

 

public class ImageAddWaterMark {
        
        public static void main(String[] args) {
                File originImgFile = new File("D:/originImage.jpg");
                File watermarkImgFile = new File("D:/watermarkImage.jpg");
                File targetImgFile = new File("D:/targetImage.jpg");
                addWatermark(originImgFile,watermarkImgFile,targetImgFile);
        }
        
        /**
     * 图片上加水印
     * @param originImgFile 原图
     * @param watermarkImgFile 水印图
     * @param targetImgFile 输出文件
     */
    public static File addWatermark(File originImgFile,File watermarkImgFile,File targetImgFile){
        FileOutputStream fileOutputStream = null;
        try {
            //原图
            Image originImage = ImageIO.read(originImgFile);
            int originWidth = originImage.getWidth(null);//原图长
            int originHeight = originImage.getHeight(null);//原图宽
            //表示一个图像,它具有合成整数像素的 8位 RGB颜色分量.(以原图的长宽作为新图片的长宽)
            BufferedImage image = new BufferedImage(originWidth,originHeight,BufferedImage.TYPE_INT_RGB);
            Graphics g = image.createGraphics();//获取Graphics以渲染BufferedImage
            g.drawImage(originImage,0,0,originWidth,originHeight,null);//从渲染原图
            
            //水印图
            Image waterMarkImage = ImageIO.read(watermarkImgFile);
            int waterMarkWidth = waterMarkImage.getWidth(null);//水印图长
            int waterMarkHeight = waterMarkImage.getHeight(null);//水印图宽
            //在原图指定位置渲染水印图(此处设置为右下角)
            g.drawImage(waterMarkImage,(originWidth-waterMarkWidth),(originHeight-waterMarkHeight),waterMarkWidth,waterMarkHeight,null);
            //释放此图形的上下文以及它使用的所有系统资源
            g.dispose();
            
            ImageWriter imgWrier = ImageIO.getImageWritersByFormatName("jpg").next();
            imgWrier.reset();
            
            fileOutputStream = new FileOutputStream(targetImgFile);//目标文件
            imgWrier.setOutput(ImageIO.createImageOutputStream(fileOutputStream));
            //IIOImage:聚集图像、缩略(预览)图像集和表示与图像关联的元数据对象的简单容器类
            imgWrier.write(new IIOImage(image,null,null));
            fileOutputStream.flush();
            return targetImgFile;
        }catch(Exception e){
                e.printStackTrace();
        }finally {
                try{
                        if(fileOutputStream!=null) fileOutputStream.close();
                }catch(Exception e){
                        e.printStackTrace();
                }
        }
        return null;
    }

}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Java EasyExcel 是一款非常好用的 Excel 操作工具,提供了丰富的 API 接口,可以方便地进行 Excel 文件的读写操作。而在实际应用中,我们可能需要在 Excel 文件中添加一些自定义的水印,以便于区分文件的来源或者保护文件的安全性。本文将介绍如何使用 Java EasyExcel 自定义水印插件。 1. 创建水印插件类 首先,我们需要创建一个自定义的水印插件类,实现 com.alibaba.excel.write.handler.WriteHandler 接口,该接口提供了 beforeSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) 方法,允许我们在生成 Excel 文件之前对 Sheet 进行一些自定义操作。以下是一个简单的水印插件类示例: ```java public class WaterMarkHandler implements WriteHandler { private String waterMark; // 水印内容 public WaterMarkHandler(String waterMark) { this.waterMark = waterMark; } @Override public void beforeSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) { Sheet sheet = writeSheetHolder.getSheet(); Drawing<?> drawing = sheet.createDrawingPatriarch(); // 创建水印文本框 ClientAnchor anchor = drawing.createAnchor(0, 0, 0, 0, 0, 0, 10, 10); Textbox textBox = drawing.createTextbox(anchor); // 设置文本框属性 textBox.setLineStyleColor(255, 255, 255); textBox.setFillColor(255, 255, 255); textBox.setVerticalAlignment(VerticalAlignment.CENTER); textBox.setHorizontalAlignment(HorizontalAlignment.CENTER); textBox.setText(new HSSFRichTextString(waterMark)); } @Override public void afterSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) { // do nothing } } ``` 在上面的代码中,我们通过实现 beforeSheetCreate() 方法,在生成 Sheet 之前创建了一个水印文本框,并设置了文本框的属性和文本内容。 2. 使用水印插件 在使用 Java EasyExcel 生成 Excel 文件时,我们可以将上面定义的 WaterMarkHandler 类作为参数传递给 ExcelWriter 的构造函数,从而实现添加自定义水印的功能。以下是一个简单的示例: ```java public void writeExcelWithWaterMark(OutputStream outputStream, List<List<String>> data, String waterMark) throws IOException { ExcelWriter writer = new ExcelWriter(outputStream, ExcelTypeEnum.XLSX); WriteSheet sheet = new WriteSheet(); sheet.setSheetName("Sheet1"); sheet.setClazz(String.class); sheet.setHead(Collections.emptyList()); sheet.setTableStyle(new TableStyle()); sheet.setWriteHandler(new WaterMarkHandler(waterMark)); // 设置水印插件 writer.write1(data, sheet); writer.finish(); } ``` 在上面的代码中,我们通过调用 setWriteHandler() 方法将 WaterMarkHandler 对象传递给 WriteSheet,从而在生成 Excel 文件时添加自定义水印。完整的代码实现可以参考下面的示例: ```java import com.alibaba.excel.ExcelWriter; import com.alibaba.excel.metadata.TableStyle; import com.alibaba.excel.metadata.VerticalAlignment; import com.alibaba.excel.metadata.WriteSheet; import com.alibaba.excel.write.handler.WriteHandler; import com.alibaba.excel.write.handler.WriteHandlerHolder; import com.alibaba.excel.write.handler.WriteWorkbookHolder; import com.alibaba.excel.write.handler.context.CellWriteHandlerContext; import com.alibaba.excel.write.handler.context.RowWriteHandlerContext; import com.alibaba.excel.write.handler.context.SheetWriteHandlerContext; import org.apache.poi.hssf.usermodel.HSSFRichTextString; import org.apache.poi.ss.usermodel.HorizontalAlignment; import org.apache.poi.ss.usermodel.Sheet; import org.apache.poi.ss.usermodel.VerticalAlignment; import org.apache.poi.ss.usermodel.drawing.ClientAnchor; import org.apache.poi.ss.usermodel.drawing.Drawing; import org.apache.poi.ss.usermodel.drawing.Textbox; import org.apache.poi.ss.util.CellAddress; import java.io.IOException; import java.io.OutputStream; import java.util.Collections; import java.util.List; public class WaterMarkHandler implements WriteHandler { private String waterMark; public WaterMarkHandler(String waterMark) { this.waterMark = waterMark; } @Override public void beforeSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) { Sheet sheet = writeSheetHolder.getSheet(); Drawing<?> drawing = sheet.createDrawingPatriarch(); // 创建水印文本框 ClientAnchor anchor = drawing.createAnchor(0, 0, 0, 0, 0, 0, 10, 10); Textbox textBox = drawing.createTextbox(anchor); // 设置文本框属性 textBox.setLineStyleColor(255, 255, 255); textBox.setFillColor(255, 255, 255); textBox.setVerticalAlignment(VerticalAlignment.CENTER); textBox.setHorizontalAlignment(HorizontalAlignment.CENTER); textBox.setText(new HSSFRichTextString(waterMark)); } @Override public void afterSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) { // do nothing } public static void writeExcelWithWaterMark(OutputStream outputStream, List<List<String>> data, String waterMark) throws IOException { ExcelWriter writer = new ExcelWriter(outputStream, ExcelTypeEnum.XLSX); WriteSheet sheet = new WriteSheet(); sheet.setSheetName("Sheet1"); sheet.setClazz(String.class); sheet.setHead(Collections.emptyList()); sheet.setTableStyle(new TableStyle()); sheet.setWriteHandler(new WaterMarkHandler(waterMark)); writer.write1(data, sheet); writer.finish(); } } ``` 这样,我们就可以使用 Java EasyExcel 实现自定义水印插件的功能了。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值