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

gui/builtinViewColumns/__init__.py

@@ -0,0 +1,8 @@
+__all__ = ["moduleState", "moduleNameOrSlot"]
+
+columns = {}
+def registerColumn(column):
+    columns[column.name] = column
+
+def getColumn(name):
+    return columns[name]

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)

gui/builtinViewColumns/moduleState.py

@@ -0,0 +1,46 @@
+#===============================================================================
+# 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 State
+
+class ModuleState(ViewColumn):
+    name = "Module state"
+    def __init__(self, fittingView, params):
+        ViewColumn.__init__(self, fittingView)
+        self.resizable = False
+        self.size = 16
+        self.stateNameMap = {}
+        for state in State.__dict__:
+            if state[0:2] == "__":
+                continue
+
+            id = State.__dict__[state]
+            self.stateNameMap[id] = state
+
+    def getText(self, mod):
+        return ""
+
+    def getImageId(self, mod):
+        bitmap = bitmapLoader.getBitmap("state_%s_small" % self.stateNameMap[mod.state], "icons")
+        return self.fittingView.imageList.Add(bitmap)
+
+builtinViewColumns.registerColumn(ModuleState)

gui/fittingView.py

@@ -19,6 +19,32 @@
 
 import wx
 
-class FittingView(wx.TreeCtrl):
+import gui.builtinViewColumns
+from gui.builtinViewColumns import *
+
+class FittingView(wx.ListCtrl):
+    DEFAULT_COLS = ["Module state", "Module name/slot"]
     def __init__(self, parent):
-        wx.TreeCtrl.__init__(self, parent, wx.ID_ANY)
+        listStyle = wx.LC_REPORT | wx.BORDER_NONE
+        wx.ListCtrl.__init__(self, parent, wx.ID_ANY, style=listStyle)
+
+        self.imageList = wx.ImageList(16, 16)
+        self.SetImageList(self.imageList, wx.IMAGE_LIST_SMALL)
+        self.activeColumns = []
+        self.Bind(wx.EVT_LIST_COL_BEGIN_DRAG, self.resizeChecker)
+        i = 0
+        for colName in FittingView.DEFAULT_COLS:
+            col = gui.builtinViewColumns.getColumn(colName)(self, None)
+            self.activeColumns.append(col)
+
+            info = wx.ListItem()
+            info.m_mask = col.mask
+            info.m_image = col.imageId
+            info.m_text = col.columnText
+            self.InsertColumnInfo(i, info)
+            self.SetColumnWidth(i, wx.LIST_AUTOSIZE_USEHEADER if col.size is wx.LIST_AUTOSIZE else col.size)
+            i += 1
+
+    def resizeChecker(self, event):
+        if self.activeColumns[event.Column].resizable is False:
+            event.Veto()

gui/viewColumn.py

@@ -0,0 +1,49 @@
+#===============================================================================
+# 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/>.
+#===============================================================================
+
+import wx
+
+class ViewColumn(object):
+    '''
+    Abstract class that columns can inherit from.
+    Once the missing methods are correctly implemented,
+    they can be used as columns in a view.
+    '''
+    columns = []
+    def __init__(self, fittingView):
+        ViewColumn.columns.append(self)
+
+        self.fittingView = fittingView
+        self.columnText = ""
+        self.imageId = -1
+        self.size = wx.LIST_AUTOSIZE
+        self.mask = wx.LIST_MASK_TEXT | wx.LIST_MASK_IMAGE
+        self.resizable = True
+
+    def getRestrictions(self):
+        raise NotImplementedError()
+
+    def getText(self, mod):
+        raise NotImplementedError()
+
+    def getImage(self, mod):
+        raise NotImplementedError()
+
+    def getParameters(self):
+        raise NotImplementedError()