commonMain.jetbrains.datalore.plot.builder.data.YOrientationUtil.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of lets-plot-common Show documentation

Lets-Plot JVM package without rendering part

There is a newer version: 4.5.3-alpha1

/*
 * Copyright (c) 2022. JetBrains s.r.o.
 * Use of this source code is governed by the MIT license that can be found in the LICENSE file.
 */

package jetbrains.datalore.plot.builder.data

import jetbrains.datalore.plot.base.Aes
import jetbrains.datalore.plot.base.DataFrame
import jetbrains.datalore.plot.base.data.TransformVar
import jetbrains.datalore.plot.base.util.YOrientationBaseUtil
import jetbrains.datalore.plot.builder.VarBinding

object YOrientationUtil {
    fun flipDataFrame(data: DataFrame): DataFrame {
        val positionalTransformVars = data.variables()
            .filter { it.isTransform }
            .associateBy { TransformVar.toAes(it) }
            .filterKeys { Aes.isPositionalXY(it) }
            .values


        // Clean target data builder.
        var toDataBuilder: DataFrame.Builder = data.builder()
        for (transformVar in positionalTransformVars) {
            toDataBuilder.remove(transformVar)
        }

        // Update positional transform vars.
        for (transformVar in positionalTransformVars) {
            val aes = TransformVar.toAes(transformVar)
            val flippedAes = YOrientationBaseUtil.flipAes(aes)
            val toVar = TransformVar.forAes(flippedAes)
            val serie = data.getNumeric(transformVar)
            toDataBuilder.putNumeric(toVar, serie)
        }

        return toDataBuilder.build()
    }

    fun flipVarBinding(bindings: List): List {
        return bindings.map {
            if (Aes.isPositionalXY(it.aes)) {
                val flippedAes = YOrientationBaseUtil.flipAes(it.aes)
                VarBinding(
                    it.variable,
                    flippedAes
                )
            } else {
                it
            }
        }
    }
}