215 lines
6.4 KiB
Plaintext
215 lines
6.4 KiB
Plaintext
package com.bonus.core.qrcode;
|
||
|
||
import java.awt.Color;
|
||
import java.awt.Graphics2D;
|
||
import java.awt.Image;
|
||
import java.awt.image.BufferedImage;
|
||
import java.io.File;
|
||
import java.io.IOException;
|
||
import java.util.Hashtable;
|
||
|
||
import javax.imageio.ImageIO;
|
||
|
||
import com.google.zxing.BarcodeFormat;
|
||
import com.google.zxing.EncodeHintType;
|
||
import com.google.zxing.MultiFormatWriter;
|
||
import com.google.zxing.client.j2se.MatrixToImageConfig;
|
||
import com.google.zxing.client.j2se.MatrixToImageWriter;
|
||
import com.google.zxing.common.BitMatrix;
|
||
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
|
||
|
||
public class QRCodeUtil {
|
||
private static int width = 280; // 二维码宽度
|
||
private static int height = 280; // 二维码高度
|
||
private static int onColor = 0xFF000000; // 前景色
|
||
private static int offColor = 0xFFFFFFFF; // 背景色
|
||
private static int margin = 0; // 白边大小，取值范围0~4
|
||
private static ErrorCorrectionLevel level = ErrorCorrectionLevel.L; // 二维码容错率
|
||
|
||
/**
|
||
* 生成二维码
|
||
*
|
||
* @param params
|
||
* QRCodeParams属性：txt、fileName、filePath不得为空；
|
||
* @throws QRParamsException
|
||
*/
|
||
public static void generateQRImage(QRCodeParams params) throws QRParamsException {
|
||
if (params == null || params.getFileName() == null || params.getFilePath() == null || params.getTxt() == null) {
|
||
|
||
throw new QRParamsException("参数错误");
|
||
}
|
||
try {
|
||
initData(params);
|
||
|
||
String imgPath = params.getFilePath();
|
||
String imgName = params.getFileName();
|
||
String txt = params.getTxt();
|
||
|
||
if (params.getLogoPath() != null && !"".equals(params.getLogoPath().trim())) {
|
||
generateQRImage(txt, params.getLogoPath(), imgPath, imgName, params.getSuffixName());
|
||
} else {
|
||
generateQRImage(txt, imgPath, imgName, params.getSuffixName());
|
||
}
|
||
} catch (Exception e) {
|
||
e.printStackTrace();
|
||
}
|
||
|
||
}
|
||
|
||
/**
|
||
* 生成二维码
|
||
*
|
||
* @param txt
|
||
* //二维码内容
|
||
* @param imgPath
|
||
* //二维码保存物理路径
|
||
* @param imgName
|
||
* //二维码文件名称
|
||
* @param suffix
|
||
* //图片后缀名
|
||
*/
|
||
public static void generateQRImage(String txt, String imgPath, String imgName, String suffix) {
|
||
|
||
File filePath = new File(imgPath);
|
||
if (!filePath.exists()) {
|
||
filePath.mkdirs();
|
||
}
|
||
|
||
File imageFile = new File(imgPath, imgName);
|
||
Hashtable<EncodeHintType, Object> hints = new Hashtable<EncodeHintType, Object>();
|
||
// 指定纠错等级
|
||
hints.put(EncodeHintType.ERROR_CORRECTION, level);
|
||
// 指定编码格式
|
||
hints.put(EncodeHintType.CHARACTER_SET, "UTF-8");
|
||
hints.put(EncodeHintType.MARGIN, margin); // 设置白边
|
||
try {
|
||
MatrixToImageConfig config = new MatrixToImageConfig(onColor, offColor);
|
||
BitMatrix bitMatrix = new MultiFormatWriter().encode(txt, BarcodeFormat.QR_CODE, width, height, hints);
|
||
// bitMatrix = deleteWhite(bitMatrix);
|
||
MatrixToImageWriter.writeToPath(bitMatrix, suffix, imageFile.toPath(), config);
|
||
} catch (Exception e) {
|
||
e.printStackTrace();
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 生成带logo的二维码图片
|
||
*
|
||
* @param txt
|
||
* //二维码内容
|
||
* @param logoPath
|
||
* //logo绝对物理路径
|
||
* @param imgPath
|
||
* //二维码保存绝对物理路径
|
||
* @param imgName
|
||
* //二维码文件名称
|
||
* @param suffix
|
||
* //图片后缀名
|
||
* @throws Exception
|
||
*/
|
||
public static void generateQRImage(String txt, String logoPath, String imgPath, String imgName, String suffix)
|
||
throws Exception {
|
||
File filePath = new File(imgPath);
|
||
if (!filePath.exists()) {
|
||
filePath.mkdirs();
|
||
}
|
||
|
||
if (imgPath.endsWith("/")) {
|
||
imgPath += imgName;
|
||
} else {
|
||
imgPath += "/" + imgName;
|
||
}
|
||
|
||
Hashtable<EncodeHintType, Object> hints = new Hashtable<EncodeHintType, Object>();
|
||
hints.put(EncodeHintType.CHARACTER_SET, "utf-8");
|
||
hints.put(EncodeHintType.ERROR_CORRECTION, level);
|
||
hints.put(EncodeHintType.MARGIN, margin); // 设置白边
|
||
BitMatrix bitMatrix = new MultiFormatWriter().encode(txt, BarcodeFormat.QR_CODE, width, height, hints);
|
||
File qrcodeFile = new File(imgPath);
|
||
writeToFile(bitMatrix, suffix, qrcodeFile, logoPath);
|
||
}
|
||
|
||
/**
|
||
*
|
||
* @param matrix
|
||
* 二维码矩阵相关
|
||
* @param format
|
||
* 二维码图片格式
|
||
* @param file
|
||
* 二维码图片文件
|
||
* @param logoPath
|
||
* logo路径
|
||
* @throws IOException
|
||
*/
|
||
public static void writeToFile(BitMatrix matrix, String format, File file, String logoPath) throws IOException {
|
||
BufferedImage image = toBufferedImage(matrix);
|
||
Graphics2D gs = image.createGraphics();
|
||
|
||
int ratioWidth = image.getWidth() * 2 / 10;
|
||
int ratioHeight = image.getHeight() * 2 / 10;
|
||
// 载入logo
|
||
Image img = ImageIO.read(new File(logoPath));
|
||
int logoWidth = img.getWidth(null) > ratioWidth ? ratioWidth : img.getWidth(null);
|
||
int logoHeight = img.getHeight(null) > ratioHeight ? ratioHeight : img.getHeight(null);
|
||
|
||
int x = (image.getWidth() - logoWidth) / 2;
|
||
int y = (image.getHeight() - logoHeight) / 2;
|
||
|
||
gs.drawImage(img, x, y, logoWidth, logoHeight, null);
|
||
gs.setColor(Color.black);
|
||
gs.setBackground(Color.WHITE);
|
||
gs.dispose();
|
||
img.flush();
|
||
if (!ImageIO.write(image, format, file)) {
|
||
throw new IOException("Could not write an image of format " + format + " to " + file);
|
||
}
|
||
}
|
||
|
||
public static BufferedImage toBufferedImage(BitMatrix matrix) {
|
||
int width = matrix.getWidth();
|
||
int height = matrix.getHeight();
|
||
BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
|
||
|
||
for (int x = 0; x < width; x++) {
|
||
for (int y = 0; y < height; y++) {
|
||
image.setRGB(x, y, matrix.get(x, y) ? onColor : offColor);
|
||
}
|
||
}
|
||
return image;
|
||
}
|
||
|
||
public static BitMatrix deleteWhite(BitMatrix matrix) {
|
||
int[] rec = matrix.getEnclosingRectangle();
|
||
int resWidth = rec[2] + 1;
|
||
int resHeight = rec[3] + 1;
|
||
|
||
BitMatrix resMatrix = new BitMatrix(resWidth, resHeight);
|
||
resMatrix.clear();
|
||
for (int i = 0; i < resWidth; i++) {
|
||
for (int j = 0; j < resHeight; j++) {
|
||
if (matrix.get(i + rec[0], j + rec[1]))
|
||
resMatrix.set(i, j);
|
||
}
|
||
}
|
||
return resMatrix;
|
||
}
|
||
|
||
private static void initData(QRCodeParams params) {
|
||
if (params.getWidth() != null) {
|
||
width = params.getWidth();
|
||
}
|
||
if (params.getHeight() != null) {
|
||
height = params.getHeight();
|
||
}
|
||
if (params.getOnColor() != null) {
|
||
onColor = params.getOnColor();
|
||
}
|
||
if (params.getOffColor() != null) {
|
||
offColor = params.getOffColor();
|
||
}
|
||
if (params.getLevel() != null) {
|
||
level = params.getLevel();
|
||
}
|
||
|
||
}
|
||
} |