优化生成AsciiArt字符画二维码

This commit is contained in:
TomXin 2022-08-21 21:43:15 +08:00
parent 7a233005e8
commit b5b1418c3e
2 changed files with 77 additions and 28 deletions

View File

@ -3,7 +3,9 @@ package cn.hutool.extra.qrcode;
import cn.hutool.core.codec.Base64;
import cn.hutool.core.img.Img;
import cn.hutool.core.img.ImgUtil;
import cn.hutool.core.lang.ansi.*;
import cn.hutool.core.util.CharsetUtil;
import cn.hutool.core.util.NumberUtil;
import cn.hutool.core.util.StrUtil;
import com.google.zxing.BarcodeFormat;
import com.google.zxing.Binarizer;
@ -109,14 +111,19 @@ public class QrCodeUtil {
}
/**
* 生成ASCII Art字符画形式的二维码
*
* @param content 内容
* @return ASCII Art字符画形式的二维码
* @return ASCII Art字符画形式的二维码字符串
* @since 5.8.6
*/
public static String generateAsAsciiArt(String content) {
return generateAsAsciiArt(content, 0, 0, 1);
}
/**
* 生成ASCII Art字符画形式的二维码
*
* @param content 内容
* @param qrConfig 二维码配置仅长边距配置有效
* @return ASCII Art字符画形式的二维码
@ -124,7 +131,7 @@ public class QrCodeUtil {
*/
public static String generateAsAsciiArt(String content, QrConfig qrConfig) {
BitMatrix bitMatrix = encode(content, qrConfig);
return toAsciiArt(bitMatrix);
return toAsciiArt(bitMatrix, qrConfig);
}
/**
@ -464,6 +471,8 @@ public class QrCodeUtil {
}
/**
* BitMatrix转SVG(字符串)
*
* @param matrix BitMatrix
* @param qrConfig 二维码配置包括长边距颜色等
* @return SVG矢量图字符串
@ -528,34 +537,65 @@ public class QrCodeUtil {
}
/**
* BitMatrix转ASCII Art字符画形式的二维码
*
* @param bitMatrix
* @return ASCII Art字符画形式的二维码
* @since 5.8.6
*/
public static String toAsciiArt(BitMatrix bitMatrix) {
public static String toAsciiArt(BitMatrix bitMatrix, QrConfig qrConfig) {
int width = bitMatrix.getWidth();
int height = bitMatrix.getHeight();
StringBuilder result = new StringBuilder(height * (width + 1));
AnsiElement foreground = Ansi8BitColor.foreground(rgbToAnsi8BitValue(qrConfig.getForeColor()));
AnsiElement background = Ansi8BitColor.background(rgbToAnsi8BitValue(qrConfig.getBackColor()));
StringBuilder builder = new StringBuilder();
for (int i = 0; i <= height; i += 2) {
StringBuilder rowBuilder = new StringBuilder();
for (int j = 0; j < width; j++) {
boolean tp = bitMatrix.get(i, j);
boolean bt = i + 1 >= height || bitMatrix.get(i + 1, j);
if (tp && bt) {
result.append(' ');//'\u0020'
rowBuilder.append(' ');//'\u0020'
} else if (tp) {
result.append('▄');//'\u2584'
rowBuilder.append('▄');//'\u2584'
} else if (bt) {
result.append('▀');//'\u2580'
rowBuilder.append('▀');//'\u2580'
} else {
result.append('█');//'\u2588'
rowBuilder.append('█');//'\u2588'
}
}
builder.append(AnsiEncoder.encode(foreground, background, rowBuilder)).append('\n');
}
return builder.toString();
}
/**
* rgb转Ansi8Bit值
*
* @param rgb rgb颜色值
* @return Ansi8bit颜色值
* @since 5.8.6
*/
private static int rgbToAnsi8BitValue(int rgb) {
int l;
int r = (rgb >> 16) & 0xff;
int g = (rgb >> 8) & 0xff;
int b = (rgb) & 0xff;
if (r < 0) r += 256;
if (g < 0) g += 256;
if (b < 0) b += 256;
if (r == g && g == b) {
int i = (int) (NumberUtil.div(NumberUtil.mul(r - 10.625, 23), (255 - 10.625), 0));
l = i >= 0 ? 232 + i : 0;
} else {
l = 16 + (int) (36 * NumberUtil.div(NumberUtil.mul(r, 5), 255, 0)) + (int) (6.0 * (g / 256.0 * 6.0)) + (int) (b / 256.0 * 6.0);
}
result.append('\n');
}
return result.toString();
return l;
}
/**
* 创建解码选项
*

View File

@ -4,6 +4,8 @@ import cn.hutool.core.codec.Base64;
import cn.hutool.core.img.ImgUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.lang.Console;
import cn.hutool.core.lang.ansi.Ansi8BitColor;
import cn.hutool.core.lang.ansi.AnsiEncoder;
import com.google.zxing.BarcodeFormat;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.datamatrix.encoder.SymbolShapeHint;
@ -112,7 +114,10 @@ public class QrCodeUtilTest {
@Test
public void generateSvgTest() {
QrConfig qrConfig = QrConfig.create().setImg("d:/test/logo.png").setForeColor(Color.GREEN).setBackColor(Color.pink);
QrConfig qrConfig = QrConfig.create().setImg("d:/test/logo.png")
.setForeColor(Color.GREEN)
.setBackColor(Color.pink)
.setMargin(1);
String svg = QrCodeUtil.generateAsSvg("https://hutool.cn/", qrConfig);
Assert.assertNotNull(svg);
FileUtil.writeString(svg, "d:/test/qr.svg", StandardCharsets.UTF_8);
@ -120,8 +125,12 @@ public class QrCodeUtilTest {
@Test
public void generateAsciiArtTest() {
QrConfig qrConfig = QrConfig.create();
String asciiArt = QrCodeUtil.generateAsAsciiArt("https://hutool.cn/呱呱呱呱呱呱呱呱呱呱呱呱呱呱呱古古怪怪");
QrConfig qrConfig = QrConfig.create()
.setForeColor(new Color(0,255,0))
.setBackColor(new Color(0,0,0))
.setWidth(0)
.setHeight(0).setMargin(1);
String asciiArt = QrCodeUtil.generateAsAsciiArt("https://hutool.cn/",qrConfig);
Assert.assertNotNull(asciiArt);
System.out.println(asciiArt);
}