Make GivesBounty conditional.

This commit is contained in:
Paul Chote
2017-11-15 17:03:16 +00:00
committed by reaperrr
parent e79680e22c
commit 1a73a2578b

View File

@@ -17,7 +17,7 @@ using OpenRA.Traits;
namespace OpenRA.Mods.Common.Traits namespace OpenRA.Mods.Common.Traits
{ {
[Desc("When killed, this actor causes the attacking player to receive money.")] [Desc("When killed, this actor causes the attacking player to receive money.")]
class GivesBountyInfo : ITraitInfo class GivesBountyInfo : ConditionalTraitInfo
{ {
[Desc("Percentage of the killed actor's Cost or CustomSellValue to be given.")] [Desc("Percentage of the killed actor's Cost or CustomSellValue to be given.")]
public readonly int Percentage = 10; public readonly int Percentage = 10;
@@ -35,22 +35,21 @@ namespace OpenRA.Mods.Common.Traits
"Use an empty list (the default) to allow all DeathTypes.")] "Use an empty list (the default) to allow all DeathTypes.")]
public readonly HashSet<string> DeathTypes = new HashSet<string>(); public readonly HashSet<string> DeathTypes = new HashSet<string>();
public object Create(ActorInitializer init) { return new GivesBounty(init.Self, this); } public override object Create(ActorInitializer init) { return new GivesBounty(this); }
} }
class GivesBounty : INotifyKilled, INotifyCreated class GivesBounty : ConditionalTrait<GivesBountyInfo>, INotifyKilled
{ {
readonly GivesBountyInfo info;
GainsExperience gainsExp; GainsExperience gainsExp;
Cargo cargo; Cargo cargo;
public GivesBounty(Actor self, GivesBountyInfo info) public GivesBounty(GivesBountyInfo info)
{ : base(info) { }
this.info = info;
}
void INotifyCreated.Created(Actor self) protected override void Created(Actor self)
{ {
base.Created(self);
gainsExp = self.TraitOrDefault<GainsExperience>(); gainsExp = self.TraitOrDefault<GainsExperience>();
cargo = self.TraitOrDefault<Cargo>(); cargo = self.TraitOrDefault<Cargo>();
} }
@@ -62,13 +61,13 @@ namespace OpenRA.Mods.Common.Traits
return 100; return 100;
var slevel = gainsExp.Level; var slevel = gainsExp.Level;
return (slevel > 0) ? slevel * info.LevelMod : 100; return (slevel > 0) ? slevel * Info.LevelMod : 100;
} }
int GetBountyValue(Actor self) int GetBountyValue(Actor self)
{ {
// Divide by 10000 because of GetMultiplier and info.Percentage. // Divide by 10000 because of GetMultiplier and info.Percentage.
return self.GetSellValue() * GetMultiplier() * info.Percentage / 10000; return self.GetSellValue() * GetMultiplier() * Info.Percentage / 10000;
} }
int GetDisplayedBountyValue(Actor self) int GetDisplayedBountyValue(Actor self)
@@ -89,17 +88,17 @@ namespace OpenRA.Mods.Common.Traits
void INotifyKilled.Killed(Actor self, AttackInfo e) void INotifyKilled.Killed(Actor self, AttackInfo e)
{ {
if (e.Attacker == null || e.Attacker.Disposed) if (e.Attacker == null || e.Attacker.Disposed || IsTraitDisabled)
return; return;
if (!info.ValidStances.HasStance(e.Attacker.Owner.Stances[self.Owner])) if (!Info.ValidStances.HasStance(e.Attacker.Owner.Stances[self.Owner]))
return; return;
if (info.DeathTypes.Count > 0 && !e.Damage.DamageTypes.Overlaps(info.DeathTypes)) if (Info.DeathTypes.Count > 0 && !e.Damage.DamageTypes.Overlaps(Info.DeathTypes))
return; return;
var displayedBounty = GetDisplayedBountyValue(self); var displayedBounty = GetDisplayedBountyValue(self);
if (info.ShowBounty && self.IsInWorld && displayedBounty > 0 && e.Attacker.Owner.IsAlliedWith(self.World.RenderPlayer)) if (Info.ShowBounty && self.IsInWorld && displayedBounty > 0 && e.Attacker.Owner.IsAlliedWith(self.World.RenderPlayer))
e.Attacker.World.AddFrameEndTask(w => w.Add(new FloatingText(self.CenterPosition, e.Attacker.Owner.Color.RGB, FloatingText.FormatCashTick(displayedBounty), 30))); e.Attacker.World.AddFrameEndTask(w => w.Add(new FloatingText(self.CenterPosition, e.Attacker.Owner.Color.RGB, FloatingText.FormatCashTick(displayedBounty), 30)));
e.Attacker.Owner.PlayerActor.Trait<PlayerResources>().GiveCash(GetBountyValue(self)); e.Attacker.Owner.PlayerActor.Trait<PlayerResources>().GiveCash(GetBountyValue(self));