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

Safeguard CashTrickler negative values

Browse Source 
By making sure to never remove more cash than the player has.
This commit is contained in:
reaperrr
2016-11-12 19:07:17 +01:00
parent 7dcda5db30
commit 8699d0fca3
1 changed files with 34 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
OpenRA.Mods.Common/Traits/CashTrickler.cs
View File

@@ -9,13 +9,14 @@
*/
#endregion
using System;
using OpenRA.Mods.Common.Effects;
using OpenRA.Traits;
namespace OpenRA.Mods.Common.Traits
{
[Desc("Lets the actor generate cash in a set periodic time.")]
class CashTricklerInfo : ConditionalTraitInfo
public class CashTricklerInfo : ConditionalTraitInfo
{
[Desc("Number of ticks to wait between giving money.")]
public readonly int Interval = 50;
@@ -32,9 +33,10 @@ namespace OpenRA.Mods.Common.Traits
public override object Create(ActorInitializer init) { return new CashTrickler(this); }
}
class CashTrickler : ConditionalTrait<CashTricklerInfo>, ITick, ISync
public class CashTrickler : ConditionalTrait<CashTricklerInfo>, ITick, ISync, INotifyCreated, INotifyOwnerChanged
{
readonly CashTricklerInfo info;
PlayerResources resources;
[Sync] int ticks;
public CashTrickler(CashTricklerInfo info)
@@ -43,6 +45,16 @@ namespace OpenRA.Mods.Common.Traits
this.info = info;
}
void INotifyCreated.Created(Actor self)
{
resources = self.Owner.PlayerActor.Trait<PlayerResources>();
}
void INotifyOwnerChanged.OnOwnerChanged(Actor self, Player oldOwner, Player newOwner)
{
resources = newOwner.PlayerActor.Trait<PlayerResources>();
}
void ITick.Tick(Actor self)
{
if (IsTraitDisabled)
@@ -51,10 +63,7 @@ namespace OpenRA.Mods.Common.Traits
if (--ticks < 0)
{
ticks = info.Interval;
self.Owner.PlayerActor.Trait<PlayerResources>().GiveCash(info.Amount);
if (info.ShowTicks)
AddCashTick(self, info.Amount);
ModifyCash(self, self.Owner, info.Amount);
}
}
@@ -63,5 +72,24 @@ namespace OpenRA.Mods.Common.Traits
self.World.AddFrameEndTask(w => w.Add(
new FloatingText(self.CenterPosition, self.Owner.Color.RGB, FloatingText.FormatCashTick(amount), info.DisplayDuration)));
}
void ModifyCash(Actor self, Player newOwner, int amount)
{
if (amount < 0)
{
// Check whether the amount of cash to be removed would exceed available player cash, in that case only remove all the player cash
var drain = Math.Min(resources.Cash + resources.Resources, -amount);
resources.TakeCash(drain);
if (info.ShowTicks)
AddCashTick(self, -drain);
}
else
{
resources.GiveCash(amount);
if (info.ShowTicks)
AddCashTick(self, amount);
}
}
}
}