Add color column

graphs/gui/lists.py

@@ -31,10 +31,6 @@ from service.fit import Fit
 
 class BaseWrapperList(gui.display.Display):
 
-    DEFAULT_COLS = (
-        'Base Icon',
-        'Base Name')
-
     def __init__(self, graphFrame, parent):
         super().__init__(parent)
         self.graphFrame = graphFrame
@@ -239,6 +235,10 @@ class BaseWrapperList(gui.display.Display):
 
 class SourceWrapperList(BaseWrapperList):
 
+    DEFAULT_COLS = (
+        'Color',
+        'Base Icon',
+        'Base Name')
     wrapperClass = SourceWrapper
 
     def __init__(self, graphFrame, parent):
@@ -268,6 +268,9 @@ class SourceWrapperList(BaseWrapperList):
 
 class TargetWrapperList(BaseWrapperList):
 
+    DEFAULT_COLS = (
+        'Base Icon',
+        'Base Name')
     wrapperClass = TargetWrapper
 
     def __init__(self, graphFrame, parent):

gui/builtinViewColumns/__init__.py

@@ -1,16 +0,0 @@
-__all__ = [
-    "ammo",
-    "ammoIcon",
-    "attributeDisplay",
-    "attributeDisplayGraph",
-    "baseIcon",
-    "baseName",
-    "capacitorUse",
-    "maxRange",
-    "price",
-    "propertyDisplay",
-    "state",
-    "misc",
-    "abilities",
-    "sideEffects",
-    "targetResists"]

gui/builtinViewColumns/color.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 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/>.
+# =============================================================================
+
+# noinspection PyPackageRequirements
+import wx
+
+from graphs.wrapper import SourceWrapper
+from gui.bitmap_loader import BitmapLoader
+from gui.viewColumn import ViewColumn
+
+
+class LineColor(ViewColumn):
+
+    name = 'Color'
+
+    def __init__(self, fittingView, params):
+        ViewColumn.__init__(self, fittingView)
+        self.imageId = fittingView.imageList.GetImageIndex(1432, 'icons')
+        self.bitmap = BitmapLoader.getBitmap(1432, 'icons')
+        self.mask = wx.LIST_MASK_TEXT
+
+    def getText(self, stuff):
+        if isinstance(stuff, SourceWrapper):
+            return ''
+        return ''
+
+    def getToolTip(self, stuff):
+        if isinstance(stuff, SourceWrapper):
+            return ''
+        return ''
+
+
+LineColor.register()

gui/viewColumn.py

@@ -76,6 +76,7 @@ from gui.builtinViewColumns import (  # noqa: E402, F401
     baseIcon,
     baseName,
     capacitorUse,
+    color,
     maxRange,
     misc,
     price,