diff options
Diffstat (limited to 'share/extensions/render_barcode_qrcode.py')
| -rwxr-xr-x | share/extensions/render_barcode_qrcode.py | 1352 |
1 files changed, 1352 insertions, 0 deletions
diff --git a/share/extensions/render_barcode_qrcode.py b/share/extensions/render_barcode_qrcode.py new file mode 100755 index 0000000..5593b5e --- /dev/null +++ b/share/extensions/render_barcode_qrcode.py @@ -0,0 +1,1352 @@ +#!/usr/bin/env python +# coding=utf-8 +# +# Copyright (C) 2009 Kazuhiko Arase (http://www.d-project.com/) +# 2010 Bulia Byak <buliabyak@gmail.com> +# 2018 Kirill Okhotnikov <kirill.okhotnikov@gmail.com> +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software +# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110, USA. +# +""" +Provide the QR Code rendering. +""" + +from __future__ import print_function + +from itertools import product + +import inkex +from inkex import Group, Rectangle, Use, PathElement +from inkex.localization import inkex_gettext as _ + + +class QRLengthError(Exception): + def __init__(self, message): + self.message = message + + +class QRCode(object): + PAD0 = 0xEC + PAD1 = 0x11 + + def __init__(self, correction): + self.typeNumber = 1 + self.errorCorrectLevel = correction + self.qrDataList = [] + self.modules = [] + self.moduleCount = 0 + + def getTypeNumber(self): + return self.typeNumber + + def setTypeNumber(self, typeNumber): + self.typeNumber = typeNumber + + def clearData(self): + self.qrDataList = [] + + def addData(self, data): + self.qrDataList.append(data) + + def getDataCount(self): + return len(self.qrDataList) + + def getData(self, index): + return self.qrDataList[index] + + def isDark(self, row, col): + return self.modules[row][col] if self.modules[row][col] is not None else False + + def getModuleCount(self): + return self.moduleCount + + def make(self): + self._make(False, self._getBestMaskPattern()) + + def _getBestMaskPattern(self): + minLostPoint = 0 + pattern = 0 + for i in range(8): + self._make(True, i) + lostPoint = QRUtil.getLostPoint(self) + if i == 0 or minLostPoint > lostPoint: + minLostPoint = lostPoint + pattern = i + return pattern + + def _make(self, test, maskPattern): + + self.moduleCount = self.typeNumber * 4 + 17 + self.modules = [[None] * self.moduleCount for i in range(self.moduleCount)] + + self._setupPositionProbePattern(0, 0) + self._setupPositionProbePattern(self.moduleCount - 7, 0) + self._setupPositionProbePattern(0, self.moduleCount - 7) + + self._setupPositionAdjustPattern() + self._setupTimingPattern() + + self._setupTypeInfo(test, maskPattern) + + if self.typeNumber >= 7: + self._setupTypeNumber(test) + + data = QRCode._createData( + self.typeNumber, self.errorCorrectLevel, self.qrDataList + ) + + self._mapData(data, maskPattern) + + def _mapData(self, data, maskPattern): + + rows = list(range(self.moduleCount)) + cols = [ + col - 1 if col <= 6 else col for col in range(self.moduleCount - 1, 0, -2) + ] + maskFunc = QRUtil.getMaskFunction(maskPattern) + + byteIndex = 0 + bitIndex = 7 + + for col in cols: + rows.reverse() + for row in rows: + for c in range(2): + if self.modules[row][col - c] is None: + + dark = False + if byteIndex < len(data): + dark = ((data[byteIndex] >> bitIndex) & 1) == 1 + if maskFunc(row, col - c): + dark = not dark + self.modules[row][col - c] = dark + + bitIndex -= 1 + if bitIndex == -1: + byteIndex += 1 + bitIndex = 7 + + def _setupPositionAdjustPattern(self): + pos = QRUtil.getPatternPosition(self.typeNumber) + for row in pos: + for col in pos: + if self.modules[row][col] is not None: + continue + for r in range(-2, 3): + for c in range(-2, 3): + self.modules[row + r][col + c] = ( + r == -2 + or r == 2 + or c == -2 + or c == 2 + or (r == 0 and c == 0) + ) + + def _setupPositionProbePattern(self, row, col): + for r in range(-1, 8): + for c in range(-1, 8): + if ( + row + r <= -1 + or self.moduleCount <= row + r + or col + c <= -1 + or self.moduleCount <= col + c + ): + continue + self.modules[row + r][col + c] = ( + (0 <= r <= 6 and (c == 0 or c == 6)) + or (0 <= c <= 6 and (r == 0 or r == 6)) + or (2 <= r <= 4 and 2 <= c <= 4) + ) + + def _setupTimingPattern(self): + for r in range(8, self.moduleCount - 8): + if self.modules[r][6] is not None: + continue + self.modules[r][6] = r % 2 == 0 + for c in range(8, self.moduleCount - 8): + if self.modules[6][c] is not None: + continue + self.modules[6][c] = c % 2 == 0 + + def _setupTypeNumber(self, test): + bits = QRUtil.getBCHTypeNumber(self.typeNumber) + for i in range(18): + self.modules[i // 3][i % 3 + self.moduleCount - 8 - 3] = ( + not test and ((bits >> i) & 1) == 1 + ) + for i in range(18): + self.modules[i % 3 + self.moduleCount - 8 - 3][i // 3] = ( + not test and ((bits >> i) & 1) == 1 + ) + + def _setupTypeInfo(self, test, maskPattern): + + data = (self.errorCorrectLevel << 3) | maskPattern + bits = QRUtil.getBCHTypeInfo(data) + + # vertical + for i in range(15): + mod = not test and ((bits >> i) & 1) == 1 + if i < 6: + self.modules[i][8] = mod + elif i < 8: + self.modules[i + 1][8] = mod + else: + self.modules[self.moduleCount - 15 + i][8] = mod + + # horizontal + for i in range(15): + mod = not test and ((bits >> i) & 1) == 1 + if i < 8: + self.modules[8][self.moduleCount - i - 1] = mod + elif i < 9: + self.modules[8][15 - i - 1 + 1] = mod + else: + self.modules[8][15 - i - 1] = mod + + # fixed + self.modules[self.moduleCount - 8][8] = not test + + @staticmethod + def _createData(typeNumber, errorCorrectLevel, dataArray): + + rsBlocks = RSBlock.getRSBlocks(typeNumber, errorCorrectLevel) + + buffer = BitBuffer() + + for data in dataArray: + buffer.put(data.getMode(), 4) + buffer.put(data.getLength(), data.getLengthInBits(typeNumber)) + data.write(buffer) + + totalDataCount = sum(rsBlock.getDataCount() for rsBlock in rsBlocks) + + if buffer.getLengthInBits() > totalDataCount * 8: + raise QRLengthError( + "code length overflow. (%s>%s)" + % (buffer.getLengthInBits(), totalDataCount * 8) + ) + + # end code + if buffer.getLengthInBits() + 4 <= totalDataCount * 8: + buffer.put(0, 4) + + # padding + while buffer.getLengthInBits() % 8 != 0: + buffer.put(False, 1) + + # padding + while True: + if buffer.getLengthInBits() >= totalDataCount * 8: + break + buffer.put(QRCode.PAD0, 8) + if buffer.getLengthInBits() >= totalDataCount * 8: + break + buffer.put(QRCode.PAD1, 8) + + return QRCode._createBytes(buffer, rsBlocks) + + @staticmethod + def _createBytes(buffer, rsBlocks): + + offset = 0 + + maxDcCount = 0 + maxEcCount = 0 + + dcdata = [None] * len(rsBlocks) + ecdata = [None] * len(rsBlocks) + + for r in range(len(rsBlocks)): + + dcCount = rsBlocks[r].getDataCount() + ecCount = rsBlocks[r].getTotalCount() - dcCount + + maxDcCount = max(maxDcCount, dcCount) + maxEcCount = max(maxEcCount, ecCount) + + dcdata[r] = [0] * dcCount + for i in range(len(dcdata[r])): + dcdata[r][i] = 0xFF & buffer.getBuffer()[i + offset] + offset += dcCount + + rsPoly = QRUtil.getErrorCorrectPolynomial(ecCount) + rawPoly = Polynomial(dcdata[r], rsPoly.getLength() - 1) + + modPoly = rawPoly.mod(rsPoly) + ecdata[r] = [0] * (rsPoly.getLength() - 1) + for i in range(len(ecdata[r])): + modIndex = i + modPoly.getLength() - len(ecdata[r]) + ecdata[r][i] = modPoly.get(modIndex) if modIndex >= 0 else 0 + + totalCodeCount = sum(rsBlock.getTotalCount() for rsBlock in rsBlocks) + + data = [0] * totalCodeCount + + index = 0 + + for i in range(maxDcCount): + for r in range(len(rsBlocks)): + if i < len(dcdata[r]): + data[index] = dcdata[r][i] + index += 1 + + for i in range(maxEcCount): + for r in range(len(rsBlocks)): + if i < len(ecdata[r]): + data[index] = ecdata[r][i] + index += 1 + + return data + + @staticmethod + def getMinimumQRCode(data, errorCorrectLevel): + qr = QRCode(correction=errorCorrectLevel) + qr.addData(data) + lv = 1 + rv = 40 + while rv - lv > 0: + mid = (3 * lv + rv) // 4 + qr.setTypeNumber(mid) + try: + qr.make() + except QRLengthError: + if mid == 40: + raise inkex.AbortExtension( + _("The string is too large to represent as QR code") + ) + lv = mid + 1 + else: + rv = mid + + qr.setTypeNumber(rv) + qr.make() + return qr + + +class Mode(object): + MODE_NUMBER = 1 << 0 + MODE_ALPHA_NUM = 1 << 1 + MODE_8BIT_BYTE = 1 << 2 + MODE_KANJI = 1 << 3 + + +class MaskPattern(object): + PATTERN000 = 0 + PATTERN001 = 1 + PATTERN010 = 2 + PATTERN011 = 3 + PATTERN100 = 4 + PATTERN101 = 5 + PATTERN110 = 6 + PATTERN111 = 7 + + +class QRUtil(object): + @staticmethod + def getPatternPosition(typeNumber): + return QRUtil.PATTERN_POSITION_TABLE[typeNumber - 1] + + PATTERN_POSITION_TABLE = [ + [], + [6, 18], + [6, 22], + [6, 26], + [6, 30], + [6, 34], + [6, 22, 38], + [6, 24, 42], + [6, 26, 46], + [6, 28, 50], + [6, 30, 54], + [6, 32, 58], + [6, 34, 62], + [6, 26, 46, 66], + [6, 26, 48, 70], + [6, 26, 50, 74], + [6, 30, 54, 78], + [6, 30, 56, 82], + [6, 30, 58, 86], + [6, 34, 62, 90], + [6, 28, 50, 72, 94], + [6, 26, 50, 74, 98], + [6, 30, 54, 78, 102], + [6, 28, 54, 80, 106], + [6, 32, 58, 84, 110], + [6, 30, 58, 86, 114], + [6, 34, 62, 90, 118], + [6, 26, 50, 74, 98, 122], + [6, 30, 54, 78, 102, 126], + [6, 26, 52, 78, 104, 130], + [6, 30, 56, 82, 108, 134], + [6, 34, 60, 86, 112, 138], + [6, 30, 58, 86, 114, 142], + [6, 34, 62, 90, 118, 146], + [6, 30, 54, 78, 102, 126, 150], + [6, 24, 50, 76, 102, 128, 154], + [6, 28, 54, 80, 106, 132, 158], + [6, 32, 58, 84, 110, 136, 162], + [6, 26, 54, 82, 110, 138, 166], + [6, 30, 58, 86, 114, 142, 170], + ] + + @staticmethod + def getErrorCorrectPolynomial(errorCorrectLength): + a = Polynomial([1]) + for i in range(errorCorrectLength): + a = a.multiply(Polynomial([1, QRMath.gexp(i)])) + return a + + @staticmethod + def getMaskFunction(maskPattern): + return { + MaskPattern.PATTERN000: lambda i, j: (i + j) % 2 == 0, + MaskPattern.PATTERN001: lambda i, j: i % 2 == 0, + MaskPattern.PATTERN010: lambda i, j: j % 3 == 0, + MaskPattern.PATTERN011: lambda i, j: (i + j) % 3 == 0, + MaskPattern.PATTERN100: lambda i, j: (i // 2 + j // 3) % 2 == 0, + MaskPattern.PATTERN101: lambda i, j: (i * j) % 2 + (i * j) % 3 == 0, + MaskPattern.PATTERN110: lambda i, j: ((i * j) % 2 + (i * j) % 3) % 2 == 0, + MaskPattern.PATTERN111: lambda i, j: ((i * j) % 3 + (i + j) % 2) % 2 == 0, + }[maskPattern] + + @staticmethod + def getLostPoint(qrcode): + + moduleCount = qrcode.getModuleCount() + lostPoint = 0 + + # LEVEL1 + for row in range(moduleCount): + for col in range(moduleCount): + sameCount = 0 + dark = qrcode.isDark(row, col) + for r in range(-1, 2): + if row + r < 0 or moduleCount <= row + r: + continue + for c in range(-1, 2): + if col + c < 0 or moduleCount <= col + c: + continue + if r == 0 and c == 0: + continue + if dark == qrcode.isDark(row + r, col + c): + sameCount += 1 + if sameCount > 5: + lostPoint += 3 + sameCount - 5 + + # LEVEL2 + for row in range(moduleCount - 1): + for col in range(moduleCount - 1): + count = 0 + if qrcode.isDark(row, col): + count += 1 + if qrcode.isDark(row + 1, col): + count += 1 + if qrcode.isDark(row, col + 1): + count += 1 + if qrcode.isDark(row + 1, col + 1): + count += 1 + if count == 0 or count == 4: + lostPoint += 3 + + # LEVEL3 + for row in range(moduleCount): + for col in range(moduleCount - 6): + if ( + qrcode.isDark(row, col) + and not qrcode.isDark(row, col + 1) + and qrcode.isDark(row, col + 2) + and qrcode.isDark(row, col + 3) + and qrcode.isDark(row, col + 4) + and not qrcode.isDark(row, col + 5) + and qrcode.isDark(row, col + 6) + ): + lostPoint += 40 + + for col in range(moduleCount): + for row in range(moduleCount - 6): + if ( + qrcode.isDark(row, col) + and not qrcode.isDark(row + 1, col) + and qrcode.isDark(row + 2, col) + and qrcode.isDark(row + 3, col) + and qrcode.isDark(row + 4, col) + and not qrcode.isDark(row + 5, col) + and qrcode.isDark(row + 6, col) + ): + lostPoint += 40 + + # LEVEL4 + darkCount = 0 + for col in range(moduleCount): + for row in range(moduleCount): + if qrcode.isDark(row, col): + darkCount += 1 + + ratio = abs(100 * darkCount // moduleCount // moduleCount - 50) // 5 + lostPoint += ratio * 10 + + return lostPoint + + G15 = (1 << 10) | (1 << 8) | (1 << 5) | (1 << 4) | (1 << 2) | (1 << 1) | (1 << 0) + G18 = ( + (1 << 12) + | (1 << 11) + | (1 << 10) + | (1 << 9) + | (1 << 8) + | (1 << 5) + | (1 << 2) + | (1 << 0) + ) + G15_MASK = (1 << 14) | (1 << 12) | (1 << 10) | (1 << 4) | (1 << 1) + + @staticmethod + def getBCHTypeInfo(data): + d = data << 10 + while QRUtil.getBCHDigit(d) - QRUtil.getBCHDigit(QRUtil.G15) >= 0: + d ^= QRUtil.G15 << (QRUtil.getBCHDigit(d) - QRUtil.getBCHDigit(QRUtil.G15)) + return ((data << 10) | d) ^ QRUtil.G15_MASK + + @staticmethod + def getBCHTypeNumber(data): + d = data << 12 + while QRUtil.getBCHDigit(d) - QRUtil.getBCHDigit(QRUtil.G18) >= 0: + d ^= QRUtil.G18 << (QRUtil.getBCHDigit(d) - QRUtil.getBCHDigit(QRUtil.G18)) + return (data << 12) | d + + @staticmethod + def getBCHDigit(data): + digit = 0 + while data != 0: + digit += 1 + data >>= 1 + return digit + + +class QRData(object): + def __init__(self, mode, data): + self._mode = mode + self.data = data + + def getMode(self): + return self._mode + + def getData(self): + return self.data + + def getLength(self): + return len(self.data) + + # @abstractmethod + def write(self, buffer): + pass + + def getLengthInBits(self, type): + if 1 <= type < 10: # 1 - 9 + return { + Mode.MODE_NUMBER: 10, + Mode.MODE_ALPHA_NUM: 9, + Mode.MODE_8BIT_BYTE: 8, + Mode.MODE_KANJI: 8, + }[self._mode] + + elif type < 27: # 10 - 26 + return { + Mode.MODE_NUMBER: 12, + Mode.MODE_ALPHA_NUM: 11, + Mode.MODE_8BIT_BYTE: 16, + Mode.MODE_KANJI: 10, + }[self._mode] + + elif type < 41: # 27 - 40 + return { + Mode.MODE_NUMBER: 14, + Mode.MODE_ALPHA_NUM: 13, + Mode.MODE_8BIT_BYTE: 16, + Mode.MODE_KANJI: 12, + }[self._mode] + + else: + raise Exception("type:%s" % type) + + +class QR8BitByte(QRData): + def __init__(self, data): + super(QR8BitByte, self).__init__(Mode.MODE_8BIT_BYTE, data) + if isinstance(data, str): + data = data.encode("ascii", "ignore") + if not isinstance(data, bytes): + raise ValueError("Data must be in bytes!") + + def write(self, buffer): + for d in self.data: + buffer.put(ord(d), 8) + + +class QRAlphaNum(QRData): + def __init__(self, data): + super(QRAlphaNum, self).__init__(Mode.MODE_ALPHA_NUM, data) + + def write(self, buffer): + i = 0 + while i + 1 < len(self.data): + buffer.put( + QRAlphaNum._getCode(self.data[i]) * 45 + + QRAlphaNum._getCode(self.data[i + 1]), + 11, + ) + i += 2 + if i < len(self.data): + buffer.put(QRAlphaNum._getCode(self.data[i]), 6) + + @staticmethod + def _getCode(c): + if "0" <= c and c <= "9": + return ord(c) - ord("0") + elif "A" <= c and c <= "Z": + return ord(c) - ord("A") + 10 + else: + dct = { + " ": 36, + "$": 37, + "%": 38, + "*": 39, + "+": 40, + "-": 41, + ".": 42, + "/": 43, + ":": 44, + } + if c in dct.keys(): + return dct[c] + else: + raise inkex.AbortExtension( + _( + "Wrong symbol '{}' in alphanumeric representation: Should be [A-Z, 0-9] or {}" + ).format(c, dct.keys()) + ) + + +class QRNumber(QRData): + def __init__(self, data): + super(QRNumber, self).__init__(Mode.MODE_NUMBER, data) + + def write(self, buffer): + i = 0 + try: + while i + 2 < len(self.data): + num = int(self.data[i : i + 3]) + buffer.put(num, 10) + i += 3 + + if i < len(self.data): + ln = 4 if len(self.data) - i == 1 else 7 + buffer.put(int(self.data[i:]), ln) + except: + raise ValueError("The string '{}' is not a valid number".format(self.data)) + + +class QRKanji(QRData): + def __init__(self, data): + super(QRKanji, self).__init__(Mode.MODE_KANJI, data) + raise RuntimeError("Class QRKanji is not implemented") + + +class QRMath(object): + EXP_TABLE = None + LOG_TABLE = None + + @staticmethod + def _init(): + + QRMath.EXP_TABLE = [0] * 256 + for i in range(256): + QRMath.EXP_TABLE[i] = ( + 1 << i + if i < 8 + else QRMath.EXP_TABLE[i - 4] + ^ QRMath.EXP_TABLE[i - 5] + ^ QRMath.EXP_TABLE[i - 6] + ^ QRMath.EXP_TABLE[i - 8] + ) + + QRMath.LOG_TABLE = [0] * 256 + for i in range(255): + QRMath.LOG_TABLE[QRMath.EXP_TABLE[i]] = i + + @staticmethod + def glog(n): + if n < 1: + raise Exception("log(%s)" % n) + return QRMath.LOG_TABLE[n] + + @staticmethod + def gexp(n): + while n < 0: + n += 255 + while n >= 256: + n -= 255 + return QRMath.EXP_TABLE[n] + + +# initialize statics +QRMath._init() + + +class Polynomial(object): + def __init__(self, num, shift=0): + offset = 0 + length = len(num) + while offset < length and num[offset] == 0: + offset += 1 + self.num = num[offset:] + [0] * shift + + def get(self, index): + return self.num[index] + + def getLength(self): + return len(self.num) + + def __repr__(self): + return ",".join([str(self.get(i)) for i in range(self.getLength())]) + + def toLogString(self): + return ",".join( + [str(QRMath.glog(self.get(i))) for i in range(self.getLength())] + ) + + def multiply(self, e): + num = [0] * (self.getLength() + e.getLength() - 1) + for i in range(self.getLength()): + for j in range(e.getLength()): + num[i + j] ^= QRMath.gexp( + QRMath.glog(self.get(i)) + QRMath.glog(e.get(j)) + ) + return Polynomial(num) + + def mod(self, e): + if self.getLength() - e.getLength() < 0: + return self + ratio = QRMath.glog(self.get(0)) - QRMath.glog(e.get(0)) + num = self.num[:] + for i in range(e.getLength()): + num[i] ^= QRMath.gexp(QRMath.glog(e.get(i)) + ratio) + return Polynomial(num).mod(e) + + +class RSBlock(object): + RS_BLOCK_TABLE = [ + # L + # M + # Q + # H + # 1 + [1, 26, 19], + [1, 26, 16], + [1, 26, 13], + [1, 26, 9], + # 2 + [1, 44, 34], + [1, 44, 28], + [1, 44, 22], + [1, 44, 16], + # 3 + [1, 70, 55], + [1, 70, 44], + [2, 35, 17], + [2, 35, 13], + # 4 + [1, 100, 80], + [2, 50, 32], + [2, 50, 24], + [4, 25, 9], + # 5 + [1, 134, 108], + [2, 67, 43], + [2, 33, 15, 2, 34, 16], + [2, 33, 11, 2, 34, 12], + # 6 + [2, 86, 68], + [4, 43, 27], + [4, 43, 19], + [4, 43, 15], + # 7 + [2, 98, 78], + [4, 49, 31], + [2, 32, 14, 4, 33, 15], + [4, 39, 13, 1, 40, 14], + # 8 + [2, 121, 97], + [2, 60, 38, 2, 61, 39], + [4, 40, 18, 2, 41, 19], + [4, 40, 14, 2, 41, 15], + # 9 + [2, 146, 116], + [3, 58, 36, 2, 59, 37], + [4, 36, 16, 4, 37, 17], + [4, 36, 12, 4, 37, 13], + # 10 + [2, 86, 68, 2, 87, 69], + [4, 69, 43, 1, 70, 44], + [6, 43, 19, 2, 44, 20], + [6, 43, 15, 2, 44, 16], + # 11 + [4, 101, 81], + [1, 80, 50, 4, 81, 51], + [4, 50, 22, 4, 51, 23], + [3, 36, 12, 8, 37, 13], + # 12 + [2, 116, 92, 2, 117, 93], + [6, 58, 36, 2, 59, 37], + [4, 46, 20, 6, 47, 21], + [7, 42, 14, 4, 43, 15], + # 13 + [4, 133, 107], + [8, 59, 37, 1, 60, 38], + [8, 44, 20, 4, 45, 21], + [12, 33, 11, 4, 34, 12], + # 14 + [3, 145, 115, 1, 146, 116], + [4, 64, 40, 5, 65, 41], + [11, 36, 16, 5, 37, 17], + [11, 36, 12, 5, 37, 13], + # 15 + [5, 109, 87, 1, 110, 88], + [5, 65, 41, 5, 66, 42], + [5, 54, 24, 7, 55, 25], + [11, 36, 12, 7, 37, 13], + # 16 + [5, 122, 98, 1, 123, 99], + [7, 73, 45, 3, 74, 46], + [15, 43, 19, 2, 44, 20], + [3, 45, 15, 13, 46, 16], + # 17 + [1, 135, 107, 5, 136, 108], + [10, 74, 46, 1, 75, 47], + [1, 50, 22, 15, 51, 23], + [2, 42, 14, 17, 43, 15], + # 18 + [5, 150, 120, 1, 151, 121], + [9, 69, 43, 4, 70, 44], + [17, 50, 22, 1, 51, 23], + [2, 42, 14, 19, 43, 15], + # 19 + [3, 141, 113, 4, 142, 114], + [3, 70, 44, 11, 71, 45], + [17, 47, 21, 4, 48, 22], + [9, 39, 13, 16, 40, 14], + # 20 + [3, 135, 107, 5, 136, 108], + [3, 67, 41, 13, 68, 42], + [15, 54, 24, 5, 55, 25], + [15, 43, 15, 10, 44, 16], + # 21 + [4, 144, 116, 4, 145, 117], + [17, 68, 42], + [17, 50, 22, 6, 51, 23], + [19, 46, 16, 6, 47, 17], + # 22 + [2, 139, 111, 7, 140, 112], + [17, 74, 46], + [7, 54, 24, 16, 55, 25], + [34, 37, 13], + # 23 + [4, 151, 121, 5, 152, 122], + [4, 75, 47, 14, 76, 48], + [11, 54, 24, 14, 55, 25], + [16, 45, 15, 14, 46, 16], + # 24 + [6, 147, 117, 4, 148, 118], + [6, 73, 45, 14, 74, 46], + [11, 54, 24, 16, 55, 25], + [30, 46, 16, 2, 47, 17], + # 25 + [8, 132, 106, 4, 133, 107], + [8, 75, 47, 13, 76, 48], + [7, 54, 24, 22, 55, 25], + [22, 45, 15, 13, 46, 16], + # 26 + [10, 142, 114, 2, 143, 115], + [19, 74, 46, 4, 75, 47], + [28, 50, 22, 6, 51, 23], + [33, 46, 16, 4, 47, 17], + # 27 + [8, 152, 122, 4, 153, 123], + [22, 73, 45, 3, 74, 46], + [8, 53, 23, 26, 54, 24], + [12, 45, 15, 28, 46, 16], + # 28 + [3, 147, 117, 10, 148, 118], + [3, 73, 45, 23, 74, 46], + [4, 54, 24, 31, 55, 25], + [11, 45, 15, 31, 46, 16], + # 29 + [7, 146, 116, 7, 147, 117], + [21, 73, 45, 7, 74, 46], + [1, 53, 23, 37, 54, 24], + [19, 45, 15, 26, 46, 16], + # 30 + [5, 145, 115, 10, 146, 116], + [19, 75, 47, 10, 76, 48], + [15, 54, 24, 25, 55, 25], + [23, 45, 15, 25, 46, 16], + # 31 + [13, 145, 115, 3, 146, 116], + [2, 74, 46, 29, 75, 47], + [42, 54, 24, 1, 55, 25], + [23, 45, 15, 28, 46, 16], + # 32 + [17, 145, 115], + [10, 74, 46, 23, 75, 47], + [10, 54, 24, 35, 55, 25], + [19, 45, 15, 35, 46, 16], + # 33 + [17, 145, 115, 1, 146, 116], + [14, 74, 46, 21, 75, 47], + [29, 54, 24, 19, 55, 25], + [11, 45, 15, 46, 46, 16], + # 34 + [13, 145, 115, 6, 146, 116], + [14, 74, 46, 23, 75, 47], + [44, 54, 24, 7, 55, 25], + [59, 46, 16, 1, 47, 17], + # 35 + [12, 151, 121, 7, 152, 122], + [12, 75, 47, 26, 76, 48], + [39, 54, 24, 14, 55, 25], + [22, 45, 15, 41, 46, 16], + # 36 + [6, 151, 121, 14, 152, 122], + [6, 75, 47, 34, 76, 48], + [46, 54, 24, 10, 55, 25], + [2, 45, 15, 64, 46, 16], + # 37 + [17, 152, 122, 4, 153, 123], + [29, 74, 46, 14, 75, 47], + [49, 54, 24, 10, 55, 25], + [24, 45, 15, 46, 46, 16], + # 38 + [4, 152, 122, 18, 153, 123], + [13, 74, 46, 32, 75, 47], + [48, 54, 24, 14, 55, 25], + [42, 45, 15, 32, 46, 16], + # 39 + [20, 147, 117, 4, 148, 118], + [40, 75, 47, 7, 76, 48], + [43, 54, 24, 22, 55, 25], + [10, 45, 15, 67, 46, 16], + # 40 + [19, 148, 118, 6, 149, 119], + [18, 75, 47, 31, 76, 48], + [34, 54, 24, 34, 55, 25], + [20, 45, 15, 61, 46, 16], + ] + + def __init__(self, totalCount, dataCount): + self.totalCount = totalCount + self.dataCount = dataCount + + def getDataCount(self): + return self.dataCount + + def getTotalCount(self): + return self.totalCount + + def __repr__(self): + return "(total=%s,data=%s)" % (self.totalCount, self.dataCount) + + @staticmethod + def getRSBlocks(typeNumber, errorCorrectLevel): + rsBlock = RSBlock.getRsBlockTable(typeNumber, errorCorrectLevel) + length = len(rsBlock) // 3 + list = [] + for i in range(length): + count = rsBlock[i * 3 + 0] + totalCount = rsBlock[i * 3 + 1] + dataCount = rsBlock[i * 3 + 2] + list += [RSBlock(totalCount, dataCount)] * count + return list + + @staticmethod + def getRsBlockTable(typeNumber, errorCorrectLevel): + return { + 1: RSBlock.RS_BLOCK_TABLE[(typeNumber - 1) * 4 + 0], + 0: RSBlock.RS_BLOCK_TABLE[(typeNumber - 1) * 4 + 1], + 3: RSBlock.RS_BLOCK_TABLE[(typeNumber - 1) * 4 + 2], + 2: RSBlock.RS_BLOCK_TABLE[(typeNumber - 1) * 4 + 3], + }[errorCorrectLevel] + + +class BitBuffer(object): + def __init__(self, inclements=32): + self.inclements = inclements + self.buffer = [0] * self.inclements + self.length = 0 + + def getBuffer(self): + return self.buffer + + def getLengthInBits(self): + return self.length + + def get(self, index): + return ((self.buffer[index // 8] >> (7 - index % 8)) & 1) == 1 + + def putBit(self, bit): + if self.length == len(self.buffer) * 8: + self.buffer += [0] * self.inclements + if bit: + self.buffer[self.length // 8] |= 0x80 >> (self.length % 8) + self.length += 1 + + def put(self, num, length): + for i in range(length): + self.putBit(((num >> (length - i - 1)) & 1) == 1) + + def __repr__(self): + return "".join( + "1" if self.get(i) else "0" for i in range(self.getLengthInBits()) + ) + + +class GridDrawer(object): + """Mechanism to draw grids of boxes""" + + def __init__(self, invert_code, smooth_factor): + self.invert_code = invert_code + self.smoothFactor = smooth_factor + self.grid = None + + def set_grid(self, grid): + if len({len(g) for g in grid}) != 1: + raise Exception("The array is not rectangular") + else: + self.grid = grid + + def row_count(self): + return len(self.grid) if self.grid is not None else 0 + + def col_count(self): + return len(self.grid[0]) if self.row_count() > 0 else 0 + + def isDark(self, col, row): + inside = col >= 0 and 0 <= row < self.row_count() and col < self.col_count() + return False if not inside else self.grid[row][col] != self.invert_code + + @staticmethod + def moveByDirection(xyd): + dm = {0: (1, 0), 1: (0, -1), 2: (-1, 0), 3: (0, 1)} + return xyd[0] + dm[xyd[2]][0], xyd[1] + dm[xyd[2]][1] + + @staticmethod + def makeDirectionsTable(): + result = [] + for cfg in product(range(2), repeat=4): + result.append([]) + for d in range(4): + if cfg[3 - d] == 0 and cfg[3 - (d - 1) % 4] != 0: + result[-1].append(d) + return result + + def createVertexesForAdvDrawer(self): + dirTable = self.makeDirectionsTable() + result = [] + # Create vertex + for row in range(self.row_count() + 1): + for col in range(self.col_count() + 1): + indx = ( + (2**0 if self.isDark(col - 0, row - 1) else 0) + + (2**1 if self.isDark(col - 1, row - 1) else 0) + + (2**2 if self.isDark(col - 1, row - 0) else 0) + + (2**3 if self.isDark(col - 0, row - 0) else 0) + ) + + for d in dirTable[indx]: + result.append((col, row, d, len(dirTable[indx]) > 1)) + + return result + + def getSmoothPosition(self, v, extraSmoothFactor=1.0): + vn = self.moveByDirection(v) + sc = extraSmoothFactor * self.smoothFactor / 2.0 + sc1 = 1.0 - sc + return (v[0] * sc1 + vn[0] * sc, v[1] * sc1 + vn[1] * sc), ( + v[0] * sc + vn[0] * sc1, + v[1] * sc + vn[1] * sc1, + ) + + +class QrCode(inkex.GenerateExtension): + """Generate QR Code Extension""" + + def add_arguments(self, pars): + pars.add_argument("--text", default="https://inkscape.org") + pars.add_argument("--typenumber", type=int, default=0) + pars.add_argument("--correctionlevel", type=int, default=0) + pars.add_argument("--qrmode", type=int, default=0) + pars.add_argument("--encoding", default="latin_1") + pars.add_argument("--modulesize", type=float, default=4.0) + pars.add_argument("--invert", type=inkex.Boolean, default="false") + pars.add_argument( + "--drawtype", + default="smooth", + choices=["smooth", "pathpreset", "selection", "symbol"], + ) + pars.add_argument( + "--smoothness", default="neutral", choices=["neutral", "greedy", "proud"] + ) + pars.add_argument("--pathtype", default="simple", choices=["simple", "circle"]) + pars.add_argument("--smoothval", type=float, default=0.2) + pars.add_argument("--symbolid", default="") + pars.add_argument("--groupid", default="") + + def generate(self): + + scale = self.svg.unittouu("1px") # convert to document units + opt = self.options + + if not opt.text: + raise inkex.AbortExtension(_("Please enter an input text")) + elif opt.drawtype == "symbol" and opt.symbolid == "": + raise inkex.AbortExtension(_("Please enter symbol id")) + + # for Python 3 ugly hack to represent bytes as str for Python2 compatibility + text_str = str(opt.text) + cmode = [QR8BitByte, QRNumber, QRAlphaNum, QRKanji][opt.qrmode] + text_data = cmode(bytes(opt.text, opt.encoding).decode("latin_1")) + + grp = Group() + grp.set("inkscape:label", "QR Code: " + text_str) + if opt.groupid: + grp.set("id", opt.groupid) + pos_x, pos_y = self.svg.namedview.center + grp.transform.add_translate(pos_x, pos_y) + if scale: + grp.transform.add_scale(scale) + + # GENERATE THE QRCODE + if opt.typenumber == 0: + # Automatic QR code size` + code = QRCode.getMinimumQRCode(text_data, opt.correctionlevel) + else: + # Manual QR code size + code = QRCode(correction=opt.correctionlevel) + code.setTypeNumber(int(opt.typenumber)) + code.addData(text_data) + code.make() + + self.boxsize = opt.modulesize + self.invert_code = opt.invert + self.margin = 4 + self.draw = GridDrawer(opt.invert, opt.smoothval) + self.draw.set_grid(code.modules) + self.render_svg(grp, opt.drawtype) + return grp + + def render_adv(self, greedy): + + verts = self.draw.createVertexesForAdvDrawer() + qrPathStr = "" + while len(verts) > 0: + vertsIndexStart = len(verts) - 1 + vertsIndexCur = vertsIndexStart + ringIndexes = [] + ci = {} + for i, v in enumerate(verts): + ci.setdefault(v[0], []).append(i) + while True: + ringIndexes.append(vertsIndexCur) + nextPos = self.draw.moveByDirection(verts[vertsIndexCur]) + nextIndexes = [i for i in ci[nextPos[0]] if verts[i][1] == nextPos[1]] + if len(nextIndexes) == 0 or len(nextIndexes) > 2: + raise Exception("Vertex " + str(next_c) + " has no connections") + elif len(nextIndexes) == 1: + vertsIndexNext = nextIndexes[0] + else: + if {verts[nextIndexes[0]][2], verts[nextIndexes[1]][2]} != { + (verts[vertsIndexCur][2] - 1) % 4, + (verts[vertsIndexCur][2] + 1) % 4, + }: + raise Exception( + "Bad next vertex directions " + + str(verts[nextIndexes[0]]) + + str(verts[nextIndexes[1]]) + ) + + # Greedy - CCW turn, proud and neutral CW turn + vertsIndexNext = ( + nextIndexes[0] + if (greedy == "g") + == ( + verts[nextIndexes[0]][2] + == (verts[vertsIndexCur][2] + 1) % 4 + ) + else nextIndexes[1] + ) + + if vertsIndexNext == vertsIndexStart: + break + + vertsIndexCur = vertsIndexNext + + posStart, _ = self.draw.getSmoothPosition(verts[ringIndexes[0]]) + qrPathStr += "M %f,%f " % self.get_svg_pos(posStart[0], posStart[1]) + for ri in range(len(ringIndexes)): + vc = verts[ringIndexes[ri]] + vn = verts[ringIndexes[(ri + 1) % len(ringIndexes)]] + if vn[2] != vc[2]: + if (greedy != "n") or not vn[3]: + # Add bezier + # Opt length http://spencermortensen.com/articles/bezier-circle/ + # c = 0.552284749 + ex = 1 - 0.552284749 + _, bs = self.draw.getSmoothPosition(vc) + _, bp1 = self.draw.getSmoothPosition(vc, ex) + bp2, _ = self.draw.getSmoothPosition(vn, ex) + bf, _ = self.draw.getSmoothPosition(vn) + qrPathStr += "L %f,%f " % self.get_svg_pos(bs[0], bs[1]) + qrPathStr += "C %f,%f %f,%f %f,%f " % ( + self.get_svg_pos(bp1[0], bp1[1]) + + self.get_svg_pos(bp2[0], bp2[1]) + + self.get_svg_pos(bf[0], bf[1]) + ) + else: + # Add straight + qrPathStr += "L %f,%f " % self.get_svg_pos(vn[0], vn[1]) + + qrPathStr += "z " + + # Delete already processed vertex + for i in sorted(ringIndexes, reverse=True): + del verts[i] + + path = PathElement() + path.set("d", qrPathStr) + return path + + def render_obsolete(self): + for row in range(self.draw.row_count()): + for col in range(self.draw.col_count()): + if self.draw.isDark(col, row): + x, y = self.get_svg_pos(col, row) + return Rectangle.new(x, y, self.boxsize, self.boxsize) + + def render_path(self, pointStr): + singlePath = self.get_icon_path_str(pointStr) + pathStr = "" + for row in range(self.draw.row_count()): + for col in range(self.draw.col_count()): + if self.draw.isDark(col, row): + x, y = self.get_svg_pos(col, row) + pathStr += "M %f,%f " % (x, y) + singlePath + " z " + + path = PathElement() + path.set("d", pathStr) + return path + + def render_selection(self): + if len(self.svg.selection) > 0: + self.options.symbolid = self.svg.selection.first().get_id() + else: + raise inkex.AbortExtension(_("Please select an element to clone")) + return self.render_symbol() + + def render_symbol(self): + symbol = self.svg.getElementById(self.options.symbolid) + if symbol is None: + raise inkex.AbortExtension( + _("Can't find symbol {}").format(self.options.symbolid) + ) + bbox = symbol.path.bounding_box() + transform = inkex.Transform( + scale=( + float(self.boxsize) / bbox.width, + float(self.boxsize) / bbox.height, + ) + ) + result = Group() + for row in range(self.draw.row_count()): + for col in range(self.draw.col_count()): + if self.draw.isDark(col, row): + x, y = self.get_svg_pos(col, row) + # Inkscape doesn't support width/height on use tags + result.append( + Use.new( + symbol, + x / transform.a, + y / transform.d, + transform=transform, + ) + ) + return result + + def render_pathpreset(self): + if self.options.pathtype == "simple": + return self.render_path("h 1 v 1 h -1") + else: + s = ( + "m 0.5,0.5 " + "c 0.2761423745,0 0.5,0.2238576255 0.5,0.5 " + "c 0,0.2761423745 -0.2238576255,0.5 -0.5,0.5 " + "c -0.2761423745,0 -0.5,-0.2238576255 -0.5,-0.5 " + "c 0,-0.2761423745 0.2238576255,-0.5 0.5,-0.5" + ) + return self.render_path(s) + + render_smooth = lambda self: self.render_adv(self.options.smoothness[0]) + + def render_svg(self, grp, drawtype): + """Render to svg""" + drawer = getattr(self, f"render_{drawtype}", self.render_obsolete) + if drawer is None: + raise Exception("Unknown draw type: " + drawtype) + + canvas_width = (self.draw.col_count() + 2 * self.margin) * self.boxsize + canvas_height = (self.draw.row_count() + 2 * self.margin) * self.boxsize + + # white background providing margin: + rect = grp.add(Rectangle.new(0, 0, canvas_width, canvas_height)) + rect.style["stroke"] = "none" + rect.style["fill"] = "black" if self.invert_code else "white" + + qrg = grp.add(Group()) + qrg.style["stroke"] = "none" + qrg.style["fill"] = "white" if self.invert_code else "black" + qrg.add(drawer()) + + def get_svg_pos(self, col, row): + return (col + self.margin) * self.boxsize, (row + self.margin) * self.boxsize + + def get_icon_path_str(self, pointStr): + result = "" + digBuffer = "" + for c in pointStr: + if c.isdigit() or c == "-" or c == ".": + digBuffer += c + else: + if len(digBuffer) > 0: + result += str(float(digBuffer) * self.boxsize) + digBuffer = "" + result += c + + if len(digBuffer) > 0: + result += str(float(digBuffer) * self.boxsize) + + return result + + +if __name__ == "__main__": + QrCode().run() |