45 lines
1.4 KiB
C#
45 lines
1.4 KiB
C#
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Content.Server.NPC.Components;
|
|
using Robust.Shared.Random;
|
|
|
|
namespace Content.Server.NPC.HTN.PrimitiveTasks.Operators.Test;
|
|
|
|
public sealed class PickPathfindPointOperator : HTNOperator
|
|
{
|
|
[Dependency] private readonly IEntityManager _entManager = default!;
|
|
[Dependency] private readonly IRobustRandom _random = default!;
|
|
|
|
public override async Task<(bool Valid, Dictionary<string, object>? Effects)> Plan(NPCBlackboard blackboard)
|
|
{
|
|
var owner = blackboard.GetValue<EntityUid>(NPCBlackboard.Owner);
|
|
|
|
// Find all pathfind points on the same grid and choose to move to it.
|
|
var xform = _entManager.GetComponent<TransformComponent>(owner);
|
|
var gridUid = xform.GridUid;
|
|
|
|
if (gridUid == null)
|
|
return (false, null);
|
|
|
|
var points = new List<TransformComponent>();
|
|
|
|
foreach (var (point, pointXform) in _entManager.EntityQuery<NPCPathfindPointComponent, TransformComponent>(true))
|
|
{
|
|
if (gridUid != pointXform.GridUid)
|
|
continue;
|
|
|
|
points.Add(pointXform);
|
|
}
|
|
|
|
if (points.Count == 0)
|
|
return (false, null);
|
|
|
|
var selected = _random.Pick(points);
|
|
|
|
return (true, new Dictionary<string, object>()
|
|
{
|
|
{ NPCBlackboard.MovementTarget, selected.Coordinates }
|
|
});
|
|
}
|
|
}
|