Files
tbd-station-14/Content.Server/AI/Utility/Considerations/Clothing/ClothingInInventoryCon.cs
2020-07-07 00:23:41 +02:00

39 lines
1.1 KiB
C#

using Content.Server.AI.Utility.Curves;
using Content.Server.AI.WorldState;
using Content.Server.AI.WorldState.States.Inventory;
using Content.Server.GameObjects;
using Content.Shared.GameObjects.Components.Inventory;
namespace Content.Server.AI.Utility.Considerations.Clothing
{
public sealed class ClothingInInventoryCon : Consideration
{
private readonly EquipmentSlotDefines.SlotFlags _slot;
public ClothingInInventoryCon(EquipmentSlotDefines.SlotFlags slotFlags, IResponseCurve curve) : base(curve)
{
_slot = slotFlags;
}
public override float GetScore(Blackboard context)
{
var inventory = context.GetState<InventoryState>().GetValue();
foreach (var entity in inventory)
{
if (!entity.TryGetComponent(out ClothingComponent clothingComponent))
{
continue;
}
if ((clothingComponent.SlotFlags & _slot) != 0)
{
return 1.0f;
}
}
return 0.0f;
}
}
}