fuookami.ospf.kotlin.framework.model.Pipeline.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ospf-kotlin-framework Show documentation

There is a newer version: 1.0.31

Show newest version

package fuookami.ospf.kotlin.framework.model

import fuookami.ospf.kotlin.utils.math.*
import fuookami.ospf.kotlin.utils.error.*
import fuookami.ospf.kotlin.utils.functional.*
import fuookami.ospf.kotlin.core.frontend.model.*
import fuookami.ospf.kotlin.core.frontend.model.mechanism.*

interface Pipeline {
    val name: String

    fun register(model: M) {
        if (model is MetaModel) {
            model.registerConstraintGroup(name)
        }
    }

    operator fun invoke(model: M): Try
}

interface CGPipeline<
    in Args : Any,
    in Model : MetaModel,
    in Map : AbstractShadowPriceMap
> : Pipeline {
    fun extractor(): ShadowPriceExtractor<@UnsafeVariance Args, @UnsafeVariance Map>? {
        return null
    }

    fun refresh(map: Map, model: Model, shadowPrices: List): Try {
        return ok
    }
}

interface HAPipeline : Pipeline {
    data class Obj(
        val tag: String,
        val value: Flt64
    )

    override operator fun invoke(model: M): Try = ok

    operator fun invoke(model: M, solution: List): Ret =
        when (val obj = calculate(model, solution)) {
            is Ok -> if (obj.value != null) {
                Ok(Obj(this.name, obj.value!!))
            } else {
                Failed(Err(ErrorCode.ORSolutionInvalid, this.name))
            }

            is Failed -> Failed(obj.error)
        }

    fun calculate(model: M, solution: List): Ret

    fun check(model: M, solution: List): Try = when (val obj = calculate(model, solution)) {
        is Ok -> if (obj.value != null) {
            ok
        } else {
            Failed(Err(ErrorCode.ORSolutionInvalid, this.name))
        }

        is Failed -> Failed(obj.error)
    }
}

typealias PipelineList = List>

operator fun  PipelineList.invoke(model: M): Try {
    for (pipeline in this) {
        pipeline.register(model)
        when (val ret = pipeline(model)) {
            is Ok -> {}
            is Failed -> {
                return Failed(ret.error)
            }
        }
    }
    return ok
}

typealias CGPipelineList = List>
typealias HAPipelineList = List>