dave/pyfa
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 5 Wiki Activity

Save & restore RAH-specific damage pattern through do-undo actions

Browse Source
This commit is contained in:
DarkPhoenix
2022-03-12 13:49:18 +03:00
parent ae115b640a
commit 9e06106a1c
1 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
gui/fitCommands/helpers.py
View File

@@ -58,7 +58,9 @@ class InternalCommandHistory:
class ModuleInfo:
def __init__(self, itemID, baseItemID=None, mutaplasmidID=None, mutations=None, chargeID=None, state=None, spoolType=None, spoolAmount=None):
def __init__(
self, itemID, baseItemID=None, mutaplasmidID=None, mutations=None, chargeID=None,
state=None, spoolType=None, spoolAmount=None, rahPattern=None):
self.itemID = itemID
self.baseItemID = baseItemID
self.mutaplasmidID = mutaplasmidID
@@ -67,6 +69,7 @@ class ModuleInfo:
self.state = state
self.spoolType = spoolType
self.spoolAmount = spoolAmount
self.rahPattern = rahPattern
@classmethod
def fromModule(cls, mod, unmutate=False):
@@ -81,7 +84,8 @@ class ModuleInfo:
chargeID=mod.chargeID,
state=mod.state,
spoolType=mod.spoolType,
spoolAmount=mod.spoolAmount)
spoolAmount=mod.spoolAmount,
rahPattern=mod.rahPatternOverride)
else:
info = cls(
itemID=mod.itemID,
@@ -91,7 +95,8 @@ class ModuleInfo:
chargeID=mod.chargeID,
state=mod.state,
spoolType=mod.spoolType,
spoolAmount=mod.spoolAmount)
spoolAmount=mod.spoolAmount,
rahPattern=mod.rahPatternOverride)
return info
def toModule(self, fallbackState=None):
@@ -119,6 +124,8 @@ class ModuleInfo:
mod.spoolType = self.spoolType
mod.spoolAmount = self.spoolAmount
mod.rahPatternOverride = self.rahPattern
if self.state is not None:
if mod.isValidState(self.state):
mod.state = self.state
@@ -148,12 +155,13 @@ class ModuleInfo:
self.chargeID == other.chargeID,
self.state == other.state,
self.spoolType == other.spoolType,
self.spoolAmount == other.spoolAmount))
self.spoolAmount == other.spoolAmount,
self.rahPattern == other.rahPattern))
def __repr__(self):
return makeReprStr(self, [
'itemID', 'baseItemID', 'mutaplasmidID', 'mutations',
'chargeID', 'state', 'spoolType', 'spoolAmount'])
'chargeID', 'state', 'spoolType', 'spoolAmount', 'rahPattern'])
class DroneInfo: