osu-lazer/osu.Game/Beatmaps/WorkingBeatmap_VirtualBeatmapTrack.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.Linq;
using osu.Framework.Audio.Track;
using osu.Game.Rulesets.Objects;
using osu.Game.Rulesets.Objects.Types;

namespace osu.Game.Beatmaps
{
    public partial class WorkingBeatmap
    {
        /// <summary>
        /// A type of <see cref="TrackVirtual"/> which provides a valid length based on the <see cref="HitObject"/>s of an <see cref="IBeatmap"/>.
        /// </summary>
        protected class VirtualBeatmapTrack : TrackVirtual
        {
            private const double excess_length = 1000;

            public VirtualBeatmapTrack(IBeatmap beatmap)
            {
                var lastObject = beatmap.HitObjects.LastOrDefault();

                switch (lastObject)
                {
                    case null:
                        Length = excess_length;
                        break;
                    case IHasEndTime endTime:
                        Length = endTime.EndTime + excess_length;
                        break;
                    default:
                        Length = lastObject.StartTime + excess_length;
                        break;
                }
            }
        }
    }
}