19 lines
549 B
Python
19 lines
549 B
Python
|
import numpy as np
|
||
|
from PyQt4 import QtGui
|
||
|
import os, pickle
|
||
|
|
||
|
path = os.path.split(__file__)[0]
|
||
|
pixmaps = {}
|
||
|
for f in os.listdir(path):
|
||
|
if not f.endswith('.png'):
|
||
|
continue
|
||
|
print f
|
||
|
img = QtGui.QImage(os.path.join(path, f))
|
||
|
ptr = img.bits()
|
||
|
ptr.setsize(img.byteCount())
|
||
|
arr = np.asarray(ptr).reshape(img.height(), img.width(), 4).transpose(1,0,2)
|
||
|
pixmaps[f] = pickle.dumps(arr)
|
||
|
fh = open(os.path.join(path, 'pixmapData.py'), 'w')
|
||
|
fh.write("import numpy as np; pixmapData=%s" % repr(pixmaps))
|
||
|
|