diff --git a/src/utils/itertools.ts b/src/utils/itertools.ts index 221d6c0..e52e034 100644 --- a/src/utils/itertools.ts +++ b/src/utils/itertools.ts @@ -22,6 +22,34 @@ export function * MakeIterator(root: IContainerModel): Generator { + const queue: IContainerModel[] = [root]; + let depth = 0; + while (queue.length > 0) { + let levelSize = queue.length; + while (levelSize-- !== 0) { + const container = queue.shift() as IContainerModel; + yield { + container, + depth + }; + + for (let i = container.children.length - 1; i >= 0; i--) { + const child = container.children[i]; + queue.push(child); + } + } + depth++; + } +} + /** * Returns the depth of the container * @returns The depth of the container