Replace old Overflow algorithm (#6280)
Co-authored-by: metalgearsloth <31366439+metalgearsloth@users.noreply.github.com>
This commit is contained in:
@@ -2,6 +2,7 @@ using System;
|
|||||||
using System.Collections.Generic;
|
using System.Collections.Generic;
|
||||||
using System.Diagnostics.CodeAnalysis;
|
using System.Diagnostics.CodeAnalysis;
|
||||||
using Content.Server.Chemistry.Components.SolutionManager;
|
using Content.Server.Chemistry.Components.SolutionManager;
|
||||||
|
using Content.Server.Fluids.Components;
|
||||||
using Content.Shared.Chemistry;
|
using Content.Shared.Chemistry;
|
||||||
using Content.Shared.Chemistry.Components;
|
using Content.Shared.Chemistry.Components;
|
||||||
using Content.Shared.Chemistry.Reaction;
|
using Content.Shared.Chemistry.Reaction;
|
||||||
@@ -213,17 +214,46 @@ namespace Content.Server.Chemistry.EntitySystems
|
|||||||
/// <summary>
|
/// <summary>
|
||||||
/// Adds a solution to the container, if it can fully fit.
|
/// Adds a solution to the container, if it can fully fit.
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <param name="targetUid"></param>
|
/// <param name="targetUid">entity holding targetSolution</param>
|
||||||
/// <param name="targetSolution">The container to which we try to add.</param>
|
/// <param name="targetSolution">entity holding targetSolution</param>
|
||||||
/// <param name="solution">The solution to try to add.</param>
|
/// <param name="addedSolution">solution being added</param>
|
||||||
/// <returns>If the solution could be added.</returns>
|
/// <returns>If the solution could be added.</returns>
|
||||||
public bool TryAddSolution(EntityUid targetUid, Solution? targetSolution, Solution solution)
|
public bool TryAddSolution(EntityUid targetUid, Solution? targetSolution, Solution addedSolution)
|
||||||
{
|
{
|
||||||
if (targetSolution == null || !targetSolution.CanAddSolution(solution) || solution.TotalVolume == 0)
|
if (targetSolution == null
|
||||||
|
|| !targetSolution.CanAddSolution(addedSolution) || targetSolution.TotalVolume == 0)
|
||||||
return false;
|
return false;
|
||||||
|
|
||||||
targetSolution.AddSolution(solution);
|
addedSolution.AddSolution(targetSolution);
|
||||||
|
UpdateChemicals(targetUid, addedSolution, true);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Adds a solution to the container, overflowing the rest.
|
||||||
|
/// Unlike TryAddSolution it will ignore size limits.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="targetUid">entity holding targetSolution</param>
|
||||||
|
/// <param name="targetSolution">The container to which we try to add.</param>
|
||||||
|
/// <param name="addedSolution">solution being added</param>
|
||||||
|
/// <param name="overflowThreshold">After addition this much will be left in targetSolution</param>
|
||||||
|
/// <param name="overflowingSolution">Solution that exceeded overflowThreshold</param>
|
||||||
|
/// <returns></returns>
|
||||||
|
public bool TryMixAndOverflow(EntityUid targetUid, Solution targetSolution,
|
||||||
|
Solution addedSolution,
|
||||||
|
FixedPoint2 overflowThreshold,
|
||||||
|
[NotNullWhen(true)] out Solution? overflowingSolution)
|
||||||
|
{
|
||||||
|
if (addedSolution.TotalVolume == 0)
|
||||||
|
{
|
||||||
|
overflowingSolution = null;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
targetSolution.AddSolution(addedSolution);
|
||||||
UpdateChemicals(targetUid, targetSolution, true);
|
UpdateChemicals(targetUid, targetSolution, true);
|
||||||
|
overflowingSolution = targetSolution.SplitSolution(FixedPoint2.Max(FixedPoint2.Zero,
|
||||||
|
targetSolution.CurrentVolume - overflowThreshold));
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -295,19 +325,6 @@ namespace Content.Server.Chemistry.EntitySystems
|
|||||||
return removedReagent;
|
return removedReagent;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void TryRemoveAllReagents(EntityUid uid, Solution solution, List<Solution.ReagentQuantity> removeReagents)
|
|
||||||
{
|
|
||||||
if (removeReagents.Count == 0)
|
|
||||||
return;
|
|
||||||
|
|
||||||
foreach (var reagent in removeReagents)
|
|
||||||
{
|
|
||||||
solution.RemoveReagent(reagent.ReagentId, reagent.Quantity);
|
|
||||||
}
|
|
||||||
|
|
||||||
UpdateChemicals(uid, solution);
|
|
||||||
}
|
|
||||||
|
|
||||||
public FixedPoint2 GetReagentQuantity(EntityUid owner, string reagentId)
|
public FixedPoint2 GetReagentQuantity(EntityUid owner, string reagentId)
|
||||||
{
|
{
|
||||||
var reagentQuantity = FixedPoint2.New(0);
|
var reagentQuantity = FixedPoint2.New(0);
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
using Content.Server.Fluids.Components;
|
using Content.Server.Fluids.EntitySystems;
|
||||||
using Content.Server.Fluids.EntitySystems;
|
|
||||||
using Content.Shared.Chemistry.Components;
|
using Content.Shared.Chemistry.Components;
|
||||||
using Content.Shared.Chemistry.Reaction;
|
using Content.Shared.Chemistry.Reaction;
|
||||||
using Content.Shared.Chemistry.Reagent;
|
using Content.Shared.Chemistry.Reagent;
|
||||||
|
|||||||
@@ -37,7 +37,7 @@ namespace Content.Server.Destructible.Thresholds.Behaviors
|
|||||||
else if (Solution != null &&
|
else if (Solution != null &&
|
||||||
solutionContainerSystem.TryGetSolution(owner, Solution, out var behaviorSolution))
|
solutionContainerSystem.TryGetSolution(owner, Solution, out var behaviorSolution))
|
||||||
{
|
{
|
||||||
spillableSystem.SpillAt(behaviorSolution, coordinates, "PuddleSmear", false);
|
spillableSystem.SpillAt(behaviorSolution, coordinates, "PuddleSmear");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
14
Content.Server/Fluids/Components/FluidSpreaderComponent.cs
Normal file
14
Content.Server/Fluids/Components/FluidSpreaderComponent.cs
Normal file
@@ -0,0 +1,14 @@
|
|||||||
|
using Content.Server.Fluids.EntitySystems;
|
||||||
|
using Content.Shared.Chemistry.Components;
|
||||||
|
|
||||||
|
namespace Content.Server.Fluids.Components;
|
||||||
|
|
||||||
|
[RegisterComponent]
|
||||||
|
[Friend(typeof(FluidSpreaderSystem))]
|
||||||
|
public sealed class FluidSpreaderComponent : Component
|
||||||
|
{
|
||||||
|
[ViewVariables]
|
||||||
|
public Solution OverflownSolution = default!;
|
||||||
|
|
||||||
|
public bool Enabled { get; set; }
|
||||||
|
}
|
||||||
@@ -48,11 +48,6 @@ namespace Content.Server.Fluids.Components
|
|||||||
[DataField("spillSound")]
|
[DataField("spillSound")]
|
||||||
public SoundSpecifier SpillSound = new SoundPathSpecifier("/Audio/Effects/Fluids/splat.ogg");
|
public SoundSpecifier SpillSound = new SoundPathSpecifier("/Audio/Effects/Fluids/splat.ogg");
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// Whether or not this puddle is currently overflowing onto its neighbors
|
|
||||||
/// </summary>
|
|
||||||
public bool Overflown;
|
|
||||||
|
|
||||||
[ViewVariables(VVAccess.ReadOnly)]
|
[ViewVariables(VVAccess.ReadOnly)]
|
||||||
public FixedPoint2 CurrentVolume => EntitySystem.Get<PuddleSystem>().CurrentVolume(Owner);
|
public FixedPoint2 CurrentVolume => EntitySystem.Get<PuddleSystem>().CurrentVolume(Owner);
|
||||||
|
|
||||||
|
|||||||
208
Content.Server/Fluids/EntitySystems/FluidSpreaderSystem.cs
Normal file
208
Content.Server/Fluids/EntitySystems/FluidSpreaderSystem.cs
Normal file
@@ -0,0 +1,208 @@
|
|||||||
|
using System.Diagnostics.CodeAnalysis;
|
||||||
|
using System.Linq;
|
||||||
|
using Content.Server.Chemistry.EntitySystems;
|
||||||
|
using Content.Server.Fluids.Components;
|
||||||
|
using Content.Shared.Chemistry.Components;
|
||||||
|
using Content.Shared.Directions;
|
||||||
|
using Content.Shared.FixedPoint;
|
||||||
|
using Content.Shared.Physics;
|
||||||
|
using JetBrains.Annotations;
|
||||||
|
using Robust.Shared.Map;
|
||||||
|
using Robust.Shared.Physics;
|
||||||
|
using Robust.Shared.Utility;
|
||||||
|
|
||||||
|
namespace Content.Server.Fluids.EntitySystems;
|
||||||
|
|
||||||
|
[UsedImplicitly]
|
||||||
|
public sealed class FluidSpreaderSystem : EntitySystem
|
||||||
|
{
|
||||||
|
[Dependency] private readonly IMapManager _mapManager = default!;
|
||||||
|
[Dependency] private readonly PuddleSystem _puddleSystem = default!;
|
||||||
|
[Dependency] private readonly SolutionContainerSystem _solutionContainerSystem = default!;
|
||||||
|
|
||||||
|
|
||||||
|
private float _accumulatedTimeFrame;
|
||||||
|
private HashSet<EntityUid> _fluidSpread = new();
|
||||||
|
|
||||||
|
public override void Initialize()
|
||||||
|
{
|
||||||
|
SubscribeLocalEvent<FluidSpreaderComponent, ComponentAdd>((uid, component, _) =>
|
||||||
|
FluidSpreaderAdd(uid, component));
|
||||||
|
}
|
||||||
|
|
||||||
|
public void AddOverflowingPuddle(PuddleComponent puddleComponent, Solution? solution = null)
|
||||||
|
{
|
||||||
|
var puddleSolution = solution;
|
||||||
|
if (puddleSolution == null && !_solutionContainerSystem.TryGetSolution(puddleComponent.Owner,
|
||||||
|
puddleComponent.SolutionName,
|
||||||
|
out puddleSolution)) return;
|
||||||
|
|
||||||
|
if (puddleSolution.CurrentVolume <= puddleComponent.OverflowVolume)
|
||||||
|
return;
|
||||||
|
|
||||||
|
var spreaderComponent = EntityManager.EnsureComponent<FluidSpreaderComponent>(puddleComponent.Owner);
|
||||||
|
spreaderComponent.OverflownSolution = puddleSolution;
|
||||||
|
spreaderComponent.Enabled = true;
|
||||||
|
FluidSpreaderAdd(spreaderComponent.Owner, spreaderComponent);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void FluidSpreaderAdd(EntityUid uid, FluidSpreaderComponent component)
|
||||||
|
{
|
||||||
|
if (component.Enabled)
|
||||||
|
_fluidSpread.Add(uid);
|
||||||
|
}
|
||||||
|
|
||||||
|
public override void Update(float frameTime)
|
||||||
|
{
|
||||||
|
_accumulatedTimeFrame += frameTime;
|
||||||
|
|
||||||
|
if (!(_accumulatedTimeFrame >= 1.0f))
|
||||||
|
return;
|
||||||
|
|
||||||
|
_accumulatedTimeFrame -= 1.0f;
|
||||||
|
|
||||||
|
base.Update(frameTime);
|
||||||
|
|
||||||
|
var remQueue = new RemQueue<EntityUid>();
|
||||||
|
foreach (var uid in _fluidSpread)
|
||||||
|
{
|
||||||
|
MetaDataComponent? meta = null;
|
||||||
|
|
||||||
|
if (Paused(uid, meta))
|
||||||
|
continue;
|
||||||
|
|
||||||
|
// If not paused
|
||||||
|
// it's either Deleted or will be via SpreadFluid
|
||||||
|
remQueue.Add(uid);
|
||||||
|
|
||||||
|
if (Deleted(uid, meta))
|
||||||
|
continue;
|
||||||
|
|
||||||
|
SpreadFluid(uid);
|
||||||
|
}
|
||||||
|
|
||||||
|
foreach (var removeUid in remQueue)
|
||||||
|
{
|
||||||
|
_fluidSpread.Remove(removeUid);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void SpreadFluid(EntityUid suid)
|
||||||
|
{
|
||||||
|
PuddleComponent? puddleComponent = null;
|
||||||
|
MetaDataComponent? metadataOriginal = null;
|
||||||
|
TransformComponent? transformOrig = null;
|
||||||
|
FluidSpreaderComponent? spreader = null;
|
||||||
|
|
||||||
|
if (!Resolve(suid, ref puddleComponent, ref metadataOriginal, ref transformOrig, ref spreader, false))
|
||||||
|
return;
|
||||||
|
|
||||||
|
var prototypeName = metadataOriginal.EntityPrototype!.ID;
|
||||||
|
|
||||||
|
var puddles = new List<PuddleComponent> { puddleComponent };
|
||||||
|
var visitedTiles = new HashSet<Vector2i>();
|
||||||
|
|
||||||
|
if (!_mapManager.TryGetGrid(transformOrig.GridID, out var mapGrid))
|
||||||
|
return;
|
||||||
|
|
||||||
|
while (puddles.Count > 0
|
||||||
|
&& spreader.OverflownSolution.CurrentVolume > FixedPoint2.Zero)
|
||||||
|
{
|
||||||
|
var nextToExpand = new List<(Vector2i, EntityUid?)>();
|
||||||
|
|
||||||
|
var divided = spreader.OverflownSolution.CurrentVolume / puddles.Count;
|
||||||
|
|
||||||
|
foreach (var puddle in puddles)
|
||||||
|
{
|
||||||
|
if (puddle.CurrentVolume >= puddle.OverflowVolume) continue;
|
||||||
|
|
||||||
|
// -puddle.OverflowLeft is guaranteed to be >= 0
|
||||||
|
// iff puddle.CurrentVolume >= puddle.OverflowVolume
|
||||||
|
var split = FixedPoint2.Min(divided, -puddle.OverflowLeft);
|
||||||
|
_puddleSystem.TryAddSolution(
|
||||||
|
puddle.Owner,
|
||||||
|
spreader.OverflownSolution.SplitSolution(split),
|
||||||
|
false, false, puddle);
|
||||||
|
}
|
||||||
|
|
||||||
|
// if solution is spent do not explore
|
||||||
|
if (spreader.OverflownSolution.CurrentVolume <= FixedPoint2.Zero)
|
||||||
|
continue;
|
||||||
|
|
||||||
|
// find edges
|
||||||
|
foreach (var puddle in puddles)
|
||||||
|
{
|
||||||
|
TransformComponent? transform = null;
|
||||||
|
|
||||||
|
if (!Resolve(puddle.Owner, ref transform, false))
|
||||||
|
continue;
|
||||||
|
|
||||||
|
// prepare next set of puddles to be expanded
|
||||||
|
var puddlePos = transform.Coordinates.ToVector2i(EntityManager, _mapManager);
|
||||||
|
foreach (var direction in SharedDirectionExtensions.RandomDirections().ToArray())
|
||||||
|
{
|
||||||
|
var newPos = puddlePos.Offset(direction);
|
||||||
|
if (visitedTiles.Contains(newPos))
|
||||||
|
continue;
|
||||||
|
|
||||||
|
visitedTiles.Add(newPos);
|
||||||
|
|
||||||
|
if (CanExpand(newPos, mapGrid, out var uid))
|
||||||
|
nextToExpand.Add((newPos, uid));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
puddles = new List<PuddleComponent>();
|
||||||
|
|
||||||
|
// prepare edges for next iteration
|
||||||
|
foreach (var (pos, uid) in nextToExpand)
|
||||||
|
{
|
||||||
|
if (spreader.OverflownSolution.CurrentVolume <= FixedPoint2.Zero)
|
||||||
|
continue;
|
||||||
|
|
||||||
|
var puddleUid = uid!.Value;
|
||||||
|
var coordinate = mapGrid.GridTileToWorld(pos);
|
||||||
|
if (uid == EntityUid.Invalid)
|
||||||
|
{
|
||||||
|
puddleUid = EntityManager.SpawnEntity(prototypeName, coordinate);
|
||||||
|
}
|
||||||
|
|
||||||
|
puddles.Add(EntityManager.GetComponent<PuddleComponent>(puddleUid));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private bool CanExpand(Vector2i newPos, IMapGrid mapGrid,
|
||||||
|
[NotNullWhen(true)] out EntityUid? uid)
|
||||||
|
{
|
||||||
|
if (!mapGrid.TryGetTileRef(newPos, out var tileRef)
|
||||||
|
|| tileRef.Tile.IsEmpty)
|
||||||
|
{
|
||||||
|
uid = null;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
foreach (var entity in mapGrid.GetAnchoredEntities(newPos))
|
||||||
|
{
|
||||||
|
IPhysBody? physics = null;
|
||||||
|
PuddleComponent? existingPuddle = null;
|
||||||
|
|
||||||
|
// This is an invalid location
|
||||||
|
if (Resolve(entity, ref physics, false)
|
||||||
|
&& (physics.CollisionLayer & (int) CollisionGroup.Impassable) != 0)
|
||||||
|
{
|
||||||
|
uid = null;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!Resolve(entity, ref existingPuddle, false))
|
||||||
|
continue;
|
||||||
|
|
||||||
|
uid = entity;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
uid = EntityUid.Invalid;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,28 +1,12 @@
|
|||||||
using System;
|
|
||||||
using System.Collections.Generic;
|
|
||||||
using System.Diagnostics.CodeAnalysis;
|
|
||||||
using System.Linq;
|
|
||||||
using Content.Server.Chemistry.EntitySystems;
|
using Content.Server.Chemistry.EntitySystems;
|
||||||
using Content.Server.Construction.Components;
|
|
||||||
using Content.Server.Fluids.Components;
|
using Content.Server.Fluids.Components;
|
||||||
using Content.Shared.Chemistry.Components;
|
using Content.Shared.Chemistry.Components;
|
||||||
using Content.Shared.Database;
|
|
||||||
using Content.Shared.Directions;
|
|
||||||
using Content.Shared.Examine;
|
using Content.Shared.Examine;
|
||||||
using Content.Shared.FixedPoint;
|
using Content.Shared.FixedPoint;
|
||||||
using Content.Shared.Fluids;
|
using Content.Shared.Fluids;
|
||||||
using Content.Shared.Maps;
|
|
||||||
using Content.Shared.Physics;
|
|
||||||
using Content.Shared.Slippery;
|
using Content.Shared.Slippery;
|
||||||
using Content.Shared.Verbs;
|
|
||||||
using JetBrains.Annotations;
|
using JetBrains.Annotations;
|
||||||
using Robust.Shared.Audio;
|
using Robust.Shared.Audio;
|
||||||
using Robust.Shared.GameObjects;
|
|
||||||
using Robust.Shared.IoC;
|
|
||||||
using Robust.Shared.Localization;
|
|
||||||
using Robust.Shared.Map;
|
|
||||||
using Robust.Shared.Maths;
|
|
||||||
using Robust.Shared.Physics;
|
|
||||||
using Robust.Shared.Player;
|
using Robust.Shared.Player;
|
||||||
|
|
||||||
namespace Content.Server.Fluids.EntitySystems
|
namespace Content.Server.Fluids.EntitySystems
|
||||||
@@ -30,8 +14,8 @@ namespace Content.Server.Fluids.EntitySystems
|
|||||||
[UsedImplicitly]
|
[UsedImplicitly]
|
||||||
public sealed class PuddleSystem : EntitySystem
|
public sealed class PuddleSystem : EntitySystem
|
||||||
{
|
{
|
||||||
[Dependency] private readonly IMapManager _mapManager = default!;
|
|
||||||
[Dependency] private readonly SolutionContainerSystem _solutionContainerSystem = default!;
|
[Dependency] private readonly SolutionContainerSystem _solutionContainerSystem = default!;
|
||||||
|
[Dependency] private readonly FluidSpreaderSystem _fluidSpreaderSystem = default!;
|
||||||
|
|
||||||
public override void Initialize()
|
public override void Initialize()
|
||||||
{
|
{
|
||||||
@@ -45,7 +29,7 @@ namespace Content.Server.Fluids.EntitySystems
|
|||||||
|
|
||||||
private void OnInit(EntityUid uid, PuddleComponent component, ComponentInit args)
|
private void OnInit(EntityUid uid, PuddleComponent component, ComponentInit args)
|
||||||
{
|
{
|
||||||
var solution = _solutionContainerSystem.EnsureSolution(uid, component.SolutionName);
|
var solution = _solutionContainerSystem.EnsureSolution(uid, component.SolutionName);
|
||||||
solution.MaxVolume = FixedPoint2.New(1000);
|
solution.MaxVolume = FixedPoint2.New(1000);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -111,21 +95,6 @@ namespace Content.Server.Fluids.EntitySystems
|
|||||||
QueueDel(uid);
|
QueueDel(uid);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// Whether adding this solution to this puddle would overflow.
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="uid">Uid of owning entity</param>
|
|
||||||
/// <param name="puddle">Puddle to which we are adding solution</param>
|
|
||||||
/// <param name="solution">Solution we intend to add</param>
|
|
||||||
/// <returns></returns>
|
|
||||||
public bool WouldOverflow(EntityUid uid, Solution solution, PuddleComponent? puddle = null)
|
|
||||||
{
|
|
||||||
if (!Resolve(uid, ref puddle))
|
|
||||||
return false;
|
|
||||||
|
|
||||||
return puddle.CurrentVolume + solution.TotalVolume > puddle.OverflowVolume;
|
|
||||||
}
|
|
||||||
|
|
||||||
public bool EmptyHolder(EntityUid uid, PuddleComponent? puddleComponent = null)
|
public bool EmptyHolder(EntityUid uid, PuddleComponent? puddleComponent = null)
|
||||||
{
|
{
|
||||||
if (!Resolve(uid, ref puddleComponent))
|
if (!Resolve(uid, ref puddleComponent))
|
||||||
@@ -147,24 +116,40 @@ namespace Content.Server.Fluids.EntitySystems
|
|||||||
: FixedPoint2.Zero;
|
: FixedPoint2.Zero;
|
||||||
}
|
}
|
||||||
|
|
||||||
public bool TryAddSolution(EntityUid uid, Solution solution,
|
/// <summary>
|
||||||
|
///
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="puddleUid">Puddle to which we add</param>
|
||||||
|
/// <param name="addedSolution">Solution that is added to puddleComponent</param>
|
||||||
|
/// <param name="sound">Play sound on overflow</param>
|
||||||
|
/// <param name="checkForOverflow">Overflow on encountered values</param>
|
||||||
|
/// <param name="puddleComponent">Optional resolved PuddleComponent</param>
|
||||||
|
/// <returns></returns>
|
||||||
|
public bool TryAddSolution(EntityUid puddleUid,
|
||||||
|
Solution addedSolution,
|
||||||
bool sound = true,
|
bool sound = true,
|
||||||
bool checkForOverflow = true,
|
bool checkForOverflow = true,
|
||||||
PuddleComponent? puddleComponent = null)
|
PuddleComponent? puddleComponent = null)
|
||||||
{
|
{
|
||||||
if (!Resolve(uid, ref puddleComponent))
|
if (!Resolve(puddleUid, ref puddleComponent))
|
||||||
return false;
|
return false;
|
||||||
|
|
||||||
if (solution.TotalVolume == 0 ||
|
if (addedSolution.TotalVolume == 0 ||
|
||||||
!_solutionContainerSystem.TryGetSolution(puddleComponent.Owner, puddleComponent.SolutionName,
|
!_solutionContainerSystem.TryGetSolution(puddleComponent.Owner, puddleComponent.SolutionName,
|
||||||
out var puddleSolution))
|
out var puddleSolution))
|
||||||
{
|
{
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
var result = _solutionContainerSystem
|
var result = _solutionContainerSystem
|
||||||
.TryAddSolution(puddleComponent.Owner, puddleSolution, solution);
|
.TryMixAndOverflow(puddleComponent.Owner, puddleSolution, addedSolution, puddleComponent.OverflowVolume,
|
||||||
|
out var overflowSolution);
|
||||||
|
|
||||||
|
if (checkForOverflow && overflowSolution != null)
|
||||||
|
{
|
||||||
|
_fluidSpreaderSystem.AddOverflowingPuddle(puddleComponent, overflowSolution);
|
||||||
|
}
|
||||||
|
|
||||||
if (!result)
|
if (!result)
|
||||||
{
|
{
|
||||||
return false;
|
return false;
|
||||||
@@ -172,11 +157,6 @@ namespace Content.Server.Fluids.EntitySystems
|
|||||||
|
|
||||||
RaiseLocalEvent(puddleComponent.Owner, new SolutionChangedEvent());
|
RaiseLocalEvent(puddleComponent.Owner, new SolutionChangedEvent());
|
||||||
|
|
||||||
if (checkForOverflow)
|
|
||||||
{
|
|
||||||
CheckOverflow(puddleComponent);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!sound)
|
if (!sound)
|
||||||
{
|
{
|
||||||
return true;
|
return true;
|
||||||
@@ -188,139 +168,18 @@ namespace Content.Server.Fluids.EntitySystems
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// Will overflow this entity to neighboring entities if required
|
/// Whether adding this solution to this puddle would overflow.
|
||||||
/// </summary>
|
/// </summary>
|
||||||
private void CheckOverflow(PuddleComponent puddleComponent)
|
/// <param name="uid">Uid of owning entity</param>
|
||||||
|
/// <param name="puddle">Puddle to which we are adding solution</param>
|
||||||
|
/// <param name="solution">Solution we intend to add</param>
|
||||||
|
/// <returns></returns>
|
||||||
|
public bool WouldOverflow(EntityUid uid, Solution solution, PuddleComponent? puddle = null)
|
||||||
{
|
{
|
||||||
if (puddleComponent.CurrentVolume <= puddleComponent.OverflowVolume
|
if (!Resolve(uid, ref puddle))
|
||||||
|| puddleComponent.Overflown)
|
|
||||||
return;
|
|
||||||
|
|
||||||
var nextPuddles = new List<PuddleComponent>() { puddleComponent };
|
|
||||||
var overflownPuddles = new List<PuddleComponent>();
|
|
||||||
|
|
||||||
while (puddleComponent.OverflowLeft > FixedPoint2.Zero && nextPuddles.Count > 0)
|
|
||||||
{
|
|
||||||
foreach (var next in nextPuddles.ToArray())
|
|
||||||
{
|
|
||||||
nextPuddles.Remove(next);
|
|
||||||
|
|
||||||
next.Overflown = true;
|
|
||||||
overflownPuddles.Add(next);
|
|
||||||
|
|
||||||
var adjacentPuddles = GetAllAdjacentOverflow(next).ToArray();
|
|
||||||
if (puddleComponent.OverflowLeft <= FixedPoint2.Epsilon * adjacentPuddles.Length)
|
|
||||||
{
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (adjacentPuddles.Length == 0)
|
|
||||||
{
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
var numberOfAdjacent = FixedPoint2.New(adjacentPuddles.Length);
|
|
||||||
var overflowSplit = puddleComponent.OverflowLeft / numberOfAdjacent;
|
|
||||||
foreach (var adjacent in adjacentPuddles)
|
|
||||||
{
|
|
||||||
var adjacentPuddle = adjacent();
|
|
||||||
var quantity = FixedPoint2.Min(overflowSplit, adjacentPuddle.OverflowVolume);
|
|
||||||
var puddleSolution = _solutionContainerSystem.EnsureSolution(puddleComponent.Owner,
|
|
||||||
puddleComponent.SolutionName);
|
|
||||||
var spillAmount = _solutionContainerSystem.SplitSolution(puddleComponent.Owner,
|
|
||||||
puddleSolution, quantity);
|
|
||||||
|
|
||||||
TryAddSolution(adjacentPuddle.Owner, spillAmount, false, false);
|
|
||||||
nextPuddles.Add(adjacentPuddle);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
foreach (var puddle in overflownPuddles)
|
|
||||||
{
|
|
||||||
puddle.Overflown = false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// Finds or creates adjacent puddles in random directions from this one
|
|
||||||
/// </summary>
|
|
||||||
/// <returns>Enumerable of the puddles found or to be created</returns>
|
|
||||||
private IEnumerable<Func<PuddleComponent>> GetAllAdjacentOverflow(PuddleComponent puddleComponent)
|
|
||||||
{
|
|
||||||
foreach (var direction in SharedDirectionExtensions.RandomDirections())
|
|
||||||
{
|
|
||||||
if (TryGetAdjacentOverflow(puddleComponent, direction, out var puddle))
|
|
||||||
{
|
|
||||||
yield return puddle;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// Tries to get an adjacent coordinate to overflow to, unless it is blocked by a wall on the
|
|
||||||
/// same tile or the tile is empty
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="puddleComponent"></param>
|
|
||||||
/// <param name="direction">The direction to get the puddle from, respective to this one</param>
|
|
||||||
/// <param name="puddle">The puddle that was found or is to be created, or null if there
|
|
||||||
/// is a wall in the way</param>
|
|
||||||
/// <returns>true if a puddle was found or created, false otherwise</returns>
|
|
||||||
private bool TryGetAdjacentOverflow(PuddleComponent puddleComponent, Direction direction,
|
|
||||||
[NotNullWhen(true)] out Func<PuddleComponent>? puddle)
|
|
||||||
{
|
|
||||||
puddle = default;
|
|
||||||
|
|
||||||
// We're most likely in space, do nothing.
|
|
||||||
if (!EntityManager.GetComponent<TransformComponent>(puddleComponent.Owner).GridID.IsValid())
|
|
||||||
return false;
|
return false;
|
||||||
|
|
||||||
var mapGrid = _mapManager.GetGrid(EntityManager.GetComponent<TransformComponent>(puddleComponent.Owner).GridID);
|
return puddle.CurrentVolume + solution.TotalVolume > puddle.OverflowVolume;
|
||||||
var coords = EntityManager.GetComponent<TransformComponent>(puddleComponent.Owner).Coordinates;
|
|
||||||
|
|
||||||
if (!coords.Offset(direction).TryGetTileRef(out var tile))
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// If space return early, let that spill go out into the void
|
|
||||||
if (tile.Value.Tile.IsEmpty)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!EntityManager.GetComponent<TransformComponent>(puddleComponent.Owner).Anchored)
|
|
||||||
return false;
|
|
||||||
|
|
||||||
foreach (var entity in mapGrid.GetInDir(coords, direction))
|
|
||||||
{
|
|
||||||
if (EntityManager.TryGetComponent(entity, out IPhysBody? physics) &&
|
|
||||||
(physics.CollisionLayer & (int)CollisionGroup.Impassable) != 0)
|
|
||||||
{
|
|
||||||
puddle = default;
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (EntityManager.TryGetComponent(entity, out PuddleComponent? existingPuddle))
|
|
||||||
{
|
|
||||||
if (existingPuddle.Overflown)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
puddle = () => existingPuddle;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
puddle ??= () =>
|
|
||||||
{
|
|
||||||
var id = EntityManager.SpawnEntity(
|
|
||||||
EntityManager.GetComponent<MetaDataComponent>(puddleComponent.Owner).EntityPrototype?.ID,
|
|
||||||
mapGrid.DirectionToGrid(coords, direction));
|
|
||||||
return EntityManager.GetComponent<PuddleComponent>(id);
|
|
||||||
};
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,7 +3,6 @@ using System.Linq;
|
|||||||
using Content.Server.Administration.Logs;
|
using Content.Server.Administration.Logs;
|
||||||
using Content.Server.Chemistry.EntitySystems;
|
using Content.Server.Chemistry.EntitySystems;
|
||||||
using Content.Server.Clothing.Components;
|
using Content.Server.Clothing.Components;
|
||||||
using Content.Server.Coordinates.Helpers;
|
|
||||||
using Content.Server.Fluids.Components;
|
using Content.Server.Fluids.Components;
|
||||||
using Content.Shared.Chemistry.Components;
|
using Content.Shared.Chemistry.Components;
|
||||||
using Content.Shared.Chemistry.Reagent;
|
using Content.Shared.Chemistry.Reagent;
|
||||||
@@ -13,17 +12,13 @@ using Content.Shared.Inventory.Events;
|
|||||||
using Content.Shared.Throwing;
|
using Content.Shared.Throwing;
|
||||||
using Content.Shared.Verbs;
|
using Content.Shared.Verbs;
|
||||||
using JetBrains.Annotations;
|
using JetBrains.Annotations;
|
||||||
using Robust.Server.GameObjects;
|
|
||||||
using Robust.Shared.GameObjects;
|
|
||||||
using Robust.Shared.IoC;
|
|
||||||
using Robust.Shared.Localization;
|
|
||||||
using Robust.Shared.Map;
|
using Robust.Shared.Map;
|
||||||
using Robust.Shared.Prototypes;
|
using Robust.Shared.Prototypes;
|
||||||
|
|
||||||
namespace Content.Server.Fluids.EntitySystems;
|
namespace Content.Server.Fluids.EntitySystems;
|
||||||
|
|
||||||
[UsedImplicitly]
|
[UsedImplicitly]
|
||||||
public class SpillableSystem : EntitySystem
|
public sealed class SpillableSystem : EntitySystem
|
||||||
{
|
{
|
||||||
[Dependency] private readonly SolutionContainerSystem _solutionContainerSystem = default!;
|
[Dependency] private readonly SolutionContainerSystem _solutionContainerSystem = default!;
|
||||||
[Dependency] private readonly PuddleSystem _puddleSystem = default!;
|
[Dependency] private readonly PuddleSystem _puddleSystem = default!;
|
||||||
@@ -61,7 +56,6 @@ public class SpillableSystem : EntitySystem
|
|||||||
// spill all solution on the player
|
// spill all solution on the player
|
||||||
var drainedSolution = _solutionContainerSystem.Drain(uid, solution, solution.DrainAvailable);
|
var drainedSolution = _solutionContainerSystem.Drain(uid, solution, solution.DrainAvailable);
|
||||||
SpillAt(args.Equipee, drainedSolution, "PuddleSmear");
|
SpillAt(args.Equipee, drainedSolution, "PuddleSmear");
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
@@ -174,10 +168,13 @@ public class SpillableSystem : EntitySystem
|
|||||||
if (!noTileReact)
|
if (!noTileReact)
|
||||||
{
|
{
|
||||||
// First, do all tile reactions
|
// First, do all tile reactions
|
||||||
foreach (var (reagentId, quantity) in solution.Contents)
|
for (var i = 0; i < solution.Contents.Count; i++)
|
||||||
{
|
{
|
||||||
|
var (reagentId, quantity) = solution.Contents[i];
|
||||||
var proto = _prototypeManager.Index<ReagentPrototype>(reagentId);
|
var proto = _prototypeManager.Index<ReagentPrototype>(reagentId);
|
||||||
proto.ReactionTile(tileRef, quantity);
|
var removed = proto.ReactionTile(tileRef, quantity);
|
||||||
|
if (removed <= FixedPoint2.Zero) continue;
|
||||||
|
solution.RemoveReagent(reagentId, removed);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -188,40 +185,34 @@ public class SpillableSystem : EntitySystem
|
|||||||
// Get normalized co-ordinate for spill location and spill it in the centre
|
// Get normalized co-ordinate for spill location and spill it in the centre
|
||||||
// TODO: Does SnapGrid or something else already do this?
|
// TODO: Does SnapGrid or something else already do this?
|
||||||
var spillGridCoords = mapGrid.GridTileToWorld(tileRef.GridIndices);
|
var spillGridCoords = mapGrid.GridTileToWorld(tileRef.GridIndices);
|
||||||
|
var startEntity = EntityUid.Invalid;
|
||||||
var spillEntities = _entityLookup.GetEntitiesIntersecting(mapGrid.ParentMapId, spillGridCoords.Position).ToArray();
|
PuddleComponent? puddleComponent = null;
|
||||||
foreach (var spillEntity in spillEntities)
|
|
||||||
{
|
|
||||||
if (_solutionContainerSystem.TryGetRefillableSolution(spillEntity, out var solutionContainerComponent))
|
|
||||||
{
|
|
||||||
_solutionContainerSystem.Refill(spillEntity, solutionContainerComponent,
|
|
||||||
solution.SplitSolution(FixedPoint2.Min(
|
|
||||||
solutionContainerComponent.AvailableVolume,
|
|
||||||
solutionContainerComponent.MaxSpillRefill))
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (combine)
|
if (combine)
|
||||||
{
|
{
|
||||||
|
var spillEntities = _entityLookup.GetEntitiesIntersecting(tileRef).ToArray();
|
||||||
|
|
||||||
foreach (var spillEntity in spillEntities)
|
foreach (var spillEntity in spillEntities)
|
||||||
{
|
{
|
||||||
if (!EntityManager.TryGetComponent(spillEntity, out PuddleComponent? puddleComponent)) continue;
|
if (!EntityManager.TryGetComponent(spillEntity, out puddleComponent)) continue;
|
||||||
|
|
||||||
if (!overflow && _puddleSystem.WouldOverflow(puddleComponent.Owner, solution, puddleComponent))
|
if (!overflow && _puddleSystem.WouldOverflow(puddleComponent.Owner, solution, puddleComponent))
|
||||||
return null;
|
return null;
|
||||||
|
|
||||||
if (!_puddleSystem.TryAddSolution(puddleComponent.Owner, solution, sound)) continue;
|
if (!_puddleSystem.TryAddSolution(puddleComponent.Owner, solution, sound, overflow)) continue;
|
||||||
|
|
||||||
return puddleComponent;
|
startEntity = puddleComponent.Owner;
|
||||||
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
var puddleEnt = EntityManager.SpawnEntity(prototype, spillGridCoords);
|
if (startEntity != EntityUid.Invalid)
|
||||||
var newPuddleComponent = EntityManager.GetComponent<PuddleComponent>(puddleEnt);
|
return puddleComponent;
|
||||||
|
|
||||||
_puddleSystem.TryAddSolution(newPuddleComponent.Owner, solution, sound);
|
startEntity = EntityManager.SpawnEntity(prototype, spillGridCoords);
|
||||||
|
puddleComponent = EntityManager.EnsureComponent<PuddleComponent>(startEntity);
|
||||||
|
_puddleSystem.TryAddSolution(startEntity, solution, sound, overflow);
|
||||||
|
|
||||||
return newPuddleComponent;
|
return puddleComponent;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user