commit b079252251deeb0ea45e92ff1eff479da22f9115
parent 9f797d4c5d53d38537d4f92ca5341399e334ffd9
Author: ThomasV <thomasv@gitorious>
Date: Tue, 22 Nov 2011 16:41:41 +0300
add qr support from slush
Diffstat:
2 files changed, 993 insertions(+), 0 deletions(-)
diff --git a/client/gui.py b/client/gui.py
@@ -22,6 +22,7 @@ import socket, traceback
import pygtk
pygtk.require('2.0')
import gtk, gobject
+import pyqrnative
gtk.gdk.threads_init()
APP_NAME = "Electrum"
@@ -722,6 +723,36 @@ class BitcoinGUI:
button.show()
hbox.pack_start(button,False)
+ def showqrcode(w, treeview, liststore):
+ path, col = treeview.get_cursor()
+ if not path: return
+ address = liststore.get_value(liststore.get_iter(path), 0)
+ qr = pyqrnative.QRCode(4, pyqrnative.QRErrorCorrectLevel.H)
+ qr.addData(address)
+ qr.make()
+ boxsize = 7
+ size = qr.getModuleCount()*boxsize
+ def area_expose_cb(area, event):
+ style = area.get_style()
+ k = qr.getModuleCount()
+ for r in range(k):
+ for c in range(k):
+ gc = style.black_gc if qr.isDark(r, c) else style.white_gc
+ area.window.draw_rectangle(gc, True, c*boxsize, r*boxsize, boxsize, boxsize)
+ area = gtk.DrawingArea()
+ area.set_size_request(size, size)
+ area.connect("expose-event", area_expose_cb)
+ area.show()
+ dialog = gtk.Dialog("QR Code", parent=self.window, flags=gtk.DIALOG_MODAL|gtk.DIALOG_NO_SEPARATOR, buttons = ("ok",1))
+ dialog.vbox.add(area)
+ dialog.run()
+ dialog.destroy()
+
+ button = gtk.Button("QR")
+ button.connect("clicked", showqrcode, treeview, liststore)
+ button.show()
+ hbox.pack_start(button,False)
+
button = gtk.Button("Copy to clipboard")
def copy2clipboard(w, treeview, liststore):
import platform
diff --git a/client/pyqrnative.py b/client/pyqrnative.py
@@ -0,0 +1,962 @@
+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
