Port RepairBuilding to the new Enter activity.
This commit is contained in:
committed by
Oliver Brakmann
parent
167371d540
commit
49e3c46d00
@@ -14,35 +14,61 @@ using OpenRA.Traits;
|
|||||||
|
|
||||||
namespace OpenRA.Mods.Common.Activities
|
namespace OpenRA.Mods.Common.Activities
|
||||||
{
|
{
|
||||||
class RepairBuilding : LegacyEnter
|
class RepairBuilding : Enter
|
||||||
{
|
{
|
||||||
readonly Actor target;
|
readonly EnterBehaviour enterBehaviour;
|
||||||
readonly IHealth health;
|
|
||||||
readonly Stance validStances;
|
readonly Stance validStances;
|
||||||
|
|
||||||
public RepairBuilding(Actor self, Actor target, EnterBehaviour enterBehaviour, Stance validStances)
|
Actor enterActor;
|
||||||
: base(self, target, enterBehaviour, targetLineColor: Color.Yellow)
|
IHealth enterHealth;
|
||||||
|
|
||||||
|
public RepairBuilding(Actor self, Target target, EnterBehaviour enterBehaviour, Stance validStances)
|
||||||
|
: base(self, target, Color.Yellow)
|
||||||
{
|
{
|
||||||
this.target = target;
|
this.enterBehaviour = enterBehaviour;
|
||||||
this.validStances = validStances;
|
this.validStances = validStances;
|
||||||
health = target.Trait<IHealth>();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
protected override bool CanReserve(Actor self)
|
protected override bool TryStartEnter(Actor self, Actor targetActor)
|
||||||
{
|
{
|
||||||
return health.DamageState != DamageState.Undamaged;
|
enterActor = targetActor;
|
||||||
|
enterHealth = targetActor.TraitOrDefault<IHealth>();
|
||||||
|
|
||||||
|
// Make sure we can still repair the target before entering
|
||||||
|
// (but not before, because this may stop the actor in the middle of nowhere)
|
||||||
|
var stance = self.Owner.Stances[enterActor.Owner];
|
||||||
|
if (enterHealth == null || enterHealth.DamageState == DamageState.Undamaged || !stance.HasStance(validStances))
|
||||||
|
{
|
||||||
|
Cancel(self, true);
|
||||||
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
protected override void OnInside(Actor self)
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
protected override void OnEnterComplete(Actor self, Actor targetActor)
|
||||||
{
|
{
|
||||||
var stance = self.Owner.Stances[target.Owner];
|
// Make sure the target hasn't changed while entering
|
||||||
|
// OnEnterComplete is only called if targetActor is alive
|
||||||
|
if (targetActor != enterActor)
|
||||||
|
return;
|
||||||
|
|
||||||
|
if (enterHealth.DamageState == DamageState.Undamaged)
|
||||||
|
return;
|
||||||
|
|
||||||
|
var stance = self.Owner.Stances[enterActor.Owner];
|
||||||
if (!stance.HasStance(validStances))
|
if (!stance.HasStance(validStances))
|
||||||
return;
|
return;
|
||||||
|
|
||||||
if (health.DamageState == DamageState.Undamaged)
|
if (enterHealth.DamageState == DamageState.Undamaged)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
target.InflictDamage(self, new Damage(-health.MaxHP));
|
enterActor.InflictDamage(self, new Damage(-enterHealth.MaxHP));
|
||||||
|
|
||||||
|
if (enterBehaviour == EnterBehaviour.Dispose)
|
||||||
|
self.Dispose();
|
||||||
|
else if (enterBehaviour == EnterBehaviour.Suicide)
|
||||||
|
self.Kill(self);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -76,15 +76,11 @@ namespace OpenRA.Mods.Common.Traits
|
|||||||
if (order.OrderString != "EngineerRepair" || !IsValidOrder(self, order))
|
if (order.OrderString != "EngineerRepair" || !IsValidOrder(self, order))
|
||||||
return;
|
return;
|
||||||
|
|
||||||
var target = self.ResolveFrozenActorOrder(order, Color.Yellow);
|
|
||||||
if (target.Type != TargetType.Actor)
|
|
||||||
return;
|
|
||||||
|
|
||||||
if (!order.Queued)
|
if (!order.Queued)
|
||||||
self.CancelActivity();
|
self.CancelActivity();
|
||||||
|
|
||||||
self.SetTargetLine(target, Color.Yellow);
|
self.SetTargetLine(order.Target, Color.Yellow);
|
||||||
self.QueueActivity(new RepairBuilding(self, target.Actor, info.EnterBehaviour, info.ValidStances));
|
self.QueueActivity(new RepairBuilding(self, order.Target, info.EnterBehaviour, info.ValidStances));
|
||||||
}
|
}
|
||||||
|
|
||||||
class EngineerRepairOrderTargeter : UnitOrderTargeter
|
class EngineerRepairOrderTargeter : UnitOrderTargeter
|
||||||
|
|||||||
Reference in New Issue
Block a user