civ14/Content.Server/ParticleAccelerator/EntitySystems/ParticleAcceleratorSystem.Emitter.cs

using Content.Server.ParticleAccelerator.Components;
using Content.Server.Singularity.Components;
using Content.Shared.Projectiles;
using Content.Shared.Singularity.Components;
using Robust.Shared.Physics.Components;

namespace Content.Server.ParticleAccelerator.EntitySystems;

public sealed partial class ParticleAcceleratorSystem
{
    private void FireEmitter(EntityUid uid, ParticleAcceleratorPowerState strength, ParticleAcceleratorEmitterComponent? emitter = null)
    {
        if (!Resolve(uid, ref emitter))
            return;

        var xformQuery = GetEntityQuery<TransformComponent>();
        if (!xformQuery.TryGetComponent(uid, out var xform))
        {
            Log.Error("ParticleAccelerator attempted to emit a particle without (having) a transform from which to base its initial position and orientation.");
            return;
        }

        var emitted = Spawn(emitter.EmittedPrototype, xform.Coordinates);

        if (xformQuery.TryGetComponent(emitted, out var particleXform))
            _transformSystem.SetLocalRotation(emitted, xform.LocalRotation, particleXform);

        if (TryComp<PhysicsComponent>(emitted, out var particlePhys))
        {
            var angle = _transformSystem.GetWorldRotation(uid, xformQuery);
            _physicsSystem.SetBodyStatus(emitted, particlePhys, BodyStatus.InAir);

            var velocity = angle.ToWorldVec() * 20f;
            if (TryComp<PhysicsComponent>(uid, out var phys))
                velocity += phys.LinearVelocity; // Inherit velocity from parent so if the clown has strapped a dozen engines to departures we don't outpace the particles.

            _physicsSystem.SetLinearVelocity(emitted, velocity, body: particlePhys);
        }

        if (TryComp<ProjectileComponent>(emitted, out var proj))
            _projectileSystem.SetShooter(emitted, proj, uid);

        if (TryComp<SinguloFoodComponent>(emitted, out var food))
        {
            // TODO: Unhardcode this.
            food.Energy = strength switch
            {
                ParticleAcceleratorPowerState.Standby => 0,
                ParticleAcceleratorPowerState.Level0 => 1,
                ParticleAcceleratorPowerState.Level1 => 2,
                ParticleAcceleratorPowerState.Level2 => 3,
                ParticleAcceleratorPowerState.Level3 => 6,
                _ => 0,
            } * 10;
        }

        if (TryComp<ParticleProjectileComponent>(emitted, out var particle))
            particle.State = strength;

        _appearanceSystem.SetData(emitted, ParticleAcceleratorVisuals.VisualState, strength);
    }
}