epic/OpenRA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Changes to ExternalCondition.TryRevokeCondition

Browse Source
This commit is contained in:
atlimit8
2017-07-18 13:34:05 -05:00
committed by Paul Chote
parent 72236b66f6
commit c7d3c3ec73
4 changed files with 19 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
OpenRA.Mods.Common/Traits/Conditions/ExternalCondition.cs
View File

@@ -130,10 +130,20 @@ namespace OpenRA.Mods.Common.Traits
if (conditionManager == null || source == null)
return false;
var removed = permanentTokens.GetOrAdd(source).Remove(token) ||
timedTokens.GetOrAdd(source).RemoveWhere(t => t.Token == token) > 0;
HashSet<int> permanentTokensForSource;
if (permanentTokens.TryGetValue(source, out permanentTokensForSource))
{
if (!permanentTokensForSource.Remove(token))
return false;
}
else
{
HashSet<TimedToken> timedTokensForSource = null;
if (timedTokens.TryGetValue(source, out timedTokensForSource) && timedTokensForSource.RemoveWhere(t => t.Token == token) == 0)
return false;
}
if (removed && conditionManager.TokenValid(self, token))
if (conditionManager.TokenValid(self, token))
conditionManager.RevokeCondition(self, token);
return true;