6 changed files with 41 additions and 1012 deletions
@ -1,962 +0,0 @@ |
|||||||
import math |
|
||||||
|
|
||||||
#from PIL import Image, ImageDraw |
|
||||||
|
|
||||||
#QRCode for Python |
|
||||||
# |
|
||||||
#Ported from the Javascript library by Sam Curren |
|
||||||
# |
|
||||||
#QRCode for Javascript |
|
||||||
#http://d-project.googlecode.com/svn/trunk/misc/qrcode/js/qrcode.js |
|
||||||
# |
|
||||||
#Copyright (c) 2009 Kazuhiko Arase |
|
||||||
# |
|
||||||
#URL: http://www.d-project.com/ |
|
||||||
# |
|
||||||
#Licensed under the MIT license: |
|
||||||
# http://www.opensource.org/licenses/mit-license.php |
|
||||||
# |
|
||||||
# The word "QR Code" is registered trademark of |
|
||||||
# DENSO WAVE INCORPORATED |
|
||||||
# http://www.denso-wave.com/qrcode/faqpatent-e.html |
|
||||||
|
|
||||||
|
|
||||||
class QR8bitByte: |
|
||||||
|
|
||||||
def __init__(self, data): |
|
||||||
self.mode = QRMode.MODE_8BIT_BYTE |
|
||||||
self.data = data |
|
||||||
|
|
||||||
def getLength(self): |
|
||||||
return len(self.data) |
|
||||||
|
|
||||||
def write(self, buffer): |
|
||||||
for i in range(len(self.data)): |
|
||||||
#// not JIS ... |
|
||||||
buffer.put(ord(self.data[i]), 8) |
|
||||||
def __repr__(self): |
|
||||||
return self.data |
|
||||||
|
|
||||||
class QRCode: |
|
||||||
def __init__(self, typeNumber, errorCorrectLevel): |
|
||||||
self.typeNumber = typeNumber |
|
||||||
self.errorCorrectLevel = errorCorrectLevel |
|
||||||
self.modules = None |
|
||||||
self.moduleCount = 0 |
|
||||||
self.dataCache = None |
|
||||||
self.dataList = [] |
|
||||||
def addData(self, data): |
|
||||||
newData = QR8bitByte(data) |
|
||||||
self.dataList.append(newData) |
|
||||||
self.dataCache = None |
|
||||||
def isDark(self, row, col): |
|
||||||
if (row < 0 or self.moduleCount <= row or col < 0 or self.moduleCount <= col): |
|
||||||
raise Exception("%s,%s - %s" % (row, col, self.moduleCount)) |
|
||||||
return self.modules[row][col] |
|
||||||
def getModuleCount(self): |
|
||||||
return self.moduleCount |
|
||||||
def make(self): |
|
||||||
self.makeImpl(False, self.getBestMaskPattern() ) |
|
||||||
def makeImpl(self, test, maskPattern): |
|
||||||
|
|
||||||
self.moduleCount = self.typeNumber * 4 + 17 |
|
||||||
self.modules = [None for x in range(self.moduleCount)] |
|
||||||
|
|
||||||
for row in range(self.moduleCount): |
|
||||||
|
|
||||||
self.modules[row] = [None for x in range(self.moduleCount)] |
|
||||||
|
|
||||||
for col in range(self.moduleCount): |
|
||||||
self.modules[row][col] = None #//(col + row) % 3; |
|
||||||
|
|
||||||
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) |
|
||||||
|
|
||||||
if (self.dataCache == None): |
|
||||||
self.dataCache = QRCode.createData(self.typeNumber, self.errorCorrectLevel, self.dataList) |
|
||||||
self.mapData(self.dataCache, maskPattern) |
|
||||||
|
|
||||||
def setupPositionProbePattern(self, row, col): |
|
||||||
|
|
||||||
for r in range(-1, 8): |
|
||||||
|
|
||||||
if (row + r <= -1 or self.moduleCount <= row + r): continue |
|
||||||
|
|
||||||
for c in range(-1, 8): |
|
||||||
|
|
||||||
if (col + c <= -1 or self.moduleCount <= col + c): continue |
|
||||||
|
|
||||||
if ( (0 <= r and r <= 6 and (c == 0 or c == 6) ) |
|
||||||
or (0 <= c and c <= 6 and (r == 0 or r == 6) ) |
|
||||||
or (2 <= r and r <= 4 and 2 <= c and c <= 4) ): |
|
||||||
self.modules[row + r][col + c] = True; |
|
||||||
else: |
|
||||||
self.modules[row + r][col + c] = False; |
|
||||||
|
|
||||||
def getBestMaskPattern(self): |
|
||||||
|
|
||||||
minLostPoint = 0 |
|
||||||
pattern = 0 |
|
||||||
|
|
||||||
for i in range(8): |
|
||||||
|
|
||||||
self.makeImpl(True, i); |
|
||||||
|
|
||||||
lostPoint = QRUtil.getLostPoint(self); |
|
||||||
|
|
||||||
if (i == 0 or minLostPoint > lostPoint): |
|
||||||
minLostPoint = lostPoint |
|
||||||
pattern = i |
|
||||||
|
|
||||||
return pattern |
|
||||||
|
|
||||||
|
|
||||||
def setupTimingPattern(self): |
|
||||||
|
|
||||||
for r in range(8, self.moduleCount - 8): |
|
||||||
if (self.modules[r][6] != None): |
|
||||||
continue |
|
||||||
self.modules[r][6] = (r % 2 == 0) |
|
||||||
|
|
||||||
for c in range(8, self.moduleCount - 8): |
|
||||||
if (self.modules[6][c] != None): |
|
||||||
continue |
|
||||||
self.modules[6][c] = (c % 2 == 0) |
|
||||||
|
|
||||||
def setupPositionAdjustPattern(self): |
|
||||||
|
|
||||||
pos = QRUtil.getPatternPosition(self.typeNumber) |
|
||||||
|
|
||||||
for i in range(len(pos)): |
|
||||||
|
|
||||||
for j in range(len(pos)): |
|
||||||
|
|
||||||
row = pos[i] |
|
||||||
col = pos[j] |
|
||||||
|
|
||||||
if (self.modules[row][col] != None): |
|
||||||
continue |
|
||||||
|
|
||||||
for r in range(-2, 3): |
|
||||||
|
|
||||||
for c in range(-2, 3): |
|
||||||
|
|
||||||
if (r == -2 or r == 2 or c == -2 or c == 2 or (r == 0 and c == 0) ): |
|
||||||
self.modules[row + r][col + c] = True |
|
||||||
else: |
|
||||||
self.modules[row + r][col + c] = False |
|
||||||
|
|
||||||
def setupTypeNumber(self, test): |
|
||||||
|
|
||||||
bits = QRUtil.getBCHTypeNumber(self.typeNumber) |
|
||||||
|
|
||||||
for i in range(18): |
|
||||||
mod = (not test and ( (bits >> i) & 1) == 1) |
|
||||||
self.modules[i // 3][i % 3 + self.moduleCount - 8 - 3] = mod; |
|
||||||
|
|
||||||
for i in range(18): |
|
||||||
mod = (not test and ( (bits >> i) & 1) == 1) |
|
||||||
self.modules[i % 3 + self.moduleCount - 8 - 3][i // 3] = mod; |
|
||||||
|
|
||||||
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 module |
|
||||||
self.modules[self.moduleCount - 8][8] = (not test) |
|
||||||
|
|
||||||
def mapData(self, data, maskPattern): |
|
||||||
|
|
||||||
inc = -1 |
|
||||||
row = self.moduleCount - 1 |
|
||||||
bitIndex = 7 |
|
||||||
byteIndex = 0 |
|
||||||
|
|
||||||
for col in range(self.moduleCount - 1, 0, -2): |
|
||||||
|
|
||||||
if (col == 6): col-=1 |
|
||||||
|
|
||||||
while (True): |
|
||||||
|
|
||||||
for c in range(2): |
|
||||||
|
|
||||||
if (self.modules[row][col - c] == None): |
|
||||||
|
|
||||||
dark = False |
|
||||||
|
|
||||||
if (byteIndex < len(data)): |
|
||||||
dark = ( ( (data[byteIndex] >> bitIndex) & 1) == 1) |
|
||||||
|
|
||||||
mask = QRUtil.getMask(maskPattern, row, col - c) |
|
||||||
|
|
||||||
if (mask): |
|
||||||
dark = not dark |
|
||||||
|
|
||||||
self.modules[row][col - c] = dark |
|
||||||
bitIndex-=1 |
|
||||||
|
|
||||||
if (bitIndex == -1): |
|
||||||
byteIndex+=1 |
|
||||||
bitIndex = 7 |
|
||||||
|
|
||||||
row += inc |
|
||||||
|
|
||||||
if (row < 0 or self.moduleCount <= row): |
|
||||||
row -= inc |
|
||||||
inc = -inc |
|
||||||
break |
|
||||||
PAD0 = 0xEC |
|
||||||
PAD1 = 0x11 |
|
||||||
|
|
||||||
@staticmethod |
|
||||||
def createData(typeNumber, errorCorrectLevel, dataList): |
|
||||||
|
|
||||||
rsBlocks = QRRSBlock.getRSBlocks(typeNumber, errorCorrectLevel) |
|
||||||
|
|
||||||
buffer = QRBitBuffer(); |
|
||||||
|
|
||||||
for i in range(len(dataList)): |
|
||||||
data = dataList[i] |
|
||||||
buffer.put(data.mode, 4) |
|
||||||
buffer.put(data.getLength(), QRUtil.getLengthInBits(data.mode, typeNumber) ) |
|
||||||
data.write(buffer) |
|
||||||
|
|
||||||
#// calc num max data. |
|
||||||
totalDataCount = 0; |
|
||||||
for i in range(len(rsBlocks)): |
|
||||||
totalDataCount += rsBlocks[i].dataCount |
|
||||||
|
|
||||||
if (buffer.getLengthInBits() > totalDataCount * 8): |
|
||||||
raise Exception("code length overflow. (" |
|
||||||
+ buffer.getLengthInBits() |
|
||||||
+ ">" |
|
||||||
+ totalDataCount * 8 |
|
||||||
+ ")") |
|
||||||
|
|
||||||
#// end code |
|
||||||
if (buffer.getLengthInBits() + 4 <= totalDataCount * 8): |
|
||||||
buffer.put(0, 4) |
|
||||||
|
|
||||||
#// padding |
|
||||||
while (buffer.getLengthInBits() % 8 != 0): |
|
||||||
buffer.putBit(False) |
|
||||||
|
|
||||||
#// 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 = [0 for x in range(len(rsBlocks))] |
|
||||||
ecdata = [0 for x in range(len(rsBlocks))] |
|
||||||
|
|
||||||
for r in range(len(rsBlocks)): |
|
||||||
|
|
||||||
dcCount = rsBlocks[r].dataCount |
|
||||||
ecCount = rsBlocks[r].totalCount - dcCount |
|
||||||
|
|
||||||
maxDcCount = max(maxDcCount, dcCount) |
|
||||||
maxEcCount = max(maxEcCount, ecCount) |
|
||||||
|
|
||||||
dcdata[r] = [0 for x in range(dcCount)] |
|
||||||
|
|
||||||
for i in range(len(dcdata[r])): |
|
||||||
dcdata[r][i] = 0xff & buffer.buffer[i + offset] |
|
||||||
offset += dcCount |
|
||||||
|
|
||||||
rsPoly = QRUtil.getErrorCorrectPolynomial(ecCount) |
|
||||||
rawPoly = QRPolynomial(dcdata[r], rsPoly.getLength() - 1) |
|
||||||
|
|
||||||
modPoly = rawPoly.mod(rsPoly) |
|
||||||
ecdata[r] = [0 for x in range(rsPoly.getLength()-1)] |
|
||||||
for i in range(len(ecdata[r])): |
|
||||||
modIndex = i + modPoly.getLength() - len(ecdata[r]) |
|
||||||
if (modIndex >= 0): |
|
||||||
ecdata[r][i] = modPoly.get(modIndex) |
|
||||||
else: |
|
||||||
ecdata[r][i] = 0 |
|
||||||
|
|
||||||
totalCodeCount = 0 |
|
||||||
for i in range(len(rsBlocks)): |
|
||||||
totalCodeCount += rsBlocks[i].totalCount |
|
||||||
|
|
||||||
data = [None for x in range(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 |
|
||||||
|
|
||||||
|
|
||||||
class QRMode: |
|
||||||
MODE_NUMBER = 1 << 0 |
|
||||||
MODE_ALPHA_NUM = 1 << 1 |
|
||||||
MODE_8BIT_BYTE = 1 << 2 |
|
||||||
MODE_KANJI = 1 << 3 |
|
||||||
|
|
||||||
class QRErrorCorrectLevel: |
|
||||||
L = 1 |
|
||||||
M = 0 |
|
||||||
Q = 3 |
|
||||||
H = 2 |
|
||||||
|
|
||||||
class QRMaskPattern: |
|
||||||
PATTERN000 = 0 |
|
||||||
PATTERN001 = 1 |
|
||||||
PATTERN010 = 2 |
|
||||||
PATTERN011 = 3 |
|
||||||
PATTERN100 = 4 |
|
||||||
PATTERN101 = 5 |
|
||||||
PATTERN110 = 6 |
|
||||||
PATTERN111 = 7 |
|
||||||
|
|
||||||
class QRUtil(object): |
|
||||||
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] |
|
||||||
] |
|
||||||
|
|
||||||
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 |
|
||||||
@staticmethod |
|
||||||
def getPatternPosition(typeNumber): |
|
||||||
return QRUtil.PATTERN_POSITION_TABLE[typeNumber - 1] |
|
||||||
@staticmethod |
|
||||||
def getMask(maskPattern, i, j): |
|
||||||
if maskPattern == QRMaskPattern.PATTERN000 : return (i + j) % 2 == 0 |
|
||||||
if maskPattern == QRMaskPattern.PATTERN001 : return i % 2 == 0 |
|
||||||
if maskPattern == QRMaskPattern.PATTERN010 : return j % 3 == 0 |
|
||||||
if maskPattern == QRMaskPattern.PATTERN011 : return (i + j) % 3 == 0 |
|
||||||
if maskPattern == QRMaskPattern.PATTERN100 : return (math.floor(i / 2) + math.floor(j / 3) ) % 2 == 0 |
|
||||||
if maskPattern == QRMaskPattern.PATTERN101 : return (i * j) % 2 + (i * j) % 3 == 0 |
|
||||||
if maskPattern == QRMaskPattern.PATTERN110 : return ( (i * j) % 2 + (i * j) % 3) % 2 == 0 |
|
||||||
if maskPattern == QRMaskPattern.PATTERN111 : return ( (i * j) % 3 + (i + j) % 2) % 2 == 0 |
|
||||||
raise Exception("bad maskPattern:" + maskPattern); |
|
||||||
@staticmethod |
|
||||||
def getErrorCorrectPolynomial(errorCorrectLength): |
|
||||||
a = QRPolynomial([1], 0); |
|
||||||
for i in range(errorCorrectLength): |
|
||||||
a = a.multiply(QRPolynomial([1, QRMath.gexp(i)], 0) ) |
|
||||||
return a |
|
||||||
@staticmethod |
|
||||||
def getLengthInBits(mode, type): |
|
||||||
|
|
||||||
if 1 <= type and type < 10: |
|
||||||
|
|
||||||
#// 1 - 9 |
|
||||||
|
|
||||||
if mode == QRMode.MODE_NUMBER : return 10 |
|
||||||
if mode == QRMode.MODE_ALPHA_NUM : return 9 |
|
||||||
if mode == QRMode.MODE_8BIT_BYTE : return 8 |
|
||||||
if mode == QRMode.MODE_KANJI : return 8 |
|
||||||
raise Exception("mode:" + mode) |
|
||||||
|
|
||||||
elif (type < 27): |
|
||||||
|
|
||||||
#// 10 - 26 |
|
||||||
|
|
||||||
if mode == QRMode.MODE_NUMBER : return 12 |
|
||||||
if mode == QRMode.MODE_ALPHA_NUM : return 11 |
|
||||||
if mode == QRMode.MODE_8BIT_BYTE : return 16 |
|
||||||
if mode == QRMode.MODE_KANJI : return 10 |
|
||||||
raise Exception("mode:" + mode) |
|
||||||
|
|
||||||
elif (type < 41): |
|
||||||
|
|
||||||
#// 27 - 40 |
|
||||||
|
|
||||||
if mode == QRMode.MODE_NUMBER : return 14 |
|
||||||
if mode == QRMode.MODE_ALPHA_NUM : return 13 |
|
||||||
if mode == QRMode.MODE_8BIT_BYTE : return 16 |
|
||||||
if mode == QRMode.MODE_KANJI : return 12 |
|
||||||
raise Exception("mode:" + mode) |
|
||||||
|
|
||||||
else: |
|
||||||
raise Exception("type:" + type) |
|
||||||
@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 |
|
||||||
|
|
||||||
class QRMath: |
|
||||||
|
|
||||||
@staticmethod |
|
||||||
def glog(n): |
|
||||||
if (n < 1): |
|
||||||
raise Exception("glog(" + n + ")") |
|
||||||
return LOG_TABLE[n]; |
|
||||||
@staticmethod |
|
||||||
def gexp(n): |
|
||||||
while n < 0: |
|
||||||
n += 255 |
|
||||||
while n >= 256: |
|
||||||
n -= 255 |
|
||||||
return EXP_TABLE[n]; |
|
||||||
|
|
||||||
EXP_TABLE = [x for x in range(256)] |
|
||||||
|
|
||||||
LOG_TABLE = [x for x in range(256)] |
|
||||||
|
|
||||||
for i in range(8): |
|
||||||
EXP_TABLE[i] = 1 << i; |
|
||||||
|
|
||||||
for i in range(8, 256): |
|
||||||
EXP_TABLE[i] = EXP_TABLE[i - 4] ^ EXP_TABLE[i - 5] ^ EXP_TABLE[i - 6] ^ EXP_TABLE[i - 8] |
|
||||||
|
|
||||||
for i in range(255): |
|
||||||
LOG_TABLE[EXP_TABLE[i] ] = i |
|
||||||
|
|
||||||
class QRPolynomial: |
|
||||||
|
|
||||||
def __init__(self, num, shift): |
|
||||||
|
|
||||||
if (len(num) == 0): |
|
||||||
raise Exception(num.length + "/" + shift) |
|
||||||
|
|
||||||
offset = 0 |
|
||||||
|
|
||||||
while offset < len(num) and num[offset] == 0: |
|
||||||
offset += 1 |
|
||||||
|
|
||||||
self.num = [0 for x in range(len(num)-offset+shift)] |
|
||||||
for i in range(len(num) - offset): |
|
||||||
self.num[i] = num[i + offset] |
|
||||||
|
|
||||||
|
|
||||||
def get(self, index): |
|
||||||
return self.num[index] |
|
||||||
def getLength(self): |
|
||||||
return len(self.num) |
|
||||||
def multiply(self, e): |
|
||||||
num = [0 for x in range(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 QRPolynomial(num, 0); |
|
||||||
def mod(self, e): |
|
||||||
|
|
||||||
if (self.getLength() - e.getLength() < 0): |
|
||||||
return self; |
|
||||||
|
|
||||||
ratio = QRMath.glog(self.get(0) ) - QRMath.glog(e.get(0) ) |
|
||||||
|
|
||||||
num = [0 for x in range(self.getLength())] |
|
||||||
|
|
||||||
for i in range(self.getLength()): |
|
||||||
num[i] = self.get(i); |
|
||||||
|
|
||||||
for i in range(e.getLength()): |
|
||||||
num[i] ^= QRMath.gexp(QRMath.glog(e.get(i) ) + ratio) |
|
||||||
|
|
||||||
# recursive call |
|
||||||
return QRPolynomial(num, 0).mod(e); |
|
||||||
|
|
||||||
class QRRSBlock: |
|
||||||
|
|
||||||
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], |
|
||||||
|
|
||||||
# 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 |
|
||||||
|
|
||||||
@staticmethod |
|
||||||
def getRSBlocks(typeNumber, errorCorrectLevel): |
|
||||||
rsBlock = QRRSBlock.getRsBlockTable(typeNumber, errorCorrectLevel); |
|
||||||
if rsBlock == None: |
|
||||||
raise Exception("bad rs block @ typeNumber:" + typeNumber + "/errorCorrectLevel:" + 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] |
|
||||||
|
|
||||||
for j in range(count): |
|
||||||
list.append(QRRSBlock(totalCount, dataCount)) |
|
||||||
|
|
||||||
return list; |
|
||||||
|
|
||||||
@staticmethod |
|
||||||
def getRsBlockTable(typeNumber, errorCorrectLevel): |
|
||||||
if errorCorrectLevel == QRErrorCorrectLevel.L: |
|
||||||
return QRRSBlock.RS_BLOCK_TABLE[(typeNumber - 1) * 4 + 0]; |
|
||||||
elif errorCorrectLevel == QRErrorCorrectLevel.M: |
|
||||||
return QRRSBlock.RS_BLOCK_TABLE[(typeNumber - 1) * 4 + 1]; |
|
||||||
elif errorCorrectLevel == QRErrorCorrectLevel.Q: |
|
||||||
return QRRSBlock.RS_BLOCK_TABLE[(typeNumber - 1) * 4 + 2]; |
|
||||||
elif errorCorrectLevel == QRErrorCorrectLevel.H: |
|
||||||
return QRRSBlock.RS_BLOCK_TABLE[(typeNumber - 1) * 4 + 3]; |
|
||||||
else: |
|
||||||
return None; |
|
||||||
|
|
||||||
class QRBitBuffer: |
|
||||||
def __init__(self): |
|
||||||
self.buffer = [] |
|
||||||
self.length = 0 |
|
||||||
def __repr__(self): |
|
||||||
return ".".join([str(n) for n in self.buffer]) |
|
||||||
def get(self, index): |
|
||||||
bufIndex = math.floor(index / 8) |
|
||||||
val = ( (self.buffer[bufIndex] >> (7 - index % 8) ) & 1) == 1 |
|
||||||
print "get ", val |
|
||||||
return ( (self.buffer[bufIndex] >> (7 - index % 8) ) & 1) == 1 |
|
||||||
def put(self, num, length): |
|
||||||
for i in range(length): |
|
||||||
self.putBit( ( (num >> (length - i - 1) ) & 1) == 1) |
|
||||||
def getLengthInBits(self): |
|
||||||
return self.length |
|
||||||
def putBit(self, bit): |
|
||||||
bufIndex = self.length // 8 |
|
||||||
if len(self.buffer) <= bufIndex: |
|
||||||
self.buffer.append(0) |
|
||||||
if bit: |
|
||||||
self.buffer[bufIndex] |= (0x80 >> (self.length % 8) ) |
|
||||||
self.length+=1 |
|
||||||
Loading…
Reference in new issue