dmitry1100 · March 12, 2024 13:09 · Nov 14, 2019 · Nov 14, 2019 · Nov 14, 2019 · Nov 14, 2019
diff --git a/TransformAllChildren.cs b/TransformAllChildren.cs
@@ -14,7 +14,7 @@ public static IEnumerable<Transform> AllChildren ( this Transform t )
 
         private class DeepEnumerator : IEnumerator<Transform>, IEnumerable<Transform>
         {
-            private Transform _root;
+            private readonly Transform _root;
             private Transform _current;
             private Stack<Transform> _stack;
 

diff --git a/TransformAllChildren.cs b/TransformAllChildren.cs
@@ -3,7 +3,6 @@
 using System.Collections.Generic;
 using UnityEngine;
 
-
 namespace Fiftytwo
 {
     public static class TransformAllChildren

diff --git a/TransformAllChildrenTest.cs b/TransformAllChildrenTest.cs
@@ -4,6 +4,7 @@ namespace Fiftytwo
 {
     public class TransformAllChildrenTest : MonoBehaviour
     {
+        // Will be called just after component attached to GameObject
         private void Reset ()
         {
             foreach ( var child in transform.AllChildren() )

diff --git a/TransformAllChildren → TransformAllChildren.cs b/TransformAllChildren → TransformAllChildren.cs
diff --git a/TestTransformAllChildren.cs → TransformAllChildrenTest.cs b/TestTransformAllChildren.cs → TransformAllChildrenTest.cs
@@ -2,7 +2,7 @@
 
 namespace Fiftytwo
 {
-    public class TestTransformAllChildren : MonoBehaviour
+    public class TransformAllChildrenTest : MonoBehaviour
     {
         private void Reset ()
         {

diff --git a/TestTransformAllChildren.cs b/TestTransformAllChildren.cs
@@ -0,0 +1,15 @@
+using UnityEngine;
+
+namespace Fiftytwo
+{
+    public class TestTransformAllChildren : MonoBehaviour
+    {
+        private void Reset ()
+        {
+            foreach ( var child in transform.AllChildren() )
+            {
+                Debug.Log( child.name );
+            }
+        }
+    }
+}
diff --git a/TransformAllChildren b/TransformAllChildren
@@ -0,0 +1,66 @@
+using System;
+using System.Collections;
+using System.Collections.Generic;
+using UnityEngine;
+
+
+namespace Fiftytwo
+{
+    public static class TransformAllChildren
+    {
+        public static IEnumerable<Transform> AllChildren ( this Transform t )
+        {
+            return new DeepEnumerator( t );
+        }
+
+        private class DeepEnumerator : IEnumerator<Transform>, IEnumerable<Transform>
+        {
+            private Transform _root;
+            private Transform _current;
+            private Stack<Transform> _stack;
+
+            internal DeepEnumerator ( Transform root )
+            {
+                _root = root;
+                _current = root;
+                _stack = new Stack<Transform>();
+            }
+
+            public Transform Current { get { return _current; } }
+
+            object IEnumerator.Current { get { return _current; } }
+
+            void IDisposable.Dispose () { }
+
+            public bool MoveNext ()
+            {
+                for( var i = _current.childCount; --i >= 0; )
+                    _stack.Push( _current.GetChild( i ) );
+
+                if( _stack.Count == 0 )
+                    return false;
+
+                _current = _stack.Pop();
+
+                return true;
+            }
+
+            public void Reset ()
+            {
+                _current = _root;
+                _stack.Clear();
+            }
+
+            public IEnumerator<Transform> GetEnumerator ()
+            {
+                Reset();
+                return this;
+            }
+
+            IEnumerator IEnumerable.GetEnumerator ()
+            {
+                return GetEnumerator();
+            }
+        }
+    }
+}