java二维码生成工具

package com.shopwinph.mobile.util;

import java.awt.Color;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.geom.AffineTransform;
import java.awt.image.AffineTransformOp;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

import javax.imageio.ImageIO;

import org.apache.commons.lang.StringUtils;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.WriterException;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;

/**
 * 二维码生成工具
 *
 * @datetime: 2018年11月30日 上午11:14:51
 * @copyright: ShangYun network
 * @author: 10000/sunaiqiang [email protected]
 * @version: 1.0
 */
public class QRCreateUtil {
    /** 二维码中间的LOGO图片设置 */
    /**
     * 宽度
     */
    private static final int IMAGE_WIDTH = 60;
    /**
     * 高度
     */
    private static final int IMAGE_HEIGHT = 60;

    /**
     * 
     */
    private static final int IMAGE_HALF_WIDTH = IMAGE_WIDTH / 2;

    /**
     * 距离二维码
     */
    private static final int FRAME_WIDTH = 6;

    /**
     * 二维码写码器
     */
    private static MultiFormatWriter mutiWriter = new MultiFormatWriter();

    /**
     * 
     * @param waterImage 水印文件地址
     * @param img        目标
     * @param x          x坐标
     * @param y          y坐标
     * @return
     */
    public final static BufferedImage addWaterImage(String waterImage, BufferedImage img, int x, int y) {
        try {
            // 目标文件
            int width = img.getWidth(null);
            int height = img.getHeight(null);
            File filebiao = new File(waterImage);
            BufferedImage image = ImageIO.read(filebiao);
            Graphics g = image.createGraphics();
            g.drawImage(img, x, y, width, height, null);
            g.dispose();
            return image;
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

    /**
     * 生成对应的二维码
     * 
     * @param text      二维码内容
     * @param width     二维码图片宽度
     * @param height    二维码图片高度
     * @param imagePath logo图片地址
     * @return
     * @throws WriterException
     * @throws IOException
     */
    public static BufferedImage createQR(String text, int width, int height, String imagePath)
        throws WriterException, IOException {

        Map<EncodeHintType, Object> hint = new HashMap<EncodeHintType, Object>();
        hint.put(EncodeHintType.CHARACTER_SET, "utf-8");
        hint.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H);
        hint.put(EncodeHintType.MARGIN, 0);
        // 生成二维码
        BitMatrix matrix = mutiWriter.encode(text, BarcodeFormat.QR_CODE, width, height, hint);
        // 输出
        BufferedImage image = null;
        if (StringUtils.isNotBlank(imagePath)) {
            // 读取源图像
            BufferedImage scaleImage = scale(imagePath, IMAGE_WIDTH, IMAGE_HEIGHT, true);
            int[][] srcPixels = new int[IMAGE_WIDTH][IMAGE_HEIGHT];
            for (int i = 0; i < scaleImage.getWidth(); i++) {
                for (int j = 0; j < scaleImage.getHeight(); j++) {
                    srcPixels[i][j] = scaleImage.getRGB(i, j);
                }
            }

            // 二维矩阵转为一维像素数组
            int halfW = matrix.getWidth() / 2;
            int halfH = matrix.getHeight() / 2;
            int[] pixels = new int[width * height];

            for (int y = 0; y < matrix.getHeight(); y++) {
                for (int x = 0; x < matrix.getWidth(); x++) {
                    // 读取图片
                    if (x > halfW - IMAGE_HALF_WIDTH && x < halfW + IMAGE_HALF_WIDTH && y > halfH - IMAGE_HALF_WIDTH
                        && y < halfH + IMAGE_HALF_WIDTH) {
                        pixels[y * width + x] = srcPixels[x - halfW + IMAGE_HALF_WIDTH][y - halfH + IMAGE_HALF_WIDTH];
                    }
                    // 在图片四周形成边框
                    else if ((x > halfW - IMAGE_HALF_WIDTH - FRAME_WIDTH && x < halfW - IMAGE_HALF_WIDTH + FRAME_WIDTH
                        && y > halfH - IMAGE_HALF_WIDTH - FRAME_WIDTH && y < halfH + IMAGE_HALF_WIDTH + FRAME_WIDTH)
                        || (x > halfW + IMAGE_HALF_WIDTH - FRAME_WIDTH && x < halfW + IMAGE_HALF_WIDTH + FRAME_WIDTH
                            && y > halfH - IMAGE_HALF_WIDTH - FRAME_WIDTH && y < halfH + IMAGE_HALF_WIDTH + FRAME_WIDTH)
                        || (x > halfW - IMAGE_HALF_WIDTH - FRAME_WIDTH && x < halfW + IMAGE_HALF_WIDTH + FRAME_WIDTH
                            && y > halfH - IMAGE_HALF_WIDTH - FRAME_WIDTH && y < halfH - IMAGE_HALF_WIDTH + FRAME_WIDTH)
                        || (x > halfW - IMAGE_HALF_WIDTH - FRAME_WIDTH && x < halfW + IMAGE_HALF_WIDTH + FRAME_WIDTH
                            && y > halfH + IMAGE_HALF_WIDTH - FRAME_WIDTH
                            && y < halfH + IMAGE_HALF_WIDTH + FRAME_WIDTH)) {
                        pixels[y * width + x] = 0xfffffff;
                    } else {
                        // 此处可以修改二维码的颜色，可以分别制定二维码和背景的颜色；
                        pixels[y * width + x] = matrix.get(x, y) ? 0xff000000 : 0xfffffff;
                    }
                }
            }
            image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
            image.getRaster().setDataElements(0, 0, width, height, pixels);
        } else {
            image = MatrixToImageWriter.toBufferedImage(matrix);
        }
        return image;
    }

    /**
     * 把传入的原始图像按高度和宽度进行缩放，生成符合要求的图标
     * 
     * @param imageFile 源文件地址
     * @param height    目标高度
     * @param width     目标宽度
     * @param hasFiller 比例不对时是否需要补白：true为补白; false为不补白;
     * @return
     * @throws IOException
     */
    private static BufferedImage scale(String imageFile, int height, int width, boolean hasFiller)
        throws IOException {
        // 缩放比例
        double ratio = 0.0;
        File file = new File(imageFile);
        BufferedImage srcImage = ImageIO.read(file);
        Image destImage = srcImage.getScaledInstance(width, height, BufferedImage.SCALE_SMOOTH);
        // 计算比例
        if ((srcImage.getHeight() > height) || (srcImage.getWidth() > width)) {
            if (srcImage.getHeight() > srcImage.getWidth()) {
                ratio = (new Integer(height)).doubleValue() / srcImage.getHeight();
            } else {
                ratio = (new Integer(width)).doubleValue() / srcImage.getWidth();
            }
            AffineTransformOp op = new AffineTransformOp(AffineTransform.getScaleInstance(ratio, ratio), null);
            destImage = op.filter(srcImage, null);
        }
        if (hasFiller) {
            // 补白
            BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
            Graphics2D graphic = image.createGraphics();
            graphic.setColor(Color.white);
            graphic.fillRect(0, 0, width, height);
            if (width == destImage.getWidth(null)) {
                graphic.drawImage(destImage, 0, (height - destImage.getHeight(null)) / 2, destImage.getWidth(null),
                    destImage.getHeight(null), Color.white, null);
            } else {
                graphic.drawImage(destImage, (width - destImage.getWidth(null)) / 2, 0, destImage.getWidth(null),
                    destImage.getHeight(null), Color.white, null);
                graphic.dispose();
                destImage = image;
            }
        }
        return (BufferedImage) destImage;
    }

    public static void main(String[] args) throws WriterException, IOException {
        // 生成二维码
        BufferedImage img = QRCreateUtil.createQR("http://m.bhzxdy.net/index.php/mobile/mobile/init", 324, 324,
            null);
        // 追加二维码模板
        img = QRCreateUtil.addWaterImage("F:\\le\\code_box.png", img, 132, 318);
        ImageIO.write(img, "png", new File("F:\\le\\generator.png"));
    }

}