添加区间的 Clamp 方法

dotnet-campus · Jan 10, 2025 · 73beb72 · 73beb72
1 parent b724dce
commit 73beb72
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/DotNetCampus.Numerics/FloatingPointHelper.cs b/DotNetCampus.Numerics/FloatingPointHelper.cs
@@ -66,5 +66,24 @@ public static TNum Sqrt<TNum>(this TNum a)
         throw new NotSupportedException();
     }
 
+    public static TNum Clamp<TNum>(this TNum value, TNum min, TNum max)
+        where TNum : unmanaged, IFloatingPoint<TNum>
+    {
+        if (typeof(TNum) == typeof(float))
+            return Unsafe.BitCast<float, TNum>(Math.Clamp(Unsafe.BitCast<TNum, float>(value), Unsafe.BitCast<TNum, float>(min), Unsafe.BitCast<TNum, float>(max)));
+
+        if (typeof(TNum) == typeof(double))
+            return Unsafe.BitCast<double, TNum>(Math.Clamp(Unsafe.BitCast<TNum, double>(value), Unsafe.BitCast<TNum, double>(min), Unsafe.BitCast<TNum, double>(max)));
+
+        if (typeof(TNum) == typeof(NFloat))
+            return Unsafe.BitCast<NFloat, TNum>(NFloat.Clamp(Unsafe.BitCast<TNum, NFloat>(value), Unsafe.BitCast<TNum, NFloat>(min), Unsafe.BitCast<TNum, NFloat>(max)));
+
+        return value <= min
+            ? min
+            : value >= max
+                ? max
+                : value;
+    }
+
     #endregion
 }
diff --git a/DotNetCampus.Numerics/Interval.cs b/DotNetCampus.Numerics/Interval.cs
@@ -67,6 +67,16 @@ public bool Contains(TNum value)
         return Start <= value && value <= End;
     }
 
+    /// <summary>
+    /// 将目标值限制在区间内。
+    /// </summary>
+    /// <param name="value">需要限制的值。</param>
+    /// <returns>限制后的值。</returns>
+    public TNum Clamp(ref TNum value)
+    {
+        return value.Clamp(Start, End);
+    }
+
     /// <inheritdoc />
     public override string ToString()
     {