kscience/kmath
9
0
Fork 1
Code Issues Pull Requests 6 Packages Projects Releases 16 Wiki Activity

Simplify asm.kt

Browse Source
This commit is contained in:
Iaroslav Postovalov 2020-09-25 20:49:08 +07:00
parent 1e50587da4
commit 8014b3df0b
No known key found for this signature in database
GPG Key ID: 46E15E4A31B3BCD7
1 changed files with 27 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
kmath-ast/src/jvmMain/kotlin/kscience/kmath/asm/asm.kt
View File

@ -19,39 +19,37 @@ import kotlin.reflect.KClass
* @author Alexander Nozik * @author Alexander Nozik
*/ */
public fun <T : Any> MST.compileWith(type: KClass<T>, algebra: Algebra<T>): Expression<T> { public fun <T : Any> MST.compileWith(type: KClass<T>, algebra: Algebra<T>): Expression<T> {
fun AsmBuilder<T>.visit(node: MST) { fun AsmBuilder<T>.visit(node: MST): Unit = when (node) {
when (node) { is MST.Symbolic -> {
is MST.Symbolic -> { val symbol = try {
val symbol = try { algebra.symbol(node.value)
algebra.symbol(node.value) } catch (ignored: Throwable) {
} catch (ignored: Throwable) { null
null
}
if (symbol != null)
loadTConstant(symbol)
else
loadVariable(node.value)
} }
is MST.Numeric -> loadNumeric(node.value) if (symbol != null)
loadTConstant(symbol)
else
loadVariable(node.value)
}
is MST.Unary -> buildAlgebraOperationCall( is MST.Numeric -> loadNumeric(node.value)
context = algebra,
name = node.operation,
fallbackMethodName = "unaryOperation",
parameterTypes = arrayOf(MstType.fromMst(node.value))
) { visit(node.value) }
is MST.Binary -> buildAlgebraOperationCall( is MST.Unary -> buildAlgebraOperationCall(
context = algebra, context = algebra,
name = node.operation, name = node.operation,
fallbackMethodName = "binaryOperation", fallbackMethodName = "unaryOperation",
parameterTypes = arrayOf(MstType.fromMst(node.left), MstType.fromMst(node.right)) parameterTypes = arrayOf(MstType.fromMst(node.value))
) { ) { visit(node.value) }
visit(node.left)
visit(node.right) is MST.Binary -> buildAlgebraOperationCall(
} context = algebra,
name = node.operation,
fallbackMethodName = "binaryOperation",
parameterTypes = arrayOf(MstType.fromMst(node.left), MstType.fromMst(node.right))
) {
visit(node.left)
visit(node.right)
} }
} }