82 lines
2.7 KiB
C#
82 lines
2.7 KiB
C#
using System.Linq;
|
|
using Content.Server.Atmos;
|
|
using Content.Server.GameObjects.Components.NodeContainer;
|
|
using Content.Server.GameObjects.Components.NodeContainer.Nodes;
|
|
using Content.Server.GameObjects.EntitySystems;
|
|
using Content.Shared.GameObjects.Components.Atmos;
|
|
using Robust.Server.GameObjects;
|
|
using Robust.Shared.GameObjects.Systems;
|
|
using Robust.Shared.Interfaces.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Log;
|
|
using Robust.Shared.ViewVariables;
|
|
|
|
namespace Content.Server.GameObjects.Components.Atmos.Piping.Vents
|
|
{
|
|
/// <summary>
|
|
/// Transfers gas from a <see cref="PipeNode"/> to the tile it is on.
|
|
/// </summary>
|
|
public abstract class BaseVentComponent : PipeNetDeviceComponent
|
|
{
|
|
|
|
[ViewVariables]
|
|
private PipeNode _ventInlet;
|
|
|
|
private AtmosphereSystem _atmosSystem;
|
|
|
|
[ViewVariables(VVAccess.ReadWrite)]
|
|
public bool VentEnabled
|
|
{
|
|
get => _ventEnabled;
|
|
set
|
|
{
|
|
_ventEnabled = value;
|
|
UpdateAppearance();
|
|
}
|
|
}
|
|
private bool _ventEnabled = true;
|
|
|
|
private AppearanceComponent _appearance;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
_atmosSystem = EntitySystem.Get<AtmosphereSystem>();
|
|
if (!Owner.TryGetComponent<NodeContainerComponent>(out var container))
|
|
{
|
|
JoinedGridAtmos?.RemovePipeNetDevice(this);
|
|
Logger.Error($"{typeof(BaseVentComponent)} on entity {Owner.Uid} did not have a {nameof(NodeContainerComponent)}.");
|
|
return;
|
|
}
|
|
_ventInlet = container.Nodes.OfType<PipeNode>().FirstOrDefault();
|
|
if (_ventInlet == null)
|
|
{
|
|
JoinedGridAtmos?.RemovePipeNetDevice(this);
|
|
Logger.Error($"{typeof(BaseVentComponent)} on entity {Owner.Uid} could not find compatible {nameof(PipeNode)}s on its {nameof(NodeContainerComponent)}.");
|
|
return;
|
|
}
|
|
Owner.TryGetComponent(out _appearance);
|
|
UpdateAppearance();
|
|
}
|
|
|
|
public override void Update()
|
|
{
|
|
if (!VentEnabled)
|
|
return;
|
|
|
|
var tileAtmos = Owner.Transform.Coordinates.GetTileAtmosphere(Owner.EntityManager);
|
|
if (tileAtmos == null)
|
|
return;
|
|
VentGas(_ventInlet.Air, tileAtmos.Air);
|
|
tileAtmos.Invalidate();
|
|
}
|
|
|
|
protected abstract void VentGas(GasMixture inletGas, GasMixture outletGas);
|
|
|
|
private void UpdateAppearance()
|
|
{
|
|
_appearance?.SetData(VentVisuals.VisualState, new VentVisualState(VentEnabled));
|
|
}
|
|
}
|
|
}
|