Add more fine-grained converters from MST to SVar and SConst
This commit is contained in:
parent
84f7535fdd
commit
5de9d69237
@ -8,8 +8,8 @@ import kscience.kmath.operations.*
|
|||||||
/**
|
/**
|
||||||
* Maps [SFun] objects to [MST]. Some unsupported operations like [Derivative] are bound and converted then.
|
* Maps [SFun] objects to [MST]. Some unsupported operations like [Derivative] are bound and converted then.
|
||||||
*
|
*
|
||||||
* @receiver a scalar function.
|
* @receiver the scalar function.
|
||||||
* @return the [MST].
|
* @return a node.
|
||||||
*/
|
*/
|
||||||
public fun <X : SFun<X>> SFun<X>.mst(): MST = MstExtendedField {
|
public fun <X : SFun<X>> SFun<X>.mst(): MST = MstExtendedField {
|
||||||
when (this@mst) {
|
when (this@mst) {
|
||||||
@ -30,16 +30,33 @@ public fun <X : SFun<X>> SFun<X>.mst(): MST = MstExtendedField {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Maps [MST.Numeric] to [SConst] directly.
|
||||||
|
*
|
||||||
|
* @receiver the node.
|
||||||
|
* @return a new constant.
|
||||||
|
*/
|
||||||
|
public fun <X : SFun<X>> MST.Numeric.sconst(): SConst<X> = SConst(value)
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Maps [MST.Symbolic] to [SVar] directly.
|
||||||
|
*
|
||||||
|
* @receiver the node.
|
||||||
|
* @param proto the prototype instance.
|
||||||
|
* @return a new variable.
|
||||||
|
*/
|
||||||
|
public fun <X : SFun<X>> MST.Symbolic.svar(proto: X): SVar<X> = SVar(proto, value)
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Maps [MST] objects to [SFun]. Unsupported operations throw [IllegalStateException].
|
* Maps [MST] objects to [SFun]. Unsupported operations throw [IllegalStateException].
|
||||||
*
|
*
|
||||||
* @receiver an [MST].
|
* @receiver the node.
|
||||||
* @return the scalar function.
|
* @param proto the prototype instance.
|
||||||
|
* @return a scalar function.
|
||||||
*/
|
*/
|
||||||
public fun <X : SFun<X>> MST.sfun(proto: X): SFun<X> {
|
public fun <X : SFun<X>> MST.sfun(proto: X): SFun<X> = when (this) {
|
||||||
return when (this) {
|
is MST.Numeric -> sconst()
|
||||||
is MST.Numeric -> SConst(value)
|
is MST.Symbolic -> svar(proto)
|
||||||
is MST.Symbolic -> SVar(proto, value)
|
|
||||||
|
|
||||||
is MST.Unary -> when (operation) {
|
is MST.Unary -> when (operation) {
|
||||||
SpaceOperations.PLUS_OPERATION -> value.sfun(proto)
|
SpaceOperations.PLUS_OPERATION -> value.sfun(proto)
|
||||||
@ -62,4 +79,3 @@ public fun <X : SFun<X>> MST.sfun(proto: X): SFun<X> {
|
|||||||
else -> error("Binary operation $operation not defined in $this")
|
else -> error("Binary operation $operation not defined in $this")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
Loading…
Reference in New Issue
Block a user