commit

Assets/Data/TraitUtils.cs

@@ -1,6 +1,52 @@
-namespace Assets.Data
+using System.Collections.Generic;
+using System.Numerics;
+
+namespace Assets.Data
 {
     public class TraitUtils
     {
+        public static HashSet<TraitsEnum> FromInt(int traits)
+        {
+            HashSet<TraitsEnum> result = new HashSet<TraitsEnum>();
+            foreach (TraitsEnum trait in System.Enum.GetValues(typeof(TraitsEnum)))
+            {
+                if ((traits & (int)trait) != 0)
+                {
+                    result.Add(trait);
+                }
+            }
+            return result;
+        }
+
+        public static int ToInt(HashSet<TraitsEnum> traits)
+        {
+            int result = 0;
+            foreach (TraitsEnum trait in traits)
+            {
+                result |= (int)trait;
+            }
+            return result;
+        }
+
+        public static int TraitCountFromInt(int traits)
+        {
+            int count = 0;
+            while (traits != 0)
+            {
+                count += traits & 1;
+                traits >>= 1;
+            }
+            return count;
+        }
+
+        public static bool ContainsTrait(int traits, TraitsEnum trait)
+        {
+            return (traits & (int)trait) != 0;
+        }
+        public static bool ContainsTrait(int traits, int trait)
+        {
+            return (traits & trait) != 0;
+        }
+        
     }
 }