osukey/osu.Game.Rulesets.Mania/Utils/SnapFinder.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 osu.Framework.Utils;
using osu.Game.Beatmaps.ControlPoints;
using osu.Game.Rulesets.Mania.Beatmaps;
using osu.Game.Rulesets.Objects;

namespace osu.Game.Rulesets.Mania.Utils
{
    public class SnapFinder
    {
        private readonly ManiaBeatmap beatmap;

        public SnapFinder(ManiaBeatmap beatmap)
        {
            this.beatmap = beatmap;
        }

        private readonly static int[] snaps = { 1, 2, 3, 4, 6, 8, 12, 16 };

        public int FindSnap(HitObject hitObject)
        {
            TimingControlPoint currentTimingPoint = beatmap.ControlPointInfo.TimingPointAt(hitObject.StartTime);
            double snapResult = (hitObject.StartTime - currentTimingPoint.Time) % (currentTimingPoint.BeatLength * 4);

            foreach (var snap in snaps)
            {
                if (almostDivisibleBy(snapResult, currentTimingPoint.BeatLength / (double)snap))
                    return snap;
            }

            return 0;
        }

        private const double leniency_ms = 1.0;

        private static bool almostDivisibleBy(double dividend, double divisor)
        {
            double remainder = Math.Abs(dividend) % divisor;
            return Precision.AlmostEquals(remainder, 0, leniency_ms) || Precision.AlmostEquals(remainder - divisor, 0, leniency_ms);
        }
    }
}