50 lines
1.6 KiB
C#
50 lines
1.6 KiB
C#
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Content.Server.Disposal.Unit.Components;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Maths;
|
|
using Robust.Shared.Random;
|
|
using Robust.Shared.Serialization.Manager.Attributes;
|
|
using Robust.Shared.ViewVariables;
|
|
|
|
namespace Content.Server.Disposal.Tube.Components
|
|
{
|
|
[RegisterComponent]
|
|
[ComponentReference(typeof(IDisposalTubeComponent))]
|
|
public class DisposalJunctionComponent : DisposalTubeComponent
|
|
{
|
|
[Dependency] private readonly IRobustRandom _random = default!;
|
|
|
|
/// <summary>
|
|
/// The angles to connect to.
|
|
/// </summary>
|
|
[ViewVariables]
|
|
[DataField("degrees")]
|
|
private List<Angle> _degrees = new();
|
|
|
|
public override string Name => "DisposalJunction";
|
|
|
|
protected override Direction[] ConnectableDirections()
|
|
{
|
|
var direction = IoCManager.Resolve<IEntityManager>().GetComponent<TransformComponent>(Owner.Uid).LocalRotation;
|
|
|
|
return _degrees.Select(degree => new Angle(degree.Theta + direction.Theta).GetDir()).ToArray();
|
|
}
|
|
|
|
public override Direction NextDirection(DisposalHolderComponent holder)
|
|
{
|
|
var next = IoCManager.Resolve<IEntityManager>().GetComponent<TransformComponent>(Owner.Uid).LocalRotation.GetDir();
|
|
var directions = ConnectableDirections().Skip(1).ToArray();
|
|
|
|
if (holder.PreviousDirectionFrom == Direction.Invalid ||
|
|
holder.PreviousDirectionFrom == next)
|
|
{
|
|
return _random.Pick(directions);
|
|
}
|
|
|
|
return next;
|
|
}
|
|
}
|
|
}
|