Allow RAH-specific pattern control

gui/builtinContextMenus/damagePatternChange.py

@@ -1,20 +1,16 @@
-from collections import OrderedDict
-from itertools import chain
-
 # noinspection PyPackageRequirements
 import wx
 
 import gui.globalEvents as GE
 import gui.mainFrame
+from gui.builtinContextMenus.shared.patterns import DamagePatternMixin
 from gui.contextMenu import ContextMenuUnconditional
-from gui.utils.sorter import smartSort
-from service.damagePattern import DamagePattern as DmgPatternSvc
 from service.fit import Fit
 
 _t = wx.GetTranslation
 
 
-class ChangeDamagePattern(ContextMenuUnconditional):
+class ChangeDamagePattern(ContextMenuUnconditional, DamagePatternMixin):
 
     def __init__(self):
         self.mainFrame = gui.mainFrame.MainFrame.getInstance()
@@ -27,29 +23,12 @@ class ChangeDamagePattern(ContextMenuUnconditional):
         return self.mainFrame.getActiveFit() is not None
 
     def getText(self, callingWindow, itmContext):
-        sDP = DmgPatternSvc.getInstance()
         sFit = Fit.getInstance()
         fitID = self.mainFrame.getActiveFit()
         self.fit = sFit.getFit(fitID)
-
-        builtinPatterns = sDP.getBuiltinDamagePatternList()
-        userPatterns = sorted(sDP.getUserDamagePatternList(), key=lambda p: smartSort(p.fullName))
-        # Order here is important: patterns with duplicate names from the latter will overwrite
-        # patterns from the former
-        self.patterns = sorted(
-                chain(builtinPatterns, userPatterns),
-                key=lambda p: p.fullName not in ["Uniform", "Selected Ammo"])
-
         self.patternEventMap = {}
-        self.items = (OrderedDict(), OrderedDict())
-        for pattern in self.patterns:
-            container = self.items
-            for categoryName in pattern.hierarchy:
-                categoryName = _t(categoryName) if pattern.builtin else categoryName
-                container = container[1].setdefault(categoryName, (OrderedDict(), OrderedDict()))
-            shortName = _t(pattern.shortName) if pattern.builtin else pattern.shortName
-            container[0][shortName] = pattern
-
+        self.patterns = self._getPatterns()
+        self.items = self._getItems(self.patterns)
         return list(self.items[0].keys()) + list(self.items[1].keys())
 
     def _addPattern(self, parentMenu, pattern, name):