Files
tbd-station-14/Content.Server/AI/Operators/Movement/MoveToGridOperator.cs
metalgearsloth 098b536fb8 NPC Steering refactor (#10190)
Co-authored-by: metalgearsloth <metalgearsloth@gmail.com>
2022-08-03 16:02:50 +10:00

62 lines
1.8 KiB
C#

using Content.Server.AI.Steering;
using Robust.Shared.Map;
using Robust.Shared.Utility;
namespace Content.Server.AI.Operators.Movement
{
public sealed class MoveToGridOperator : AiOperator
{
private readonly EntityUid _owner;
private readonly EntityCoordinates _target;
public float DesiredRange { get; set; }
public MoveToGridOperator(EntityUid owner, EntityCoordinates target, float desiredRange = 1.5f)
{
_owner = owner;
_target = target;
DesiredRange = desiredRange;
}
public override bool Startup()
{
if (!base.Startup())
{
return true;
}
var steering = EntitySystem.Get<NPCSteeringSystem>();
var comp = steering.Register(_owner, _target);
comp.Range = DesiredRange;
return true;
}
public override bool Shutdown(Outcome outcome)
{
if (!base.Shutdown(outcome))
return false;
var steering = EntitySystem.Get<NPCSteeringSystem>();
steering.Unregister(_owner);
return true;
}
public override Outcome Execute(float frameTime)
{
if (!IoCManager.Resolve<IEntityManager>().TryGetComponent<NPCSteeringComponent>(_owner, out var steering))
return Outcome.Failed;
switch (steering.Status)
{
case SteeringStatus.NoPath:
return Outcome.Failed;
case SteeringStatus.InRange:
return Outcome.Success;
case SteeringStatus.Moving:
return Outcome.Continuing;
default:
throw new ArgumentOutOfRangeException();
}
}
}
}