tbd-station-14/Content.Server/AI/Utility/Actions/Combat/Melee/EquipMelee.cs

using System.Collections.Generic;
using Content.Server.AI.Operators;
using Content.Server.AI.Operators.Inventory;
using Content.Server.AI.Utility.Considerations;
using Content.Server.AI.Utility.Considerations.Combat.Melee;
using Content.Server.AI.Utility.Considerations.Combat.Ranged;
using Content.Server.AI.Utility.Considerations.Inventory;
using Content.Server.AI.Utility.Curves;
using Content.Server.AI.WorldState;
using Content.Server.AI.WorldState.States;
using Content.Server.AI.WorldState.States.Combat;
using Robust.Shared.Interfaces.GameObjects;

namespace Content.Server.AI.Utility.Actions.Combat.Melee
{
    public sealed class EquipMelee : UtilityAction
    {
        private IEntity _entity;

        public EquipMelee(IEntity owner, IEntity entity, float weight) : base(owner)
        {
            _entity = entity;
            Bonus = weight;
        }

        public override void SetupOperators(Blackboard context)
        {
            ActionOperators = new Queue<AiOperator>(new AiOperator[]
            {
                new EquipEntityOperator(Owner, _entity)
            });
        }

        protected override void UpdateBlackboard(Blackboard context)
        {
            base.UpdateBlackboard(context);
            context.GetState<WeaponEntityState>().SetValue(_entity);
            context.GetState<TargetEntityState>().SetValue(_entity);
        }

        protected override Consideration[] Considerations { get; } = {
            new MeleeWeaponEquippedCon(
                new InverseBoolCurve()),
            // We'll prioritise equipping ranged weapons; If we try and score this then it'll just keep swapping between ranged and melee
            new RangedWeaponEquippedCon(
                new InverseBoolCurve()),
            new CanPutTargetInHandsCon(
                new BoolCurve()),
            new MeleeWeaponSpeedCon(
                new QuadraticCurve(1.0f, 0.5f, 0.0f, 0.0f)),
            new MeleeWeaponDamageCon(
                new QuadraticCurve(1.0f, 0.25f, 0.0f, 0.0f)),
        };
    }
}