Excel添加水印

Excel添加水印【源码下载】

步骤一：根据生成图片的类创建水印图片

步骤二：

•   代码在Excel中根据第一行获取sheet的列数【sheet.getRow(0).getLastCellNum() 】，根据【sheet.getLastRowNum()】获取整个sheet的行数。
•   使用【sheet.protectSheet(UUID.randomUUID().toString());】为每个sheet添加密码，防止sheet被修改
•   在Excel中添加水印

                    /*
                     * 参数定义： 第一个参数是（x轴的开始节点）； 第二个参数是（是y轴的开始节点）； 第三个参数是（是x轴的结束节点）；
                     * 第四个参数是（是y轴的结束节点）； 第五个参数是（是从Excel的第几列开始插入图片，从0开始计数）；
                     * 第六个参数是（是从excel的第几行开始插入图片，从0开始计数）； 第七个参数是（图片宽度，共多少列）；
                     * 第8个参数是（图片高度，共多少行）；
                     */
                    ClientAnchor anchor = drawing.createAnchor(0, 0, 0, 0, xIndexInteger,
                            yIndexInteger, xIndexInteger+waterRemarkWidth, yIndexInteger+waterRemarkHeight);
                    
                    Picture pic = drawing.createPicture(anchor,
                            wb.addPicture(byteArrayOut.toByteArray(), Workbook.PICTURE_TYPE_PNG));
                    pic.resize();

案例源码：

import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Date;
import java.util.UUID;

import javax.imageio.ImageIO;

import org.apache.poi.ss.usermodel.ClientAnchor;
import org.apache.poi.ss.usermodel.Drawing;
import org.apache.poi.ss.usermodel.Picture;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

public class ExcelWaterRemarkUtils {
    public static void main(String[] args) throws Exception {
        String waterRemarkPath = "E:\NO5\123.png";
        String tmpName = "E:\NO5\测试.xlsx";
        String tmpName2 = "E:\NO5\测试2.xlsx";
        Date a = new Date();
        XSSFWorkbook wb=new XSSFWorkbook(new FileInputStream(tmpName));
        //添加水印
        ExcelWaterRemarkUtils.betweenYRow(wb, waterRemarkPath);
        
        ByteArrayOutputStream os = new ByteArrayOutputStream();
        wb.write(os);
        byte[] content = os.toByteArray();
        File file1 = new File(tmpName2);// Excel文件生成后存储的位置。
        OutputStream fos = new FileOutputStream(file1);
        fos.write(content);
        os.close();
        fos.close();

        Date b = new Date();
        long interval = (b.getTime() - a.getTime())/1000;
        System.out.println("两个时间相差"+interval+"秒");//会打印出相差几秒
        System.out.println("jdldbjo");
    }
    
