tommy/tbd-station-14
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix fish petting (#16094)

Browse Source 
Co-authored-by: deltanedas <@deltanedas:kde.org>
This commit is contained in:
deltanedas
2023-05-05 13:03:46 +00:00
committed by GitHub
parent 2e2621396f
commit 1ca9328c15
1 changed files with 198 additions and 190 deletions
Download Patch File Download Diff File Expand all files Collapse all files

388
Content.Server/NPC/Systems/FactionSystem.cs
View File

@@ -1,221 +1,229 @@
using System.Linq;
using Content.Server.NPC.Components;
using Robust.Shared.Prototypes;
using System.Linq;
namespace Content.Server.NPC.Systems
namespace Content.Server.NPC.Systems;
/// <summary>
/// Outlines faction relationships with each other.
/// </summary>
public sealed class FactionSystem : EntitySystem
{
[Dependency] private readonly FactionExceptionSystem _factionException = default!;
[Dependency] private readonly EntityLookupSystem _lookup = default!;
[Dependency] private readonly IPrototypeManager _protoManager = default!;
private ISawmill _sawmill = default!;
/// <summary>
/// Outlines faction relationships with each other.
/// To avoid prototype mutability we store an intermediary data class that gets used instead.
/// </summary>
public sealed class FactionSystem : EntitySystem
private Dictionary<string, FactionData> _factions = new();
public override void Initialize()
{
[Dependency] private readonly IPrototypeManager _protoManager = default!;
[Dependency] private readonly EntityLookupSystem _lookup = default!;
base.Initialize();
_sawmill = Logger.GetSawmill("faction");
SubscribeLocalEvent<FactionComponent, ComponentStartup>(OnFactionStartup);
_protoManager.PrototypesReloaded += OnProtoReload;
RefreshFactions();
}
private ISawmill _sawmill = default!;
public override void Shutdown()
{
base.Shutdown();
_protoManager.PrototypesReloaded -= OnProtoReload;
}
/// <summary>
/// To avoid prototype mutability we store an intermediary data class that gets used instead.
/// </summary>
private Dictionary<string, FactionData> _factions = new();
private void OnProtoReload(PrototypesReloadedEventArgs obj)
{
if (!obj.ByType.ContainsKey(typeof(FactionPrototype)))
return;
public override void Initialize()
RefreshFactions();
}
private void OnFactionStartup(EntityUid uid, FactionComponent component, ComponentStartup args)
{
RefreshFactions(component);
}
/// <summary>
/// Refreshes the cached factions for this component.
/// </summary>
private void RefreshFactions(FactionComponent component)
{
component.FriendlyFactions.Clear();
component.HostileFactions.Clear();
foreach (var faction in component.Factions)
{
base.Initialize();
_sawmill = Logger.GetSawmill("faction");
SubscribeLocalEvent<FactionComponent, ComponentStartup>(OnFactionStartup);
_protoManager.PrototypesReloaded += OnProtoReload;
RefreshFactions();
// YAML Linter already yells about this
if (!_factions.TryGetValue(faction, out var factionData))
continue;
component.FriendlyFactions.UnionWith(factionData.Friendly);
component.HostileFactions.UnionWith(factionData.Hostile);
}
}
/// <summary>
/// Adds this entity to the particular faction.
/// </summary>
public void AddFaction(EntityUid uid, string faction, bool dirty = true)
{
if (!_protoManager.HasIndex<FactionPrototype>(faction))
{
_sawmill.Error($"Unable to find faction {faction}");
return;
}
public override void Shutdown()
var comp = EnsureComp<FactionComponent>(uid);
if (!comp.Factions.Add(faction))
return;
if (dirty)
{
base.Shutdown();
_protoManager.PrototypesReloaded -= OnProtoReload;
RefreshFactions(comp);
}
}
/// <summary>
/// Removes this entity from the particular faction.
/// </summary>
public void RemoveFaction(EntityUid uid, string faction, bool dirty = true)
{
if (!_protoManager.HasIndex<FactionPrototype>(faction))
{
_sawmill.Error($"Unable to find faction {faction}");
return;
}
private void OnProtoReload(PrototypesReloadedEventArgs obj)
{
if (!obj.ByType.ContainsKey(typeof(FactionPrototype)))
return;
if (!TryComp<FactionComponent>(uid, out var component))
return;
RefreshFactions();
}
if (!component.Factions.Remove(faction))
return;
private void OnFactionStartup(EntityUid uid, FactionComponent component, ComponentStartup args)
if (dirty)
{
RefreshFactions(component);
}
}
/// <summary>
/// Refreshes the cached factions for this component.
/// </summary>
private void RefreshFactions(FactionComponent component)
public IEnumerable<EntityUid> GetNearbyHostiles(EntityUid entity, float range, FactionComponent? component = null)
{
if (!Resolve(entity, ref component, false))
return Array.Empty<EntityUid>();
var hostiles = GetNearbyFactions(entity, range, component.HostileFactions);
if (TryComp<FactionExceptionComponent>(entity, out var factionException))
{
component.FriendlyFactions.Clear();
component.HostileFactions.Clear();
foreach (var faction in component.Factions)
{
// YAML Linter already yells about this
if (!_factions.TryGetValue(faction, out var factionData))
continue;
component.FriendlyFactions.UnionWith(factionData.Friendly);
component.HostileFactions.UnionWith(factionData.Hostile);
}
// ignore anything from enemy faction that we are explicitly friendly towards
return hostiles.Where(target => !_factionException.IsIgnored(factionException, target));
}
/// <summary>
/// Adds this entity to the particular faction.
/// </summary>
public void AddFaction(EntityUid uid, string faction, bool dirty = true)
return hostiles;
}
public IEnumerable<EntityUid> GetNearbyFriendlies(EntityUid entity, float range, FactionComponent? component = null)
{
if (!Resolve(entity, ref component, false))
return Array.Empty<EntityUid>();
return GetNearbyFactions(entity, range, component.FriendlyFactions);
}
private IEnumerable<EntityUid> GetNearbyFactions(EntityUid entity, float range, HashSet<string> factions)
{
var xformQuery = GetEntityQuery<TransformComponent>();
if (!xformQuery.TryGetComponent(entity, out var entityXform))
yield break;
foreach (var comp in _lookup.GetComponentsInRange<FactionComponent>(entityXform.MapPosition, range))
{
if (!_protoManager.HasIndex<FactionPrototype>(faction))
{
_sawmill.Error($"Unable to find faction {faction}");
return;
}
if (comp.Owner == entity)
continue;
var comp = EnsureComp<FactionComponent>(uid);
if (!comp.Factions.Add(faction))
return;
if (!factions.Overlaps(comp.Factions))
continue;
if (dirty)
{
RefreshFactions(comp);
}
}
/// <summary>
/// Removes this entity from the particular faction.
/// </summary>
public void RemoveFaction(EntityUid uid, string faction, bool dirty = true)
{
if (!_protoManager.HasIndex<FactionPrototype>(faction))
{
_sawmill.Error($"Unable to find faction {faction}");
return;
}
if (!TryComp<FactionComponent>(uid, out var component))
return;
if (!component.Factions.Remove(faction))
return;
if (dirty)
{
RefreshFactions(component);
}
}
public IEnumerable<EntityUid> GetNearbyHostiles(EntityUid entity, float range, FactionComponent? component = null)
{
if (!Resolve(entity, ref component, false))
return Array.Empty<EntityUid>();
return GetNearbyFactions(entity, range, component.HostileFactions);
}
public IEnumerable<EntityUid> GetNearbyFriendlies(EntityUid entity, float range, FactionComponent? component = null)
{
if (!Resolve(entity, ref component, false))
return Array.Empty<EntityUid>();
return GetNearbyFactions(entity, range, component.FriendlyFactions);
}
private IEnumerable<EntityUid> GetNearbyFactions(EntityUid entity, float range, HashSet<string> factions)
{
var xformQuery = GetEntityQuery<TransformComponent>();
if (!xformQuery.TryGetComponent(entity, out var entityXform))
yield break;
foreach (var comp in _lookup.GetComponentsInRange<FactionComponent>(entityXform.MapPosition, range))
{
if (comp.Owner == entity)
continue;
if (!factions.Overlaps(comp.Factions))
continue;
yield return comp.Owner;
}
}
public bool IsFriendly(EntityUid uidA, EntityUid uidB, FactionComponent? factionA = null, FactionComponent? factionB = null)
{
if (!Resolve(uidA, ref factionA, false) || !Resolve(uidB, ref factionB, false))
return false;
return factionA.Factions.Overlaps(factionB.Factions) || factionA.FriendlyFactions.Overlaps(factionB.Factions);
}
/// <summary>
/// Makes the source faction friendly to the target faction, 1-way.
/// </summary>
public void MakeFriendly(string source, string target)
{
if (!_factions.TryGetValue(source, out var sourceFaction))
{
_sawmill.Error($"Unable to find faction {source}");
return;
}
if (!_factions.ContainsKey(target))
{
_sawmill.Error($"Unable to find faction {target}");
return;
}
sourceFaction.Friendly.Add(target);
sourceFaction.Hostile.Remove(target);
RefreshFactions();
}
private void RefreshFactions()
{
_factions.Clear();
foreach (var faction in _protoManager.EnumeratePrototypes<FactionPrototype>())
{
_factions[faction.ID] = new FactionData()
{
Friendly = faction.Friendly.ToHashSet(),
Hostile = faction.Hostile.ToHashSet(),
};
}
foreach (var comp in EntityQuery<FactionComponent>(true))
{
comp.FriendlyFactions.Clear();
comp.HostileFactions.Clear();
RefreshFactions(comp);
}
}
/// <summary>
/// Makes the source faction hostile to the target faction, 1-way.
/// </summary>
public void MakeHostile(string source, string target)
{
if (!_factions.TryGetValue(source, out var sourceFaction))
{
_sawmill.Error($"Unable to find faction {source}");
return;
}
if (!_factions.ContainsKey(target))
{
_sawmill.Error($"Unable to find faction {target}");
return;
}
sourceFaction.Friendly.Remove(target);
sourceFaction.Hostile.Add(target);
RefreshFactions();
yield return comp.Owner;
}
}
public bool IsFriendly(EntityUid uidA, EntityUid uidB, FactionComponent? factionA = null, FactionComponent? factionB = null)
{
if (!Resolve(uidA, ref factionA, false) || !Resolve(uidB, ref factionB, false))
return false;
return factionA.Factions.Overlaps(factionB.Factions) || factionA.FriendlyFactions.Overlaps(factionB.Factions);
}
/// <summary>
/// Makes the source faction friendly to the target faction, 1-way.
/// </summary>
public void MakeFriendly(string source, string target)
{
if (!_factions.TryGetValue(source, out var sourceFaction))
{
_sawmill.Error($"Unable to find faction {source}");
return;
}
if (!_factions.ContainsKey(target))
{
_sawmill.Error($"Unable to find faction {target}");
return;
}
sourceFaction.Friendly.Add(target);
sourceFaction.Hostile.Remove(target);
RefreshFactions();
}
private void RefreshFactions()
{
_factions.Clear();
foreach (var faction in _protoManager.EnumeratePrototypes<FactionPrototype>())
{
_factions[faction.ID] = new FactionData()
{
Friendly = faction.Friendly.ToHashSet(),
Hostile = faction.Hostile.ToHashSet(),
};
}
foreach (var comp in EntityQuery<FactionComponent>(true))
{
comp.FriendlyFactions.Clear();
comp.HostileFactions.Clear();
RefreshFactions(comp);
}
}
/// <summary>
/// Makes the source faction hostile to the target faction, 1-way.
/// </summary>
public void MakeHostile(string source, string target)
{
if (!_factions.TryGetValue(source, out var sourceFaction))
{
_sawmill.Error($"Unable to find faction {source}");
return;
}
if (!_factions.ContainsKey(target))
{
_sawmill.Error($"Unable to find faction {target}");
return;
}
sourceFaction.Friendly.Remove(target);
sourceFaction.Hostile.Add(target);
RefreshFactions();
}
}