java生成二维码,并添加logo的工具类,可以改变二维码颜色

添加依赖

        
            com.google.zxing
            javase
            3.3.0
        

QRCodeUtils工具类

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;

import javax.imageio.ImageIO;
import java.awt.*;
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;

public class QRCodeUtil {
    private static final int LOGO_WIDTH = 50;
    private static final int LOGO_HEIGHT = 50;
    private static final int IMAGE_WIDTH = 300;
    private static final int IMAGE_HEIGHT = 300;
    private static final int IMAGE_HALF_WIDTH = LOGO_WIDTH / 2;
    private static final int FRAME_WIDTH = 2;
    private static final String srcImagePath = "图片地址";
    private static final MultiFormatWriter mutiWriter = new MultiFormatWriter();

    public static BufferedImage genBarcode(String content)
            throws WriterException, IOException {
        BufferedImage scaleImage = scale(srcImagePath);
        int[][] srcPixels = new int[LOGO_WIDTH][LOGO_HEIGHT];
        for (int i = 0; i < scaleImage.getWidth(); i++) {
            for (int j = 0; j < scaleImage.getHeight(); j++) {
                srcPixels[i][j] = scaleImage.getRGB(i, j);
            }
        }
        Map hint = new HashMap<>();
        hint.put(EncodeHintType.CHARACTER_SET, "utf-8");
        hint.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H);
        hint.put(EncodeHintType.MARGIN, 0);
        // 生成二维码
        BitMatrix matrix = mutiWriter.encode(content, BarcodeFormat.QR_CODE, IMAGE_WIDTH, IMAGE_HEIGHT, hint);
        // 二维矩阵转为一维像素数组
        int halfW = matrix.getWidth() / 2;
        int halfH = matrix.getHeight() / 2;
        int[] pixels = new int[IMAGE_WIDTH * IMAGE_HEIGHT];
        for (int y = 0; y < matrix.getHeight(); y++) {
            for (int x = 0; x < matrix.getWidth(); x++) {
                // 左上角颜色,根据自己需要调整颜色范围和颜色
                if (x > 0 && x < 55 && y > 0 && y < 55) {
                    Color color = new Color(231, 144, 56);
                    int colorInt = color.getRGB();
                    pixels[y * IMAGE_WIDTH + x] = matrix.get(x, y) ? colorInt : 16777215;
                }
                // 读取图片
                else if (x > halfW - IMAGE_HALF_WIDTH && x < halfW + IMAGE_HALF_WIDTH && y > halfH - IMAGE_HALF_WIDTH
                        && y < halfH + IMAGE_HALF_WIDTH) {
                    pixels[y * IMAGE_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 > halfW - 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 * IMAGE_WIDTH + x] = 0xfffffff;
                    // 在图片四周形成边框
                } else {
                    // 二维码颜色
                    int num1 = (int) (50 - (50.0 - 13.0) / matrix.getHeight() * (y + 1));
                    int num2 = (int) (200 - (165.0 - 72.0) / matrix.getHeight() * (y + 1));
                    int num3 = (int) (162 - (162.0 - 107.0) / matrix.getHeight() * (y + 1));
                    Color color = new Color(num1, num2, num3);
                    int colorInt = color.getRGB();
                    // 此处可以修改二维码的颜色,可以分别制定二维码和背景的颜色;
                    pixels[y * IMAGE_WIDTH + x] = matrix.get(x, y) ? colorInt : 16777215;
                    // 0x000000:0xffffff
                }
            }
        }
        BufferedImage image = new BufferedImage(IMAGE_WIDTH, IMAGE_HEIGHT, BufferedImage.TYPE_INT_RGB);
        image.getRaster().setDataElements(0, 0, IMAGE_WIDTH, IMAGE_HEIGHT, pixels);

        return image;
    }

    private static BufferedImage scale(String srcImageFile) throws IOException {
        // 缩放比例
        double ratio;
        File file = new File(srcImageFile);
        BufferedImage srcImage = ImageIO.read(file);
        Image destImage = srcImage.getScaledInstance(QRCodeUtil.LOGO_WIDTH, QRCodeUtil.LOGO_HEIGHT,
                BufferedImage.SCALE_SMOOTH);
        // 计算比例
        if ((srcImage.getHeight() > QRCodeUtil.LOGO_WIDTH) || (srcImage.getWidth() > QRCodeUtil.LOGO_HEIGHT)) {
            if (srcImage.getHeight() > srcImage.getWidth()) {
                ratio = (new Integer(QRCodeUtil.LOGO_WIDTH)).doubleValue() / srcImage.getHeight();
            } else {
                ratio = (new Integer(QRCodeUtil.LOGO_HEIGHT)).doubleValue() / srcImage.getWidth();
            }
            AffineTransformOp op = new AffineTransformOp(AffineTransform.getScaleInstance(ratio, ratio), null);
            destImage = op.filter(srcImage, null);
        }
        // 补白
        BufferedImage image = new BufferedImage(QRCodeUtil.LOGO_WIDTH, QRCodeUtil.LOGO_HEIGHT,
                BufferedImage.TYPE_INT_RGB);
        Graphics2D graphic = image.createGraphics();
        graphic.setColor(Color.white);
        graphic.fillRect(0, 0, QRCodeUtil.LOGO_WIDTH, QRCodeUtil.LOGO_HEIGHT);
        if (QRCodeUtil.LOGO_HEIGHT == destImage.getWidth(null))
            graphic.drawImage(destImage, 0, (QRCodeUtil.LOGO_WIDTH - destImage.getHeight(null)) / 2,
                    destImage.getWidth(null), destImage.getHeight(null), Color.white, null);
        else
            graphic.drawImage(destImage, (QRCodeUtil.LOGO_HEIGHT - destImage.getWidth(null)) / 2, 0,
                    destImage.getWidth(null), destImage.getHeight(null), Color.white, null);
        graphic.dispose();
        destImage = image;
        return (BufferedImage) destImage;
    }
}

你可能感兴趣的:(java生成二维码,并添加logo的工具类,可以改变二维码颜色)