Improve Node.GetChildKeys()

JimmyCushnie · Apr 26, 2024 · ea5920a · ea5920a
1 parent 361c087
commit ea5920a
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/SUCC/Parsing Logic/Nodes/Node.cs b/SUCC/Parsing Logic/Nodes/Node.cs
@@ -182,14 +182,13 @@ public void CapChildCount(int count)
             }
         }
 
-        public string[] GetChildKeys()
+        public IEnumerable<string> GetChildKeys()
         {
-            var keys = new string[ChildNodes.Count];
+            if (ChildNodeType != NodeChildrenType.Key)
+                yield break;
 
-            for (int i = 0; i < ChildNodes.Count; i++)
-                keys[i] = (ChildNodes[i] as KeyNode).Key;
-
-            return keys;
+            foreach (var node in ChildNodes)
+                yield return (node as KeyNode).Key;
         }