Fix EngineerRepairable stance checks.

This commit is contained in:
Mustafa Alperen Seki
2019-04-13 20:28:24 +03:00
committed by reaperrr
parent 8752e76d1e
commit 2194584b25
2 changed files with 18 additions and 14 deletions

View File

@@ -9,6 +9,7 @@
*/ */
#endregion #endregion
using OpenRA.Mods.Common.Traits;
using OpenRA.Primitives; using OpenRA.Primitives;
using OpenRA.Traits; using OpenRA.Traits;
@@ -16,17 +17,15 @@ namespace OpenRA.Mods.Common.Activities
{ {
class RepairBuilding : Enter class RepairBuilding : Enter
{ {
readonly EnterBehaviour enterBehaviour; readonly EngineerRepairInfo info;
readonly Stance validStances;
Actor enterActor; Actor enterActor;
IHealth enterHealth; IHealth enterHealth;
public RepairBuilding(Actor self, Target target, EnterBehaviour enterBehaviour, Stance validStances) public RepairBuilding(Actor self, Target target, EngineerRepairInfo info)
: base(self, target, Color.Yellow) : base(self, target, Color.Yellow)
{ {
this.enterBehaviour = enterBehaviour; this.info = info;
this.validStances = validStances;
} }
protected override bool TryStartEnter(Actor self, Actor targetActor) protected override bool TryStartEnter(Actor self, Actor targetActor)
@@ -37,7 +36,7 @@ namespace OpenRA.Mods.Common.Activities
// Make sure we can still repair the target before entering // Make sure we can still repair the target before entering
// (but not before, because this may stop the actor in the middle of nowhere) // (but not before, because this may stop the actor in the middle of nowhere)
var stance = self.Owner.Stances[enterActor.Owner]; var stance = self.Owner.Stances[enterActor.Owner];
if (enterHealth == null || enterHealth.DamageState == DamageState.Undamaged || !stance.HasStance(validStances)) if (enterHealth == null || enterHealth.DamageState == DamageState.Undamaged || !info.ValidStances.HasStance(stance))
{ {
Cancel(self, true); Cancel(self, true);
return false; return false;
@@ -57,7 +56,7 @@ namespace OpenRA.Mods.Common.Activities
return; return;
var stance = self.Owner.Stances[enterActor.Owner]; var stance = self.Owner.Stances[enterActor.Owner];
if (!stance.HasStance(validStances)) if (!info.ValidStances.HasStance(stance))
return; return;
if (enterHealth.DamageState == DamageState.Undamaged) if (enterHealth.DamageState == DamageState.Undamaged)
@@ -65,9 +64,9 @@ namespace OpenRA.Mods.Common.Activities
enterActor.InflictDamage(self, new Damage(-enterHealth.MaxHP)); enterActor.InflictDamage(self, new Damage(-enterHealth.MaxHP));
if (enterBehaviour == EnterBehaviour.Dispose) if (info.EnterBehaviour == EnterBehaviour.Dispose)
self.Dispose(); self.Dispose();
else if (enterBehaviour == EnterBehaviour.Suicide) else if (info.EnterBehaviour == EnterBehaviour.Suicide)
self.Kill(self); self.Kill(self);
} }
} }

View File

@@ -80,20 +80,25 @@ namespace OpenRA.Mods.Common.Traits
self.CancelActivity(); self.CancelActivity();
self.SetTargetLine(order.Target, Color.Yellow); self.SetTargetLine(order.Target, Color.Yellow);
self.QueueActivity(new RepairBuilding(self, order.Target, info.EnterBehaviour, info.ValidStances)); self.QueueActivity(new RepairBuilding(self, order.Target, info));
} }
class EngineerRepairOrderTargeter : UnitOrderTargeter class EngineerRepairOrderTargeter : UnitOrderTargeter
{ {
public EngineerRepairOrderTargeter() EngineerRepairInfo info;
: base("EngineerRepair", 6, "goldwrench", false, true) { }
public EngineerRepairOrderTargeter(EngineerRepairInfo info)
: base("EngineerRepair", 6, "goldwrench", true, true)
{
this.info = info;
}
public override bool CanTargetActor(Actor self, Actor target, TargetModifiers modifiers, ref string cursor) public override bool CanTargetActor(Actor self, Actor target, TargetModifiers modifiers, ref string cursor)
{ {
if (!target.Info.HasTraitInfo<EngineerRepairableInfo>()) if (!target.Info.HasTraitInfo<EngineerRepairableInfo>())
return false; return false;
if (self.Owner.Stances[target.Owner] != Stance.Ally) if (!info.ValidStances.HasStance(self.Owner.Stances[target.Owner]))
return false; return false;
if (target.GetDamageState() == DamageState.Undamaged) if (target.GetDamageState() == DamageState.Undamaged)
@@ -107,7 +112,7 @@ namespace OpenRA.Mods.Common.Traits
if (!target.Info.HasTraitInfo<EngineerRepairableInfo>()) if (!target.Info.HasTraitInfo<EngineerRepairableInfo>())
return false; return false;
if (self.Owner.Stances[target.Owner] != Stance.Ally) if (!info.ValidStances.HasStance(self.Owner.Stances[target.Owner]))
return false; return false;
if (target.DamageState == DamageState.Undamaged) if (target.DamageState == DamageState.Undamaged)