osu-lazer/osu.Game.Rulesets.Osu/Edit/Masks/SliderMasks/Components/ControlPointVisualiser.cs

// Copyright (c) 2007-2018 ppy Pty Ltd <contact@ppy.sh>.
// Licensed under the MIT Licence - https://raw.githubusercontent.com/ppy/osu/master/LICENCE

using osu.Framework.Graphics;
using osu.Framework.Graphics.Containers;
using osu.Game.Rulesets.Osu.Objects;

namespace osu.Game.Rulesets.Osu.Edit.Masks.SliderMasks.Components
{
    public class ControlPointVisualiser : CompositeDrawable
    {
        private readonly Slider slider;

        private readonly Container<ControlPointPiece> pieces;

        public ControlPointVisualiser(Slider slider)
        {
            this.slider = slider;

            InternalChild = pieces = new Container<ControlPointPiece> { RelativeSizeAxes = Axes.Both };

            slider.ControlPointsChanged += _ => updateControlPoints();
            updateControlPoints();
        }

        private void updateControlPoints()
        {
            while (slider.ControlPoints.Length > pieces.Count)
                pieces.Add(new ControlPointPiece(slider, pieces.Count));
            while (slider.ControlPoints.Length < pieces.Count)
                pieces.Remove(pieces[pieces.Count - 1]);
        }
    }
}