Start work on an generic implementation for showing fitting/drone/implant/booster/etc. view columns.

gui/builtinViewColumns/moduleNameOrSlot.py

@@ -0,0 +1,61 @@
+#===============================================================================
+# 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 Affero 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 Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with pyfa.  If not, see <http://www.gnu.org/licenses/>.
+#===============================================================================
+
+from gui import builtinViewColumns
+from gui.viewColumn import ViewColumn
+from gui import bitmapLoader
+from eos.types import Slot
+
+class ModuleNameOrSlot(ViewColumn):
+    name = "Module name/slot"
+    def __init__(self, fittingView, params):
+        ViewColumn.__init__(self, fittingView)
+        self.columnText = "Name"
+        self.slotNameMap = {}
+        for slot in Slot.__dict__:
+            if slot[0:2] == "__":
+                continue
+
+            id = Slot.__dict__[slot]
+            self.slotNameMap[id] = slot
+
+    def getText(self, mod):
+        if mod.isEmpty():
+            return "%s Slot" % self.slotNameMap[mod.slot].capitalize()
+        else:
+            return mod.item.name
+
+    def getImageId(self, mod):
+        if mod.isEmpty():
+            bitmap = bitmapLoader.getBitmap("slot_%s_small" % self.slotNameMap[mod.state], "icons")
+            iconId = self.fittingView.imageList.Add(bitmap)
+        else:
+            iconFile = mod.item.icon.iconFile if mod.item.icon else ""
+            if iconFile:
+                bitmap = bitmapLoader.getBitmap(iconFile, "pack")
+                if bitmap is None:
+                    iconId = -1
+                else:
+                    iconId = self.fittingView.imageList.Add(bitmap)
+            else:
+                iconId = -1
+
+        return iconId
+
+builtinViewColumns.registerColumn(ModuleNameOrSlot)