101 lines
3.2 KiB
C#
101 lines
3.2 KiB
C#
using Content.Server.CombatMode;
|
|
using Content.Server.NPC.Components;
|
|
using Content.Shared.MobState;
|
|
using Content.Shared.MobState.Components;
|
|
using Content.Shared.Weapons.Melee;
|
|
using Robust.Shared.Map;
|
|
|
|
namespace Content.Server.NPC.Systems;
|
|
|
|
public sealed partial class NPCCombatSystem
|
|
{
|
|
private const float TargetMeleeLostRange = 14f;
|
|
|
|
private void InitializeMelee()
|
|
{
|
|
SubscribeLocalEvent<NPCMeleeCombatComponent, ComponentStartup>(OnMeleeStartup);
|
|
SubscribeLocalEvent<NPCMeleeCombatComponent, ComponentShutdown>(OnMeleeShutdown);
|
|
}
|
|
|
|
private void OnMeleeShutdown(EntityUid uid, NPCMeleeCombatComponent component, ComponentShutdown args)
|
|
{
|
|
if (TryComp<CombatModeComponent>(uid, out var combatMode))
|
|
{
|
|
combatMode.IsInCombatMode = false;
|
|
}
|
|
|
|
_steering.Unregister(component.Owner);
|
|
}
|
|
|
|
private void OnMeleeStartup(EntityUid uid, NPCMeleeCombatComponent component, ComponentStartup args)
|
|
{
|
|
if (TryComp<CombatModeComponent>(uid, out var combatMode))
|
|
{
|
|
combatMode.IsInCombatMode = true;
|
|
}
|
|
|
|
// TODO: Cleanup later, just looking for parity for now.
|
|
component.Weapon = uid;
|
|
}
|
|
|
|
private void UpdateMelee(float frameTime)
|
|
{
|
|
var combatQuery = GetEntityQuery<CombatModeComponent>();
|
|
var xformQuery = GetEntityQuery<TransformComponent>();
|
|
|
|
foreach (var (comp, _) in EntityQuery<NPCMeleeCombatComponent, ActiveNPCComponent>())
|
|
{
|
|
if (!combatQuery.TryGetComponent(comp.Owner, out var combat) || !combat.IsInCombatMode)
|
|
{
|
|
RemComp<NPCMeleeCombatComponent>(comp.Owner);
|
|
continue;
|
|
}
|
|
|
|
Attack(comp, xformQuery);
|
|
}
|
|
}
|
|
|
|
private void Attack(NPCMeleeCombatComponent component, EntityQuery<TransformComponent> xformQuery)
|
|
{
|
|
component.Status = CombatStatus.Normal;
|
|
|
|
// TODO:
|
|
// Also need some blackboard data for stuff like juke frequency, assigning target slots (to surround targets), etc.
|
|
// miss %
|
|
if (!TryComp<MeleeWeaponComponent>(component.Weapon, out var weapon))
|
|
{
|
|
component.Status = CombatStatus.NoWeapon;
|
|
return;
|
|
}
|
|
|
|
if (!xformQuery.TryGetComponent(component.Owner, out var xform) ||
|
|
!xformQuery.TryGetComponent(component.Target, out var targetXform))
|
|
{
|
|
component.Status = CombatStatus.TargetUnreachable;
|
|
return;
|
|
}
|
|
|
|
if (!xform.Coordinates.TryDistance(EntityManager, targetXform.Coordinates, out var distance))
|
|
{
|
|
component.Status = CombatStatus.TargetUnreachable;
|
|
return;
|
|
}
|
|
|
|
if (distance > TargetMeleeLostRange)
|
|
{
|
|
component.Status = CombatStatus.TargetUnreachable;
|
|
return;
|
|
}
|
|
|
|
if (distance > weapon.Range)
|
|
{
|
|
component.Status = CombatStatus.TargetOutOfRange;
|
|
return;
|
|
}
|
|
|
|
// Gets unregistered on component shutdown.
|
|
_steering.TryRegister(component.Owner, new EntityCoordinates(component.Target, Vector2.Zero));
|
|
_melee.AttemptLightAttack(component.Owner, weapon, component.Target);
|
|
}
|
|
}
|