io.github.danielTucano.matplotlib.np.meshgrid.kt Maven / Gradle / Ivy

Go to download
package io.github.danielTucano.matplotlib.np

import io.github.danielTucano.python.PythonScriptBuilder

fun meshgrid(xMeshValues: NPVar, yMeshValues: NPVar): Pair {
    val gridPair = Pair(
        object : NPVar {
            override val variableName: String = "np_var_${NPVar.npVarNumber}"
        },
        object : NPVar {
            override val variableName: String = "np_var_${NPVar.npVarNumber}"
        }
    )
    PythonScriptBuilder.addCommand(
        "${gridPair.first.variableName}, ${gridPair.second.variableName} =" +
                " np.meshgrid(${xMeshValues.variableName},${yMeshValues.variableName})"
    )

    return gridPair
}

fun meshgrid(xMeshValues: NPVar, yMeshValues: NPVar, zMeshValues: NPVar): Triple {
    val gridTriple = Triple(
        object : NPVar {
            override val variableName: String = "np_var_${NPVar.npVarNumber}"
        },
        object : NPVar {
            override val variableName: String = "np_var_${NPVar.npVarNumber}"
        },
        object : NPVar {
            override val variableName: String = "np_var_${NPVar.npVarNumber}"
        }
    )
    PythonScriptBuilder.addCommand(
        "${gridTriple.first.variableName}, ${gridTriple.second.variableName}, ${gridTriple.third.variableName} =" +
                " np.meshgrid(${xMeshValues.variableName},${yMeshValues.variableName},${zMeshValues.variableName})"
    )

    return gridTriple
}