commonMain.jetbrains.datalore.plot.config.GeomProto.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.config

import jetbrains.datalore.base.gcommon.base.Preconditions.checkArgument
import jetbrains.datalore.plot.base.Aes
import jetbrains.datalore.plot.base.GeomKind
import jetbrains.datalore.plot.base.GeomKind.*
import jetbrains.datalore.plot.base.GeomMeta
import jetbrains.datalore.plot.base.pos.PositionAdjustments
import jetbrains.datalore.plot.builder.assemble.PosProvider
import jetbrains.datalore.plot.builder.assemble.geom.DefaultSampling
import jetbrains.datalore.plot.builder.sampling.Sampling
import jetbrains.datalore.plot.builder.sampling.Samplings
import jetbrains.datalore.plot.config.Option.Meta

open class GeomProto constructor(val geomKind: GeomKind) {

    fun defaultOptions(): Map {
        checkArgument(DEFAULTS.containsKey(geomKind), "Default values doesn't support geom kind: '$geomKind'")
        return DEFAULTS[geomKind]!!
    }

    fun preferredSampling(): Sampling {
        return when (geomKind) {
            POINT -> DefaultSampling.POINT
            PATH -> DefaultSampling.PATH
            LINE -> DefaultSampling.LINE
            SMOOTH -> DefaultSampling.SMOOTH
            BAR -> DefaultSampling.BAR
            HISTOGRAM -> DefaultSampling.HISTOGRAM
            TILE -> DefaultSampling.TILE
            BIN_2D -> DefaultSampling.BIN_2D
            ERROR_BAR -> DefaultSampling.ERROR_BAR
            CROSS_BAR -> DefaultSampling.CROSS_BAR
            LINE_RANGE -> DefaultSampling.LINE_RANGE
            POINT_RANGE -> DefaultSampling.POINT_RANGE
            CONTOUR -> DefaultSampling.CONTOUR
            CONTOURF -> DefaultSampling.CONTOURF
            POLYGON -> DefaultSampling.POLYGON
            MAP -> DefaultSampling.MAP
            AB_LINE -> DefaultSampling.AB_LINE
            H_LINE -> DefaultSampling.H_LINE
            V_LINE -> DefaultSampling.V_LINE
            BOX_PLOT -> Samplings.NONE // DefaultSampling.BOX_PLOT
            RIBBON -> DefaultSampling.RIBBON
            AREA -> DefaultSampling.AREA
            DENSITY -> DefaultSampling.DENSITY
            DENSITY2D -> DefaultSampling.DENSITY2D
            DENSITY2DF -> DefaultSampling.DENSITY2DF
            JITTER -> DefaultSampling.JITTER
            FREQPOLY -> DefaultSampling.FREQPOLY
            STEP -> DefaultSampling.STEP
            RECT -> DefaultSampling.RECT
            SEGMENT -> DefaultSampling.SEGMENT
            TEXT -> DefaultSampling.TEXT

            LIVE_MAP,
            RASTER,
            IMAGE -> Samplings.NONE
        }
    }

    fun renders(): List> {
        return GeomMeta.renders(geomKind)
    }

    fun preferredPositionAdjustments(layerOptions: OptionsAccessor): PosProvider {
        return when (geomKind) {
            JITTER -> PosProvider.jitter(
                layerOptions.getDouble(Option.Geom.Jitter.WIDTH),
                layerOptions.getDouble(Option.Geom.Jitter.HEIGHT)
            )

            // Some other geoms has stateless position adjustments defined in `defaults`
            // Otherwise it's just `identity`
            else -> PosProvider.wrap(PositionAdjustments.identity())
        }
    }

    private companion object {
        private val DEFAULTS = HashMap>()
        private val COMMON = commonDefaults()

        init {
            for (geomKind in values()) {
                DEFAULTS[geomKind] = COMMON
            }

            DEFAULTS[SMOOTH] =
                smoothDefaults()
            DEFAULTS[BAR] =
                barDefaults()
            DEFAULTS[HISTOGRAM] =
                histogramDefaults()
            DEFAULTS[CONTOUR] =
                contourDefaults()
            DEFAULTS[CONTOURF] =
                contourfDefaults()
            DEFAULTS[CROSS_BAR] =
                crossBarDefaults()
            DEFAULTS[BOX_PLOT] =
                boxplotDefaults()
            DEFAULTS[AREA] =
                areaDefaults()
            DEFAULTS[DENSITY] =
                densityDefaults()
            DEFAULTS[DENSITY2D] =
                density2dDefaults()
            DEFAULTS[DENSITY2DF] =
                density2dfDefaults()
            DEFAULTS[FREQPOLY] =
                freqpolyDefaults()
            DEFAULTS[BIN_2D] =
                bin2dDefaults()
        }

        private fun commonDefaults(): Map {
            val defaults = HashMap()
            defaults["stat"] = "identity"
            return defaults
        }

        private fun smoothDefaults(): Map {
            val defaults = HashMap()
            defaults["stat"] = "smooth"
            return defaults
        }

        private fun barDefaults(): Map {
            val defaults = HashMap()
            defaults["stat"] = "count"
            defaults["position"] = "stack"
            return defaults
        }

        private fun histogramDefaults(): Map {
            val defaults = HashMap()
            defaults["stat"] = "bin"
            defaults["position"] = "stack"
            return defaults
        }

        private fun contourDefaults(): Map {
            val defaults = HashMap()
            defaults["stat"] = "contour"
            return defaults
        }

        private fun contourfDefaults(): Map {
            val defaults = HashMap()
            defaults["stat"] = "contourf"
            return defaults
        }


        private fun crossBarDefaults(): Map {
            val defaults = HashMap()
            defaults["stat"] = "identity"
            defaults["position"] = mapOf(Meta.NAME to "dodge", "width" to 0.95)
            return defaults
        }

        private fun boxplotDefaults(): Map {
            val defaults = HashMap()
            defaults["stat"] = "boxplot"
            defaults["position"] = mapOf(Meta.NAME to "dodge", "width" to 0.95)
            return defaults
        }

        private fun areaDefaults(): Map {
            val defaults = HashMap()
            defaults["stat"] = "identity"
            defaults["position"] = "stack"
            return defaults
        }

        private fun densityDefaults(): Map {
            val defaults = HashMap()
            defaults["stat"] = "density"
            return defaults
        }

        private fun density2dDefaults(): Map {
            val defaults = HashMap()
            defaults["stat"] = "density2d"
            return defaults
        }

        private fun density2dfDefaults(): Map {
            val defaults = HashMap()
            defaults["stat"] = "density2df"
            return defaults
        }

        private fun freqpolyDefaults(): Map {
            val defaults = HashMap()
            defaults["stat"] = "bin"
            return defaults
        }

        private fun bin2dDefaults(): Map {
            return mapOf(
                "stat" to "bin2d"
            )
        }
    }
}