🥈 Allow enumeration of priority queues (#225)

beardgame · May 31, 2021 · 9196603 · 9196603
1 parent f9a2118
commit 9196603
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/Bearded.Utilities/Collections/StaticPriorityQueue.cs b/Bearded.Utilities/Collections/StaticPriorityQueue.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Collections;
 using System.Collections.Generic;
 using System.Linq;
 
@@ -9,7 +10,9 @@ namespace Bearded.Utilities.Collections
     /// </summary>
     /// <typeparam name="TPriority"></typeparam>
     /// <typeparam name="TValue"></typeparam>
-    public class StaticPriorityQueue<TPriority, TValue> where TPriority : IComparable<TPriority>
+    public class StaticPriorityQueue<TPriority, TValue>
+        : IEnumerable<KeyValuePair<TPriority, TValue>>
+        where TPriority : IComparable<TPriority>
     {
         /// <summary>
         /// Array-representation of the entire heap.
@@ -203,5 +206,15 @@ private static int getRightChild(int i)
             return 2 * i + 2;
         }
         #endregion
+
+        public IEnumerator<KeyValuePair<TPriority, TValue>> GetEnumerator()
+        {
+            for (var i = 0; i < Count; i++)
+            {
+                yield return data[i];
+            }
+        }
+
+        IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
     }
 }