262 lines
10 KiB
C#
262 lines
10 KiB
C#
using Content.Shared.Actions;
|
|
using Content.Shared.Actions.Components;
|
|
using Content.Shared.Administration.Logs;
|
|
using Content.Shared.Alert;
|
|
using Content.Shared.Atmos.Components; // DeltaV
|
|
using Content.Shared.Body.Components;
|
|
using Content.Shared.Body.Systems;
|
|
using Content.Shared.Chemistry;
|
|
using Content.Shared.Chemistry.Components;
|
|
using Content.Shared.Chemistry.EntitySystems;
|
|
using Content.Shared.Cloning.Events;
|
|
using Content.Shared.Coordinates;
|
|
using Content.Shared.Database;
|
|
using Content.Shared.FixedPoint;
|
|
using Content.Shared.Fluids.Components;
|
|
using Content.Shared.Gravity;
|
|
using Content.Shared.Mobs;
|
|
using Content.Shared.Movement.Systems;
|
|
using Content.Shared.Slippery;
|
|
using Content.Shared.Toggleable;
|
|
using Content.Shared.Trigger.Components.Effects;
|
|
using Robust.Shared.Audio.Systems;
|
|
using Robust.Shared.Physics.Components;
|
|
using Robust.Shared.Physics.Events;
|
|
using Robust.Shared.Physics.Systems;
|
|
using Robust.Shared.Timing;
|
|
|
|
namespace Content.Shared.Rootable;
|
|
|
|
/// <summary>
|
|
/// Adds an action to toggle rooting to the ground, primarily for the Diona species.
|
|
/// Being rooted prevents weighlessness and slipping, but causes any floor contents to transfer its reagents to the bloodstream.
|
|
/// </summary>
|
|
public sealed class RootableSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly AlertsSystem _alerts = default!;
|
|
[Dependency] private readonly IGameTiming _timing = default!;
|
|
[Dependency] private readonly ISharedAdminLogManager _logger = default!;
|
|
[Dependency] private readonly MovementSpeedModifierSystem _movementSpeedModifier = default!;
|
|
[Dependency] private readonly ReactiveSystem _reactive = default!;
|
|
[Dependency] private readonly SharedActionsSystem _actions = default!;
|
|
[Dependency] private readonly SharedAudioSystem _audio = default!;
|
|
[Dependency] private readonly SharedBloodstreamSystem _blood = default!;
|
|
[Dependency] private readonly SharedGravitySystem _gravity = default!;
|
|
[Dependency] private readonly SharedPhysicsSystem _physics = default!;
|
|
[Dependency] private readonly SharedSolutionContainerSystem _solutionContainer = default!;
|
|
|
|
private EntityQuery<PuddleComponent> _puddleQuery;
|
|
private EntityQuery<PhysicsComponent> _physicsQuery;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
_puddleQuery = GetEntityQuery<PuddleComponent>();
|
|
_physicsQuery = GetEntityQuery<PhysicsComponent>();
|
|
|
|
SubscribeLocalEvent<RootableComponent, MapInitEvent>(OnRootableMapInit);
|
|
SubscribeLocalEvent<RootableComponent, ComponentShutdown>(OnRootableShutdown);
|
|
SubscribeLocalEvent<RootableComponent, StartCollideEvent>(OnStartCollide);
|
|
SubscribeLocalEvent<RootableComponent, EndCollideEvent>(OnEndCollide);
|
|
SubscribeLocalEvent<RootableComponent, ToggleActionEvent>(OnRootableToggle);
|
|
SubscribeLocalEvent<RootableComponent, MobStateChangedEvent>(OnMobStateChanged);
|
|
SubscribeLocalEvent<RootableComponent, IsWeightlessEvent>(OnIsWeightless);
|
|
SubscribeLocalEvent<RootableComponent, SlipAttemptEvent>(OnSlipAttempt);
|
|
SubscribeLocalEvent<RootableComponent, RefreshMovementSpeedModifiersEvent>(OnRefreshMovementSpeed);
|
|
SubscribeLocalEvent<RootableComponent, CloningEvent>(OnCloning);
|
|
}
|
|
|
|
public override void Update(float frameTime)
|
|
{
|
|
base.Update(frameTime);
|
|
|
|
var query = EntityQueryEnumerator<RootableComponent, BloodstreamComponent>();
|
|
var curTime = _timing.CurTime;
|
|
while (query.MoveNext(out var uid, out var rooted, out var bloodstream))
|
|
{
|
|
if (!rooted.Rooted || rooted.PuddleEntity == null || curTime < rooted.NextUpdate || !_puddleQuery.TryComp(rooted.PuddleEntity, out var puddleComp))
|
|
continue;
|
|
|
|
rooted.NextUpdate += rooted.TransferFrequency;
|
|
Dirty(uid, rooted);
|
|
PuddleReact((uid, rooted, bloodstream), (rooted.PuddleEntity.Value, puddleComp!));
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Determines if the puddle is set up properly and if so, moves on to reacting.
|
|
/// </summary>
|
|
private void PuddleReact(Entity<RootableComponent, BloodstreamComponent> ent, Entity<PuddleComponent> puddleEntity)
|
|
{
|
|
if (!_solutionContainer.ResolveSolution(puddleEntity.Owner, puddleEntity.Comp.SolutionName, ref puddleEntity.Comp.Solution, out var solution) ||
|
|
solution.Contents.Count == 0)
|
|
{
|
|
return;
|
|
}
|
|
|
|
ReactWithEntity(ent, puddleEntity, solution);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Attempt to transfer an amount of the solution to the ent's bloodstream.
|
|
/// </summary>
|
|
private void ReactWithEntity(Entity<RootableComponent, BloodstreamComponent> ent, Entity<PuddleComponent> puddleEntity, Solution solution)
|
|
{
|
|
if (!_solutionContainer.ResolveSolution(ent.Owner, ent.Comp2.BloodSolutionName, ref ent.Comp2.BloodSolution, out var bloodSolution) || bloodSolution.AvailableVolume <= 0)
|
|
return;
|
|
|
|
var availableTransfer = FixedPoint2.Min(solution.Volume, ent.Comp1.TransferRate);
|
|
var transferAmount = FixedPoint2.Min(availableTransfer, bloodSolution.AvailableVolume);
|
|
var transferSolution = _solutionContainer.SplitSolution(puddleEntity.Comp.Solution!.Value, transferAmount);
|
|
|
|
_reactive.DoEntityReaction(ent, transferSolution, ReactionMethod.Ingestion);
|
|
|
|
// Log solution addition by puddle.
|
|
if (_blood.TryAddToBloodstream((ent, ent.Comp2), transferSolution))
|
|
_logger.Add(LogType.ForceFeed, LogImpact.Medium, $"{ToPrettyString(ent):target} absorbed puddle {SharedSolutionContainerSystem.ToPrettyString(transferSolution)}");
|
|
}
|
|
|
|
private void OnCloning(Entity<RootableComponent> ent, ref CloningEvent args)
|
|
{
|
|
if (!args.Settings.EventComponents.Contains(Factory.GetRegistration(ent.Comp.GetType()).Name))
|
|
return;
|
|
|
|
var cloneComp = EnsureComp<RootableComponent>(args.CloneUid);
|
|
cloneComp.TransferRate = ent.Comp.TransferRate;
|
|
cloneComp.TransferFrequency = ent.Comp.TransferFrequency;
|
|
cloneComp.SpeedModifier = ent.Comp.SpeedModifier;
|
|
cloneComp.RootSound = ent.Comp.RootSound;
|
|
Dirty(args.CloneUid, cloneComp);
|
|
}
|
|
|
|
private void OnRootableMapInit(Entity<RootableComponent> ent, ref MapInitEvent args)
|
|
{
|
|
if (!TryComp(ent, out ActionsComponent? comp))
|
|
return;
|
|
|
|
ent.Comp.NextUpdate = _timing.CurTime;
|
|
Dirty(ent);
|
|
_actions.AddAction(ent, ref ent.Comp.ActionEntity, ent.Comp.Action, component: comp);
|
|
}
|
|
|
|
private void OnRootableShutdown(Entity<RootableComponent> ent, ref ComponentShutdown args)
|
|
{
|
|
if (!TryComp(ent, out ActionsComponent? comp))
|
|
return;
|
|
|
|
var actions = new Entity<ActionsComponent?>(ent, comp);
|
|
_actions.RemoveAction(actions, ent.Comp.ActionEntity);
|
|
_alerts.ClearAlert(ent.Owner, ent.Comp.RootedAlert);
|
|
}
|
|
|
|
private void OnRootableToggle(Entity<RootableComponent> ent, ref ToggleActionEvent args)
|
|
{
|
|
args.Handled = TryToggleRooting((ent, ent));
|
|
}
|
|
|
|
private void OnMobStateChanged(Entity<RootableComponent> ent, ref MobStateChangedEvent args)
|
|
{
|
|
if (ent.Comp.Rooted)
|
|
TryToggleRooting((ent, ent));
|
|
}
|
|
|
|
public bool TryToggleRooting(Entity<RootableComponent?> ent)
|
|
{
|
|
if (!Resolve(ent, ref ent.Comp))
|
|
return false;
|
|
|
|
ent.Comp.Rooted = !ent.Comp.Rooted;
|
|
if (TryComp<MovedByPressureComponent>(ent, out var moved)) // Delta-V - Prevent spacewind
|
|
moved.Enabled = !ent.Comp.Rooted;
|
|
_movementSpeedModifier.RefreshMovementSpeedModifiers(ent);
|
|
_gravity.RefreshWeightless(ent.Owner);
|
|
|
|
if (ent.Comp.Rooted)
|
|
{
|
|
_alerts.ShowAlert(ent.Owner, ent.Comp.RootedAlert);
|
|
var curTime = _timing.CurTime;
|
|
if (curTime > ent.Comp.NextUpdate)
|
|
ent.Comp.NextUpdate = curTime;
|
|
}
|
|
else
|
|
{
|
|
_alerts.ClearAlert(ent.Owner, ent.Comp.RootedAlert);
|
|
}
|
|
|
|
_audio.PlayPredicted(ent.Comp.RootSound, ent.Owner.ToCoordinates(), ent);
|
|
Dirty(ent);
|
|
|
|
return true;
|
|
}
|
|
|
|
private void OnIsWeightless(Entity<RootableComponent> ent, ref IsWeightlessEvent args)
|
|
{
|
|
if (args.Handled || !ent.Comp.Rooted)
|
|
return;
|
|
|
|
// Do not cancel weightlessness if the person is in off-grid.
|
|
if (!_gravity.EntityOnGravitySupportingGridOrMap(ent.Owner))
|
|
return;
|
|
|
|
args.IsWeightless = false;
|
|
args.Handled = true;
|
|
}
|
|
|
|
private void OnSlipAttempt(Entity<RootableComponent> ent, ref SlipAttemptEvent args)
|
|
{
|
|
if (!ent.Comp.Rooted)
|
|
return;
|
|
|
|
if (args.SlipCausingEntity != null && HasComp<DamageOnTriggerComponent>(args.SlipCausingEntity))
|
|
return;
|
|
|
|
args.NoSlip = true;
|
|
}
|
|
|
|
private void OnStartCollide(Entity<RootableComponent> ent, ref StartCollideEvent args)
|
|
{
|
|
if (!_puddleQuery.HasComp(args.OtherEntity))
|
|
return;
|
|
|
|
ent.Comp.PuddleEntity = args.OtherEntity;
|
|
|
|
if (ent.Comp.NextUpdate < _timing.CurTime) // To prevent constantly moving to new puddles resetting the timer.
|
|
ent.Comp.NextUpdate = _timing.CurTime;
|
|
|
|
Dirty(ent);
|
|
}
|
|
|
|
private void OnEndCollide(Entity<RootableComponent> ent, ref EndCollideEvent args)
|
|
{
|
|
if (ent.Comp.PuddleEntity != args.OtherEntity)
|
|
return;
|
|
|
|
var exists = Exists(args.OtherEntity);
|
|
|
|
if (!_physicsQuery.TryComp(ent, out var body))
|
|
return;
|
|
|
|
foreach (var entContact in _physics.GetContactingEntities(ent, body))
|
|
{
|
|
if (exists && entContact == args.OtherEntity)
|
|
continue;
|
|
|
|
if (!_puddleQuery.HasComponent(entContact))
|
|
continue;
|
|
|
|
ent.Comp.PuddleEntity = ent;
|
|
return; // New puddle found, no need to continue.
|
|
}
|
|
|
|
ent.Comp.PuddleEntity = null;
|
|
Dirty(ent);
|
|
}
|
|
|
|
private void OnRefreshMovementSpeed(Entity<RootableComponent> ent, ref RefreshMovementSpeedModifiersEvent args)
|
|
{
|
|
if (ent.Comp.Rooted)
|
|
args.ModifySpeed(ent.Comp.SpeedModifier);
|
|
}
|
|
}
|