    /**
     * 为Excel打上水印工具函数 请自行确保参数值，以保证水印图片之间不会覆盖。 在计算水印的位置的时候，并没有考虑到单元格合并的情况，请注意
     * 
     * @param wb
     *            Excel Workbook
     * @param sheet
     *            需要打水印的Excel
     * @param waterRemarkPath
     *            水印地址，classPath，目前只支持png格式的图片，
     *            因为非png格式的图片打到Excel上后可能会有图片变红的问题，且不容易做出透明效果。
     *            同时请注意传入的地址格式，应该为类似："\excelTemplate\test.png"
     * @param startXCol
     *            水印起始列
     * @param startYRow
     *            水印起始行
     * @param betweenXCol
     *            水印横向之间间隔多少列
     * @param betweenYRow
     *            水印纵向之间间隔多少行
     * @param XCount
     *            横向共有水印多少个
     * @param YCount
     *            纵向共有水印多少个
     * @param waterRemarkWidth
     *            水印图片宽度为多少列
     * @param waterRemarkHeight
     *            水印图片高度为多少行
     * @throws IOException
     */
    public static void betweenYRow(Workbook wb, String waterRemarkPath) throws Exception {
        int startXCol=0;         //水印起始列
        int startYRow=0;         // 水印起始行
        int betweenXCol=1;       //水印横向之间间隔多少列
        int betweenYRow=3;       //水印纵向之间间隔多少行
        int XCount=0;            //横向共有水印多少个
        int YCount=0;            //纵向共有水印多少个
        int waterRemarkWidth=0;  //水印图片宽度为多少列
        int waterRemarkHeight=0; // 水印图片高度为多少行
        
        // 校验传入的水印图片格式
        if (!waterRemarkPath.endsWith("png") && !waterRemarkPath.endsWith("PNG")) {
            throw new RuntimeException("向Excel上面打印水印，目前支持png格式的图片。");
        }

        // 加载图片
        ByteArrayOutputStream byteArrayOut = new ByteArrayOutputStream();
        InputStream imageIn=new FileInputStream(waterRemarkPath);
        if (null == imageIn || imageIn.available() < 1) {
            throw new RuntimeException("向Excel上面打印水印，读取水印图片失败(1)。");
        }
        BufferedImage bufferImg = ImageIO.read(imageIn);
        if (null == bufferImg) {
            throw new RuntimeException("向Excel上面打印水印，读取水印图片失败(2)。");
        }
        ImageIO.write(bufferImg, "png", byteArrayOut);
        
        for(int i=0; i<wb.getNumberOfSheets(); i++){  // wb.getNumberOfSheets()
            System.err.println("num:"+ wb.getNumberOfSheets());
            System.err.println("i:"+ i);
            Sheet sheet = wb.getSheetAt(i);
            sheet.protectSheet(UUID.randomUUID().toString());
            try {
                XCount = (sheet.getRow(0).getLastCellNum()); //横向共有水印多少个
                if(XCount == 0){
                    XCount = 2;
                }
                if(XCount >= 5){
                    XCount = 5;
                }
                
            } catch (Exception e) {
                XCount = 10;
            }
            try {
                YCount=sheet.getLastRowNum()/(betweenYRow); //纵向共有水印多少个
                if(YCount < 6){
                    YCount = 6;
                }
                
            } catch (Exception e) {
                YCount = 50;
            }
            
            System.err.println("XCount:"+ XCount);
            System.err.println("YCount:"+ YCount);
            // 开始打水印
            Drawing drawing = sheet.createDrawingPatriarch();
            // 按照共需打印多少行水印进行循环
            for (int yCount = 0; yCount < YCount; yCount++) {
                // 按照每行需要打印多少个水印进行循环
                for (int xCount = 0; xCount < XCount; xCount++) {
                    // 创建水印图片位置
                    int xIndexInteger = startXCol + (xCount * waterRemarkWidth) + (xCount * betweenXCol);
                    int yIndexInteger = startYRow + (yCount * waterRemarkHeight) + (yCount * betweenYRow);
                    /*
                     * 参数定义： 第一个参数是（x轴的开始节点）； 第二个参数是（是y轴的开始节点）； 第三个参数是（是x轴的结束节点）；
                     * 第四个参数是（是y轴的结束节点）； 第五个参数是（是从Excel的第几列开始插入图片，从0开始计数）；
                     * 第六个参数是（是从excel的第几行开始插入图片，从0开始计数）； 第七个参数是（图片宽度，共多少列）；
                     * 第8个参数是（图片高度，共多少行）；
                     */
                    //System.err.println("xIndexInteger:"+ xIndexInteger+", yIndexInteger:"+ yIndexInteger);
                    ClientAnchor anchor = drawing.createAnchor(0, 0, 0, 0, xIndexInteger,
                            yIndexInteger, xIndexInteger+waterRemarkWidth, yIndexInteger+waterRemarkHeight);
                    
                    Picture pic = drawing.createPicture(anchor,
                            wb.addPicture(byteArrayOut.toByteArray(), Workbook.PICTURE_TYPE_PNG));
                    pic.resize();
                    
                }
            }
        }
    }
}

生成水印图片的源码：

/**
     * 
     * @param content  水印文字
     * @param path     图片生成路径
     * @throws IOException 
     * 
     */
    public static void createWaterMark(String content,String path) throws IOException{
        Integer width = 120;
        Integer height = 90;
        BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);// 获取bufferedImage对象
        String fontType = "宋体";
        Integer fontStyle = 0;
        Integer fontSize = 16;
        Font font = new Font(fontType, fontStyle, fontSize);
        Graphics2D g2d = image.createGraphics(); // 获取Graphics2d对象
        image = g2d.getDeviceConfiguration().createCompatibleImage(width, height, Transparency.TRANSLUCENT);
        g2d.dispose();
        g2d = image.createGraphics();
        g2d.setColor(new Color(0, 0, 0, 50)); //设置字体颜色和透明度
        g2d.setStroke(new BasicStroke(1)); // 设置字体
        g2d.setFont(font); // 设置字体类型  加粗 大小
        
        g2d.rotate(Math.toRadians(-10),(double) image.getWidth() / 2, (double) image.getHeight() / 2);//设置倾斜度
        
        FontRenderContext context = g2d.getFontRenderContext();
        Rectangle2D bounds = font.getStringBounds(content, context);
        double x = (width - bounds.getWidth()) / 2;
        double y = (height - bounds.getHeight()) / 2;
        double ascent = -bounds.getY();
        double baseY = y + ascent;
        // 写入水印文字原定高度过小，所以累计写水印，增加高度
        g2d.drawString(content, (int)x, (int)baseY);
        // 设置透明度
        g2d.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER));
        // 释放对象
        g2d.dispose();
        ImageIO.write(image, "png", new File(path));
    }