--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/srcanamdw/codescanner/pyinstaller/icon.py Thu Feb 18 12:29:02 2010 +0530
@@ -0,0 +1,156 @@
+#! /usr/bin/env python
+# Copyright (C) 2005, Giovanni Bajo
+# Based on previous work under copyright (c) 2002 McMillan Enterprises, Inc.
+#
+# 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-1301, USA
+
+# This code is courtesy of Thomas Heller, who
+# has kindly donated it to this project.
+RT_ICON = 3
+RT_GROUP_ICON = 14
+LOAD_LIBRARY_AS_DATAFILE = 2
+
+import struct
+
+class Structure:
+ def __init__ (self):
+ size = self._sizeInBytes = struct.calcsize (self._format_)
+ self._fields_ = list (struct.unpack (self._format_, '\000' * size))
+ indexes = self._indexes_ = {}
+ for i in range (len (self._names_)):
+ indexes[self._names_[i]] = i
+ def dump (self):
+ print "I: DUMP of", self
+ for name in self._names_:
+ if name[0] != '_':
+ print "I: %20s = %s" % (name, getattr (self, name))
+ print
+ def __getattr__ (self, name):
+ if name in self._names_:
+ index = self._indexes_[name]
+ return self._fields_[index]
+ try:
+ return self.__dict__[name]
+ except KeyError:
+ raise AttributeError, name
+ def __setattr__ (self, name, value):
+ if name in self._names_:
+ index = self._indexes_[name]
+ self._fields_[index] = value
+ else:
+ self.__dict__[name] = value
+ def tostring (self):
+ return apply (struct.pack, [self._format_,] + self._fields_)
+ def fromfile (self, file):
+ data = file.read (self._sizeInBytes)
+ self._fields_ = list (struct.unpack (self._format_, data))
+
+class ICONDIRHEADER (Structure):
+ _names_ = "idReserved", "idType", "idCount"
+ _format_ = "hhh"
+
+class ICONDIRENTRY (Structure):
+ _names_ = "bWidth", "bHeight", "bColorCount", "bReserved", "wPlanes", "wBitCount", "dwBytesInRes", "dwImageOffset"
+ _format_ = "bbbbhhii"
+
+class GRPICONDIR (Structure):
+ _names_ = "idReserved", "idType", "idCount"
+ _format_ = "hhh"
+
+class GRPICONDIRENTRY (Structure):
+ _names_ = "bWidth", "bHeight", "bColorCount", "bReserved", "wPlanes", "wBitCount", "dwBytesInRes", "nID"
+ _format_ = "bbbbhhih"
+
+class IconFile:
+ def __init__ (self, path):
+ self.path = path
+ file = open (path, "rb")
+ self.entries = []
+ self.images = []
+ header = self.header = ICONDIRHEADER()
+ header.fromfile (file)
+ for i in range (header.idCount):
+ entry = ICONDIRENTRY()
+ entry.fromfile (file)
+ self.entries.append (entry)
+ for e in self.entries:
+ file.seek (e.dwImageOffset, 0)
+ self.images.append (file.read (e.dwBytesInRes))
+
+ def grp_icon_dir (self):
+ return self.header.tostring()
+
+ def grp_icondir_entries (self):
+ data = ""
+ i = 1
+ for entry in self.entries:
+ e = GRPICONDIRENTRY()
+ for n in e._names_[:-1]:
+ setattr(e, n, getattr (entry, n))
+ e.nID = i
+ i = i + 1
+ data = data + e.tostring()
+ return data
+
+
+def CopyIcons_FromIco (dstpath, srcpath):
+ f = IconFile (srcpath)
+ print "I: Updating icons from", srcpath, "to", dstpath
+ import win32api #, win32con
+ hdst = win32api.BeginUpdateResource (dstpath, 0)
+ data = f.grp_icon_dir()
+ data = data + f.grp_icondir_entries()
+ win32api.UpdateResource (hdst, RT_GROUP_ICON, 1, data)
+ print "I: Writing RT_GROUP_ICON resource with %d bytes" % len (data)
+ i = 1
+ for data in f.images:
+ win32api.UpdateResource (hdst, RT_ICON, i, data)
+ print "I: Writing RT_ICON resource with %d bytes" % len (data)
+ i = i + 1
+ win32api.EndUpdateResource (hdst, 0)
+
+def CopyIcons (dstpath, srcpath):
+ import os.path, string
+ index = None
+ try:
+ srcpath, index = map (string.strip, string.split (srcpath, ','))
+ index = int (index)
+ except:
+ pass
+ print "I: PATH, INDEX", srcpath, index
+ srcext = os.path.splitext (srcpath)[1]
+ if string.lower (srcext) == '.ico':
+ return CopyIcons_FromIco (dstpath, srcpath)
+ if index is not None:
+ print "I: Updating icons from", srcpath, ", %d to" % index, dstpath
+ else:
+ print "I: Updating icons from", srcpath, "to", dstpath
+ import win32api #, win32con
+ hdst = win32api.BeginUpdateResource (dstpath, 0)
+ hsrc = win32api.LoadLibraryEx (srcpath, 0, LOAD_LIBRARY_AS_DATAFILE)
+ if index is None:
+ grpname = win32api.EnumResourceNames (hsrc, RT_GROUP_ICON)[0]
+ elif index >= 0:
+ grpname = win32api.EnumResourceNames (hsrc, RT_GROUP_ICON)[index]
+ else:
+ grpname = -index
+ data = win32api.LoadResource (hsrc, RT_GROUP_ICON, grpname)
+ win32api.UpdateResource (hdst, RT_GROUP_ICON, grpname, data)
+ for iconname in win32api.EnumResourceNames (hsrc, RT_ICON):
+ data = win32api.LoadResource (hsrc, RT_ICON, iconname)
+ win32api.UpdateResource (hdst, RT_ICON, iconname, data)
+ win32api.FreeLibrary (hsrc)
+ win32api.EndUpdateResource (hdst, 0)
+