mirror of
https://github.com/DeltaV-Station/Delta-v.git
synced 2026-10-01 16:47:26 +00:00
* Reapply "Nubody Merge" (#5807) This reverts commitd62f7cef10. * Revert "Nubody Merge (#5670)" This reverts commitb465cfb3fd. * Revert "IPC fibers (#4968)" This reverts commitc2b42211f0. * Revert "IPC Polymorph fix (#4545)" This reverts commitfa03d8c655. * Revert "fix IPC not having humanoid emotes (#4445)" This reverts commit216fdf999f. * Revert "ipc heat/cold (#4306)" This reverts commitfafd5cf816. * Revert "ipc cables (#4283)" This reverts commit54958039a1. * Revert "feat: ipcs now have dynamic power draw based off movement speed (#3917)" This reverts commit71e8b7e64b. * Revert "fix ipcs not being able to disarm (#3374)" This reverts commit764c4dcf54. * Revert "make steel heal caustic for ipcs (#3206)" This reverts commit450ac82550. * Revert "add missing ipc survival boxes (#3065)" This reverts commit888c9cdb00. * Revert "make borgs use ipc repairing (#3064)" This reverts commitf77a49dd0b. * Revert "Update hud.yml to add ipcs to diagnostic hud (#3019)" This reverts commit7220643531. * Revert "Tweak IPCs (#2979)" This reverts commit98b47476de. * Revert "fix nukies ipc test fail (#2957)" This reverts commit4a1acbf863. * Revert "fix ipc nukies not getting comms (#2884)" This reverts commit8a2254770e. * Revert "fix ipc encryption key mispredict (#2879)" This reverts commitf08726efc1. * Revert "Merge pull request #2800 from Spielern/port-ipc" This reverts commit822fa4312c, reversing changes made tofe4c23c702. * Remove and Fix rest of IPC Hangons causing crashes * Integration test Errors * Expand migration file with removed IPC stuff * IPC Leg was found and migrated to naught * Revert "Merge pull request #2542 from deltanedas/shitmed-ops" This reverts commit83a8002922, reversing changes made tobaa89b0052. * Revert "Set up surgical augments infrastructure & add tool panel augments (#3059)" This reverts commita40f0be6fd. * Revert "Psionic Prosthetics (#4689)" This reverts commit2679bb238a. * Revert "Add surgical cleanliness and sanitation (#2847)" This reverts commit64565ba4f7. * Remove unremoved mentions * Revert "Add autoclave to easily sanitize large amounts of items and faster (#2933)" This reverts commitd0da90fa2e. * Shit Med Removed * Remove _ShitMed Folders and fix Errors in Console * Add OrganAnimalBrain into the mirgraiton file * Heads no Input Mover * Monke * Full migration file I guess * oops * shiv state fix * Restore accidental deleted recipe * Forgot one flt mention * Fix more issues * Comment out Damage Visual for motorkind for now * Motorkind... * Commenting stuff out for now * thought there'd be more of this to gut --------- Co-authored-by: Janet Blackquill <uhhadd@gmail.com>
811 lines
26 KiB
C#
811 lines
26 KiB
C#
using System.Diagnostics.CodeAnalysis;
|
|
using System.Linq;
|
|
using Content.Shared.Body.Components;
|
|
using Content.Shared.Body.Events;
|
|
using Content.Shared.Body.Organ;
|
|
using Content.Shared.Body.Part;
|
|
using Content.Shared.Damage;
|
|
using Content.Shared.Damage.Prototypes;
|
|
using Content.Shared.Movement.Components;
|
|
using Content.Shared.Standing;
|
|
using Robust.Shared.Containers;
|
|
using Robust.Shared.Prototypes;
|
|
using Robust.Shared.Utility;
|
|
|
|
namespace Content.Shared.Body.Systems;
|
|
|
|
public partial class SharedBodySystem
|
|
{
|
|
private static readonly ProtoId<DamageTypePrototype> BloodlossDamageType = "Bloodloss";
|
|
private void InitializeParts()
|
|
{
|
|
// TODO: This doesn't handle comp removal on child ents.
|
|
|
|
// If you modify this also see the Body partial for root parts.
|
|
SubscribeLocalEvent<BodyPartComponent, EntInsertedIntoContainerMessage>(OnBodyPartInserted);
|
|
SubscribeLocalEvent<BodyPartComponent, EntRemovedFromContainerMessage>(OnBodyPartRemoved);
|
|
}
|
|
|
|
private void OnBodyPartInserted(Entity<BodyPartComponent> ent, ref EntInsertedIntoContainerMessage args)
|
|
{
|
|
// Body part inserted into another body part.
|
|
var insertedUid = args.Entity;
|
|
var slotId = args.Container.ID;
|
|
|
|
if (ent.Comp.Body is null)
|
|
return;
|
|
|
|
if (TryComp(insertedUid, out BodyPartComponent? part))
|
|
{
|
|
AddPart(ent.Comp.Body.Value, (insertedUid, part), slotId);
|
|
RecursiveBodyUpdate((insertedUid, part), ent.Comp.Body.Value);
|
|
}
|
|
|
|
if (TryComp(insertedUid, out OrganComponent? organ))
|
|
AddOrgan((insertedUid, organ), ent.Comp.Body.Value, ent);
|
|
}
|
|
|
|
private void OnBodyPartRemoved(Entity<BodyPartComponent> ent, ref EntRemovedFromContainerMessage args)
|
|
{
|
|
// Body part removed from another body part.
|
|
var removedUid = args.Entity;
|
|
var slotId = args.Container.ID;
|
|
|
|
DebugTools.Assert(!TryComp(removedUid, out BodyPartComponent? b) || b.Body == ent.Comp.Body);
|
|
DebugTools.Assert(!TryComp(removedUid, out OrganComponent? o) || o.Body == ent.Comp.Body);
|
|
|
|
if (TryComp(removedUid, out BodyPartComponent? part) && part.Body is not null)
|
|
{
|
|
RemovePart(part.Body.Value, (removedUid, part), slotId);
|
|
RecursiveBodyUpdate((removedUid, part), null);
|
|
}
|
|
|
|
if (TryComp(removedUid, out OrganComponent? organ))
|
|
RemoveOrgan((removedUid, organ), ent);
|
|
}
|
|
|
|
private void RecursiveBodyUpdate(Entity<BodyPartComponent> ent, EntityUid? bodyUid)
|
|
{
|
|
ent.Comp.Body = bodyUid;
|
|
Dirty(ent, ent.Comp);
|
|
|
|
foreach (var slotId in ent.Comp.Organs.Keys)
|
|
{
|
|
if (!Containers.TryGetContainer(ent, GetOrganContainerId(slotId), out var container))
|
|
continue;
|
|
|
|
foreach (var organ in container.ContainedEntities)
|
|
{
|
|
if (!TryComp(organ, out OrganComponent? organComp))
|
|
continue;
|
|
|
|
Dirty(organ, organComp);
|
|
|
|
if (organComp.Body is { Valid: true } oldBodyUid)
|
|
{
|
|
var removedEv = new OrganRemovedFromBodyEvent(oldBodyUid, ent);
|
|
RaiseLocalEvent(organ, ref removedEv);
|
|
}
|
|
|
|
organComp.Body = bodyUid;
|
|
if (bodyUid is not null)
|
|
{
|
|
var addedEv = new OrganAddedToBodyEvent(bodyUid.Value, ent);
|
|
RaiseLocalEvent(organ, ref addedEv);
|
|
}
|
|
}
|
|
}
|
|
|
|
foreach (var slotId in ent.Comp.Children.Keys)
|
|
{
|
|
if (!Containers.TryGetContainer(ent, GetPartSlotContainerId(slotId), out var container))
|
|
continue;
|
|
|
|
foreach (var containedUid in container.ContainedEntities)
|
|
{
|
|
if (TryComp(containedUid, out BodyPartComponent? childPart))
|
|
RecursiveBodyUpdate((containedUid, childPart), bodyUid);
|
|
}
|
|
}
|
|
}
|
|
|
|
protected virtual void AddPart(
|
|
Entity<BodyComponent?> bodyEnt,
|
|
Entity<BodyPartComponent> partEnt,
|
|
string slotId)
|
|
{
|
|
Dirty(partEnt, partEnt.Comp);
|
|
partEnt.Comp.Body = bodyEnt;
|
|
|
|
var ev = new BodyPartAddedEvent(slotId, partEnt);
|
|
RaiseLocalEvent(bodyEnt, ref ev);
|
|
|
|
AddLeg(partEnt, bodyEnt);
|
|
}
|
|
|
|
protected virtual void RemovePart(
|
|
Entity<BodyComponent?> bodyEnt,
|
|
Entity<BodyPartComponent> partEnt,
|
|
string slotId)
|
|
{
|
|
Resolve(bodyEnt, ref bodyEnt.Comp, logMissing: false);
|
|
Dirty(partEnt, partEnt.Comp);
|
|
partEnt.Comp.Body = null;
|
|
|
|
var ev = new BodyPartRemovedEvent(slotId, partEnt);
|
|
RaiseLocalEvent(bodyEnt, ref ev);
|
|
|
|
RemoveLeg(partEnt, bodyEnt);
|
|
PartRemoveDamage(bodyEnt, partEnt);
|
|
}
|
|
|
|
private void AddLeg(Entity<BodyPartComponent> legEnt, Entity<BodyComponent?> bodyEnt)
|
|
{
|
|
if (!Resolve(bodyEnt, ref bodyEnt.Comp, logMissing: false))
|
|
return;
|
|
|
|
if (legEnt.Comp.PartType == BodyPartType.Leg)
|
|
{
|
|
bodyEnt.Comp.LegEntities.Add(legEnt);
|
|
UpdateMovementSpeed(bodyEnt);
|
|
Dirty(bodyEnt, bodyEnt.Comp);
|
|
}
|
|
}
|
|
|
|
private void RemoveLeg(Entity<BodyPartComponent> legEnt, Entity<BodyComponent?> bodyEnt)
|
|
{
|
|
if (!Resolve(bodyEnt, ref bodyEnt.Comp, logMissing: false))
|
|
return;
|
|
|
|
if (legEnt.Comp.PartType != BodyPartType.Leg)
|
|
return;
|
|
|
|
bodyEnt.Comp.LegEntities.Remove(legEnt);
|
|
UpdateMovementSpeed(bodyEnt);
|
|
Dirty(bodyEnt, bodyEnt.Comp);
|
|
|
|
if (bodyEnt.Comp.LegEntities.Count != 0)
|
|
return;
|
|
|
|
if (!TryComp<StandingStateComponent>(bodyEnt, out var standingState)
|
|
|| !standingState.Standing
|
|
|| !Standing.Down(bodyEnt, standingState: standingState))
|
|
return;
|
|
|
|
var ev = new DropHandItemsEvent();
|
|
RaiseLocalEvent(bodyEnt, ref ev);
|
|
}
|
|
|
|
private void PartRemoveDamage(Entity<BodyComponent?> bodyEnt, Entity<BodyPartComponent> partEnt)
|
|
{
|
|
if (!Resolve(bodyEnt, ref bodyEnt.Comp, logMissing: false))
|
|
return;
|
|
|
|
if (!_timing.ApplyingState
|
|
&& partEnt.Comp.IsVital
|
|
&& !GetBodyChildrenOfType(bodyEnt, partEnt.Comp.PartType, bodyEnt.Comp).Any()
|
|
)
|
|
{
|
|
// TODO BODY SYSTEM KILL : remove this when wounding and required parts are implemented properly
|
|
var damage = new DamageSpecifier(Prototypes.Index(BloodlossDamageType), 300);
|
|
Damageable.ChangeDamage(bodyEnt.Owner, damage);
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Tries to get the parent body part to this if applicable.
|
|
/// Doesn't validate if it's a part of body system.
|
|
/// </summary>
|
|
public EntityUid? GetParentPartOrNull(EntityUid uid)
|
|
{
|
|
if (!Containers.TryGetContainingContainer((uid, null, null), out var container))
|
|
return null;
|
|
|
|
var parent = container.Owner;
|
|
|
|
if (!HasComp<BodyPartComponent>(parent))
|
|
return null;
|
|
|
|
return parent;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Tries to get the parent body part and slot to this if applicable.
|
|
/// </summary>
|
|
public (EntityUid Parent, string Slot)? GetParentPartAndSlotOrNull(EntityUid uid)
|
|
{
|
|
if (!Containers.TryGetContainingContainer((uid, null, null), out var container))
|
|
return null;
|
|
|
|
var slotId = GetPartSlotContainerIdFromContainer(container.ID);
|
|
|
|
if (string.IsNullOrEmpty(slotId))
|
|
return null;
|
|
|
|
var parent = container.Owner;
|
|
|
|
if (!TryComp<BodyPartComponent>(parent, out var parentBody)
|
|
|| !parentBody.Children.ContainsKey(slotId))
|
|
return null;
|
|
|
|
return (parent, slotId);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Tries to get the relevant parent body part to this if it exists.
|
|
/// It won't exist if this is the root body part or if it's not in a body.
|
|
/// </summary>
|
|
public bool TryGetParentBodyPart(
|
|
EntityUid partUid,
|
|
[NotNullWhen(true)] out EntityUid? parentUid,
|
|
[NotNullWhen(true)] out BodyPartComponent? parentComponent)
|
|
{
|
|
DebugTools.Assert(HasComp<BodyPartComponent>(partUid));
|
|
parentUid = null;
|
|
parentComponent = null;
|
|
|
|
if (Containers.TryGetContainingContainer((partUid, null, null), out var container) &&
|
|
TryComp(container.Owner, out parentComponent))
|
|
{
|
|
parentUid = container.Owner;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
#region Slots
|
|
|
|
/// <summary>
|
|
/// Creates a BodyPartSlot on the specified partUid.
|
|
/// </summary>
|
|
private BodyPartSlot? CreatePartSlot(
|
|
EntityUid partUid,
|
|
string slotId,
|
|
BodyPartType partType,
|
|
BodyPartComponent? part = null)
|
|
{
|
|
if (!Resolve(partUid, ref part, logMissing: false))
|
|
return null;
|
|
|
|
Containers.EnsureContainer<ContainerSlot>(partUid, GetPartSlotContainerId(slotId));
|
|
var partSlot = new BodyPartSlot(slotId, partType);
|
|
try
|
|
{
|
|
part.Children.Add(slotId, partSlot);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
throw new Exception($"{ToPrettyString(partUid)}", e);
|
|
}
|
|
Dirty(partUid, part);
|
|
return partSlot;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Tries to create a BodyPartSlot on the specified partUid.
|
|
/// </summary>
|
|
/// <returns>false if not relevant or can't add it.</returns>
|
|
public bool TryCreatePartSlot(
|
|
EntityUid? partId,
|
|
string slotId,
|
|
BodyPartType partType,
|
|
[NotNullWhen(true)] out BodyPartSlot? slot,
|
|
BodyPartComponent? part = null)
|
|
{
|
|
slot = null;
|
|
|
|
if (partId is null
|
|
|| !Resolve(partId.Value, ref part, logMissing: false))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
Containers.EnsureContainer<ContainerSlot>(partId.Value, GetPartSlotContainerId(slotId));
|
|
slot = new BodyPartSlot(slotId, partType);
|
|
|
|
if (!part.Children.TryAdd(slotId, slot.Value))
|
|
return false;
|
|
|
|
Dirty(partId.Value, part);
|
|
return true;
|
|
}
|
|
|
|
public bool TryCreatePartSlotAndAttach(
|
|
EntityUid parentId,
|
|
string slotId,
|
|
EntityUid childId,
|
|
BodyPartType partType,
|
|
BodyPartComponent? parent = null,
|
|
BodyPartComponent? child = null)
|
|
{
|
|
return TryCreatePartSlot(parentId, slotId, partType, out _, parent)
|
|
&& AttachPart(parentId, slotId, childId, parent, child);
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region RootPartManagement
|
|
|
|
/// <summary>
|
|
/// Returns true if the partId is the root body container for the specified bodyId.
|
|
/// </summary>
|
|
public bool IsPartRoot(
|
|
EntityUid bodyId,
|
|
EntityUid partId,
|
|
BodyComponent? body = null,
|
|
BodyPartComponent? part = null)
|
|
{
|
|
return Resolve(partId, ref part)
|
|
&& Resolve(bodyId, ref body)
|
|
&& Containers.TryGetContainingContainer(bodyId, partId, out var container)
|
|
&& container.ID == BodyRootContainerId;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns true if we can attach the partId to the bodyId as the root entity.
|
|
/// </summary>
|
|
public bool CanAttachToRoot(
|
|
EntityUid bodyId,
|
|
EntityUid partId,
|
|
BodyComponent? body = null,
|
|
BodyPartComponent? part = null)
|
|
{
|
|
return Resolve(bodyId, ref body)
|
|
&& Resolve(partId, ref part)
|
|
&& body.RootContainer.ContainedEntity is null
|
|
&& bodyId != part.Body;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns the root part of this body if it exists.
|
|
/// </summary>
|
|
public (EntityUid Entity, BodyPartComponent BodyPart)? GetRootPartOrNull(EntityUid bodyId, BodyComponent? body = null)
|
|
{
|
|
if (!Resolve(bodyId, ref body)
|
|
|| body.RootContainer.ContainedEntity is null)
|
|
{
|
|
return null;
|
|
}
|
|
|
|
return (body.RootContainer.ContainedEntity.Value,
|
|
Comp<BodyPartComponent>(body.RootContainer.ContainedEntity.Value));
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns true if the partId can be attached to the parentId in the specified slot.
|
|
/// </summary>
|
|
public bool CanAttachPart(
|
|
EntityUid parentId,
|
|
BodyPartSlot slot,
|
|
EntityUid partId,
|
|
BodyPartComponent? parentPart = null,
|
|
BodyPartComponent? part = null)
|
|
{
|
|
return Resolve(partId, ref part, logMissing: false)
|
|
&& Resolve(parentId, ref parentPart, logMissing: false)
|
|
&& CanAttachPart(parentId, slot.Id, partId, parentPart, part);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns true if we can attach the specified partId to the parentId in the specified slot.
|
|
/// </summary>
|
|
public bool CanAttachPart(
|
|
EntityUid parentId,
|
|
string slotId,
|
|
EntityUid partId,
|
|
BodyPartComponent? parentPart = null,
|
|
BodyPartComponent? part = null)
|
|
{
|
|
return Resolve(partId, ref part, logMissing: false)
|
|
&& Resolve(parentId, ref parentPart, logMissing: false)
|
|
&& parentPart.Children.TryGetValue(slotId, out var parentSlotData)
|
|
&& part.PartType == parentSlotData.Type
|
|
&& Containers.TryGetContainer(parentId, GetPartSlotContainerId(slotId), out var container)
|
|
&& Containers.CanInsert(partId, container);
|
|
}
|
|
|
|
public bool AttachPartToRoot(
|
|
EntityUid bodyId,
|
|
EntityUid partId,
|
|
BodyComponent? body = null,
|
|
BodyPartComponent? part = null)
|
|
{
|
|
return Resolve(bodyId, ref body)
|
|
&& Resolve(partId, ref part)
|
|
&& CanAttachToRoot(bodyId, partId, body, part)
|
|
&& Containers.Insert(partId, body.RootContainer);
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region Attach/Detach
|
|
|
|
/// <summary>
|
|
/// Attaches a body part to the specified body part parent.
|
|
/// </summary>
|
|
public bool AttachPart(
|
|
EntityUid parentPartId,
|
|
string slotId,
|
|
EntityUid partId,
|
|
BodyPartComponent? parentPart = null,
|
|
BodyPartComponent? part = null)
|
|
{
|
|
return Resolve(parentPartId, ref parentPart, logMissing: false)
|
|
&& parentPart.Children.TryGetValue(slotId, out var slot)
|
|
&& AttachPart(parentPartId, slot, partId, parentPart, part);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Attaches a body part to the specified body part parent.
|
|
/// </summary>
|
|
public bool AttachPart(
|
|
EntityUid parentPartId,
|
|
BodyPartSlot slot,
|
|
EntityUid partId,
|
|
BodyPartComponent? parentPart = null,
|
|
BodyPartComponent? part = null)
|
|
{
|
|
if (!Resolve(parentPartId, ref parentPart, logMissing: false)
|
|
|| !Resolve(partId, ref part, logMissing: false)
|
|
|| !CanAttachPart(parentPartId, slot.Id, partId, parentPart, part)
|
|
|| !parentPart.Children.ContainsKey(slot.Id))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (!Containers.TryGetContainer(parentPartId, GetPartSlotContainerId(slot.Id), out var container))
|
|
{
|
|
DebugTools.Assert($"Unable to find body slot {slot.Id} for {ToPrettyString(parentPartId)}");
|
|
return false;
|
|
}
|
|
|
|
return Containers.Insert(partId, container);
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region Misc
|
|
|
|
public void UpdateMovementSpeed(
|
|
EntityUid bodyId,
|
|
BodyComponent? body = null,
|
|
MovementSpeedModifierComponent? movement = null)
|
|
{
|
|
if (!Resolve(bodyId, ref body, ref movement, logMissing: false)
|
|
|| body.RequiredLegs <= 0)
|
|
{
|
|
return;
|
|
}
|
|
|
|
var walkSpeed = 0f;
|
|
var sprintSpeed = 0f;
|
|
var acceleration = 0f;
|
|
foreach (var legEntity in body.LegEntities)
|
|
{
|
|
if (!TryComp<MovementBodyPartComponent>(legEntity, out var legModifier))
|
|
continue;
|
|
|
|
walkSpeed += legModifier.WalkSpeed;
|
|
sprintSpeed += legModifier.SprintSpeed;
|
|
acceleration += legModifier.Acceleration;
|
|
}
|
|
walkSpeed /= body.RequiredLegs;
|
|
sprintSpeed /= body.RequiredLegs;
|
|
acceleration /= body.RequiredLegs;
|
|
Movement.ChangeBaseSpeed(bodyId, walkSpeed, sprintSpeed, acceleration, movement);
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region Queries
|
|
|
|
/// <summary>
|
|
/// Get all organs for the specified body part.
|
|
/// </summary>
|
|
public IEnumerable<(EntityUid Id, OrganComponent Component)> GetPartOrgans(EntityUid partId, BodyPartComponent? part = null)
|
|
{
|
|
if (!Resolve(partId, ref part, logMissing: false))
|
|
yield break;
|
|
|
|
foreach (var slotId in part.Organs.Keys)
|
|
{
|
|
var containerSlotId = GetOrganContainerId(slotId);
|
|
|
|
if (!Containers.TryGetContainer(partId, containerSlotId, out var container))
|
|
continue;
|
|
|
|
foreach (var containedEnt in container.ContainedEntities)
|
|
{
|
|
if (!TryComp(containedEnt, out OrganComponent? organ))
|
|
continue;
|
|
|
|
yield return (containedEnt, organ);
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets all BaseContainers for body parts on this entity and its child entities.
|
|
/// </summary>
|
|
public IEnumerable<BaseContainer> GetPartContainers(EntityUid id, BodyPartComponent? part = null)
|
|
{
|
|
if (!Resolve(id, ref part, logMissing: false) ||
|
|
part.Children.Count == 0)
|
|
{
|
|
yield break;
|
|
}
|
|
|
|
foreach (var slotId in part.Children.Keys)
|
|
{
|
|
var containerSlotId = GetPartSlotContainerId(slotId);
|
|
|
|
if (!Containers.TryGetContainer(id, containerSlotId, out var container))
|
|
continue;
|
|
|
|
yield return container;
|
|
|
|
foreach (var ent in container.ContainedEntities)
|
|
{
|
|
foreach (var childContainer in GetPartContainers(ent))
|
|
{
|
|
yield return childContainer;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns all body part components for this entity including itself.
|
|
/// </summary>
|
|
public IEnumerable<(EntityUid Id, BodyPartComponent Component)> GetBodyPartChildren(
|
|
EntityUid partId,
|
|
BodyPartComponent? part = null)
|
|
{
|
|
if (!Resolve(partId, ref part, logMissing: false))
|
|
yield break;
|
|
|
|
yield return (partId, part);
|
|
|
|
foreach (var slotId in part.Children.Keys)
|
|
{
|
|
var containerSlotId = GetPartSlotContainerId(slotId);
|
|
|
|
if (Containers.TryGetContainer(partId, containerSlotId, out var container))
|
|
{
|
|
foreach (var containedEnt in container.ContainedEntities)
|
|
{
|
|
if (!TryComp(containedEnt, out BodyPartComponent? childPart))
|
|
continue;
|
|
|
|
foreach (var value in GetBodyPartChildren(containedEnt, childPart))
|
|
{
|
|
yield return value;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns all body part slots for this entity.
|
|
/// </summary>
|
|
public IEnumerable<BodyPartSlot> GetAllBodyPartSlots(
|
|
EntityUid partId,
|
|
BodyPartComponent? part = null)
|
|
{
|
|
if (!Resolve(partId, ref part, logMissing: false))
|
|
yield break;
|
|
|
|
foreach (var (slotId, slot) in part.Children)
|
|
{
|
|
yield return slot;
|
|
|
|
var containerSlotId = GetOrganContainerId(slotId);
|
|
|
|
if (Containers.TryGetContainer(partId, containerSlotId, out var container))
|
|
{
|
|
foreach (var containedEnt in container.ContainedEntities)
|
|
{
|
|
if (!TryComp(containedEnt, out BodyPartComponent? childPart))
|
|
continue;
|
|
|
|
foreach (var subSlot in GetAllBodyPartSlots(containedEnt, childPart))
|
|
{
|
|
yield return subSlot;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns true if the bodyId has any parts of this type.
|
|
/// </summary>
|
|
public bool BodyHasPartType(
|
|
EntityUid bodyId,
|
|
BodyPartType type,
|
|
BodyComponent? body = null)
|
|
{
|
|
return GetBodyChildrenOfType(bodyId, type, body).Any();
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns true if the parentId has the specified childId.
|
|
/// </summary>
|
|
public bool PartHasChild(
|
|
EntityUid parentId,
|
|
EntityUid childId,
|
|
BodyPartComponent? parent,
|
|
BodyPartComponent? child)
|
|
{
|
|
if (!Resolve(parentId, ref parent, logMissing: false)
|
|
|| !Resolve(childId, ref child, logMissing: false))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
foreach (var (foundId, _) in GetBodyPartChildren(parentId, parent))
|
|
{
|
|
if (foundId == childId)
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns true if the bodyId has the specified partId.
|
|
/// </summary>
|
|
public bool BodyHasChild(
|
|
EntityUid bodyId,
|
|
EntityUid partId,
|
|
BodyComponent? body = null,
|
|
BodyPartComponent? part = null)
|
|
{
|
|
return Resolve(bodyId, ref body, logMissing: false)
|
|
&& body.RootContainer.ContainedEntity is not null
|
|
&& Resolve(partId, ref part, logMissing: false)
|
|
&& TryComp(body.RootContainer.ContainedEntity, out BodyPartComponent? rootPart)
|
|
&& PartHasChild(body.RootContainer.ContainedEntity.Value, partId, rootPart, part);
|
|
}
|
|
|
|
public IEnumerable<(EntityUid Id, BodyPartComponent Component)> GetBodyChildrenOfType(
|
|
EntityUid bodyId,
|
|
BodyPartType type,
|
|
BodyComponent? body = null)
|
|
{
|
|
foreach (var part in GetBodyChildren(bodyId, body))
|
|
{
|
|
if (part.Component.PartType == type)
|
|
yield return part;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns a list of ValueTuples of <see cref="T"/> and OrganComponent on each organ
|
|
/// in the given part.
|
|
/// </summary>
|
|
/// <param name="uid">The part entity id to check on.</param>
|
|
/// <param name="part">The part to check for organs on.</param>
|
|
/// <typeparam name="T">The component to check for.</typeparam>
|
|
public List<(T Comp, OrganComponent Organ)> GetBodyPartOrganComponents<T>(
|
|
EntityUid uid,
|
|
BodyPartComponent? part = null)
|
|
where T : IComponent
|
|
{
|
|
if (!Resolve(uid, ref part))
|
|
return new List<(T Comp, OrganComponent Organ)>();
|
|
|
|
var query = GetEntityQuery<T>();
|
|
var list = new List<(T Comp, OrganComponent Organ)>();
|
|
|
|
foreach (var organ in GetPartOrgans(uid, part))
|
|
{
|
|
if (query.TryGetComponent(organ.Id, out var comp))
|
|
list.Add((comp, organ.Component));
|
|
}
|
|
|
|
return list;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Tries to get a list of ValueTuples of <see cref="T"/> and OrganComponent on each organs
|
|
/// in the given part.
|
|
/// </summary>
|
|
/// <param name="uid">The part entity id to check on.</param>
|
|
/// <param name="comps">The list of components.</param>
|
|
/// <param name="part">The part to check for organs on.</param>
|
|
/// <typeparam name="T">The component to check for.</typeparam>
|
|
/// <returns>Whether any were found.</returns>
|
|
public bool TryGetBodyPartOrganComponents<T>(
|
|
EntityUid uid,
|
|
[NotNullWhen(true)] out List<(T Comp, OrganComponent Organ)>? comps,
|
|
BodyPartComponent? part = null)
|
|
where T : IComponent
|
|
{
|
|
if (!Resolve(uid, ref part))
|
|
{
|
|
comps = null;
|
|
return false;
|
|
}
|
|
|
|
comps = GetBodyPartOrganComponents<T>(uid, part);
|
|
|
|
if (comps.Count != 0)
|
|
return true;
|
|
|
|
comps = null;
|
|
return false;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the parent body part and all immediate child body parts for the partId.
|
|
/// </summary>
|
|
public IEnumerable<EntityUid> GetBodyPartAdjacentParts(
|
|
EntityUid partId,
|
|
BodyPartComponent? part = null)
|
|
{
|
|
if (!Resolve(partId, ref part, logMissing: false))
|
|
yield break;
|
|
|
|
if (TryGetParentBodyPart(partId, out var parentUid, out _))
|
|
yield return parentUid.Value;
|
|
|
|
foreach (var slotId in part.Children.Keys)
|
|
{
|
|
var container = Containers.GetContainer(partId, GetPartSlotContainerId(slotId));
|
|
|
|
foreach (var containedEnt in container.ContainedEntities)
|
|
{
|
|
yield return containedEnt;
|
|
}
|
|
}
|
|
}
|
|
|
|
public IEnumerable<(EntityUid AdjacentId, T Component)> GetBodyPartAdjacentPartsComponents<T>(
|
|
EntityUid partId,
|
|
BodyPartComponent? part = null)
|
|
where T : IComponent
|
|
{
|
|
if (!Resolve(partId, ref part, logMissing: false))
|
|
yield break;
|
|
|
|
var query = GetEntityQuery<T>();
|
|
foreach (var adjacentId in GetBodyPartAdjacentParts(partId, part))
|
|
{
|
|
if (query.TryGetComponent(adjacentId, out var component))
|
|
yield return (adjacentId, component);
|
|
}
|
|
}
|
|
|
|
public bool TryGetBodyPartAdjacentPartsComponents<T>(
|
|
EntityUid partId,
|
|
[NotNullWhen(true)] out List<(EntityUid AdjacentId, T Component)>? comps,
|
|
BodyPartComponent? part = null)
|
|
where T : IComponent
|
|
{
|
|
if (!Resolve(partId, ref part, logMissing: false))
|
|
{
|
|
comps = null;
|
|
return false;
|
|
}
|
|
|
|
var query = GetEntityQuery<T>();
|
|
comps = new List<(EntityUid AdjacentId, T Component)>();
|
|
|
|
foreach (var adjacentId in GetBodyPartAdjacentParts(partId, part))
|
|
{
|
|
if (query.TryGetComponent(adjacentId, out var component))
|
|
comps.Add((adjacentId, component));
|
|
}
|
|
|
|
if (comps.Count != 0)
|
|
return true;
|
|
|
|
comps = null;
|
|
return false;
|
|
}
|
|
|
|
#endregion
|
|
}
|