BarGeomAndCountStat.kt

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

package frontendContextDemo.scripts

import frontendContextDemo.ScriptInBrowserContext
import org.jetbrains.letsPlot.geom.geomBar
import org.jetbrains.letsPlot.ggplot
import org.jetbrains.letsPlot.stat.statCount

object BarGeomAndCountStat {
    @JvmStatic
    @Suppress("DuplicatedCode")
    fun main(args: Array<String>) {
        ScriptInBrowserContext.eval("'geom_bar()' == 'stat_count'") {
            val data = mapOf<String, Any>(
                "cat1" to listOf("a", "a", "b", "a", "a", "a", "a", "b", "b", "b", "b"),
                "cat2" to listOf("c", "c", "d", "d", "d", "c", "c", "d", "c", "c", "d")
            )
            val p = ggplot(data)

            // bar (with count stat by default)
            val barLayer = geomBar {
                x = "cat1"
                fill = "cat2"
            }

            // count stat (with bar geom by default)
            val countLayer = statCount {
                x = "cat1"
                fill = "cat2"
            }

            // show two identical plots
            (p + barLayer).show()
            (p + countLayer).show()
        }
    }
}