diff --git a/gui/builtinViewColumns/attributeDisplayGraph.py b/gui/builtinViewColumns/attributeDisplayGraph.py index 7fd4d0df5..3f2e1991b 100644 --- a/gui/builtinViewColumns/attributeDisplayGraph.py +++ b/gui/builtinViewColumns/attributeDisplayGraph.py @@ -34,6 +34,8 @@ from gui.viewColumn import ViewColumn class GraphColumn(ViewColumn, metaclass=ABCMeta): + stickPrefixToValue = False + def __init__(self, fittingView, iconID, formatSpec=(3, 0, 3)): ViewColumn.__init__(self, fittingView) self.imageId = fittingView.imageList.GetImageIndex(iconID, 'icons') @@ -52,7 +54,12 @@ class GraphColumn(ViewColumn, metaclass=ABCMeta): val, unit = self._getValue(stuff) if val is None: return '' - return formatAmount(val, *self.formatSpec, unitName=unit) + # Stick to value - 25k GJ + if self.stickPrefixToValue: + return '{} {}'.format(formatAmount(val, *self.formatSpec), unit) + # Stick to unit - 25 km + else: + return formatAmount(val, *self.formatSpec, unitName=unit) return '' @abstractmethod @@ -192,6 +199,7 @@ SignatureRadiusColumn.register() class ShieldAmountColumn(GraphColumn): name = 'ShieldAmount' + stickPrefixToValue = True def __init__(self, fittingView, params): super().__init__(fittingView, 1384) @@ -199,14 +207,6 @@ class ShieldAmountColumn(GraphColumn): def _getValue(self, fit): return fit.ship.getModifiedItemAttr('shieldCapacity'), 'HP' - def getText(self, stuff): - if isinstance(stuff, Fit): - val, unit = self._getValue(stuff) - if val is None: - return '' - return '{} {}'.format(formatAmount(val, *self.formatSpec), unit) - return '' - def _getFitTooltip(self): return 'Maximum shield amount' @@ -234,6 +234,7 @@ ShieldTimeColumn.register() class CapAmountColumn(GraphColumn): name = 'CapAmount' + stickPrefixToValue = True def __init__(self, fittingView, params): super().__init__(fittingView, 1668) @@ -241,14 +242,6 @@ class CapAmountColumn(GraphColumn): def _getValue(self, fit): return fit.ship.getModifiedItemAttr('capacitorCapacity'), 'GJ' - def getText(self, stuff): - if isinstance(stuff, Fit): - val, unit = self._getValue(stuff) - if val is None: - return '' - return '{} {}'.format(formatAmount(val, *self.formatSpec), unit) - return '' - def _getFitTooltip(self): return 'Maximum capacitor amount'