[WIP] redo DataSet fill and updates
This commit is contained in:
parent
3334380693
commit
7e4d1af55f
@ -5,7 +5,7 @@ import kotlinx.coroutines.CoroutineScope
|
||||
import kotlinx.coroutines.flow.*
|
||||
|
||||
/**
|
||||
* A simple data transformation on a data node. Actions should avoid doing actual dependency evaluation in [run].
|
||||
* A simple data transformation on a data node. Actions should avoid doing actual dependency evaluation in [execute].
|
||||
*/
|
||||
public interface Action<in T : Any, out R : Any> {
|
||||
/**
|
||||
@ -14,7 +14,7 @@ public interface Action<in T : Any, out R : Any> {
|
||||
*
|
||||
* [scope] context used to compute the initial result, also it is used for updates propagation
|
||||
*/
|
||||
public suspend fun run(set: DataSet<T>, meta: Meta, scope: CoroutineScope): DataSet<R>
|
||||
public suspend fun execute(dataSet: DataSet<T>, meta: Meta = Meta.EMPTY, scope: CoroutineScope? = null): DataSet<R>
|
||||
|
||||
public companion object
|
||||
}
|
||||
@ -25,8 +25,8 @@ public interface Action<in T : Any, out R : Any> {
|
||||
public infix fun <T : Any, I : Any, R : Any> Action<T, I>.then(action: Action<I, R>): Action<T, R> {
|
||||
// TODO introduce composite action and add optimize by adding action to the list
|
||||
return object : Action<T, R> {
|
||||
override suspend fun run(set: DataSet<T>, meta: Meta, scope: CoroutineScope): DataSet<R> {
|
||||
return action.run(this@then.run(set, meta, scope), meta, scope)
|
||||
override suspend fun execute(dataSet: DataSet<T>, meta: Meta, scope: CoroutineScope?): DataSet<R> {
|
||||
return action.execute(this@then.execute(dataSet, meta, scope), meta, scope)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -4,6 +4,7 @@ import hep.dataforge.meta.Meta
|
||||
import hep.dataforge.names.Name
|
||||
import hep.dataforge.names.startsWith
|
||||
import kotlinx.coroutines.CoroutineScope
|
||||
import kotlinx.coroutines.coroutineScope
|
||||
import kotlinx.coroutines.flow.Flow
|
||||
import kotlinx.coroutines.flow.collect
|
||||
import kotlin.reflect.KClass
|
||||
@ -29,18 +30,20 @@ public abstract class CachingAction<in T : Any, out R : Any>(
|
||||
key: Name = Name.EMPTY,
|
||||
): Flow<NamedData<R>>
|
||||
|
||||
override suspend fun run(
|
||||
set: DataSet<T>,
|
||||
override suspend fun execute(
|
||||
dataSet: DataSet<T>,
|
||||
meta: Meta,
|
||||
scope: CoroutineScope,
|
||||
): DataSet<R> = DataTree.dynamic(outputType,scope) {
|
||||
collectFrom(scope.transform(set, meta))
|
||||
scope.let {
|
||||
set.updates.collect {
|
||||
scope: CoroutineScope?,
|
||||
): DataSet<R> = DataTree.dynamic(outputType) {
|
||||
coroutineScope {
|
||||
collectFrom(transform(dataSet, meta))
|
||||
}
|
||||
scope?.let {
|
||||
dataSet.updates.collect {
|
||||
//clear old nodes
|
||||
remove(it)
|
||||
//collect new items
|
||||
collectFrom(scope.transform(set, meta, it))
|
||||
collectFrom(scope.transform(dataSet, meta, it))
|
||||
//FIXME if the target is data, updates are fired twice
|
||||
}
|
||||
}
|
||||
|
@ -44,13 +44,15 @@ public interface Data<out T : Any> : Goal<T>, MetaRepr {
|
||||
/**
|
||||
* An empty data containing only meta
|
||||
*/
|
||||
public fun <T> empty(meta: Meta): Data<Nothing> = object : Data<Nothing> {
|
||||
public fun empty(meta: Meta): Data<Nothing> = object : Data<Nothing> {
|
||||
override val type: KClass<out Nothing> = Nothing::class
|
||||
override val meta: Meta = meta
|
||||
override val dependencies: Collection<Goal<*>> = emptyList()
|
||||
override val deferred: Deferred<Nothing> get() = GlobalScope.async(start = CoroutineStart.LAZY) {
|
||||
override val deferred: Deferred<Nothing>
|
||||
get() = GlobalScope.async(start = CoroutineStart.LAZY) {
|
||||
error("The Data is empty and could not be computed")
|
||||
}
|
||||
|
||||
override fun async(coroutineScope: CoroutineScope): Deferred<Nothing> = deferred
|
||||
override fun reset() {}
|
||||
}
|
||||
@ -92,7 +94,18 @@ public inline fun <reified T : Any> Data(
|
||||
public class NamedData<out T : Any> internal constructor(
|
||||
override val name: Name,
|
||||
public val data: Data<T>,
|
||||
) : Data<T> by data, Named
|
||||
) : Data<T> by data, Named {
|
||||
override fun toString(): String = buildString {
|
||||
append("NamedData(name=\"$name\"")
|
||||
if(data is StaticData){
|
||||
append(", value=${data.value}")
|
||||
}
|
||||
if(!data.meta.isEmpty()){
|
||||
append(", meta=${data.meta}")
|
||||
}
|
||||
append(")")
|
||||
}
|
||||
}
|
||||
|
||||
public fun <T : Any> Data<T>.named(name: Name): NamedData<T> = if (this is NamedData) {
|
||||
NamedData(name, this.data)
|
||||
|
@ -77,3 +77,5 @@ public suspend fun DataSet<*>.toMeta(): Meta = Meta {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public val <T : Any> DataSet<T>.updatesWithData: Flow<NamedData<T>> get() = updates.mapNotNull { getData(it)?.named(it) }
|
@ -4,50 +4,81 @@ import hep.dataforge.meta.DFExperimental
|
||||
import hep.dataforge.meta.Meta
|
||||
import hep.dataforge.meta.MetaBuilder
|
||||
import hep.dataforge.names.Name
|
||||
import hep.dataforge.names.plus
|
||||
import hep.dataforge.names.toName
|
||||
import kotlinx.coroutines.coroutineScope
|
||||
import kotlinx.coroutines.flow.Flow
|
||||
import kotlinx.coroutines.flow.collect
|
||||
|
||||
public interface DataSetBuilder<in T : Any> {
|
||||
public fun remove(name: Name)
|
||||
/**
|
||||
* Remove all data items starting with [name]
|
||||
*/
|
||||
public suspend fun remove(name: Name)
|
||||
|
||||
public operator fun set(name: Name, data: Data<T>?)
|
||||
public suspend fun set(name: Name, data: Data<T>?)
|
||||
|
||||
public suspend fun set(name: Name, dataSet: DataSet<T>)
|
||||
|
||||
public operator fun set(name: Name, block: DataSetBuilder<T>.() -> Unit)
|
||||
/**
|
||||
* Set a current state of given [dataSet] into a branch [name]. Does not propagate updates
|
||||
*/
|
||||
public suspend fun set(name: Name, dataSet: DataSet<T>){
|
||||
//remove previous items
|
||||
remove(name)
|
||||
//Set new items
|
||||
dataSet.flow().collect {
|
||||
set(name + it.name, it.data)
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Append data to node
|
||||
*/
|
||||
public infix fun String.put(data: Data<T>): Unit = set(toName(), data)
|
||||
public suspend infix fun String.put(data: Data<T>): Unit = set(toName(), data)
|
||||
|
||||
/**
|
||||
* Append node
|
||||
*/
|
||||
public suspend infix fun String.put(tree: DataSet<T>): Unit = set(toName(), tree)
|
||||
public suspend infix fun String.put(dataSet: DataSet<T>): Unit = set(toName(), dataSet)
|
||||
|
||||
/**
|
||||
* Build and append node
|
||||
*/
|
||||
public infix fun String.put(block: DataSetBuilder<T>.() -> Unit): Unit = set(toName(), block)
|
||||
public suspend infix fun String.put(block: suspend DataSetBuilder<T>.() -> Unit): Unit = set(toName(), block)
|
||||
}
|
||||
|
||||
private class SubSetBuilder<in T : Any>(private val parent: DataSetBuilder<T>, private val branch: Name) :
|
||||
DataSetBuilder<T> {
|
||||
override suspend fun remove(name: Name) {
|
||||
parent.remove(branch + name)
|
||||
}
|
||||
|
||||
override suspend fun set(name: Name, data: Data<T>?) {
|
||||
parent.set(branch + name, data)
|
||||
}
|
||||
|
||||
override suspend fun set(name: Name, dataSet: DataSet<T>) {
|
||||
parent.set(branch + name, dataSet)
|
||||
}
|
||||
}
|
||||
|
||||
public suspend fun <T : Any> DataSetBuilder<T>.set(name: Name, block: suspend DataSetBuilder<T>.() -> Unit){
|
||||
SubSetBuilder(this,name).apply { block() }
|
||||
}
|
||||
|
||||
|
||||
public operator fun <T : Any> DataSetBuilder<T>.set(name: String, data: Data<T>) {
|
||||
this@set[name.toName()] = data
|
||||
public suspend fun <T : Any> DataSetBuilder<T>.set(name: String, data: Data<T>) {
|
||||
set(name.toName(), data)
|
||||
}
|
||||
|
||||
public fun <T : Any> DataSetBuilder<T>.data(name: Name, data: T, meta: Meta = Meta.EMPTY) {
|
||||
public suspend fun <T : Any> DataSetBuilder<T>.data(name: Name, data: T, meta: Meta = Meta.EMPTY) {
|
||||
set(name, Data.static(data, meta))
|
||||
}
|
||||
|
||||
public fun <T : Any> DataSetBuilder<T>.data(name: Name, data: T, block: MetaBuilder.() -> Unit = {}) {
|
||||
public suspend fun <T : Any> DataSetBuilder<T>.data(name: Name, data: T, block: MetaBuilder.() -> Unit = {}) {
|
||||
set(name, Data.static(data, Meta(block)))
|
||||
}
|
||||
|
||||
public fun <T : Any> DataSetBuilder<T>.data(name: String, data: T, block: MetaBuilder.() -> Unit = {}) {
|
||||
public suspend fun <T : Any> DataSetBuilder<T>.data(name: String, data: T, block: MetaBuilder.() -> Unit = {}) {
|
||||
set(name.toName(), Data.static(data, Meta(block)))
|
||||
}
|
||||
|
||||
@ -55,7 +86,7 @@ public suspend fun <T : Any> DataSetBuilder<T>.set(name: String, set: DataSet<T>
|
||||
this.set(name.toName(), set)
|
||||
}
|
||||
|
||||
public operator fun <T : Any> DataSetBuilder<T>.set(name: String, block: DataSetBuilder<T>.() -> Unit): Unit =
|
||||
public suspend fun <T : Any> DataSetBuilder<T>.set(name: String, block: suspend DataSetBuilder<T>.() -> Unit): Unit =
|
||||
this@set.set(name.toName(), block)
|
||||
|
||||
|
||||
@ -63,9 +94,15 @@ public operator fun <T : Any> DataSetBuilder<T>.set(name: String, block: DataSet
|
||||
* Update data with given node data and meta with node meta.
|
||||
*/
|
||||
@DFExperimental
|
||||
public suspend fun <T: Any> DataSetBuilder<T>.update(tree: DataSet<T>): Unit = coroutineScope{
|
||||
public suspend fun <T : Any> DataSetBuilder<T>.update(tree: DataSet<T>): Unit = coroutineScope {
|
||||
tree.flow().collect {
|
||||
//TODO check if the place is occupied
|
||||
set(it.name, it.data)
|
||||
}
|
||||
}
|
||||
|
||||
public suspend fun <T : Any> DataSetBuilder<T>.collectFrom(flow: Flow<NamedData<T>>) {
|
||||
flow.collect {
|
||||
set(it.name, it.data)
|
||||
}
|
||||
}
|
||||
|
@ -46,12 +46,14 @@ public interface DataTree<out T : Any> : DataSet<T> {
|
||||
|
||||
override fun flow(): Flow<NamedData<T>> = flow {
|
||||
items().forEach { (token, childItem: DataTreeItem<T>) ->
|
||||
if(!token.body.startsWith("@")) {
|
||||
when (childItem) {
|
||||
is DataTreeItem.Leaf -> emit(childItem.data.named(token.asName()))
|
||||
is DataTreeItem.Node -> emitAll(childItem.tree.flow().map { it.named(token + it.name) })
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
override suspend fun listChildren(prefix: Name): List<Name> =
|
||||
getItem(prefix).tree?.items()?.keys?.map { prefix + it } ?: emptyList()
|
||||
|
@ -41,7 +41,7 @@ public interface GroupRule {
|
||||
|
||||
set.flow().collect { data ->
|
||||
val tagValue = data.meta[key]?.string ?: defaultTagValue
|
||||
map.getOrPut(tagValue) { MutableDataTree(dataType, scope) }.set(data.name, data.data)
|
||||
map.getOrPut(tagValue) { MutableDataTree(dataType) }.set(data.name, data.data)
|
||||
}
|
||||
|
||||
return map
|
||||
|
@ -33,16 +33,16 @@ public class MapActionBuilder<T, R>(public var name: Name, public var meta: Meta
|
||||
}
|
||||
|
||||
|
||||
public class MapAction<T : Any, out R : Any>(
|
||||
public class MapAction<in T : Any, out R : Any>(
|
||||
public val outputType: KClass<out R>,
|
||||
private val block: MapActionBuilder<T, R>.() -> Unit,
|
||||
) : Action<T, R> {
|
||||
|
||||
override suspend fun run(
|
||||
set: DataSet<T>,
|
||||
override suspend fun execute(
|
||||
dataSet: DataSet<T>,
|
||||
meta: Meta,
|
||||
scope: CoroutineScope,
|
||||
): DataSet<R> = DataTree.dynamic(outputType, scope) {
|
||||
scope: CoroutineScope?,
|
||||
): DataSet<R> {
|
||||
suspend fun mapOne(data: NamedData<T>): NamedData<R> {
|
||||
// Creating a new environment for action using **old** name, old meta and task meta
|
||||
val env = ActionEnv(data.name, data.meta, meta)
|
||||
@ -65,23 +65,26 @@ public class MapAction<T : Any, out R : Any>(
|
||||
return newData.named(newName)
|
||||
}
|
||||
|
||||
collectFrom(set.flow().map(::mapOne))
|
||||
scope.launch {
|
||||
set.updates.collect { name ->
|
||||
val flow = dataSet.flow().map(::mapOne)
|
||||
|
||||
return DataTree.dynamic(outputType) {
|
||||
collectFrom(flow)
|
||||
scope?.launch {
|
||||
dataSet.updates.collect { name ->
|
||||
//clear old nodes
|
||||
remove(name)
|
||||
//collect new items
|
||||
collectFrom(set.flowChildren(name).map(::mapOne))
|
||||
collectFrom(dataSet.flowChildren(name).map(::mapOne))
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public suspend inline fun <T : Any, reified R : Any> DataSet<T>.map(
|
||||
meta: Meta,
|
||||
updatesScope: CoroutineScope,
|
||||
noinline action: MapActionBuilder<in T, out R>.() -> Unit,
|
||||
): DataSet<R> = MapAction(R::class, action).run(this, meta, updatesScope)
|
||||
|
||||
|
||||
@Suppress("FunctionName")
|
||||
public inline fun <T : Any, reified R : Any> MapAction(
|
||||
noinline builder: MapActionBuilder<T, R>.() -> Unit,
|
||||
): MapAction<T, R> = MapAction(R::class, builder)
|
||||
|
||||
|
||||
|
@ -3,7 +3,7 @@ package hep.dataforge.data
|
||||
import hep.dataforge.meta.*
|
||||
import hep.dataforge.names.*
|
||||
import kotlinx.coroutines.CoroutineScope
|
||||
import kotlinx.coroutines.coroutineScope
|
||||
import kotlinx.coroutines.Job
|
||||
import kotlinx.coroutines.flow.*
|
||||
import kotlinx.coroutines.launch
|
||||
import kotlinx.coroutines.sync.Mutex
|
||||
@ -15,12 +15,13 @@ import kotlin.reflect.KClass
|
||||
*/
|
||||
public class MutableDataTree<T : Any>(
|
||||
override val dataType: KClass<out T>,
|
||||
public val scope: CoroutineScope,
|
||||
) : DataTree<T>, DataSetBuilder<T> {
|
||||
private val mutex = Mutex()
|
||||
private val treeItems = HashMap<NameToken, DataTreeItem<T>>()
|
||||
|
||||
override suspend fun items(): Map<NameToken, DataTreeItem<T>> = mutex.withLock { treeItems }
|
||||
override suspend fun items(): Map<NameToken, DataTreeItem<T>> = mutex.withLock {
|
||||
treeItems.filter { !it.key.body.startsWith("@") }
|
||||
}
|
||||
|
||||
private val _updates = MutableSharedFlow<Name>()
|
||||
|
||||
@ -35,36 +36,37 @@ public class MutableDataTree<T : Any>(
|
||||
}
|
||||
}
|
||||
|
||||
override fun remove(name: Name) {
|
||||
scope.launch {
|
||||
override suspend fun remove(name: Name) {
|
||||
if (name.isEmpty()) error("Can't remove the root node")
|
||||
(getItem(name.cutLast()).tree as? MutableDataTree)?.remove(name.lastOrNull()!!)
|
||||
}
|
||||
}
|
||||
|
||||
private suspend fun set(token: NameToken, node: DataSet<T>) {
|
||||
//if (_map.containsKey(token)) error("Tree entry with name $token is not empty")
|
||||
mutex.withLock {
|
||||
treeItems[token] = DataTreeItem.Node(node.toMutableTree(scope))
|
||||
coroutineScope {
|
||||
node.updates.onEach {
|
||||
_updates.emit(token + it)
|
||||
}.launchIn(this)
|
||||
}
|
||||
_updates.emit(token.asName())
|
||||
}
|
||||
}
|
||||
// private suspend fun set(token: NameToken, node: DataSet<T>) {
|
||||
// //if (_map.containsKey(token)) error("Tree entry with name $token is not empty")
|
||||
// mutex.withLock {
|
||||
// treeItems[token] = DataTreeItem.Node(node.toMutableTree())
|
||||
// coroutineScope {
|
||||
// node.updates.onEach {
|
||||
// _updates.emit(token + it)
|
||||
// }.launchIn(this)
|
||||
// }
|
||||
// _updates.emit(token.asName())
|
||||
// }
|
||||
// }
|
||||
|
||||
private suspend fun set(token: NameToken, data: Data<T>) {
|
||||
mutex.withLock {
|
||||
treeItems[token] = DataTreeItem.Leaf(data)
|
||||
_updates.emit(token.asName())
|
||||
}
|
||||
}
|
||||
|
||||
private suspend fun getOrCreateNode(token: NameToken): MutableDataTree<T> =
|
||||
(treeItems[token] as? DataTreeItem.Node<T>)?.tree as? MutableDataTree<T>
|
||||
?: MutableDataTree(dataType, scope).also { set(token, it) }
|
||||
?: MutableDataTree(dataType).also {
|
||||
mutex.withLock {
|
||||
treeItems[token] = DataTreeItem.Node(it)
|
||||
}
|
||||
}
|
||||
|
||||
private suspend fun getOrCreateNode(name: Name): MutableDataTree<T> {
|
||||
return when (name.length) {
|
||||
@ -74,83 +76,53 @@ public class MutableDataTree<T : Any>(
|
||||
}
|
||||
}
|
||||
|
||||
override fun set(name: Name, data: Data<T>?) {
|
||||
override suspend fun set(name: Name, data: Data<T>?) {
|
||||
if (data == null) {
|
||||
remove(name)
|
||||
} else {
|
||||
scope.launch {
|
||||
when (name.length) {
|
||||
0 -> error("Can't add data with empty name")
|
||||
1 -> set(name.firstOrNull()!!, data)
|
||||
2 -> getOrCreateNode(name.cutLast()).set(name.lastOrNull()!!, data)
|
||||
}
|
||||
}
|
||||
}
|
||||
_updates.emit(name)
|
||||
}
|
||||
|
||||
private suspend fun setTree(name: Name, node: MutableDataTree<out T>) {
|
||||
when (name.length) {
|
||||
0 -> error("Can't add data with empty name")
|
||||
1 -> set(name.firstOrNull()!!, node)
|
||||
2 -> getOrCreateNode(name.cutLast()).set(name.lastOrNull()!!, node)
|
||||
/**
|
||||
* Copy given data set and mirror its changes to this [MutableDataTree] in [this@setAndObserve]. Returns an update [Job]
|
||||
*/
|
||||
public fun CoroutineScope.setAndObserve(name: Name, dataSet: DataSet<T>): Job = launch {
|
||||
set(name, dataSet)
|
||||
dataSet.updates.collect { nameInBranch ->
|
||||
set(name + nameInBranch, dataSet.getData(nameInBranch))
|
||||
}
|
||||
}
|
||||
|
||||
override suspend fun set(name: Name, dataSet: DataSet<T>): Unit {
|
||||
if (dataSet is MutableDataTree) {
|
||||
setTree(name, dataSet)
|
||||
} else {
|
||||
setTree(name, dataSet.toMutableTree(scope))
|
||||
}
|
||||
}
|
||||
|
||||
override fun set(name: Name, block: DataSetBuilder<T>.() -> Unit) {
|
||||
scope.launch {
|
||||
setTree(name, MutableDataTree(dataType, scope).apply(block))
|
||||
}
|
||||
}
|
||||
|
||||
public fun collectFrom(flow: Flow<NamedData<T>>) {
|
||||
flow.onEach {
|
||||
set(it.name, it.data)
|
||||
}.launchIn(scope)
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a dynamic tree. Initial data is placed synchronously. Updates are propagated via [updatesScope]
|
||||
*/
|
||||
public suspend fun <T : Any> DataTree.Companion.dynamic(
|
||||
type: KClass<out T>,
|
||||
updatesScope: CoroutineScope,
|
||||
block: suspend MutableDataTree<T>.() -> Unit,
|
||||
): DataTree<T> {
|
||||
val tree = MutableDataTree(type, updatesScope)
|
||||
val tree = MutableDataTree(type)
|
||||
tree.block()
|
||||
return tree
|
||||
}
|
||||
|
||||
public suspend inline fun <reified T : Any> DataTree.Companion.dynamic(
|
||||
updatesScope: CoroutineScope,
|
||||
crossinline block: suspend MutableDataTree<T>.() -> Unit,
|
||||
): DataTree<T> = MutableDataTree(T::class, updatesScope).apply { block() }
|
||||
): DataTree<T> = MutableDataTree(T::class).apply { block() }
|
||||
|
||||
|
||||
public suspend inline fun <reified T : Any> MutableDataTree<T>.set(
|
||||
name: Name,
|
||||
noinline block: suspend MutableDataTree<T>.() -> Unit,
|
||||
): Unit = set(name, DataTree.dynamic(T::class, scope, block))
|
||||
): Unit = set(name, DataTree.dynamic(T::class, block))
|
||||
|
||||
public suspend inline fun <reified T : Any> MutableDataTree<T>.set(
|
||||
name: String,
|
||||
noinline block: suspend MutableDataTree<T>.() -> Unit,
|
||||
): Unit = set(name.toName(), DataTree.dynamic(T::class, scope, block))
|
||||
|
||||
/**
|
||||
* Generate a mutable builder from this node. Node content is not changed
|
||||
*/
|
||||
public suspend fun <T : Any> DataSet<T>.toMutableTree(
|
||||
scope: CoroutineScope,
|
||||
): MutableDataTree<T> = MutableDataTree(dataType, scope).apply {
|
||||
flow().collect { set(it.name, it.data) }
|
||||
this@toMutableTree.updates.onEach {
|
||||
set(it, getData(it))
|
||||
}.launchIn(scope)
|
||||
}
|
||||
): Unit = set(name.toName(), DataTree.dynamic(T::class, block))
|
||||
|
@ -13,7 +13,6 @@ import kotlin.collections.set
|
||||
import kotlin.reflect.KClass
|
||||
|
||||
|
||||
|
||||
public class SplitBuilder<T : Any, R : Any>(public val name: Name, public val meta: Meta) {
|
||||
|
||||
public class FragmentRule<T : Any, R : Any>(public val name: Name, public var meta: MetaBuilder) {
|
||||
@ -44,11 +43,11 @@ public class SplitAction<T : Any, R : Any>(
|
||||
private val action: SplitBuilder<T, R>.() -> Unit,
|
||||
) : Action<T, R> {
|
||||
|
||||
override suspend fun run(
|
||||
set: DataSet<T>,
|
||||
override suspend fun execute(
|
||||
dataSet: DataSet<T>,
|
||||
meta: Meta,
|
||||
scope: CoroutineScope,
|
||||
): DataSet<R> = DataTree.dynamic(outputType, scope) {
|
||||
scope: CoroutineScope?,
|
||||
): DataSet<R> {
|
||||
|
||||
suspend fun splitOne(data: NamedData<T>): Flow<NamedData<R>> {
|
||||
val laminate = Laminate(data.meta, meta)
|
||||
@ -63,13 +62,15 @@ public class SplitAction<T : Any, R : Any>(
|
||||
}
|
||||
}
|
||||
|
||||
collectFrom(set.flow().flatMapConcat(transform = ::splitOne))
|
||||
scope.launch {
|
||||
set.updates.collect { name ->
|
||||
return DataTree.dynamic(outputType) {
|
||||
collectFrom(dataSet.flow().flatMapConcat(transform = ::splitOne))
|
||||
scope?.launch {
|
||||
dataSet.updates.collect { name ->
|
||||
//clear old nodes
|
||||
remove(name)
|
||||
//collect new items
|
||||
collectFrom(set.flowChildren(name).flatMapConcat(transform = ::splitOne))
|
||||
collectFrom(dataSet.flowChildren(name).flatMapConcat(transform = ::splitOne))
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -5,7 +5,8 @@ import kotlinx.coroutines.coroutineScope
|
||||
import kotlinx.coroutines.flow.*
|
||||
import kotlin.reflect.KClass
|
||||
|
||||
private class StaticDataTree<T : Any>(
|
||||
@PublishedApi
|
||||
internal class StaticDataTree<T : Any>(
|
||||
override val dataType: KClass<out T>,
|
||||
) : DataSetBuilder<T>, DataTree<T> {
|
||||
|
||||
@ -13,9 +14,9 @@ private class StaticDataTree<T : Any>(
|
||||
|
||||
override val updates: Flow<Name> = emptyFlow()
|
||||
|
||||
override suspend fun items(): Map<NameToken, DataTreeItem<T>> = items
|
||||
override suspend fun items(): Map<NameToken, DataTreeItem<T>> = items.filter { !it.key.body.startsWith("@") }
|
||||
|
||||
override fun remove(name: Name) {
|
||||
override suspend fun remove(name: Name) {
|
||||
when (name.length) {
|
||||
0 -> error("Can't remove root tree node")
|
||||
1 -> items.remove(name.firstOrNull()!!)
|
||||
@ -34,7 +35,7 @@ private class StaticDataTree<T : Any>(
|
||||
else -> getOrCreateNode(name.cutLast()).getOrCreateNode(name.lastOrNull()!!.asName())
|
||||
}
|
||||
|
||||
private operator fun set(name: Name, item: DataTreeItem<T>?) {
|
||||
private suspend fun set(name: Name, item: DataTreeItem<T>?) {
|
||||
if (name.isEmpty()) error("Can't set top level tree node")
|
||||
if (item == null) {
|
||||
remove(name)
|
||||
@ -43,7 +44,7 @@ private class StaticDataTree<T : Any>(
|
||||
}
|
||||
}
|
||||
|
||||
override fun set(name: Name, data: Data<T>?) {
|
||||
override suspend fun set(name: Name, data: Data<T>?) {
|
||||
set(name, data?.let { DataTreeItem.Leaf(it) })
|
||||
}
|
||||
|
||||
@ -58,20 +59,15 @@ private class StaticDataTree<T : Any>(
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
override fun set(name: Name, block: DataSetBuilder<T>.() -> Unit) {
|
||||
val tree = StaticDataTree(dataType).apply(block)
|
||||
set(name, DataTreeItem.Node(tree))
|
||||
}
|
||||
}
|
||||
|
||||
public fun <T : Any> DataTree.Companion.static(
|
||||
public suspend fun <T : Any> DataTree.Companion.static(
|
||||
dataType: KClass<out T>,
|
||||
block: DataSetBuilder<T>.() -> Unit,
|
||||
): DataTree<T> = StaticDataTree(dataType).apply(block)
|
||||
block: suspend DataSetBuilder<T>.() -> Unit,
|
||||
): DataTree<T> = StaticDataTree(dataType).apply { block() }
|
||||
|
||||
public inline fun <reified T : Any> DataTree.Companion.static(
|
||||
noinline block: DataSetBuilder<T>.() -> Unit,
|
||||
public suspend inline fun <reified T : Any> DataTree.Companion.static(
|
||||
noinline block: suspend DataSetBuilder<T>.() -> Unit,
|
||||
): DataTree<T> = static(T::class, block)
|
||||
|
||||
public suspend fun <T : Any> DataSet<T>.toStaticTree(): DataTree<T> = StaticDataTree(dataType).apply {
|
||||
|
@ -12,9 +12,9 @@ public suspend fun DataSet<*>.getMeta(): Meta? = getData(DataSet.META_KEY)?.meta
|
||||
/**
|
||||
* Add meta-data node to a [DataSet]
|
||||
*/
|
||||
public fun DataSetBuilder<*>.meta(meta: Meta): Unit = set(DataSet.META_KEY, Data.empty(meta))
|
||||
public suspend fun DataSetBuilder<*>.meta(meta: Meta): Unit = set(DataSet.META_KEY, Data.empty(meta))
|
||||
|
||||
/**
|
||||
* Add meta-data node to a [DataSet]
|
||||
*/
|
||||
public fun DataSetBuilder<*>.meta(metaBuilder: MetaBuilder.() -> Unit): Unit = meta(Meta(metaBuilder))
|
||||
public suspend fun DataSetBuilder<*>.meta(metaBuilder: MetaBuilder.() -> Unit): Unit = meta(Meta(metaBuilder))
|
@ -0,0 +1,39 @@
|
||||
package hep.dataforge.data
|
||||
|
||||
import kotlinx.coroutines.runBlocking
|
||||
import org.junit.jupiter.api.Test
|
||||
import kotlin.test.assertEquals
|
||||
|
||||
class ActionsTest {
|
||||
val data: DataTree<Int> = runBlocking {
|
||||
DataTree.static {
|
||||
repeat(10) {
|
||||
data(it.toString(), it)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun testStaticMapAction() {
|
||||
val plusOne = MapAction<Int, Int> {
|
||||
result { it + 1 }
|
||||
}
|
||||
runBlocking {
|
||||
val result = plusOne.execute(data)
|
||||
assertEquals(2, result.getData("1")?.value())
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun testDynamicMapAction() {
|
||||
val plusOne = MapAction<Int, Int> {
|
||||
result { it + 1 }
|
||||
}
|
||||
val datum = runBlocking {
|
||||
val result = plusOne.execute(data, scope = this)
|
||||
result.getData("1")?.value()
|
||||
}
|
||||
assertEquals(2, datum)
|
||||
}
|
||||
|
||||
}
|
@ -1,34 +1,69 @@
|
||||
package hep.dataforge.data
|
||||
|
||||
import kotlinx.coroutines.runBlocking
|
||||
import hep.dataforge.names.toName
|
||||
import kotlinx.coroutines.*
|
||||
import kotlinx.coroutines.flow.collect
|
||||
import kotlin.test.Test
|
||||
import kotlin.test.assertTrue
|
||||
import kotlin.test.assertEquals
|
||||
|
||||
|
||||
internal class DataTreeBuilderTest{
|
||||
internal class DataTreeBuilderTest {
|
||||
@Test
|
||||
fun testDataUpdate(){
|
||||
val updateData: DataTree<Any> = DataTree.static{
|
||||
fun testDataUpdate() = runBlocking {
|
||||
val updateData: DataTree<Any> = DataTree.static {
|
||||
"update" put {
|
||||
"a" put Data.static("a")
|
||||
"b" put Data.static("b")
|
||||
}
|
||||
}
|
||||
|
||||
val node = DataTree.static<Any>{
|
||||
set("primary"){
|
||||
data("a","a")
|
||||
data("b","b")
|
||||
val node = DataTree.static<Any> {
|
||||
set("primary") {
|
||||
data("a", "a")
|
||||
data("b", "b")
|
||||
}
|
||||
data("root","root")
|
||||
runBlocking {
|
||||
data("root", "root")
|
||||
update(updateData)
|
||||
}
|
||||
|
||||
|
||||
assertEquals("a", node.getData("update.a")?.value())
|
||||
assertEquals("a", node.getData("primary.a")?.value())
|
||||
}
|
||||
|
||||
@Test
|
||||
fun testDynamicUpdates() = runBlocking {
|
||||
try {
|
||||
supervisorScope {
|
||||
val subNode = DataTree.dynamic<Int> {
|
||||
launch {
|
||||
repeat(10) {
|
||||
delay(10)
|
||||
data("value", it)
|
||||
}
|
||||
}
|
||||
}
|
||||
launch {
|
||||
subNode.updatesWithData.collect {
|
||||
println(it)
|
||||
}
|
||||
}
|
||||
val rootNode = DataTree.dynamic<Int> {
|
||||
setAndObserve("sub".toName(), subNode)
|
||||
}
|
||||
|
||||
assertTrue { node.branch("update.a") != null }
|
||||
assertTrue { node.branch("primary.a") != null }
|
||||
launch {
|
||||
rootNode.updatesWithData.collect {
|
||||
println(it)
|
||||
}
|
||||
}
|
||||
delay(200)
|
||||
assertEquals(9, rootNode.getData("sub.value")?.value())
|
||||
cancel()
|
||||
}
|
||||
} catch (t: Throwable) {
|
||||
if (t !is CancellationException) throw t
|
||||
}
|
||||
|
||||
}
|
||||
}
|
@ -193,7 +193,7 @@ public fun Name.endsWith(name: Name): Boolean =
|
||||
* if [this] starts with given [head] name, returns the reminder of the name (could be empty). Otherwise returns null
|
||||
*/
|
||||
public fun Name.removeHeadOrNull(head: Name): Name? = if (startsWith(head)) {
|
||||
Name(tokens.subList(head.length, head.length))
|
||||
Name(tokens.subList(head.length, length))
|
||||
} else {
|
||||
null
|
||||
}
|
@ -3,13 +3,12 @@ package hep.dataforge.workspace
|
||||
import hep.dataforge.data.DataSet
|
||||
import hep.dataforge.meta.Meta
|
||||
import hep.dataforge.meta.descriptors.Described
|
||||
import hep.dataforge.misc.Named
|
||||
import hep.dataforge.misc.Type
|
||||
import hep.dataforge.workspace.Task.Companion.TYPE
|
||||
import kotlin.reflect.KClass
|
||||
|
||||
@Type(TYPE)
|
||||
public interface Task<out R : Any> : Named, Described {
|
||||
public interface Task<out R : Any> : Described {
|
||||
|
||||
/**
|
||||
* The explicit type of the node returned by the task
|
||||
|
@ -104,12 +104,11 @@ public fun TaskDependencyContainer.data(
|
||||
pattern: String? = null,
|
||||
from: String? = null,
|
||||
to: String? = null,
|
||||
): DataDependency =
|
||||
data {
|
||||
): DataDependency = data {
|
||||
pattern?.let { this.pattern = it }
|
||||
from?.let { this.from = it }
|
||||
to?.let { this.to = it }
|
||||
}
|
||||
}
|
||||
|
||||
///**
|
||||
// * Add all data as root node
|
||||
|
@ -35,14 +35,7 @@ public class TaskBuilder<R : Any>(private val name: Name, public val type: KClas
|
||||
public val meta: Meta,
|
||||
public val context: Context,
|
||||
public val data: DataSet<Any>,
|
||||
) {
|
||||
// public operator fun <T : Any> DirectTaskDependency<T>.invoke(): DataSet<T> = if (placement.isEmpty()) {
|
||||
// data.cast(task.type)
|
||||
// } else {
|
||||
// data[placement].tree?.cast(task.type)
|
||||
// ?: error("Could not find results of direct task dependency $this at \"$placement\"")
|
||||
// }
|
||||
}
|
||||
)
|
||||
|
||||
/**
|
||||
* Add a transformation on untyped data
|
||||
@ -86,7 +79,7 @@ public class TaskBuilder<R : Any>(private val name: Name, public val type: KClas
|
||||
crossinline block: TaskEnv.() -> Action<T, R>,
|
||||
) {
|
||||
transform { data: DataSet<T> ->
|
||||
block().run(data, meta, context)
|
||||
block().execute(data, meta, context)
|
||||
}
|
||||
}
|
||||
|
||||
@ -194,15 +187,19 @@ public class TaskBuilder<R : Any>(private val name: Name, public val type: KClas
|
||||
logger.warn { "No transformation present, returning input data" }
|
||||
dataSet.castOrNull(type) ?: error("$type expected, but $type received")
|
||||
} else {
|
||||
val builder = MutableDataTree(type, workspace.context)
|
||||
DataTree.dynamic(type, workspace.context){
|
||||
dataTransforms.forEach { transformation ->
|
||||
val res = transformation(workspace.context, model, dataSet)
|
||||
builder.update(res)
|
||||
update(res)
|
||||
}
|
||||
}
|
||||
builder
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@DFExperimental
|
||||
public suspend inline fun <reified T : Any> TaskBuilder.TaskEnv.dataTree(
|
||||
crossinline block: suspend MutableDataTree<T>.() -> Unit,
|
||||
): DataTree<T> = DataTree.dynamic(context, block)
|
@ -37,6 +37,8 @@ public fun WorkspaceBuilder.target(name: String, block: MetaBuilder.() -> Unit)
|
||||
targets[name] = Meta(block).seal()
|
||||
}
|
||||
|
||||
class WorkspaceTask(val workspace: Workspace, val name: String)
|
||||
|
||||
/**
|
||||
* Use existing target as a base updating it with the block
|
||||
*/
|
||||
@ -52,7 +54,7 @@ public fun <T : Any> WorkspaceBuilder.task(
|
||||
name: String,
|
||||
type: KClass<out T>,
|
||||
builder: TaskBuilder<T>.() -> Unit,
|
||||
): Task<T> = TaskBuilder(name.toName(), type).apply(builder).build().also { tasks.add(it) }
|
||||
): WorkspaceTask = TaskBuilder(name.toName(), type).apply(builder).build().also { tasks.add(it) }
|
||||
|
||||
public inline fun <reified T : Any> WorkspaceBuilder.task(
|
||||
name: String,
|
||||
|
@ -33,7 +33,7 @@ class DataPropagationTestPlugin : WorkspacePlugin() {
|
||||
|
||||
val testSingleData = task("singleData", Int::class) {
|
||||
model {
|
||||
data("myData\\[12\\]")
|
||||
data(pattern = "myData\\[12\\]")
|
||||
}
|
||||
transform<Int> { data ->
|
||||
DataTree.dynamic(context) {
|
||||
|
@ -85,7 +85,7 @@ class SimpleWorkspaceTest {
|
||||
transform<Int> { data ->
|
||||
val squareNode = data.branch("square").filterIsInstance<Int>() //squareDep()
|
||||
val linearNode = data.branch("linear").filterIsInstance<Int>() //linearDep()
|
||||
DataTree.dynamic<Int>(context) {
|
||||
dataTree {
|
||||
squareNode.flow().collect {
|
||||
val newData: Data<Int> = Data {
|
||||
val squareValue = squareNode.getData(it.name)!!.value()
|
||||
@ -150,6 +150,7 @@ class SimpleWorkspaceTest {
|
||||
}
|
||||
|
||||
@Test
|
||||
@Timeout(1)
|
||||
fun testWorkspace() {
|
||||
val node = workspace.runBlocking("sum")
|
||||
val res = node.first()
|
||||
@ -157,7 +158,7 @@ class SimpleWorkspaceTest {
|
||||
}
|
||||
|
||||
@Test
|
||||
@Timeout(400)
|
||||
@Timeout(1)
|
||||
fun testMetaPropagation() {
|
||||
val node = workspace.runBlocking("sum") { "testFlag" put true }
|
||||
val res = node.first()?.value()
|
||||
@ -179,7 +180,7 @@ class SimpleWorkspaceTest {
|
||||
}
|
||||
|
||||
@Test
|
||||
fun testGather() {
|
||||
fun testFilter() {
|
||||
val node = workspace.runBlocking("filterOne")
|
||||
runBlocking {
|
||||
assertEquals(12, node.first()?.value())
|
||||
|
Loading…
Reference in New Issue
Block a user