Files
tbd-station-14/Content.Server/AI/Utility/Considerations/Inventory/CanPutTargetInInventoryCon.cs
2022-05-13 17:59:03 +10:00

34 lines
1.1 KiB
C#

using Content.Server.AI.WorldState;
using Content.Server.AI.WorldState.States;
using Content.Server.AI.WorldState.States.Hands;
using Content.Server.AI.WorldState.States.Inventory;
using Content.Shared.Item;
namespace Content.Server.AI.Utility.Considerations.Inventory
{
public sealed class CanPutTargetInInventoryCon : Consideration
{
protected override float GetScore(Blackboard context)
{
// First check if target in inventory already
// If not then check if we have a free hand
var target = context.GetState<TargetEntityState>().GetValue();
if (target == null || !IoCManager.Resolve<IEntityManager>().HasComponent<SharedItemComponent>(target))
{
return 0.0f;
}
foreach (var item in context.GetState<EnumerableInventoryState>().GetValue())
{
if (item == target)
{
return 1.0f;
}
}
return context.GetState<AnyFreeHandState>().GetValue() ? 1.0f : 0.0f;
}
}
}