Do some renaming of some files

gui/bitmap_loader.py

@@ -0,0 +1,101 @@
+# =============================================================================
+# Copyright (C) 2010 Diego Duclos
+#
+# This file is part of pyfa.
+#
+# pyfa 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 3 of the License, or
+# (at your option) any later version.
+#
+# pyfa 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 pyfa.  If not, see <http://www.gnu.org/licenses/>.
+# =============================================================================
+
+import io
+import os.path
+import zipfile
+
+# noinspection PyPackageRequirements
+import wx
+
+import config
+
+from logbook import Logger
+logging = Logger(__name__)
+
+try:
+    from collections import OrderedDict
+except ImportError:
+    from .utils.compat import OrderedDict
+
+
+class BitmapLoader(object):
+    try:
+        archive = zipfile.ZipFile(os.path.join(config.pyfaPath, 'imgs.zip'), 'r')
+        logging.info("Using zipped image files.")
+    except IOError:
+        logging.info("Using local image files.")
+        archive = None
+
+    cachedBitmaps = OrderedDict()
+    dontUseCachedBitmaps = False
+    max_bmps = 500
+
+    @classmethod
+    def getStaticBitmap(cls, name, parent, location):
+        static = wx.StaticBitmap(parent)
+        static.SetBitmap(cls.getBitmap(name, location))
+        return static
+
+    @classmethod
+    def getBitmap(cls, name, location):
+        if cls.dontUseCachedBitmaps:
+            img = cls.getImage(name, location)
+            if img is not None:
+                return img.ConvertToBitmap()
+
+        path = "%s%s" % (name, location)
+
+        if len(cls.cachedBitmaps) == cls.max_bmps:
+            cls.cachedBitmaps.popitem(False)
+
+        if path not in cls.cachedBitmaps:
+            img = cls.getImage(name, location)
+            if img is not None:
+                bmp = img.ConvertToBitmap()
+            else:
+                bmp = None
+            cls.cachedBitmaps[path] = bmp
+        else:
+            bmp = cls.cachedBitmaps[path]
+
+        return bmp
+
+    @classmethod
+    def getImage(cls, name, location):
+        filename = "{0}.png".format(name)
+
+        if cls.archive:
+            path = os.path.join(location, filename)
+            if os.sep != "/" and os.sep in path:
+                path = path.replace(os.sep, "/")
+
+            try:
+                img_data = cls.archive.read(path)
+                sbuf = io.StringIO(img_data)
+                return wx.ImageFromStream(sbuf)
+            except KeyError:
+                print(("Missing icon file from zip: {0}".format(path)))
+        else:
+            path = os.path.join(config.pyfaPath, 'imgs' + os.sep + location + os.sep + filename)
+
+            if os.path.exists(path):
+                return wx.Image(path)
+            else:
+                print(("Missing icon file: {0}".format(path)))