index.js 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462
  1. //---------------------------------------------------------------------
  2. // QRCode for JavaScript
  3. //
  4. // Copyright (c) 2009 Kazuhiko Arase
  5. //
  6. // URL: http://www.d-project.com/
  7. //
  8. // Licensed under the MIT license:
  9. // http://www.opensource.org/licenses/mit-license.php
  10. //
  11. // The word "QR Code" is registered trademark of
  12. // DENSO WAVE INCORPORATED
  13. // http://www.denso-wave.com/qrcode/faqpatent-e.html
  14. //
  15. //---------------------------------------------------------------------
  16. // Modified to work in node for this project (and some refactoring)
  17. //---------------------------------------------------------------------
  18. var QR8bitByte = require('./QR8bitByte');
  19. var QRUtil = require('./QRUtil');
  20. var QRPolynomial = require('./QRPolynomial');
  21. var QRRSBlock = require('./QRRSBlock');
  22. var QRBitBuffer = require('./QRBitBuffer');
  23. function QRCode(typeNumber, errorCorrectLevel) {
  24. this.typeNumber = typeNumber;
  25. this.errorCorrectLevel = errorCorrectLevel;
  26. this.modules = null;
  27. this.moduleCount = 0;
  28. this.dataCache = null;
  29. this.dataList = [];
  30. }
  31. QRCode.prototype = {
  32. addData : function(data) {
  33. var newData = new QR8bitByte(data);
  34. this.dataList.push(newData);
  35. this.dataCache = null;
  36. },
  37. isDark : function(row, col) {
  38. if (row < 0 || this.moduleCount <= row || col < 0 || this.moduleCount <= col) {
  39. throw new Error(row + "," + col);
  40. }
  41. return this.modules[row][col];
  42. },
  43. getModuleCount : function() {
  44. return this.moduleCount;
  45. },
  46. make : function() {
  47. // Calculate automatically typeNumber if provided is < 1
  48. if (this.typeNumber < 1 ){
  49. var typeNumber = 1;
  50. for (typeNumber = 1; typeNumber < 40; typeNumber++) {
  51. var rsBlocks = QRRSBlock.getRSBlocks(typeNumber, this.errorCorrectLevel);
  52. var buffer = new QRBitBuffer();
  53. var totalDataCount = 0;
  54. for (var i = 0; i < rsBlocks.length; i++) {
  55. totalDataCount += rsBlocks[i].dataCount;
  56. }
  57. for (var x = 0; x < this.dataList.length; x++) {
  58. var data = this.dataList[x];
  59. buffer.put(data.mode, 4);
  60. buffer.put(data.getLength(), QRUtil.getLengthInBits(data.mode, typeNumber) );
  61. data.write(buffer);
  62. }
  63. if (buffer.getLengthInBits() <= totalDataCount * 8)
  64. break;
  65. }
  66. this.typeNumber = typeNumber;
  67. }
  68. this.makeImpl(false, this.getBestMaskPattern() );
  69. },
  70. makeImpl : function(test, maskPattern) {
  71. this.moduleCount = this.typeNumber * 4 + 17;
  72. this.modules = new Array(this.moduleCount);
  73. for (var row = 0; row < this.moduleCount; row++) {
  74. this.modules[row] = new Array(this.moduleCount);
  75. for (var col = 0; col < this.moduleCount; col++) {
  76. this.modules[row][col] = null;//(col + row) % 3;
  77. }
  78. }
  79. this.setupPositionProbePattern(0, 0);
  80. this.setupPositionProbePattern(this.moduleCount - 7, 0);
  81. this.setupPositionProbePattern(0, this.moduleCount - 7);
  82. this.setupPositionAdjustPattern();
  83. this.setupTimingPattern();
  84. this.setupTypeInfo(test, maskPattern);
  85. if (this.typeNumber >= 7) {
  86. this.setupTypeNumber(test);
  87. }
  88. if (this.dataCache === null) {
  89. this.dataCache = QRCode.createData(this.typeNumber, this.errorCorrectLevel, this.dataList);
  90. }
  91. this.mapData(this.dataCache, maskPattern);
  92. },
  93. setupPositionProbePattern : function(row, col) {
  94. for (var r = -1; r <= 7; r++) {
  95. if (row + r <= -1 || this.moduleCount <= row + r) continue;
  96. for (var c = -1; c <= 7; c++) {
  97. if (col + c <= -1 || this.moduleCount <= col + c) continue;
  98. if ( (0 <= r && r <= 6 && (c === 0 || c === 6) ) ||
  99. (0 <= c && c <= 6 && (r === 0 || r === 6) ) ||
  100. (2 <= r && r <= 4 && 2 <= c && c <= 4) ) {
  101. this.modules[row + r][col + c] = true;
  102. } else {
  103. this.modules[row + r][col + c] = false;
  104. }
  105. }
  106. }
  107. },
  108. getBestMaskPattern : function() {
  109. var minLostPoint = 0;
  110. var pattern = 0;
  111. for (var i = 0; i < 8; i++) {
  112. this.makeImpl(true, i);
  113. var lostPoint = QRUtil.getLostPoint(this);
  114. if (i === 0 || minLostPoint > lostPoint) {
  115. minLostPoint = lostPoint;
  116. pattern = i;
  117. }
  118. }
  119. return pattern;
  120. },
  121. createMovieClip : function(target_mc, instance_name, depth) {
  122. var qr_mc = target_mc.createEmptyMovieClip(instance_name, depth);
  123. var cs = 1;
  124. this.make();
  125. for (var row = 0; row < this.modules.length; row++) {
  126. var y = row * cs;
  127. for (var col = 0; col < this.modules[row].length; col++) {
  128. var x = col * cs;
  129. var dark = this.modules[row][col];
  130. if (dark) {
  131. qr_mc.beginFill(0, 100);
  132. qr_mc.moveTo(x, y);
  133. qr_mc.lineTo(x + cs, y);
  134. qr_mc.lineTo(x + cs, y + cs);
  135. qr_mc.lineTo(x, y + cs);
  136. qr_mc.endFill();
  137. }
  138. }
  139. }
  140. return qr_mc;
  141. },
  142. setupTimingPattern : function() {
  143. for (var r = 8; r < this.moduleCount - 8; r++) {
  144. if (this.modules[r][6] !== null) {
  145. continue;
  146. }
  147. this.modules[r][6] = (r % 2 === 0);
  148. }
  149. for (var c = 8; c < this.moduleCount - 8; c++) {
  150. if (this.modules[6][c] !== null) {
  151. continue;
  152. }
  153. this.modules[6][c] = (c % 2 === 0);
  154. }
  155. },
  156. setupPositionAdjustPattern : function() {
  157. var pos = QRUtil.getPatternPosition(this.typeNumber);
  158. for (var i = 0; i < pos.length; i++) {
  159. for (var j = 0; j < pos.length; j++) {
  160. var row = pos[i];
  161. var col = pos[j];
  162. if (this.modules[row][col] !== null) {
  163. continue;
  164. }
  165. for (var r = -2; r <= 2; r++) {
  166. for (var c = -2; c <= 2; c++) {
  167. if (Math.abs(r) === 2 ||
  168. Math.abs(c) === 2 ||
  169. (r === 0 && c === 0) ) {
  170. this.modules[row + r][col + c] = true;
  171. } else {
  172. this.modules[row + r][col + c] = false;
  173. }
  174. }
  175. }
  176. }
  177. }
  178. },
  179. setupTypeNumber : function(test) {
  180. var bits = QRUtil.getBCHTypeNumber(this.typeNumber);
  181. var mod;
  182. for (var i = 0; i < 18; i++) {
  183. mod = (!test && ( (bits >> i) & 1) === 1);
  184. this.modules[Math.floor(i / 3)][i % 3 + this.moduleCount - 8 - 3] = mod;
  185. }
  186. for (var x = 0; x < 18; x++) {
  187. mod = (!test && ( (bits >> x) & 1) === 1);
  188. this.modules[x % 3 + this.moduleCount - 8 - 3][Math.floor(x / 3)] = mod;
  189. }
  190. },
  191. setupTypeInfo : function(test, maskPattern) {
  192. var data = (this.errorCorrectLevel << 3) | maskPattern;
  193. var bits = QRUtil.getBCHTypeInfo(data);
  194. var mod;
  195. // vertical
  196. for (var v = 0; v < 15; v++) {
  197. mod = (!test && ( (bits >> v) & 1) === 1);
  198. if (v < 6) {
  199. this.modules[v][8] = mod;
  200. } else if (v < 8) {
  201. this.modules[v + 1][8] = mod;
  202. } else {
  203. this.modules[this.moduleCount - 15 + v][8] = mod;
  204. }
  205. }
  206. // horizontal
  207. for (var h = 0; h < 15; h++) {
  208. mod = (!test && ( (bits >> h) & 1) === 1);
  209. if (h < 8) {
  210. this.modules[8][this.moduleCount - h - 1] = mod;
  211. } else if (h < 9) {
  212. this.modules[8][15 - h - 1 + 1] = mod;
  213. } else {
  214. this.modules[8][15 - h - 1] = mod;
  215. }
  216. }
  217. // fixed module
  218. this.modules[this.moduleCount - 8][8] = (!test);
  219. },
  220. mapData : function(data, maskPattern) {
  221. var inc = -1;
  222. var row = this.moduleCount - 1;
  223. var bitIndex = 7;
  224. var byteIndex = 0;
  225. for (var col = this.moduleCount - 1; col > 0; col -= 2) {
  226. if (col === 6) col--;
  227. while (true) {
  228. for (var c = 0; c < 2; c++) {
  229. if (this.modules[row][col - c] === null) {
  230. var dark = false;
  231. if (byteIndex < data.length) {
  232. dark = ( ( (data[byteIndex] >>> bitIndex) & 1) === 1);
  233. }
  234. var mask = QRUtil.getMask(maskPattern, row, col - c);
  235. if (mask) {
  236. dark = !dark;
  237. }
  238. this.modules[row][col - c] = dark;
  239. bitIndex--;
  240. if (bitIndex === -1) {
  241. byteIndex++;
  242. bitIndex = 7;
  243. }
  244. }
  245. }
  246. row += inc;
  247. if (row < 0 || this.moduleCount <= row) {
  248. row -= inc;
  249. inc = -inc;
  250. break;
  251. }
  252. }
  253. }
  254. }
  255. };
  256. QRCode.PAD0 = 0xEC;
  257. QRCode.PAD1 = 0x11;
  258. QRCode.createData = function(typeNumber, errorCorrectLevel, dataList) {
  259. var rsBlocks = QRRSBlock.getRSBlocks(typeNumber, errorCorrectLevel);
  260. var buffer = new QRBitBuffer();
  261. for (var i = 0; i < dataList.length; i++) {
  262. var data = dataList[i];
  263. buffer.put(data.mode, 4);
  264. buffer.put(data.getLength(), QRUtil.getLengthInBits(data.mode, typeNumber) );
  265. data.write(buffer);
  266. }
  267. // calc num max data.
  268. var totalDataCount = 0;
  269. for (var x = 0; x < rsBlocks.length; x++) {
  270. totalDataCount += rsBlocks[x].dataCount;
  271. }
  272. if (buffer.getLengthInBits() > totalDataCount * 8) {
  273. throw new Error("code length overflow. (" +
  274. buffer.getLengthInBits() +
  275. ">" +
  276. totalDataCount * 8 +
  277. ")");
  278. }
  279. // end code
  280. if (buffer.getLengthInBits() + 4 <= totalDataCount * 8) {
  281. buffer.put(0, 4);
  282. }
  283. // padding
  284. while (buffer.getLengthInBits() % 8 !== 0) {
  285. buffer.putBit(false);
  286. }
  287. // padding
  288. while (true) {
  289. if (buffer.getLengthInBits() >= totalDataCount * 8) {
  290. break;
  291. }
  292. buffer.put(QRCode.PAD0, 8);
  293. if (buffer.getLengthInBits() >= totalDataCount * 8) {
  294. break;
  295. }
  296. buffer.put(QRCode.PAD1, 8);
  297. }
  298. return QRCode.createBytes(buffer, rsBlocks);
  299. };
  300. QRCode.createBytes = function(buffer, rsBlocks) {
  301. var offset = 0;
  302. var maxDcCount = 0;
  303. var maxEcCount = 0;
  304. var dcdata = new Array(rsBlocks.length);
  305. var ecdata = new Array(rsBlocks.length);
  306. for (var r = 0; r < rsBlocks.length; r++) {
  307. var dcCount = rsBlocks[r].dataCount;
  308. var ecCount = rsBlocks[r].totalCount - dcCount;
  309. maxDcCount = Math.max(maxDcCount, dcCount);
  310. maxEcCount = Math.max(maxEcCount, ecCount);
  311. dcdata[r] = new Array(dcCount);
  312. for (var i = 0; i < dcdata[r].length; i++) {
  313. dcdata[r][i] = 0xff & buffer.buffer[i + offset];
  314. }
  315. offset += dcCount;
  316. var rsPoly = QRUtil.getErrorCorrectPolynomial(ecCount);
  317. var rawPoly = new QRPolynomial(dcdata[r], rsPoly.getLength() - 1);
  318. var modPoly = rawPoly.mod(rsPoly);
  319. ecdata[r] = new Array(rsPoly.getLength() - 1);
  320. for (var x = 0; x < ecdata[r].length; x++) {
  321. var modIndex = x + modPoly.getLength() - ecdata[r].length;
  322. ecdata[r][x] = (modIndex >= 0)? modPoly.get(modIndex) : 0;
  323. }
  324. }
  325. var totalCodeCount = 0;
  326. for (var y = 0; y < rsBlocks.length; y++) {
  327. totalCodeCount += rsBlocks[y].totalCount;
  328. }
  329. var data = new Array(totalCodeCount);
  330. var index = 0;
  331. for (var z = 0; z < maxDcCount; z++) {
  332. for (var s = 0; s < rsBlocks.length; s++) {
  333. if (z < dcdata[s].length) {
  334. data[index++] = dcdata[s][z];
  335. }
  336. }
  337. }
  338. for (var xx = 0; xx < maxEcCount; xx++) {
  339. for (var t = 0; t < rsBlocks.length; t++) {
  340. if (xx < ecdata[t].length) {
  341. data[index++] = ecdata[t][xx];
  342. }
  343. }
  344. }
  345. return data;
  346. };
  347. module.exports = QRCode;