commonMain.org.jetbrains.letsPlot.intern.layer.SamplingOptions.kt Maven / Gradle / Ivy

Go to download

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

Lets-Plot Kotlin API without dependencies.

There is a newer version: 4.9.2

Show newest version

/*
 * 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 org.jetbrains.letsPlot.intern.layer

import org.jetbrains.letsPlot.intern.Options
import org.jetbrains.letsPlot.intern.SamplingKind
import org.jetbrains.letsPlot.intern.filterNonNullValues

class SamplingOptions internal constructor(
    val kind: SamplingKind,
    private val options: Map
) {
    internal constructor(kind: SamplingKind, n: Int, seed: Int? = null, minSubsample: Int? = null) : this(
        kind,
        mapOf(
            "n" to n,
            "seed" to seed,
            "min_subsample" to minSubsample
        ).filterNonNullValues()
    )

    val isNone: Boolean = (kind == SamplingKind.NONE)

    val mapping: Options
        get() {
            check(!isNone) { "Not applicable to `none` sampling." }
            return Options(options) +
                    Options(mapOf("name" to kind.optionName()))
        }
}