diff --git a/gui/builtinViewColumns/__init__.py b/gui/builtinViewColumns/__init__.py index 9aca80794..ebf450cb2 100644 --- a/gui/builtinViewColumns/__init__.py +++ b/gui/builtinViewColumns/__init__.py @@ -1,3 +1,3 @@ __all__ = ["moduleState", "moduleNameOrSlot", "attributeDisplay", "maxRange", "name", "droneDps", "droneNameAmount", "droneCheckbox", "moduleAmmo", - "capacitorUse", "activityCheckbox"] + "capacitorUse", "activityCheckbox", "moduleAmmoIcon"] diff --git a/gui/builtinViewColumns/moduleAmmoIcon.py b/gui/builtinViewColumns/moduleAmmoIcon.py new file mode 100755 index 000000000..055e69a40 --- /dev/null +++ b/gui/builtinViewColumns/moduleAmmoIcon.py @@ -0,0 +1,50 @@ +#=============================================================================== +# 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 . +#=============================================================================== + +from gui import builtinViewColumns +from gui.viewColumn import ViewColumn +from gui import bitmapLoader + +class ModuleAmmoIcon(ViewColumn): + name = "Module Ammo Icon" + def __init__(self, fittingView, params): + ViewColumn.__init__(self, fittingView) + self.size = 20 + self.columnText = "" + + def getText(self, mod): + return "" + + def getImageId(self, mod): + if mod.charge is None: + iconId = -1 + else: + iconFile = mod.charge.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 + +ModuleAmmoIcon.register() diff --git a/gui/fittingView.py b/gui/fittingView.py index a7fb9c32a..16103e39c 100644 --- a/gui/fittingView.py +++ b/gui/fittingView.py @@ -31,13 +31,14 @@ FitChanged, FIT_CHANGED = wx.lib.newevent.NewEvent() class FittingView(d.Display): DEFAULT_COLS = ["Module state", + "Module Ammo Icon", "Module name/slot", - "Module Ammo", "attr:trackingSpeed", "Max range", "Capacitor Usage", "attr:power", "attr:cpu", + "Module Ammo", ] class FittingViewDrop(wx.PyDropTarget):