feat: TimeSpan - rounding to interval

jirikostiha · Nov 20, 2024 · 910df1d · 910df1d
1 parent 90f5b81
commit 910df1d
Show file tree

Hide file tree

Showing 2 changed files with 69 additions and 0 deletions.
diff --git a/src/Atin.Tests/TimeSpanExtensionsTests.cs b/src/Atin.Tests/TimeSpanExtensionsTests.cs
@@ -31,4 +31,37 @@ public void ToAtin_ForTwoDigits()
 
         Assert.Equal("M10", result);
     }
+
+    [Fact]
+    public void RoundTo_M5()
+    {
+        var subInterval = TimeSpan.FromMinutes(5);
+        var totalInterval = new TimeSpan(5, 12, 45);
+
+        var result = totalInterval.RoundTo(subInterval);
+
+        Assert.Equal(new TimeSpan(5, 15, 00), result);
+    }
+
+    [Fact]
+    public void RoundDown_M5()
+    {
+        var subInterval = TimeSpan.FromMinutes(5);
+        var totalInterval = new TimeSpan(5, 12, 45);
+
+        var result = totalInterval.RoundDown(subInterval);
+
+        Assert.Equal(new TimeSpan(5, 10, 00), result);
+    }
+
+    [Fact]
+    public void RoundUp_M5()
+    {
+        var subInterval = TimeSpan.FromMinutes(5);
+        var totalInterval = new TimeSpan(5, 12, 45);
+
+        var result = totalInterval.RoundUp(subInterval);
+
+        Assert.Equal(new TimeSpan(5, 15, 00), result);
+    }
 }
diff --git a/src/Atin/TimeSpanExtensions.cs b/src/Atin/TimeSpanExtensions.cs
@@ -66,4 +66,40 @@ public static string ToAtin(this TimeSpan timeSpan)
 
         return result.ToString();
     }
+
+    /// <summary>
+    /// Rounds the <see cref="TimeSpan"/> value to the nearest interval.
+    /// </summary>
+    /// <param name="timeSpan">The original <see cref="TimeSpan"/> to round.</param>
+    /// <param name="subInterval">The interval to which the value should be rounded.</param>
+    /// <returns>The rounded <see cref="TimeSpan"/>.</returns>
+    public static TimeSpan RoundTo(this TimeSpan timeSpan, TimeSpan subInterval)
+    {
+        long ticks = (timeSpan.Ticks + subInterval.Ticks / 2) / subInterval.Ticks * subInterval.Ticks;
+        return TimeSpan.FromTicks(ticks);
+    }
+
+    /// <summary>
+    /// Rounds the <see cref="TimeSpan"/> value down to the nearest interval.
+    /// </summary>
+    /// <param name="timeSpan">The original <see cref="TimeSpan"/> to round.</param>
+    /// <param name="subInterval">The interval to which the value should be rounded down.</param>
+    /// <returns>The rounded-down <see cref="TimeSpan"/>.</returns>
+    public static TimeSpan RoundDown(this TimeSpan timeSpan, TimeSpan subInterval)
+    {
+        long ticks = timeSpan.Ticks / subInterval.Ticks * subInterval.Ticks;
+        return TimeSpan.FromTicks(ticks);
+    }
+
+    /// <summary>
+    /// Rounds the <see cref="TimeSpan"/> value up to the nearest interval.
+    /// </summary>
+    /// <param name="timeSpan">The original <see cref="TimeSpan"/> to round.</param>
+    /// <param name="subInterval">The interval to which the value should be rounded up.</param>
+    /// <returns>The rounded-up <see cref="TimeSpan"/>.</returns>
+    public static TimeSpan RoundUp(this TimeSpan timeSpan, TimeSpan subInterval)
+    {
+        long ticks = (timeSpan.Ticks + subInterval.Ticks - 1) / subInterval.Ticks * subInterval.Ticks;
+        return TimeSpan.FromTicks(ticks);
+    }
 }