CatStation/Content.Client/Rotation/RotationVisualizerSystem.cs

using Content.Shared.Rotation;
using Robust.Client.Animations;
using Robust.Client.GameObjects;
using Robust.Shared.Animations;

namespace Content.Client.Rotation;

public sealed partial class RotationVisualizerSystem : SharedRotationVisualsSystem
{
    [Dependency] private AnimationPlayerSystem _animation = default!;

    [SubscribeLocalEvent]
    private void OnAppearanceChange(Entity<RotationVisualsComponent> ent, ref AppearanceChangeEvent args)
    {
        if (args.Sprite == null)
            return;

        if (!args.TryGetData<RotationState>(RotationVisuals.RotationState, out var state))
            state = RotationState.Vertical;

        switch (state)
        {
            case RotationState.Vertical:
                AnimateSpriteRotation(ent, args.Sprite, ent.Comp.VerticalRotation, ent.Comp.AnimationTime);
                break;
            case RotationState.Horizontal:
                AnimateSpriteRotation(ent, args.Sprite, ent.Comp.HorizontalRotation, ent.Comp.AnimationTime);
                break;
        }
    }

    /// <summary>
    ///     Rotates a sprite between two animated keyframes given a certain time.
    /// </summary>
    public void AnimateSpriteRotation(EntityUid uid, SpriteComponent spriteComp, Angle rotation, float animationTime)
    {
        if (spriteComp.Rotation.Equals(rotation))
        {
            return;
        }

        var animationComp = EnsureComp<AnimationPlayerComponent>(uid);
        const string animationKey = "rotate";
        // Stop the current rotate animation and then start a new one
        if (_animation.HasRunningAnimation(animationComp, animationKey))
        {
            _animation.Stop((uid, animationComp), animationKey);
        }

        var animation = new Animation
        {
            Length = TimeSpan.FromSeconds(animationTime),
            AnimationTracks =
            {
                new AnimationTrackComponentProperty
                {
                    ComponentType = typeof(SpriteComponent),
                    Property = nameof(SpriteComponent.Rotation),
                    InterpolationMode = AnimationInterpolationMode.Linear,
                    KeyFrames =
                    {
                        new AnimationTrackProperty.KeyFrame(spriteComp.Rotation, 0),
                        new AnimationTrackProperty.KeyFrame(rotation, animationTime)
                    }
                }
            }
        };

        _animation.Play((uid, animationComp), animation, animationKey);
    }
}