osu-lazer/osu.Game.Rulesets.Osu/Objects/Drawables/Connections/FollowPointGroup.cs

// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
// See the LICENCE file in the repository root for full licence text.

using System;
using JetBrains.Annotations;
using osu.Framework.Graphics;
using osu.Framework.Graphics.Containers;
using osu.Game.Rulesets.Objects.Types;
using osuTK;

namespace osu.Game.Rulesets.Osu.Objects.Drawables.Connections
{
    public class FollowPointGroup : CompositeDrawable
    {
        // Todo: These shouldn't be constants
        private const int spacing = 32;
        private const double preempt = 800;

        /// <summary>
        /// The <see cref="DrawableOsuHitObject"/> which <see cref="FollowPoint"/>s will exit from.
        /// </summary>
        [NotNull]
        public readonly DrawableOsuHitObject Start;

        public FollowPointGroup(DrawableOsuHitObject start)
        {
            Start = start;
            RelativeSizeAxes = Axes.Both;
        }

        protected override void LoadComplete()
        {
            base.LoadComplete();
            bindHitObject(Start);
        }

        private DrawableOsuHitObject end;

        /// <summary>
        /// The <see cref="DrawableOsuHitObject"/> which <see cref="FollowPoint"/>s will enter.
        /// </summary>
        [CanBeNull]
        public DrawableOsuHitObject End
        {
            get => end;
            set
            {
                end = value;

                if (end != null)
                    bindHitObject(end);

                refreshFollowPoints();
            }
        }

        private void bindHitObject(DrawableOsuHitObject drawableObject)
        {
            drawableObject.HitObject.StartTimeBindable.BindValueChanged(_ => refreshFollowPoints());
            drawableObject.HitObject.PositionBindable.BindValueChanged(_ => refreshFollowPoints());
            drawableObject.HitObject.DefaultsApplied += refreshFollowPoints;
        }

        private void refreshFollowPoints()
        {
            ClearInternal();

            if (End == null)
                return;

            OsuHitObject osuStart = Start.HitObject;
            OsuHitObject osuEnd = End.HitObject;

            if (osuEnd.NewCombo)
                return;

            if (osuStart is Spinner || osuEnd is Spinner)
                return;

            Vector2 startPosition = osuStart.EndPosition;
            Vector2 endPosition = osuEnd.Position;
            double startTime = (osuStart as IHasEndTime)?.EndTime ?? osuStart.StartTime;
            double endTime = osuEnd.StartTime;

            Vector2 distanceVector = endPosition - startPosition;
            int distance = (int)distanceVector.Length;
            float rotation = (float)(Math.Atan2(distanceVector.Y, distanceVector.X) * (180 / Math.PI));
            double duration = endTime - startTime;

            for (int d = (int)(spacing * 1.5); d < distance - spacing; d += spacing)
            {
                float fraction = (float)d / distance;
                Vector2 pointStartPosition = startPosition + (fraction - 0.1f) * distanceVector;
                Vector2 pointEndPosition = startPosition + fraction * distanceVector;
                double fadeOutTime = startTime + fraction * duration;
                double fadeInTime = fadeOutTime - preempt;

                FollowPoint fp;

                AddInternal(fp = new FollowPoint
                {
                    Position = pointStartPosition,
                    Rotation = rotation,
                    Alpha = 0,
                    Scale = new Vector2(1.5f * osuEnd.Scale),
                });

                using (fp.BeginAbsoluteSequence(fadeInTime))
                {
                    fp.FadeIn(osuEnd.TimeFadeIn);
                    fp.ScaleTo(osuEnd.Scale, osuEnd.TimeFadeIn, Easing.Out);
                    fp.MoveTo(pointEndPosition, osuEnd.TimeFadeIn, Easing.Out);
                    fp.Delay(fadeOutTime - fadeInTime).FadeOut(osuEnd.TimeFadeIn);
                }

                fp.Expire(true);
            }
        }
    }
}