Add line lightness column

gui/builtinViewColumns/graphColor.py

@@ -25,9 +25,9 @@ from graphs.wrapper import SourceWrapper
 from gui.viewColumn import ViewColumn
 
 
-class LineColor(ViewColumn):
+class GraphColor(ViewColumn):
 
-    name = 'Color'
+    name = 'Graph Color'
 
     def __init__(self, fittingView, params):
         ViewColumn.__init__(self, fittingView)
@@ -39,7 +39,7 @@ class LineColor(ViewColumn):
     def getImageId(self, stuff):
         if isinstance(stuff, SourceWrapper):
             try:
-                color_data = BASE_COLORS[stuff.color]
+                color_data = BASE_COLORS[stuff.colorID]
             except KeyError:
                 return -1
             img = self.fittingView.imageList.GetImageIndex(color_data.iconName, 'gui')
@@ -52,4 +52,4 @@ class LineColor(ViewColumn):
         return ''
 
 
-LineColor.register()
+GraphColor.register()

gui/builtinViewColumns/graphLightness.py

@@ -0,0 +1,51 @@
+# =============================================================================
+# 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.colors import BASE_COLORS
+from graphs.wrapper import TargetWrapper
+from gui.viewColumn import ViewColumn
+
+
+class GraphLightness(ViewColumn):
+
+    name = 'Graph Lightness'
+
+    def __init__(self, fittingView, params):
+        ViewColumn.__init__(self, fittingView)
+        self.resizable = False
+        self.size = 20
+        self.maxsize = self.size
+        self.mask = wx.LIST_MASK_TEXT
+
+    def getImageId(self, stuff):
+        if isinstance(stuff, TargetWrapper):
+            # img = self.fittingView.imageList.GetImageIndex(color_data.iconName, 'gui')
+            return -1
+        return -1
+
+    def getToolTip(self, stuff):
+        if isinstance(stuff, TargetWrapper):
+            return 'Change line brightness'
+        return ''
+
+
+GraphLightness.register()

gui/viewColumn.py

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