240e9b5b58
The original source is licensed under a BSD-License, so it should be fine to use.
319 lines
11 KiB
Python
319 lines
11 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding:utf-8 -*-
|
|
|
|
###
|
|
# qidenticon.py is Licesensed under FreeBSD License.
|
|
# (http://www.freebsd.org/copyright/freebsd-license.html)
|
|
#
|
|
# Copyright 2013 "Sendiulo". All rights reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
|
|
#
|
|
# 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
|
|
# 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDER ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
###
|
|
|
|
###
|
|
# identicon.py is Licesensed under FreeBSD License.
|
|
# (http://www.freebsd.org/copyright/freebsd-license.html)
|
|
#
|
|
# Copyright 1994-2009 Shin Adachi. All rights reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
|
|
#
|
|
# 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
|
|
# 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDER ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
###
|
|
|
|
"""
|
|
qidenticon.py
|
|
identicon python implementation with QPixmap output
|
|
by sendiulo <sendiulo@gmx.net>
|
|
|
|
based on
|
|
identicon.py
|
|
identicon python implementation.
|
|
by Shin Adachi <shn@glucose.jp>
|
|
|
|
= usage =
|
|
|
|
== commandline ==
|
|
>>> python identicon.py [code]
|
|
|
|
== python ==
|
|
>>> import qtidenticon
|
|
>>> qtidenticon.render_identicon(code, size)
|
|
|
|
Return a PIL Image class instance which have generated identicon image.
|
|
```size``` specifies `patch size`. Generated image size is 3 * ```size```.
|
|
"""
|
|
|
|
# we probably don't need all of them, but i don't want to check now
|
|
from PyQt4 import QtCore, QtGui
|
|
from PyQt4.QtCore import *
|
|
from PyQt4.QtGui import *
|
|
|
|
__all__ = ['render_identicon', 'IdenticonRendererBase']
|
|
|
|
|
|
class Matrix2D(list):
|
|
"""Matrix for Patch rotation"""
|
|
def __init__(self, initial=[0.] * 9):
|
|
assert isinstance(initial, list) and len(initial) == 9
|
|
list.__init__(self, initial)
|
|
|
|
def clear(self):
|
|
for i in xrange(9):
|
|
self[i] = 0.
|
|
|
|
def set_identity(self):
|
|
self.clear()
|
|
for i in xrange(3):
|
|
self[i] = 1.
|
|
|
|
def __str__(self):
|
|
return '[%s]' % ', '.join('%3.2f' % v for v in self)
|
|
|
|
def __mul__(self, other):
|
|
r = []
|
|
if isinstance(other, Matrix2D):
|
|
for y in xrange(3):
|
|
for x in xrange(3):
|
|
v = 0.0
|
|
for i in xrange(3):
|
|
v += (self[i * 3 + x] * other[y * 3 + i])
|
|
r.append(v)
|
|
else:
|
|
raise NotImplementedError
|
|
return Matrix2D(r)
|
|
|
|
def for_PIL(self):
|
|
return self[0:6]
|
|
|
|
@classmethod
|
|
def translate(kls, x, y):
|
|
return kls([1.0, 0.0, float(x),
|
|
0.0, 1.0, float(y),
|
|
0.0, 0.0, 1.0])
|
|
|
|
@classmethod
|
|
def scale(kls, x, y):
|
|
return kls([float(x), 0.0, 0.0,
|
|
0.0, float(y), 0.0,
|
|
0.0, 0.0, 1.0])
|
|
|
|
"""
|
|
# need `import math`
|
|
@classmethod
|
|
def rotate(kls, theta, pivot=None):
|
|
c = math.cos(theta)
|
|
s = math.sin(theta)
|
|
|
|
matR = kls([c, -s, 0., s, c, 0., 0., 0., 1.])
|
|
if not pivot:
|
|
return matR
|
|
return kls.translate(-pivot[0], -pivot[1]) * matR *
|
|
kls.translate(*pivot)
|
|
"""
|
|
|
|
@classmethod
|
|
def rotateSquare(kls, theta, pivot=None):
|
|
theta = theta % 4
|
|
c = [1., 0., -1., 0.][theta]
|
|
s = [0., 1., 0., -1.][theta]
|
|
|
|
matR = kls([c, -s, 0., s, c, 0., 0., 0., 1.])
|
|
if not pivot:
|
|
return matR
|
|
return kls.translate(-pivot[0], -pivot[1]) * matR * \
|
|
kls.translate(*pivot)
|
|
|
|
|
|
class IdenticonRendererBase(object):
|
|
PATH_SET = []
|
|
|
|
def __init__(self, code):
|
|
"""
|
|
@param code code for icon
|
|
"""
|
|
if not isinstance(code, int):
|
|
code = int(code)
|
|
self.code = code
|
|
|
|
def render(self, size):
|
|
"""
|
|
render identicon to QPixmap
|
|
|
|
@param size identicon patchsize. (image size is 3 * [size])
|
|
@return QPixmap
|
|
"""
|
|
|
|
# decode the code
|
|
middle, corner, side, foreColor, backColor = self.decode(self.code)
|
|
|
|
# make image
|
|
image = QPixmap(QSize(size * 3, size * 3))
|
|
|
|
# fill background
|
|
image.fill(QtGui.QColor(0,0,0))
|
|
|
|
kwds = {
|
|
'image': image,
|
|
'size': size,
|
|
'foreColor': foreColor,
|
|
'backColor': backColor}
|
|
|
|
# middle patch
|
|
image = self.drawPatchQt((1, 1), middle[2], middle[1], middle[0], **kwds)
|
|
|
|
# side patch
|
|
kwds['type'] = side[0]
|
|
for i in xrange(4):
|
|
pos = [(1, 0), (2, 1), (1, 2), (0, 1)][i]
|
|
image = self.drawPatchQt(pos, side[2] + 1 + i, side[1], **kwds)
|
|
|
|
# corner patch
|
|
kwds['type'] = corner[0]
|
|
for i in xrange(4):
|
|
pos = [(0, 0), (2, 0), (2, 2), (0, 2)][i]
|
|
image = self.drawPatchQt(pos, corner[2] + 1 + i, corner[1], **kwds)
|
|
|
|
return image
|
|
|
|
|
|
def drawPatchQt(self, pos, turn, invert, type, image, size, foreColor,
|
|
backColor):
|
|
"""
|
|
@param size patch size
|
|
"""
|
|
path = self.PATH_SET[type]
|
|
if not path:
|
|
# blank patch
|
|
invert = not invert
|
|
path = [(0., 0.), (1., 0.), (1., 1.), (0., 1.), (0., 0.)]
|
|
|
|
|
|
polygon = QPolygonF([QPointF(x*size,y*size) for x,y in path])
|
|
|
|
rot = turn % 4
|
|
rot_trans = [QPointF(0.,0.), QPointF(size, 0.), QPointF(size, size), QPointF(0., size)]
|
|
rotation = [0,90,180,270]
|
|
# print rotation[rot], rot_trans[rot], turn
|
|
pen_color = QtGui.QColor(255, 255, 255, 0)
|
|
pen = QtGui.QPen(pen_color, Qt.SolidPattern)
|
|
foreBrush = QtGui.QBrush(foreColor, Qt.SolidPattern)
|
|
backBrush = QtGui.QBrush(backColor, Qt.SolidPattern)
|
|
if invert:
|
|
foreBrush, backBrush = backBrush, foreBrush
|
|
|
|
painter = QPainter()
|
|
painter.begin(image)
|
|
painter.setPen(pen)
|
|
|
|
painter.translate(pos[0]*size, pos[1]*size)
|
|
painter.translate(rot_trans[rot])
|
|
painter.rotate(rotation[rot])
|
|
|
|
painter.setBrush(backBrush)
|
|
painter.drawRect(0,0, size, size)
|
|
|
|
painter.setBrush(foreBrush)
|
|
painter.drawPolygon(polygon, Qt.WindingFill)
|
|
|
|
painter.end()
|
|
|
|
return image
|
|
|
|
### virtual functions
|
|
def decode(self, code):
|
|
raise NotImplementedError
|
|
|
|
class DonRenderer(IdenticonRendererBase):
|
|
"""
|
|
Don Park's implementation of identicon
|
|
see : http://www.docuverse.com/blog/donpark/2007/01/19/identicon-updated-and-source-released
|
|
"""
|
|
|
|
PATH_SET = [
|
|
[(0, 0), (4, 0), (4, 4), (0, 4)], # 0
|
|
[(0, 0), (4, 0), (0, 4)],
|
|
[(2, 0), (4, 4), (0, 4)],
|
|
[(0, 0), (2, 0), (2, 4), (0, 4)],
|
|
[(2, 0), (4, 2), (2, 4), (0, 2)], # 4
|
|
[(0, 0), (4, 2), (4, 4), (2, 4)],
|
|
[(2, 0), (4, 4), (2, 4), (3, 2), (1, 2), (2, 4), (0, 4)],
|
|
[(0, 0), (4, 2), (2, 4)],
|
|
[(1, 1), (3, 1), (3, 3), (1, 3)], # 8
|
|
[(2, 0), (4, 0), (0, 4), (0, 2), (2, 2)],
|
|
[(0, 0), (2, 0), (2, 2), (0, 2)],
|
|
[(0, 2), (4, 2), (2, 4)],
|
|
[(2, 2), (4, 4), (0, 4)],
|
|
[(2, 0), (2, 2), (0, 2)],
|
|
[(0, 0), (2, 0), (0, 2)],
|
|
[]] # 15
|
|
MIDDLE_PATCH_SET = [0, 4, 8, 15]
|
|
|
|
# modify path set
|
|
for idx in xrange(len(PATH_SET)):
|
|
if PATH_SET[idx]:
|
|
p = map(lambda vec: (vec[0] / 4.0, vec[1] / 4.0), PATH_SET[idx])
|
|
PATH_SET[idx] = p + p[:1]
|
|
|
|
def decode(self, code):
|
|
# decode the code
|
|
middleType = self.MIDDLE_PATCH_SET[code & 0x03]
|
|
middleInvert= (code >> 2) & 0x01
|
|
cornerType = (code >> 3) & 0x0F
|
|
cornerInvert= (code >> 7) & 0x01
|
|
cornerTurn = (code >> 8) & 0x03
|
|
sideType = (code >> 10) & 0x0F
|
|
sideInvert = (code >> 14) & 0x01
|
|
sideTurn = (code >> 15) & 0x03
|
|
# bug reported by Masato Hagiwara
|
|
# http://lilyx.net/iconlang-en/
|
|
# blue = (code >> 16) & 0x1F
|
|
# green = (code >> 21) & 0x1F
|
|
blue = (code >> 17) & 0x1F
|
|
green = (code >> 22) & 0x1F
|
|
red = (code >> 27) & 0x1F
|
|
|
|
foreColor = (red << 3, green << 3, blue << 3)
|
|
foreColor = QtGui.QColor(*foreColor)
|
|
|
|
bgcolor = QtGui.QColor(255,255,255)
|
|
|
|
return (middleType, middleInvert, 0),\
|
|
(cornerType, cornerInvert, cornerTurn),\
|
|
(sideType, sideInvert, sideTurn),\
|
|
foreColor, bgcolor
|
|
|
|
|
|
def render_identicon(code, size, renderer=None):
|
|
if not renderer:
|
|
renderer = DonRenderer
|
|
return renderer(code).render(size)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import sys
|
|
|
|
if len(sys.argv) < 2:
|
|
print 'usage: python identicon.py [CODE]....'
|
|
raise SystemExit
|
|
|
|
for code in sys.argv[1:]:
|
|
if code.startswith('0x') or code.startswith('0X'):
|
|
code = int(code[2:], 16)
|
|
elif code.startswith('0'):
|
|
code = int(code[1:], 8)
|
|
else:
|
|
code = int(code)
|
|
|
|
app = Qt.QApplication(sys.argv)
|
|
icon = render_identicon(code, 24)
|
|
icon.save('%08x.png' % code, 'PNG')
|