tbd-station-14/Content.Shared/Singularity/SharedSingularitySystem.cs

using Content.Shared.Ghost;
using Content.Shared.Radiation;
using Content.Shared.Radiation.Components;
using Content.Shared.Singularity.Components;
using Robust.Shared.Physics;
using Robust.Shared.Physics.Collision.Shapes;
using Robust.Shared.Physics.Components;
using Robust.Shared.Physics.Dynamics;
using Robust.Shared.Physics.Events;
using Robust.Shared.Physics.Systems;

namespace Content.Shared.Singularity
{
    public abstract class SharedSingularitySystem : EntitySystem
    {
        [Dependency] private readonly FixtureSystem _fixtures = default!;
        [Dependency] private readonly SharedAppearanceSystem _appearance = default!;
        [Dependency] private readonly SharedPhysicsSystem _physics = default!;

        public const string DeleteFixture = "DeleteCircle";

        private float GetFalloff(int level)
        {
            return level switch
            {
                0 => 9999f,
                1 => MathF.Sqrt(6.4f),
                2 => MathF.Sqrt(7.0f),
                3 => MathF.Sqrt(8.0f),
                4 => MathF.Sqrt(10.0f),
                5 => MathF.Sqrt(12.0f),
                6 => MathF.Sqrt(12.0f),
                _ => -1.0f
            };
        }

        private float GetIntensity(int level)
        {
            return level switch
            {
                0 => 0.0f,
                1 => 3645f,
                2 => 103680f,
                3 => 1113920f,
                4 => 16200000f,
                5 => 180000000f,
                6 => 180000000f,
                _ => -1.0f
            };
        }

        public override void Initialize()
        {
            base.Initialize();
            SubscribeLocalEvent<SharedSingularityComponent, PreventCollideEvent>(OnPreventCollide);
        }

        protected void OnPreventCollide(EntityUid uid, SharedSingularityComponent component, ref PreventCollideEvent args)
        {
            PreventCollide(uid, component, ref args);
        }

        protected virtual bool PreventCollide(EntityUid uid, SharedSingularityComponent component, ref PreventCollideEvent args)
        {
            var otherUid = args.BodyB.Owner;

            // For prediction reasons always want the client to ignore these.
            if (EntityManager.HasComponent<IMapGridComponent>(otherUid) ||
                EntityManager.HasComponent<SharedGhostComponent>(otherUid))
            {
                args.Cancelled = true;
                return true;
            }

            // If we're above 4 then breach containment
            // otherwise, check if it's containment and just keep the collision
            if (EntityManager.HasComponent<SharedContainmentFieldComponent>(otherUid) ||
                EntityManager.HasComponent<SharedContainmentFieldGeneratorComponent>(otherUid))
            {
                if (component.Level > 4)
                {
                    args.Cancelled = true;
                }

                return true;
            }

            return false;
        }

        public void ChangeSingularityLevel(SharedSingularityComponent singularity, int value)
        {
            if (value == singularity.Level)
            {
                return;
            }

            value = Math.Clamp(value, 0, 6);

            var physics = EntityManager.GetComponentOrNull<PhysicsComponent>(singularity.Owner);

            if (singularity.Level > 1 && value <= 1)
            {
                // Prevents it getting stuck (see SingularityController.MoveSingulo)
                if (physics != null)
                {
                    _physics.SetLinearVelocity(physics, Vector2.Zero);
                }
            }

            singularity.Level = value;

            if (EntityManager.TryGetComponent(singularity.Owner, out RadiationSourceComponent? source))
            {
                source.Intensity = singularity.RadsPerLevel * value;
            }

            _appearance.SetData(singularity.Owner, SingularityVisuals.Level, value);

            if (physics != null)
            {
                var fixture = _fixtures.GetFixtureOrNull(physics, DeleteFixture);

                if (fixture != null)
                {
                    var circle = (PhysShapeCircle) fixture.Shape;
                    circle.Radius = value - 0.5f;

                    fixture.Hard = value <= 4;
                }
            }

            if (EntityManager.TryGetComponent(singularity.Owner, out SingularityDistortionComponent? distortion))
            {
                distortion.FalloffPower = GetFalloff(value);
                distortion.Intensity = GetIntensity(value);
            }

            singularity.Dirty();
        }
    }
}