Files
WAI_Project_UNIX/uni_modules/cool-ui/components/cl-qrcode/draw.ts

392 lines
11 KiB
TypeScript
Raw Normal View History

2025-07-21 16:47:04 +08:00
/**
*
*/
import { isNull } from "@/cool";
import { generateFrame } from "./qrcode";
import type { ClQrcodeMode } from "../../types";
/**
*
*
*/
export type QrcodeOptions = {
ecc: string; // 纠错级别,可选 L/M/Q/H,纠错能力依次增强
text: string; // 二维码内容,要编码的文本
size: number; // 二维码尺寸,单位px
foreground: string; // 前景色,二维码数据点的颜色
background: string; // 背景色,二维码背景的颜色
padding: number; // 内边距,二维码四周留白的距离
logo: string; // logo图片地址,可以在二维码中心显示logo
logoSize: number; // logo尺寸,logo图片的显示大小
mode: ClQrcodeMode; // 二维码样式模式,支持矩形、圆形、线条、小方块
pdColor: string | null; // 定位点颜色,三个角上定位图案的颜色,为null时使用前景色
pdRadius: number; // 定位图案圆角半径,为0时绘制直角矩形
};
/**
*
*
* @returns
*/
function getRatio() {
// #ifdef APP || MP-WEIXIN
return uni.getWindowInfo().pixelRatio; // App和小程序环境
// #endif
// #ifdef H5
return window.devicePixelRatio; // H5环境
// #endif
}
/**
*
*
* @param ctx Canvas上下文
* @param x x坐标
* @param y y坐标
* @param width
* @param height
* @param radius
*/
function drawRoundedRect(
ctx: CanvasRenderingContext2D,
x: number,
y: number,
width: number,
height: number,
radius: number
) {
if (radius <= 0) {
// 圆角半径为0时直接绘制矩形
ctx.fillRect(x, y, width, height);
return;
}
// 限制圆角半径不超过矩形的一半
const maxRadius = Math.min(width, height) / 2;
const r = Math.min(radius, maxRadius);
ctx.beginPath();
ctx.moveTo(x + r, y);
ctx.lineTo(x + width - r, y);
ctx.arcTo(x + width, y, x + width, y + r, r);
ctx.lineTo(x + width, y + height - r);
ctx.arcTo(x + width, y + height, x + width - r, y + height, r);
ctx.lineTo(x + r, y + height);
ctx.arcTo(x, y + height, x, y + height - r, r);
ctx.lineTo(x, y + r);
ctx.arcTo(x, y, x + r, y, r);
ctx.closePath();
ctx.fill();
}
/**
*
* 7x7的定位图案,
* @param ctx Canvas上下文
* @param startX X坐标
* @param startY Y坐标
* @param px
* @param pdColor
* @param background
* @param radius
*/
function drawPositionPattern(
ctx: CanvasRenderingContext2D,
startX: number,
startY: number,
px: number,
pdColor: string,
background: string,
radius: number
) {
const patternSize = px * 7; // 定位图案总尺寸 7x7
// 绘制外层边框 (7x7)
ctx.fillStyle = pdColor;
drawRoundedRect(ctx, startX, startY, patternSize, patternSize, radius);
// 绘制内层空心区域 (5x5)
ctx.fillStyle = background;
const innerStartX = startX + px;
const innerStartY = startY + px;
const innerSize = px * 5;
const innerRadius = Math.max(0, radius - px); // 内层圆角适当减小
drawRoundedRect(ctx, innerStartX, innerStartY, innerSize, innerSize, innerRadius);
// 绘制中心实心区域 (3x3)
ctx.fillStyle = pdColor;
const centerStartX = startX + px * 2;
const centerStartY = startY + px * 2;
const centerSize = px * 3;
const centerRadius = Math.max(0, radius - px * 2); // 中心圆角适当减小
drawRoundedRect(ctx, centerStartX, centerStartY, centerSize, centerSize, centerRadius);
}
/**
* Canvas上下文
* ,
* @param context Canvas上下文对象
* @param options
*/
export function drawQrcode(context: CanvasContext, options: QrcodeOptions) {
// 获取2D绘图上下文
const ctx = context.getContext("2d")!;
if (isNull(ctx)) return;
// 获取设备像素比,用于高清适配
const ratio = getRatio();
// App和小程序平台的画布初始化
// #ifdef APP || MP-WEIXIN
const c1 = ctx.canvas;
// 清空画布并设置尺寸
ctx.clearRect(0, 0, c1.offsetWidth, c1.offsetHeight);
c1.width = c1.offsetWidth * ratio;
c1.height = c1.offsetHeight * ratio;
// #endif
// #ifdef APP
ctx.reset();
// #endif
// H5平台的画布初始化
// #ifdef H5
const c2 = context as HTMLCanvasElement;
c2.width = c2.offsetWidth * ratio;
c2.height = c2.offsetHeight * ratio;
// #endif
// 缩放画布以适配高分屏
ctx.scale(ratio, ratio);
// 生成二维码数据矩阵
const frame = generateFrame(options.text, options.ecc);
const points = frame.frameBuffer; // 点阵数据
const width = frame.width; // 矩阵宽度
// 计算二维码内容区域大小减去四周的padding
const contentSize = options.size - options.padding * 2;
// 计算每个数据点的实际像素大小
const px = contentSize / width;
// 二维码内容的起始位置考虑padding
const offsetX = options.padding;
const offsetY = options.padding;
// 绘制整个画布背景
ctx.fillStyle = options.background;
ctx.fillRect(0, 0, options.size, options.size);
/**
*
* 7x7的方块
* @param i
* @param j
* @param width
* @returns
*/
function isPositionDetectionPattern(i: number, j: number, width: number): boolean {
// 判断三个角的定位图案(7x7)
if (i < 7 && j < 7) return true; // 左上角
if (i > width - 8 && j < 7) return true; // 右上角
if (i < 7 && j > width - 8) return true; // 左下角
return false;
}
/**
* Logo区域内()
* @param i
* @param j
* @param width
* @param logoSize logo尺寸()
* @param px
* @returns logo区域内
*/
function isInLogoArea(
i: number,
j: number,
width: number,
logoSize: number,
px: number
): boolean {
if (logoSize <= 0) return false;
// 计算logo在矩阵中占用的点数
const logoPoints = Math.ceil(logoSize / px);
// 添加缓冲区logo周围额外空出1个点的距离
const buffer = 1;
const totalLogoPoints = logoPoints + buffer * 2;
// 计算logo区域在矩阵中的中心位置
const centerI = Math.floor(width / 2);
const centerJ = Math.floor(width / 2);
// 计算logo区域的边界
const halfSize = Math.floor(totalLogoPoints / 2);
const minI = centerI - halfSize;
const maxI = centerI + halfSize;
const minJ = centerJ - halfSize;
const maxJ = centerJ + halfSize;
// 判断当前点是否在logo区域内
return i >= minI && i <= maxI && j >= minJ && j <= maxJ;
}
// 先绘制定位图案
const pdColor = options.pdColor ?? options.foreground;
const radius = options.pdRadius;
// 绘制三个定位图案
// 左上角 (0, 0)
drawPositionPattern(ctx, offsetX, offsetY, px, pdColor, options.background, radius);
// 右上角 (width-7, 0)
drawPositionPattern(
ctx,
offsetX + (width - 7) * px,
offsetY,
px,
pdColor,
options.background,
radius
);
// 左下角 (0, width-7)
drawPositionPattern(
ctx,
offsetX,
offsetY + (width - 7) * px,
px,
pdColor,
options.background,
radius
);
// 点的间距,用于圆形和小方块模式
const dot = px * 0.1;
// 遍历绘制数据点(跳过定位图案区域和logo区域)
for (let i = 0; i < width; i++) {
for (let j = 0; j < width; j++) {
if (points[j * width + i] > 0) {
// 跳过定位图案区域
if (isPositionDetectionPattern(i, j, width)) {
continue;
}
// 跳过logo区域(包含缓冲区)
if (options.logo != "" && isInLogoArea(i, j, width, options.logoSize, px)) {
continue;
}
// 绘制数据点
ctx.fillStyle = options.foreground;
const x = offsetX + px * i;
const y = offsetY + px * j;
// 根据不同模式绘制数据点
switch (options.mode) {
case "line": // 线条模式 - 绘制水平线条
ctx.fillRect(x, y, px, px / 2);
break;
case "circular": // 圆形模式 - 绘制圆点
ctx.beginPath();
const rx = x + px / 2 - dot;
const ry = y + px / 2 - dot;
ctx.arc(rx, ry, px / 2 - dot, 0, 2 * Math.PI);
ctx.fill();
ctx.closePath();
break;
case "rectSmall": // 小方块模式 - 绘制小一号的方块
ctx.fillRect(x + dot, y + dot, px - dot * 2, px - dot * 2);
break;
default: // 默认实心方块模式
ctx.fillRect(x, y, px, px);
}
}
}
}
// 绘制 Logo
if (options.logo != "") {
let img: Image;
// 微信小程序环境创建图片
// #ifdef MP-WEIXIN || APP-HARMONY
img = context.createImage();
// #endif
// 其他环境创建图片
// #ifndef MP-WEIXIN || APP-HARMONY
img = new Image(options.logoSize, options.logoSize);
// #endif
// 设置图片源并在加载完成后绘制
img.src = options.logo;
img.onload = () => {
drawLogo(ctx, options, img);
};
}
}
/**
* Logo
* Logo图片,
* @param ctx Canvas上下文
* @param options
* @param img Logo图片对象
*/
function drawLogo(ctx: CanvasRenderingContext2D, options: QrcodeOptions, img: Image) {
ctx.save(); // 保存当前绘图状态
// 计算二维码内容区域的中心位置考虑padding
const contentSize = options.size - options.padding * 2;
const contentCenterX = options.padding + contentSize / 2;
const contentCenterY = options.padding + contentSize / 2;
// 添加额外的背景边距,与数据点避让区域保持一致
const backgroundPadding = 6; // 背景比logo大6px
const backgroundSize = options.logoSize + backgroundPadding * 2;
// 绘制白色背景作为Logo的底色稍大于logo
ctx.fillStyle = "#fff";
const backgroundX = contentCenterX - backgroundSize / 2;
const backgroundY = contentCenterY - backgroundSize / 2;
ctx.fillRect(backgroundX, backgroundY, backgroundSize, backgroundSize);
// 获取图片信息后绘制Logo
uni.getImageInfo({
src: options.logo,
success: (imgInfo) => {
// 计算logo的精确位置
const logoX = contentCenterX - options.logoSize / 2;
const logoY = contentCenterY - options.logoSize / 2;
// 绘制Logo图片,四周留出3px边距
// #ifdef APP-HARMONY
ctx.drawImage(
img,
logoX + 3,
logoY + 3,
options.logoSize - 6,
options.logoSize - 6,
0,
0,
imgInfo.width,
imgInfo.height
);
// #endif
// #ifndef APP-HARMONY
ctx.drawImage(img, logoX + 3, logoY + 3, options.logoSize - 6, options.logoSize - 6);
// #endif
ctx.restore(); // 恢复之前的绘图状态
},
fail(err) {
console.error(err);
}
});
}