commonMain.jetbrains.datalore.plot.builder.assemble.StitchedPlotLayers.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) 2019. 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.assemble

import jetbrains.datalore.base.gcommon.collect.ClosedRange
import jetbrains.datalore.plot.base.Aes
import jetbrains.datalore.plot.base.DataFrame
import jetbrains.datalore.plot.base.Scale
import jetbrains.datalore.plot.base.aes.AestheticsDefaults
import jetbrains.datalore.plot.base.render.LegendKeyElementFactory
import jetbrains.datalore.plot.builder.VarBinding
import jetbrains.datalore.plot.common.data.SeriesUtil

internal class StitchedPlotLayers(layers: List) {
    private val myLayers: List

    val legendKeyElementFactory: LegendKeyElementFactory
        get() {
            check(myLayers.isNotEmpty())
            return myLayers[0].legendKeyElementFactory
        }

    val aestheticsDefaults: AestheticsDefaults
        get() {
            check(myLayers.isNotEmpty())
            return myLayers[0].aestheticsDefaults
        }

    val isLegendDisabled: Boolean
        get() {
            check(myLayers.isNotEmpty())
            return myLayers[0].isLegendDisabled
        }

    init {
        myLayers = ArrayList(layers)
    }

    fun renderedAes(): List> {
        return if (myLayers.isEmpty()) {
            emptyList()
        } else myLayers[0].renderedAes()
    }

    fun hasBinding(aes: Aes<*>): Boolean {
        return myLayers.isNotEmpty() && myLayers[0].hasBinding(aes)
    }

    fun hasConstant(aes: Aes<*>): Boolean {
        return myLayers.isNotEmpty() && myLayers[0].hasConstant(aes)
    }

    fun  getConstant(aes: Aes): T {
        check(myLayers.isNotEmpty())
        return myLayers[0].getConstant(aes)
    }

    fun getBinding(aes: Aes<*>): VarBinding {
        check(myLayers.isNotEmpty())
        return myLayers[0].getBinding(aes)
    }

    fun getScale(aes: Aes<*>): Scale<*> {
        check(myLayers.isNotEmpty())
        return myLayers[0].scaleMap[aes]
    }

    fun getScaleMap(): TypedScaleMap {
        check(myLayers.isNotEmpty())
        return myLayers[0].scaleMap
    }

    fun getDataRange(variable: DataFrame.Variable): ClosedRange? {
        check(isNumericData(variable)) { "Not numeric data [$variable]" }
        var result: ClosedRange? = null
        for (layer in myLayers) {
            val range = layer.dataFrame.range(variable)
            result = SeriesUtil.span(result, range)
        }
        return result
    }

    private fun isNumericData(variable: DataFrame.Variable): Boolean {
        check(myLayers.isNotEmpty())
        for (layer in myLayers) {
            if (!layer.dataFrame.isNumeric(variable)) {
                return false
            }
        }
        return true
    }
}