From 635e684c3a1652dd3313f979cf374f61a51a6e5f Mon Sep 17 00:00:00 2001
From: Headpigeon <thierry.steebierg@gmail.com>
Date: Sun, 2 Nov 2014 19:12:03 +0100
Subject: [PATCH] Added module context menu entry for changing charges on all
 modules of same type at once (no need to manually select them)

---
 gui/builtinContextMenus/__init__.py           |  6 +--
 .../moduleGlobalAmmoPicker.py                 | 51 +++++++++++++++++++
 2 files changed, 54 insertions(+), 3 deletions(-)
 create mode 100644 gui/builtinContextMenus/moduleGlobalAmmoPicker.py

diff --git a/gui/builtinContextMenus/__init__.py b/gui/builtinContextMenus/__init__.py
index f497c9d27..c03a033cd 100644
--- a/gui/builtinContextMenus/__init__.py
+++ b/gui/builtinContextMenus/__init__.py
@@ -1,3 +1,3 @@
-__all__ = ["openFit", "moduleAmmoPicker", "itemStats", "damagePattern", "marketJump", "droneSplit", "itemRemove",
-           "droneRemoveStack", "ammoPattern", "project", "factorReload", "whProjector", "cargo", "shipJump",
-           "targetResists"]
+__all__ = ["openFit", "moduleGlobalAmmoPicker", "moduleAmmoPicker", "itemStats", "damagePattern", "marketJump",
+		   "droneSplit", "itemRemove", "droneRemoveStack", "ammoPattern", "project", "factorReload", "whProjector",
+		   "cargo", "shipJump", "targetResists"]
diff --git a/gui/builtinContextMenus/moduleGlobalAmmoPicker.py b/gui/builtinContextMenus/moduleGlobalAmmoPicker.py
new file mode 100644
index 000000000..44a1adced
--- /dev/null
+++ b/gui/builtinContextMenus/moduleGlobalAmmoPicker.py
@@ -0,0 +1,51 @@
+# -*- coding: utf-8 -*-
+from gui.contextMenu import ContextMenu
+import gui.mainFrame
+import service
+import wx
+from gui import bitmapLoader
+from eos.types import Hardpoint
+import gui.globalEvents as GE
+from gui.builtinContextMenus.moduleAmmoPicker import ModuleAmmoPicker
+import eos.db
+
+class ModuleGlobalAmmoPicker(ModuleAmmoPicker):
+    def __init__(self):
+        self.mainFrame = gui.mainFrame.MainFrame.getInstance()
+
+    def getText(self, itmContext, selection):
+        return "Charge (All)"
+
+    def handleAmmoSwitch(self, event):
+        if len(self.modules) != 1:
+            event.Skip()
+            return
+
+        charge = self.chargeIds.get(event.Id, False)
+        if charge is False:
+            event.Skip()
+            return
+
+        sFit = service.Fit.getInstance()
+        fitID = self.mainFrame.getActiveFit()
+        fit = eos.db.getFit(fitID)
+
+        selectedModule = self.modules[0]
+        allModules = []
+        for mod in fit.modules:
+            if mod.itemID == None:
+                continue
+            if mod.itemID == selectedModule.itemID:
+                allModules.append(mod)
+
+        sFit.setAmmo(fitID, charge.ID if charge is not None else None, allModules)
+        wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=fitID))
+
+    def display(self, srcContext, selection):
+        if len(selection) != 1:
+            return False
+
+        return super(ModuleGlobalAmmoPicker, self).display(srcContext, selection)
+
+
+ModuleGlobalAmmoPicker.register()