qrcode.ts 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972
  1. export type GenerateFrameResult = {
  2. frameBuffer: Uint8Array;
  3. width: number;
  4. };
  5. /**
  6. * 二维码生成器
  7. * @description 纯 UTS 实现的二维码生成算法,支持多平台,兼容 uni-app x。核心算法参考 QR Code 标准,支持自定义纠错级别、自动适配内容长度。
  8. * @version 1.0.0
  9. * @平台兼容性 App、H5、微信小程序、UTS
  10. * @注意事项
  11. * - 仅支持 8bit 字符串内容,不支持数字/字母/汉字等模式优化
  12. * - 生成结果为二维码点阵数据和宽度,需配合 canvas 绘制
  13. * - 纠错级别支持 'L'/'M'/'Q'/'H',默认 'L'
  14. */
  15. // 对齐块间距表 - 不同版本二维码的对齐块分布位置
  16. const ALIGNMENT_DELTA = [
  17. 0, 11, 15, 19, 23, 27, 31, 16, 18, 20, 22, 24, 26, 28, 20, 22, 24, 24, 26, 28, 28, 22, 24, 24,
  18. 26, 26, 28, 28, 24, 24, 26, 26, 26, 28, 28, 24, 26, 26, 26, 28, 28
  19. ] as number[];
  20. // 纠错块参数表 - 每个版本包含4个参数:块数、数据宽度、纠错宽度
  21. const ECC_BLOCKS = [
  22. 1, 0, 19, 7, 1, 0, 16, 10, 1, 0, 13, 13, 1, 0, 9, 17, 1, 0, 34, 10, 1, 0, 28, 16, 1, 0, 22, 22,
  23. 1, 0, 16, 28, 1, 0, 55, 15, 1, 0, 44, 26, 2, 0, 17, 18, 2, 0, 13, 22, 1, 0, 80, 20, 2, 0, 32,
  24. 18, 2, 0, 24, 26, 4, 0, 9, 16, 1, 0, 108, 26, 2, 0, 43, 24, 2, 2, 15, 18, 2, 2, 11, 22, 2, 0,
  25. 68, 18, 4, 0, 27, 16, 4, 0, 19, 24, 4, 0, 15, 28, 2, 0, 78, 20, 4, 0, 31, 18, 2, 4, 14, 18, 4,
  26. 1, 13, 26, 2, 0, 97, 24, 2, 2, 38, 22, 4, 2, 18, 22, 4, 2, 14, 26, 2, 0, 116, 30, 3, 2, 36, 22,
  27. 4, 4, 16, 20, 4, 4, 12, 24, 2, 2, 68, 18, 4, 1, 43, 26, 6, 2, 19, 24, 6, 2, 15, 28, 4, 0, 81,
  28. 20, 1, 4, 50, 30, 4, 4, 22, 28, 3, 8, 12, 24, 2, 2, 92, 24, 6, 2, 36, 22, 4, 6, 20, 26, 7, 4,
  29. 14, 28, 4, 0, 107, 26, 8, 1, 37, 22, 8, 4, 20, 24, 12, 4, 11, 22, 3, 1, 115, 30, 4, 5, 40, 24,
  30. 11, 5, 16, 20, 11, 5, 12, 24, 5, 1, 87, 22, 5, 5, 41, 24, 5, 7, 24, 30, 11, 7, 12, 24, 5, 1, 98,
  31. 24, 7, 3, 45, 28, 15, 2, 19, 24, 3, 13, 15, 30, 1, 5, 107, 28, 10, 1, 46, 28, 1, 15, 22, 28, 2,
  32. 17, 14, 28, 5, 1, 120, 30, 9, 4, 43, 26, 17, 1, 22, 28, 2, 19, 14, 28, 3, 4, 113, 28, 3, 11, 44,
  33. 26, 17, 4, 21, 26, 9, 16, 13, 26, 3, 5, 107, 28, 3, 13, 41, 26, 15, 5, 24, 30, 15, 10, 15, 28,
  34. 4, 4, 116, 28, 17, 0, 42, 26, 17, 6, 22, 28, 19, 6, 16, 30, 2, 7, 111, 28, 17, 0, 46, 28, 7, 16,
  35. 24, 30, 34, 0, 13, 24, 4, 5, 121, 30, 4, 14, 47, 28, 11, 14, 24, 30, 16, 14, 15, 30, 6, 4, 117,
  36. 30, 6, 14, 45, 28, 11, 16, 24, 30, 30, 2, 16, 30, 8, 4, 106, 26, 8, 13, 47, 28, 7, 22, 24, 30,
  37. 22, 13, 15, 30, 10, 2, 114, 28, 19, 4, 46, 28, 28, 6, 22, 28, 33, 4, 16, 30, 8, 4, 122, 30, 22,
  38. 3, 45, 28, 8, 26, 23, 30, 12, 28, 15, 30, 3, 10, 117, 30, 3, 23, 45, 28, 4, 31, 24, 30, 11, 31,
  39. 15, 30, 7, 7, 116, 30, 21, 7, 45, 28, 1, 37, 23, 30, 19, 26, 15, 30, 5, 10, 115, 30, 19, 10, 47,
  40. 28, 15, 25, 24, 30, 23, 25, 15, 30, 13, 3, 115, 30, 2, 29, 46, 28, 42, 1, 24, 30, 23, 28, 15,
  41. 30, 17, 0, 115, 30, 10, 23, 46, 28, 10, 35, 24, 30, 19, 35, 15, 30, 17, 1, 115, 30, 14, 21, 46,
  42. 28, 29, 19, 24, 30, 11, 46, 15, 30, 13, 6, 115, 30, 14, 23, 46, 28, 44, 7, 24, 30, 59, 1, 16,
  43. 30, 12, 7, 121, 30, 12, 26, 47, 28, 39, 14, 24, 30, 22, 41, 15, 30, 6, 14, 121, 30, 6, 34, 47,
  44. 28, 46, 10, 24, 30, 2, 64, 15, 30, 17, 4, 122, 30, 29, 14, 46, 28, 49, 10, 24, 30, 24, 46, 15,
  45. 30, 4, 18, 122, 30, 13, 32, 46, 28, 48, 14, 24, 30, 42, 32, 15, 30, 20, 4, 117, 30, 40, 7, 47,
  46. 28, 43, 22, 24, 30, 10, 67, 15, 30, 19, 6, 118, 30, 18, 31, 47, 28, 34, 34, 24, 30, 20, 61, 15,
  47. 30
  48. ] as number[];
  49. // 纠错级别映射表 - 将人类可读的纠错级别映射为内部数值
  50. const ECC_LEVELS = new Map<string, number>([
  51. ["L", 1],
  52. ["M", 0],
  53. ["Q", 3],
  54. ["H", 2]
  55. ]);
  56. // 最终格式信息掩码表 - 用于格式信息区域的掩码计算(level << 3 | mask)
  57. const FINAL_FORMAT = [
  58. 0x77c4, 0x72f3, 0x7daa, 0x789d, 0x662f, 0x6318, 0x6c41, 0x6976 /* L */, 0x5412, 0x5125, 0x5e7c,
  59. 0x5b4b, 0x45f9, 0x40ce, 0x4f97, 0x4aa0 /* M */, 0x355f, 0x3068, 0x3f31, 0x3a06, 0x24b4, 0x2183,
  60. 0x2eda, 0x2bed /* Q */, 0x1689, 0x13be, 0x1ce7, 0x19d0, 0x0762, 0x0255, 0x0d0c, 0x083b /* H */
  61. ];
  62. // Galois域指数表 - 用于纠错码计算的查找表
  63. const GALOIS_EXPONENT = [
  64. 0x01, 0x02, 0x04, 0x08, 0x10, 0x20, 0x40, 0x80, 0x1d, 0x3a, 0x74, 0xe8, 0xcd, 0x87, 0x13, 0x26,
  65. 0x4c, 0x98, 0x2d, 0x5a, 0xb4, 0x75, 0xea, 0xc9, 0x8f, 0x03, 0x06, 0x0c, 0x18, 0x30, 0x60, 0xc0,
  66. 0x9d, 0x27, 0x4e, 0x9c, 0x25, 0x4a, 0x94, 0x35, 0x6a, 0xd4, 0xb5, 0x77, 0xee, 0xc1, 0x9f, 0x23,
  67. 0x46, 0x8c, 0x05, 0x0a, 0x14, 0x28, 0x50, 0xa0, 0x5d, 0xba, 0x69, 0xd2, 0xb9, 0x6f, 0xde, 0xa1,
  68. 0x5f, 0xbe, 0x61, 0xc2, 0x99, 0x2f, 0x5e, 0xbc, 0x65, 0xca, 0x89, 0x0f, 0x1e, 0x3c, 0x78, 0xf0,
  69. 0xfd, 0xe7, 0xd3, 0xbb, 0x6b, 0xd6, 0xb1, 0x7f, 0xfe, 0xe1, 0xdf, 0xa3, 0x5b, 0xb6, 0x71, 0xe2,
  70. 0xd9, 0xaf, 0x43, 0x86, 0x11, 0x22, 0x44, 0x88, 0x0d, 0x1a, 0x34, 0x68, 0xd0, 0xbd, 0x67, 0xce,
  71. 0x81, 0x1f, 0x3e, 0x7c, 0xf8, 0xed, 0xc7, 0x93, 0x3b, 0x76, 0xec, 0xc5, 0x97, 0x33, 0x66, 0xcc,
  72. 0x85, 0x17, 0x2e, 0x5c, 0xb8, 0x6d, 0xda, 0xa9, 0x4f, 0x9e, 0x21, 0x42, 0x84, 0x15, 0x2a, 0x54,
  73. 0xa8, 0x4d, 0x9a, 0x29, 0x52, 0xa4, 0x55, 0xaa, 0x49, 0x92, 0x39, 0x72, 0xe4, 0xd5, 0xb7, 0x73,
  74. 0xe6, 0xd1, 0xbf, 0x63, 0xc6, 0x91, 0x3f, 0x7e, 0xfc, 0xe5, 0xd7, 0xb3, 0x7b, 0xf6, 0xf1, 0xff,
  75. 0xe3, 0xdb, 0xab, 0x4b, 0x96, 0x31, 0x62, 0xc4, 0x95, 0x37, 0x6e, 0xdc, 0xa5, 0x57, 0xae, 0x41,
  76. 0x82, 0x19, 0x32, 0x64, 0xc8, 0x8d, 0x07, 0x0e, 0x1c, 0x38, 0x70, 0xe0, 0xdd, 0xa7, 0x53, 0xa6,
  77. 0x51, 0xa2, 0x59, 0xb2, 0x79, 0xf2, 0xf9, 0xef, 0xc3, 0x9b, 0x2b, 0x56, 0xac, 0x45, 0x8a, 0x09,
  78. 0x12, 0x24, 0x48, 0x90, 0x3d, 0x7a, 0xf4, 0xf5, 0xf7, 0xf3, 0xfb, 0xeb, 0xcb, 0x8b, 0x0b, 0x16,
  79. 0x2c, 0x58, 0xb0, 0x7d, 0xfa, 0xe9, 0xcf, 0x83, 0x1b, 0x36, 0x6c, 0xd8, 0xad, 0x47, 0x8e, 0x00
  80. ];
  81. // Galois域对数表 - 用于纠错码计算的反向查找表
  82. const GALOIS_LOG = [
  83. 0xff, 0x00, 0x01, 0x19, 0x02, 0x32, 0x1a, 0xc6, 0x03, 0xdf, 0x33, 0xee, 0x1b, 0x68, 0xc7, 0x4b,
  84. 0x04, 0x64, 0xe0, 0x0e, 0x34, 0x8d, 0xef, 0x81, 0x1c, 0xc1, 0x69, 0xf8, 0xc8, 0x08, 0x4c, 0x71,
  85. 0x05, 0x8a, 0x65, 0x2f, 0xe1, 0x24, 0x0f, 0x21, 0x35, 0x93, 0x8e, 0xda, 0xf0, 0x12, 0x82, 0x45,
  86. 0x1d, 0xb5, 0xc2, 0x7d, 0x6a, 0x27, 0xf9, 0xb9, 0xc9, 0x9a, 0x09, 0x78, 0x4d, 0xe4, 0x72, 0xa6,
  87. 0x06, 0xbf, 0x8b, 0x62, 0x66, 0xdd, 0x30, 0xfd, 0xe2, 0x98, 0x25, 0xb3, 0x10, 0x91, 0x22, 0x88,
  88. 0x36, 0xd0, 0x94, 0xce, 0x8f, 0x96, 0xdb, 0xbd, 0xf1, 0xd2, 0x13, 0x5c, 0x83, 0x38, 0x46, 0x40,
  89. 0x1e, 0x42, 0xb6, 0xa3, 0xc3, 0x48, 0x7e, 0x6e, 0x6b, 0x3a, 0x28, 0x54, 0xfa, 0x85, 0xba, 0x3d,
  90. 0xca, 0x5e, 0x9b, 0x9f, 0x0a, 0x15, 0x79, 0x2b, 0x4e, 0xd4, 0xe5, 0xac, 0x73, 0xf3, 0xa7, 0x57,
  91. 0x07, 0x70, 0xc0, 0xf7, 0x8c, 0x80, 0x63, 0x0d, 0x67, 0x4a, 0xde, 0xed, 0x31, 0xc5, 0xfe, 0x18,
  92. 0xe3, 0xa5, 0x99, 0x77, 0x26, 0xb8, 0xb4, 0x7c, 0x11, 0x44, 0x92, 0xd9, 0x23, 0x20, 0x89, 0x2e,
  93. 0x37, 0x3f, 0xd1, 0x5b, 0x95, 0xbc, 0xcf, 0xcd, 0x90, 0x87, 0x97, 0xb2, 0xdc, 0xfc, 0xbe, 0x61,
  94. 0xf2, 0x56, 0xd3, 0xab, 0x14, 0x2a, 0x5d, 0x9e, 0x84, 0x3c, 0x39, 0x53, 0x47, 0x6d, 0x41, 0xa2,
  95. 0x1f, 0x2d, 0x43, 0xd8, 0xb7, 0x7b, 0xa4, 0x76, 0xc4, 0x17, 0x49, 0xec, 0x7f, 0x0c, 0x6f, 0xf6,
  96. 0x6c, 0xa1, 0x3b, 0x52, 0x29, 0x9d, 0x55, 0xaa, 0xfb, 0x60, 0x86, 0xb1, 0xbb, 0xcc, 0x3e, 0x5a,
  97. 0xcb, 0x59, 0x5f, 0xb0, 0x9c, 0xa9, 0xa0, 0x51, 0x0b, 0xf5, 0x16, 0xeb, 0x7a, 0x75, 0x2c, 0xd7,
  98. 0x4f, 0xae, 0xd5, 0xe9, 0xe6, 0xe7, 0xad, 0xe8, 0x74, 0xd6, 0xf4, 0xea, 0xa8, 0x50, 0x58, 0xaf
  99. ];
  100. // 二维码质量评估系数 - 用于计算最佳掩码模式
  101. // N1: 连续5个及以上同色模块的惩罚分数
  102. const N1 = 3;
  103. // N2: 2x2同色模块区域的惩罚分数
  104. const N2 = 3;
  105. // N3: 类似定位图形的图案(1:1:3:1:1)的惩罚分数
  106. const N3 = 40;
  107. // N4: 黑白模块比例不均衡的惩罚分数
  108. const N4 = 10;
  109. // 版本信息掩码表 - 用于在二维码中嵌入版本信息
  110. const VERSION_BLOCK = [
  111. 0xc94, 0x5bc, 0xa99, 0x4d3, 0xbf6, 0x762, 0x847, 0x60d, 0x928, 0xb78, 0x45d, 0xa17, 0x532,
  112. 0x9a6, 0x683, 0x8c9, 0x7ec, 0xec4, 0x1e1, 0xfab, 0x08e, 0xc1a, 0x33f, 0xd75, 0x250, 0x9d5,
  113. 0x6f0, 0x8ba, 0x79f, 0xb0b, 0x42e, 0xa64, 0x541, 0xc69
  114. ];
  115. /**
  116. * 生成二维码点阵
  117. * @param _str 输入字符串,支持任意文本内容,默认 null 表示空字符串
  118. * @param ecc 纠错级别,可选 'L' | 'M' | 'Q' | 'H',默认 'L'
  119. * @returns {GenerateFrameResult} 返回二维码点阵数据和宽度
  120. */
  121. export function generateFrame(
  122. _str: string | null = null,
  123. ecc: string | null = null
  124. ): GenerateFrameResult {
  125. // 变量声明区,所有临时变量、缓冲区
  126. let i: number;
  127. let t: number;
  128. let j: number;
  129. let k: number;
  130. let m: number;
  131. let v: number;
  132. let x: number;
  133. let y: number;
  134. let version: number;
  135. let str = _str == null ? "" : _str;
  136. let width = 0;
  137. // 获取纠错级别数值
  138. let eccLevel = ECC_LEVELS.get(ecc == null ? "L" : ecc)!;
  139. // Data block
  140. // 数据块、纠错块、块数
  141. let dataBlock: number;
  142. let eccBlock: number;
  143. let neccBlock1: number;
  144. let neccBlock2: number;
  145. // ECC buffer.
  146. // 纠错码缓冲区 - 先初始化为空数组,后面会重新赋值
  147. let eccBuffer: Uint8Array;
  148. // Image buffer.
  149. // 二维码点阵缓冲区 - 先初始化为空数组,后面会重新赋值
  150. let frameBuffer = new Uint8Array(0);
  151. // Fixed part of the image.
  152. // 点阵掩码缓冲区(标记不可变区域) - 先初始化为空数组,后面会重新赋值
  153. let frameMask = new Uint8Array(0);
  154. // Generator polynomial.
  155. // 生成多项式缓冲区(纠错码计算用) - 先初始化为空数组,后面会重新赋值
  156. let polynomial = new Uint8Array(0);
  157. // Data input buffer.
  158. // 数据输入缓冲区 - 先初始化为空数组,后面会重新赋值
  159. let stringBuffer = new Uint8Array(0);
  160. /**
  161. * 设置掩码位,表示该点为不可变区域(对称处理)
  162. * @param _x 横坐标
  163. * @param _y 纵坐标
  164. */
  165. function setMask(_x: number, _y: number) {
  166. let bit: number;
  167. let x = _x;
  168. let y = _y;
  169. if (x > y) {
  170. bit = x;
  171. x = y;
  172. y = bit;
  173. }
  174. bit = y;
  175. bit *= y;
  176. bit += y;
  177. bit >>= 1;
  178. bit += x;
  179. frameMask[bit] = 1;
  180. }
  181. /**
  182. * 添加对齐块,设置对应点阵和掩码
  183. * @param _x 横坐标
  184. * @param _y 纵坐标
  185. */
  186. function addAlignment(_x: number, _y: number) {
  187. let i: number;
  188. let x = _x;
  189. let y = _y;
  190. frameBuffer[x + width * y] = 1;
  191. for (i = -2; i < 2; i++) {
  192. frameBuffer[x + i + width * (y - 2)] = 1;
  193. frameBuffer[x - 2 + width * (y + i + 1)] = 1;
  194. frameBuffer[x + 2 + width * (y + i)] = 1;
  195. frameBuffer[x + i + 1 + width * (y + 2)] = 1;
  196. }
  197. for (i = 0; i < 2; i++) {
  198. setMask(x - 1, y + i);
  199. setMask(x + 1, y - i);
  200. setMask(x - i, y - 1);
  201. setMask(x + i, y + 1);
  202. }
  203. for (i = 2; i < 4; i++) {
  204. frameBuffer[x + i + width * (y - 2)] = 1;
  205. frameBuffer[x - 2 + width * (y + i - 1)] = 1;
  206. frameBuffer[x + 2 + width * (y + i - 2)] = 1;
  207. frameBuffer[x - 1 + width * (y + i - 2)] = 1;
  208. }
  209. }
  210. /**
  211. * Galois 域取模运算
  212. * @param _x 输入数值
  213. * @returns {number} 取模结果
  214. */
  215. function modN(_x: number): number {
  216. var x = _x;
  217. while (x >= 255) {
  218. x -= 255;
  219. x = (x >> 8) + (x & 255);
  220. }
  221. return x;
  222. }
  223. /**
  224. * 计算并追加纠错码到数据块
  225. * @param _data 数据起始索引
  226. * @param _dataLength 数据长度
  227. * @param _ecc 纠错码起始索引
  228. * @param _eccLength 纠错码长度
  229. */
  230. function appendData(_data: number, _dataLength: number, _ecc: number, _eccLength: number) {
  231. let bit: number;
  232. let i: number;
  233. let j: number;
  234. let data = _data;
  235. let dataLength = _dataLength;
  236. let ecc = _ecc;
  237. let eccLength = _eccLength;
  238. for (i = 0; i < eccLength; i++) {
  239. stringBuffer[ecc + i] = 0;
  240. }
  241. for (i = 0; i < dataLength; i++) {
  242. bit = GALOIS_LOG[stringBuffer[data + i] ^ stringBuffer[ecc]];
  243. if (bit != 255) {
  244. for (j = 1; j < eccLength; j++) {
  245. stringBuffer[ecc + j - 1] =
  246. stringBuffer[ecc + j] ^
  247. GALOIS_EXPONENT[modN(bit + polynomial[eccLength - j])];
  248. }
  249. } else {
  250. for (j = ecc; j < ecc + eccLength; j++) {
  251. stringBuffer[j] = stringBuffer[j + 1];
  252. }
  253. }
  254. stringBuffer[ecc + eccLength - 1] =
  255. bit == 255 ? 0 : GALOIS_EXPONENT[modN(bit + polynomial[0])];
  256. }
  257. }
  258. /**
  259. * 判断某点是否为掩码区域
  260. * @param _x 横坐标
  261. * @param _y 纵坐标
  262. * @returns {boolean} 是否为掩码
  263. */
  264. function isMasked(_x: number, _y: number): boolean {
  265. let bit: number;
  266. let x = _x;
  267. let y = _y;
  268. if (x > y) {
  269. bit = x;
  270. x = y;
  271. y = bit;
  272. }
  273. bit = y;
  274. bit += y * y;
  275. bit >>= 1;
  276. bit += x;
  277. return frameMask[bit] == 1;
  278. }
  279. /**
  280. * 根据 QR Code 标准,应用指定的掩码 pattern
  281. * @param mask 掩码编号 (0-7)
  282. */
  283. function applyMask(mask: number) {
  284. for (let y = 0; y < width; y++) {
  285. for (let x = 0; x < width; x++) {
  286. if (!isMasked(x, y)) {
  287. let shouldInvert = false;
  288. switch (mask) {
  289. case 0:
  290. shouldInvert = (x + y) % 2 == 0;
  291. break;
  292. case 1:
  293. shouldInvert = y % 2 == 0;
  294. break;
  295. case 2:
  296. shouldInvert = x % 3 == 0;
  297. break;
  298. case 3:
  299. shouldInvert = (x + y) % 3 == 0;
  300. break;
  301. case 4:
  302. shouldInvert = (Math.floor(y / 2) + Math.floor(x / 3)) % 2 == 0;
  303. break;
  304. case 5:
  305. shouldInvert = ((x * y) % 2) + ((x * y) % 3) == 0;
  306. break;
  307. case 6:
  308. shouldInvert = (((x * y) % 2) + ((x * y) % 3)) % 2 == 0;
  309. break;
  310. case 7:
  311. shouldInvert = (((x + y) % 2) + ((x * y) % 3)) % 2 == 0;
  312. break;
  313. }
  314. if (shouldInvert) {
  315. frameBuffer[x + y * width] ^= 1;
  316. }
  317. }
  318. }
  319. }
  320. }
  321. /**
  322. * 计算连续同色块的"坏度"分数
  323. * @param runLengths
  324. * @param length 块长度
  325. * @returns {number} 坏度分数
  326. */
  327. function getBadRuns(runLengths: number[], length: number): number {
  328. let badRuns = 0;
  329. let i: number;
  330. for (i = 0; i <= length; i++) {
  331. if (i < runLengths.length && runLengths[i] >= 5) {
  332. badRuns += N1 + runLengths[i] - 5;
  333. }
  334. }
  335. // FBFFFBF as in finder.
  336. for (i = 3; i < length - 1; i += 2) {
  337. // 检查数组索引是否越界
  338. if (i + 2 >= runLengths.length || i - 3 < 0) {
  339. continue;
  340. }
  341. if (
  342. runLengths[i - 2] == runLengths[i + 2] &&
  343. runLengths[i + 2] == runLengths[i - 1] &&
  344. runLengths[i - 1] == runLengths[i + 1] &&
  345. runLengths[i - 1] * 3 == runLengths[i] &&
  346. // Background around the foreground pattern? Not part of the specs.
  347. (runLengths[i - 3] == 0 ||
  348. i + 3 > length ||
  349. runLengths[i - 3] * 3 >= runLengths[i] * 4 ||
  350. runLengths[i + 3] * 3 >= runLengths[i] * 4)
  351. ) {
  352. badRuns += N3;
  353. }
  354. }
  355. return badRuns;
  356. }
  357. /**
  358. * 评估当前二维码点阵的整体"坏度"
  359. * @returns {number} 坏度分数
  360. */
  361. function checkBadness(): number {
  362. let b: number;
  363. let b1: number;
  364. let bad = 0;
  365. let big: number;
  366. let bw = 0;
  367. let count = 0;
  368. let h: number;
  369. let x: number;
  370. let y: number;
  371. // 优化:在函数内创建badBuffer,避免外部变量的内存泄漏风险
  372. let badBuffer = new Array<number>(width);
  373. // Blocks of same colour.
  374. for (y = 0; y < width - 1; y++) {
  375. for (x = 0; x < width - 1; x++) {
  376. // All foreground colour.
  377. if (
  378. (frameBuffer[x + width * y] == 1 &&
  379. frameBuffer[x + 1 + width * y] == 1 &&
  380. frameBuffer[x + width * (y + 1)] == 1 &&
  381. frameBuffer[x + 1 + width * (y + 1)] == 1) ||
  382. // All background colour.
  383. (frameBuffer[x + width * y] == 0 &&
  384. frameBuffer[x + 1 + width * y] == 0 &&
  385. frameBuffer[x + width * (y + 1)] == 0 &&
  386. frameBuffer[x + 1 + width * (y + 1)] == 0)
  387. ) {
  388. bad += N2;
  389. }
  390. }
  391. }
  392. // X runs
  393. for (y = 0; y < width; y++) {
  394. h = 0;
  395. badBuffer[h] = 0;
  396. b = 0;
  397. for (x = 0; x < width; x++) {
  398. b1 = frameBuffer[x + width * y];
  399. if (b1 == b) {
  400. if (h < badBuffer.length) {
  401. badBuffer[h]++;
  402. }
  403. } else {
  404. h++;
  405. if (h < badBuffer.length) {
  406. badBuffer[h] = 1;
  407. }
  408. }
  409. b = b1;
  410. bw += b > 0 ? 1 : -1;
  411. }
  412. bad += getBadRuns(badBuffer, h);
  413. }
  414. if (bw < 0) bw = -bw;
  415. big = bw;
  416. big += big << 2;
  417. big <<= 1;
  418. while (big > width * width) {
  419. big -= width * width;
  420. count++;
  421. }
  422. bad += count * N4;
  423. // Y runs.
  424. for (x = 0; x < width; x++) {
  425. h = 0;
  426. badBuffer[h] = 0;
  427. b = 0;
  428. for (y = 0; y < width; y++) {
  429. b1 = frameBuffer[x + width * y];
  430. if (b1 == b) {
  431. if (h < badBuffer.length) {
  432. badBuffer[h]++;
  433. }
  434. } else {
  435. h++;
  436. if (h < badBuffer.length) {
  437. badBuffer[h] = 1;
  438. }
  439. }
  440. b = b1;
  441. }
  442. bad += getBadRuns(badBuffer, h);
  443. }
  444. return bad;
  445. }
  446. /**
  447. * 将字符串转为 UTF-8 编码,兼容多平台
  448. * @param str 输入字符串
  449. * @returns {string} UTF-8 编码字符串
  450. */
  451. function toUtf8(str: string): string {
  452. let out = "";
  453. let i: number;
  454. let len: number;
  455. let c: number;
  456. len = str.length;
  457. for (i = 0; i < len; i++) {
  458. c = str.charCodeAt(i)!;
  459. if (c >= 0x0001 && c <= 0x007f) {
  460. out += str.charAt(i);
  461. } else if (c > 0x07ff) {
  462. out += String.fromCharCode(0xe0 | ((c >> 12) & 0x0f));
  463. out += String.fromCharCode(0x80 | ((c >> 6) & 0x3f));
  464. out += String.fromCharCode(0x80 | ((c >> 0) & 0x3f));
  465. } else {
  466. out += String.fromCharCode(0xc0 | ((c >> 6) & 0x1f));
  467. out += String.fromCharCode(0x80 | ((c >> 0) & 0x3f));
  468. }
  469. }
  470. return out;
  471. }
  472. //end functions
  473. // Find the smallest version that fits the string.
  474. // 1. 字符串转 UTF-8,计算长度
  475. str = toUtf8(str);
  476. t = str.length;
  477. // 2. 自动选择最小可用版本
  478. version = 0;
  479. do {
  480. version++;
  481. k = (eccLevel - 1) * 4 + (version - 1) * 16;
  482. neccBlock1 = ECC_BLOCKS[k++];
  483. neccBlock2 = ECC_BLOCKS[k++];
  484. dataBlock = ECC_BLOCKS[k++];
  485. eccBlock = ECC_BLOCKS[k];
  486. k = dataBlock * (neccBlock1 + neccBlock2) + neccBlock2 - 3 + (version <= 9 ? 1 : 0);
  487. if (t <= k) break;
  488. } while (version < 40);
  489. // FIXME: Ensure that it fits insted of being truncated.
  490. // 3. 计算二维码宽度
  491. width = 17 + 4 * version;
  492. // Allocate, clear and setup data structures.
  493. // 4. 分配缓冲区, 使用定长的 Uint8Array 优化内存
  494. v = dataBlock + (dataBlock + eccBlock) * (neccBlock1 + neccBlock2) + neccBlock2;
  495. eccBuffer = new Uint8Array(v);
  496. stringBuffer = new Uint8Array(v);
  497. // 5. 预分配点阵、掩码缓冲区
  498. frameBuffer = new Uint8Array(width * width);
  499. frameMask = new Uint8Array(Math.floor((width * (width + 1) + 1) / 2));
  500. // Insert finders: Foreground colour to frame and background to mask.
  501. // 插入定位点: 前景色为二维码,背景色为掩码
  502. for (t = 0; t < 3; t++) {
  503. k = 0;
  504. y = 0;
  505. if (t == 1) k = width - 7;
  506. if (t == 2) y = width - 7;
  507. frameBuffer[y + 3 + width * (k + 3)] = 1;
  508. for (x = 0; x < 6; x++) {
  509. frameBuffer[y + x + width * k] = 1;
  510. frameBuffer[y + width * (k + x + 1)] = 1;
  511. frameBuffer[y + 6 + width * (k + x)] = 1;
  512. frameBuffer[y + x + 1 + width * (k + 6)] = 1;
  513. }
  514. for (x = 1; x < 5; x++) {
  515. setMask(y + x, k + 1);
  516. setMask(y + 1, k + x + 1);
  517. setMask(y + 5, k + x);
  518. setMask(y + x + 1, k + 5);
  519. }
  520. for (x = 2; x < 4; x++) {
  521. frameBuffer[y + x + width * (k + 2)] = 1;
  522. frameBuffer[y + 2 + width * (k + x + 1)] = 1;
  523. frameBuffer[y + 4 + width * (k + x)] = 1;
  524. frameBuffer[y + x + 1 + width * (k + 4)] = 1;
  525. }
  526. }
  527. // Alignment blocks.
  528. // 插入对齐点: 前景色为二维码,背景色为掩码
  529. if (version > 1) {
  530. t = ALIGNMENT_DELTA[version];
  531. y = width - 7;
  532. for (;;) {
  533. x = width - 7;
  534. while (x > t - 3) {
  535. addAlignment(x, y);
  536. if (x < t) break;
  537. x -= t;
  538. }
  539. if (y <= t + 9) break;
  540. y -= t;
  541. addAlignment(6, y);
  542. addAlignment(y, 6);
  543. }
  544. }
  545. // Single foreground cell.
  546. // 插入单个前景色单元格: 前景色为二维码,背景色为掩码
  547. frameBuffer[8 + width * (width - 8)] = 1;
  548. // Timing gap (mask only).
  549. // 插入时间间隔: 掩码
  550. for (y = 0; y < 7; y++) {
  551. setMask(7, y);
  552. setMask(width - 8, y);
  553. setMask(7, y + width - 7);
  554. }
  555. for (x = 0; x < 8; x++) {
  556. setMask(x, 7);
  557. setMask(x + width - 8, 7);
  558. setMask(x, width - 8);
  559. }
  560. // Reserve mask, format area.
  561. // 保留掩码,格式化区域
  562. for (x = 0; x < 9; x++) {
  563. setMask(x, 8);
  564. }
  565. for (x = 0; x < 8; x++) {
  566. setMask(x + width - 8, 8);
  567. setMask(8, x);
  568. }
  569. for (y = 0; y < 7; y++) {
  570. setMask(8, y + width - 7);
  571. }
  572. // Timing row/column.
  573. // 插入时间间隔行/列: 掩码
  574. for (x = 0; x < width - 14; x++) {
  575. if ((x & 1) > 0) {
  576. setMask(8 + x, 6);
  577. setMask(6, 8 + x);
  578. } else {
  579. frameBuffer[8 + x + width * 6] = 1;
  580. frameBuffer[6 + width * (8 + x)] = 1;
  581. }
  582. }
  583. // Version block.
  584. if (version > 6) {
  585. t = VERSION_BLOCK[version - 7];
  586. k = 17;
  587. for (x = 0; x < 6; x++) {
  588. for (y = 0; y < 3; y++) {
  589. if ((1 & (k > 11 ? version >> (k - 12) : t >> k)) > 0) {
  590. frameBuffer[5 - x + width * (2 - y + width - 11)] = 1;
  591. frameBuffer[2 - y + width - 11 + width * (5 - x)] = 1;
  592. } else {
  593. setMask(5 - x, 2 - y + width - 11);
  594. setMask(2 - y + width - 11, 5 - x);
  595. }
  596. k--;
  597. }
  598. }
  599. }
  600. // Sync mask bits. Only set above for background cells, so now add the foreground.
  601. // 同步掩码位。只有上方的背景单元格需要设置,现在添加前景色。
  602. for (y = 0; y < width; y++) {
  603. for (x = 0; x <= y; x++) {
  604. if (frameBuffer[x + width * y] > 0) {
  605. setMask(x, y);
  606. }
  607. }
  608. }
  609. // Convert string to bit stream. 8-bit data to QR-coded 8-bit data (numeric, alphanum, or kanji
  610. // not supported).
  611. // 将字符串转换为位流。8位数据转换为QR编码的8位数据(不支持数字、字母或汉字)。
  612. v = str.length;
  613. // String to array.
  614. for (i = 0; i < v; i++) {
  615. // #ifdef APP-ANDROID
  616. // @ts-ignore
  617. eccBuffer[i.toInt()] = str.charCodeAt(i)!;
  618. // #endif
  619. // #ifndef APP-ANDROID
  620. eccBuffer[i] = str.charCodeAt(i)!;
  621. // #endif
  622. }
  623. //++++++++++++++++++++==============
  624. stringBuffer.set(eccBuffer.subarray(0, v));
  625. // Calculate max string length.
  626. x = dataBlock * (neccBlock1 + neccBlock2) + neccBlock2;
  627. if (v >= x - 2) {
  628. v = x - 2;
  629. if (version > 9) v--;
  630. }
  631. // Shift and re-pack to insert length prefix.
  632. // 移位并重新打包以插入长度前缀。
  633. i = v;
  634. if (version > 9) {
  635. stringBuffer[i + 2] = 0;
  636. stringBuffer[i + 3] = 0;
  637. while (i-- > 0) {
  638. t = stringBuffer[i];
  639. stringBuffer[i + 3] |= 255 & (t << 4);
  640. stringBuffer[i + 2] = t >> 4;
  641. }
  642. stringBuffer[2] |= 255 & (v << 4);
  643. stringBuffer[1] = v >> 4;
  644. stringBuffer[0] = 0x40 | (v >> 12);
  645. } else {
  646. stringBuffer[i + 1] = 0;
  647. stringBuffer[i + 2] = 0;
  648. while (i-- > 0) {
  649. t = stringBuffer[i];
  650. stringBuffer[i + 2] |= 255 & (t << 4);
  651. stringBuffer[i + 1] = t >> 4;
  652. }
  653. stringBuffer[1] |= 255 & (v << 4);
  654. stringBuffer[0] = 0x40 | (v >> 4);
  655. }
  656. // Fill to end with pad pattern.
  657. // 用填充模式填充到结束。
  658. i = v + 3 - (version < 10 ? 1 : 0);
  659. while (i < x) {
  660. stringBuffer[i++] = 0xec;
  661. stringBuffer[i++] = 0x11;
  662. }
  663. // Calculate generator polynomial.
  664. // 计算生成多项式。
  665. polynomial = new Uint8Array(eccBlock + 1);
  666. polynomial[0] = 1;
  667. for (i = 0; i < eccBlock; i++) {
  668. polynomial[i + 1] = 1;
  669. for (j = i; j > 0; j--) {
  670. polynomial[j] =
  671. polynomial[j] > 0
  672. ? polynomial[j - 1] ^ GALOIS_EXPONENT[modN(GALOIS_LOG[polynomial[j]] + i)]
  673. : polynomial[j - 1];
  674. }
  675. polynomial[0] = GALOIS_EXPONENT[modN(GALOIS_LOG[polynomial[0]] + i)];
  676. }
  677. // Use logs for generator polynomial to save calculation step.
  678. // 使用对数计算生成多项式以节省计算步骤。
  679. for (i = 0; i < eccBlock; i++) {
  680. polynomial[i] = GALOIS_LOG[polynomial[i]];
  681. }
  682. // Append ECC to data buffer.
  683. // 将ECC附加到数据缓冲区。
  684. k = x;
  685. y = 0;
  686. for (i = 0; i < neccBlock1; i++) {
  687. appendData(y, dataBlock, k, eccBlock);
  688. y += dataBlock;
  689. k += eccBlock;
  690. }
  691. for (i = 0; i < neccBlock2; i++) {
  692. appendData(y, dataBlock + 1, k, eccBlock);
  693. y += dataBlock + 1;
  694. k += eccBlock;
  695. }
  696. // Interleave blocks.
  697. y = 0;
  698. for (i = 0; i < dataBlock; i++) {
  699. for (j = 0; j < neccBlock1; j++) {
  700. eccBuffer[y++] = stringBuffer[i + j * dataBlock];
  701. }
  702. for (j = 0; j < neccBlock2; j++) {
  703. eccBuffer[y++] = stringBuffer[neccBlock1 * dataBlock + i + j * (dataBlock + 1)];
  704. }
  705. }
  706. for (j = 0; j < neccBlock2; j++) {
  707. eccBuffer[y++] = stringBuffer[neccBlock1 * dataBlock + i + j * (dataBlock + 1)];
  708. }
  709. for (i = 0; i < eccBlock; i++) {
  710. for (j = 0; j < neccBlock1 + neccBlock2; j++) {
  711. eccBuffer[y++] = stringBuffer[x + i + j * eccBlock];
  712. }
  713. }
  714. stringBuffer.set(eccBuffer);
  715. // Pack bits into frame avoiding masked area.
  716. // 将位流打包到帧中,避免掩码区域。
  717. x = width - 1;
  718. y = width - 1;
  719. k = 1;
  720. v = 1;
  721. // inteleaved data and ECC codes.
  722. // 交错数据和ECC代码。
  723. m = (dataBlock + eccBlock) * (neccBlock1 + neccBlock2) + neccBlock2;
  724. for (i = 0; i < m; i++) {
  725. t = stringBuffer[i];
  726. for (j = 0; j < 8; j++) {
  727. if ((0x80 & t) > 0) {
  728. frameBuffer[x + width * y] = 1;
  729. }
  730. // Find next fill position.
  731. // 找到下一个填充位置。
  732. do {
  733. if (v > 0) {
  734. x--;
  735. } else {
  736. x++;
  737. if (k > 0) {
  738. if (y != 0) {
  739. y--;
  740. } else {
  741. x -= 2;
  742. k = k == 0 ? 1 : 0;
  743. if (x == 6) {
  744. x--;
  745. y = 9;
  746. }
  747. }
  748. } else {
  749. if (y != width - 1) {
  750. y++;
  751. } else {
  752. x -= 2;
  753. k = k == 0 ? 1 : 0;
  754. if (x == 6) {
  755. x--;
  756. y -= 8;
  757. }
  758. }
  759. }
  760. }
  761. v = v == 0 ? 1 : 0;
  762. } while (isMasked(x, y));
  763. t <<= 1;
  764. }
  765. }
  766. // Save pre-mask copy of frame.
  767. const frameBufferCopy = frameBuffer.slice(0);
  768. t = 0;
  769. y = 30000;
  770. // Using `for` instead of `while` since in original Arduino code if an early mask was *good
  771. // enough* it wouldn't try for a better one since they get more complex and take longer.
  772. // 使用`for`而不是`while`,因为在原始Arduino代码中,如果早期掩码足够好,它不会尝试更好的掩码,因为它们变得更复杂并需要更长的时间。
  773. for (k = 0; k < 8; k++) {
  774. // Returns foreground-background imbalance.
  775. // 返回前景色和背景色的不平衡。
  776. applyMask(k);
  777. x = checkBadness();
  778. // Is current mask better than previous best?
  779. // 当前掩码是否比之前的最佳掩码更好?
  780. if (x < y) {
  781. y = x;
  782. t = k;
  783. }
  784. // Don't increment `i` to a void redoing mask.
  785. // 不要增加`i`以避免重新做掩码。
  786. if (t == 7) break;
  787. // Reset for next pass.
  788. // 重置下一个循环。
  789. frameBuffer.set(frameBufferCopy);
  790. }
  791. // Redo best mask as none were *good enough* (i.e. last wasn't `t`).
  792. // 重做最佳掩码,因为没有一个掩码足够好(即最后一个不是`t`)。
  793. if (t != k) {
  794. // Reset buffer to pre-mask state before applying the best one
  795. frameBuffer.set(frameBufferCopy);
  796. applyMask(t);
  797. }
  798. // Add in final mask/ECC level bytes.
  799. // 添加最终的掩码/ECC级别字节。
  800. y = FINAL_FORMAT[t + ((eccLevel - 1) << 3)];
  801. // Low byte.
  802. for (k = 0; k < 8; k++) {
  803. if ((y & 1) > 0) {
  804. frameBuffer[width - 1 - k + width * 8] = 1;
  805. if (k < 6) {
  806. frameBuffer[8 + width * k] = 1;
  807. } else {
  808. frameBuffer[8 + width * (k + 1)] = 1;
  809. }
  810. }
  811. y >>= 1;
  812. }
  813. // High byte.
  814. for (k = 0; k < 7; k++) {
  815. if ((y & 1) > 0) {
  816. frameBuffer[8 + width * (width - 7 + k)] = 1;
  817. if (k > 0) {
  818. frameBuffer[6 - k + width * 8] = 1;
  819. } else {
  820. frameBuffer[7 + width * 8] = 1;
  821. }
  822. }
  823. y >>= 1;
  824. }
  825. // Finally, return the image data.
  826. return {
  827. frameBuffer: frameBuffer,
  828. width: width
  829. } as GenerateFrameResult;
  830. }