kscience/dataforge-core
8
0
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases 12 Wiki Activity

Add depth-first walk for data tree

Browse Source
This commit is contained in:
Alexander Nozik 2024-02-24 15:01:27 +03:00
parent b2593d308e
commit 0e72b4b63c
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
dataforge-data/src/commonMain/kotlin/space/kscience/dataforge/data/DataSource.kt
View File

@ -96,10 +96,10 @@ public fun <T> DataTree<T>.asSequence(
/** /**
* Walk the data tree depth-first * Walk the data tree depth-first
*/ */
public fun <T, TR: GenericDataTree<T,TR>> TR.walk( private fun <T, TR: GenericDataTree<T,TR>> TR.walk(
namePrefix: Name = Name.EMPTY, namePrefix: Name = Name.EMPTY,
): Sequence<TR> = sequence { ): Sequence<Pair<Name,TR>> = sequence {
yield(this@walk) yield(namePrefix to this@walk)
items.forEach { (token, tree) -> items.forEach { (token, tree) ->
yieldAll(tree.walk(namePrefix + token)) yieldAll(tree.walk(namePrefix + token))
} }