From c57f9fdbf675675cf367cdbf11eaeda9bd0ee0b1 Mon Sep 17 00:00:00 2001 From: Ellpeck Date: Fri, 9 Aug 2024 22:37:42 +0200 Subject: [PATCH] Added GetUniqueValues --- DynamicEnums/DynamicEnum.cs | 17 +++++++++++++++++ DynamicEnums/EnumHelper.cs | 16 ++++++++++++++++ Tests/EnumTests.cs | 24 +++++++++++++++++++++++- 3 files changed, 56 insertions(+), 1 deletion(-) diff --git a/DynamicEnums/DynamicEnum.cs b/DynamicEnums/DynamicEnum.cs index 9050077..5cf45ab 100644 --- a/DynamicEnums/DynamicEnum.cs +++ b/DynamicEnums/DynamicEnum.cs @@ -200,6 +200,23 @@ namespace DynamicEnums { return DynamicEnum.GetValues(typeof(T)).Cast(); } + /// + /// Returns a collection of all of the enum values that are explicitly defined for the given dynamic enum type , excluding any explicitly defined combined flags. + /// A value counts as explicitly defined if it has been added using , or . + /// + /// The type whose values to get. + /// The defined values for the given type, excluding combined flags. + public static IEnumerable GetUniqueValues() where T : DynamicEnum { + var used = BigInteger.Zero; + foreach (var value in DynamicEnum.GetValues()) { + var iValue = DynamicEnum.GetValue(value); + if ((used & iValue) == 0) { + yield return value; + used |= iValue; + } + } + } + /// /// Returns a collection of all of the enum values that are explicitly defined for the given dynamic enum type . /// A value counts as explicitly defined if it has been added using , or . diff --git a/DynamicEnums/EnumHelper.cs b/DynamicEnums/EnumHelper.cs index 2da36f6..e7c5b69 100644 --- a/DynamicEnums/EnumHelper.cs +++ b/DynamicEnums/EnumHelper.cs @@ -21,6 +21,22 @@ namespace DynamicEnums { #endif } + /// + /// Returns an array containing all of the values of the given enum type, excluding any explicitly defined combined flags. + /// + /// The type whose enum to get. + /// An enumerable of the values of the enum, in declaration order, excluding combined flags. + public static IEnumerable GetUniqueValues() where T : struct, Enum { + var used = 0L; + foreach (var value in EnumHelper.GetValues()) { + var lValue = Convert.ToInt64(value); + if ((used & lValue) == 0) { + yield return value; + used |= lValue; + } + } + } + /// /// Returns true if the given has all of the given flags on it. /// This operation is equivalent to , but doesn't do any additional checks, making it faster. diff --git a/Tests/EnumTests.cs b/Tests/EnumTests.cs index e0635c2..acc421b 100644 --- a/Tests/EnumTests.cs +++ b/Tests/EnumTests.cs @@ -1,4 +1,5 @@ using System; +using System.Linq; using System.Numerics; using DynamicEnums; using NUnit.Framework; @@ -9,8 +10,18 @@ public class EnumTests { [Test] public void TestRegularEnums() { + Assert.AreEqual( + new[] {TestEnum.Zero, TestEnum.One, TestEnum.Two, TestEnum.Eight, TestEnum.Sixteen, TestEnum.EightSixteen, TestEnum.ThirtyTwo, TestEnum.OneTwentyEight, TestEnum.OneTwentyEightTwoOne}, + EnumHelper.GetValues()); + Assert.AreEqual( + new[] {TestEnum.Zero, TestEnum.One, TestEnum.Two, TestEnum.Eight, TestEnum.Sixteen, TestEnum.ThirtyTwo, TestEnum.OneTwentyEight}, + EnumHelper.GetUniqueValues()); + Assert.AreEqual( new[] {TestEnum.One, TestEnum.Two, TestEnum.Eight, TestEnum.Sixteen, TestEnum.EightSixteen}, + EnumHelper.GetFlags(TestEnum.One | TestEnum.Sixteen | TestEnum.Eight | TestEnum.Two, false)); + Assert.AreEqual( + new[] {TestEnum.Zero, TestEnum.One, TestEnum.Two, TestEnum.Eight, TestEnum.Sixteen, TestEnum.EightSixteen}, EnumHelper.GetFlags(TestEnum.One | TestEnum.Sixteen | TestEnum.Eight | TestEnum.Two)); Assert.AreEqual( @@ -35,9 +46,16 @@ public class EnumTests { var zero = DynamicEnum.Add("Zero", 0); var combined = DynamicEnum.Add("Combined", DynamicEnum.GetValue(DynamicEnum.Or(flags[7], flags[13]))); - DynamicEnum.Add("Test", 10); + var test = DynamicEnum.Add("Test", 10); Assert.AreEqual(DynamicEnum.GetEnumValue(10).ToString(), "TestModified"); + Assert.AreEqual( + flags.Append(zero).Append(combined), + DynamicEnum.GetValues()); + Assert.AreEqual( + flags.Append(zero), + DynamicEnum.GetUniqueValues()); + Assert.AreEqual(DynamicEnum.GetValue(flags[7]), BigInteger.One << 7); Assert.AreEqual(DynamicEnum.GetEnumValue(BigInteger.One << 75), flags[75]); @@ -73,6 +91,9 @@ public class EnumTests { Assert.AreEqual( new[] {flags[0], flags[7], flags[13], combined}, DynamicEnum.GetFlags(DynamicEnum.Or(DynamicEnum.Or(flags[0], flags[13]), flags[7]), false)); + Assert.AreEqual( + new[] {flags[0], flags[7], flags[13], zero, combined}, + DynamicEnum.GetFlags(DynamicEnum.Or(DynamicEnum.Or(flags[0], flags[13]), flags[7]))); Assert.AreEqual( new[] {flags[0], flags[7], flags[13]}, @@ -82,6 +103,7 @@ public class EnumTests { [Flags] private enum TestEnum { + Zero = 0, One = 1, Two = 2, Eight = 8,